diff --git a/dsLightRag/Start.py b/dsLightRag/Start.py index 8923fe23..f0da83c4 100644 --- a/dsLightRag/Start.py +++ b/dsLightRag/Start.py @@ -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: diff --git a/dsLightRag/static/ai.html b/dsLightRag/static/ai.html index 0ed164c2..b946d816 100644 --- a/dsLightRag/static/ai.html +++ b/dsLightRag/static/ai.html @@ -232,6 +232,20 @@ +