Fix: Switch to v1 API endpoint for gemini-pro
This commit is contained in:
@@ -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": [{
|
||||
|
||||
Reference in New Issue
Block a user