Upload 2 files

main.py

@@ -68,6 +68,8 @@ from pymongo import MongoClient
 from RyuzakiLib.hackertools.chatgpt import RendyDevChat
 from RyuzakiLib.hackertools.gemini import GeminiLatest
 from RyuzakiLib.hackertools.openai_api import OpenAiToken
+from RyuzakiLib.hackertools.huggingface import BetaRag
+
 from RyuzakiLib.mental import BadWordsList
 from RyuzakiLib.system import OpenReadSystem
 
@@ -844,6 +846,23 @@ def v1beta3_google_ai(
     except:
         return SuccessResponse(status="False", randydev={"message": internal_error})
 
+@app.post("/ryuzaki/beta-rag", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
+def beta_rag(item: BetaRags):
+    try:
+        rag = BetaRag(
+            token=HUGGING_TOKEN,
+            user_id=item.user_id,
+            mongo_url=item.mongo_url
+        )
+        cybersecuritydb(item.user_id, item.mongo_url)
+        answer = rag.rag_chat(item.query)
+        return SuccessResponse(
+            status="True",
+                randydev={"message": answer}
+        )
+    except Exception:
+        return SuccessResponse(status="False", randydev={"message": contact_support})
+
 @app.post("/ryuzaki/gemini-ai-pro", response_model=SuccessResponse, responses={422: {"model": ErrorStatus}})
 def gemini_pro(item: GeminiPro):
     owner_base = f"""
@@ -1438,7 +1457,7 @@ def custom_openapi():
         return app.openapi_schema
     openapi_schema = get_openapi(
         title="RyuzakiLib API",
-        version="2.3.1",
+        version="3.3.0",
         summary="Use It Only For Personal Project Else I Need To Delete The Api",
         description=description,
         routes=app.routes,

models.py

@@ -69,6 +69,11 @@ class ChatgptModel(BaseModel):
 class ChatgptCustom(BaseModel):
     query: str
 
+class BetaRags(BaseModel):
+    query: str
+    mongo_url: Optional[str] = None
+    user_id: Optional[int] = None
+
 class GeminiPro(BaseModel):
     query: str
     bard_api_key: Optional[str] = None