main
HuangHai 2 weeks ago
parent 39afdbb181
commit 3bbb67186a

@ -67,7 +67,7 @@ app.mount("/static", StaticFiles(directory="Static"), name="static")
async def rag(request: fastapi.Request): async def rag(request: fastapi.Request):
data = await request.json() data = await request.json()
query = data.get("query") query = data.get("query")
lightrag_working_dir = "./rag_storage" lightrag_working_dir = "./Topic/Chinese"
async def generate_response_stream(query: str): async def generate_response_stream(query: str):
try: try:
llm_model_func = create_llm_model_func() llm_model_func = create_llm_model_func()
@ -89,7 +89,7 @@ async def rag(request: fastapi.Request):
) )
resp = await rag.aquery( resp = await rag.aquery(
"平台安全的保证方法有哪些?", query,
mode="hybrid" mode="hybrid"
) )
print("查询结果:", resp) print("查询结果:", resp)

File diff suppressed because one or more lines are too long
Loading…
Cancel
Save