Fix: Fallback to gemini-pro model for stability

This commit is contained in:
Jarvis
2026-01-31 06:46:37 +00:00
parent 608a71f14c
commit a764ae6ace

View File

@@ -27,7 +27,8 @@ def call_gemini_api(prompt: str, temperature: float = 0.7, retries: int = 3, tim
logging.error("GEMINI_API_KEY environment variable not set.")
raise ValueError("API key not found.")
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash-latest:generateContent?key={api_key}"
# Fallback to the stable gemini-pro model to ensure functionality
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={api_key}"
headers = {'Content-Type': 'application/json'}
payload = {
"contents": [{