adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files

2
api/app/services/memory_dashboard_service.py
View File

@@ -423,7 +423,7 @@ def get_rag_user_kb_total_chunk(
# 通过 App 关联取该 workspace 下所有 end_user_id # 通过 App 关联取该 workspace 下所有 end_user_id
end_user_ids = [ 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) .join(App, EndUser.app_id == App.id)
.filter(App.workspace_id == workspace_id) .filter(App.workspace_id == workspace_id)
.all() .all()

6
api/app/tasks.py
View File

@@ -74,11 +74,11 @@ def parse_document(file_path: str, document_id: str):
db = next(get_db()) # Manually call the generator db = next(get_db()) # Manually call the generator
db_document = None db_document = None
db_knowledge = 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" progress_msg = f"{datetime.now().strftime('%H:%M:%S')} Task has been received.\n"
try: 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_document = db.query(Document).filter(Document.id == document_id).first()
db_knowledge = db.query(Knowledge).filter(Knowledge.id == db_document.kb_id).first() db_knowledge = db.query(Knowledge).filter(Knowledge.id == db_document.kb_id).first()
# 1. Document parsing & segmentation # 1. Document parsing & segmentation