agentex-sdk 0.9.1__tar.gz → 0.9.2__tar.gz

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

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

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/CHANGELOG.md

@@ -1,5 +1,18 @@
 # Changelog
 
+## 0.9.2 (2026-02-06)
+
+Full Changelog: [v0.9.1...v0.9.2](https://github.com/scaleapi/scale-agentex-python/compare/v0.9.1...v0.9.2)
+
+### Features
+
+* **client:** add custom JSON encoder for extended type support ([a0720ab](https://github.com/scaleapi/scale-agentex-python/commit/a0720abb088583ce4b596e464f7483a4be728e29))
+
+
+### Bug Fixes
+
+* add litellm retry with exponential backoff for rate limit errors ([ccdb24a](https://github.com/scaleapi/scale-agentex-python/commit/ccdb24a08607298f8dafd748ee9e7fe8ba13d5fe))
+
 ## 0.9.1 (2026-01-26)
 
 Full Changelog: [v0.9.0...v0.9.1](https://github.com/scaleapi/scale-agentex-python/compare/v0.9.0...v0.9.1)

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: agentex-sdk
-Version: 0.9.1
+Version: 0.9.2
 Summary: The official Python library for the agentex API
 Project-URL: Homepage, https://github.com/scaleapi/scale-agentex-python
 Project-URL: Repository, https://github.com/scaleapi/scale-agentex-python

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/pyproject.toml

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

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/src/agentex/_base_client.py

@@ -86,6 +86,7 @@ from ._exceptions import (
     APIConnectionError,
     APIResponseValidationError,
 )
+from ._utils._json import openapi_dumps
 
 log: logging.Logger = logging.getLogger(__name__)
 
@@ -554,8 +555,10 @@ class BaseClient(Generic[_HttpxClientT, _DefaultStreamT]):
                 kwargs["content"] = options.content
             elif isinstance(json_data, bytes):
                 kwargs["content"] = json_data
-            else:
-                kwargs["json"] = json_data if is_given(json_data) else None
+            elif not files:
+                # Don't set content when JSON is sent as multipart/form-data,
+                # since httpx's content param overrides other body arguments
+                kwargs["content"] = openapi_dumps(json_data) if is_given(json_data) and json_data is not None else None
             kwargs["files"] = files
         else:
             headers.pop("Content-Type", None)

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/src/agentex/_compat.py

@@ -139,6 +139,7 @@ def model_dump(
     exclude_defaults: bool = False,
     warnings: bool = True,
     mode: Literal["json", "python"] = "python",
+    by_alias: bool | None = None,
 ) -> dict[str, Any]:
     if (not PYDANTIC_V1) or hasattr(model, "model_dump"):
         return model.model_dump(
@@ -148,13 +149,12 @@ def model_dump(
             exclude_defaults=exclude_defaults,
             # warnings are not supported in Pydantic v1
             warnings=True if PYDANTIC_V1 else warnings,
+            by_alias=by_alias,
         )
     return cast(
         "dict[str, Any]",
         model.dict(  # pyright: ignore[reportDeprecated, reportUnnecessaryCast]
-            exclude=exclude,
-            exclude_unset=exclude_unset,
-            exclude_defaults=exclude_defaults,
+            exclude=exclude, exclude_unset=exclude_unset, exclude_defaults=exclude_defaults, by_alias=bool(by_alias)
         ),
     )
 

agentex_sdk-0.9.2/src/agentex/_utils/_json.py

@@ -0,0 +1,35 @@
+import json
+from typing import Any
+from datetime import datetime
+from typing_extensions import override
+
+import pydantic
+
+from .._compat import model_dump
+
+
+def openapi_dumps(obj: Any) -> bytes:
+    """
+    Serialize an object to UTF-8 encoded JSON bytes.
+
+    Extends the standard json.dumps with support for additional types
+    commonly used in the SDK, such as `datetime`, `pydantic.BaseModel`, etc.
+    """
+    return json.dumps(
+        obj,
+        cls=_CustomEncoder,
+        # Uses the same defaults as httpx's JSON serialization
+        ensure_ascii=False,
+        separators=(",", ":"),
+        allow_nan=False,
+    ).encode()
+
+
+class _CustomEncoder(json.JSONEncoder):
+    @override
+    def default(self, o: Any) -> Any:
+        if isinstance(o, datetime):
+            return o.isoformat()
+        if isinstance(o, pydantic.BaseModel):
+            return model_dump(o, exclude_unset=True, mode="json", by_alias=True)
+        return super().default(o)

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/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.9.1"  # x-release-please-version
+__version__ = "0.9.2"  # x-release-please-version

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/src/agentex/lib/adk/providers/_modules/litellm.py

@@ -26,7 +26,13 @@ from agentex.lib.core.temporal.activities.adk.providers.litellm_activities impor
 logger = make_logger(__name__)
 
 # Default retry policy for all LiteLLM operations
-DEFAULT_RETRY_POLICY = RetryPolicy(maximum_attempts=1)
+# Retries with exponential backoff: 1s, 2s, 4s, ... up to 30s between attempts
+DEFAULT_RETRY_POLICY = RetryPolicy(
+    maximum_attempts=3,
+    initial_interval=timedelta(seconds=1),
+    backoff_coefficient=2.0,
+    maximum_interval=timedelta(seconds=30),
+)
 
 
 class LiteLLMModule:

{agentex_sdk-0.9.1 → agentex_sdk-0.9.2}/src/agentex/lib/types/llm_messages.py

@@ -58,6 +58,7 @@ class LLMConfig(BaseModel):
     parallel_tool_calls: bool | None = None
     logprobs: bool | None = None
     top_logprobs: int | None = None
+    num_retries: int | None = 3
 
 
 class ContentPartText(BaseModel):

agentex_sdk-0.9.2/tests/test_utils/test_json.py

@@ -0,0 +1,126 @@
+from __future__ import annotations
+
+import datetime
+from typing import Union
+
+import pydantic
+
+from agentex import _compat
+from agentex._utils._json import openapi_dumps
+
+
+class TestOpenapiDumps:
+    def test_basic(self) -> None:
+        data = {"key": "value", "number": 42}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"key":"value","number":42}'
+
+    def test_datetime_serialization(self) -> None:
+        dt = datetime.datetime(2023, 1, 1, 12, 0, 0)
+        data = {"datetime": dt}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"datetime":"2023-01-01T12:00:00"}'
+
+    def test_pydantic_model_serialization(self) -> None:
+        class User(pydantic.BaseModel):
+            first_name: str
+            last_name: str
+            age: int
+
+        model_instance = User(first_name="John", last_name="Kramer", age=83)
+        data = {"model": model_instance}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"first_name":"John","last_name":"Kramer","age":83}}'
+
+    def test_pydantic_model_with_default_values(self) -> None:
+        class User(pydantic.BaseModel):
+            name: str
+            role: str = "user"
+            active: bool = True
+            score: int = 0
+
+        model_instance = User(name="Alice")
+        data = {"model": model_instance}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"name":"Alice"}}'
+
+    def test_pydantic_model_with_default_values_overridden(self) -> None:
+        class User(pydantic.BaseModel):
+            name: str
+            role: str = "user"
+            active: bool = True
+
+        model_instance = User(name="Bob", role="admin", active=False)
+        data = {"model": model_instance}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"name":"Bob","role":"admin","active":false}}'
+
+    def test_pydantic_model_with_alias(self) -> None:
+        class User(pydantic.BaseModel):
+            first_name: str = pydantic.Field(alias="firstName")
+            last_name: str = pydantic.Field(alias="lastName")
+
+        model_instance = User(firstName="John", lastName="Doe")
+        data = {"model": model_instance}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"firstName":"John","lastName":"Doe"}}'
+
+    def test_pydantic_model_with_alias_and_default(self) -> None:
+        class User(pydantic.BaseModel):
+            user_name: str = pydantic.Field(alias="userName")
+            user_role: str = pydantic.Field(default="member", alias="userRole")
+            is_active: bool = pydantic.Field(default=True, alias="isActive")
+
+        model_instance = User(userName="charlie")
+        data = {"model": model_instance}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"userName":"charlie"}}'
+
+        model_with_overrides = User(userName="diana", userRole="admin", isActive=False)
+        data = {"model": model_with_overrides}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"userName":"diana","userRole":"admin","isActive":false}}'
+
+    def test_pydantic_model_with_nested_models_and_defaults(self) -> None:
+        class Address(pydantic.BaseModel):
+            street: str
+            city: str = "Unknown"
+
+        class User(pydantic.BaseModel):
+            name: str
+            address: Address
+            verified: bool = False
+
+        if _compat.PYDANTIC_V1:
+            # to handle forward references in Pydantic v1
+            User.update_forward_refs(**locals())  # type: ignore[reportDeprecated]
+
+        address = Address(street="123 Main St")
+        user = User(name="Diana", address=address)
+        data = {"user": user}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"user":{"name":"Diana","address":{"street":"123 Main St"}}}'
+
+        address_with_city = Address(street="456 Oak Ave", city="Boston")
+        user_verified = User(name="Eve", address=address_with_city, verified=True)
+        data = {"user": user_verified}
+        json_bytes = openapi_dumps(data)
+        assert (
+            json_bytes == b'{"user":{"name":"Eve","address":{"street":"456 Oak Ave","city":"Boston"},"verified":true}}'
+        )
+
+    def test_pydantic_model_with_optional_fields(self) -> None:
+        class User(pydantic.BaseModel):
+            name: str
+            email: Union[str, None]
+            phone: Union[str, None]
+
+        model_with_none = User(name="Eve", email=None, phone=None)
+        data = {"model": model_with_none}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"name":"Eve","email":null,"phone":null}}'
+
+        model_with_values = User(name="Frank", email="frank@example.com", phone=None)
+        data = {"model": model_with_values}
+        json_bytes = openapi_dumps(data)
+        assert json_bytes == b'{"model":{"name":"Frank","email":"frank@example.com","phone":null}}'

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

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