Add file search stores, register_files, chat support, and full test coverage

- Add TemporalAsyncFileSearchStores: overrides upload_to_file_search_store
  to dispatch through a Temporal activity on the worker.
- Add TemporalAsyncFiles.register_files(): dispatches through activity,
  auth param ignored — uses extra_credentials from GeminiPlugin or the
  client's own credentials. Raises ValueError when neither is available.
- Add extra_credentials param to GeminiPlugin for operations needing
  explicit auth (e.g. GCS file registration).
- Wire TemporalAsyncFileSearchStores into TemporalAsyncClient.
- Extract all Pydantic models into _models.py; move _validate_http_options
  to _temporal_api_client.py.
- Remove all AFC disabling from tests.
- Add chat_example for multi-turn conversation via client.chats.
- Add FileSearchStoreWorkflow exercising full RAG flow with file_search tool.
- Add full integration test running real activities with mocked client.
- Add tests for file upload/download, store upload, chat, register_files
  credential error, TemporalAsyncClient wiring, and low-level raises.
- Fix all lint issues: pydocstyle, mypy, basedpyright, ruff.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>

Author: JasonSteving99

temporalio/contrib/google_gemini_sdk/_gemini_activity.py

@@ -9,15 +9,16 @@
 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, types
+from google.genai import Client as GeminiClient
+from google.genai import types
 from google.genai.types import HttpOptions
 from google.genai.types import HttpResponse as SdkHttpResponse
 
 from temporalio import activity
-from temporalio.contrib.google_gemini_sdk._temporal_api_client import (
+from temporalio.contrib.google_gemini_sdk._models import (
     _GeminiApiRequest,
     _GeminiApiResponse,
     _GeminiApiStreamedResponse,
@@ -48,8 +49,9 @@ class GeminiApiCaller:
     def __init__(
         self,
         client: GeminiClient,
-        credentials: Optional[google.auth.credentials.Credentials] = None,
+        credentials: google.auth.credentials.Credentials | None = None,
     ) -> None:
+        """Initialize with a genai.Client and optional extra credentials."""
         self._client = client
         self._credentials = credentials
 
@@ -107,9 +109,7 @@ async def gemini_files_upload(
             else:
                 file_arg = req.file_path
 
-            return await self._client.aio.files.upload(
-                file=file_arg, config=req.config
-            )
+            return await self._client.aio.files.upload(file=file_arg, config=req.config)
 
         @activity.defn(name="gemini_files_download")
         async def gemini_files_download(
@@ -131,8 +131,15 @@ async def gemini_files_register(
             Token refresh happens here on the worker side, so no auth
             material enters the workflow event history.
             """
+            auth = self._credentials or self._client._api_client._credentials
+            if auth is None:
+                raise ValueError(
+                    "No credentials available for register_files(). "
+                    "Pass extra_credentials to GeminiPlugin or initialize "
+                    "the genai.Client with credentials."
+                )
             return await self._client.aio.files.register_files(
-                auth=self._credentials or self._client._api_client._credentials,
+                auth=auth,
                 uris=req.uris,
                 config=req.config,
             )
@@ -149,10 +156,12 @@ async def gemini_file_search_stores_upload(
             else:
                 file_arg = req.file_path
 
-            return await self._client.aio.file_search_stores.upload_to_file_search_store(
-                file_search_store_name=req.file_search_store_name,
-                file=file_arg,
-                config=req.config,
+            return (
+                await self._client.aio.file_search_stores.upload_to_file_search_store(
+                    file_search_store_name=req.file_search_store_name,
+                    file=file_arg,
+                    config=req.config,
+                )
             )
 
         return [

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.
 
@@ -77,9 +76,7 @@ def __init__(
                 ``files.register_files()``).  If not provided, the
                 client's own credentials are used.
         """
-        self._api_caller = GeminiApiCaller(
-            client, credentials=extra_credentials
-        )
+        self._api_caller = GeminiApiCaller(client, credentials=extra_credentials)
 
         def workflow_runner(runner: WorkflowRunner | None) -> WorkflowRunner:
             if not runner:

temporalio/contrib/google_gemini_sdk/_models.py

@@ -0,0 +1,107 @@
+"""Serializable Pydantic models for the Gemini SDK Temporal integration.
+
+These models cross the activity boundary — they're constructed on the
+workflow side and deserialized on the activity side (or vice versa).
+"""
+
+from __future__ import annotations
+
+from typing import Any
+
+from google.genai import types
+from pydantic import BaseModel
+
+__all__ = [
+    "_GeminiApiRequest",
+    "_GeminiApiResponse",
+    "_GeminiApiStreamedResponse",
+    "_GeminiDownloadFileRequest",
+    "_GeminiRegisterFilesRequest",
+    "_GeminiUploadFileRequest",
+    "_GeminiUploadToFileSearchStoreRequest",
+    "_SerializableHttpOptions",
+]
+
+
+class _SerializableHttpOptions(BaseModel):
+    """Per-request HTTP options that can be serialized across the activity boundary.
+
+    Non-serializable fields (httpx_client, httpx_async_client, aiohttp_client,
+    client_args, async_client_args) must be configured at GeminiPlugin init.
+
+    ``timeout`` is excluded because Temporal owns timeouts/retries — configure
+    via ``ActivityConfig`` instead.
+    """
+
+    base_url: str | None = None
+    base_url_resource_scope: str | None = None
+    api_version: str | None = None
+    headers: dict[str, str] | None = None
+    extra_body: dict[str, Any] | None = None
+
+
+# ── async_request models ──────────────────────────────────────────────────
+
+
+class _GeminiApiRequest(BaseModel):
+    """Serializable activity input for a Gemini SDK API call."""
+
+    http_method: str
+    path: str
+    request_dict: dict[str, object]
+    http_options_overrides: _SerializableHttpOptions | None = None
+
+
+class _GeminiApiResponse(BaseModel):
+    """Serializable activity output for a Gemini SDK API call."""
+
+    headers: dict[str, str]
+    body: str
+
+
+class _GeminiApiStreamedResponse(BaseModel):
+    """Serializable activity output for a batched streamed API call.
+
+    The activity collects all streamed chunks and returns them as a list.
+    The ``TemporalApiClient`` then yields them one at a time to the SDK.
+    """
+
+    chunks: list[_GeminiApiResponse]
+
+
+# ── files upload/download models ──────────────────────────────────────────
+
+
+class _GeminiUploadFileRequest(BaseModel):
+    """Serializable activity input for a file upload.
+
+    For file path uploads the path is resolved on the worker.  For
+    in-memory uploads the raw bytes are sent across the activity boundary.
+    """
+
+    file_bytes: bytes | None = None
+    file_path: str | None = None
+    config: types.UploadFileConfig | None = None
+
+
+class _GeminiDownloadFileRequest(BaseModel):
+    """Serializable activity input for a file download."""
+
+    file: str
+    config: types.DownloadFileConfig | None = None
+
+
+class _GeminiRegisterFilesRequest(BaseModel):
+    """Serializable activity input for registering GCS files."""
+
+    uris: list[str]
+    config: types.RegisterFilesConfig | None = None
+
+
+class _GeminiUploadToFileSearchStoreRequest(BaseModel):
+    """Serializable activity input for uploading a file to a file search store."""
+
+    file_search_store_name: str
+    file_bytes: bytes | None = None
+    file_path: str | None = None
+    config: types.UploadToFileSearchStoreConfig | None = None