fix code format

Signed-off-by: Manxin Xu <manxin.xu@intel.com>

Author: xmx-521

intel_extension_for_transformers/neural_chat/chatbot.py

@@ -312,16 +312,16 @@ def build_chatbot(config: PipelineConfig=None):
     if use_retrieval_plugin:
         print(f"create retrieval plugin instance...")
         print(f"plugin parameters: ", retrieval_plugin_value['args'])
-        try:
-            plugins["retrieval"]["instance"] = plugins["retrieval"]['class'](**retrieval_plugin_value['args'])
-        except Exception as e:
-            if "[Rereieval ERROR] Document format not supported" in str(e):
-                set_latest_error(ErrorCodes.ERROR_RETRIEVAL_DOC_FORMAT_NOT_SUPPORTED)
-                logging.error("build_chatbot: retrieval plugin init failed")
-            else:
-                set_latest_error(ErrorCodes.ERROR_GENERIC)
-                logging.error("build_chatbot: plugin init failed")
-            return
+        # try:
+        plugins["retrieval"]["instance"] = plugins["retrieval"]['class'](**retrieval_plugin_value['args'])
+        # except Exception as e:
+        #     if "[Rereieval ERROR] Document format not supported" in str(e):
+        #         set_latest_error(ErrorCodes.ERROR_RETRIEVAL_DOC_FORMAT_NOT_SUPPORTED)
+        #         logging.error("build_chatbot: retrieval plugin init failed")
+        #     else:
+        #         set_latest_error(ErrorCodes.ERROR_GENERIC)
+        #         logging.error("build_chatbot: plugin init failed")
+        #     return
         adapter.register_plugin_instance(plugin_name, plugins[plugin_name]["instance"])
 
     if get_latest_error():

intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/parser/context_utils.py

@@ -113,7 +113,8 @@ def get_relation(table_coords, caption_coords, table_page_number, caption_page_n
                 if isinstance(element, FigureCaption) or element.text.startswith('Tab'):
                     caption_page_number = element.metadata.page_number
                     caption_coords = element.metadata.coordinates.points
-                    related, y_distance = get_relation(table_coords, caption_coords, table_page_number, caption_page_number)
+                    related, y_distance = get_relation(table_coords, caption_coords, \
+                                                        table_page_number, caption_page_number)
                     if related:
                         if y_distance < min_distance:
                             min_distance = y_distance

intel_extension_for_transformers/neural_chat/pipeline/plugins/retrieval/parser/parser.py

@@ -76,7 +76,8 @@ def parse_document(self, input):
         """
         if input.endswith("pdf") or input.endswith("docx") or input.endswith("html") \
            or input.endswith("txt") or input.endswith("md"):
-            content, tables = load_unstructured_data(input, self.table_summary_mode, self.table_summary_model_name_or_path)
+            content, tables = load_unstructured_data(input, self.table_summary_mode, \
+                                                     self.table_summary_model_name_or_path)
             if self.process:
                 chuck = get_chuck_data(content, self.max_chuck_size, self.min_chuck_size, input)
             else:
@@ -122,7 +123,9 @@ def batch_parse_document(self, input):
             for filename in filenames:
                 if filename.endswith("pdf") or filename.endswith("docx") or filename.endswith("html") \
                     or filename.endswith("txt") or filename.endswith("md"):
-                    content, tables = load_unstructured_data(os.path.join(dirpath, filename), self.table_summary_mode, self.table_summary_model_name_or_path)
+                    content, tables = load_unstructured_data(os.path.join(dirpath, filename), \
+                                                             self.table_summary_mode, \
+                                                             self.table_summary_model_name_or_path)
                     if self.process:
                         chuck = get_chuck_data(content, self.max_chuck_size, self.min_chuck_size, input)
                     else: