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

Merge pull request #809 from jsbxyyx/main

Browse Source 
Refactor gemini provider configuration
This commit is contained in:
Harry 2025-12-14 11:49:15 +08:00 committed by GitHub
commit f8f10669cb
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)