dopyDocument multi agent

documents/adapter/input/web/document_router.py

@@ -1,6 +1,10 @@
 from fastapi import APIRouter, UploadFile, File, Depends
 
+from documents.adapter.input.web.request.rag_query_request import RAGQueryRequest
+from documents.adapter.input.web.request.sentiment_request import SentimentRequest
 from documents.application.factory.analyze_document_usecase_factory import get_analyze_document_usecase
+from documents.application.factory.rag_query_usecase_factory import get_rag_query_usecase
+from documents.application.factory.sentiment_analysis_usecase_factory import get_sentiment_analysis_usecase
 from documents.application.factory.upload_file_usecase_factory import get_upload_document_usecase
 
 documentRouter = APIRouter()
@@ -19,4 +23,20 @@ async def analyze_document(
     usecase = Depends(get_analyze_document_usecase)
 ):
     result = await usecase.execute(document_id)
-    return {"result": result}
+    return {"result": result}
+
+@documentRouter.post("/rag/query")
+async def rag_query(
+    request: RAGQueryRequest,
+    usecase = Depends(get_rag_query_usecase)
+):
+    result = usecase.ask(request.query)
+    return {"result": result}
+
+@documentRouter.post("/sentiment")
+async def sentiment_analysis(
+    request: SentimentRequest,
+    usecase = Depends(get_sentiment_analysis_usecase)
+):
+    result = usecase.analyze(request.text)
+    return {"result": result}

documents/adapter/input/web/request/rag_query_request.py

@@ -0,0 +1,5 @@
+from pydantic import BaseModel
+
+class RAGQueryRequest(BaseModel):
+    query: str
+    top_k: int = 5

documents/adapter/input/web/request/sentiment_request.py

@@ -0,0 +1,4 @@
+from pydantic import BaseModel
+
+class SentimentRequest(BaseModel):
+    text: str