From 3473ebf7f52083cbcf5b779e3c9d709d1174178e Mon Sep 17 00:00:00 2001 From: jsbxyyx Date: Thu, 11 Dec 2025 10:24:49 +0800 Subject: [PATCH] Refactor gemini provider configuration Updated base URL retrieval for 'gemini' provider and adjusted configuration logic. --- app/services/llm.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/app/services/llm.py b/app/services/llm.py index 6c954a8..cbddcf9 100644 --- a/app/services/llm.py +++ b/app/services/llm.py @@ -58,7 +58,7 @@ def _generate_response(prompt: str) -> str: elif llm_provider == "gemini": api_key = config.app.get("gemini_api_key") model_name = config.app.get("gemini_model_name") - base_url = "***" + base_url = config.app.get("gemini_base_url", "") elif llm_provider == "qwen": api_key = config.app.get("qwen_api_key") model_name = config.app.get("qwen_model_name") @@ -167,7 +167,10 @@ def _generate_response(prompt: str) -> str: if llm_provider == "gemini": import google.generativeai as genai - genai.configure(api_key=api_key, transport="rest") + if not base_url: + genai.configure(api_key=api_key, transport="rest") + else: + genai.configure(api_key=api_key, transport="rest", client_options={'api_endpoint': base_url}) generation_config = { "temperature": 0.5, @@ -441,4 +444,4 @@ if __name__ == "__main__": ) print("######################") print(search_terms) - \ No newline at end of file +