1 files changed, 49 insertions, 7 deletions

diff --git a/Omni/Agent/Worker.hs b/Omni/Agent/Worker.hs
index 01099a0..a29feb4 100644
--- a/Omni/Agent/Worker.hs
+++ b/Omni/Agent/Worker.hs
@@ -1,11 +1,11 @@
 {-# LANGUAGE OverloadedStrings #-}
 {-# LANGUAGE NoImplicitPrelude #-}
 
--- : out omni-agent-worker
 module Omni.Agent.Worker where
 
 import Alpha
 import qualified Data.Text as Text
+import qualified Data.Text.IO as TIO
 import qualified Omni.Agent.Core as Core
 import qualified Omni.Agent.Git as Git
 import qualified Omni.Agent.Log as AgentLog
@@ -13,6 +13,7 @@ import qualified Omni.Task.Core as TaskCore
 import qualified System.Directory as Directory
 import qualified System.Exit as Exit
 import System.FilePath ((</>))
+import qualified System.IO as IO
 import qualified System.Process as Process
 
 start :: Core.Worker -> IO ()
@@ -58,7 +59,7 @@ processTask worker task = do
   AgentLog.updateActivity ("Claiming task " <> tid)
 
   -- Claim task
-  TaskCore.updateTaskStatus tid TaskCore.InProgress
+  TaskCore.updateTaskStatus tid TaskCore.InProgress []
 
   -- Commit claim locally
   Git.commit repo ("task: claim " <> tid)
@@ -94,7 +95,7 @@ processTask worker task = do
       AgentLog.log "Agent finished successfully"
 
       -- Update status to Review (bundled with feature commit)
-      TaskCore.updateTaskStatus tid TaskCore.Review
+      TaskCore.updateTaskStatus tid TaskCore.Review []
 
       -- Commit changes
       -- We should check if there are changes, but 'git add .' is safe.
@@ -111,12 +112,11 @@ processTask worker task = do
       Git.syncWithLive repo
 
       -- Update status to Review (for signaling)
-      TaskCore.updateTaskStatus tid TaskCore.Review
+      TaskCore.updateTaskStatus tid TaskCore.Review []
       Git.commit repo ("task: review " <> tid)
-      
+
       AgentLog.log ("[✓] Task " <> tid <> " completed")
       AgentLog.update (\s -> s {AgentLog.statusTask = Nothing})
-      
     Exit.ExitFailure code -> do
       AgentLog.log ("Agent failed with code " <> tshow code)
       AgentLog.updateActivity "Agent failed, retrying..."
@@ -143,6 +143,12 @@ runAmp repo task = do
           <> fromMaybe "root" (TaskCore.taskNamespace task)
           <> "'.\n"
 
+  let logFile = repo </> "_/llm/amp.log"
+
+  -- Remove old log file
+  exists <- Directory.doesFileExist logFile
+  when exists (Directory.removeFile logFile)
+
   Directory.createDirectoryIfMissing True (repo </> "_/llm")
 
   -- Assume amp is in PATH
@@ -150,7 +156,12 @@ runAmp repo task = do
 
   let cp = (Process.proc "amp" args) {Process.cwd = Just repo}
   (_, _, _, ph) <- Process.createProcess cp
-  Process.waitForProcess ph
+
+  tid <- forkIO <| monitorLog logFile ph
+
+  exitCode <- Process.waitForProcess ph
+  killThread tid
+  pure exitCode
 
 formatTask :: TaskCore.Task -> Text
 formatTask t =
@@ -182,6 +193,37 @@ formatTask t =
   where
     formatDep dep = "  - " <> TaskCore.depId dep <> " [" <> Text.pack (show (TaskCore.depType dep)) <> "]"
 
+monitorLog :: FilePath -> Process.ProcessHandle -> IO ()
+monitorLog path ph = do
+  waitForFile path
+  IO.withFile path IO.ReadMode <| \h -> do
+    IO.hSetBuffering h IO.LineBuffering
+    go h
+  where
+    go h = do
+      eof <- IO.hIsEOF h
+      if eof
+        then do
+          mExit <- Process.getProcessExitCode ph
+          case mExit of
+            Nothing -> do
+              threadDelay 100000 -- 0.1s
+              go h
+            Just _ -> pure ()
+        else do
+          line <- TIO.hGetLine h
+          AgentLog.processLogLine line
+          go h
+
+waitForFile :: FilePath -> IO ()
+waitForFile path = do
+  exists <- Directory.doesFileExist path
+  if exists
+    then pure ()
+    else do
+      threadDelay 100000
+      waitForFile path
+
 findBaseBranch :: FilePath -> TaskCore.Task -> IO Text
 findBaseBranch repo task = do
   let deps = TaskCore.taskDependencies task