dsLightRag/Start.py

@@ -335,37 +335,5 @@ async def get_articles(page: int = 1, limit: int = 10):
         return {"code": 1, "msg": str(e)}
 
 
-@app.post("/api/chat")
-async def chat(request: fastapi.Request):
-    data = await request.json()
-    topic = 'ShiJi'
-    mode = data.get("mode", "hybrid")  # 默认为hybrid模式
-    # 拼接路径
-    WORKING_PATH = "./Topic/" + topic
-    # 查询的问题
-    query = data.get("query")
-    # 关闭参考资料
-    user_prompt = "\n 1、总结回答时，要注意不要太繁琐！"
-
-    async def generate_response_stream(query: str):
-        try:
-            rag = await initialize_rag(WORKING_PATH)
-            resp = await rag.aquery(
-                query=query,
-                param=QueryParam(mode=mode, stream=True, user_prompt=user_prompt, enable_rerank=True))
-
-            async for chunk in resp:
-                if not chunk:
-                    continue
-                yield f"data: {json.dumps({'reply': chunk})}\n\n"
-                print(chunk, end='', flush=True)
-        except Exception as e:
-            yield f"data: {json.dumps({'error': str(e)})}\n\n"
-        finally:
-            # 清理资源
-            await rag.finalize_storages()
-
-    return EventSourceResponse(generate_response_stream(query=query))
-
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=8000)