Merge pull request #119 from Fr4nc3/main

GroupChatManager Rewrite

Author: Fr4nc3

src/backend/agents/base_agent.py

@@ -21,7 +21,7 @@
     Step,
     StepStatus,
 )
-from models.messages_kernel import AgentType
+
 from src.backend.event_utils import track_event_if_configured
 
 
@@ -77,7 +77,7 @@ async def handle_action_request(
                 AssistantMessage(content=message.action, source="GroupChatManager"),
                 UserMessage(
                     content=f"{step.human_feedback}. Now make the function call",
-                    source=AgentType.HUMAN.value,
+                    source="HumanAgent",
                 ),
             ]
         )

src/backend/agents/group_chat_manager.py

@@ -21,7 +21,7 @@
     Step,
     StepStatus,
 )
-from models.messages_kernel import AgentType
+
 from src.backend.event_utils import track_event_if_configured
 
 
@@ -57,7 +57,7 @@ async def handle_input_task(
                 user_id=self._user_id,
                 plan_id="",
                 content=f"{message.description}",
-                source=AgentType.HUMAN.value,
+                source="HumanAgent",
                 step_id="",
             )
         )
@@ -68,7 +68,7 @@ async def handle_input_task(
                 "session_id": message.session_id,
                 "user_id": self._user_id,
                 "content": message.description,
-                "source": AgentType.HUMAN.value,
+                "source": "HumanAgent",
             },
         )
 

src/backend/agents/human.py

@@ -13,7 +13,6 @@
     Step,
 )
 from src.backend.event_utils import track_event_if_configured
-from models.messages_kernel import AgentType
 
 
 @default_subscription
@@ -53,7 +52,7 @@ async def handle_step_feedback(
                 user_id=self.user_id,
                 plan_id=step.plan_id,
                 content=f"Received feedback for step: {step.action}",
-                source=AgentType.HUMAN.value,
+                source="HumanAgent",
                 step_id=message.step_id,
             )
         )
@@ -65,7 +64,7 @@ async def handle_step_feedback(
                 "user_id": self.user_id,
                 "plan_id": step.plan_id,
                 "content": f"Received feedback for step: {step.action}",
-                "source": AgentType.HUMAN.value,
+                "source": "HumanAgent",
                 "step_id": message.step_id,
             },
         )

src/backend/agents/planner.py

@@ -25,7 +25,7 @@
     StepStatus,
     HumanFeedbackStatus,
 )
-from models.messages_kernel import AgentType
+
 from src.backend.event_utils import track_event_if_configured
 
 
@@ -133,7 +133,7 @@ async def handle_plan_clarification(
                 user_id=self._user_id,
                 plan_id="",
                 content=f"{message.human_clarification}",
-                source=AgentType.HUMAN.value,
+                source="HumanAgent",
                 step_id="",
             )
         )
@@ -144,7 +144,7 @@ async def handle_plan_clarification(
                 "session_id": message.session_id,
                 "user_id": self._user_id,
                 "content": f"{message.human_clarification}",
-                "source": AgentType.HUMAN.value,
+                "source": "HumanAgent",
             },
         )