diff --git a/dsRag/Start.py b/dsRag/Start.py index fdb6eb33..4547f2e1 100644 --- a/dsRag/Start.py +++ b/dsRag/Start.py @@ -123,17 +123,16 @@ async def rag(request: fastapi.Request): return {"data": "没有在知识库中找到相关的信息,无法回答此问题。"} -@app.post("/api/rag_stream", response_model=None) -async def rag_stream(request: fastapi.Request): - data = await request.json() - query = data.get('query', '') - query_tags = data.get('tags', []) - - # 调用es进行混合搜索 - search_results = queryByEs(query, query_tags, logger) - - # 流式调用大模型 - return callLLM(request, query, search_results, logger, True) +@app.post("/api/helloWorld", response_model=None) +async def helloWorld(request: fastapi.Request): + async def generate_hello_world(): + message = "Hello,World," + for char in message: + yield f"data: {char}\n\n" + import asyncio + await asyncio.sleep(0.5) + + return StreamingResponse(generate_hello_world(), media_type="text/event-stream") if __name__ == "__main__": diff --git a/dsRag/Util/__pycache__/SearchUtil.cpython-310.pyc b/dsRag/Util/__pycache__/SearchUtil.cpython-310.pyc index f6193525..e8e5b99d 100644 Binary files a/dsRag/Util/__pycache__/SearchUtil.cpython-310.pyc and b/dsRag/Util/__pycache__/SearchUtil.cpython-310.pyc differ