diff --git a/libs/partners/huggingface/tests/integration_tests/test_standard.py b/libs/partners/huggingface/tests/integration_tests/test_standard.py index 682a4c625ee9f..ce9b5b6b44b9f 100644 --- a/libs/partners/huggingface/tests/integration_tests/test_standard.py +++ b/libs/partners/huggingface/tests/integration_tests/test_standard.py @@ -4,6 +4,7 @@ import pytest from langchain_core.language_models import BaseChatModel +from langchain_core.tools import BaseTool from langchain_tests.integration_tests import ChatModelIntegrationTests from langchain_huggingface import ChatHuggingFace, HuggingFaceEndpoint @@ -82,9 +83,15 @@ def test_structured_output_optional_param(self, model: BaseChatModel) -> None: super().test_structured_output_optional_param(model) @pytest.mark.xfail(reason=("Not implemented")) - def test_tool_message_histories_list_content(self, model: BaseChatModel) -> None: - super().test_tool_message_histories_list_content(model) + def test_tool_message_histories_list_content( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + super().test_tool_message_histories_list_content( + model, my_adder_tool=my_adder_tool + ) @pytest.mark.xfail(reason=("Not implemented")) - def test_structured_few_shot_examples(self, model: BaseChatModel) -> None: - super().test_structured_few_shot_examples(model) + def test_structured_few_shot_examples( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + super().test_structured_few_shot_examples(model, my_adder_tool=my_adder_tool)