Merge pull request #98 from gregdrizz/auto_agent_choosing

Auto agent choosing
assafelovic · Jul 25, 2023 · f9b59b6 · f9b59b6
2 parents 5d0a911 + 6ebb884
commit f9b59b6
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/agent/llm_utils.py b/agent/llm_utils.py
@@ -151,6 +151,7 @@ def choose_agent_configuration():
                                  "Academic Research Agent", "Computer Security Analyst Agent"
                                  
                                  if an agent for the field required doesn't exist make one up
+                                 fit an emoji to every agent before the agent name
                             """,
                     },
                     "instructions": {

diff --git a/main.py b/main.py
@@ -55,7 +55,7 @@ async def websocket_endpoint(websocket: WebSocket):
                 else:
                     agent_role_prompt = None
 
-                await websocket.send_json({"type": "logs", "output": f"🕵️ Agent: {agent}"})
+                await websocket.send_json({"type": "logs", "output": f"{agent}"})
                 if task and report_type and agent:
                     await manager.start_streaming(task, report_type, agent, agent_role_prompt, websocket)
                 else: