fix bugs in agent selection

ceylonai · Aug 16, 2024 · c390ffc · c390ffc
1 parent 0ca85a7
commit c390ffc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bindings/ceylon/ceylon/llm/llm_task_coordinator.py b/bindings/ceylon/ceylon/llm/llm_task_coordinator.py
@@ -100,7 +100,7 @@ async def get_best_agent_for_subtask(self, subtask: SubTask) -> str:
         runnable = prompt_template | self.llm | StrOutputParser()
 
         def get_valid_agent_name(max_attempts=3):
-            agent_names = [agent.details().name for agent in self.agents]
+            agent_names = [agent.details().name for agent in self.get_llm_operators]
 
             for attempt in range(max_attempts):
                 response = runnable.invoke({