feat: unified models integration

This commit is contained in:
Timothy J. Baek committed 2024-05-24 03:02:56 -07:00
1 parent e80e4c304a
commit 468c6398cd
9 files changed
+94 -92

No files matched your search

-2
View File
@@ -276,13 +276,11 @@ async def get_models(user=Depends(get_verified_user)):
if app.state.config.ENABLE_OPENAI_API:
openai_models = await get_openai_models()
openai_app.state.MODELS = openai_models
openai_models = openai_models["data"]
if app.state.config.ENABLE_OLLAMA_API:
ollama_models = await get_ollama_models()
ollama_app.state.MODELS = ollama_models
print(ollama_models)