[fix] Fixed tuple unpacking and moved UUID conversion into the try block.

This commit is contained in:
lanceyq
2026-03-04 18:01:56 +08:00
parent 13dce5e265
commit 420f391f3c
2 changed files with 4 additions and 4 deletions

View File

@@ -423,7 +423,7 @@ def get_rag_user_kb_total_chunk(
# 通过 App 关联取该 workspace 下所有 end_user_id
end_user_ids = [
str(u.id) for u in db.query(EndUser.id)
str(eid) for (eid,) in db.query(EndUser.id)
.join(App, EndUser.app_id == App.id)
.filter(App.workspace_id == workspace_id)
.all()

View File

@@ -74,11 +74,11 @@ def parse_document(file_path: str, document_id: str):
db = next(get_db()) # Manually call the generator
db_document = None
db_knowledge = None
# 确保 document_id 是 UUID 对象
if not isinstance(document_id, uuid.UUID):
document_id = uuid.UUID(str(document_id))
progress_msg = f"{datetime.now().strftime('%H:%M:%S')} Task has been received.\n"
try:
# 确保 document_id 是 UUID 对象
if not isinstance(document_id, uuid.UUID):
document_id = uuid.UUID(str(document_id))
db_document = db.query(Document).filter(Document.id == document_id).first()
db_knowledge = db.query(Knowledge).filter(Knowledge.id == db_document.kb_id).first()
# 1. Document parsing & segmentation