fix: allow litellm security patch

Author: scale-ballen

pyproject.toml

@@ -9,7 +9,7 @@ authors = [
 ]
 
 dependencies = [
-    "httpx>=0.27.2,<0.28",
+    "httpx>=0.28.1,<0.29",
     "pydantic>=2.0.0, <3",
   "typing-extensions>=4.14, <5",
   "anyio>=3.5.0, <5",
@@ -28,7 +28,7 @@ dependencies = [
     "temporalio>=1.26.0,<2",
     "aiohttp>=3.10.10,<4",
     "redis>=5.2.0,<6",
-    "litellm>=1.83.0,<2",
+    "litellm>=1.83.7,<2",
     "kubernetes>=25.0.0,<36.0.0",
     "jinja2>=3.1.3,<4",
     "mcp[cli]>=1.4.1",

src/agentex/lib/core/tracing/processors/agentex_tracing_processor.py

@@ -26,6 +26,7 @@ def on_span_start(self, span: Span) -> None:
             input=span.input,
             output=span.output,
             parent_id=span.parent_id,
+            task_id=span.task_id,
         )
 
     @override
@@ -94,6 +95,7 @@ async def on_span_start(self, span: Span) -> None:
             input=span.input,
             output=span.output,
             data=span.data,
+            task_id=span.task_id,
         )
 
     @override

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,
@@ -300,6 +308,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 +338,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 +361,7 @@ async def create(
                     "input": input,
                     "output": output,
                     "parent_id": parent_id,
+                    "task_id": task_id,
                 },
                 span_create_params.SpanCreateParams,
             ),
@@ -403,6 +415,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 +442,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 +467,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,

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_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"])
@@ -244,6 +246,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 +342,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"])