Fix: Switch to v1 API endpoint for gemini-pro

This commit is contained in:
Jarvis
2026-01-31 06:49:29 +00:00
parent 0e4118a932
commit 1fe2621d98

View File

@@ -27,8 +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.")
# Fallback to the stable gemini-pro model to ensure functionality
url = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:generateContent?key={api_key}"
# Fallback to v1 (stable) and gemini-pro
url = f"https://generativelanguage.googleapis.com/v1/models/gemini-pro:generateContent?key={api_key}"
headers = {'Content-Type': 'application/json'}
payload = {
"contents": [{