Merge branch 'manxin/rag_table_summary' of https://github.com/intel/i…

…ntel-extension-for-transformers into manxin/rag_table_summary
intel · Mar 25, 2024 · 76ef38f · 76ef38f
2 parents 21f1320 + ea66051
commit 76ef38f
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 4 deletions.
diff --git a/intel_extension_for_transformers/neural_chat/chatbot.py b/intel_extension_for_transformers/neural_chat/chatbot.py
@@ -323,7 +323,7 @@ def build_chatbot(config: PipelineConfig=None):
         #         logging.error("build_chatbot: plugin init failed")
         #     return
         adapter.register_plugin_instance(plugin_name, plugins[plugin_name]["instance"])
-    
+
     if get_latest_error():
         return
     else:

diff --git a/...extension_for_transformers/neural_chat/pipeline/plugins/retrieval/parser/context_utils.py b/...extension_for_transformers/neural_chat/pipeline/plugins/retrieval/parser/context_utils.py
@@ -95,7 +95,7 @@ def get_relation(table_coords, caption_coords, table_page_number, caption_page_n
             y_distance = 0
         y_close = y_distance < threshold
         return same_page and x_overlap and y_close, y_distance
-    
+
     raw_pdf_elements = partition_pdf(
         filename=pdf_path,
         infer_table_structure=True,

diff --git a/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/parser/parser.py b/intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/parser/parser.py
@@ -51,7 +51,7 @@ def load(self, input, **kwargs):
             self.process = kwargs['process']
         self.table_summary_model_name_or_path = kwargs['table_summary_model_name_or_path']
         self.table_summary_mode = kwargs['table_summary_mode'] if 'table_summary_mode' in kwargs else 'none'
-        
+
         if isinstance(input, str):
             if os.path.isfile(input):
                 data_collection = self.parse_document(input)

diff --git a/intel_extension_for_transformers/neural_chat/prompts/prompt.py b/intel_extension_for_transformers/neural_chat/prompts/prompt.py
@@ -331,4 +331,4 @@ def generate_sqlcoder_prompt(qurey, metadata_file):
 {table_content}
 ---
 ### Generated Summary:
-"""
+"""