From b2588720a50a090f43e53be4d164f7dd7b3b6d99 Mon Sep 17 00:00:00 2001
From: Kye Gomez <swarms_wd@Kyes-MacBook-Pro.local>
Date: Wed, 26 Jun 2024 11:42:59 -0700
Subject: [PATCH] [CLEANUP]

---
 api.py               | 4 ++--
 servers/agent/api.py | 3 +--
 2 files changed, 3 insertions(+), 4 deletions(-)

diff --git a/api.py b/api.py
index 832ea05..9d0f6b5 100644
--- a/api.py
+++ b/api.py
@@ -177,9 +177,9 @@ async def agent_completions(agent_input: AgentInput):
                 ),
             ),
         )
-        
+
         return out.json()
-        
+
     except Exception as e:
         raise HTTPException(status_code=400, detail=str(e))
 
diff --git a/servers/agent/api.py b/servers/agent/api.py
index 1baf60d..199c1cb 100644
--- a/servers/agent/api.py
+++ b/servers/agent/api.py
@@ -1,4 +1,3 @@
-
 import asyncio
 import os
 from typing import List
@@ -61,7 +60,7 @@ def count_tokens(
         raise HTTPException(status_code=400, detail=str(e))
 
 
-async def model_router(model_name: str):
+def model_router(model_name: str):
     """
     Function to switch to the specified model.