Sync updates from stainless: a3a4573 (#249)

Syncing stainless changes as the title says. The repository is at:
https://github.com/llamastack/llama-stack-client-python/

Please see
#2 for the
source of the changes. The commit hash in the PR title is the commit
hash after the PR above landed in the source repository.

> [!NOTE]
> This repository will soon be deprecated. We will be moving source of
truth to https://github.com/llamastack/llama-stack-client-python/

Author: ashwinb

src/llama_stack_client/__init__.py

@@ -36,7 +36,7 @@
     UnprocessableEntityError,
     APIResponseValidationError,
 )
-from ._base_client import DefaultHttpxClient, DefaultAsyncHttpxClient
+from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
 from ._utils._logs import setup_logging as _setup_logging
 
 from .lib.agents.agent import Agent
@@ -84,6 +84,7 @@
     "DEFAULT_CONNECTION_LIMITS",
     "DefaultHttpxClient",
     "DefaultAsyncHttpxClient",
+    "DefaultAioHttpClient",
 ]
 
 if not _t.TYPE_CHECKING:

src/llama_stack_client/_base_client.py

@@ -1289,6 +1289,24 @@ def __init__(self, **kwargs: Any) -> None:
         super().__init__(**kwargs)
 
 
+try:
+    import httpx_aiohttp
+except ImportError:
+
+    class _DefaultAioHttpClient(httpx.AsyncClient):
+        def __init__(self, **_kwargs: Any) -> None:
+            raise RuntimeError("To use the aiohttp client you must have installed the package with the `aiohttp` extra")
+else:
+
+    class _DefaultAioHttpClient(httpx_aiohttp.HttpxAiohttpClient):  # type: ignore
+        def __init__(self, **kwargs: Any) -> None:
+            kwargs.setdefault("timeout", DEFAULT_TIMEOUT)
+            kwargs.setdefault("limits", DEFAULT_CONNECTION_LIMITS)
+            kwargs.setdefault("follow_redirects", True)
+
+            super().__init__(**kwargs)
+
+
 if TYPE_CHECKING:
     DefaultAsyncHttpxClient = httpx.AsyncClient
     """An alias to `httpx.AsyncClient` that provides the same defaults that this SDK
@@ -1297,8 +1315,12 @@ def __init__(self, **kwargs: Any) -> None:
     This is useful because overriding the `http_client` with your own instance of
     `httpx.AsyncClient` will result in httpx's defaults being used, not ours.
     """
+
+    DefaultAioHttpClient = httpx.AsyncClient
+    """An alias to `httpx.AsyncClient` that changes the default HTTP transport to `aiohttp`."""
 else:
     DefaultAsyncHttpxClient = _DefaultAsyncHttpxClient
+    DefaultAioHttpClient = _DefaultAioHttpClient
 
 
 class AsyncHttpxClientWrapper(DefaultAsyncHttpxClient):

src/llama_stack_client/_files.py

@@ -34,7 +34,7 @@ def assert_is_file_content(obj: object, *, key: str | None = None) -> None:
     if not is_file_content(obj):
         prefix = f"Expected entry at `{key}`" if key is not None else f"Expected file input `{obj!r}`"
         raise RuntimeError(
-            f"{prefix} to be bytes, an io.IOBase instance, PathLike or a tuple but received {type(obj)} instead. See https://github.com/stainless-sdks/llama-stack-python/tree/main#file-uploads"
+            f"{prefix} to be bytes, an io.IOBase instance, PathLike or a tuple but received {type(obj)} instead. See https://github.com/llamastack/llama-stack-client-python/tree/main#file-uploads"
         ) from None
 
 

src/llama_stack_client/resources/agents/agents.py

@@ -65,7 +65,7 @@ def with_raw_response(self) -> AgentsResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AgentsResourceWithRawResponse(self)
 
@@ -74,7 +74,7 @@ def with_streaming_response(self) -> AgentsResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AgentsResourceWithStreamingResponse(self)
 
@@ -166,7 +166,7 @@ def with_raw_response(self) -> AsyncAgentsResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncAgentsResourceWithRawResponse(self)
 
@@ -175,7 +175,7 @@ def with_streaming_response(self) -> AsyncAgentsResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncAgentsResourceWithStreamingResponse(self)
 

src/llama_stack_client/resources/agents/session.py

@@ -31,7 +31,7 @@ def with_raw_response(self) -> SessionResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return SessionResourceWithRawResponse(self)
 
@@ -40,7 +40,7 @@ def with_streaming_response(self) -> SessionResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return SessionResourceWithStreamingResponse(self)
 
@@ -169,7 +169,7 @@ def with_raw_response(self) -> AsyncSessionResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncSessionResourceWithRawResponse(self)
 
@@ -178,7 +178,7 @@ def with_streaming_response(self) -> AsyncSessionResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncSessionResourceWithStreamingResponse(self)
 

src/llama_stack_client/resources/agents/steps.py

@@ -26,7 +26,7 @@ def with_raw_response(self) -> StepsResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return StepsResourceWithRawResponse(self)
 
@@ -35,7 +35,7 @@ def with_streaming_response(self) -> StepsResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return StepsResourceWithStreamingResponse(self)
 
@@ -89,7 +89,7 @@ def with_raw_response(self) -> AsyncStepsResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncStepsResourceWithRawResponse(self)
 
@@ -98,7 +98,7 @@ def with_streaming_response(self) -> AsyncStepsResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncStepsResourceWithStreamingResponse(self)
 

src/llama_stack_client/resources/agents/turn.py

@@ -34,7 +34,7 @@ def with_raw_response(self) -> TurnResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return TurnResourceWithRawResponse(self)
 
@@ -43,7 +43,7 @@ def with_streaming_response(self) -> TurnResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return TurnResourceWithStreamingResponse(self)
 
@@ -428,7 +428,7 @@ def with_raw_response(self) -> AsyncTurnResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncTurnResourceWithRawResponse(self)
 
@@ -437,7 +437,7 @@ def with_streaming_response(self) -> AsyncTurnResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncTurnResourceWithStreamingResponse(self)
 

src/llama_stack_client/resources/benchmarks.py

@@ -32,7 +32,7 @@ def with_raw_response(self) -> BenchmarksResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return BenchmarksResourceWithRawResponse(self)
 
@@ -41,7 +41,7 @@ def with_streaming_response(self) -> BenchmarksResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return BenchmarksResourceWithStreamingResponse(self)
 
@@ -169,7 +169,7 @@ def with_raw_response(self) -> AsyncBenchmarksResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncBenchmarksResourceWithRawResponse(self)
 
@@ -178,7 +178,7 @@ def with_streaming_response(self) -> AsyncBenchmarksResourceWithStreamingRespons
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncBenchmarksResourceWithStreamingResponse(self)
 

src/llama_stack_client/resources/chat/chat.py

@@ -27,7 +27,7 @@ def with_raw_response(self) -> ChatResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return ChatResourceWithRawResponse(self)
 
@@ -36,7 +36,7 @@ def with_streaming_response(self) -> ChatResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return ChatResourceWithStreamingResponse(self)
 
@@ -52,7 +52,7 @@ def with_raw_response(self) -> AsyncChatResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncChatResourceWithRawResponse(self)
 
@@ -61,7 +61,7 @@ def with_streaming_response(self) -> AsyncChatResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncChatResourceWithStreamingResponse(self)
 

src/llama_stack_client/resources/chat/completions.py

@@ -35,7 +35,7 @@ def with_raw_response(self) -> CompletionsResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return CompletionsResourceWithRawResponse(self)
 
@@ -44,7 +44,7 @@ def with_streaming_response(self) -> CompletionsResourceWithStreamingResponse:
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return CompletionsResourceWithStreamingResponse(self)
 
@@ -515,7 +515,7 @@ def with_raw_response(self) -> AsyncCompletionsResourceWithRawResponse:
         This property can be used as a prefix for any HTTP method call to return
         the raw response object instead of the parsed content.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#accessing-raw-response-data-eg-headers
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#accessing-raw-response-data-eg-headers
         """
         return AsyncCompletionsResourceWithRawResponse(self)
 
@@ -524,7 +524,7 @@ def with_streaming_response(self) -> AsyncCompletionsResourceWithStreamingRespon
         """
         An alternative to `.with_raw_response` that doesn't eagerly read the response body.
 
-        For more information, see https://www.github.com/stainless-sdks/llama-stack-python#with_streaming_response
+        For more information, see https://www.github.com/llamastack/llama-stack-client-python#with_streaming_response
         """
         return AsyncCompletionsResourceWithStreamingResponse(self)