Initial codes commit
Browse files- api/endpoints.py +3 -3
- 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 |
|