poe format

Author: JasonSteving99

temporalio/contrib/google_gemini_sdk/_gemini_activity.py

@@ -12,7 +12,8 @@
 from typing import Any, Callable, Optional
 
 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
 
@@ -107,9 +108,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(
@@ -149,10 +148,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

@@ -77,9 +77,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/_temporal_api_client.py

@@ -74,6 +74,7 @@ def _validate_http_options(http_options: Optional[HttpOptions]) -> None:
 
 # ── async_request models ──────────────────────────────────────────────────
 
+
 class _GeminiApiRequest(BaseModel):
     """Serializable activity input for a Gemini SDK API call."""
 
@@ -313,7 +314,7 @@ async def async_request_streamed(
             **config,
         )
 
-        async def _yield_chunks(): 
+        async def _yield_chunks():
             for chunk in resp.chunks:
                 yield SdkHttpResponse(headers=chunk.headers, body=chunk.body)
 

temporalio/contrib/google_gemini_sdk/_temporal_async_client.py

@@ -9,8 +9,6 @@
 
 from google.genai.client import AsyncClient
 
-from temporalio.workflow import ActivityConfig
-
 from temporalio.contrib.google_gemini_sdk._temporal_api_client import (
     TemporalApiClient,
 )
@@ -20,6 +18,7 @@
 from temporalio.contrib.google_gemini_sdk._temporal_files import (
     TemporalAsyncFiles,
 )
+from temporalio.workflow import ActivityConfig
 
 
 class TemporalAsyncClient(AsyncClient):

temporalio/contrib/google_gemini_sdk/_temporal_file_search_stores.py

@@ -17,13 +17,12 @@
 from google.genai.types import HttpOptions
 
 from temporalio import workflow as temporal_workflow
-from temporalio.workflow import ActivityConfig
-
 from temporalio.contrib.google_gemini_sdk._temporal_api_client import (
     TemporalApiClient,
     _GeminiUploadToFileSearchStoreRequest,
     _validate_http_options,
 )
+from temporalio.workflow import ActivityConfig
 
 
 class TemporalAsyncFileSearchStores(AsyncFileSearchStores):

temporalio/contrib/google_gemini_sdk/_temporal_files.py

@@ -12,22 +12,20 @@
 from datetime import timedelta
 from typing import Optional, Union
 
+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.workflow import ActivityConfig
-
-import google.auth.credentials
-
 from temporalio.contrib.google_gemini_sdk._temporal_api_client import (
     TemporalApiClient,
     _GeminiDownloadFileRequest,
     _GeminiRegisterFilesRequest,
     _GeminiUploadFileRequest,
     _validate_http_options,
 )
+from temporalio.workflow import ActivityConfig
 
 
 class TemporalAsyncFiles(AsyncFiles):
@@ -74,13 +72,9 @@ async def upload(
             _validate_http_options(upload_config.http_options)
 
         if isinstance(file, io.IOBase):
-            req = _GeminiUploadFileRequest(
-                file_bytes=file.read(), config=upload_config
-            )
+            req = _GeminiUploadFileRequest(file_bytes=file.read(), config=upload_config)
         elif isinstance(file, str):
-            req = _GeminiUploadFileRequest(
-                file_path=file, config=upload_config
-            )
+            req = _GeminiUploadFileRequest(file_path=file, config=upload_config)
         else:
             # os.PathLike — convert via __fspath__() to avoid importing os
             req = _GeminiUploadFileRequest(

tests/contrib/google_gemini_sdk/test_gemini.py

@@ -218,9 +218,7 @@ async def gemini_api_client_async_request_streamed(
         return _GeminiApiStreamedResponse(chunks=chunks)
 
     @activity.defn(name="gemini_files_upload")
-    async def gemini_files_upload(
-        self, req: _GeminiUploadFileRequest
-    ) -> types.File:
+    async def gemini_files_upload(self, req: _GeminiUploadFileRequest) -> types.File:
         self.file_upload_requests.append(req)
         return types.File(
             name="files/test-uploaded-file",
@@ -229,9 +227,7 @@ async def gemini_files_upload(
         )
 
     @activity.defn(name="gemini_files_download")
-    async def gemini_files_download(
-        self, req: _GeminiDownloadFileRequest
-    ) -> bytes:
+    async def gemini_files_download(self, req: _GeminiDownloadFileRequest) -> bytes:
         self.file_download_requests.append(req)
         return b"fake file content"
 
@@ -1042,9 +1038,7 @@ async def test_chat_multi_turn(client: Client):
 # ===========================================================================
 
 
-def _apply_plugin_with_mock_client(
-    client: Client, mock_responses: list[str]
-) -> Client:
+def _apply_plugin_with_mock_client(client: Client, mock_responses: list[str]) -> Client:
     """Create a real GeminiPlugin with real activities but a mocked client.
 
     Unlike ``apply_plugin``, this does NOT replace the activities.  The
@@ -1101,16 +1095,18 @@ async def _gen():
 
     # Mock file operations at the high-level SDK interface (these are what
     # the real activities call).
-    gemini.aio.files.upload = AsyncMock( 
+    gemini.aio.files.upload = AsyncMock(
         return_value=types.File(
             name="files/mock-uploaded",
             uri="https://fake.uri/files/mock-uploaded",
             size_bytes=42,
         )
     )
     gemini.aio.files.download = AsyncMock(return_value=b"mock download content")
-    gemini.aio.file_search_stores.upload_to_file_search_store = AsyncMock( 
-        return_value=types.UploadToFileSearchStoreOperation.model_construct(name="operations/mock-op")
+    gemini.aio.file_search_stores.upload_to_file_search_store = AsyncMock(
+        return_value=types.UploadToFileSearchStoreOperation.model_construct(
+            name="operations/mock-op"
+        )
     )
 
     plugin = GeminiPlugin(gemini)
@@ -1133,10 +1129,10 @@ async def test_full_integration_with_mock_client(client: Client):
     new_client = _apply_plugin_with_mock_client(
         client,
         [
-            make_text_response("Real activity response"),       # generate_content
-            make_text_response("Streamed via real activity"),    # generate_content_stream
-            make_text_response("Grounded RAG answer"),          # RAG query with file_search
-            make_text_response(""),                             # file_search_stores.delete
+            make_text_response("Real activity response"),  # generate_content
+            make_text_response("Streamed via real activity"),  # generate_content_stream
+            make_text_response("Grounded RAG answer"),  # RAG query with file_search
+            make_text_response(""),  # file_search_stores.delete
         ],
     )