From 31c97775e357faa9489717bb0447c496a6488386 Mon Sep 17 00:00:00 2001 From: cthomas Date: Sun, 22 Dec 2024 11:31:03 -0800 Subject: [PATCH] fix: deprecate function return in new test (#2309) --- tests/test_client.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/tests/test_client.py b/tests/test_client.py index 1a9b2602..ac0f4f18 100644 --- a/tests/test_client.py +++ b/tests/test_client.py @@ -400,16 +400,16 @@ def test_function_always_error(client: Union[LocalClient, RESTClient]): response_message = None for message in response.messages: - if isinstance(message, FunctionReturn): + if isinstance(message, ToolReturnMessage): response_message = message break - assert response_message, "FunctionReturn message not found in response" + assert response_message, "ToolReturnMessage message not found in response" assert response_message.status == "error" if isinstance(client, RESTClient): - assert response_message.function_return == "Error executing function always_error: ZeroDivisionError: division by zero" + assert response_message.tool_return == "Error executing function always_error: ZeroDivisionError: division by zero" else: - response_json = json.loads(response_message.function_return) + response_json = json.loads(response_message.tool_return) assert response_json['status'] == "Failed" assert response_json['message'] == "Error executing function always_error: ZeroDivisionError: division by zero"