more typing, comment

Author: eberki-scale

src/agentex/lib/sdk/fastacp/impl/temporal_acp.py

@@ -4,6 +4,7 @@
 from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
+from temporalio.converter import PayloadCodec
 
 from agentex.lib.types.acp import (
     SendEventParams,
@@ -31,7 +32,7 @@ def __init__(
         temporal_task_service: TemporalTaskService | None = None,
         plugins: list[Any] | None = None,
         interceptors: list[Any] | None = None,
-        payload_codec: Any | None = None,
+        payload_codec: PayloadCodec | None = None,
     ):
         super().__init__()
         self._temporal_task_service = temporal_task_service
@@ -42,7 +43,7 @@ def __init__(
 
     @classmethod
     @override
-    def create(cls, temporal_address: str, plugins: list[Any] | None = None, interceptors: list[Any] | None = None, payload_codec: Any | None = None) -> "TemporalACP":
+    def create(cls, temporal_address: str, plugins: list[Any] | None = None, interceptors: list[Any] | None = None, payload_codec: PayloadCodec | None = None) -> "TemporalACP":
         logger.info("Initializing TemporalACP instance")
 
         # Create instance without temporal client initially

src/agentex/lib/types/fastacp.py

@@ -50,6 +50,11 @@ class TemporalACPConfig(AsyncACPConfig):
         temporal_address: The address of the temporal server
         plugins: List of Temporal client plugins
         interceptors: List of Temporal worker interceptors
+        payload_codec: Optional ``temporalio.converter.PayloadCodec`` for
+            encoding/decoding payloads (e.g. encryption, compression). NOTE:
+            this only configures the ACP (client) side. The worker side must
+            be configured separately via ``AgentexWorker(payload_codec=...)``
+            with the SAME codec, or decode will fail at runtime.
     """
 
     type: Literal["temporal"] = Field(default="temporal", frozen=True)