feat: make custom code changes (#3)

These are the changes we had made on top of whatever Stainless
generates. We are going to get them into this repository now so (a)
future updates can just be simple PRs from Stainless and (b) we can
directly publish from the repository.

Author: ashwinb

.github/CODEOWNERS

@@ -0,0 +1 @@
+@ehhuang @ashwinb @raghotham @reluctantfuturist

src/llama_stack_client/__init__.py

@@ -39,6 +39,12 @@
 from ._base_client import DefaultHttpxClient, DefaultAioHttpClient, DefaultAsyncHttpxClient
 from ._utils._logs import setup_logging as _setup_logging
 
+from .lib.agents.agent import Agent
+from .lib.agents.event_logger import EventLogger as AgentEventLogger
+from .lib.inference.event_logger import EventLogger as InferenceEventLogger
+from .types.agents.turn_create_params import Document
+from .types.shared_params.document import Document as RAGDocument
+
 __all__ = [
     "types",
     "__version__",

src/llama_stack_client/_client.py

@@ -1,6 +1,7 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 from __future__ import annotations
+import json
 
 import os
 from typing import Any, Union, Mapping
@@ -126,6 +127,7 @@ def __init__(
         # outlining your use-case to help us decide if it should be
         # part of our public interface in the future.
         _strict_response_validation: bool = False,
+        provider_data: Mapping[str, Any] | None = None,
     ) -> None:
         """Construct a new synchronous LlamaStackClient client instance.
 
@@ -140,13 +142,18 @@ def __init__(
         if base_url is None:
             base_url = f"http://any-hosted-llama-stack.com"
 
+        custom_headers = default_headers or {}
+        custom_headers["X-LlamaStack-Client-Version"] = __version__
+        if provider_data is not None:
+            custom_headers["X-LlamaStack-Provider-Data"] = json.dumps(provider_data)
+
         super().__init__(
             version=__version__,
             base_url=base_url,
             max_retries=max_retries,
             timeout=timeout,
             http_client=http_client,
-            custom_headers=default_headers,
+            custom_headers=custom_headers,
             custom_query=default_query,
             _strict_response_validation=_strict_response_validation,
         )
@@ -344,6 +351,7 @@ def __init__(
         # outlining your use-case to help us decide if it should be
         # part of our public interface in the future.
         _strict_response_validation: bool = False,
+        provider_data: Mapping[str, Any] | None = None,
     ) -> None:
         """Construct a new async AsyncLlamaStackClient client instance.
 
@@ -358,13 +366,18 @@ def __init__(
         if base_url is None:
             base_url = f"http://any-hosted-llama-stack.com"
 
+        custom_headers = default_headers or {}
+        custom_headers["X-LlamaStack-Client-Version"] = __version__
+        if provider_data is not None:
+            custom_headers["X-LlamaStack-Provider-Data"] = json.dumps(provider_data)
+
         super().__init__(
             version=__version__,
             base_url=base_url,
             max_retries=max_retries,
             timeout=timeout,
             http_client=http_client,
-            custom_headers=default_headers,
+            custom_headers=custom_headers,
             custom_query=default_query,
             _strict_response_validation=_strict_response_validation,
         )

src/llama_stack_client/_utils/_logs.py

@@ -1,5 +1,6 @@
 import os
 import logging
+from rich.logging import RichHandler
 
 logger: logging.Logger = logging.getLogger("llama_stack_client")
 httpx_logger: logging.Logger = logging.getLogger("httpx")
@@ -10,6 +11,7 @@ def _basic_config() -> None:
     logging.basicConfig(
         format="[%(asctime)s - %(name)s:%(lineno)d - %(levelname)s] %(message)s",
         datefmt="%Y-%m-%d %H:%M:%S",
+        handlers=[RichHandler(rich_tracebacks=True)],
     )
 
 

src/llama_stack_client/_version.py

@@ -1,4 +1,4 @@
 # File generated from our OpenAPI spec by Stainless. See CONTRIBUTING.md for details.
 
 __title__ = "llama_stack_client"
-__version__ = "0.1.0-alpha.2"  # x-release-please-version
+__version__ = "0.2.12"

src/llama_stack_client/lib/.keep

@@ -1,4 +1,4 @@
 File generated from our OpenAPI spec by Stainless.
 
 This directory can be used to store custom files to expand the SDK.
-It is ignored by Stainless code generation and its content (other than this keep file) won't be touched.
+It is ignored by Stainless code generation and its content (other than this keep file) won't be touched.

src/llama_stack_client/lib/__init__.py

@@ -0,0 +1,9 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.
+
+from .tools.mcp_oauth import get_oauth_token_for_mcp_server
+
+__all__ = ["get_oauth_token_for_mcp_server"]

src/llama_stack_client/lib/agents/__init__.py

@@ -0,0 +1,5 @@
+# Copyright (c) Meta Platforms, Inc. and affiliates.
+# All rights reserved.
+#
+# This source code is licensed under the terms described in the LICENSE file in
+# the root directory of this source tree.