better naming

Author: marcusmotill

src/google/adk/integrations/temporal/README.md

@@ -4,9 +4,9 @@ This package provides the integration layer between the Google ADK and Temporal.
 
 ## Core Concepts
 
-### 1. Interception Flow (`TemporalPlugin`)
+### 1. Interception Flow (`AgentPlugin`)
 
-The `TemporalPlugin` acts as a middleware that intercepts model calls (e.g., `agent.generate_content`) *before* they execute.
+The `AgentPlugin` acts as a middleware that intercepts model calls (e.g., `agent.generate_content`) *before* they execute.
 
 **Workflow Interception:**
 1.  **Intercept**: The ADK invokes `before_model_callback` when an agent attempts to call a model.
@@ -32,12 +32,12 @@ When the workflow executes an activity with an unknown name (e.g., `MyAgent.gene
 The integration requires setup on both the Agent (Workflow) side and the Worker side.
 
 #### Agent Setup (Workflow Side)
-Attach the `TemporalPlugin` to your ADK agent. This safely routes model calls through Temporal activities. You **must** provide activity options (e.g., timeouts) as there are no defaults.
+Attach the `AgentPlugin` to your ADK agent. This safely routes model calls through Temporal activities. You **must** provide activity options (e.g., timeouts) as there are no defaults.
 
 ```python
 from datetime import timedelta
 from temporalio.common import RetryPolicy
-from google.adk.integrations.temporal import TemporalPlugin
+from google.adk.integrations.temporal import AgentPlugin
 
 # 1. Define Temporal Activity Options
 activity_options = {
@@ -50,7 +50,7 @@ agent = Agent(
     model="gemini-2.5-pro",
     plugins=[
         # Routes model calls to Temporal Activities
-        TemporalPlugin(activity_options=activity_options)  
+        AgentPlugin(activity_options=activity_options)  
     ]
 )
 
@@ -59,23 +59,23 @@ agent = Agent(
 ```
 
 #### Worker Setup
-Install the `AdkWorkerPlugin` on your Temporal Worker. This handles serialization and runtime determinism.
+Install the `WorkerPlugin` on your Temporal Worker. This handles serialization and runtime determinism.
 
 ```python
 from temporalio.worker import Worker
-from google.adk.integrations.temporal import AdkWorkerPlugin
+from google.adk.integrations.temporal import WorkerPlugin
 
 async def main():
     worker = Worker(
         client,
         task_queue="my-queue",
         # Configures ADK Runtime & Pydantic Support
-        plugins=[AdkWorkerPlugin()]
+        plugins=[WorkerPlugin()]
     )
     await worker.run()
 ```
 
-**What `AdkWorkerPlugin` Does:**
+**What `WorkerPlugin` Does:**
 *   **Data Converter**: Enables Pydantic serialization for ADK objects.
 *   **Interceptors**: Sets up specific ADK runtime hooks for determinism (replacing `time.time`, `uuid.uuid4`) before workflow execution.
 *   TODO: is this enough . **Unsandboxed Workflow Runner**: Configures the worker to use the `UnsandboxedWorkflowRunner`, allowing standard imports in ADK agents.

src/google/adk/integrations/temporal/__init__.py

@@ -91,7 +91,7 @@ def workflow_interceptor_class(
     ) -> type[WorkflowInboundInterceptor] | None:
         return AdkWorkflowInboundInterceptor
 
-class TemporalPlugin(BasePlugin):
+class AgentPlugin(BasePlugin):
     """ADK Plugin for Temporal integration.
     
     This plugin automatically configures the ADK runtime to be deterministic when running
@@ -167,7 +167,7 @@ async def before_model_callback(
 
 
 
-class AdkWorkerPlugin(SimplePlugin):
+class WorkerPlugin(SimplePlugin):
     """A Temporal Worker Plugin configured for ADK.
     
     This plugin configures:
@@ -192,7 +192,7 @@ async def dynamic_activity(args: Sequence[RawValue]) -> Any:
 
         # Check if this is a generate_content call
         if activity_type.endswith(".generate_content") or activity_type == "google.adk.generate_content":
-            return await AdkWorkerPlugin._handle_generate_content(args)
+            return await WorkerPlugin._handle_generate_content(args)
 
         raise ValueError(f"Unknown dynamic activity: {activity_type}")
 

tests/integration/manual_test_temporal_integration.py

@@ -39,7 +39,7 @@
 from google.adk.sessions import InMemorySessionService
 from google.adk.utils.context_utils import Aclosing
 from google.adk.events import Event
-from google.adk.integrations.temporal import AdkWorkerPlugin, TemporalPlugin
+from google.adk.integrations.temporal import WorkerPlugin, AgentPlugin
 
 # Required Environment Variables for this test:
 # in this folder update .env.example to be .env and have the following vars:
@@ -68,11 +68,11 @@ async def run(self, prompt: str) -> Event | None:
         logger.info("Workflow started.")
 
         # 1. Define Agent using Temporal Helpers
-        # Note: TemporalPlugin in the Runner automatically handles Runtime setup
+        # Note: AgentPlugin in the Runner automatically handles Runtime setup
         # and Model Activity interception. We use standard ADK models now.
 
         # Wraps 'get_weather' activity as a Tool
-        weather_tool = TemporalPlugin.activity_tool(
+        weather_tool = AgentPlugin.activity_tool(
             get_weather,
             start_to_close_timeout=timedelta(seconds=60)
         )
@@ -90,12 +90,12 @@ async def run(self, prompt: str) -> Event | None:
 
         logger.info(f"Session created with ID: {session.id}")
 
-        # 3. Run Agent with TemporalPlugin
+        # 3. Run Agent with AgentPlugin
         runner = Runner(
             agent=agent,
             app_name='test_app',
             session_service=session_service,
-            plugins=[TemporalPlugin(activity_options={'start_to_close_timeout': timedelta(minutes=2)})]
+            plugins=[AgentPlugin(activity_options={'start_to_close_timeout': timedelta(minutes=2)})]
         )
 
         logger.info("Starting runner.")
@@ -154,7 +154,7 @@ async def run(self, topic: str) -> str:
             agent=coordinator, 
             app_name="multi_agent_app",
             session_service=session_service,
-            plugins=[TemporalPlugin(activity_options={'start_to_close_timeout': timedelta(minutes=2)})]
+            plugins=[AgentPlugin(activity_options={'start_to_close_timeout': timedelta(minutes=2)})]
         )
 
         # 4. Run
@@ -198,7 +198,7 @@ async def test_temporal_integration():
             get_weather, 
         ],
         workflows=[WeatherAgent, MultiAgentWorkflow],
-        plugins=[AdkWorkerPlugin()]
+        plugins=[WorkerPlugin()]
     ):
         print("Worker started.")
         # Test Weather Agent

tests/unittests/integrations/test_temporal.py

@@ -96,7 +96,7 @@ async def my_activity(arg: str) -> str:
             return f"Hello {arg}"
 
         # Wrap it
-        tool = temporal.TemporalPlugin.activity_tool(
+        tool = temporal.AgentPlugin.activity_tool(
              my_activity,
              start_to_close_timeout=100
         )
@@ -122,7 +122,7 @@ async def my_activity(arg: str) -> str:
 
 
     def test_temporal_plugin_before_model(self):
-        plugin = temporal.TemporalPlugin(activity_options={"start_to_close_timeout": 60})
+        plugin = temporal.AgentPlugin(activity_options={"start_to_close_timeout": 60})
 
         # Setup mocks
         mock_workflow.reset_mock()