fix: The knowledge base table file upload is missing a header (#2185)

1Panel-dev · Feb 10, 2025 · f16f417 · f16f417
1 parent ce79e39
commit f16f417
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/apps/common/handle/impl/csv_split_handle.py b/apps/common/handle/impl/csv_split_handle.py
@@ -55,7 +55,7 @@ def handle(self, file, pattern_list: List, with_filter: bool, limit: int, get_bu
                         result_item_content += next_md_content
                     else:
                         paragraphs.append({'content': result_item_content, 'title': ''})
-                        result_item_content = next_md_content
+                        result_item_content = title_md_content + next_md_content
             if len(result_item_content) > 0:
                 paragraphs.append({'content': result_item_content, 'title': ''})
             return result

diff --git a/apps/common/handle/impl/xls_split_handle.py b/apps/common/handle/impl/xls_split_handle.py
@@ -48,7 +48,7 @@ def handle_sheet(file_name, sheet, limit: int):
                 result_item_content += next_md_content
             else:
                 paragraphs.append({'content': result_item_content, 'title': ''})
-                result_item_content = next_md_content
+                result_item_content = title_md_content + next_md_content
     if len(result_item_content) > 0:
         paragraphs.append({'content': result_item_content, 'title': ''})
     return result

diff --git a/apps/common/handle/impl/xlsx_split_handle.py b/apps/common/handle/impl/xlsx_split_handle.py
@@ -54,7 +54,7 @@ def handle_sheet(file_name, sheet, image_dict, limit: int):
                 result_item_content += next_md_content
             else:
                 paragraphs.append({'content': result_item_content, 'title': ''})
-                result_item_content = next_md_content
+                result_item_content = title_md_content + next_md_content
     if len(result_item_content) > 0:
         paragraphs.append({'content': result_item_content, 'title': ''})
     return result