swarmauri_llm_azureopenai 0.11.0.dev2__tar.gz

swarmauri_llm_azureopenai-0.11.0.dev2/PKG-INFO

@@ -0,0 +1,79 @@
+Metadata-Version: 2.4
+Name: swarmauri_llm_azureopenai
+Version: 0.11.0.dev2
+Summary: Provider-native Swarmauri chat and tool-calling adapters for Azure OpenAI.
+License-Expression: Apache-2.0
+Keywords: swarmauri,llm,azure openai,microsoft foundry,entra id,tool calling,streaming
+Author: Jacob Stewart
+Author-email: jacob@swarmauri.com
+Requires-Python: >=3.10,<3.13
+Classifier: Development Status :: 1 - Planning
+Classifier: License :: OSI Approved :: Apache Software License
+Classifier: Programming Language :: Python :: 3.10
+Classifier: Programming Language :: Python :: 3.11
+Classifier: Programming Language :: Python :: 3.12
+Requires-Dist: swarmauri_base (>=0.11.0.dev5)
+Requires-Dist: swarmauri_core (>=0.11.0.dev5)
+Requires-Dist: swarmauri_standard (>=0.11.0.dev5)
+Requires-Dist: swarmauri_transport_httpx (>=0.1.0.dev4)
+Requires-Dist: swarmauri_transport_retry (>=0.1.0.dev1)
+Description-Content-Type: text/markdown
+
+![Swarmauri](https://raw.githubusercontent.com/swarmauri/swarmauri-sdk/3d4d1cfa949399d7019ae9d8f296afba773dfb7f/assets/swarmauri.brand.theme.svg)
+
+<p align="center">
+  <a href="https://pepy.tech/project/swarmauri_llm_azureopenai"><img src="https://static.pepy.tech/badge/swarmauri_llm_azureopenai/month" alt="Downloads"></a>
+  <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/community/swarmauri_llm_azureopenai/"><img src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/community/swarmauri_llm_azureopenai.svg" alt="Hits"></a>
+  <a href="https://pypi.org/project/swarmauri_llm_azureopenai/"><img src="https://img.shields.io/badge/python-3.10%20%7C%203.11%20%7C%203.12-blue" alt="Python"></a>
+  <a href="https://pypi.org/project/swarmauri_llm_azureopenai/"><img src="https://img.shields.io/pypi/l/swarmauri_llm_azureopenai" alt="License"></a>
+  <a href="https://pypi.org/project/swarmauri_llm_azureopenai/"><img src="https://img.shields.io/pypi/v/swarmauri_llm_azureopenai" alt="Release"></a>
+</p>
+
+# Swarmauri Azure OpenAI LLM
+
+`swarmauri_llm_azureopenai` provides provider-native Swarmauri adapters for Azure OpenAI and Microsoft Foundry chat-completions deployments.
+
+## Features
+
+- `AzureOpenAIModel` directly implements the Swarmauri LLM base contract.
+- `AzureOpenAIToolModel` directly implements the tool-LLM base contract.
+- API-key authentication with Azure's `api-key` header.
+- Refreshable Microsoft Entra bearer-token providers kept outside serialized state.
+- Sync, async, streaming, and batch workflows against the current `/openai/v1` API.
+- Persistent, injectable Swarmauri HTTP clients with conservative rate-limit retries.
+
+## Installation
+
+```bash
+uv add swarmauri_llm_azureopenai
+```
+
+```bash
+pip install swarmauri_llm_azureopenai
+```
+
+## Usage
+
+```python
+import os
+
+from swarmauri_llm_azureopenai import AzureOpenAIModel
+from swarmauri_standard.conversations.Conversation import Conversation
+from swarmauri_standard.messages.HumanMessage import HumanMessage
+
+conversation = Conversation(messages=[HumanMessage(content="Summarize this release.")])
+model = AzureOpenAIModel(
+    endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
+    api_key=os.environ["AZURE_OPENAI_API_KEY"],
+    name="my-gpt-deployment",
+)
+model.predict(conversation)
+print(conversation.get_last().content)
+```
+
+For keyless authentication, omit `api_key` and pass a callable as `token_provider`; it is invoked for every request so refreshed Entra tokens are used. Use `AzureOpenAIToolModel` with a Swarmauri `Toolkit` for function calling.
+
+Pass an `IHttpClient` as `http_client` to integrate a custom transport or to test without network access. Pass an `IHttpRetryPolicy` as `retry_policy` to replace the default policy, which retries safe reads and rate-limited chat submissions without replaying failed POST requests for generic server errors.
+
+See the [Azure OpenAI chat REST reference](https://learn.microsoft.com/en-us/rest/api/microsoft-foundry/azureopenai/chat) and [Entra authentication guide](https://learn.microsoft.com/en-us/azure/foundry/foundry-models/how-to/configure-entra-id).
+

swarmauri_llm_azureopenai-0.11.0.dev2/README.md

@@ -0,0 +1,57 @@
+![Swarmauri](https://raw.githubusercontent.com/swarmauri/swarmauri-sdk/3d4d1cfa949399d7019ae9d8f296afba773dfb7f/assets/swarmauri.brand.theme.svg)
+
+<p align="center">
+  <a href="https://pepy.tech/project/swarmauri_llm_azureopenai"><img src="https://static.pepy.tech/badge/swarmauri_llm_azureopenai/month" alt="Downloads"></a>
+  <a href="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/community/swarmauri_llm_azureopenai/"><img src="https://hits.sh/github.com/swarmauri/swarmauri-sdk/tree/master/pkgs/community/swarmauri_llm_azureopenai.svg" alt="Hits"></a>
+  <a href="https://pypi.org/project/swarmauri_llm_azureopenai/"><img src="https://img.shields.io/badge/python-3.10%20%7C%203.11%20%7C%203.12-blue" alt="Python"></a>
+  <a href="https://pypi.org/project/swarmauri_llm_azureopenai/"><img src="https://img.shields.io/pypi/l/swarmauri_llm_azureopenai" alt="License"></a>
+  <a href="https://pypi.org/project/swarmauri_llm_azureopenai/"><img src="https://img.shields.io/pypi/v/swarmauri_llm_azureopenai" alt="Release"></a>
+</p>
+
+# Swarmauri Azure OpenAI LLM
+
+`swarmauri_llm_azureopenai` provides provider-native Swarmauri adapters for Azure OpenAI and Microsoft Foundry chat-completions deployments.
+
+## Features
+
+- `AzureOpenAIModel` directly implements the Swarmauri LLM base contract.
+- `AzureOpenAIToolModel` directly implements the tool-LLM base contract.
+- API-key authentication with Azure's `api-key` header.
+- Refreshable Microsoft Entra bearer-token providers kept outside serialized state.
+- Sync, async, streaming, and batch workflows against the current `/openai/v1` API.
+- Persistent, injectable Swarmauri HTTP clients with conservative rate-limit retries.
+
+## Installation
+
+```bash
+uv add swarmauri_llm_azureopenai
+```
+
+```bash
+pip install swarmauri_llm_azureopenai
+```
+
+## Usage
+
+```python
+import os
+
+from swarmauri_llm_azureopenai import AzureOpenAIModel
+from swarmauri_standard.conversations.Conversation import Conversation
+from swarmauri_standard.messages.HumanMessage import HumanMessage
+
+conversation = Conversation(messages=[HumanMessage(content="Summarize this release.")])
+model = AzureOpenAIModel(
+    endpoint=os.environ["AZURE_OPENAI_ENDPOINT"],
+    api_key=os.environ["AZURE_OPENAI_API_KEY"],
+    name="my-gpt-deployment",
+)
+model.predict(conversation)
+print(conversation.get_last().content)
+```
+
+For keyless authentication, omit `api_key` and pass a callable as `token_provider`; it is invoked for every request so refreshed Entra tokens are used. Use `AzureOpenAIToolModel` with a Swarmauri `Toolkit` for function calling.
+
+Pass an `IHttpClient` as `http_client` to integrate a custom transport or to test without network access. Pass an `IHttpRetryPolicy` as `retry_policy` to replace the default policy, which retries safe reads and rate-limited chat submissions without replaying failed POST requests for generic server errors.
+
+See the [Azure OpenAI chat REST reference](https://learn.microsoft.com/en-us/rest/api/microsoft-foundry/azureopenai/chat) and [Entra authentication guide](https://learn.microsoft.com/en-us/azure/foundry/foundry-models/how-to/configure-entra-id).

swarmauri_llm_azureopenai-0.11.0.dev2/pyproject.toml

@@ -0,0 +1,46 @@
+[project]
+name = "swarmauri_llm_azureopenai"
+version = "0.11.0.dev2"
+description = "Provider-native Swarmauri chat and tool-calling adapters for Azure OpenAI."
+license = "Apache-2.0"
+readme = "README.md"
+requires-python = ">=3.10,<3.13"
+authors = [{ name = "Jacob Stewart", email = "jacob@swarmauri.com" }]
+keywords = ["swarmauri", "llm", "azure openai", "microsoft foundry", "entra id", "tool calling", "streaming"]
+classifiers = [
+  "Development Status :: 1 - Planning",
+  "License :: OSI Approved :: Apache Software License",
+  "Programming Language :: Python :: 3.10",
+  "Programming Language :: Python :: 3.11",
+  "Programming Language :: Python :: 3.12",
+]
+dependencies = [
+  "swarmauri_base>=0.11.0.dev5",
+  "swarmauri_core>=0.11.0.dev5",
+  "swarmauri_standard>=0.11.0.dev5",
+  "swarmauri_transport_httpx>=0.1.0.dev4",
+  "swarmauri_transport_retry>=0.1.0.dev1",
+]
+
+[tool.uv.sources]
+swarmauri_base = { workspace = true }
+swarmauri_core = { workspace = true }
+swarmauri_standard = { workspace = true }
+swarmauri_transport_httpx = { workspace = true }
+swarmauri_transport_retry = { workspace = true }
+
+[project.entry-points."swarmauri.llms"]
+AzureOpenAIModel = "swarmauri_llm_azureopenai:AzureOpenAIModel"
+
+[project.entry-points."swarmauri.tool_llms"]
+AzureOpenAIToolModel = "swarmauri_llm_azureopenai:AzureOpenAIToolModel"
+
+[build-system]
+requires = ["poetry-core>=1.0.0"]
+build-backend = "poetry.core.masonry.api"
+
+[dependency-groups]
+dev = ["pytest>=8.0", "pytest-asyncio>=0.24", "ruff>=0.9.9"]
+
+[tool.pytest.ini_options]
+markers = ["unit: Unit tests"]

swarmauri_llm_azureopenai-0.11.0.dev2/swarmauri_llm_azureopenai/AzureOpenAIChatRetryPolicy.py

@@ -0,0 +1,34 @@
+"""Retry policy for Azure OpenAI chat operations."""
+
+from swarmauri_core.transports.i_http_response import IHttpResponse
+from swarmauri_transport_retry import ExponentialBackoffRetryPolicy
+
+
+class AzureOpenAIChatRetryPolicy(ExponentialBackoffRetryPolicy):
+    """Retry safe reads and rate-limited chat submissions."""
+
+    def __init__(self, **kwargs) -> None:
+        kwargs.setdefault("retryable_methods", {"GET", "POST"})
+        super().__init__(**kwargs)
+
+    def should_retry(
+        self,
+        *,
+        method: str,
+        attempt: int,
+        response: IHttpResponse | None = None,
+        error: Exception | None = None,
+    ) -> bool:
+        if method.upper() == "POST":
+            return (
+                attempt < self.max_attempts
+                and error is None
+                and response is not None
+                and response.status_code == 429
+            )
+        return super().should_retry(
+            method=method,
+            attempt=attempt,
+            response=response,
+            error=error,
+        )

swarmauri_llm_azureopenai-0.11.0.dev2/swarmauri_llm_azureopenai/AzureOpenAIModel.py

@@ -0,0 +1,111 @@
+from typing import Any, Callable, ClassVar, FrozenSet, List, Literal, Optional
+
+from pydantic import Field, PrivateAttr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.llms.LLMBase import LLMBase
+from swarmauri_core.transports.i_http_client import IHttpClient
+from swarmauri_core.transports.i_http_retry_policy import IHttpRetryPolicy
+from swarmauri_core.transports.i_managed_http_client import IManagedHttpClient
+from swarmauri_standard.utils.provider_chat_transport import (
+    abatch_chat,
+    apredict_chat,
+    astream_chat,
+    batch_chat,
+    predict_chat,
+    stream_chat,
+)
+
+from ._client import build_azure_openai_client
+
+
+@ComponentBase.register_type()
+class AzureOpenAIModel(LLMBase):
+    """Provider-native Azure OpenAI chat-completions model."""
+
+    type: Literal["AzureOpenAIModel"] = "AzureOpenAIModel"
+    endpoint: str = Field(min_length=1)
+    api_version: str = "v1"
+    name: str = Field(min_length=1)
+    discover_models: bool = False
+    capabilities: ClassVar[FrozenSet[str]] = frozenset(
+        {"chat_completions", "structured_output", "streaming", "tools"}
+    )
+    _token_provider: Optional[Callable[[], Any]] = PrivateAttr(default=None)
+    _http_client: IHttpClient = PrivateAttr()
+    _retry_owned_by_transport: bool = PrivateAttr(default=True)
+
+    predict = predict_chat
+    apredict = apredict_chat
+    stream = stream_chat
+    astream = astream_chat
+    batch = batch_chat
+    abatch = abatch_chat
+
+    def __init__(
+        self,
+        *,
+        token_provider: Optional[Callable[[], Any]] = None,
+        http_client: IHttpClient | None = None,
+        retry_policy: IHttpRetryPolicy | None = None,
+        **data: Any,
+    ) -> None:
+        selected_name = data.get("name")
+        if selected_name:
+            data.setdefault("allowed_models", [selected_name])
+        super().__init__(**data)
+        self._token_provider = token_provider
+        if self.api_key is None and self._token_provider is None:
+            raise ValueError("api_key or token_provider is required")
+        self._http_client = build_azure_openai_client(
+            base_url=self._api_root(),
+            timeout=self.timeout,
+            max_attempts=self.max_retries,
+            http_client=http_client,
+            retry_policy=retry_policy,
+        )
+        if self.discover_models:
+            self.allowed_models = self._get_models()
+        elif not self.allowed_models:
+            self.allowed_models = [self.name]
+
+    def _api_root(self) -> str:
+        root = self.endpoint.rstrip("/")
+        suffix = f"/openai/{self.api_version}"
+        return root if root.endswith(suffix) else f"{root}{suffix}"
+
+    def _build_endpoint(self) -> str:
+        return f"{self._api_root()}/chat/completions"
+
+    def _build_headers(self) -> dict[str, str]:
+        headers = {"Content-Type": "application/json"}
+        if self.api_key is not None:
+            headers["api-key"] = self.api_key.get_secret_value()
+        else:
+            token = self._token_provider()
+            headers["Authorization"] = (
+                f"Bearer {getattr(token, 'token', token)}"
+            )
+        return headers
+
+    def _get_models(self) -> List[str]:
+        if not self.discover_models:
+            return [self.name]
+        response = self._http_client.request(
+            "GET",
+            f"{self._api_root()}/models",
+            headers=self._build_headers(),
+        )
+        response.raise_for_status()
+        return [
+            item["id"]
+            for item in response.json().get("data", [])
+            if item.get("id")
+        ]
+
+    def close(self) -> None:
+        if isinstance(self._http_client, IManagedHttpClient):
+            self._http_client.close()
+
+    async def aclose(self) -> None:
+        if isinstance(self._http_client, IManagedHttpClient):
+            await self._http_client.aclose()

swarmauri_llm_azureopenai-0.11.0.dev2/swarmauri_llm_azureopenai/AzureOpenAIToolModel.py

@@ -0,0 +1,121 @@
+from typing import Any, Callable, ClassVar, FrozenSet, List, Literal, Optional
+
+from pydantic import Field, PrivateAttr
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.tool_llms.ToolLLMBase import ToolLLMBase
+from swarmauri_core.transports.i_http_client import IHttpClient
+from swarmauri_core.transports.i_http_retry_policy import IHttpRetryPolicy
+from swarmauri_core.transports.i_managed_http_client import IManagedHttpClient
+from swarmauri_standard.schema_converters.OpenAISchemaConverter import (
+    OpenAISchemaConverter,
+)
+
+from ._client import build_azure_openai_client
+from swarmauri_standard.utils.provider_chat_transport import (
+    abatch_tools,
+    apredict_tools,
+    astream_tools,
+    batch_tools,
+    format_messages,
+    predict_tools,
+    stream_tools,
+)
+
+
+@ComponentBase.register_type()
+class AzureOpenAIToolModel(ToolLLMBase):
+    """Provider-native Azure OpenAI tool-calling model."""
+
+    type: Literal["AzureOpenAIToolModel"] = "AzureOpenAIToolModel"
+    endpoint: str = Field(min_length=1)
+    api_version: str = "v1"
+    name: str = Field(min_length=1)
+    discover_models: bool = False
+    capabilities: ClassVar[FrozenSet[str]] = frozenset(
+        {"chat_completions", "streaming", "tools"}
+    )
+    _token_provider: Optional[Callable[[], Any]] = PrivateAttr(default=None)
+    _http_client: IHttpClient = PrivateAttr()
+    _retry_owned_by_transport: bool = PrivateAttr(default=True)
+
+    predict = predict_tools
+    apredict = apredict_tools
+    stream = stream_tools
+    astream = astream_tools
+    batch = batch_tools
+    abatch = abatch_tools
+
+    def __init__(
+        self,
+        *,
+        token_provider: Optional[Callable[[], Any]] = None,
+        http_client: IHttpClient | None = None,
+        retry_policy: IHttpRetryPolicy | None = None,
+        **data: Any,
+    ) -> None:
+        selected_name = data.get("name")
+        if selected_name:
+            data.setdefault("allowed_models", [selected_name])
+        super().__init__(**data)
+        self._token_provider = token_provider
+        if self.api_key is None and self._token_provider is None:
+            raise ValueError("api_key or token_provider is required")
+        self._http_client = build_azure_openai_client(
+            base_url=self._api_root(),
+            timeout=self.timeout,
+            max_attempts=self.max_retries,
+            http_client=http_client,
+            retry_policy=retry_policy,
+        )
+        if self.discover_models:
+            self.allowed_models = self._get_models()
+        elif not self.allowed_models:
+            self.allowed_models = [self.name]
+
+    def get_schema_converter(self):
+        return OpenAISchemaConverter()
+
+    def _format_messages(self, messages):
+        return format_messages(messages)
+
+    def _api_root(self) -> str:
+        root = self.endpoint.rstrip("/")
+        suffix = f"/openai/{self.api_version}"
+        return root if root.endswith(suffix) else f"{root}{suffix}"
+
+    def _build_endpoint(self) -> str:
+        return f"{self._api_root()}/chat/completions"
+
+    def _build_headers(self) -> dict[str, str]:
+        headers = {"Content-Type": "application/json"}
+        if self.api_key is not None:
+            headers["api-key"] = self.api_key.get_secret_value()
+        else:
+            token = self._token_provider()
+            headers["Authorization"] = (
+                f"Bearer {getattr(token, 'token', token)}"
+            )
+        return headers
+
+    def _get_models(self) -> List[str]:
+        if not self.discover_models:
+            return [self.name]
+        response = self._http_client.request(
+            "GET",
+            f"{self._api_root()}/models",
+            headers=self._build_headers(),
+        )
+        response.raise_for_status()
+        return [
+            item["id"]
+            for item in response.json().get("data", [])
+            if item.get("id")
+        ]
+
+    def close(self) -> None:
+        if isinstance(self._http_client, IManagedHttpClient):
+            self._http_client.close()
+
+    async def aclose(self) -> None:
+        if isinstance(self._http_client, IManagedHttpClient):
+            await self._http_client.aclose()

swarmauri_llm_azureopenai-0.11.0.dev2/swarmauri_llm_azureopenai/__init__.py

@@ -0,0 +1,16 @@
+from importlib.metadata import PackageNotFoundError, version
+
+from .AzureOpenAIChatRetryPolicy import AzureOpenAIChatRetryPolicy
+from .AzureOpenAIModel import AzureOpenAIModel
+from .AzureOpenAIToolModel import AzureOpenAIToolModel
+
+try:
+    __version__ = version("swarmauri_llm_azureopenai")
+except PackageNotFoundError:
+    __version__ = "0.0.0"
+
+__all__ = [
+    "AzureOpenAIChatRetryPolicy",
+    "AzureOpenAIModel",
+    "AzureOpenAIToolModel",
+]

swarmauri_llm_azureopenai-0.11.0.dev2/swarmauri_llm_azureopenai/_client.py

@@ -0,0 +1,29 @@
+"""HTTP client composition for Azure OpenAI models."""
+
+from swarmauri_core.transports.i_http_client import IHttpClient
+from swarmauri_core.transports.i_http_retry_policy import IHttpRetryPolicy
+from swarmauri_transport_httpx import HttpxClient
+from swarmauri_transport_retry import RetryingHttpClient
+
+from .AzureOpenAIChatRetryPolicy import AzureOpenAIChatRetryPolicy
+
+
+def build_azure_openai_client(
+    *,
+    base_url: str,
+    timeout: float,
+    max_attempts: int,
+    http_client: IHttpClient | None,
+    retry_policy: IHttpRetryPolicy | None,
+) -> IHttpClient:
+    """Build a persistent client while preserving injected ownership."""
+    concrete_client = http_client or HttpxClient(
+        base_url=base_url,
+        timeout=timeout,
+    )
+    return RetryingHttpClient(
+        client=concrete_client,
+        policy=retry_policy
+        or AzureOpenAIChatRetryPolicy(max_attempts=max(max_attempts, 1)),
+        own_client=http_client is None,
+    )