fix re.escape problem

infiniflow · Oct 8, 2024 · b88245d · b88245d
1 parent 6f7fcdc
commit b88245d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/agent/component/generate.py b/agent/component/generate.py
@@ -112,7 +112,7 @@ def _run(self, history, **kwargs):
 
         kwargs["input"] = input
         for n, v in kwargs.items():
-            prompt = re.sub(r"\{%s\}" % n, re.escape(str(v)), prompt)
+            prompt = re.sub(r"\{%s\}" % re.escape(n), str(v), prompt)
 
         downstreams = self._canvas.get_component(self._id)["downstream"]
         if kwargs.get("stream") and len(downstreams) == 1 and self._canvas.get_component(downstreams[0])[