diff --git a/libs/ai-endpoints/tests/integration_tests/test_standard.py b/libs/ai-endpoints/tests/integration_tests/test_standard.py index 540d6ac3..3e07a866 100644 --- a/libs/ai-endpoints/tests/integration_tests/test_standard.py +++ b/libs/ai-endpoints/tests/integration_tests/test_standard.py @@ -20,8 +20,10 @@ def chat_model_params(self) -> dict: return {"model": "meta/llama-3.1-8b-instruct"} @pytest.mark.xfail(reason="anthropic-style list content not supported") - def test_tool_message_histories_list_content(self, model: BaseChatModel) -> None: - return super().test_tool_message_histories_list_content(model) + def test_tool_message_histories_list_content( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + return super().test_tool_message_histories_list_content(model, my_adder_tool) @pytest.mark.xfail(reason="Empty AIMessage content not supported") def test_tool_message_error_status(