Spaces:
Running
Running
Girish Jeswani commited on
Commit ·
87fd07d
1
Parent(s): 21e8a1d
update gemini model
Browse files
multi_llm_chatbot_backend/app/api/routes.py
CHANGED
|
@@ -22,7 +22,7 @@ def create_llm_client(provider: str = None) -> LLMClient:
|
|
| 22 |
|
| 23 |
if provider == "gemini":
|
| 24 |
try:
|
| 25 |
-
return GeminiClient(model_name=
|
| 26 |
except ValueError as e:
|
| 27 |
# Fallback to Ollama if Gemini API key is not available
|
| 28 |
print(f"Gemini API key not found, falling back to Ollama: {e}")
|
|
@@ -170,7 +170,7 @@ async def get_current_provider():
|
|
| 170 |
"current_provider": current_provider,
|
| 171 |
"available_providers": available_providers,
|
| 172 |
"model_info": {
|
| 173 |
-
"name": llm.model_name if hasattr(llm, 'model_name') else "gemini-
|
| 174 |
"provider": current_provider
|
| 175 |
}
|
| 176 |
}
|
|
@@ -206,7 +206,7 @@ async def switch_provider(provider_data: ProviderSwitch):
|
|
| 206 |
"message": f"Successfully switched to {current_provider}",
|
| 207 |
"current_provider": current_provider,
|
| 208 |
"model_info": {
|
| 209 |
-
"name": new_llm.model_name if hasattr(new_llm, 'model_name') else "gemini-
|
| 210 |
"provider": current_provider
|
| 211 |
}
|
| 212 |
}
|
|
@@ -387,7 +387,7 @@ async def switch_model(model_name: str = Body(...)):
|
|
| 387 |
@router.get("/current-model")
|
| 388 |
async def get_current_model():
|
| 389 |
# For backward compatibility
|
| 390 |
-
model_name = llm.model_name if hasattr(llm, 'model_name') else "gemini-
|
| 391 |
return {
|
| 392 |
"model": model_name,
|
| 393 |
"provider": current_provider
|
|
|
|
| 22 |
|
| 23 |
if provider == "gemini":
|
| 24 |
try:
|
| 25 |
+
return GeminiClient(model_name = os.getenv("GEMINI_MODEL"))
|
| 26 |
except ValueError as e:
|
| 27 |
# Fallback to Ollama if Gemini API key is not available
|
| 28 |
print(f"Gemini API key not found, falling back to Ollama: {e}")
|
|
|
|
| 170 |
"current_provider": current_provider,
|
| 171 |
"available_providers": available_providers,
|
| 172 |
"model_info": {
|
| 173 |
+
"name": llm.model_name if hasattr(llm, 'model_name') else "gemini-2.0-flash",
|
| 174 |
"provider": current_provider
|
| 175 |
}
|
| 176 |
}
|
|
|
|
| 206 |
"message": f"Successfully switched to {current_provider}",
|
| 207 |
"current_provider": current_provider,
|
| 208 |
"model_info": {
|
| 209 |
+
"name": new_llm.model_name if hasattr(new_llm, 'model_name') else "gemini-2.0-flash",
|
| 210 |
"provider": current_provider
|
| 211 |
}
|
| 212 |
}
|
|
|
|
| 387 |
@router.get("/current-model")
|
| 388 |
async def get_current_model():
|
| 389 |
# For backward compatibility
|
| 390 |
+
model_name = llm.model_name if hasattr(llm, 'model_name') else "gemini-2.0-flash"
|
| 391 |
return {
|
| 392 |
"model": model_name,
|
| 393 |
"provider": current_provider
|