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