pls-rag / api /endpoints.py
m97j's picture
Initial codes commit
de65949
raw
history blame contribute delete
359 Bytes
# rag/api/endpoints.py
from fastapi import APIRouter, Request
from api.schemas import SearchRequest, SearchResponse
from service.search import search
router = APIRouter()
@router.post("/search", response_model=SearchResponse)
def search_context(request: Request, req: SearchRequest):
results = search(request, req.query)
return {"results": results}