diff --git a/src/promptflow-core/promptflow/_utils/tool_utils.py b/src/promptflow-core/promptflow/_utils/tool_utils.py index 003d227a67b..ab802ea36ab 100644 --- a/src/promptflow-core/promptflow/_utils/tool_utils.py +++ b/src/promptflow-core/promptflow/_utils/tool_utils.py @@ -301,7 +301,7 @@ def validate_tool_func_result(func_call_scenario: str, result): if func_call_scenario == ToolFuncCallScenario.REVERSE_GENERATED_BY: if not isinstance(result, Dict): raise RetrieveToolFuncResultValidationError( - f"ToolFuncCallScenario {func_call_scenario} response must be a dict. " f"{result} is not a dict." + f"ToolFuncCallScenario {func_call_scenario.value} response must be a dict. " f"{result} is not a dict." ) elif func_call_scenario == ToolFuncCallScenario.DYNAMIC_LIST: validate_dynamic_list_func_response_type(result, f"ToolFuncCallScenario {func_call_scenario}") diff --git a/src/promptflow/tests/executor/unittests/_core/test_tools_manager.py b/src/promptflow/tests/executor/unittests/_core/test_tools_manager.py index f8e46f74238..d2bedd721c1 100644 --- a/src/promptflow/tests/executor/unittests/_core/test_tools_manager.py +++ b/src/promptflow/tests/executor/unittests/_core/test_tools_manager.py @@ -366,8 +366,9 @@ def test_retrieve_tool_func_result_error( def test_register_apis(self): from typing import Union - from promptflow._core.tools_manager import register_apis, connection_type_to_api_mapping + from promptflow._core.tool import ToolProvider + from promptflow._core.tools_manager import connection_type_to_api_mapping, register_apis from promptflow.connections import AzureOpenAIConnection, OpenAIConnection, ServerlessConnection class MockAI1(ToolProvider):