feat(api): api update

Author: stainless-app[bot]

.stats.yml

@@ -1,4 +1,4 @@
 configured_endpoints: 45
-openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-636aa63c588134e6f47fc45212049593d91f810a9c7bd8d7a57810cf1b5ffc92.yml
-openapi_spec_hash: c76a42d4510aeafd896bc0d596f17af4
+openapi_spec_url: https://storage.googleapis.com/stainless-sdk-openapi-specs/sgp%2Fagentex-sdk-eeb5bf63b18d948611eec48d0225e9bba63b170f64eeeb35d91825724b7cf6c3.yml
+openapi_spec_hash: df8eb2eb2e32b18c1e3a5d8879d14270
 config_hash: fb079ef7936611b032568661b8165f19

src/agentex/resources/spans.py

@@ -57,6 +57,7 @@ def create(
         input: Union[Dict[str, object], Iterable[Dict[str, object]], None] | Omit = omit,
         output: Union[Dict[str, object], Iterable[Dict[str, object]], None] | Omit = omit,
         parent_id: Optional[str] | Omit = omit,
+        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,
@@ -86,6 +87,8 @@ def create(
 
           parent_id: ID of the parent span if this is a child span in a trace
 
+          task_id: ID of the task this span belongs to
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -107,6 +110,7 @@ def create(
                     "input": input,
                     "output": output,
                     "parent_id": parent_id,
+                    "task_id": task_id,
                 },
                 span_create_params.SpanCreateParams,
             ),
@@ -160,6 +164,7 @@ def update(
         output: Union[Dict[str, object], Iterable[Dict[str, object]], None] | Omit = omit,
         parent_id: Optional[str] | Omit = omit,
         start_time: Union[str, datetime, None] | Omit = omit,
+        task_id: Optional[str] | Omit = omit,
         trace_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.
@@ -186,6 +191,8 @@ def update(
 
           start_time: The time the span started
 
+          task_id: ID of the task this span belongs to
+
           trace_id: Unique identifier for the trace this span belongs to
 
           extra_headers: Send extra headers
@@ -209,6 +216,7 @@ def update(
                     "output": output,
                     "parent_id": parent_id,
                     "start_time": start_time,
+                    "task_id": task_id,
                     "trace_id": trace_id,
                 },
                 span_update_params.SpanUpdateParams,
@@ -226,6 +234,7 @@ def list(
         order_by: Optional[str] | Omit = omit,
         order_direction: str | Omit = omit,
         page_number: int | Omit = omit,
+        task_id: Optional[str] | Omit = omit,
         trace_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.
@@ -235,7 +244,7 @@ def list(
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SpanListResponse:
         """
-        List all spans for a given trace ID
+        List spans, optionally filtered by trace_id and/or task_id
 
         Args:
           extra_headers: Send extra headers
@@ -259,6 +268,7 @@ def list(
                         "order_by": order_by,
                         "order_direction": order_direction,
                         "page_number": page_number,
+                        "task_id": task_id,
                         "trace_id": trace_id,
                     },
                     span_list_params.SpanListParams,
@@ -300,6 +310,7 @@ async def create(
         input: Union[Dict[str, object], Iterable[Dict[str, object]], None] | Omit = omit,
         output: Union[Dict[str, object], Iterable[Dict[str, object]], None] | Omit = omit,
         parent_id: Optional[str] | Omit = omit,
+        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,
@@ -329,6 +340,8 @@ async def create(
 
           parent_id: ID of the parent span if this is a child span in a trace
 
+          task_id: ID of the task this span belongs to
+
           extra_headers: Send extra headers
 
           extra_query: Add additional query parameters to the request
@@ -350,6 +363,7 @@ async def create(
                     "input": input,
                     "output": output,
                     "parent_id": parent_id,
+                    "task_id": task_id,
                 },
                 span_create_params.SpanCreateParams,
             ),
@@ -403,6 +417,7 @@ async def update(
         output: Union[Dict[str, object], Iterable[Dict[str, object]], None] | Omit = omit,
         parent_id: Optional[str] | Omit = omit,
         start_time: Union[str, datetime, None] | Omit = omit,
+        task_id: Optional[str] | Omit = omit,
         trace_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.
@@ -429,6 +444,8 @@ async def update(
 
           start_time: The time the span started
 
+          task_id: ID of the task this span belongs to
+
           trace_id: Unique identifier for the trace this span belongs to
 
           extra_headers: Send extra headers
@@ -452,6 +469,7 @@ async def update(
                     "output": output,
                     "parent_id": parent_id,
                     "start_time": start_time,
+                    "task_id": task_id,
                     "trace_id": trace_id,
                 },
                 span_update_params.SpanUpdateParams,
@@ -469,6 +487,7 @@ async def list(
         order_by: Optional[str] | Omit = omit,
         order_direction: str | Omit = omit,
         page_number: int | Omit = omit,
+        task_id: Optional[str] | Omit = omit,
         trace_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.
@@ -478,7 +497,7 @@ async def list(
         timeout: float | httpx.Timeout | None | NotGiven = not_given,
     ) -> SpanListResponse:
         """
-        List all spans for a given trace ID
+        List spans, optionally filtered by trace_id and/or task_id
 
         Args:
           extra_headers: Send extra headers
@@ -502,6 +521,7 @@ async def list(
                         "order_by": order_by,
                         "order_direction": order_direction,
                         "page_number": page_number,
+                        "task_id": task_id,
                         "trace_id": trace_id,
                     },
                     span_list_params.SpanListParams,

src/agentex/types/span.py

@@ -34,3 +34,6 @@ class Span(BaseModel):
 
     parent_id: Optional[str] = None
     """ID of the parent span if this is a child span in a trace"""
+
+    task_id: Optional[str] = None
+    """ID of the task this span belongs to"""

src/agentex/types/span_create_params.py

@@ -38,3 +38,6 @@ class SpanCreateParams(TypedDict, total=False):
 
     parent_id: Optional[str]
     """ID of the parent span if this is a child span in a trace"""
+
+    task_id: Optional[str]
+    """ID of the task this span belongs to"""

src/agentex/types/span_list_params.py

@@ -17,4 +17,6 @@ class SpanListParams(TypedDict, total=False):
 
     page_number: int
 
+    task_id: Optional[str]
+
     trace_id: Optional[str]

src/agentex/types/span_update_params.py

@@ -33,5 +33,8 @@ class SpanUpdateParams(TypedDict, total=False):
     start_time: Annotated[Union[str, datetime, None], PropertyInfo(format="iso8601")]
     """The time the span started"""
 
+    task_id: Optional[str]
+    """ID of the task this span belongs to"""
+
     trace_id: Optional[str]
     """Unique identifier for the trace this span belongs to"""

tests/api_resources/test_spans.py

@@ -42,6 +42,7 @@ def test_method_create_with_all_params(self, client: Agentex) -> None:
             input={"foo": "bar"},
             output={"foo": "bar"},
             parent_id="parent_id",
+            task_id="task_id",
         )
         assert_matches_type(Span, span, path=["response"])
 
@@ -137,6 +138,7 @@ def test_method_update_with_all_params(self, client: Agentex) -> None:
             output={"foo": "bar"},
             parent_id="parent_id",
             start_time=parse_datetime("2019-12-27T18:11:19.117Z"),
+            task_id="task_id",
             trace_id="trace_id",
         )
         assert_matches_type(Span, span, path=["response"])
@@ -189,6 +191,7 @@ def test_method_list_with_all_params(self, client: Agentex) -> None:
             order_by="order_by",
             order_direction="order_direction",
             page_number=1,
+            task_id="task_id",
             trace_id="trace_id",
         )
         assert_matches_type(SpanListResponse, span, path=["response"])
@@ -244,6 +247,7 @@ async def test_method_create_with_all_params(self, async_client: AsyncAgentex) -
             input={"foo": "bar"},
             output={"foo": "bar"},
             parent_id="parent_id",
+            task_id="task_id",
         )
         assert_matches_type(Span, span, path=["response"])
 
@@ -339,6 +343,7 @@ async def test_method_update_with_all_params(self, async_client: AsyncAgentex) -
             output={"foo": "bar"},
             parent_id="parent_id",
             start_time=parse_datetime("2019-12-27T18:11:19.117Z"),
+            task_id="task_id",
             trace_id="trace_id",
         )
         assert_matches_type(Span, span, path=["response"])
@@ -391,6 +396,7 @@ async def test_method_list_with_all_params(self, async_client: AsyncAgentex) ->
             order_by="order_by",
             order_direction="order_direction",
             page_number=1,
+            task_id="task_id",
             trace_id="trace_id",
         )
         assert_matches_type(SpanListResponse, span, path=["response"])