feat: make custom code changes

Author: ashwinb

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.