Add Strands Agents plugin (contrib)

.github/CODEOWNERS

@@ -6,13 +6,16 @@
 
 
 # Below are owners for modules in the temporalio/contrib/
-# and tests/contrib/ directories that are owned by teams 
-# other than the SDK team. For each one, we add the owning team, 
+# and tests/contrib/ directories that are owned by teams
+# other than the SDK team. For each one, we add the owning team,
 # as well as @temporalio/sdk, so the SDK team can continue to
 # manage repo-wide concerns.
+/temporalio/contrib/common/                 @temporalio/ai-sdk     @temporalio/sdk
 /temporalio/contrib/google_adk_agents/      @temporalio/ai-sdk     @temporalio/sdk
 /temporalio/contrib/langsmith/              @temporalio/ai-sdk     @temporalio/sdk
 /temporalio/contrib/openai_agents/          @temporalio/ai-sdk     @temporalio/sdk
+/temporalio/contrib/strands/                @temporalio/ai-sdk     @temporalio/sdk
 /tests/contrib/google_adk_agents/           @temporalio/ai-sdk     @temporalio/sdk
 /tests/contrib/langsmith/                   @temporalio/ai-sdk     @temporalio/sdk
 /tests/contrib/openai_agents/               @temporalio/ai-sdk     @temporalio/sdk
+/tests/contrib/strands/                     @temporalio/ai-sdk     @temporalio/sdk

pyproject.toml

@@ -40,6 +40,7 @@ lambda-worker-otel = [
   "opentelemetry-sdk-extension-aws>=2.0.0,<3",
 ]
 aioboto3 = ["aioboto3>=10.4.0", "types-aioboto3[s3]>=10.4.0"]
+strands-agents = ["strands-agents>=1.39.0"]
 
 [project.urls]
 Homepage = "https://github.com/temporalio/sdk-python"
@@ -85,6 +86,8 @@ dev = [
   "opentelemetry-sdk-extension-aws>=2.0.0,<3",
   "pytest-flakefinder>=1.1.0",
   "async-timeout>=4.0,<6; python_version < '3.11'",
+  "strands-agents>=1.39.0",
+  "strands-agents-tools>=0.5.2",
 ]
 
 [tool.poe.tasks]

temporalio/contrib/openai_agents/_heartbeat_decorator.py

@@ -8,23 +8,22 @@
 F = TypeVar("F", bound=Callable[..., Awaitable[Any]])
 
 
-def _auto_heartbeater(fn: F) -> F:  # type:ignore[reportUnusedClass]
-    # Propagate type hints from the original callable.
+def auto_heartbeater(fn: F) -> F:
+    """Decorator that heartbeats at half the activity's heartbeat timeout."""
+
     @wraps(fn)
     async def wrapper(*args: Any, **kwargs: Any) -> Any:
         heartbeat_timeout = activity.info().heartbeat_timeout
         heartbeat_task = None
         if heartbeat_timeout:
-            # Heartbeat twice as often as the timeout
             heartbeat_task = asyncio.create_task(
-                heartbeat_every(heartbeat_timeout.total_seconds() / 2)
+                _heartbeat_every(heartbeat_timeout.total_seconds() / 2)
             )
         try:
             return await fn(*args, **kwargs)
         finally:
             if heartbeat_task:
                 heartbeat_task.cancel()
-                # Wait for heartbeat cancellation to complete
                 try:
                     await heartbeat_task
                 except asyncio.CancelledError:
@@ -33,8 +32,7 @@ async def wrapper(*args: Any, **kwargs: Any) -> Any:
     return cast(F, wrapper)
 
 
-async def heartbeat_every(delay: float, *details: Any) -> None:
-    """Heartbeat every so often while not cancelled"""
+async def _heartbeat_every(delay: float) -> None:
     while True:
         await asyncio.sleep(delay)
-        activity.heartbeat(*details)
+        activity.heartbeat()

temporalio/contrib/openai_agents/_invoke_model_activity.py

@@ -43,7 +43,7 @@
 from typing_extensions import Required, TypedDict
 
 from temporalio import activity
-from temporalio.contrib.openai_agents._heartbeat_decorator import _auto_heartbeater
+from temporalio.contrib.openai_agents._heartbeat_decorator import auto_heartbeater
 from temporalio.contrib.workflow_streams import WorkflowStreamClient
 from temporalio.exceptions import ApplicationError
 
@@ -314,7 +314,7 @@ def __init__(self, model_provider: ModelProvider | None = None):
         )
 
     @activity.defn
-    @_auto_heartbeater
+    @auto_heartbeater
     async def invoke_model_activity(self, input: ActivityModelInput) -> ModelResponse:
         """Activity that invokes a model with the given input."""
         model = self._model_provider.get_model(input.get("model_name"))
@@ -337,7 +337,7 @@ async def invoke_model_activity(self, input: ActivityModelInput) -> ModelRespons
             _raise_for_openai_status(e)
 
     @activity.defn
-    @_auto_heartbeater
+    @auto_heartbeater
     async def invoke_model_activity_streaming(
         self, input: StreamingActivityModelInput
     ) -> list[TResponseStreamEvent]:
@@ -357,7 +357,7 @@ async def invoke_model_activity_streaming(
         ``streaming_topic`` so external consumers (UIs, tracing,
         etc.) can observe events as they arrive.
 
-        Heartbeats run on a background task via ``_auto_heartbeater`` so
+        Heartbeats run on a background task via ``auto_heartbeater`` so
         long initial-token latency or long pauses between chunks do not
         trip ``heartbeat_timeout``.
         """