Update main.py

main.py

@@ -224,6 +224,35 @@ def ryuzaki_ai(
     except Exception as e:
         return {"status": "false", "message": f"error: {e}"}
 
+@app.post("/ryuzaki/google-ai", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def v1beta3_google_ai(
+    item: ChatgptCustom,
+    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}"
+    try:
+        headers = {"Content-Type": "application/json"}
+        data = {
+            "prompt": {
+                "text": item.query
+            }
+        }
+        response = requests.post(api_url, headers=headers, json=data)
+        response_str = response.json()
+        answer = response_str["candidates"]
+        for results in answer:
+            message = results.get("output")
+        return SuccessResponse(
+            status="True",
+            randydev={
+                "message": message
+            }
+        )
+    except:
+        return {"status": "false", "message": "Error response."}
+
 @app.get("/ryuzaki/unsplash")
 async def get_image_unsplash(query: str, size: str="500x500"):
     url = SOURCE_UNSPLASH_URL