agentex-sdk 0.1.0a6__tar.gz → 0.2.0__tar.gz

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/.gitignore

@@ -1,5 +1,4 @@
 .prism.log
-.vscode
 _dev
 
 __pycache__

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

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

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/CHANGELOG.md

@@ -1,5 +1,36 @@
 # Changelog
 
+## 0.2.0 (2025-07-25)
+
+Full Changelog: [v0.1.1...v0.2.0](https://github.com/scaleapi/agentex-python/compare/v0.1.1...v0.2.0)
+
+### Features
+
+* **api:** update typescript sdk with big changes ([2c75d64](https://github.com/scaleapi/agentex-python/commit/2c75d642348df727505778c347efa568930ea4f0))
+
+
+### Chores
+
+* **project:** add settings file for vscode ([0f926cc](https://github.com/scaleapi/agentex-python/commit/0f926cce7df375de33627f8212caacf64f89b1ed))
+
+## 0.1.1 (2025-07-24)
+
+Full Changelog: [v0.1.0...v0.1.1](https://github.com/scaleapi/agentex-python/compare/v0.1.0...v0.1.1)
+
+### Features
+
+* **api:** manual updates ([714e97e](https://github.com/scaleapi/agentex-python/commit/714e97ed1813a4a91b421fb77fadaf2afac2450d))
+* **api:** manual updates ([8dccfbd](https://github.com/scaleapi/agentex-python/commit/8dccfbdd9b8b887bfb99c79a9a28163215560ae4))
+* **api:** manual updates ([03af884](https://github.com/scaleapi/agentex-python/commit/03af884e31a3df4d42a863c06c5ab4dfc2374374))
+
+## 0.1.0 (2025-07-23)
+
+Full Changelog: [v0.1.0-alpha.6...v0.1.0](https://github.com/scaleapi/agentex-python/compare/v0.1.0-alpha.6...v0.1.0)
+
+### Features
+
+* **api:** manual updates ([84010e4](https://github.com/scaleapi/agentex-python/commit/84010e4adecf7c779abd9a828000a3b50d9d3ac3))
+
 ## 0.1.0-alpha.6 (2025-07-23)
 
 Full Changelog: [v0.1.0-alpha.5...v0.1.0-alpha.6](https://github.com/scaleapi/agentex-python/compare/v0.1.0-alpha.5...v0.1.0-alpha.6)

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.1.0a6
+Version: 0.2.0
 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
@@ -75,13 +75,10 @@ The full API of this library can be found in [api.md](https://github.com/scaleap
 ## Installation
 
 ```sh
-# install from the production repo
-pip install git+ssh://git@github.com/scaleapi/agentex-python.git
+# install from PyPI
+pip install agentex-sdk
 ```
 
-> [!NOTE]
-> Once this package is [published to PyPI](https://www.stainless.com/docs/guides/publish), this will become: `pip install --pre agentex-sdk`
-
 ## Usage
 
 The full API of this library can be found in [api.md](https://github.com/scaleapi/agentex-python/tree/main/api.md).
@@ -136,8 +133,8 @@ By default, the async client uses `httpx` for HTTP requests. However, for improv
 You can enable this by installing `aiohttp`:
 
 ```sh
-# install from the production repo
-pip install 'agentex-sdk[aiohttp] @ git+ssh://git@github.com/scaleapi/agentex-python.git'
+# install from PyPI
+pip install agentex-sdk[aiohttp]
 ```
 
 Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/README.md

@@ -16,13 +16,10 @@ The full API of this library can be found in [api.md](api.md).
 ## Installation
 
 ```sh
-# install from the production repo
-pip install git+ssh://git@github.com/scaleapi/agentex-python.git
+# install from PyPI
+pip install agentex-sdk
 ```
 
-> [!NOTE]
-> Once this package is [published to PyPI](https://www.stainless.com/docs/guides/publish), this will become: `pip install --pre agentex-sdk`
-
 ## Usage
 
 The full API of this library can be found in [api.md](api.md).
@@ -77,8 +74,8 @@ By default, the async client uses `httpx` for HTTP requests. However, for improv
 You can enable this by installing `aiohttp`:
 
 ```sh
-# install from the production repo
-pip install 'agentex-sdk[aiohttp] @ git+ssh://git@github.com/scaleapi/agentex-python.git'
+# install from PyPI
+pip install agentex-sdk[aiohttp]
 ```
 
 Then you can enable it by instantiating the client with `http_client=DefaultAioHttpClient()`:

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/api.md

@@ -1,9 +1,3 @@
-# Shared Types
-
-```python
-from agentex.types import TaskMessageUpdate
-```
-
 # Agents
 
 Types:
@@ -12,7 +6,6 @@ Types:
 from agentex.types import (
     AcpType,
     Agent,
-    AgentRpcParams,
     AgentRpcRequest,
     AgentRpcResponse,
     AgentRpcResult,

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/pyproject.toml

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

{agentex_sdk-0.1.0a6 → agentex_sdk-0.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.1.0-alpha.6"  # x-release-please-version
+__version__ = "0.2.0"  # x-release-please-version

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/cli/handlers/deploy_handlers.py

@@ -8,8 +8,10 @@ import yaml
 from pydantic import BaseModel, Field
 from rich.console import Console
 
+from agentex.lib.cli.utils.auth_utils import _encode_principal_context
 from agentex.lib.cli.utils.exceptions import DeploymentError, HelmError
 from agentex.lib.cli.utils.kubectl_utils import check_and_switch_cluster_context
+from agentex.lib.environment_variables import EnvVarKeys
 from agentex.lib.sdk.config.agent_config import AgentConfig
 from agentex.lib.sdk.config.agent_manifest import AgentManifest
 from agentex.lib.sdk.config.deployment_config import ClusterConfig
@@ -92,6 +94,12 @@ def load_override_config(override_file_path: str | None = None) -> ClusterConfig
         ) from e
 
 
+
+def convert_env_vars_dict_to_list(env_vars: dict[str, str]) -> list[dict[str, str]]:
+    """Convert a dictionary of environment variables to a list of dictionaries"""
+    return [{"name": key, "value": value} for key, value in env_vars.items()]
+
+
 def merge_deployment_configs(
     manifest: AgentManifest,
     cluster_config: ClusterConfig | None,
@@ -168,6 +176,10 @@ def merge_deployment_configs(
         if TEMPORAL_WORKER_KEY in helm_values:
             helm_values[TEMPORAL_WORKER_KEY]["env"] = agent_config.env
 
+        encoded_principal = _encode_principal_context(manifest)
+        if encoded_principal:
+            helm_values["env"][EnvVarKeys.AUTH_PRINCIPAL_B64] = encoded_principal
+
     if manifest.deployment and manifest.deployment.imagePullSecrets:
         pull_secrets = [
             pull_secret.to_dict()
@@ -213,6 +225,14 @@ def merge_deployment_configs(
         if cluster_config.additional_overrides:
             _deep_merge(helm_values, cluster_config.additional_overrides)
 
+    # Convert the env vars to a list of dictionaries
+    if "env" in helm_values:
+        helm_values["env"] = convert_env_vars_dict_to_list(helm_values["env"])
+    if TEMPORAL_WORKER_KEY in helm_values and "env" in helm_values[TEMPORAL_WORKER_KEY]:
+        helm_values[TEMPORAL_WORKER_KEY]["env"] = convert_env_vars_dict_to_list(
+            helm_values[TEMPORAL_WORKER_KEY]["env"]
+        )
+    print("Deploying with the following helm values: ", helm_values)
     return helm_values
 
 

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/cli/handlers/run_handlers.py

@@ -6,10 +6,12 @@ from pathlib import Path
 from rich.console import Console
 from rich.panel import Panel
 
+from agentex.lib.cli.utils.auth_utils import _encode_principal_context
 from agentex.lib.cli.handlers.cleanup_handlers import (
     cleanup_agent_workflows,
     should_cleanup_on_restart
 )
+from agentex.lib.environment_variables import EnvVarKeys
 from agentex.lib.sdk.config.agent_manifest import AgentManifest
 from agentex.lib.utils.logging import make_logger
 
@@ -427,6 +429,11 @@ def create_agent_environment(manifest: AgentManifest) -> dict[str, str]:
         "ACP_PORT": str(manifest.local_development.agent.port),
     }
 
+    # Add authorization principal if set
+    encoded_principal = _encode_principal_context(manifest)
+    if encoded_principal:
+        env_vars[EnvVarKeys.AUTH_PRINCIPAL_B64] = encoded_principal
+
     # Add description if available
     if manifest.agent.description:
         env_vars["AGENT_DESCRIPTION"] = manifest.agent.description

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/cli/templates/temporal/project/workflow.py.j2

@@ -7,7 +7,7 @@ from agentex.lib.types.acp import CreateTaskParams, SendEventParams
 from agentex.lib.core.temporal.workflows.workflow import BaseWorkflow
 from agentex.lib.core.temporal.types.workflow import SignalName
 from agentex.lib.utils.logging import make_logger
-from agentex.types.task_message import TextContent
+from agentex.types.text_content import TextContent
 from agentex.lib.environment_variables import EnvironmentVariables
 
 environment_variables = EnvironmentVariables.refresh()

agentex_sdk-0.2.0/src/agentex/lib/cli/utils/auth_utils.py

@@ -0,0 +1,18 @@
+import base64
+import json
+
+from agentex.lib.sdk.config.agent_manifest import AgentManifest
+
+
+# Base 64 encode principal dictionary
+def _encode_principal_context(manifest: AgentManifest):
+    if manifest.auth is None:
+        return None
+
+    principal = manifest.auth.principal
+    if principal is None:
+        return None
+
+    json_str = json.dumps(principal, separators=(',', ':'))
+    encoded_bytes = base64.b64encode(json_str.encode('utf-8'))
+    return encoded_bytes.decode('utf-8')

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/environment_variables.py

@@ -24,9 +24,11 @@ class EnvVarKeys(str, Enum):
     ACP_URL = "ACP_URL"
     ACP_PORT = "ACP_PORT"
     ACP_TYPE = "ACP_TYPE"
-    # Workflow Configuraiton
+    # Workflow Configuration
     WORKFLOW_NAME = "WORKFLOW_NAME"
     WORKFLOW_TASK_QUEUE = "WORKFLOW_TASK_QUEUE"
+    # Auth Configuration
+    AUTH_PRINCIPAL_B64 = "AUTH_PRINCIPAL_B64"
 
 
 class Environment(str, Enum):
@@ -54,6 +56,7 @@ class EnvironmentVariables(BaseModel):
     # Workflow Configuration
     WORKFLOW_TASK_QUEUE: str | None = None
     WORKFLOW_NAME: str | None = None
+    AUTH_PRINCIPAL_B64: str | None = None
 
     @classmethod
     def refresh(cls) -> EnvironmentVariables | None:

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/sdk/config/agent_manifest.py

@@ -15,7 +15,7 @@ from pydantic import Field
 
 from agentex.lib.sdk.config.agent_config import AgentConfig
 from agentex.lib.sdk.config.build_config import BuildConfig
-from agentex.lib.sdk.config.deployment_config import DeploymentConfig
+from agentex.lib.sdk.config.deployment_config import DeploymentConfig, AuthenticationConfig
 from agentex.lib.sdk.config.local_development_config import LocalDevelopmentConfig
 from agentex.lib.utils.logging import make_logger
 from agentex.lib.utils.model_utils import BaseModel
@@ -36,6 +36,7 @@ class AgentManifest(BaseModel):
     deployment: DeploymentConfig | None = Field(
         default=None, description="Deployment configuration for the agent"
     )
+    auth: AuthenticationConfig | None = Field(default=None, description="Authentication configuration")
 
     def context_manager(self, build_context_root: Path) -> BuildContextManager:
         """

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/sdk/config/deployment_config.py

@@ -1,4 +1,4 @@
-from typing import Any
+from typing import Any, Dict
 
 from pydantic import Field
 
@@ -90,6 +90,10 @@ class ClusterConfig(BaseModel):
     )
 
 
+class AuthenticationConfig(BaseModel):
+    principal: Dict[str, Any] = Field(description="Principal used for authorization on registration")
+
+
 class InjectedImagePullSecretValues(BaseModel):
     """Values for image pull secrets"""
 

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/lib/sdk/fastacp/base/base_acp_server.py

@@ -1,5 +1,7 @@
 import asyncio
+import base64
 import inspect
+import json
 from collections.abc import AsyncGenerator, Awaitable, Callable
 from contextlib import asynccontextmanager
 from typing import Any
@@ -341,6 +343,16 @@ class BaseACPServer(FastAPI):
         """Start the Uvicorn server for async handlers."""
         uvicorn.run(self, host=host, port=port, **kwargs)
 
+    def _get_auth_principal(self, env_vars: EnvironmentVariables):
+        if not env_vars.AUTH_PRINCIPAL_B64:
+            return None
+
+        try:
+            decoded_str = base64.b64decode(env_vars.AUTH_PRINCIPAL_B64).decode('utf-8')
+            return json.loads(decoded_str)
+        except Exception:
+            return None
+
     async def _register_agent(self, env_vars: EnvironmentVariables):
         """Register this agent with the Agentex server"""
         # Build the agent's own URL
@@ -350,12 +362,14 @@ class BaseACPServer(FastAPI):
             env_vars.AGENT_DESCRIPTION
             or f"Generic description for agent: {env_vars.AGENT_NAME}"
         )
+
         # Prepare registration data
         registration_data = {
             "name": env_vars.AGENT_NAME,
             "description": description,
             "acp_url": full_acp_url,
             "acp_type": env_vars.ACP_TYPE,
+            "principal_context": self._get_auth_principal(env_vars)
         }
 
         if env_vars.AGENT_ID:

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/resources/agents.py

@@ -7,7 +7,7 @@ from typing_extensions import Literal
 
 import httpx
 
-from ..types import AgentRpcParams, agent_rpc_params, agent_list_params, agent_rpc_by_name_params
+from ..types import agent_rpc_params, agent_list_params, agent_rpc_by_name_params
 from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
@@ -20,7 +20,6 @@ from .._response import (
 )
 from ..types.agent import Agent
 from .._base_client import make_request_options
-from ..types.agent_rpc_params import AgentRpcParams
 from ..types.agent_rpc_response import AgentRpcResponse
 from ..types.agent_list_response import AgentListResponse
 
@@ -221,7 +220,7 @@ class AgentsResource(SyncAPIResource):
         agent_id: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_params.Params,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -269,7 +268,7 @@ class AgentsResource(SyncAPIResource):
         agent_name: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_by_name_params.Params,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -507,7 +506,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         agent_id: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_params.Params,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
@@ -555,7 +554,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         agent_name: str,
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
-        params: AgentRpcParams,
+        params: agent_rpc_by_name_params.Params,
         id: Union[int, str, None] | NotGiven = NOT_GIVEN,
         jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/types/__init__.py

@@ -7,7 +7,6 @@ from .task import Task as Task
 from .agent import Agent as Agent
 from .event import Event as Event
 from .state import State as State
-from .shared import TaskMessageUpdate as TaskMessageUpdate
 from .acp_type import AcpType as AcpType
 from .data_delta import DataDelta as DataDelta
 from .text_delta import TextDelta as TextDelta
@@ -38,6 +37,7 @@ from .message_list_params import MessageListParams as MessageListParams
 from .state_create_params import StateCreateParams as StateCreateParams
 from .state_list_response import StateListResponse as StateListResponse
 from .state_update_params import StateUpdateParams as StateUpdateParams
+from .task_message_update import TaskMessageUpdate as TaskMessageUpdate
 from .tool_response_delta import ToolResponseDelta as ToolResponseDelta
 from .tracker_list_params import TrackerListParams as TrackerListParams
 from .task_delete_response import TaskDeleteResponse as TaskDeleteResponse

agentex_sdk-0.2.0/src/agentex/types/agent_rpc_by_name_params.py

@@ -0,0 +1,71 @@
+# File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
+
+from __future__ import annotations
+
+from typing import Dict, Union, Optional
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
+
+from .task_message_content_param import TaskMessageContentParam
+
+__all__ = [
+    "AgentRpcByNameParams",
+    "Params",
+    "ParamsCreateTaskRequest",
+    "ParamsCancelTaskRequest",
+    "ParamsSendMessageRequest",
+    "ParamsSendEventRequest",
+]
+
+
+class AgentRpcByNameParams(TypedDict, total=False):
+    method: Required[Literal["event/send", "task/create", "message/send", "task/cancel"]]
+
+    params: Required[Params]
+    """The parameters for the agent RPC request"""
+
+    id: Union[int, str, None]
+
+    jsonrpc: Literal["2.0"]
+
+
+class ParamsCreateTaskRequest(TypedDict, total=False):
+    name: Optional[str]
+    """The name of the task to create"""
+
+    params: Optional[Dict[str, object]]
+    """The parameters for the task"""
+
+
+class ParamsCancelTaskRequest(TypedDict, total=False):
+    task_id: Optional[str]
+    """The ID of the task to cancel. Either this or task_name must be provided."""
+
+    task_name: Optional[str]
+    """The name of the task to cancel. Either this or task_id must be provided."""
+
+
+class ParamsSendMessageRequest(TypedDict, total=False):
+    content: Required[TaskMessageContentParam]
+    """The message that was sent to the agent"""
+
+    stream: bool
+    """Whether to stream the response message back to the client"""
+
+    task_id: Optional[str]
+    """The ID of the task that the message was sent to"""
+
+
+class ParamsSendEventRequest(TypedDict, total=False):
+    content: Optional[TaskMessageContentParam]
+    """The content to send to the event"""
+
+    task_id: Optional[str]
+    """The ID of the task that the event was sent to"""
+
+    task_name: Optional[str]
+    """The name of the task that the event was sent to"""
+
+
+Params: TypeAlias = Union[
+    ParamsCreateTaskRequest, ParamsCancelTaskRequest, ParamsSendMessageRequest, ParamsSendEventRequest
+]

{agentex_sdk-0.1.0a6 → agentex_sdk-0.2.0}/src/agentex/types/agent_rpc_params.py

@@ -3,14 +3,32 @@
 from __future__ import annotations
 
 from typing import Dict, Union, Optional
-from typing_extensions import Required, TypeAlias, TypedDict
+from typing_extensions import Literal, Required, TypeAlias, TypedDict
 
 from .task_message_content_param import TaskMessageContentParam
 
-__all__ = ["AgentRpcParams", "CreateTaskRequest", "CancelTaskRequest", "SendMessageRequest", "SendEventRequest"]
+__all__ = [
+    "AgentRpcParams",
+    "Params",
+    "ParamsCreateTaskRequest",
+    "ParamsCancelTaskRequest",
+    "ParamsSendMessageRequest",
+    "ParamsSendEventRequest",
+]
 
 
-class CreateTaskRequest(TypedDict, total=False):
+class AgentRpcParams(TypedDict, total=False):
+    method: Required[Literal["event/send", "task/create", "message/send", "task/cancel"]]
+
+    params: Required[Params]
+    """The parameters for the agent RPC request"""
+
+    id: Union[int, str, None]
+
+    jsonrpc: Literal["2.0"]
+
+
+class ParamsCreateTaskRequest(TypedDict, total=False):
     name: Optional[str]
     """The name of the task to create"""
 
@@ -18,7 +36,7 @@ class CreateTaskRequest(TypedDict, total=False):
     """The parameters for the task"""
 
 
-class CancelTaskRequest(TypedDict, total=False):
+class ParamsCancelTaskRequest(TypedDict, total=False):
     task_id: Optional[str]
     """The ID of the task to cancel. Either this or task_name must be provided."""
 
@@ -26,7 +44,7 @@ class CancelTaskRequest(TypedDict, total=False):
     """The name of the task to cancel. Either this or task_id must be provided."""
 
 
-class SendMessageRequest(TypedDict, total=False):
+class ParamsSendMessageRequest(TypedDict, total=False):
     content: Required[TaskMessageContentParam]
     """The message that was sent to the agent"""
 
@@ -37,7 +55,7 @@ class SendMessageRequest(TypedDict, total=False):
     """The ID of the task that the message was sent to"""
 
 
-class SendEventRequest(TypedDict, total=False):
+class ParamsSendEventRequest(TypedDict, total=False):
     content: Optional[TaskMessageContentParam]
     """The content to send to the event"""
 
@@ -48,4 +66,6 @@ class SendEventRequest(TypedDict, total=False):
     """The name of the task that the event was sent to"""
 
 
-AgentRpcParams: TypeAlias = Union[CreateTaskRequest, CancelTaskRequest, SendMessageRequest, SendEventRequest]
+Params: TypeAlias = Union[
+    ParamsCreateTaskRequest, ParamsCancelTaskRequest, ParamsSendMessageRequest, ParamsSendEventRequest
+]

{agentex_sdk-0.1.0a6/src/agentex/types/shared → agentex_sdk-0.2.0/src/agentex/types}/task_message_update.py

@@ -3,11 +3,11 @@
 from typing import Union, Optional
 from typing_extensions import Literal, Annotated, TypeAlias
 
-from ..._utils import PropertyInfo
-from ..._models import BaseModel
-from ..task_message import TaskMessage
-from ..task_message_delta import TaskMessageDelta
-from ..task_message_content import TaskMessageContent
+from .._utils import PropertyInfo
+from .._models import BaseModel
+from .task_message import TaskMessage
+from .task_message_delta import TaskMessageDelta
+from .task_message_content import TaskMessageContent
 
 __all__ = [
     "TaskMessageUpdate",