hzhang/MoneyPrinterTurbo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #248 from elf-mouse/main

Browse Source 
fix: response parsing bug for gemini
This commit is contained in:
Harry
2024-04-12 17:03:01 +08:00
committed by GitHub
parent a8b54415a5 1dbfcfadab
commit 9ab13a74a2
1 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
app/services/llm.py
View File

@@ -100,7 +100,7 @@ def _generate_response(prompt: str) -> str:
if llm_provider == "gemini": if llm_provider == "gemini":
import google.generativeai as genai import google.generativeai as genai
genai.configure(api_key=api_key) genai.configure(api_key=api_key, transport='rest')
generation_config = { generation_config = {
"temperature": 0.5, "temperature": 0.5,
@@ -132,10 +132,14 @@ def _generate_response(prompt: str) -> str:
generation_config=generation_config, generation_config=generation_config,
safety_settings=safety_settings) safety_settings=safety_settings)
convo = model.start_chat(history=[]) try:
response = model.generate_content(prompt)
candidates = response.candidates
generated_text = candidates[0].content.parts[0].text
except (AttributeError, IndexError) as e:
print("Gemini Error:", e)
convo.send_message(prompt) return generated_text
return convo.last.text
if llm_provider == "cloudflare": if llm_provider == "cloudflare":
import requests import requests