Фиксация изменений
This commit is contained in:
@@ -1,5 +1,5 @@
|
||||
from fastapi import APIRouter
|
||||
from fastapi.responses import StreamingResponse
|
||||
from fastapi.responses import JSONResponse, StreamingResponse
|
||||
|
||||
from app.core.exceptions import AppError
|
||||
from app.modules.rag_session.embedding.gigachat_embedder import GigaChatEmbedder
|
||||
@@ -37,6 +37,7 @@ class RagModule:
|
||||
token_provider = GigaChatTokenProvider(settings)
|
||||
client = GigaChatClient(settings, token_provider)
|
||||
embedder = GigaChatEmbedder(client)
|
||||
self.embedder = embedder
|
||||
self.rag = RagService(embedder=embedder, repository=repository, chunker=TextChunker())
|
||||
self.sessions = RagSessionStore(repository)
|
||||
self.jobs = IndexJobStore(repository)
|
||||
@@ -252,12 +253,13 @@ class RagModule:
|
||||
}
|
||||
|
||||
@router.post("/retrieve")
|
||||
async def retrieve(payload: dict) -> dict:
|
||||
rag_session_id = payload.get("rag_session_id") or payload.get("project_id", "")
|
||||
ctx = await self.rag.retrieve(
|
||||
rag_session_id=rag_session_id,
|
||||
query=payload.get("query", ""),
|
||||
async def retrieve() -> JSONResponse:
|
||||
return JSONResponse(
|
||||
status_code=410,
|
||||
content={
|
||||
"error": "deprecated",
|
||||
"message": "POST /internal/rag/retrieve is deprecated.",
|
||||
},
|
||||
)
|
||||
return {"items": ctx}
|
||||
|
||||
return router
|
||||
|
||||
Reference in New Issue
Block a user