bibibi12345 commited on
Commit
da6c071
·
1 Parent(s): 7cc3183

fixed model name

Browse files
Files changed (1) hide show
  1. app/routes/chat_api.py +1 -1
app/routes/chat_api.py CHANGED
@@ -270,7 +270,7 @@ async def chat_completions(fastapi_request: Request, request: OpenAIRequest, api
270
  # This means if `request.model` was "gemini-1.5-pro-search", `base_model_name` becomes "gemini-1.5-pro"
271
  # but the API call might need the full "gemini-1.5-pro-search".
272
  # Let's use `request.model` for the API call here, and `base_model_name` for checks like Express eligibility.
273
- return await execute_gemini_call(client_to_use, api_model_string, current_prompt_func, generation_config, request)
274
 
275
  except Exception as e:
276
  error_msg = f"Unexpected error in chat_completions endpoint: {str(e)}"
 
270
  # This means if `request.model` was "gemini-1.5-pro-search", `base_model_name` becomes "gemini-1.5-pro"
271
  # but the API call might need the full "gemini-1.5-pro-search".
272
  # Let's use `request.model` for the API call here, and `base_model_name` for checks like Express eligibility.
273
+ return await execute_gemini_call(client_to_use, base_model_name, current_prompt_func, generation_config, request)
274
 
275
  except Exception as e:
276
  error_msg = f"Unexpected error in chat_completions endpoint: {str(e)}"