diff --git a/libs/partners/fireworks/tests/integration_tests/test_standard.py b/libs/partners/fireworks/tests/integration_tests/test_standard.py index 692dcb40cf3..7a595e4c7da 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 @@ from typing import Type 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 @@ class TestFireworksStandard(ChatModelIntegrationTests): } @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 3870ae953f6..b97b8c10422 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 @@ from typing import Optional, Type 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 @@ class BaseTestGroq(ChatModelIntegrationTests): 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 @@ class TestGroqLlama(BaseTestGroq): @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 3d1faa97db4..f74f1c6804f 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 @@ from typing import Tuple, Type 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 @@ class TestOpenAIStandard(ChatModelUnitTests): } @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 1152fe44946..edcaf93eebf 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 @@ from typing import Optional, Type 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 @@ class TestXAIStandard(ChatModelIntegrationTests): 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)