Skip to content

Commit 51bf4e6

Browse files
authored
Merge pull request #20 from LianjiaTech/use_parent_code
子ak无额度,需传入父级akcode
2 parents a8afd49 + 53f0ece commit 51bf4e6

File tree

2 files changed

+7
-2
lines changed

2 files changed

+7
-2
lines changed

app/workers/handlers/file_api_task.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,11 @@ def file_api_task_callback(payload: dict) -> bool:
8181
file_api_client.update_processing_status('access_denied', 0, file_id, f'ak_code : {ak_code} is not enabled for file indexing', "file_indexing")
8282
return True
8383

84+
parent_code = ak_info.get('parentCode') if ak_info else None
85+
if parent_code:
86+
# 使用parent ak code
87+
payload['ak_code'] = parent_code
88+
8489
# 根据domtree查找source文件信息
8590
domtree = file_api_client.parse_pdf_from_json(file_id)
8691
source_file_id = domtree.root.source_file.id

bella_rag/utils/trace_log_util.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -70,8 +70,8 @@ def send_event(status, message, **kwargs):
7070
except RateLimitError as re:
7171
send_event("failed", str(re))
7272
raise
73-
except Exception:
74-
send_event("failed", "internal error")
73+
except Exception as e:
74+
send_event("failed", 'Error: ' + str(e))
7575
raise
7676

7777
return wrapper

0 commit comments

Comments
 (0)