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="gemini-1.5-flash")
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-1.5-flash",
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-1.5-flash",
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-1.5-flash"
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