swarmauri_provider_amazon_nova 0.1.0.dev3__py3-none-any.whl

swarmauri_provider_amazon_nova/AmazonNovaAuthConfig.py

@@ -0,0 +1,39 @@
+from collections.abc import Mapping
+from typing import Literal
+
+from pydantic import Field, SecretStr
+from swarmauri_base.model_providers.ProviderAuthBase import ProviderAuthBase
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ProviderAuthBase, "AmazonNovaAuthConfig")
+class AmazonNovaAuthConfig(ProviderAuthBase):
+    access_key_id: SecretStr | None = Field(default=None, exclude=True)
+    secret_access_key: SecretStr | None = Field(default=None, exclude=True)
+    session_token: SecretStr | None = Field(default=None, exclude=True)
+    profile: str | None = None
+    type: Literal["AmazonNovaAuthConfig"] = "AmazonNovaAuthConfig"
+
+    def apply(self, headers: Mapping[str, str]) -> dict[str, str]:
+        """Preserve headers; the AWS transport applies request-aware SigV4."""
+        return dict(headers)
+
+    def sdk_credentials(self) -> dict[str, str]:
+        if (self.access_key_id is None) != (self.secret_access_key is None):
+            raise ValueError(
+                "AWS access_key_id and secret_access_key must be "
+                "provided together"
+            )
+        if self.access_key_id is None:
+            return {}
+        credentials = {
+            "aws_access_key_id": self.access_key_id.get_secret_value(),
+            "aws_secret_access_key": (
+                self.secret_access_key.get_secret_value()
+            ),
+        }
+        if self.session_token is not None:
+            credentials["aws_session_token"] = (
+                self.session_token.get_secret_value()
+            )
+        return credentials

swarmauri_provider_amazon_nova/AmazonNovaEndpoints.py

@@ -0,0 +1,27 @@
+from typing import Literal
+
+from swarmauri_base.model_providers.ProviderEndpointBase import (
+    ProviderEndpointBase,
+)
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ProviderEndpointBase, "AmazonNovaEndpoints")
+class AmazonNovaEndpoints(ProviderEndpointBase):
+    region: str = "us-east-1"
+    service: str = "bedrock-runtime"
+    type: Literal["AmazonNovaEndpoints"] = "AmazonNovaEndpoints"
+
+    def runtime_url(self) -> str:
+        return f"https://bedrock-runtime.{self.region}.amazonaws.com"
+
+    def resolve(self, capability: str) -> str:
+        if capability not in {
+            "bedrock.runtime",
+            "live.connect",
+            "model.invoke",
+        }:
+            raise ValueError(
+                f"Amazon Nova does not support capability {capability!r}"
+            )
+        return self.runtime_url()

swarmauri_provider_amazon_nova/AmazonNovaErrorMapper.py

@@ -0,0 +1,14 @@
+from typing import Any, Literal
+
+from swarmauri_base.model_providers.ProviderErrorMapperBase import (
+    ProviderErrorMapperBase,
+)
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ProviderErrorMapperBase, "AmazonNovaErrorMapper")
+class AmazonNovaErrorMapper(ProviderErrorMapperBase):
+    type: Literal["AmazonNovaErrorMapper"] = "AmazonNovaErrorMapper"
+
+    def map_error(self, error: BaseException | Any) -> BaseException:
+        return RuntimeError(f"Amazon Nova provider error: {error}")

swarmauri_provider_amazon_nova/AmazonNovaModelCatalog.py

@@ -0,0 +1,15 @@
+from typing import Literal
+
+from swarmauri_base.model_providers.ModelCatalogBase import ModelCatalogBase
+from swarmauri_base.ComponentBase import ComponentBase
+
+
+@ComponentBase.register_type(ModelCatalogBase, "AmazonNovaModelCatalog")
+class AmazonNovaModelCatalog(ModelCatalogBase):
+    models: dict = {
+        "amazon.nova-sonic-v1:0": {
+            "name": "amazon.nova-sonic-v1:0",
+            "features": ["realtime", "audio", "tools", "http2"],
+        }
+    }
+    type: Literal["AmazonNovaModelCatalog"] = "AmazonNovaModelCatalog"

swarmauri_provider_amazon_nova/AmazonNovaProvider.py

@@ -0,0 +1,86 @@
+from typing import Literal
+from urllib.parse import quote
+
+from pydantic import Field, SecretStr, model_validator
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.model_providers.ModelProviderBase import ModelProviderBase
+from swarmauri_base.model_providers.ProviderConnectionConfig import (
+    ProviderConnectionConfig,
+)
+
+from .AmazonNovaAuthConfig import AmazonNovaAuthConfig
+from .AmazonNovaEndpoints import AmazonNovaEndpoints
+from .AmazonNovaErrorMapper import AmazonNovaErrorMapper
+from .AmazonNovaModelCatalog import AmazonNovaModelCatalog
+
+
+@ComponentBase.register_type(ModelProviderBase, "AmazonNovaProvider")
+class AmazonNovaProvider(ModelProviderBase):
+    provider_name: Literal["amazon_nova"] = "amazon_nova"
+    access_key_id: SecretStr | None = Field(default=None, exclude=True)
+    secret_access_key: SecretStr | None = Field(default=None, exclude=True)
+    session_token: SecretStr | None = Field(default=None, exclude=True)
+    profile: str | None = None
+    auth: AmazonNovaAuthConfig = Field(
+        default_factory=AmazonNovaAuthConfig, exclude=True
+    )
+    endpoints: AmazonNovaEndpoints = Field(default_factory=AmazonNovaEndpoints)
+    catalog: AmazonNovaModelCatalog = Field(
+        default_factory=AmazonNovaModelCatalog
+    )
+    error_mapper: AmazonNovaErrorMapper = Field(
+        default_factory=AmazonNovaErrorMapper
+    )
+    type: Literal["AmazonNovaProvider"] = "AmazonNovaProvider"
+
+    @model_validator(mode="after")
+    def configure(self) -> "AmazonNovaProvider":
+        self.region = self.region or self.endpoints.region
+        self.endpoints.region = self.region
+        self.base_url = self.base_url or self.endpoints.resolve(
+            "bedrock.runtime"
+        )
+        if self.access_key_id is not None:
+            self.auth.access_key_id = self.access_key_id
+        elif self.auth.access_key_id is not None:
+            self.access_key_id = self.auth.access_key_id
+        if self.secret_access_key is not None:
+            self.auth.secret_access_key = self.secret_access_key
+        elif self.auth.secret_access_key is not None:
+            self.secret_access_key = self.auth.secret_access_key
+        if self.session_token is not None:
+            self.auth.session_token = self.session_token
+        elif self.auth.session_token is not None:
+            self.session_token = self.auth.session_token
+        if self.profile is not None:
+            self.auth.profile = self.profile
+        elif self.auth.profile is not None:
+            self.profile = self.auth.profile
+        return self
+
+    def connection_config(
+        self, capability: str, model: str | None = None
+    ) -> ProviderConnectionConfig:
+        endpoint = self.endpoints.resolve(capability)
+        options: dict[str, object] = {"region": self.region}
+        if self.profile is not None:
+            options["profile"] = self.profile
+        if model is not None:
+            options["model_id"] = model
+            endpoint = (
+                f"{endpoint.rstrip('/')}/model/{quote(model, safe='')}"
+                "/invoke-with-bidirectional-stream"
+            )
+        return ProviderConnectionConfig(
+            endpoint=endpoint,
+            transport_options=options,
+            auth_context=self.auth,
+        )
+
+    def prepare_connection(
+        self, capability: str, model: str | None = None
+    ) -> ProviderConnectionConfig:
+        config = super().prepare_connection(capability, model)
+        options = dict(config.transport_options)
+        options.update(self.auth.sdk_credentials())
+        return config.model_copy(update={"transport_options": options})

swarmauri_provider_amazon_nova/AmazonNovaToolSchemaConverter.py

@@ -0,0 +1,12 @@
+from typing import Any
+
+
+class AmazonNovaToolSchemaConverter:
+    def convert(self, tool: Any) -> dict[str, Any]:
+        return {
+            "toolSpec": {
+                "name": tool.name,
+                "description": getattr(tool, "description", ""),
+                "inputSchema": {"json": {"type": "object", "properties": {}}},
+            }
+        }

swarmauri_provider_amazon_nova/__init__.py

@@ -0,0 +1,15 @@
+from .AmazonNovaAuthConfig import AmazonNovaAuthConfig
+from .AmazonNovaEndpoints import AmazonNovaEndpoints
+from .AmazonNovaErrorMapper import AmazonNovaErrorMapper
+from .AmazonNovaModelCatalog import AmazonNovaModelCatalog
+from .AmazonNovaProvider import AmazonNovaProvider
+from .AmazonNovaToolSchemaConverter import AmazonNovaToolSchemaConverter
+
+__all__ = [
+    "AmazonNovaAuthConfig",
+    "AmazonNovaEndpoints",
+    "AmazonNovaErrorMapper",
+    "AmazonNovaModelCatalog",
+    "AmazonNovaProvider",
+    "AmazonNovaToolSchemaConverter",
+]

swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/METADATA

@@ -0,0 +1,50 @@
+Metadata-Version: 2.4
+Name: swarmauri_provider_amazon_nova
+Version: 0.1.0.dev3
+Summary: Aggregated Amazon Nova provider resources for Swarmauri models.
+License: Copyright 2024 Jacob Stewart
+         
+         Licensed under the Apache License, Version 2.0 (the "License");
+         you may not use this file except in compliance with the License.
+         You may obtain a copy of the License at
+         
+             http://www.apache.org/licenses/LICENSE-2.0
+         
+         Unless required by applicable law or agreed to in writing, software
+         distributed under the License is distributed on an "AS IS" BASIS,
+         WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+         See the License for the specific language governing permissions and
+         limitations under the License.
+License-File: LICENSE
+Keywords: swarmauri,provider,amazon nova,bedrock,nova sonic
+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)
+Description-Content-Type: text/markdown
+
+![Swarmauri](https://github.com/swarmauri/swarmauri-sdk/blob/3d4d1cfa949399d7019ae9d8f296afba773dfb7f/assets/swarmauri.brand.theme.svg)
+
+[![PyPI](https://img.shields.io/pypi/v/swarmauri_provider_amazon_nova.svg)](https://pypi.org/project/swarmauri_provider_amazon_nova/) [![Python](https://img.shields.io/pypi/pyversions/swarmauri_provider_amazon_nova.svg)](https://pypi.org/project/swarmauri_provider_amazon_nova/) [![License](https://img.shields.io/badge/license-Apache--2.0-blue)](LICENSE) [![Downloads](https://static.pepy.tech/badge/swarmauri_provider_amazon_nova)](https://pepy.tech/project/swarmauri_provider_amazon_nova) [![Hits](https://hits.sh/github.com/swarmauri/swarmauri-sdk.svg)](https://hits.sh/github.com/swarmauri/swarmauri-sdk/)
+
+# swarmauri_provider_amazon_nova
+
+## Features
+
+- AWS identity, region, Bedrock endpoint, catalog, error, and schema resources.
+- Shared provider resource for Amazon Nova model families.
+
+## Installation
+
+`uv add swarmauri_provider_amazon_nova` or `pip install swarmauri_provider_amazon_nova`.
+
+## Usage
+
+Inject `AmazonNovaProvider(region=...)` into a Nova model adapter.
+

swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/RECORD

@@ -0,0 +1,12 @@
+swarmauri_provider_amazon_nova/AmazonNovaAuthConfig.py,sha256=izjUURVbmptfojjMIGrp5zyG3GMIBsxtQZZZtmuO1uc,1587
+swarmauri_provider_amazon_nova/AmazonNovaEndpoints.py,sha256=QP5OeGioNfjzwZn33YVyD6VCH4Ir3MBQ_5Vrigyp5-s,868
+swarmauri_provider_amazon_nova/AmazonNovaErrorMapper.py,sha256=CKhm47u54p-CNjjIIAIG6Y3cyEaIz8x9g-P_bYsQKIU,531
+swarmauri_provider_amazon_nova/AmazonNovaModelCatalog.py,sha256=XuslBHjU6nLkENjgi_7PD4kQaVrEQslaO88BV5nLM84,538
+swarmauri_provider_amazon_nova/AmazonNovaProvider.py,sha256=TGaaE8MAfVXe2bdjVO828_KKXPez45gShTFS4lM5xsQ,3640
+swarmauri_provider_amazon_nova/AmazonNovaToolSchemaConverter.py,sha256=GK7RAksCSTkqsKI8BCMvFgp39eQxyvcjMMnWtg55dHg,360
+swarmauri_provider_amazon_nova/__init__.py,sha256=XpEBQalcEuIMjkHDRs9zjUtXbk4n-O2sqb18g4bhi4g,540
+swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/METADATA,sha256=2YkWjklQ4Y8a9mnviEQTPOgvHUEmv--xOxEpI4yuKvU,2508
+swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/WHEEL,sha256=L-WvLdSBvJWHXv5zrQxDvNRnIgoBzLUpcbuiw19KCuQ,88
+swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/entry_points.txt,sha256=fyOIebmTqT1zo5ONjrPJkayFvOyYZYhnyoDtm1q7WBE,98
+swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/licenses/LICENSE,sha256=kkwAgrDtfQ7c-Ozql2xlztDDjHt0YwGmRG7b_oUU9Jg,554
+swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/RECORD,,

swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: poetry-core 2.5.0
+Root-Is-Purelib: true
+Tag: py3-none-any

swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/entry_points.txt

@@ -0,0 +1,3 @@
+[swarmauri.model_providers]
+AmazonNovaProvider=swarmauri_provider_amazon_nova:AmazonNovaProvider
+

swarmauri_provider_amazon_nova-0.1.0.dev3.dist-info/licenses/LICENSE

@@ -0,0 +1,13 @@
+Copyright 2024 Jacob Stewart
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.