fix two parameters

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

Author: xmx-521

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

@@ -140,7 +140,7 @@ def get_relation(table_coords, caption_coords, table_page_number, caption_page_n
             table_summary = predict(**params)
             table_summary = table_summary[table_summary.find('### Generated Summary:\n'):]
             table_summary = re.sub('### Generated Summary:\n', '', table_summary)
-        elif table_summary_mode == 'none':
+        elif table_summary_mode == None:
             table_summary = None
         if table_summary is None:
             text = f'[Table: {content}]'

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

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

intel_extension_for_transformers/neural_chat/tests/ci/plugins/retrieval/test_parameters.py

@@ -785,7 +785,6 @@ def test_table_summary_none_mode(self):
         plugins.retrieval.args["input_path"] = "../assets/docs/LLAMA2_short.pdf"
         plugins.retrieval.args["persist_directory"] = "./table_summary_none_mode"
         plugins.retrieval.args["retrieval_type"] = 'default'
-        plugins.retrieval.args["table_summary_mode"] = 'none'
         config = PipelineConfig(model_name_or_path="facebook/opt-125m",
                                 plugins=plugins)
         chatbot = build_chatbot(config)