Merge pull request Pythagora-io#32 from Pythagora-io/remove-double-me…

…ssages Fixed various double messages in UI
abdulrahman305 · Sep 25, 2024 · d3c1026 · d3c1026
2 parents 0638e6a + f3a066e
commit d3c1026
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/core/agents/architect.py b/core/agents/architect.py
@@ -126,7 +126,7 @@ async def select_templates(self, spec: Specification) -> tuple[str, dict[Project
         """
         await self.send_message("Selecting starter templates ...")
 
-        llm = self.get_llm(stream_output=True)
+        llm = self.get_llm()
         convo = (
             AgentConvo(self)
             .template(

diff --git a/core/agents/bug_hunter.py b/core/agents/bug_hunter.py
@@ -88,7 +88,7 @@ async def check_logs(self, logs_message: str = None):
             )
             .require_schema(HuntConclusionOptions)
         )
-        llm = self.get_llm(stream_output=True)
+        llm = self.get_llm()
         hunt_conclusion = await llm(convo, parser=JSONParser(HuntConclusionOptions), temperature=0)
 
         bug_hunting_cycles = self.current_state.current_iteration.get("bug_hunting_cycles")

diff --git a/core/agents/developer.py b/core/agents/developer.py
@@ -106,7 +106,6 @@ async def breakdown_current_iteration(self) -> AgentResponse:
             source = "bug_hunt"
             n_tasks = len(self.next_state.iterations)
             log.debug(f"Breaking down the logging cycle {description}")
-            await self.send_message("Breaking down the current bug hunting cycle ...")
         else:
             iteration = self.current_state.current_iteration
             if iteration is None:
@@ -118,11 +117,11 @@ async def breakdown_current_iteration(self) -> AgentResponse:
             source = "troubleshooting"
             n_tasks = len(self.next_state.iterations)
             log.debug(f"Breaking down the iteration {description}")
-            await self.send_message("Breaking down the current task iteration ...")
 
         if self.current_state.files and self.current_state.relevant_files is None:
             return await self.get_relevant_files(user_feedback, description)
 
+        await self.send_message("Breaking down the task into steps ...")
         await self.ui.send_task_progress(
             n_tasks,  # iterations and reviews can be created only one at a time, so we are always on last one
             n_tasks,
@@ -193,7 +192,6 @@ async def breakdown_current_task(self) -> AgentResponse:
         )
 
         log.debug(f"Breaking down the current task: {current_task['description']}")
-        await self.send_message("Thinking about how to implement this task ...")
 
         log.debug(f"Current state files: {len(self.current_state.files)}, relevant {self.current_state.relevant_files}")
         # Check which files are relevant to the current task
@@ -202,6 +200,8 @@ async def breakdown_current_task(self) -> AgentResponse:
 
         current_task_index = self.current_state.tasks.index(current_task)
 
+        await self.send_message("Thinking about how to implement this task ...")
+
         llm = self.get_llm(TASK_BREAKDOWN_AGENT_NAME, stream_output=True)
         convo = AgentConvo(self).template(
             "breakdown",

diff --git a/core/agents/spec_writer.py b/core/agents/spec_writer.py
@@ -120,7 +120,7 @@ async def update_spec(self, iteration_mode) -> AgentResponse:
 
     async def check_prompt_complexity(self, prompt: str) -> str:
         await self.send_message("Checking the complexity of the prompt ...")
-        llm = self.get_llm(SPEC_WRITER_AGENT_NAME, stream_output=True)
+        llm = self.get_llm(SPEC_WRITER_AGENT_NAME)
         convo = AgentConvo(self).template("prompt_complexity", prompt=prompt)
         llm_response: str = await llm(convo, temperature=0, parser=StringParser())
         return llm_response.lower()