coosld/MoneyPrinterTurbo
1
0
Fork 0
mirror of https://github.com/harry0703/MoneyPrinterTurbo.git synced 2026-02-21 16:37:21 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactor gemini provider configuration

Browse Source 
Updated base URL retrieval for 'gemini' provider and adjusted configuration logic.
This commit is contained in:
jsbxyyx 2025-12-11 10:24:49 +08:00 committed by GitHub
parent 6cb5f23487
commit 3473ebf7f5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
app/services/llm.py
View File

@ -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)