lint

Author: JasonSteving99

temporalio/contrib/google_gemini_sdk/_gemini_activity.py

@@ -51,6 +51,7 @@ def __init__(
         client: GeminiClient,
         credentials: google.auth.credentials.Credentials | None = None,
     ) -> None:
+        """Initialize with a genai.Client and optional extra credentials."""
         self._client = client
         self._credentials = credentials
 

temporalio/contrib/google_gemini_sdk/_temporal_api_client.py

@@ -77,6 +77,7 @@ def __init__(  # pyright: ignore[reportMissingSuperCall]
         location: str | None = None,
         activity_config: ActivityConfig | None = None,
     ) -> None:
+        """Initialize without calling super (no HTTP clients needed)."""
         # Do NOT call super().__init__() — it creates HTTP clients, loads
         # credentials, etc.  We only set the properties the SDK's request
         # formatting code accesses.
@@ -164,6 +165,7 @@ async def async_request(
         request_dict: dict[str, object],
         http_options: HttpOptionsOrDict | None = None,
     ) -> SdkHttpResponse:
+        """Dispatch an async API request through a Temporal activity."""
         config: ActivityConfig = {**self._activity_config}
         if "summary" not in config:
             # Default summary is the API path (e.g. "models/gemini-2.5-flash:generateContent").
@@ -192,6 +194,7 @@ def request(
         request_dict: dict[str, object],
         http_options: HttpOptionsOrDict | None = None,
     ) -> SdkHttpResponse:
+        """Raise — sync requests not supported in workflows."""
         raise RuntimeError(
             "Synchronous requests are not supported in Temporal workflows. "
             "Use the AsyncClient returned by gemini_client() instead."
@@ -204,6 +207,7 @@ def request_streamed(
         request_dict: dict[str, object],
         http_options: HttpOptionsOrDict | None = None,
     ) -> Any:
+        """Raise — sync streaming not supported in workflows."""
         raise RuntimeError(
             "Synchronous streaming is not supported in Temporal workflows. "
             "Use the AsyncClient returned by gemini_client() instead."
@@ -216,6 +220,7 @@ async def async_request_streamed(
         request_dict: dict[str, object],
         http_options: HttpOptionsOrDict | None = None,
     ) -> Any:
+        """Dispatch a streamed request, batching chunks in the activity."""
         config: ActivityConfig = {**self._activity_config}
         if "summary" not in config:
             config["summary"] = f"{http_method.upper()} {path}"
@@ -247,21 +252,25 @@ async def _yield_chunks():
     # so we raise here to catch any unexpected direct usage.
 
     def upload_file(self, *args: Any, **kwargs: Any) -> Any:
+        """Raise — use client.files.upload() instead."""
         raise NotImplementedError(
             "Use client.files.upload() instead of the internal upload_file() method."
         )
 
     async def async_upload_file(self, *args: Any, **kwargs: Any) -> Any:
+        """Raise — use client.files.upload() instead."""
         raise NotImplementedError(
             "Use client.files.upload() instead of the internal async_upload_file() method."
         )
 
     def download_file(self, *args: Any, **kwargs: Any) -> Any:
+        """Raise — use client.files.download() instead."""
         raise NotImplementedError(
             "Use client.files.download() instead of the internal download_file() method."
         )
 
     async def async_download_file(self, *args: Any, **kwargs: Any) -> Any:
+        """Raise — use client.files.download() instead."""
         raise NotImplementedError(
             "Use client.files.download() instead of the internal async_download_file() method."
         )

temporalio/contrib/google_gemini_sdk/_temporal_async_client.py

@@ -39,6 +39,7 @@ def __init__(
         api_client: TemporalApiClient,
         activity_config: ActivityConfig | None = None,
     ) -> None:
+        """Initialize with Temporal-aware files and file search stores."""
         super().__init__(api_client)
         self._files = TemporalAsyncFiles(api_client, activity_config)
         self._file_search_stores = TemporalAsyncFileSearchStores(

temporalio/contrib/google_gemini_sdk/_temporal_file_search_stores.py

@@ -42,6 +42,7 @@ def __init__(
         api_client: TemporalApiClient,
         activity_config: ActivityConfig | None = None,
     ) -> None:
+        """Initialize with activity config for upload timeouts."""
         super().__init__(api_client)
         self._activity_config = activity_config or ActivityConfig(
             start_to_close_timeout=timedelta(seconds=60),

temporalio/contrib/google_gemini_sdk/_temporal_files.py

@@ -46,6 +46,7 @@ def __init__(
         api_client: TemporalApiClient,
         activity_config: ActivityConfig | None = None,
     ) -> None:
+        """Initialize with activity config for file operation timeouts."""
         super().__init__(api_client)
         self._activity_config = activity_config or ActivityConfig(
             start_to_close_timeout=timedelta(seconds=60),

tests/contrib/google_gemini_sdk/__init__.py

@@ -0,0 +1 @@
+"""Tests for the Google Gemini SDK Temporal integration."""