|
|
|
@ -43,6 +43,7 @@ app.mount("/static", StaticFiles(directory="Static"), name="static")
|
|
|
|
|
async def rag(request: fastapi.Request):
|
|
|
|
|
data = await request.json()
|
|
|
|
|
topic = data.get("topic") # Chinese, Math
|
|
|
|
|
mode = data.get("mode", "hybrid") # 默认为hybrid模式
|
|
|
|
|
# 拼接路径
|
|
|
|
|
WORKING_PATH = "./Topic/" + topic
|
|
|
|
|
# 查询的问题
|
|
|
|
@ -67,7 +68,7 @@ async def rag(request: fastapi.Request):
|
|
|
|
|
await initialize_pipeline_status()
|
|
|
|
|
resp = await rag.aquery(
|
|
|
|
|
query=query,
|
|
|
|
|
param=QueryParam(mode="hybrid", stream=True, user_prompt=user_prompt))
|
|
|
|
|
param=QueryParam(mode=mode, stream=True, user_prompt=user_prompt))
|
|
|
|
|
# hybrid naive
|
|
|
|
|
|
|
|
|
|
async for chunk in resp:
|
|
|
|
|