Fix agent session API

infiniflow · Nov 22, 2024 · 06c104f · 06c104f
1 parent ee7fd71
commit 06c104f
Show file tree

Hide file tree

Showing 5 changed files with 14 additions and 72 deletions.
diff --git a/agent/component/base.py b/agent/component/base.py
@@ -390,8 +390,8 @@ def __str__(self):
             "inputs": {}
         }}""".format(self.component_name,
                      self._param,
-                     json.dumps(json.loads(str(self._param))["output"], ensure_ascii=False),
-                     json.dumps(json.loads(str(self._param))["inputs"], ensure_ascii=False)
+                     json.dumps(json.loads(str(self._param)).get("output", {}), ensure_ascii=False),
+                     json.dumps(json.loads(str(self._param)).get("inputs", []), ensure_ascii=False)
         )
 
     def __init__(self, canvas, id, param: ComponentParamBase):

diff --git a/agent/component/generate.py b/agent/component/generate.py
@@ -145,7 +145,7 @@ def _run(self, history, **kwargs):
         else: retrieval_res = pd.DataFrame([])
 
         for n, v in kwargs.items():
-            prompt = re.sub(r"\{%s\}" % re.escape(n), str(v), prompt)
+            prompt = re.sub(r"\{%s\}" % re.escape(n), str(v).replace("\\", " "), prompt)
 
         if not self._param.inputs and prompt.find("{input}") >= 0:
             retrieval_res = self.get_input()

diff --git a/agent/component/template.py b/agent/component/template.py
@@ -79,7 +79,7 @@ def _run(self, history, **kwargs):
             self._param.inputs.append({"component_id": para["component_id"], "content": kwargs[para["key"]]})
 
         for n, v in kwargs.items():
-            content = re.sub(r"\{%s\}" % re.escape(n), str(v), content)
+            content = re.sub(r"\{%s\}" % re.escape(n), str(v).replace("\\", " "), content)
 
         return Template.be_output(content)