jprdonnelly/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2025-12-25 01:00:42 -05:00
Code Issues Packages Projects Releases Wiki Activity

perf: commit once (#29590)

Browse Source
This commit is contained in:
wangxiaolei
2025-12-15 11:40:26 +08:00
committed by GitHub
parent 1a18012f98
commit 8f3fd9a728
2 changed files with 89 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
api/core/rag/extractor/word_extractor.py
View File

@@ -84,7 +84,7 @@ class WordExtractor(BaseExtractor):
image_count = 0
image_map = {}
for rId, rel in doc.part.rels.items():
for r_id, rel in doc.part.rels.items():
if "image" in rel.target_ref:
image_count += 1
if rel.is_external:
@@ -121,9 +121,8 @@ class WordExtractor(BaseExtractor):
used_at=naive_utc_now(),
)
db.session.add(upload_file)
db.session.commit()
# Use rId as key for external images since target_part is undefined
image_map[rId] = f"![image]({dify_config.FILES_URL}/files/{upload_file.id}/file-preview)"
# Use r_id as key for external images since target_part is undefined
image_map[r_id] = f"![image]({dify_config.FILES_URL}/files/{upload_file.id}/file-preview)"
else:
image_ext = rel.target_ref.split(".")[-1]
if image_ext is None:
@@ -151,12 +150,11 @@ class WordExtractor(BaseExtractor):
used_at=naive_utc_now(),
)
db.session.add(upload_file)
db.session.commit()
# Use target_part as key for internal images
image_map[rel.target_part] = (
f"![image]({dify_config.FILES_URL}/files/{upload_file.id}/file-preview)"
)
db.session.commit()
return image_map
def _table_to_markdown(self, table, image_map):