fix bug

Signed-off-by: Xinyao Wang <[email protected]>
opea-project · Dec 17, 2024 · f8ca3a7 · f8ca3a7
1 parent 08f791b
commit f8ca3a7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/DocSum/docsum.py b/DocSum/docsum.py
@@ -31,8 +31,9 @@ def align_inputs(self, inputs, cur_node, runtime_graph, llm_parameters_dict, **k
     if self.services[cur_node].service_type == ServiceType.LLM:
         docsum_parameters = kwargs.get("docsum_parameters", None)
         if docsum_parameters:
+            docsum_parameters = docsum_parameters.dict()
             del docsum_parameters["query"]
-            inputs.update(docsum_parameters.dict())
+            inputs.update(docsum_parameters)
     return inputs