diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index 692dcb40cf357..7a595e4c7da44 100644 --- a/libs/partners/fireworks/tests/integration_tests/test_standard.py +++ b/libs/partners/fireworks/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 ( # type: ignore[import-not-found] ChatModelIntegrationTests, # type: ignore[import-not-found] ) @@ -24,5 +25,7 @@ def chat_model_params(self) -> dict: } @pytest.mark.xfail(reason="Not yet 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) diff --git a/libs/partners/groq/tests/integration_tests/test_standard.py b/libs/partners/groq/tests/integration_tests/test_standard.py index 3870ae953f6ee..b97b8c10422ef 100644 --- a/libs/partners/groq/tests/integration_tests/test_standard.py +++ b/libs/partners/groq/tests/integration_tests/test_standard.py @@ -5,6 +5,7 @@ import pytest from langchain_core.language_models import BaseChatModel from langchain_core.rate_limiters import InMemoryRateLimiter +from langchain_core.tools import BaseTool from langchain_tests.integration_tests import ( ChatModelIntegrationTests, ) @@ -20,8 +21,10 @@ def chat_model_class(self) -> Type[BaseChatModel]: return ChatGroq @pytest.mark.xfail(reason="Not yet 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) class TestGroqLlama(BaseTestGroq): @@ -47,8 +50,10 @@ def test_tool_calling_with_no_arguments(self, model: BaseChatModel) -> None: @pytest.mark.xfail( reason=("Fails with 'Failed to call a function. Please adjust your prompt.'") ) - def test_tool_message_histories_string_content(self, model: BaseChatModel) -> None: - super().test_tool_message_histories_string_content(model) + def test_tool_message_histories_string_content( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + super().test_tool_message_histories_string_content(model, my_adder_tool) @pytest.mark.xfail( reason=( diff --git a/libs/partners/openai/tests/unit_tests/chat_models/test_azure_standard.py b/libs/partners/openai/tests/unit_tests/chat_models/test_azure_standard.py index 3d1faa97db485..f74f1c6804f3a 100644 --- a/libs/partners/openai/tests/unit_tests/chat_models/test_azure_standard.py +++ b/libs/partners/openai/tests/unit_tests/chat_models/test_azure_standard.py @@ -4,6 +4,7 @@ import pytest from langchain_core.language_models import BaseChatModel +from langchain_core.tools import BaseTool from langchain_tests.unit_tests import ChatModelUnitTests from langchain_openai import AzureChatOpenAI @@ -23,8 +24,10 @@ def chat_model_params(self) -> dict: } @pytest.mark.xfail(reason="AzureOpenAI does not support tool_choice='any'") - def test_bind_tool_pydantic(self, model: BaseChatModel) -> None: - super().test_bind_tool_pydantic(model) + def test_bind_tool_pydantic( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + super().test_bind_tool_pydantic(model, my_adder_tool) @property def init_from_env_params(self) -> Tuple[dict, dict, dict]: diff --git a/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py b/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py index 1152fe44946f7..edcaf93eebf02 100644 --- a/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py +++ b/libs/partners/xai/tests/integration_tests/test_chat_models_standard.py @@ -5,6 +5,7 @@ import pytest # type: ignore[import-not-found] from langchain_core.language_models import BaseChatModel from langchain_core.rate_limiters import InMemoryRateLimiter +from langchain_core.tools import BaseTool from langchain_tests.integration_tests import ( # type: ignore[import-not-found] ChatModelIntegrationTests, # type: ignore[import-not-found] ) @@ -40,13 +41,19 @@ def test_usage_metadata_streaming(self, model: BaseChatModel) -> None: super().test_usage_metadata_streaming(model) @pytest.mark.xfail(reason="Can't handle AIMessage with empty content.") - def test_tool_message_error_status(self, model: BaseChatModel) -> None: - super().test_tool_message_error_status(model) + def test_tool_message_error_status( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + super().test_tool_message_error_status(model, my_adder_tool) @pytest.mark.xfail(reason="Can't handle AIMessage with empty content.") - 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) @pytest.mark.xfail(reason="Can't handle AIMessage with empty content.") - def test_tool_message_histories_string_content(self, model: BaseChatModel) -> None: - super().test_tool_message_histories_string_content(model) + def test_tool_message_histories_string_content( + self, model: BaseChatModel, my_adder_tool: BaseTool + ) -> None: + super().test_tool_message_histories_string_content(model, my_adder_tool)