From 908773aafd35aa754362fc50e895f79f094877c7 Mon Sep 17 00:00:00 2001
From: Kye Gomez <swarms_wd@Kyes-MacBook-Pro.local>
Date: Fri, 26 Jul 2024 20:50:21 -0400
Subject: [PATCH] [CLEANUP]

---
 servers/agent/api.py | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/servers/agent/api.py b/servers/agent/api.py
index 7fa0873..f532190 100644
--- a/servers/agent/api.py
+++ b/servers/agent/api.py
@@ -154,18 +154,8 @@ async def agent_completions(agent_input: AgentInput):
 
         # Model check
         model_name = agent_input.model_name
-        if model_name not in AVAILABLE_MODELS:
-            raise HTTPException(
-                status_code=400, detail=f"Invalid model name: {model_name}"
-            )
-
         model = await model_router(model_name)
-
-        # Task check
-        task = agent_input.task
-        if task not in agent_input:
-            raise HTTPException(status_code=400, detail="Task not provided")
-
+        
         # Initialize the agent
         agent = Agent(
             agent_name=agent_name,
@@ -191,7 +181,8 @@ async def agent_completions(agent_input: AgentInput):
         logger.info(f"Running agent with task: {task}")
         agent_history = agent.short_memory.return_history_as_string()
         completions = agent.run(task)
-        logger.info(f"Completions: {completions}")
+        
+        logger.info(f"Agent response: {completions}")
 
         # Costs calculation
         all_input_tokens = await count_tokens(agent_history)
@@ -209,7 +200,7 @@ async def agent_completions(agent_input: AgentInput):
                     {
                         "index": 0,
                         "message": {
-                            "role": "system",
+                            "role": "assistant",
                             "content": completions,
                             "name": agent_name,
                         },