diff --git a/test/agentchat/test_function_call.py b/test/agentchat/test_function_call.py index 5ca82d3ff5..5dfcd839f3 100755 --- a/test/agentchat/test_function_call.py +++ b/test/agentchat/test_function_call.py @@ -215,6 +215,7 @@ def __init__(self, given_num): def add(self, num_to_be_added): self.given_num = num_to_be_added + self.given_num return str(self.given_num) + user = UserProxyAgent(name="test", function_map={"add_num": AddNum(given_num=10).add}) func_call = {"name": "add_num", "arguments": '{ "num_to_be_added": 5 }'} assert (await user.a_execute_function(func_call=func_call))[1]["content"] == "15"