Upload main.py

main.py

@@ -554,8 +554,6 @@ def v1beta3_google_ai(query: str, api_key: None = Depends(validate_api_key)):
 @app.post("/ryuzaki/v1beta2-google-ai")
 def v1beta2_google_ai(
     query: str,
-    temperature=0.5,
-    candidateCount=1,
     api_key: None = Depends(validate_api_key)
 ):
     url = SOURCE_ASSISTANT_GOOGLE_AI
@@ -565,12 +563,7 @@ def v1beta2_google_ai(
         headers = {"Content-Type": "application/json"}
         data = {
             "prompt": {
-                "messages": [
-                    {"content": query}
-                ]
-            },
-            "temperature": temperature,
-            "candidateCount": candidateCount
+                "messages": [{"content": query}]}
         }
         response = requests.post(api_url, headers=headers, json=data)
         response_str = response.json()