(request: QueryRequest)
| 184 | |
| 185 | @app.post("/query", response_model=Response) |
| 186 | async def query_endpoint(request: QueryRequest): |
| 187 | # try: |
| 188 | # loop = asyncio.get_event_loop() |
| 189 | if request.mode == "naive": |
| 190 | top_k = 3 |
| 191 | else: |
| 192 | top_k = 60 |
| 193 | result = await rag.aquery( |
| 194 | request.query, |
| 195 | param=QueryParam( |
| 196 | mode=request.mode, |
| 197 | only_need_context=request.only_need_context, |
| 198 | only_need_prompt=request.only_need_prompt, |
| 199 | top_k=top_k, |
| 200 | ), |
| 201 | ) |
| 202 | return Response(status="success", data=result) |
| 203 | # except Exception as e: |
| 204 | # raise HTTPException(status_code=500, detail=str(e)) |
| 205 | |
| 206 | |
| 207 | @app.get("/data", response_model=Response) |
nothing calls this directly
no test coverage detected