Upload main.py

main.py

@@ -536,9 +536,7 @@ def google_ai(query: str, api_key: None = Depends(validate_api_key)):
                 "text": query
             }
         }
-        response = requests.post(f"{url}?key={token}", headers=headers, json=data)
-        if response.status_code != 200:
-            return {"message": f"error {response.status_code}"}
+        response = requests.post(f"{url}/v1beta3/models/text-bison-001:generateText?key={token}", headers=headers, json=data)
         response_str = response.json()
         answer = response_str["candidates"]
         for results in answer: