From f177139cbc87cd29f06c4122604f6a9b4000361f Mon Sep 17 00:00:00 2001 From: Brian Strauch Date: Sun, 30 Aug 2026 10:12:18 -0700 Subject: [PATCH 1/3] Support Google ADK 2.7 streaming import --- tests/contrib/google_adk_agents/test_adk_streaming.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/tests/contrib/google_adk_agents/test_adk_streaming.py b/tests/contrib/google_adk_agents/test_adk_streaming.py index 30aecd9f4..6cbffe71a 100644 --- a/tests/contrib/google_adk_agents/test_adk_streaming.py +++ b/tests/contrib/google_adk_agents/test_adk_streaming.py @@ -12,7 +12,13 @@ import pytest from google.adk import Agent -from google.adk.agents.run_config import RunConfig, StreamingMode + +# ADK 2.7 marks StreamingMode as a private re-export, while older supported +# versions do not have the suggested google.adk.agents._streaming_mode module. +from google.adk.agents.run_config import ( + RunConfig, + StreamingMode, # pyright: ignore[reportPrivateImportUsage] +) from google.adk.models import BaseLlm, LLMRegistry from google.adk.models.llm_request import LlmRequest from google.adk.models.llm_response import LlmResponse From 3a119e70a082cfec6c3b7f005d5e423d0fc5845e Mon Sep 17 00:00:00 2001 From: Brian Strauch Date: Sun, 30 Aug 2026 11:27:48 -0700 Subject: [PATCH 2/3] Fix tests with latest dependencies --- tests/contrib/google_genai/test_gemini.py | 8 ++++---- tests/nexus/test_temporal_system_nexus.py | 11 ++++------- 2 files changed, 8 insertions(+), 11 deletions(-) diff --git a/tests/contrib/google_genai/test_gemini.py b/tests/contrib/google_genai/test_gemini.py index 0881f63b3..880f996f9 100644 --- a/tests/contrib/google_genai/test_gemini.py +++ b/tests/contrib/google_genai/test_gemini.py @@ -292,8 +292,8 @@ async def gemini_file_search_stores_upload( self, req: _GeminiUploadToFileSearchStoreRequest ) -> types.UploadToFileSearchStoreOperation: self.file_search_store_upload_requests.append(req) - return types.UploadToFileSearchStoreOperation.model_construct( - name="operations/test-op", + return types.UploadToFileSearchStoreOperation.model_validate( + {"name": "operations/test-op"} ) @activity.defn @@ -1387,8 +1387,8 @@ async def _gen(): ) gemini.aio.files.download = AsyncMock(return_value=b"mock download content") # type: ignore[method-assign] gemini.aio.file_search_stores.upload_to_file_search_store = AsyncMock( # type: ignore[method-assign] - return_value=types.UploadToFileSearchStoreOperation.model_construct( - name="operations/mock-op" + return_value=types.UploadToFileSearchStoreOperation.model_validate( + {"name": "operations/mock-op"} ) ) diff --git a/tests/nexus/test_temporal_system_nexus.py b/tests/nexus/test_temporal_system_nexus.py index 6a9bc9959..dc49f0302 100644 --- a/tests/nexus/test_temporal_system_nexus.py +++ b/tests/nexus/test_temporal_system_nexus.py @@ -566,13 +566,10 @@ def _proto_scalar_sample(field: FieldDescriptor, *, path: str) -> Any: def _field_is_repeated(field: FieldDescriptor) -> bool: - return bool( - getattr( - field, - "is_repeated", - getattr(field, "label") == FieldDescriptor.LABEL_REPEATED, - ) - ) + is_repeated = getattr(field, "is_repeated", None) + if is_repeated is not None: + return bool(is_repeated) + return getattr(field, "label") == FieldDescriptor.LABEL_REPEATED @pytest.mark.parametrize( From e4a5ea786ebbe62f4b6e88d849e5200c11880c68 Mon Sep 17 00:00:00 2001 From: Brian Strauch Date: Sun, 30 Aug 2026 11:29:55 -0700 Subject: [PATCH 3/3] Apply suggestion from @brianstrauch --- tests/contrib/google_adk_agents/test_adk_streaming.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/contrib/google_adk_agents/test_adk_streaming.py b/tests/contrib/google_adk_agents/test_adk_streaming.py index 6cbffe71a..d426990bb 100644 --- a/tests/contrib/google_adk_agents/test_adk_streaming.py +++ b/tests/contrib/google_adk_agents/test_adk_streaming.py @@ -13,8 +13,7 @@ import pytest from google.adk import Agent -# ADK 2.7 marks StreamingMode as a private re-export, while older supported -# versions do not have the suggested google.adk.agents._streaming_mode module. +# ADK 2.7 marks StreamingMode as a private re-export. from google.adk.agents.run_config import ( RunConfig, StreamingMode, # pyright: ignore[reportPrivateImportUsage]