Merge branch 'main' into stas/trace-agentex-source

Author: smoreinis

.release-please-manifest.json

@@ -1,3 +1,3 @@
 {
-  ".": "0.4.20"
+  ".": "0.4.22"
 }

CHANGELOG.md

@@ -1,5 +1,13 @@
 # Changelog
 
+## 0.4.22 (2025-10-01)
+
+Full Changelog: [v0.4.21...v0.4.22](https://github.com/scaleapi/agentex-python/compare/v0.4.21...v0.4.22)
+
+## 0.4.21 (2025-10-01)
+
+Full Changelog: [v0.4.20...v0.4.21](https://github.com/scaleapi/agentex-python/compare/v0.4.20...v0.4.21)
+
 ## 0.4.20 (2025-10-01)
 
 Full Changelog: [v0.4.19...v0.4.20](https://github.com/scaleapi/agentex-python/compare/v0.4.19...v0.4.20)

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "agentex-sdk"
-version = "0.4.20"
+version = "0.4.22"
 description = "The official Python library for the agentex API"
 dynamic = ["readme"]
 license = "Apache-2.0"

src/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.20"  # x-release-please-version
+__version__ = "0.4.22"  # x-release-please-version

src/agentex/lib/cli/handlers/run_handlers.py

@@ -389,6 +389,10 @@ def create_agent_environment(manifest: AgentManifest) -> dict[str, str]:
             env_vars["WORKFLOW_NAME"] = temporal_config.name
             env_vars["WORKFLOW_TASK_QUEUE"] = temporal_config.queue_name
 
+        # Set health check port from temporal config
+        if manifest.agent.temporal and manifest.agent.temporal.health_check_port is not None:
+            env_vars["HEALTH_CHECK_PORT"] = str(manifest.agent.temporal.health_check_port)
+
     if agent_config.env:
         for key, value in agent_config.env.items():
             env_vars[key] = value

src/agentex/lib/cli/templates/temporal/manifest.yaml.j2

@@ -89,6 +89,10 @@ agent:
         # Convention: <agent_name>_task_queue
         queue_name: {{ queue_name }}
 
+    # Optional: Health check port for temporal worker
+    # Defaults to 80 if not specified
+    # health_check_port: 80
+
   # Optional: Credentials mapping
   # Maps Kubernetes secrets to environment variables
   # Common credentials include:

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

@@ -3,6 +3,7 @@
 from temporalio.client import Client, Plugin as ClientPlugin
 from temporalio.runtime import Runtime, TelemetryConfig, OpenTelemetryConfig
 from temporalio.contrib.pydantic import pydantic_data_converter
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 
 # class DateTimeJSONEncoder(AdvancedJSONEncoder):
 #     def default(self, o: Any) -> Any:
@@ -74,20 +75,24 @@ async def get_temporal_client(temporal_address: str, metrics_url: str | None = N
     if plugins:
         validate_client_plugins(plugins)
 
+    # Check if OpenAI plugin is present - it needs to configure its own data converter
+    has_openai_plugin = any(
+        isinstance(p, OpenAIAgentsPlugin) for p in (plugins or [])
+    )
+
+    # Only set data_converter if OpenAI plugin is not present
+    connect_kwargs = {
+        "target_host": temporal_address,
+        "plugins": plugins,
+    }
+
+    if not has_openai_plugin:
+        connect_kwargs["data_converter"] = pydantic_data_converter
+
     if not metrics_url:
-        client = await Client.connect(
-            target_host=temporal_address,
-            # data_converter=custom_data_converter,
-            data_converter=pydantic_data_converter,
-            plugins=plugins,
-        )
+        client = await Client.connect(**connect_kwargs)
     else:
         runtime = Runtime(telemetry=TelemetryConfig(metrics=OpenTelemetryConfig(url=metrics_url)))
-        client = await Client.connect(
-            target_host=temporal_address,
-            # data_converter=custom_data_converter,
-            data_converter=pydantic_data_converter,
-            runtime=runtime,
-            plugins=plugins,
-        )
+        connect_kwargs["runtime"] = runtime
+        client = await Client.connect(**connect_kwargs)
     return client

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

@@ -23,6 +23,7 @@
     JSONPlainPayloadConverter,
     _JSONTypeConverterUnhandled,
 )
+from temporalio.contrib.openai_agents import OpenAIAgentsPlugin
 
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.registration import register_agent
@@ -81,18 +82,27 @@ async def get_temporal_client(temporal_address: str, metrics_url: str | None = N
     if plugins != []:  # We don't need to validate the plugins if they are empty
         _validate_plugins(plugins)
 
+    # Check if OpenAI plugin is present - it needs to configure its own data converter
+    has_openai_plugin = any(
+        isinstance(p, OpenAIAgentsPlugin) for p in (plugins or [])
+    )
+
+    # Build connection kwargs
+    connect_kwargs = {
+        "target_host": temporal_address,
+        "plugins": plugins,
+    }
+
+    # Only set data_converter if OpenAI plugin is not present
+    if not has_openai_plugin:
+        connect_kwargs["data_converter"] = custom_data_converter
+
     if not metrics_url:
-        client = await Client.connect(
-            target_host=temporal_address, data_converter=custom_data_converter, plugins=plugins
-        )
+        client = await Client.connect(**connect_kwargs)
     else:
         runtime = Runtime(telemetry=TelemetryConfig(metrics=OpenTelemetryConfig(url=metrics_url)))
-        client = await Client.connect(
-            target_host=temporal_address,
-            data_converter=custom_data_converter,
-            runtime=runtime,
-            plugins=plugins,
-        )
+        connect_kwargs["runtime"] = runtime
+        client = await Client.connect(**connect_kwargs)
     return client
 
 
@@ -102,7 +112,7 @@ def __init__(
         task_queue,
         max_workers: int = 10,
         max_concurrent_activities: int = 10,
-        health_check_port: int = 80,
+        health_check_port: int = int(os.environ.get("HEALTH_CHECK_PORT")),
         plugins: list = [],
     ):
         self.task_queue = task_queue

src/agentex/lib/environment_variables.py

@@ -32,6 +32,8 @@ class EnvVarKeys(str, Enum):
     # Workflow Configuration
     WORKFLOW_NAME = "WORKFLOW_NAME"
     WORKFLOW_TASK_QUEUE = "WORKFLOW_TASK_QUEUE"
+    # Temporal Worker Configuration
+    HEALTH_CHECK_PORT = "HEALTH_CHECK_PORT"
     # Auth Configuration
     AUTH_PRINCIPAL_B64 = "AUTH_PRINCIPAL_B64"
     # Build Information
@@ -65,6 +67,9 @@ class EnvironmentVariables(BaseModel):
     # Workflow Configuration
     WORKFLOW_TASK_QUEUE: str | None = None
     WORKFLOW_NAME: str | None = None
+    # Temporal Worker Configuration
+    HEALTH_CHECK_PORT: int = 80
+    # Auth Configuration
     AUTH_PRINCIPAL_B64: str | None = None
     # Build Information
     BUILD_INFO_PATH: str | None = None

src/agentex/lib/types/agent_configs.py

@@ -45,6 +45,7 @@ class TemporalConfig(BaseModel):
         enabled: Whether this agent uses Temporal workflows
         workflow: The temporal workflow configuration
         workflows: The list of temporal workflow configurations
+        health_check_port: Port for temporal worker health check endpoint
     """
 
     enabled: bool = Field(
@@ -58,6 +59,10 @@ class TemporalConfig(BaseModel):
         default=None,
         description="List of temporal workflow configurations. Used when enabled=true.",
     )
+    health_check_port: int | None = Field(
+        default=None,
+        description="Port for temporal worker health check endpoint. Defaults to 80 if not specified.",
+    )
 
     @validator("workflows")
     def validate_workflows_not_empty(cls, v):