fix: classifier shouldn't select custom agents

potpie-ai · Feb 14, 2025 · c415165 · c415165
1 parent 6ef6125
commit c415165
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/app/modules/conversations/conversation/conversation_service.py b/app/modules/conversations/conversation/conversation_service.py
@@ -131,8 +131,12 @@ async def initialize(self, user_id: str):
         - Overlapping domains, choose more specialized: `choose_higher_expertise_agent|0.80`
         """
 
+        # We'll only use system agents in classifier
+        system_agents = [
+            agent for agent in self.available_agents if agent.status == "SYSTEM"
+        ]
         self.agent_descriptions = "\n".join(
-            [f"- {agent.id}: {agent.description}" for agent in self.available_agents]
+            [f"- {agent.id}: {agent.description}" for agent in system_agents]
         )
 
     class State(TypedDict):