fix: 修复agent中因为编码导致模型无法识别的问题 (#701)

* fix iam sign * fix ascii problem
baidubce · Jul 31, 2024 · b06fb71 · b06fb71
1 parent 7c90786
commit b06fb71
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/python/qianfan/extensions/langchain/agents/baidu_qianfan_endpoint.py b/python/qianfan/extensions/langchain/agents/baidu_qianfan_endpoint.py
@@ -71,7 +71,7 @@ def _convert_action_into_message(
                 additional_kwargs={
                     "function_call": {
                         "name": step.tool,
-                        "arguments": json.dumps(step.tool_input),
+                        "arguments": json.dumps(step.tool_input, ensure_ascii=False),
                     }
                 },
             )
@@ -83,7 +83,11 @@ def _convert_action_into_message(
         if not isinstance(tool_result, dict):
             dicts = {"result": tool_result}
 
-        messages.append(FunctionMessage(name=step.tool, content=json.dumps(dicts)))
+        messages.append(
+            FunctionMessage(
+                name=step.tool, content=json.dumps(dicts, ensure_ascii=False)
+            )
+        )
     return messages