Upload main.py

main.py

@@ -526,7 +526,7 @@ def free_chatgpt4_bing(query: str):
         return {"status": "false", "message": "Error response."}
 
 @app.post("/ryuzaki/google-ai")
-def google_ai(query: str, api_key: None = Depends(validate_api_key)):
+def v1beta3_google_ai(query: str, api_key: None = Depends(validate_api_key)):
     url = SOURCE_ASSISTANT_GOOGLE_AI
     token = ASSISTANT_GOOGLE_API_KEYS
     api_url = f"{SOURCE_ASSISTANT_GOOGLE_AI}/v1beta3/models/text-bison-001:generateText?key={ASSISTANT_GOOGLE_API_KEYS}"
@@ -551,6 +551,43 @@ def google_ai(query: str, api_key: None = Depends(validate_api_key)):
     except:
         return {"status": "false", "message": "Error response."}
 
+@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
+    token = ASSISTANT_GOOGLE_API_KEYS
+    api_url = f"{SOURCE_ASSISTANT_GOOGLE_AI}/v1beta2/models/text-bison-001:generateMessage?key={ASSISTANT_GOOGLE_API_KEYS}"
+    try:
+        headers = {"Content-Type": "application/json"}
+        data = {
+            "prompt": {
+                "messages": [
+                    {
+                        "content": query
+                    }
+                ]
+            },
+            "temperature": temperature,
+            "candidateCount": candidateCount
+        }
+        response = requests.post(api_url, headers=headers, json=data)
+        response_str = response.json()
+        answer = response_str["candidates"]
+        for results in answer:
+            message = results.get("content")
+        return {
+            "status": "true",
+            "randydev":{
+                "message": message
+            }
+        }
+    except:
+        return {"status": "false", "message": "Error response."}
+
 async def get_data(username):
     base_msg = ""
     async with AsyncClient() as gpx: