Merge pull request #242 from Pythagora-io/stages_upgrade

Upgraded stages that are sent to the extension

core/agents/frontend.py

@@ -76,6 +76,7 @@ class Frontend(FileDiffMixin, GitMixin, BaseAgent):
         )
         self.next_state.action = FE_START
         await self.send_message("## Building the frontend\n\nThis may take a couple of minutes.")
+        await self.ui.send_project_stage({"stage": ProjectStage.FRONTEND_STARTED})
 
         await self.ui.set_important_stream(False)
         llm = self.get_llm(FRONTEND_AGENT_NAME, stream_output=True)
@@ -162,7 +163,7 @@ class Frontend(FileDiffMixin, GitMixin, BaseAgent):
         # update the pages in the knowledge base
         await self.state_manager.update_implemented_pages_and_apis()
 
-        await self.ui.send_project_stage({"stage": ProjectStage.ITERATE_FRONTEND})
+        await self.ui.send_project_stage({"stage": ProjectStage.ITERATE_FRONTEND, "iteration_index": 1})
 
         if user_input:
             await self.send_message("Errors detected, fixing...")

core/agents/tech_lead.py

@@ -428,7 +428,7 @@ class TechLead(RelevantFilesMixin, BaseAgent):
         await self.ui.send_project_stage(
             {
                 "stage": ProjectStage.STARTING_TASK,
-                "task_index": 1,
+                "task_index": 0,
             }
         )
         await self.ui.send_front_logs_headers(
@@ -443,7 +443,7 @@ class TechLead(RelevantFilesMixin, BaseAgent):
                 {
                     "title": self.next_state.tasks[0]["description"],
                     "project_state_id": str(self.next_state.id),
-                    "labels": ["E3 / T1", "working"],
+                    "labels": ["E3 / T1", "Backend", "working"],
                 }
             ]
         )

core/ui/base.py

@@ -7,11 +7,16 @@ from pydantic import BaseModel
 class ProjectStage(str, Enum):
     PROJECT_NAME = "project_name"
     PROJECT_DESCRIPTION = "project_description"
+    SPECS_STARTED = "specs_started"
+    SPECS_FINISHED = "specs_finished"
+    FRONTEND_STARTED = "frontend_started"
+    FRONTEND_FINISHED = "frontend_finished"
     CONTINUE_FRONTEND = "continue_frontend"
     ITERATE_FRONTEND = "iterate_frontend"
     GET_USER_FEEDBACK = "get_user_feedback"
     OPEN_PLAN = "open_plan"
     STARTING_TASK = "starting_task"
+    TASK_FINISHED = "task_finished"
     BREAKDOWN_CHAT = "breakdown_chat"
     TEST_APP = "test_app"
     ADDITIONAL_FEEDBACK = "additional_feedback"

core/ui/ipc_client.py

@@ -25,6 +25,7 @@ class MessageType(str, Enum):
     RESPONSE = "response"
     USER_INPUT_REQUEST = "user_input_request"
     INFO = "info"
+    STAGE = "stage"
     PROGRESS = "progress"
     DEBUGGING_LOGS = "debugging_logs"
     RUN_COMMAND = "run_command"
@@ -393,7 +394,7 @@ class IPCClientUI(UIBase):
         return UserInput(cancelled=True, access_token=access_token)
 
     async def send_project_stage(self, data: dict):
-        await self._send(MessageType.INFO, content=json.dumps(data))
+        await self._send(MessageType.STAGE, content=json.dumps(data))
 
     async def send_epics_and_tasks(
         self,