m97j commited on
Commit
de65949
·
1 Parent(s): 835d057

Initial codes commit

Browse files
Files changed (2) hide show
  1. api/endpoints.py +3 -3
  2. app.py +1 -1
api/endpoints.py CHANGED
@@ -1,11 +1,11 @@
1
  # rag/api/endpoints.py
2
- from fastapi import APIRouter
3
  from api.schemas import SearchRequest, SearchResponse
4
  from service.search import search
5
 
6
  router = APIRouter()
7
 
8
  @router.post("/search", response_model=SearchResponse)
9
- def search_context(req: SearchRequest):
10
- results = search(req.query)
11
  return {"results": results}
 
1
  # rag/api/endpoints.py
2
+ from fastapi import APIRouter, Request
3
  from api.schemas import SearchRequest, SearchResponse
4
  from service.search import search
5
 
6
  router = APIRouter()
7
 
8
  @router.post("/search", response_model=SearchResponse)
9
+ def search_context(request: Request, req: SearchRequest):
10
+ results = search(request, req.query)
11
  return {"results": results}
app.py CHANGED
@@ -25,7 +25,7 @@ def index(request: Request):
25
 
26
  @app.post("/demo", response_class=HTMLResponse)
27
  def demo(request: Request, query: str = Form(...)):
28
- results = search(query)
29
  return templates.TemplateResponse("index.html", {"request": request, "results": results, "query": query})
30
 
31
 
 
25
 
26
  @app.post("/demo", response_class=HTMLResponse)
27
  def demo(request: Request, query: str = Form(...)):
28
+ results = search(request, query)
29
  return templates.TemplateResponse("index.html", {"request": request, "results": results, "query": query})
30
 
31