Update src/orchestrator.py

src/orchestrator.py

@@ -3,18 +3,14 @@ from typing import Dict, Any, Optional
 
 from .config import RunConfig
 from .agents import AnalyzerAgent, RefactorAgent, CriticAgent, AgentResult
-from .tasks import (
-    TaskContext,
-    build_analyzer_prompt,
-    build_refactor_prompt,
-    build_critic_prompt,
-)
+from .tasks import TaskContext, build_analyzer_prompt, build_refactor_prompt, build_critic_prompt
 
 
 @dataclass
 class RunTrace:
     task: str
     input_requirements: str
+    image_attached: bool
     analyzer: AgentResult
     refactor: AgentResult
     critic: AgentResult
@@ -23,6 +19,7 @@ class RunTrace:
         return {
             "task": self.task,
             "input_requirements": self.input_requirements,
+            "image_attached": self.image_attached,
             "analyzer": asdict(self.analyzer),
             "refactor": asdict(self.refactor),
             "critic": asdict(self.critic),
@@ -38,19 +35,22 @@ class Orchestrator:
         self.refactor = RefactorAgent(cfg.refactor)
         self.critic = CriticAgent(cfg.critic)
 
-    def run(self, requirements_text: str) -> RunTrace:
-        p1 = build_analyzer_prompt(requirements_text, self.ctx)
-        r1 = self.analyzer.run(p1)
+    def run(self, requirements_text: str, image_path: Optional[str] = None) -> RunTrace:
+        has_image = bool(image_path)
 
-        p2 = build_refactor_prompt(requirements_text, r1.output, self.ctx)
-        r2 = self.refactor.run(p2)
+        p1 = build_analyzer_prompt(requirements_text, has_image, self.ctx)
+        r1 = self.analyzer.run(p1, image_path=image_path)
 
-        p3 = build_critic_prompt(requirements_text, r2.output, self.ctx)
-        r3 = self.critic.run(p3)
+        p2 = build_refactor_prompt(requirements_text, r1.output, has_image, self.ctx)
+        r2 = self.refactor.run(p2, image_path=image_path)
+
+        p3 = build_critic_prompt(requirements_text, r2.output, has_image, self.ctx)
+        r3 = self.critic.run(p3, image_path=image_path)
 
         return RunTrace(
             task=self.ctx.name,
             input_requirements=requirements_text,
+            image_attached=has_image,
             analyzer=r1,
             refactor=r2,
             critic=r3,