api/endpoints.py

# rag/api/endpoints.py
from fastapi import APIRouter
from api.schemas import SearchRequest, SearchResponse
from rag.service.search import search

router = APIRouter()

@router.post("/search", response_model=SearchResponse)
def search_context(req: SearchRequest):
    results = search(req.query)
    return {"results": results}