github/OpenHands
1
0
Fork 0
mirror of https://github.com/OpenHands/OpenHands.git synced 2026-03-22 05:37:20 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix: add the community-maintained model list (#654)

Browse Source
This commit is contained in:
iFurySt
2024-04-05 11:21:17 +08:00
committed by GitHub
parent 12ddd973b5
commit dcaf51bc9e
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
opendevin/server/listen.py
View File

@@ -1,8 +1,8 @@
from opendevin.server.session import Session
from fastapi import FastAPI, WebSocket
from fastapi.middleware.cors import CORSMiddleware
import agenthub # noqa F401 (we import this to get the agents registered)
import litellm
import agenthub # noqa F401 (we import this to get the agents registered)
import litellm
from opendevin.agent import Agent
from opendevin import config
@@ -16,6 +16,7 @@ app.add_middleware(
allow_headers=["*"],
)
# This endpoint receives events from the client (i.e. the browser)
@app.websocket("/ws")
async def websocket_endpoint(websocket: WebSocket):
@@ -24,12 +25,14 @@ async def websocket_endpoint(websocket: WebSocket):
# TODO: should this use asyncio instead of await?
await session.start_listening()
@app.get("/litellm-models")
async def get_litellm_models():
"""
Get all models supported by LiteLLM.
"""
return litellm.model_list
return list(set(litellm.model_list + list(litellm.model_cost.keys())))
@app.get("/litellm-agents")
async def get_litellm_agents():