修复 bug

This commit is contained in:
Tian jianyong 2025-12-24 19:00:57 +08:00
parent daca304d92
commit 09ca6948f0

View File

@ -340,11 +340,17 @@ def stream_ragflow_response(result: Generator, chat_id: str, start_time: float,
total_time = time.perf_counter() - start_time
logger.info(f'[RAG_SERVER {time.time():.3f}] ⏱️ Total Stream Duration ({chat_id}): {total_time:.3f}s')
# 流结束后存储缓存(使用 create_task 确保异步任务被执行)
# 流结束后存储缓存
if cache_store_func and last_answer and len(last_answer.strip()) >= 10:
try:
asyncio.create_task(cache_store_func(last_answer))
logger.info(f'[RAG_CACHE] 缓存存储任务已调度 | chat_id={chat_id} | answer_length={len(last_answer)}')
import asyncio
loop = asyncio.new_event_loop()
asyncio.set_event_loop(loop)
try:
loop.run_until_complete(cache_store_func(last_answer))
logger.info(f'[RAG_CACHE] 缓存存储完成 | chat_id={chat_id} | answer_length={len(last_answer)}')
finally:
loop.close()
except Exception as cache_err:
logger.warning(f'[RAG_CACHE] 缓存存储失败: {cache_err}')