fix key error

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/parser.py

@@ -49,7 +49,8 @@ def load(self, input, **kwargs):
             self.min_chuck_size = kwargs['min_chuck_size']
         if 'process' in kwargs:
             self.process = kwargs['process']
-        self.table_summary_model_name_or_path = kwargs['table_summary_model_name_or_path']
+        if 'table_summary_model_name_or_path' in kwargs:
+            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):