improve typing

Author: eberki-scale

src/agentex/lib/core/clients/temporal/temporal_client.py

@@ -7,6 +7,7 @@
 from temporalio.client import Client, WorkflowExecutionStatus
 from temporalio.common import RetryPolicy as TemporalRetryPolicy, WorkflowIDReusePolicy
 from temporalio.service import RPCError, RPCStatusCode
+from temporalio.converter import PayloadCodec
 
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.model_utils import BaseModel
@@ -76,7 +77,7 @@
 
 
 class TemporalClient:
-    def __init__(self, temporal_client: Client | None = None, plugins: list[Any] = [], payload_codec: Any | None = None):
+    def __init__(self, temporal_client: Client | None = None, plugins: list[Any] = [], payload_codec: PayloadCodec | None = None):
         self._client: Client | None = temporal_client
         self._plugins = plugins
         self._payload_codec = payload_codec
@@ -89,7 +90,7 @@ def client(self) -> Client:
         return self._client
 
     @classmethod
-    async def create(cls, temporal_address: str, plugins: list[Any] = [], payload_codec: Any | None = None):
+    async def create(cls, temporal_address: str, plugins: list[Any] = [], payload_codec: PayloadCodec | None = None):
         if temporal_address in [
             "false",
             "False",