agentex-sdk 0.4.16__tar.gz → 0.4.18__tar.gz

agentex_sdk-0.4.18/.release-please-manifest.json

@@ -0,0 +1,3 @@
+{
+  ".": "0.4.18"
+}

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/CHANGELOG.md

@@ -1,5 +1,17 @@
 # Changelog
 
+## 0.4.18 (2025-09-29)
+
+Full Changelog: [v0.4.17...v0.4.18](https://github.com/scaleapi/agentex-python/compare/v0.4.17...v0.4.18)
+
+### Chores
+
+* **internal:** version bump ([eded756](https://github.com/scaleapi/agentex-python/commit/eded756bde2f3b4cfcf02c7a9cf72e70a82dd9aa))
+
+## 0.4.17 (2025-09-29)
+
+Full Changelog: [v0.4.16...v0.4.17](https://github.com/scaleapi/agentex-python/compare/v0.4.16...v0.4.17)
+
 ## 0.4.16 (2025-09-16)
 
 Full Changelog: [v0.4.15...v0.4.16](https://github.com/scaleapi/agentex-python/compare/v0.4.15...v0.4.16)

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.4.16
+Version: 0.4.18
 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.16 → agentex_sdk-0.4.18}/pyproject.toml

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

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/_constants.py

@@ -7,7 +7,7 @@ OVERRIDE_CAST_TO_HEADER = "____stainless_override_cast_to"
 
 # default timeout is 1 minute
 DEFAULT_TIMEOUT = httpx.Timeout(timeout=300, connect=5.0)
-DEFAULT_MAX_RETRIES = 0
+DEFAULT_MAX_RETRIES = 2
 DEFAULT_CONNECTION_LIMITS = httpx.Limits(max_connections=1000, max_keepalive_connections=1000)
 
 INITIAL_RETRY_DELAY = 0.5

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/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.16"  # x-release-please-version
+__version__ = "0.4.18"  # x-release-please-version

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/cli/handlers/deploy_handlers.py

@@ -260,8 +260,10 @@ def merge_deployment_configs(
         helm_values["global"]["imagePullSecrets"] = pull_secrets
         helm_values["imagePullSecrets"] = pull_secrets
 
-    # Add dynamic ACP command based on manifest configuration
-    add_acp_command_to_helm_values(helm_values, manifest, manifest_path)
+    # Add dynamic ACP command based on manifest configuration if command is not set in helm overrides
+    helm_overrides_command = agent_env_config and agent_env_config.helm_overrides and "command" in agent_env_config.helm_overrides
+    if not helm_overrides_command:
+        add_acp_command_to_helm_values(helm_values, manifest, manifest_path)
     
     print("Deploying with the following helm values: ", helm_values)
     return helm_values

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/core/clients/temporal/temporal_client.py

@@ -71,11 +71,12 @@ DUPLICATE_POLICY_TO_ID_REUSE_POLICY = {
 
 
 class TemporalClient:
-    def __init__(self, temporal_client: Client | None = None):
+    def __init__(self, temporal_client: Client | None = None, plugins: list[Any] = []):
         self._client: Client = temporal_client
+        self._plugins = plugins
 
     @classmethod
-    async def create(cls, temporal_address: str):
+    async def create(cls, temporal_address: str, plugins: list[Any] = []):
         if temporal_address in [
             "false",
             "False",
@@ -88,8 +89,11 @@ class TemporalClient:
         ]:
             _client = None
         else:
-            _client = await get_temporal_client(temporal_address)
-        return cls(_client)
+            _client = await get_temporal_client(
+                temporal_address, 
+                plugins=plugins
+            )
+        return cls(_client, plugins)
 
     async def setup(self, temporal_address: str):
         self._client = await self._get_temporal_client(
@@ -109,7 +113,10 @@ class TemporalClient:
         ]:
             return None
         else:
-            return await get_temporal_client(temporal_address)
+            return await get_temporal_client(
+                temporal_address, 
+                plugins=self._plugins
+            )
 
     async def start_workflow(
         self,

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/core/clients/temporal/utils.py

@@ -1,4 +1,5 @@
-from temporalio.client import Client
+from typing import Any
+from temporalio.client import Client, Plugin as ClientPlugin
 from temporalio.contrib.pydantic import pydantic_data_converter
 from temporalio.runtime import OpenTelemetryConfig, Runtime, TelemetryConfig
 
@@ -38,12 +39,50 @@ from temporalio.runtime import OpenTelemetryConfig, Runtime, TelemetryConfig
 # )
 
 
-async def get_temporal_client(temporal_address: str, metrics_url: str = None) -> Client:
+def validate_client_plugins(plugins: list[Any]) -> None:
+    """
+    Validate that all items in the plugins list are valid Temporal client plugins.
+    
+    Args:
+        plugins: List of plugins to validate
+        
+    Raises:
+        TypeError: If any plugin is not a valid ClientPlugin instance
+    """
+    for i, plugin in enumerate(plugins):
+        if not isinstance(plugin, ClientPlugin):
+            raise TypeError(
+                f"Plugin at index {i} must be an instance of temporalio.client.Plugin, "
+                f"got {type(plugin).__name__}. Note: WorkerPlugin is not valid for workflow clients."
+            )
+
+
+async def get_temporal_client(
+    temporal_address: str, 
+    metrics_url: str = None,
+    plugins: list[Any] = []
+) -> Client:
+    """
+    Create a Temporal client with plugin integration.
+    
+    Args:
+        temporal_address: Temporal server address
+        metrics_url: Optional metrics endpoint URL  
+        plugins: List of Temporal plugins to include
+        
+    Returns:
+        Configured Temporal client
+    """
+    # Validate plugins if any are provided
+    if plugins:
+        validate_client_plugins(plugins)
+        
     if not metrics_url:
         client = await Client.connect(
             target_host=temporal_address,
             # data_converter=custom_data_converter,
             data_converter=pydantic_data_converter,
+            plugins=plugins,
         )
     else:
         runtime = Runtime(telemetry=TelemetryConfig(metrics=OpenTelemetryConfig(url=metrics_url)))
@@ -52,5 +91,6 @@ async def get_temporal_client(temporal_address: str, metrics_url: str = None) ->
             # data_converter=custom_data_converter,
             data_converter=pydantic_data_converter,
             runtime=runtime,
+            plugins=plugins,
         )
     return client

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/core/temporal/services/temporal_task_service.py

@@ -23,6 +23,7 @@ class TemporalTaskService:
         self._temporal_client = temporal_client
         self._env_vars = env_vars
 
+
     async def submit_task(self, agent: Agent, task: Task, params: dict[str, Any] | None) -> str:
         """
         Submit a task to the async runtime for execution.

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/core/temporal/workers/worker.py

@@ -7,7 +7,7 @@ from concurrent.futures import ThreadPoolExecutor
 from typing import Any, overload
 
 from aiohttp import web
-from temporalio.client import Client
+from temporalio.client import Client, Plugin as ClientPlugin
 from temporalio.converter import (
     AdvancedJSONEncoder,
     CompositePayloadConverter,
@@ -19,6 +19,7 @@ from temporalio.converter import (
 )
 from temporalio.runtime import OpenTelemetryConfig, Runtime, TelemetryConfig
 from temporalio.worker import (
+    Plugin as WorkerPlugin,
     UnsandboxedWorkflowRunner,
     Worker,
 )
@@ -65,11 +66,25 @@ custom_data_converter = dataclasses.replace(
     payload_converter_class=DateTimePayloadConverter,
 )
 
+def _validate_plugins(plugins: list) -> None:
+    """Validate that all items in the plugins list are valid Temporal plugins."""
+    for i, plugin in enumerate(plugins):
+        if not isinstance(plugin, (ClientPlugin, WorkerPlugin)):
+            raise TypeError(
+                f"Plugin at index {i} must be an instance of temporalio.client.Plugin "
+                f"or temporalio.worker.Plugin, got {type(plugin).__name__}"
+            )
+
+
+
+async def get_temporal_client(temporal_address: str, metrics_url: str = None, plugins: list = []) -> Client:
+
+    if plugins != []: # We don't need to validate the plugins if they are empty
+        _validate_plugins(plugins)
 
-async def get_temporal_client(temporal_address: str, metrics_url: str = None) -> Client:
     if not metrics_url:
         client = await Client.connect(
-            target_host=temporal_address, data_converter=custom_data_converter
+            target_host=temporal_address, data_converter=custom_data_converter, plugins=plugins
         )
     else:
         runtime = Runtime(
@@ -90,6 +105,7 @@ class AgentexWorker:
         max_workers: int = 10,
         max_concurrent_activities: int = 10,
         health_check_port: int = 80,
+        plugins: list = [],
     ):
         self.task_queue = task_queue
         self.activity_handles = []
@@ -98,6 +114,7 @@ class AgentexWorker:
         self.health_check_server_running = False
         self.healthy = False
         self.health_check_port = health_check_port
+        self.plugins = plugins
 
     @overload
     async def run(
@@ -126,6 +143,7 @@ class AgentexWorker:
         await self._register_agent()
         temporal_client = await get_temporal_client(
             temporal_address=os.environ.get("TEMPORAL_ADDRESS", "localhost:7233"),
+            plugins=self.plugins,
         )
         
         # Enable debug mode if AgentEx debug is enabled (disables deadlock detection)

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/environment_variables.py

@@ -34,6 +34,8 @@ class EnvVarKeys(str, Enum):
     WORKFLOW_TASK_QUEUE = "WORKFLOW_TASK_QUEUE"
     # Auth Configuration
     AUTH_PRINCIPAL_B64 = "AUTH_PRINCIPAL_B64"
+    # Build Information
+    BUILD_INFO_PATH = "BUILD_INFO_PATH"
 
 
 class Environment(str, Enum):
@@ -63,6 +65,8 @@ class EnvironmentVariables(BaseModel):
     WORKFLOW_TASK_QUEUE: str | None = None
     WORKFLOW_NAME: str | None = None
     AUTH_PRINCIPAL_B64: str | None = None
+    # Build Information
+    BUILD_INFO_PATH: str | None = None
 
     @classmethod
     def refresh(cls) -> EnvironmentVariables:

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/sdk/fastacp/fastacp.py

@@ -1,4 +1,8 @@
-from typing import Literal
+import inspect
+import os
+from pathlib import Path
+
+from typing import Any, Literal
 from agentex.lib.sdk.fastacp.base.base_acp_server import BaseACPServer
 from agentex.lib.sdk.fastacp.impl.agentic_base_acp import AgenticBaseACP
 from agentex.lib.sdk.fastacp.impl.sync_acp import SyncACP
@@ -8,6 +12,7 @@ from agentex.lib.types.fastacp import (
     BaseACPConfig,
     SyncACPConfig,
 )
+from agentex.lib.utils.logging import make_logger
 
 # Add new mappings between ACP types and configs here
 # Add new mappings between ACP types and implementations here
@@ -16,6 +21,7 @@ AGENTIC_ACP_IMPLEMENTATIONS: dict[Literal["temporal", "base"], type[BaseACPServe
     "base": AgenticBaseACP,
 }
 
+logger = make_logger(__name__)
 
 class FastACP:
     """Factory for creating FastACP instances
@@ -43,17 +49,29 @@ class FastACP:
         implementation_class = AGENTIC_ACP_IMPLEMENTATIONS[config.type]
         # Handle temporal-specific configuration
         if config.type == "temporal":
-            # Extract temporal_address from config if it's a TemporalACPConfig
+            # Extract temporal_address and plugins from config if it's a TemporalACPConfig
             temporal_config = kwargs.copy()
             if hasattr(config, "temporal_address"):
                 temporal_config["temporal_address"] = config.temporal_address
+            if hasattr(config, "plugins"):
+                temporal_config["plugins"] = config.plugins
             return implementation_class.create(**temporal_config)
         else:
             return implementation_class.create(**kwargs)
 
+    @staticmethod
+    def locate_build_info_path() -> None:
+        """If a build-info.json file is present, set the BUILD_INFO_PATH environment variable"""
+        acp_root = Path(inspect.stack()[2].filename).resolve().parents[0]
+        build_info_path = acp_root / "build-info.json"
+        if build_info_path.exists():
+            os.environ["BUILD_INFO_PATH"] = str(build_info_path)
+
     @staticmethod
     def create(
-        acp_type: Literal["sync", "agentic"], config: BaseACPConfig | None = None, **kwargs
+        acp_type: Literal["sync", "agentic"], 
+        config: BaseACPConfig | None = None, 
+        **kwargs
     ) -> BaseACPServer | SyncACP | AgenticBaseACP | TemporalACP:
         """Main factory method to create any ACP type
 
@@ -63,6 +81,8 @@ class FastACP:
             **kwargs: Additional configuration parameters
         """ 
 
+        FastACP.locate_build_info_path()
+        
         if acp_type == "sync":
             sync_config = config if isinstance(config, SyncACPConfig) else None
             return FastACP.create_sync_acp(sync_config, **kwargs)

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/sdk/fastacp/impl/temporal_acp.py

@@ -1,5 +1,5 @@
 from contextlib import asynccontextmanager
-from typing import AsyncGenerator, Callable
+from typing import Any, AsyncGenerator, Callable
 
 from fastapi import FastAPI
 
@@ -24,18 +24,19 @@ class TemporalACP(BaseACPServer):
     """
 
     def __init__(
-        self, temporal_address: str, temporal_task_service: TemporalTaskService | None = None
+        self, temporal_address: str, temporal_task_service: TemporalTaskService | None = None, plugins: list[Any] | None = None
     ):
         super().__init__()
         self._temporal_task_service = temporal_task_service
         self._temporal_address = temporal_address
+        self._plugins = plugins or []
 
     @classmethod
-    def create(cls, temporal_address: str) -> "TemporalACP":
+    def create(cls, temporal_address: str, plugins: list[Any] | None = None) -> "TemporalACP":
         logger.info("Initializing TemporalACP instance")
 
         # Create instance without temporal client initially
-        temporal_acp = cls(temporal_address=temporal_address)
+        temporal_acp = cls(temporal_address=temporal_address, plugins=plugins)
         temporal_acp._setup_handlers()
         logger.info("TemporalACP instance initialized now")
         return temporal_acp
@@ -51,7 +52,8 @@ class TemporalACP(BaseACPServer):
             if self._temporal_task_service is None:
                 env_vars = EnvironmentVariables.refresh()
                 temporal_client = await TemporalClient.create(
-                    temporal_address=self._temporal_address
+                    temporal_address=self._temporal_address,
+                    plugins=self._plugins
                 )
                 self._temporal_task_service = TemporalTaskService(
                     temporal_client=temporal_client,

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/types/fastacp.py

@@ -1,6 +1,7 @@
-from typing import Literal
+from typing import Any, Literal
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, field_validator
+from agentex.lib.core.clients.temporal.utils import validate_client_plugins
 
 
 class BaseACPConfig(BaseModel):
@@ -43,12 +44,21 @@ class TemporalACPConfig(AgenticACPConfig):
     Attributes:
         type: The type of ACP implementation
         temporal_address: The address of the temporal server
+        plugins: List of Temporal client plugins
     """
 
     type: Literal["temporal"] = Field(default="temporal", frozen=True)
     temporal_address: str = Field(
         default="temporal-frontend.temporal.svc.cluster.local:7233", frozen=True
     )
+    plugins: list[Any] = Field(default=[], frozen=True)
+    
+    @field_validator("plugins")
+    @classmethod
+    def validate_plugins(cls, v: list[Any]) -> list[Any]:
+        """Validate that all plugins are valid Temporal client plugins."""
+        validate_client_plugins(v)
+        return v
 
 
 class AgenticBaseACPConfig(AgenticACPConfig):

{agentex_sdk-0.4.16 → agentex_sdk-0.4.18}/src/agentex/lib/utils/registration.py

@@ -19,6 +19,16 @@ def get_auth_principal(env_vars: EnvironmentVariables):
     except Exception:
         return None
 
+def get_build_info(env_vars: EnvironmentVariables):
+    logger.info(f"Getting build info from {env_vars.BUILD_INFO_PATH}")
+    if not env_vars.BUILD_INFO_PATH:
+        return None
+    try:
+        with open(env_vars.BUILD_INFO_PATH, "r") as f:
+            return json.load(f)
+    except Exception:
+        return None
+
 async def register_agent(env_vars: EnvironmentVariables):
     """Register this agent with the Agentex server"""
     if not env_vars.AGENTEX_BASE_URL:
@@ -38,7 +48,8 @@ async def register_agent(env_vars: EnvironmentVariables):
         "description": description,
         "acp_url": full_acp_url,
         "acp_type": env_vars.ACP_TYPE,
-        "principal_context": get_auth_principal(env_vars)
+        "principal_context": get_auth_principal(env_vars),
+        "registration_metadata": get_build_info(env_vars)
     }
 
     if env_vars.AGENT_ID:

agentex_sdk-0.4.16/.release-please-manifest.json

@@ -1,3 +0,0 @@
-{
-  ".": "0.4.16"
-}