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.