agentex-sdk 0.4.22__py3-none-any.whl → 0.4.24__py3-none-any.whl

agentex/lib/core/tracing/processors/sgp_tracing_processor.py

@@ -8,6 +8,7 @@ from scale_gp_beta.lib.tracing.span import Span as SGPSpan
 from agentex.types.span import Span
 from agentex.lib.types.tracing import SGPTracingProcessorConfig
 from agentex.lib.utils.logging import make_logger
+from agentex.lib.environment_variables import EnvironmentVariables
 from agentex.lib.core.tracing.processors.tracing_processor_interface import (
     SyncTracingProcessor,
     AsyncTracingProcessor,
@@ -24,9 +25,24 @@ class SGPSyncTracingProcessor(SyncTracingProcessor):
             disabled=disabled,
         )
         self._spans: dict[str, SGPSpan] = {}
+        self.env_vars = EnvironmentVariables.refresh()
+
+    def _add_source_to_span(self, span: Span) -> None:
+        if span.data is None:
+            span.data = {}
+        if isinstance(span.data, dict):
+            span.data["__source__"] = "agentex"
+            if self.env_vars.ACP_TYPE is not None:
+                span.data["__acp_type__"] = self.env_vars.ACP_TYPE
+            if self.env_vars.AGENT_NAME is not None:
+                span.data["__agent_name__"] = self.env_vars.AGENT_NAME
+            if self.env_vars.AGENT_ID is not None:
+                span.data["__agent_id__"] = self.env_vars.AGENT_ID
 
     @override
     def on_span_start(self, span: Span) -> None:
+        self._add_source_to_span(span)
+        
         sgp_span = create_span(
             name=span.name,
             span_id=span.id,
@@ -50,6 +66,7 @@ class SGPSyncTracingProcessor(SyncTracingProcessor):
             )
             return
 
+        self._add_source_to_span(span)
         sgp_span.output = span.output  # type: ignore[assignment]
         sgp_span.metadata = span.data  # type: ignore[assignment]
         sgp_span.end_time = span.end_time.isoformat()  # type: ignore[union-attr]
@@ -70,9 +87,23 @@ class SGPAsyncTracingProcessor(AsyncTracingProcessor):
             if not self.disabled
             else None
         )
+        self.env_vars = EnvironmentVariables.refresh()
+
+    def _add_source_to_span(self, span: Span) -> None:
+        if span.data is None:
+            span.data = {}
+        if isinstance(span.data, dict):
+            span.data["__source__"] = "agentex"
+            if self.env_vars.ACP_TYPE is not None:
+                span.data["__acp_type__"] = self.env_vars.ACP_TYPE
+            if self.env_vars.AGENT_NAME is not None:
+                span.data["__agent_name__"] = self.env_vars.AGENT_NAME
+            if self.env_vars.AGENT_ID is not None:
+                span.data["__agent_id__"] = self.env_vars.AGENT_ID
 
     @override
     async def on_span_start(self, span: Span) -> None:
+        self._add_source_to_span(span)
         sgp_span = create_span(
             name=span.name,
             span_id=span.id,
@@ -85,6 +116,7 @@ class SGPAsyncTracingProcessor(AsyncTracingProcessor):
         sgp_span.start_time = span.start_time.isoformat()  # type: ignore[union-attr]
 
         if self.disabled:
+            logger.warning("SGP is disabled, skipping span upsert")
             return
         await self.sgp_async_client.spans.upsert_batch(  # type: ignore[union-attr]
             items=[sgp_span.to_request_params()]
@@ -101,6 +133,7 @@ class SGPAsyncTracingProcessor(AsyncTracingProcessor):
             )
             return
 
+        self._add_source_to_span(span)
         sgp_span.output = span.output  # type: ignore[assignment]
         sgp_span.metadata = span.data  # type: ignore[assignment]
         sgp_span.end_time = span.end_time.isoformat()  # type: ignore[union-attr]

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

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):

agentex/lib/utils/dev_tools/async_messages.py

@@ -49,8 +49,8 @@ def print_task_message(
     
     # Skip empty reasoning messages
     if isinstance(message.content, ReasoningContent):
-        has_summary = message.content.summary and any(s for s in message.content.summary if s)
-        has_content = message.content.content and any(c for c in message.content.content if c)
+        has_summary = bool(message.content.summary) and any(s for s in message.content.summary if s)
+        has_content = bool(message.content.content) and any(c for c in message.content.content if c) if message.content.content is not None else False
         if not has_summary and not has_content:
             return
     
@@ -135,18 +135,19 @@ def print_task_message(
     
     if rich_print and console:
         author_color = "bright_cyan" if message.content.author == "user" else "green"
-        title = f"[bold {author_color}]{message.content.author.upper()}[/bold {author_color}] [{timestamp}]"
         
-        # Use different border styles for tool messages
+        # Use different border styles and colors for different content types
         if content_type == "tool_request":
             border_style = "yellow"
         elif content_type == "tool_response":
             border_style = "bright_green"
         elif content_type == "reasoning":
             border_style = "bright_magenta"
+            author_color = "bright_magenta"  # Also make the author text magenta
         else:
             border_style = author_color
-            
+        
+        title = f"[bold {author_color}]{message.content.author.upper()}[/bold {author_color}] [{timestamp}]"
         panel = Panel(Markdown(content), title=title, border_style=border_style, width=80)
         console.print(panel)
     else:
@@ -329,7 +330,7 @@ def subscribe_to_async_task_messages(
                             
                             # Deserialize the discriminated union TaskMessageUpdate based on the "type" field
                             message_type = task_message_update_data.get("type", "unknown")
-                            
+
                             # Handle different message types for streaming progress
                             if message_type == "start":
                                 task_message_update = StreamTaskMessageStart.model_validate(task_message_update_data)
@@ -359,6 +360,9 @@ def subscribe_to_async_task_messages(
                                 if index in active_spinners:
                                     active_spinners[index].stop()
                                     del active_spinners[index]
+                                    # Ensure clean line after spinner
+                                    if print_messages:
+                                        print()
                                 
                                 if task_message_update.parent_task_message and task_message_update.parent_task_message.id:
                                     finished_message = client.messages.retrieve(task_message_update.parent_task_message.id)
@@ -373,6 +377,9 @@ def subscribe_to_async_task_messages(
                                 if index in active_spinners:
                                     active_spinners[index].stop()
                                     del active_spinners[index]
+                                    # Ensure clean line after spinner
+                                    if print_messages:
+                                        print()
                                 
                                 if task_message_update.parent_task_message and task_message_update.parent_task_message.id:
                                     finished_message = client.messages.retrieve(task_message_update.parent_task_message.id)

agentex/resources/agents.py

@@ -10,7 +10,7 @@ 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
+from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -65,7 +65,7 @@ class AgentsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Agent:
         """
         Get an agent by its unique ID.
@@ -92,13 +92,13 @@ class AgentsResource(SyncAPIResource):
     def list(
         self,
         *,
-        task_id: Optional[str] | NotGiven = NOT_GIVEN,
+        task_id: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentListResponse:
         """
         List all registered agents, optionally filtered by query parameters.
@@ -135,7 +135,7 @@ class AgentsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DeleteResponse:
         """
         Delete an agent by its unique ID.
@@ -168,7 +168,7 @@ class AgentsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DeleteResponse:
         """
         Delete an agent by its unique name.
@@ -201,7 +201,7 @@ class AgentsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Agent:
         """
         Get an agent by its unique name.
@@ -231,14 +231,14 @@ class AgentsResource(SyncAPIResource):
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
         params: agent_rpc_params.Params,
-        id: Union[int, str, None] | NotGiven = NOT_GIVEN,
-        jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
+        id: Union[int, str, None] | Omit = omit,
+        jsonrpc: Literal["2.0"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique ID.
@@ -279,14 +279,14 @@ class AgentsResource(SyncAPIResource):
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
         params: agent_rpc_by_name_params.Params,
-        id: Union[int, str, None] | NotGiven = NOT_GIVEN,
-        jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
+        id: Union[int, str, None] | Omit = omit,
+        jsonrpc: Literal["2.0"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique name.
@@ -612,7 +612,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Agent:
         """
         Get an agent by its unique ID.
@@ -639,13 +639,13 @@ class AsyncAgentsResource(AsyncAPIResource):
     async def list(
         self,
         *,
-        task_id: Optional[str] | NotGiven = NOT_GIVEN,
+        task_id: Optional[str] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentListResponse:
         """
         List all registered agents, optionally filtered by query parameters.
@@ -682,7 +682,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DeleteResponse:
         """
         Delete an agent by its unique ID.
@@ -715,7 +715,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> DeleteResponse:
         """
         Delete an agent by its unique name.
@@ -748,7 +748,7 @@ class AsyncAgentsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Agent:
         """
         Get an agent by its unique name.
@@ -778,14 +778,14 @@ class AsyncAgentsResource(AsyncAPIResource):
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
         params: agent_rpc_params.Params,
-        id: Union[int, str, None] | NotGiven = NOT_GIVEN,
-        jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
+        id: Union[int, str, None] | Omit = omit,
+        jsonrpc: Literal["2.0"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique ID.
@@ -826,14 +826,14 @@ class AsyncAgentsResource(AsyncAPIResource):
         *,
         method: Literal["event/send", "task/create", "message/send", "task/cancel"],
         params: agent_rpc_by_name_params.Params,
-        id: Union[int, str, None] | NotGiven = NOT_GIVEN,
-        jsonrpc: Literal["2.0"] | NotGiven = NOT_GIVEN,
+        id: Union[int, str, None] | Omit = omit,
+        jsonrpc: Literal["2.0"] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> AgentRpcResponse:
         """
         Handle JSON-RPC requests for an agent by its unique name.

agentex/resources/events.py

@@ -7,7 +7,7 @@ from typing import Optional
 import httpx
 
 from ..types import event_list_params
-from .._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from .._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from .._utils import maybe_transform, async_maybe_transform
 from .._compat import cached_property
 from .._resource import SyncAPIResource, AsyncAPIResource
@@ -53,7 +53,7 @@ class EventsResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Event:
         """
         Get Event
@@ -82,14 +82,14 @@ class EventsResource(SyncAPIResource):
         *,
         agent_id: str,
         task_id: str,
-        last_processed_event_id: Optional[str] | NotGiven = NOT_GIVEN,
-        limit: Optional[int] | NotGiven = NOT_GIVEN,
+        last_processed_event_id: Optional[str] | Omit = omit,
+        limit: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> EventListResponse:
         """
         List events for a specific task and agent.
@@ -164,7 +164,7 @@ class AsyncEventsResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> Event:
         """
         Get Event
@@ -193,14 +193,14 @@ class AsyncEventsResource(AsyncAPIResource):
         *,
         agent_id: str,
         task_id: str,
-        last_processed_event_id: Optional[str] | NotGiven = NOT_GIVEN,
-        limit: Optional[int] | NotGiven = NOT_GIVEN,
+        last_processed_event_id: Optional[str] | Omit = omit,
+        limit: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> EventListResponse:
         """
         List events for a specific task and agent.

agentex/resources/messages/batch.py

@@ -6,7 +6,7 @@ from typing import Dict, Iterable
 
 import httpx
 
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import Body, Query, Headers, NotGiven, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -55,7 +55,7 @@ class BatchResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> BatchCreateResponse:
         """
         Batch Create Messages
@@ -94,7 +94,7 @@ class BatchResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> BatchUpdateResponse:
         """
         Batch Update Messages
@@ -154,7 +154,7 @@ class AsyncBatchResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> BatchCreateResponse:
         """
         Batch Create Messages
@@ -193,7 +193,7 @@ class AsyncBatchResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> BatchUpdateResponse:
         """
         Batch Update Messages

agentex/resources/messages/messages.py

@@ -16,7 +16,7 @@ from .batch import (
     AsyncBatchResourceWithStreamingResponse,
 )
 from ...types import message_list_params, message_create_params, message_update_params
-from ..._types import NOT_GIVEN, Body, Query, Headers, NotGiven
+from ..._types import Body, Omit, Query, Headers, NotGiven, omit, not_given
 from ..._utils import maybe_transform, async_maybe_transform
 from ..._compat import cached_property
 from ..._resource import SyncAPIResource, AsyncAPIResource
@@ -63,13 +63,13 @@ class MessagesResource(SyncAPIResource):
         *,
         content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskMessage:
         """
         Create Message
@@ -108,7 +108,7 @@ class MessagesResource(SyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskMessage:
         """
         Get Message
@@ -138,13 +138,13 @@ class MessagesResource(SyncAPIResource):
         *,
         content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskMessage:
         """
         Update Message
@@ -180,13 +180,13 @@ class MessagesResource(SyncAPIResource):
         self,
         *,
         task_id: str,
-        limit: Optional[int] | NotGiven = NOT_GIVEN,
+        limit: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> MessageListResponse:
         """
         List Messages
@@ -250,13 +250,13 @@ class AsyncMessagesResource(AsyncAPIResource):
         *,
         content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskMessage:
         """
         Create Message
@@ -295,7 +295,7 @@ class AsyncMessagesResource(AsyncAPIResource):
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskMessage:
         """
         Get Message
@@ -325,13 +325,13 @@ class AsyncMessagesResource(AsyncAPIResource):
         *,
         content: TaskMessageContentParam,
         task_id: str,
-        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | NotGiven = NOT_GIVEN,
+        streaming_status: Optional[Literal["IN_PROGRESS", "DONE"]] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> TaskMessage:
         """
         Update Message
@@ -367,13 +367,13 @@ class AsyncMessagesResource(AsyncAPIResource):
         self,
         *,
         task_id: str,
-        limit: Optional[int] | NotGiven = NOT_GIVEN,
+        limit: Optional[int] | Omit = omit,
         # Use the following arguments if you need to pass additional parameters to the API that aren't available via kwargs.
         # The extra values given here take precedence over values defined on the client or passed to this method.
         extra_headers: Headers | None = None,
         extra_query: Query | None = None,
         extra_body: Body | None = None,
-        timeout: float | httpx.Timeout | None | NotGiven = NOT_GIVEN,
+        timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> MessageListResponse:
         """
         List Messages