kris/ForcePilot
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

fix add chunks bug

Browse Source
This commit is contained in:
Wenjie Zhang 2025-05-10 00:17:38 +08:00
parent c4f475e3cd
commit f086cded44
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
server/routers/data_router.py
View File

@ -78,11 +78,7 @@ async def create_document_by_file(db_id: str = Body(...), files: List[str] = Bod
async def add_by_chunks(db_id: str = Body(...), file_chunks: dict = Body(...), current_user: User = Depends(get_admin_user)): async def add_by_chunks(db_id: str = Body(...), file_chunks: dict = Body(...), current_user: User = Depends(get_admin_user)):
# logger.debug(f"Add chunks in {db_id}: {len(file_chunks)} chunks") # logger.debug(f"Add chunks in {db_id}: {len(file_chunks)} chunks")
try: try:
loop = asyncio.get_event_loop() await knowledge_base.add_chunks(db_id, file_chunks)
await loop.run_in_executor(
executor, # 使用与chat_router相同的线程池
lambda: knowledge_base.add_chunks(db_id, file_chunks)
)
return {"message": "分块添加完成", "status": "success"} return {"message": "分块添加完成", "status": "success"}
except Exception as e: except Exception as e:
logger.error(f"添加分块失败: {e}, {traceback.format_exc()}") logger.error(f"添加分块失败: {e}, {traceback.format_exc()}")