diff --git a/libs/community/langchain_community/retrievers/google_vertex_ai_search.py b/libs/community/langchain_community/retrievers/google_vertex_ai_search.py index 41625823567..109f0385f0c 100644 --- a/libs/community/langchain_community/retrievers/google_vertex_ai_search.py +++ b/libs/community/langchain_community/retrievers/google_vertex_ai_search.py @@ -15,12 +15,7 @@ from langchain_community.utilities.vertexai import get_client_info if TYPE_CHECKING: from google.api_core.client_options import ClientOptions - from google.cloud.discoveryengine_v1beta import ( - ConversationalSearchServiceClient, - SearchRequest, - SearchResult, - SearchServiceClient, - ) + from google.cloud.discoveryengine_v1beta import SearchRequest, SearchResult class _BaseGoogleVertexAISearchRetriever(BaseModel): @@ -243,7 +238,8 @@ class GoogleVertexAISearchRetriever(BaseRetriever, _BaseGoogleVertexAISearchRetr Search will be based on the corrected query if found. """ - _client: SearchServiceClient + # type is SearchServiceClient but can't be set due to optional imports + _client: Any _serving_config: str model_config = ConfigDict( @@ -408,7 +404,8 @@ class GoogleVertexAIMultiTurnSearchRetriever( conversation_id: str = "-" """Vertex AI Search Conversation ID.""" - _client: ConversationalSearchServiceClient + # type is ConversationalSearchServiceClient but can't be set due to optional imports + _client: Any _serving_config: str model_config = ConfigDict( diff --git a/libs/community/pyproject.toml b/libs/community/pyproject.toml index d622bf0c6f3..dbc4617541f 100644 --- a/libs/community/pyproject.toml +++ b/libs/community/pyproject.toml @@ -65,6 +65,9 @@ markers = [ "compile: mark placeholder test used to compile integration tests without running them", ] asyncio_mode = "auto" +filterwarnings = [ + "ignore::langchain_core._api.deprecation.LangChainDeprecationWarning:test", +] [tool.poetry.group.test] optional = true diff --git a/libs/community/tests/unit_tests/load/test_serializable.py b/libs/community/tests/unit_tests/load/test_serializable.py index f86d2f70739..35ea4b169d7 100644 --- a/libs/community/tests/unit_tests/load/test_serializable.py +++ b/libs/community/tests/unit_tests/load/test_serializable.py @@ -120,6 +120,11 @@ def test_serializable_mapping() -> None: "bedrock", "ChatBedrock", ), + ("langchain_google_genai", "chat_models", "ChatGoogleGenerativeAI"): ( + "langchain_google_genai", + "chat_models", + "ChatGoogleGenerativeAI", + ), } serializable_modules = import_all_modules("langchain") diff --git a/libs/community/tests/unit_tests/test_dependencies.py b/libs/community/tests/unit_tests/test_dependencies.py index 47239c945d2..8fe12f140f1 100644 --- a/libs/community/tests/unit_tests/test_dependencies.py +++ b/libs/community/tests/unit_tests/test_dependencies.py @@ -93,6 +93,9 @@ def test_test_group_dependencies(poetry_conf: Mapping[str, Any]) -> None: "responses", "syrupy", "requests-mock", + # TODO: Hack to get around cffi 1.17.1 not working with py3.9, remove when + # fix is released. + "cffi", ] )