agentex-sdk 0.4.12__py3-none-any.whl → 0.4.13__py3-none-any.whl

agentex/_constants.py

@@ -6,9 +6,9 @@ RAW_RESPONSE_HEADER = "X-Stainless-Raw-Response"
 OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
 
 # default timeout is 1 minute
-DEFAULT_TIMEOUT = httpx.Timeout(timeout=60, connect=5.0)
-DEFAULT_MAX_RETRIES = 2
-DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=100, max_keepalive_connections=20)
+DEFAULT_TIMEOUT = httpx.Timeout(timeout=300, connect=5.0)
+DEFAULT_MAX_RETRIES = 0
+DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=1000, max_keepalive_connections=1000)
 
 INITIAL_RETRY_DELAY = 0.5
 MAX_RETRY_DELAY = 8.0

agentex/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "agentex"
-__version__ = "0.4.12"  # x-release-please-version
+__version__ = "0.4.13"  # x-release-please-version

agentex/lib/adk/_modules/acp.py

@@ -59,6 +59,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> Task:
         """
         Create a new task.
@@ -71,6 +72,7 @@ class ACPModule:
             start_to_close_timeout: The start to close timeout for the task.
             heartbeat_timeout: The heartbeat timeout for the task.
             retry_policy: The retry policy for the task.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The task entry.
@@ -85,6 +87,7 @@ class ACPModule:
                     params=params,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=Task,
                 start_to_close_timeout=start_to_close_timeout,
@@ -99,6 +102,7 @@ class ACPModule:
                 params=params,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def send_event(
@@ -112,15 +116,22 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> Event:
         """
         Send an event to a task.
 
         Args:
             task_id: The ID of the task to send the event to.
-            data: The data to send to the event.
+            content: The content to send to the event.
             agent_id: The ID of the agent to send the event to.
             agent_name: The name of the agent to send the event to.
+            trace_id: The trace ID for the event.
+            parent_span_id: The parent span ID for the event.
+            start_to_close_timeout: The start to close timeout for the event.
+            heartbeat_timeout: The heartbeat timeout for the event.
+            retry_policy: The retry policy for the event.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The event entry.
@@ -135,6 +146,7 @@ class ACPModule:
                     content=content,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=None,
                 start_to_close_timeout=start_to_close_timeout,
@@ -149,6 +161,7 @@ class ACPModule:
                 content=content,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def send_message(
@@ -162,15 +175,22 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> List[TaskMessage]:
         """
         Send a message to a task.
 
         Args:
-            task_id: The ID of the task to send the message to.
             content: The task message content to send to the task.
+            task_id: The ID of the task to send the message to.
             agent_id: The ID of the agent to send the message to.
             agent_name: The name of the agent to send the message to.
+            trace_id: The trace ID for the message.
+            parent_span_id: The parent span ID for the message.
+            start_to_close_timeout: The start to close timeout for the message.
+            heartbeat_timeout: The heartbeat timeout for the message.
+            retry_policy: The retry policy for the message.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The message entry.
@@ -185,6 +205,7 @@ class ACPModule:
                     content=content,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=TaskMessage,
                 start_to_close_timeout=start_to_close_timeout,
@@ -199,6 +220,7 @@ class ACPModule:
                 content=content,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def cancel_task(
@@ -212,6 +234,7 @@ class ACPModule:
         start_to_close_timeout: timedelta = timedelta(seconds=5),
         heartbeat_timeout: timedelta = timedelta(seconds=5),
         retry_policy: RetryPolicy = DEFAULT_RETRY_POLICY,
+        request: dict[str, Any] | None = None,
     ) -> Task:
         """
         Cancel a task by sending cancel request to the agent that owns the task.
@@ -226,6 +249,7 @@ class ACPModule:
             start_to_close_timeout: The start to close timeout for the task.
             heartbeat_timeout: The heartbeat timeout for the task.
             retry_policy: The retry policy for the task.
+            request: Additional request context including headers to forward to the agent.
 
         Returns:
             The task entry.
@@ -244,6 +268,7 @@ class ACPModule:
                     agent_name=agent_name,
                     trace_id=trace_id,
                     parent_span_id=parent_span_id,
+                    request=request,
                 ),
                 response_type=None,
                 start_to_close_timeout=start_to_close_timeout,
@@ -258,4 +283,5 @@ class ACPModule:
                 agent_name=agent_name,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )

agentex/lib/core/services/adk/acp/acp.py

@@ -9,6 +9,10 @@ from agentex.types.task import Task
 from agentex.types.task_message import TaskMessage
 from agentex.types.task_message_content import TaskMessageContent
 from agentex.types.task_message_content_param import TaskMessageContentParam
+from agentex.types.agent_rpc_params import (
+    ParamsCancelTaskRequest as RpcParamsCancelTaskRequest,
+    ParamsSendEventRequest as RpcParamsSendEventRequest,
+)
 
 logger = make_logger(__name__)
 
@@ -30,6 +34,7 @@ class ACPService:
         params: dict[str, Any] | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        request: dict[str, Any] | None = None,
     ) -> Task:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
@@ -43,6 +48,10 @@ class ACPService:
             },
         ) as span:
             heartbeat_if_in_workflow("task create")
+            
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
             if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
                     agent_name=agent_name,
@@ -51,6 +60,7 @@ class ACPService:
                         "name": name,
                         "params": params,
                     },
+                    extra_headers=extra_headers,
                 )
             elif agent_id:
                 json_rpc_response = await self._agentex_client.agents.rpc(
@@ -60,6 +70,7 @@ class ACPService:
                         "name": name,
                         "params": params,
                     },
+                    extra_headers=extra_headers,
                 )
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
@@ -78,6 +89,7 @@ class ACPService:
         task_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        request: dict[str, Any] | None = None,
     ) -> List[TaskMessage]:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
@@ -92,6 +104,10 @@ class ACPService:
             },
         ) as span:
             heartbeat_if_in_workflow("message send")
+            
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
             if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
                     agent_name=agent_name,
@@ -101,6 +117,7 @@ class ACPService:
                         "content": cast(TaskMessageContentParam, content.model_dump()),
                         "stream": False,
                     },
+                    extra_headers=extra_headers,
                 )
             elif agent_id:
                 json_rpc_response = await self._agentex_client.agents.rpc(
@@ -111,12 +128,13 @@ class ACPService:
                         "content": cast(TaskMessageContentParam, content.model_dump()),
                         "stream": False,
                     },
+                    extra_headers=extra_headers,
                 )
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
 
             task_messages: List[TaskMessage] = []
-            logger.info(f"json_rpc_response: {json_rpc_response}")
+            logger.info("json_rpc_response: %s", json_rpc_response)
             if isinstance(json_rpc_response.result, list):
                 for message in json_rpc_response.result:
                     task_message = TaskMessage.model_validate(message)
@@ -137,6 +155,7 @@ class ACPService:
         task_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        request: dict[str, Any] | None = None,
     ) -> Event:
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
@@ -146,27 +165,33 @@ class ACPService:
                 "agent_id": agent_id,
                 "agent_name": agent_name,
                 "task_id": task_id,
+                "task_name": task_name,
                 "content": content,
             },
         ) as span:
             heartbeat_if_in_workflow("event send")
+            
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
+            rpc_event_params: RpcParamsSendEventRequest = {
+                "task_id": task_id,
+                "task_name": task_name,
+                "content": cast(TaskMessageContentParam, content.model_dump()),
+            }
             if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
                     agent_name=agent_name,
                     method="event/send",
-                    params={
-                        "task_id": task_id,
-                        "content": cast(TaskMessageContentParam, content.model_dump()),
-                    },
+                    params=rpc_event_params,
+                    extra_headers=extra_headers,
                 )
             elif agent_id:
                 json_rpc_response = await self._agentex_client.agents.rpc(
                     agent_id=agent_id,
                     method="event/send",
-                    params={
-                        "task_id": task_id,
-                        "content": cast(TaskMessageContentParam, content.model_dump()),
-                    },
+                    params=rpc_event_params,
+                    extra_headers=extra_headers,
                 )
             else:
                 raise ValueError("Either agent_name or agent_id must be provided")
@@ -184,7 +209,27 @@ class ACPService:
         agent_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
-    ) -> Task:        
+        request: dict[str, Any] | None = None,
+    ) -> Task:
+        """
+        Cancel a task by sending cancel request to the agent that owns the task.
+        
+        Args:
+            task_id: ID of the task to cancel (passed to agent in params)
+            task_name: Name of the task to cancel (passed to agent in params)  
+            agent_id: ID of the agent that owns the task
+            agent_name: Name of the agent that owns the task
+            trace_id: Trace ID for tracing
+            parent_span_id: Parent span ID for tracing
+            request: Additional request context including headers to forward to the agent
+            
+        Returns:
+            Task entry representing the cancelled task
+            
+        Raises:
+            ValueError: If neither agent_name nor agent_id is provided,
+                       or if neither task_name nor task_id is provided
+        """
         # Require agent identification
         if not agent_name and not agent_id:
             raise ValueError("Either agent_name or agent_id must be provided to identify the agent that owns the task")
@@ -192,7 +237,6 @@ class ACPService:
         # Require task identification
         if not task_name and not task_id:
             raise ValueError("Either task_name or task_id must be provided to identify the task to cancel")
-        
         trace = self._tracer.trace(trace_id=trace_id)
         async with trace.span(
             parent_id=parent_span_id,
@@ -206,8 +250,11 @@ class ACPService:
         ) as span:
             heartbeat_if_in_workflow("task cancel")
             
+            # Extract headers from request; pass-through to agent
+            extra_headers = request.get("headers") if request else None
+            
             # Build params for the agent (task identification)
-            params = {}
+            params: RpcParamsCancelTaskRequest = {}
             if task_id:
                 params["task_id"] = task_id
             if task_name:
@@ -219,12 +266,15 @@ class ACPService:
                     agent_name=agent_name,
                     method="task/cancel",
                     params=params,
+                    extra_headers=extra_headers,
                 )
             else:  # agent_id is provided (validated above)
+                assert agent_id is not None
                 json_rpc_response = await self._agentex_client.agents.rpc(
                     agent_id=agent_id,
                     method="task/cancel",
                     params=params,
+                    extra_headers=extra_headers,
                 )
 
             task_entry = Task.model_validate(json_rpc_response.result)

agentex/lib/core/temporal/activities/adk/acp/acp_activities.py

@@ -26,6 +26,7 @@ class TaskCreateParams(BaseModelWithTraceParams):
     agent_id: str | None = None
     agent_name: str | None = None
     params: dict[str, Any] | None = None
+    request: dict[str, Any] | None = None
 
 
 class MessageSendParams(BaseModelWithTraceParams):
@@ -33,6 +34,7 @@ class MessageSendParams(BaseModelWithTraceParams):
     agent_name: str | None = None
     task_id: str | None = None
     content: TaskMessageContent
+    request: dict[str, Any] | None = None
 
 
 class EventSendParams(BaseModelWithTraceParams):
@@ -40,6 +42,7 @@ class EventSendParams(BaseModelWithTraceParams):
     agent_name: str | None = None
     task_id: str | None = None
     content: TaskMessageContent
+    request: dict[str, Any] | None = None
 
 
 class TaskCancelParams(BaseModelWithTraceParams):
@@ -47,6 +50,7 @@ class TaskCancelParams(BaseModelWithTraceParams):
     task_name: str | None = None
     agent_id: str | None = None
     agent_name: str | None = None
+    request: dict[str, Any] | None = None
 
 
 class ACPActivities:
@@ -60,6 +64,9 @@ class ACPActivities:
             agent_id=params.agent_id,
             agent_name=params.agent_name,
             params=params.params,
+            trace_id=params.trace_id,
+            parent_span_id=params.parent_span_id,
+            request=params.request,
         )
 
     @activity.defn(name=ACPActivityName.MESSAGE_SEND)
@@ -69,6 +76,9 @@ class ACPActivities:
             agent_name=params.agent_name,
             task_id=params.task_id,
             content=params.content,
+            trace_id=params.trace_id,
+            parent_span_id=params.parent_span_id,
+            request=params.request,
         )
 
     @activity.defn(name=ACPActivityName.EVENT_SEND)
@@ -78,6 +88,9 @@ class ACPActivities:
             agent_name=params.agent_name,
             task_id=params.task_id,
             content=params.content,
+            trace_id=params.trace_id,
+            parent_span_id=params.parent_span_id,
+            request=params.request,
         )
 
     @activity.defn(name=ACPActivityName.TASK_CANCEL)
@@ -89,4 +102,5 @@ class ACPActivities:
             agent_name=params.agent_name,
             trace_id=params.trace_id,
             parent_span_id=params.parent_span_id,
+            request=params.request,
         )

agentex/lib/core/temporal/workers/worker.py

@@ -4,7 +4,7 @@ import os
 import uuid
 from collections.abc import Callable
 from concurrent.futures import ThreadPoolExecutor
-from typing import Any
+from typing import Any, overload
 
 from aiohttp import web
 from temporalio.client import Client
@@ -99,10 +99,28 @@ class AgentexWorker:
         self.healthy = False
         self.health_check_port = health_check_port
 
+    @overload
     async def run(
         self,
         activities: list[Callable],
+        *,
         workflow: type,
+    ) -> None: ...
+    
+    @overload
+    async def run(
+        self,
+        activities: list[Callable],
+        *,
+        workflows: list[type],
+    ) -> None: ...
+
+    async def run(
+        self,
+        activities: list[Callable],
+        *,
+        workflow: type | None = None,
+        workflows: list[type] | None = None,
     ):
         await self.start_health_check_server()
         await self._register_agent()
@@ -115,11 +133,14 @@ class AgentexWorker:
         if debug_enabled:
             logger.info("🐛 [WORKER] Temporal debug mode enabled - deadlock detection disabled")
         
+        if workflow is None and workflows is None:
+            raise ValueError("Either workflow or workflows must be provided")
+        
         worker = Worker(
             client=temporal_client,
             task_queue=self.task_queue,
             activity_executor=ThreadPoolExecutor(max_workers=self.max_workers),
-            workflows=[workflow],
+            workflows=[workflow] if workflows is None else workflows,
             activities=activities,
             workflow_runner=UnsandboxedWorkflowRunner(),
             max_concurrent_activities=self.max_concurrent_activities,

agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -1,5 +1,6 @@
 import asyncio
 import inspect
+from datetime import datetime
 from collections.abc import AsyncGenerator, Awaitable, Callable
 from contextlib import asynccontextmanager
 from typing import Any
@@ -26,6 +27,10 @@ from agentex.types.task_message_content import TaskMessageContent
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.model_utils import BaseModel
 from agentex.lib.utils.registration import register_agent
+from agentex.lib.sdk.fastacp.base.constants import (
+    FASTACP_HEADER_SKIP_EXACT,
+    FASTACP_HEADER_SKIP_PREFIXES,
+)
 
 logger = make_logger(__name__)
 
@@ -93,6 +98,7 @@ class BaseACPServer(FastAPI):
     async def _handle_jsonrpc(self, request: Request):
         """Main JSON-RPC endpoint handler"""
         rpc_request = None
+        logger.info(f"[base_acp_server] received request: {datetime.now()}")
         try:
             data = await request.json()
             rpc_request = JSONRPCRequest(**data)
@@ -128,9 +134,23 @@ class BaseACPServer(FastAPI):
                     ),
                 )
 
-            # Parse params into appropriate model based on method
+            # Extract application headers, excluding sensitive/transport headers per FASTACP_* rules
+            # Forward filtered headers via params.request.headers to agent handlers
+            custom_headers = {
+                key: value
+                for key, value in request.headers.items()
+                if key.lower() not in FASTACP_HEADER_SKIP_EXACT
+                and not any(key.lower().startswith(p) for p in FASTACP_HEADER_SKIP_PREFIXES)
+            }
+            
+            # Parse params into appropriate model based on method and include headers
             params_model = PARAMS_MODEL_BY_METHOD[method]
-            params = params_model.model_validate(rpc_request.params)
+            params_data = dict(rpc_request.params) if rpc_request.params else {}
+            
+            # Add custom headers to the request structure if any headers were provided
+            if custom_headers:
+                params_data["request"] = {"headers": custom_headers}
+            params = params_model.model_validate(params_data)
 
             if method in RPC_SYNC_METHODS:
                 handler = self._handlers[method]

agentex/lib/sdk/fastacp/base/constants.py

@@ -0,0 +1,24 @@
+from __future__ import annotations
+
+# Header filtering rules for FastACP server
+
+# Prefixes to skip (case-insensitive beginswith checks)
+FASTACP_HEADER_SKIP_PREFIXES: tuple[str, ...] = (
+    "content-",
+    "host",
+    "user-agent",
+    "x-forwarded-",
+    "sec-",
+)
+
+# Exact header names to skip (case-insensitive matching done by lowercasing keys)
+FASTACP_HEADER_SKIP_EXACT: set[str] = {
+    "x-agent-api-key",
+    "connection",
+    "accept-encoding",
+    "cookie",
+    "content-length",
+    "transfer-encoding",
+}
+
+

agentex/lib/types/acp.py

@@ -25,6 +25,7 @@ class CreateTaskParams(BaseModel):
         agent: The agent that the task was sent to.
         task: The task to be created.
         params: The parameters for the task as inputted by the user.
+        request: Additional request context including headers forwarded to this agent.
     """
 
     agent: Agent = Field(..., description="The agent that the task was sent to")
@@ -33,6 +34,10 @@ class CreateTaskParams(BaseModel):
         None,
         description="The parameters for the task as inputted by the user",
     )
+    request: dict[str, Any] | None = Field(
+        default=None,
+        description="Additional request context including headers forwarded to this agent",
+    )
 
 
 class SendMessageParams(BaseModel):
@@ -43,6 +48,7 @@ class SendMessageParams(BaseModel):
         task: The task that the message was sent to.
         content: The message that was sent to the agent.
         stream: Whether to stream the message back to the agentex server from the agent.
+        request: Additional request context including headers forwarded to this agent.
     """
 
     agent: Agent = Field(..., description="The agent that the message was sent to")
@@ -54,6 +60,10 @@ class SendMessageParams(BaseModel):
         False,
         description="Whether to stream the message back to the agentex server from the agent",
     )
+    request: dict[str, Any] | None = Field(
+        default=None,
+        description="Additional request context including headers forwarded to this agent",
+    )
 
 
 class SendEventParams(BaseModel):
@@ -63,11 +73,16 @@ class SendEventParams(BaseModel):
         agent: The agent that the event was sent to.
         task: The task that the message was sent to.
         event: The event that was sent to the agent.
+        request: Additional request context including headers forwarded to this agent.
     """
 
     agent: Agent = Field(..., description="The agent that the event was sent to")
     task: Task = Field(..., description="The task that the message was sent to")
     event: Event = Field(..., description="The event that was sent to the agent")
+    request: dict[str, Any] | None = Field(
+        default=None,
+        description="Additional request context including headers forwarded to this agent",
+    )
 
 
 class CancelTaskParams(BaseModel):
@@ -76,10 +91,15 @@ class CancelTaskParams(BaseModel):
     Attributes:
         agent: The agent that the task was sent to.
         task: The task that was cancelled.
+        request: Additional request context including headers forwarded to this agent.
     """
 
     agent: Agent = Field(..., description="The agent that the task was sent to")
     task: Task = Field(..., description="The task that was cancelled")
+    request: dict[str, Any] | None = Field(
+        default=None,
+        description="Additional request context including headers forwarded to this agent",
+    )
 
 
 RPC_SYNC_METHODS = [

agentex/resources/agents.py

@@ -7,6 +7,7 @@ from typing import AsyncGenerator, Generator, Union, Optional
 from typing_extensions import Literal
 
 import httpx
+from pydantic import ValidationError
 
 from ..types import agent_rpc_params, agent_list_params, agent_rpc_by_name_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
@@ -1073,6 +1074,8 @@ class AsyncAgentsResource(AsyncAPIResource):
           except json.JSONDecodeError:
             # Skip invalid JSON lines
             continue
+          except ValidationError:
+            raise ValueError(f"Invalid SendMessageStreamResponse returned: {line}")
     
     async def send_event(
       self,

agentex/resources/tasks.py

@@ -232,7 +232,7 @@ class TasksResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Stream[object]:
         """
-        Stream message updates for a task by its unique ID.
+        Stream events for a task by its unique ID.
 
         Args:
           extra_headers: Send extra headers
@@ -267,7 +267,7 @@ class TasksResource(SyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> Stream[object]:
         """
-        Stream message updates for a task by its unique name.
+        Stream events for a task by its unique name.
 
         Args:
           extra_headers: Send extra headers
@@ -497,7 +497,7 @@ class AsyncTasksResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> AsyncStream[object]:
         """
-        Stream message updates for a task by its unique ID.
+        Stream events for a task by its unique ID.
 
         Args:
           extra_headers: Send extra headers
@@ -532,7 +532,7 @@ class AsyncTasksResource(AsyncAPIResource):
         timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
     ) -> AsyncStream[object]:
         """
-        Stream message updates for a task by its unique name.
+        Stream events for a task by its unique name.
 
         Args:
           extra_headers: Send extra headers

agentex/types/agent.py

@@ -1,6 +1,6 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
-from typing import Optional
+from typing import Dict, Optional
 from datetime import datetime
 from typing_extensions import Literal
 
@@ -29,6 +29,12 @@ class Agent(BaseModel):
     updated_at: datetime
     """The timestamp when the agent was last updated"""
 
+    registered_at: Optional[datetime] = None
+    """The timestamp when the agent was last registered"""
+
+    registration_metadata: Optional[Dict[str, object]] = None
+    """The metadata for the agent's registration."""
+
     status: Optional[Literal["Pending", "Building", "Ready", "Failed", "Unknown"]] = None
     """The status of the action, indicating if it's building, ready, failed, etc."""
 

agentex/types/task.py

@@ -22,4 +22,6 @@ class Task(BaseModel):
 
     status_reason: Optional[str] = None
 
+    task_metadata: Optional[Dict[str, object]] = None
+
     updated_at: Optional[datetime] = None

{agentex_sdk-0.4.12.dist-info → agentex_sdk-0.4.13.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.12
+Version: 0.4.13
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/agentex-python
 Project-URL: Repository, https://github.com/scaleapi/agentex-python

{agentex_sdk-0.4.12.dist-info → agentex_sdk-0.4.13.dist-info}/RECORD

@@ -2,7 +2,7 @@ agentex/__init__.py,sha256=j0BX5IfIjYaWdXQC7P4bh1Ev2-0bJe4Uh7IbqG551ng,2667
 agentex/_base_client.py,sha256=Y_EaL4x9g7EsRADVD6vJ303tLO0iyB3K5SbTSvKCS2k,67048
 agentex/_client.py,sha256=Fae6qw42eIjFaENpQ5nrq5SyW1310UoHZGNrG-DSa_g,20493
 agentex/_compat.py,sha256=DQBVORjFb33zch24jzkhM14msvnzY7mmSmgDLaVFUM8,6562
-agentex/_constants.py,sha256=S14PFzyN9-I31wiV7SmIlL5Ga0MLHxdvegInGdXH7tM,462
+agentex/_constants.py,sha256=UWL_R-mDWXfDVTMoEAks7F9dNEbr7NFc27WQShkBB1c,466
 agentex/_exceptions.py,sha256=B09aFjWFRSShb9BFJd-MNDblsGDyGk3w-vItYmjg_AI,3222
 agentex/_files.py,sha256=KnEzGi_O756MvKyJ4fOCW_u3JhOeWPQ4RsmDvqihDQU,3545
 agentex/_models.py,sha256=c29x_mRccdxlGwdUPfSR5eJxGXe74Ea5Dje5igZTrKQ,30024
@@ -11,7 +11,7 @@ agentex/_resource.py,sha256=S1t7wmR5WUvoDIhZjo_x-E7uoTJBynJ3d8tPJMQYdjw,1106
 agentex/_response.py,sha256=Tb9zazsnemO2rTxWtBjAD5WBqlhli5ZaXGbiKgdu5DE,28794
 agentex/_streaming.py,sha256=FNGJExRCF-vTRUZHFKUfoAWFhDGOB3XbioVCF37Jr7E,10104
 agentex/_types.py,sha256=lO491FSd7vM_uBp7-TvItbauEAH8SsEPYcyNO_5lKGM,7297
-agentex/_version.py,sha256=mc7enXMgZrtxOdwgC3XRWiyCc9frvcY3bAC3oYmJEis,160
+agentex/_version.py,sha256=N3dLunb4wcl7SK9s-2gSJpVddKvBJDMxT7Pz_dlMd8c,160
 agentex/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/_utils/__init__.py,sha256=7fch0GT9zpNnErbciSpUNa-SjTxxjY6kxHxKMOM4AGs,2305
 agentex/_utils/_compat.py,sha256=D8gtAvjJQrDWt9upS0XaG9Rr5l1QhiAx_I_1utT_tt0,1195
@@ -31,7 +31,7 @@ agentex/lib/environment_variables.py,sha256=6iWpCEURbahnLxZIjvJGES-zs7euUI8bxVXF
 agentex/lib/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/adk/__init__.py,sha256=-PpVfEvYr_HD7TnxUWU8RCW2OnxfwpPxTW97dKTnqvI,1082
 agentex/lib/adk/_modules/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/adk/_modules/acp.py,sha256=_v71srFcyuNbR1CCg4xC0iyf6G7kYD0xM48RfZmJULg,9699
+agentex/lib/adk/_modules/acp.py,sha256=_LweV8BoIyQhqL06Xie_9t_0jcbZakomb24MgojYneo,11179
 agentex/lib/adk/_modules/agent_task_tracker.py,sha256=fXMDs1zgAC6n8EmWTU5_trs79EGVBCEPxd5eBOZmdE0,7022
 agentex/lib/adk/_modules/agents.py,sha256=VlT_PXnyb3Knh-nx1rfflD9N6gjp5O0_6Z_QL5SCThU,2788
 agentex/lib/adk/_modules/events.py,sha256=RHQZ4ircC60zdm1chRzo3cdt7LNqDoJfPLNJXEz4BH0,5262
@@ -131,7 +131,7 @@ agentex/lib/core/services/adk/streaming.py,sha256=keZyy7CixM2_3tf2THv-X4gGOOSAk0
 agentex/lib/core/services/adk/tasks.py,sha256=XEX0u4skq81Fn-gjTzSE_-Iv_mKySZHchpotxAsnJwg,2618
 agentex/lib/core/services/adk/tracing.py,sha256=w4mnIoq5m5iwlkDN_NOUjhi6hUpD958cYn1QR6hFgB8,1156
 agentex/lib/core/services/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/services/adk/acp/acp.py,sha256=re6GfuivaYiNXMUxA1_rmoAKu5rrzxK2cB_YquYudos,8692
+agentex/lib/core/services/adk/acp/acp.py,sha256=ONbGc8HWNCxv5IJaItY-fvDZIwlh3d4jSmH9GGQo7q8,10971
 agentex/lib/core/services/adk/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/services/adk/providers/litellm.py,sha256=EKzus_xohNW-85V5hwvd1WqUd3ebv2wc9vDIWO2t1Mw,10044
 agentex/lib/core/services/adk/providers/openai.py,sha256=ekxh5fvsuf_WMe-gyGYmECyfjxdvZVm-dU-ndyLzFGs,52519
@@ -151,7 +151,7 @@ agentex/lib/core/temporal/activities/adk/streaming_activities.py,sha256=PEXeFU9m
 agentex/lib/core/temporal/activities/adk/tasks_activities.py,sha256=hQjdD-TfSzxsiW9eiJ5ZTvY0Ieu0tE950_Zk2PUVc9g,1442
 agentex/lib/core/temporal/activities/adk/tracing_activities.py,sha256=RF8nSBA9x6Ui7_MD9He5ljlH_xZduIhx_JUTEyKdCxA,1523
 agentex/lib/core/temporal/activities/adk/acp/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/temporal/activities/adk/acp/acp_activities.py,sha256=AkUx5ADDPZrAb0wL7lBRvwNW2zi2TNJVBGPD8AJz14M,2927
+agentex/lib/core/temporal/activities/adk/acp/acp_activities.py,sha256=FrIs0g4BqB5NwyHmb-3Do_ugGWgLhQ3Uoud1E0BnD_E,3503
 agentex/lib/core/temporal/activities/adk/providers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/temporal/activities/adk/providers/litellm_activities.py,sha256=Ix92XZIQIbhf-_fOWkUZDxZ1UUCOF0kOwxbcRYBzBQA,2583
 agentex/lib/core/temporal/activities/adk/providers/openai_activities.py,sha256=JmP1rNsVjvf9flM9ZyaFlfOsx-aQNZ4muXoYxZVqKfk,27349
@@ -163,7 +163,7 @@ agentex/lib/core/temporal/services/temporal_task_service.py,sha256=oxaegtUMfr3EC
 agentex/lib/core/temporal/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/core/temporal/types/workflow.py,sha256=o8lBUloI44NTYFfbA1BLgzUneyN7aLbt042Eq_9OKo8,89
 agentex/lib/core/temporal/workers/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/core/temporal/workers/worker.py,sha256=kmdfs67UWjzxU9ZOE7vq_q4S4jj7jB5ZMyJkZYptrAY,6670
+agentex/lib/core/temporal/workers/worker.py,sha256=XumHnctexRP3Ir7pXTid3VKaHs5vysYQO6UcCet0yvg,7211
 agentex/lib/core/temporal/workflows/workflow.py,sha256=VTS4nqwdJqqjIy6nTyvFLdAoBHnunO8-8rVY0Gx5_zo,723
 agentex/lib/core/tracing/__init__.py,sha256=3dLjXGTwAJxi_xxwdh8Mt04pdLyzN9VxAl8XHmV70Ak,229
 agentex/lib/core/tracing/trace.py,sha256=45Nn0Z97kC_9XNyb9N8IWUFo0OIdb7dDZShhgOTGJB0,8952
@@ -184,7 +184,8 @@ agentex/lib/sdk/config/project_config.py,sha256=CGH_r9KbnSFMj2CnBkZnfg41L2o0TeVN
 agentex/lib/sdk/config/validation.py,sha256=QGAlAzlVJiWRlIksqxNS-JSwkk8Z4gXMSFUJc4qPrIQ,8989
 agentex/lib/sdk/fastacp/__init__.py,sha256=UvAdexdnfb4z0F4a2sfXROFyh9EjH89kf3AxHPybzCM,75
 agentex/lib/sdk/fastacp/fastacp.py,sha256=K4D7a9EiJfCgWp2hrE_TbpZBaF4Uc46MfndZK3F9mA0,3061
-agentex/lib/sdk/fastacp/base/base_acp_server.py,sha256=vj7jPg8a8z07dTl3P-ZVTdhcuW3ulm4QrptWyYVnTAs,14598
+agentex/lib/sdk/fastacp/base/base_acp_server.py,sha256=mVi8uSOZtf70rogS7xDJwHU1hF_5Fv7S8383rUN0kK8,15604
+agentex/lib/sdk/fastacp/base/constants.py,sha256=W4vpJ-5NML7239JyqzUWdu2IypIl8Cey8CS41KR2Vk0,519
 agentex/lib/sdk/fastacp/impl/agentic_base_acp.py,sha256=LWLAlHrs-2Lc2UICBAEFL8c3JwTA6oxPnzUzW0qQWSA,2694
 agentex/lib/sdk/fastacp/impl/sync_acp.py,sha256=0y_cYD-0UJOiVJv-BBMOt6PvElDf5zmc1uvbr81VcSI,3897
 agentex/lib/sdk/fastacp/impl/temporal_acp.py,sha256=hQa_zGG6l_jIHNC75H_KXlPUuVqNBc-zIm-pmCDSZ6w,3759
@@ -203,7 +204,7 @@ agentex/lib/sdk/state_machine/state_workflow.py,sha256=ispqxlcLPT22glgOBXjyd-BXC
 agentex/lib/sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 agentex/lib/sdk/utils/messages.py,sha256=GdfT98NQRmltAnCQ2XW0XF5DO2jGi6TEZh8qqNv6WlY,8354
 agentex/lib/types/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-agentex/lib/types/acp.py,sha256=lFuWZwlXy1TNUHGvCXmyPLPbt8ME_2lejdWwx8VcodY,2970
+agentex/lib/types/acp.py,sha256=YYTXsJWcw6BdIWXliKQxAwtHn07ZINQ-RsppyjY9_64,3974
 agentex/lib/types/agent_configs.py,sha256=3wRa2okimSzi2v8sJyMyrqRlcu_4sxes-_4smEK6fq8,2798
 agentex/lib/types/agent_results.py,sha256=ev6WnPLfZRbhy2HnBmdIrZq1ExVeaweXoLRxYGspyWM,653
 agentex/lib/types/converters.py,sha256=u6fLb0rBUDA6nF5hdbC8ms6H-Z21IQfLlIvYpau_P5g,2283
@@ -230,18 +231,18 @@ agentex/lib/utils/temporal.py,sha256=sXo8OPMMXiyrF7OSBCJBuN_ufyQOD2bLOXgDbVZoyds
 agentex/lib/utils/dev_tools/__init__.py,sha256=oaHxw6ymfhNql-kzXHv3NWVHuqD4fHumasNXJG7kHTU,261
 agentex/lib/utils/dev_tools/async_messages.py,sha256=dM01spww2Fy6ej_WDnPXs2eG-9yCU7Xc1IMKwJzq6cM,19543
 agentex/resources/__init__.py,sha256=74rMqWBzQ2dSrKQqsrd7-jskPws0O_ogkFltvZO3HoU,3265
-agentex/resources/agents.py,sha256=WLG7wTQcCZCtqvNiyi2wjuDA2u85nEA5o9kUr5pBQXE,46977
+agentex/resources/agents.py,sha256=IvnUuQZ_cpP6Kz96aD6MSB0AQ5-VLo_L-VIs6MVSu1s,47132
 agentex/resources/events.py,sha256=Zc9JhUm3bq2VFnBAolC0M7KZernzj1AjZ_vj0ibP4GY,10412
 agentex/resources/spans.py,sha256=wmcUs4XbXIF5rPeyU_f39c2RTbTLnkuh2LYogZEBD6s,20936
 agentex/resources/states.py,sha256=O31A8--n7n0rHsng2e1oCUAzLNjQIxDUk7rq0IXfgGM,19262
-agentex/resources/tasks.py,sha256=R-pfvSt6W9n0QxPyxrgY6nTSG20YtkH_NXgXPNbnRuo,24780
+agentex/resources/tasks.py,sha256=ovMDt0EElGYwvhYdEiC0Noo04p0GHPWZTmDub-drl88,24744
 agentex/resources/tracker.py,sha256=YxSeiloMwIqrS9nY7SlHclauRA7142qrKw34xgwqYbA,14103
 agentex/resources/messages/__init__.py,sha256=_J1eusFtr_k6zrAntJSuqx6LWEUBSTrV1OZZh7MaDPE,1015
 agentex/resources/messages/batch.py,sha256=pegCmnjK_J0jek5ChX1pKpq5RmCucTYLbK69H6qGVS4,9629
 agentex/resources/messages/messages.py,sha256=fuFmmGNOjRJVzmYHcfP2trg0yii0n9MPPCpt7F8fDs4,17930
 agentex/types/__init__.py,sha256=lEuWxr1ggBzZAUGYrSmIhcSSZ7z5UQSIfNtmUrELC2s,3884
 agentex/types/acp_type.py,sha256=Fj-4SzmM6m95ck_ZXtNbcWggHiD9F49bxBLPbl1fxe4,208
-agentex/types/agent.py,sha256=t6CUsKSaK6Gs_X9g0YxSi5G3Hh21FrGVGnAD6MytVBk,981
+agentex/types/agent.py,sha256=ihRNn9mksULA5FuiyanKA7B-lmN4OVr6wBqNzUb8Tcg,1208
 agentex/types/agent_list_params.py,sha256=81IWnRZ2rLfHH7GB6VkXShYjb44DO0guG1znJcV3tuI,316
 agentex/types/agent_list_response.py,sha256=g0b9Cw7j2yk14veyJORpF3me8iW7g7pr2USpXGokoFI,254
 agentex/types/agent_rpc_by_name_params.py,sha256=G6xkjrZKPmJvhwqgc68tAnzmb4wYh9k0zlcm9CsLQR0,2024
@@ -275,7 +276,7 @@ agentex/types/state_create_params.py,sha256=dIQLVdFJyhIUKlFcm-7I6M4qD1BxZwH8KCt1
 agentex/types/state_list_params.py,sha256=jsBeE98mVvS-pk9iytNTVVSW6pz7OM4Zt-OxC2mEt3E,364
 agentex/types/state_list_response.py,sha256=3UyMRzP3zdEKo9hTkJdvBHjmv6II4KnwoZ8GvlzPCSI,254
 agentex/types/state_update_params.py,sha256=TpCXMrYaT2MWeOPng9-RGvGX9vE61Te9r3CFRQzzIDg,377
-agentex/types/task.py,sha256=GDPXgF2OwRCA-XFsMJpsVbUDwI99sZDN7aoWeIr00iA,597
+agentex/types/task.py,sha256=PXFMFK74V_fsNnXN8vJW45FwRcXJcxDR-cQU8tNtk5Y,652
 agentex/types/task_list_params.py,sha256=euvsWpK2kjJOfvrkUnSFiofZlU2sZHd2mar3qYtAA8I,328
 agentex/types/task_list_response.py,sha256=8Q-RIanLmUC9vOuDXoW5gjpZeE-HR6IrBugG7-cUAZQ,249
 agentex/types/task_message.py,sha256=hoLAkiQJd1Fl7EjLof-vZq6zVnDw9SKmnV6V74Po58A,918
@@ -303,8 +304,8 @@ agentex/types/messages/batch_update_params.py,sha256=Ug5CThbD49a8j4qucg04OdmVrp_
 agentex/types/messages/batch_update_response.py,sha256=TbSBe6SuPzjXXWSj-nRjT1JHGBooTshHQQDa1AixQA8,278
 agentex/types/shared/__init__.py,sha256=IKs-Qn5Yja0kFh1G1kDqYZo43qrOu1hSoxlPdN-85dI,149
 agentex/types/shared/delete_response.py,sha256=8qH3zvQXaOHYQSHyXi7UQxdR4miTzR7V9K4zXVsiUyk,215
-agentex_sdk-0.4.12.dist-info/METADATA,sha256=8y1JRtsEw-RXbWLnDVo-VK8bA6qXBH3fNYxZz8W7oDM,15095
-agentex_sdk-0.4.12.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
-agentex_sdk-0.4.12.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
-agentex_sdk-0.4.12.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
-agentex_sdk-0.4.12.dist-info/RECORD,,
+agentex_sdk-0.4.13.dist-info/METADATA,sha256=hsdN9UD0NdPOxgYvhKFYuLydspi6odLpQnTN6tVFzmM,15095
+agentex_sdk-0.4.13.dist-info/WHEEL,sha256=C2FUgwZgiLbznR-k0b_5k3Ai_1aASOXDss3lzCUsUug,87
+agentex_sdk-0.4.13.dist-info/entry_points.txt,sha256=V7vJuMZdF0UlvgX6KiBN7XUvq_cxF5kplcYvc1QlFaQ,62
+agentex_sdk-0.4.13.dist-info/licenses/LICENSE,sha256=Q1AOx2FtRcMlyMgQJ9eVN2WKPq2mQ33lnB4tvWxabLA,11337
+agentex_sdk-0.4.13.dist-info/RECORD,,