Skip to content

Commit

Permalink
[CLEANUP][Model Routews]
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 b258872 commit 9fe44a9
Showing 1 changed file with 8 additions and 10 deletions.
18 changes: 8 additions & 10 deletions servers/agent/api.py
Original file line number Diff line number Diff line change
Expand Up @@ -107,16 +107,14 @@ def model_router(model_name: str):
)


# @app.get("/v1/models", response_model=ModelList)
# async def list_models():
# """
# An endpoint to list available models. It returns a list of model cards.
# This is useful for clients to query and understand what models are available for use.
# """
# model_card = ModelCard(
# id="cogvlm-chat-17b"
# ) # can be replaced by your model id like cogagent-chat-18b
# return ModelList(data=[model_card])
@app.get("/v1/models", response_model=List[str])
async def list_models():
"""
An endpoint to list available models. It returns a list of model names.
This is useful for clients to query and understand what models are available for use.
"""
model_names = ["OpenAIChat", "GPT4o", "GPT4VisionAPI", "Anthropic"]
return model_names


@app.post("v1/agent/completions", response_model=AgentOutput)
Expand Down

0 comments on commit 9fe44a9

Please sign in to comment.