agentex-sdk 0.4.11__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.11"  # 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,32 +220,43 @@ class ACPModule:
                 content=content,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )
 
     async def cancel_task(
         self,
         task_id: str | None = None,
         task_name: str | None = None,
+        agent_id: str | None = None,
+        agent_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
         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.
+        Cancel a task by sending cancel request to the agent that owns the task.
 
         Args:
-            task_id: The ID of the task to cancel.
-            task_name: The name of the task to cancel.
+            task_id: ID of the task to cancel.
+            task_name: Name of the task to cancel.
+            agent_id: ID of the agent that owns the task.
+            agent_name: Name of the agent that owns the task.
             trace_id: The trace ID for the task.
             parent_span_id: The parent span ID for the task.
             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.
+            
+        Raises:
+            ValueError: If neither agent_name nor agent_id is provided,
+                       or if neither task_name nor task_id is provided
         """
         if in_temporal_workflow():
             return await ActivityHelpers.execute_activity(
@@ -232,8 +264,11 @@ class ACPModule:
                 request=TaskCancelParams(
                     task_id=task_id,
                     task_name=task_name,
+                    agent_id=agent_id,
+                    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,
@@ -244,6 +279,9 @@ class ACPModule:
             return await self._acp_service.task_cancel(
                 task_id=task_id,
                 task_name=task_name,
+                agent_id=agent_id,
+                agent_name=agent_name,
                 trace_id=trace_id,
                 parent_span_id=parent_span_id,
+                request=request,
             )

agentex/lib/adk/providers/_modules/openai.py

@@ -88,6 +88,7 @@ class OpenAIModule:
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> SerializableRunResult | RunResult:
         """
         Run an agent without streaming or TaskMessage creation.
@@ -114,6 +115,7 @@ class OpenAIModule:
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
             input_guardrails: Optional list of input guardrails to run on initial user input.
             output_guardrails: Optional list of output guardrails to run on final agent output.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             Union[SerializableRunResult, RunResult]: SerializableRunResult when in Temporal, RunResult otherwise.
@@ -136,6 +138,7 @@ class OpenAIModule:
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
             return await ActivityHelpers.execute_activity(
                 activity_name=OpenAIActivityName.RUN_AGENT,
@@ -163,6 +166,7 @@ class OpenAIModule:
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
 
     async def run_agent_auto_send(
@@ -191,6 +195,7 @@ class OpenAIModule:
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> SerializableRunResult | RunResult:
         """
         Run an agent with automatic TaskMessage creation.
@@ -216,6 +221,7 @@ class OpenAIModule:
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
             input_guardrails: Optional list of input guardrails to run on initial user input.
             output_guardrails: Optional list of output guardrails to run on final agent output.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             Union[SerializableRunResult, RunResult]: SerializableRunResult when in Temporal, RunResult otherwise.
@@ -239,6 +245,7 @@ class OpenAIModule:
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
             return await ActivityHelpers.execute_activity(
                 activity_name=OpenAIActivityName.RUN_AGENT_AUTO_SEND,
@@ -267,6 +274,7 @@ class OpenAIModule:
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )
 
     async def run_agent_streamed(
@@ -291,6 +299,7 @@ class OpenAIModule:
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> RunResultStreaming:
         """
         Run an agent with streaming enabled but no TaskMessage creation.
@@ -320,6 +329,7 @@ class OpenAIModule:
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
             input_guardrails: Optional list of input guardrails to run on initial user input.
             output_guardrails: Optional list of output guardrails to run on final agent output.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             RunResultStreaming: The result of the agent run with streaming.
@@ -352,6 +362,7 @@ class OpenAIModule:
             mcp_timeout_seconds=mcp_timeout_seconds,
             input_guardrails=input_guardrails,
             output_guardrails=output_guardrails,
+            max_turns=max_turns,
         )
 
     async def run_agent_streamed_auto_send(
@@ -380,6 +391,7 @@ class OpenAIModule:
         mcp_timeout_seconds: int | None = None,
         input_guardrails: list[InputGuardrail] | None = None,
         output_guardrails: list[OutputGuardrail] | None = None,
+        max_turns: int | None = None,
     ) -> SerializableRunResultStreaming | RunResultStreaming:
         """
         Run an agent with streaming enabled and automatic TaskMessage creation.
@@ -405,6 +417,7 @@ class OpenAIModule:
             output_type: Optional output type.
             tool_use_behavior: Optional tool use behavior.
             mcp_timeout_seconds: Optional param to set the timeout threshold for the MCP servers. Defaults to 5 seconds.
+            max_turns: Maximum number of turns the agent can take. Uses Runner's default if None.
 
         Returns:
             Union[SerializableRunResultStreaming, RunResultStreaming]: SerializableRunResultStreaming when in Temporal, RunResultStreaming otherwise.
@@ -428,6 +441,7 @@ class OpenAIModule:
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns
             )
             return await ActivityHelpers.execute_activity(
                 activity_name=OpenAIActivityName.RUN_AGENT_STREAMED_AUTO_SEND,
@@ -456,4 +470,5 @@ class OpenAIModule:
                 mcp_timeout_seconds=mcp_timeout_seconds,
                 input_guardrails=input_guardrails,
                 output_guardrails=output_guardrails,
+                max_turns=max_turns,
             )

agentex/lib/cli/handlers/deploy_handlers.py

@@ -229,9 +229,12 @@ def merge_deployment_configs(
                 all_env_vars[EnvVarKeys.AUTH_PRINCIPAL_B64.value] = encoded_principal
             else:
                 raise DeploymentError(f"Auth principal unable to be encoded for agent_env_config: {agent_env_config}")
-                
+        
+        logger.info(f"Defined agent helm overrides: {agent_env_config.helm_overrides}")
+        logger.info(f"Before-merge helm values: {helm_values}")
         if agent_env_config.helm_overrides:
             _deep_merge(helm_values, agent_env_config.helm_overrides)
+        logger.info(f"After-merge helm values: {helm_values}")
 
     # Set final environment variables
     # Environment variable precedence: manifest -> environments.yaml -> secrets (highest)

agentex/lib/cli/templates/temporal/environments.yaml.j2

@@ -52,7 +52,7 @@ environments:
         limits:
           cpu: "1000m"
           memory: "2Gi"
-      temporal:
+      temporal-worker:
         enabled: true
         replicaCount: 2
         resources:

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")
@@ -180,9 +205,38 @@ class ACPService:
         self,
         task_id: str | None = None,
         task_name: str | None = None,
+        agent_id: str | None = None,
+        agent_name: str | None = None,
         trace_id: str | None = None,
         parent_span_id: str | None = None,
+        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")
+            
+        # 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,
@@ -190,27 +244,38 @@ class ACPService:
             input={
                 "task_id": task_id,
                 "task_name": task_name,
+                "agent_id": agent_id,
+                "agent_name": agent_name,
             },
         ) 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: RpcParamsCancelTaskRequest = {}
+            if task_id:
+                params["task_id"] = task_id
             if task_name:
+                params["task_name"] = task_name
+            
+            # Send cancel request to the correct agent
+            if agent_name:
                 json_rpc_response = await self._agentex_client.agents.rpc_by_name(
-                    agent_name=task_name,
+                    agent_name=agent_name,
                     method="task/cancel",
-                    params={
-                        "task_name": task_name,
-                    },
+                    params=params,
+                    extra_headers=extra_headers,
                 )
-            elif task_id:
+            else:  # agent_id is provided (validated above)
+                assert agent_id is not None
                 json_rpc_response = await self._agentex_client.agents.rpc(
-                    agent_id=task_id,
+                    agent_id=agent_id,
                     method="task/cancel",
-                    params={
-                        "task_id": task_id,
-                    },
+                    params=params,
+                    extra_headers=extra_headers,
                 )
-            else:
-                raise ValueError("Either task_name or task_id must be provided")
 
             task_entry = Task.model_validate(json_rpc_response.result)
             if span: