Skip to content

Commit

Permalink
[CLEANUP]
Browse files Browse the repository at this point in the history
  • Loading branch information
Kye Gomez authored and Kye Gomez committed Jun 26, 2024
1 parent adfd58f commit b258872
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 4 deletions.
4 changes: 2 additions & 2 deletions api.py
Original file line number Diff line number Diff line change
Expand Up @@ -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))

Expand Down
3 changes: 1 addition & 2 deletions servers/agent/api.py
Original file line number Diff line number Diff line change
@@ -1,4 +1,3 @@

import asyncio
import os
from typing import List
Expand Down Expand Up @@ -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.
Expand Down

0 comments on commit b258872

Please sign in to comment.