Upload 2 files

main.py

@@ -847,16 +847,16 @@ def v1beta3_google_ai(
 @app.post("/ryuzaki/beta-rag", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def beta_rag(item: BetaRags):
     try:
-        rag = BetaRag(
-            token=HUGGING_TOKEN,
+        x = BetaRag(
+            token=str(HUGGING_TOKEN),
             user_id=item.user_id,
-            mongo_url=item.mongo_url
+            mongo_url=MONGO_URL
         )
         cybersecuritydb(item.user_id, item.mongo_url)
-        answer = rag.rag_chat(item.query)
+        response = x.rag_chat(item.query)
         return SuccessResponse(
             status="True",
-                randydev={"message": answer}
+                randydev={"message": response}
         )
     except Exception as e:
         return SuccessResponse(status="False", randydev={"message": str(e)})

models.py

@@ -71,7 +71,6 @@ class ChatgptCustom(BaseModel):
 
 class BetaRags(BaseModel):
     query: str
-    mongo_url: Optional[str] = None
     user_id: Optional[int] = None
 
 class GeminiPro(BaseModel):