feat: include AGENTS.md in worker prompt

1 files changed, 14 insertions, 1 deletions

diff --git a/Omni/Agent/Worker.hs b/Omni/Agent/Worker.hs
index 3504765..a861173 100644
--- a/Omni/Agent/Worker.hs
+++ b/Omni/Agent/Worker.hs
@@ -151,6 +151,19 @@ runAmp repo task = do
   Directory.createDirectoryIfMissing True (repo </> "_/llm")
   let logFile = repo </> "_/llm/amp.log"
 
+  -- Read AGENTS.md
+  agentsMd <-
+    fmap (fromMaybe "") <| do
+      exists <- Directory.doesFileExist (repo </> "AGENTS.md")
+      if exists
+        then Just </ readFile (repo </> "AGENTS.md")
+        else pure Nothing
+
+  let fullPrompt =
+        prompt
+          <> "\n\nREPOSITORY GUIDELINES (AGENTS.md):\n"
+          <> agentsMd
+
   -- Clean up previous log
   exists <- Directory.doesFileExist logFile
   when exists (Directory.removeFile logFile)
@@ -160,7 +173,7 @@ runAmp repo task = do
   tidLog <- forkIO (monitorLog logFile)
 
   -- Assume amp is in PATH
-  let args = ["--log-level", "debug", "--log-file", "_/llm/amp.log", "--dangerously-allow-all", "-x", Text.unpack prompt]
+  let args = ["--log-level", "debug", "--log-file", "_/llm/amp.log", "--dangerously-allow-all", "-x", Text.unpack fullPrompt]
 
   let cp = (Process.proc "amp" args) {Process.cwd = Just repo}
   (_, _, _, ph) <- Process.createProcess cp