lint

Author: JasonSteving99

temporalio/contrib/google_gemini_sdk/_gemini_activity.py

@@ -9,7 +9,7 @@
 from __future__ import annotations
 
 from collections.abc import Sequence
-from typing import Any, Callable, Optional
+from typing import Any, Callable
 
 import google.auth.credentials
 from google.genai import Client as GeminiClient
@@ -49,7 +49,7 @@ class GeminiApiCaller:
     def __init__(
         self,
         client: GeminiClient,
-        credentials: Optional[google.auth.credentials.Credentials] = None,
+        credentials: google.auth.credentials.Credentials | None = None,
     ) -> None:
         self._client = client
         self._credentials = credentials

temporalio/contrib/google_gemini_sdk/_gemini_plugin.py

@@ -3,7 +3,6 @@
 from __future__ import annotations
 
 import dataclasses
-from typing import Optional
 
 import google.auth.credentials
 from google.genai import Client as GeminiClient
@@ -64,7 +63,7 @@ class GeminiPlugin(SimplePlugin):
     def __init__(
         self,
         client: GeminiClient,
-        extra_credentials: Optional[google.auth.credentials.Credentials] = None,
+        extra_credentials: google.auth.credentials.Credentials | None = None,
     ) -> None:
         """Initialize the Gemini plugin.
 

temporalio/contrib/google_gemini_sdk/_temporal_api_client.py

@@ -15,7 +15,7 @@
 from __future__ import annotations
 
 from datetime import timedelta
-from typing import Any, Optional
+from typing import Any
 
 from google.genai import types
 from google.genai._api_client import BaseApiClient
@@ -56,7 +56,7 @@ class _SerializableHttpOptions(BaseModel):
 )
 
 
-def _validate_http_options(http_options: Optional[HttpOptions]) -> None:
+def _validate_http_options(http_options: HttpOptions | None) -> None:
     """Raise if http_options contains non-serializable fields."""
     if http_options is None:
         return
@@ -150,7 +150,7 @@ class TemporalApiClient(BaseApiClient):
     All actual HTTP calls are dispatched via ``workflow.execute_activity``.
     """
 
-    def __init__(
+    def __init__(  # pyright: ignore[reportMissingSuperCall]
         self,
         *,
         vertexai: bool = False,
@@ -189,7 +189,7 @@ def __del__(self) -> None:
 
     @staticmethod
     def _process_http_options(
-        http_options: Optional[HttpOptionsOrDict],
+        http_options: HttpOptionsOrDict | None,
         config: ActivityConfig,
     ) -> _SerializableHttpOptions | None:
         """Validate and extract serializable per-request HTTP options.
@@ -243,7 +243,7 @@ async def async_request(
         http_method: str,
         path: str,
         request_dict: dict[str, object],
-        http_options: Optional[HttpOptionsOrDict] = None,
+        http_options: HttpOptionsOrDict | None = None,
     ) -> SdkHttpResponse:
         config: ActivityConfig = {**self._activity_config}
         if "summary" not in config:
@@ -271,7 +271,7 @@ def request(
         http_method: str,
         path: str,
         request_dict: dict[str, object],
-        http_options: Optional[HttpOptionsOrDict] = None,
+        http_options: HttpOptionsOrDict | None = None,
     ) -> SdkHttpResponse:
         raise RuntimeError(
             "Synchronous requests are not supported in Temporal workflows. "
@@ -283,7 +283,7 @@ def request_streamed(
         http_method: str,
         path: str,
         request_dict: dict[str, object],
-        http_options: Optional[HttpOptionsOrDict] = None,
+        http_options: HttpOptionsOrDict | None = None,
     ) -> Any:
         raise RuntimeError(
             "Synchronous streaming is not supported in Temporal workflows. "
@@ -295,7 +295,7 @@ async def async_request_streamed(
         http_method: str,
         path: str,
         request_dict: dict[str, object],
-        http_options: Optional[HttpOptionsOrDict] = None,
+        http_options: HttpOptionsOrDict | None = None,
     ) -> Any:
         config: ActivityConfig = {**self._activity_config}
         if "summary" not in config:

temporalio/contrib/google_gemini_sdk/_temporal_file_search_stores.py

@@ -11,11 +11,9 @@
 import io
 import os
 from datetime import timedelta
-from typing import Optional, Union
 
 from google.genai import types
 from google.genai.file_search_stores import AsyncFileSearchStores
-from google.genai.types import HttpOptions
 
 from temporalio import workflow as temporal_workflow
 from temporalio.contrib.google_gemini_sdk._temporal_api_client import (
@@ -51,8 +49,8 @@ async def upload_to_file_search_store(
         self,
         *,
         file_search_store_name: str,
-        file: Union[str, "os.PathLike[str]", io.IOBase],
-        config: Optional[types.UploadToFileSearchStoreConfigOrDict] = None,
+        file: str | os.PathLike[str] | io.IOBase,
+        config: types.UploadToFileSearchStoreConfigOrDict | None = None,
     ) -> types.UploadToFileSearchStoreOperation:
         """Upload a file to a file search store via a Temporal activity.
 

temporalio/contrib/google_gemini_sdk/_temporal_files.py

@@ -11,13 +11,12 @@
 import io
 import os
 from datetime import timedelta
-from typing import TYPE_CHECKING, Optional, Union
+from typing import TYPE_CHECKING
 
 if TYPE_CHECKING:
     import google.auth.credentials
 from google.genai import types
 from google.genai.files import AsyncFiles
-from google.genai.types import HttpOptions
 
 from temporalio import workflow as temporal_workflow
 from temporalio.contrib.google_gemini_sdk._temporal_api_client import (
@@ -53,8 +52,8 @@ def __init__(
     async def upload(
         self,
         *,
-        file: Union[str, "os.PathLike[str]", io.IOBase],
-        config: Optional[types.UploadFileConfigOrDict] = None,
+        file: str | os.PathLike[str] | io.IOBase,
+        config: types.UploadFileConfigOrDict | None = None,
     ) -> types.File:
         """Upload a file via a Temporal activity.
 
@@ -93,8 +92,8 @@ async def upload(
     async def download(
         self,
         *,
-        file: Union[str, types.File],
-        config: Optional[types.DownloadFileConfigOrDict] = None,
+        file: str | types.File,
+        config: types.DownloadFileConfigOrDict | None = None,
     ) -> bytes:
         """Download a file via a Temporal activity."""
         act_config: ActivityConfig = {**self._activity_config}
@@ -128,7 +127,7 @@ async def register_files(
         *,
         auth: google.auth.credentials.Credentials,
         uris: list[str],
-        config: Optional[types.RegisterFilesConfigOrDict] = None,
+        config: types.RegisterFilesConfigOrDict | None = None,
     ) -> types.RegisterFilesResponse:
         """Register GCS files via a Temporal activity.
 

temporalio/contrib/google_gemini_sdk/justfile

@@ -9,3 +9,38 @@ worker:
 
 query q="What's the weather right now?":
     uv run python first_class_example/start_workflow.py "{{q}}"
+
+embed-worker:
+    uv run python embed_example/worker.py
+
+embed prompt="A serene mountain lake at sunset with pine trees":
+    uv run python embed_example/start_workflow.py "{{prompt}}"
+
+vibe-worker:
+    uv run python vibe_convergence_example/worker.py
+
+set positional-arguments
+
+vibe *ARGS:
+    uv run python vibe_convergence_example/start_workflow.py "$@"
+
+spy-worker:
+    uv run python i_spy_example/worker.py
+
+spy:
+    uv run python i_spy_example/start_workflow.py
+
+file-worker:
+    uv run python file_upload_example/worker.py
+
+file q="Summarize the key financial metrics and top 3 risks from this report.":
+    uv run python file_upload_example/start_workflow.py "{{q}}"
+
+file-store:
+    uv run python file_upload_example/start_workflow.py --store
+
+chat-worker:
+    uv run python chat_example/worker.py
+
+chat topic="recursion in programming":
+    uv run python chat_example/start_workflow.py "{{topic}}"

tests/contrib/google_gemini_sdk/test_gemini.py

@@ -1068,12 +1068,7 @@ def _apply_plugin_with_mock_client(client: Client, mock_responses: list[str]) ->
 
     call_state = {"index": 0}
 
-    async def fake_async_request(
-        http_method: str,
-        path: str,
-        request_dict: dict[str, object],
-        http_options: Any = None,
-    ) -> SdkHttpResponse:
+    async def fake_async_request(*_args: Any, **_kwargs: Any) -> SdkHttpResponse:
         idx = call_state["index"]
         call_state["index"] += 1
         if idx >= len(mock_responses):
@@ -1086,12 +1081,7 @@ async def fake_async_request(
             body=mock_responses[idx],
         )
 
-    async def fake_async_request_streamed(
-        http_method: str,
-        path: str,
-        request_dict: dict[str, object],
-        http_options: Any = None,
-    ) -> Any:
+    async def fake_async_request_streamed(*_args: Any, **_kwargs: Any) -> Any:
         idx = call_state["index"]
         call_state["index"] += 1
         if idx >= len(mock_responses):
@@ -1190,14 +1180,9 @@ async def test_register_files_without_credentials_fails(client: Client):
             )
 
     # The error is nested: WorkflowFailureError → ActivityError → ApplicationError
-    cause: BaseException | None = exc_info.value.cause
-    while (
-        cause is not None
-        and hasattr(cause, "__cause__")
-        and cause.__cause__ is not None
-    ):
+    cause = exc_info.value.cause
+    while cause.__cause__ is not None:
         cause = cause.__cause__
-    assert cause is not None
     assert "No credentials available for register_files" in str(cause)