1 files changed, 36 insertions, 0 deletions
diff --git a/Omni/Agent/Worker.hs b/Omni/Agent/Worker.hs
index c07ed71..424a838 100644
--- a/Omni/Agent/Worker.hs
+++ b/Omni/Agent/Worker.hs
@@ -7,12 +7,14 @@ import Alpha
 import qualified Data.Aeson as Aeson
 import qualified Data.Aeson.Key as AesonKey
 import qualified Data.ByteString.Lazy as BSL
+import qualified Data.List as List
 import qualified Data.Text as Text
 import qualified Data.Text.Encoding as TE
 import qualified Data.Text.IO as TIO
 import qualified Data.Time
 import qualified Omni.Agent.Core as Core
 import qualified Omni.Agent.Log as AgentLog
+import qualified Omni.Fact as Fact
 import qualified Omni.Task.Core as TaskCore
 import qualified System.Directory as Directory
 import qualified System.Exit as Exit
@@ -269,10 +271,15 @@ runAmp repo task = do
         then Just </ readFile (repo </> "AGENTS.md")
         else pure Nothing
 
+  -- Get relevant facts from the knowledge base
+  relevantFacts <- getRelevantFacts task
+  let factsSection = formatFacts relevantFacts
+
   let fullPrompt =
         prompt
           <> "\n\nREPOSITORY GUIDELINES (AGENTS.md):\n"
           <> agentsMd
+          <> factsSection
 
   -- Remove old log file
   exists <- Directory.doesFileExist logFile
@@ -358,6 +365,35 @@ formatCommitMessage task ampOutput =
                   cleaned = [Text.take 72 ln | ln <- lns]
                in Text.intercalate "\n" cleaned
 
+-- | Get facts relevant to a task based on namespace/project
+getRelevantFacts :: TaskCore.Task -> IO [TaskCore.Fact]
+getRelevantFacts task = do
+  let namespace = fromMaybe "Omni" (TaskCore.taskNamespace task)
+  projectFacts <- Fact.getFactsByProject namespace
+  let sorted = List.sortBy (comparing (Down <. TaskCore.factConfidence)) projectFacts
+  pure (take 10 sorted)
+
+-- | Format facts for inclusion in the prompt
+formatFacts :: [TaskCore.Fact] -> Text
+formatFacts [] = ""
+formatFacts facts =
+  Text.unlines
+    [ "\n\nKNOWLEDGE BASE FACTS:",
+      "(These are learned patterns/conventions from previous work)",
+      ""
+    ]
+    <> Text.unlines (map formatFact facts)
+
+-- | Format a single fact for the prompt
+formatFact :: TaskCore.Fact -> Text
+formatFact f =
+  "- "
+    <> TaskCore.factContent f
+    <> ( if null (TaskCore.factRelatedFiles f)
+           then ""
+           else " [" <> Text.intercalate ", " (TaskCore.factRelatedFiles f) <> "]"
+       )
+
 monitorLog :: FilePath -> Process.ProcessHandle -> IO ()
 monitorLog path ph = do
   waitForFile path