swarmauri_live_model_amazon_nova 0.1.0.dev3__tar.gz

swarmauri_live_model_amazon_nova-0.1.0.dev3/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.

swarmauri_live_model_amazon_nova-0.1.0.dev3/PKG-INFO

@@ -0,0 +1,52 @@
+Metadata-Version: 2.4
+Name: swarmauri_live_model_amazon_nova
+Version: 0.1.0.dev3
+Summary: Amazon Nova Sonic bidirectional HTTP/2 model adapter for Swarmauri.
+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,amazon nova,nova sonic,audio,http2
+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_provider_amazon_nova (>=0.1.0.dev3)
+Requires-Dist: swarmauri_transport_aws_eventstream (>=0.1.0.dev1)
+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_live_model_amazon_nova.svg)](https://pypi.org/project/swarmauri_live_model_amazon_nova/) [![Python](https://img.shields.io/pypi/pyversions/swarmauri_live_model_amazon_nova.svg)](https://pypi.org/project/swarmauri_live_model_amazon_nova/) [![License](https://img.shields.io/badge/license-Apache--2.0-blue)](LICENSE) [![Downloads](https://static.pepy.tech/badge/swarmauri_live_model_amazon_nova)](https://pepy.tech/project/swarmauri_live_model_amazon_nova) [![Hits](https://hits.sh/github.com/swarmauri/swarmauri-sdk.svg)](https://hits.sh/github.com/swarmauri/swarmauri-sdk/)
+
+# swarmauri_live_model_amazon_nova
+
+## Features
+
+- Nova Sonic text/audio sessions over an injected bidirectional HTTP/2 transport.
+- Typed audio, text, tool, usage, completion, and error events.
+
+## Installation
+
+`uv add swarmauri_live_model_amazon_nova` or `pip install swarmauri_live_model_amazon_nova`.
+
+## Usage
+
+Inject `AmazonNovaProvider` and an `IBidirectionalStreamClient` into `NovaSonicLiveModel`.
+

swarmauri_live_model_amazon_nova-0.1.0.dev3/README.md

@@ -0,0 +1,18 @@
+![Swarmauri](https://github.com/swarmauri/swarmauri-sdk/blob/3d4d1cfa949399d7019ae9d8f296afba773dfb7f/assets/swarmauri.brand.theme.svg)
+
+[![PyPI](https://img.shields.io/pypi/v/swarmauri_live_model_amazon_nova.svg)](https://pypi.org/project/swarmauri_live_model_amazon_nova/) [![Python](https://img.shields.io/pypi/pyversions/swarmauri_live_model_amazon_nova.svg)](https://pypi.org/project/swarmauri_live_model_amazon_nova/) [![License](https://img.shields.io/badge/license-Apache--2.0-blue)](LICENSE) [![Downloads](https://static.pepy.tech/badge/swarmauri_live_model_amazon_nova)](https://pepy.tech/project/swarmauri_live_model_amazon_nova) [![Hits](https://hits.sh/github.com/swarmauri/swarmauri-sdk.svg)](https://hits.sh/github.com/swarmauri/swarmauri-sdk/)
+
+# swarmauri_live_model_amazon_nova
+
+## Features
+
+- Nova Sonic text/audio sessions over an injected bidirectional HTTP/2 transport.
+- Typed audio, text, tool, usage, completion, and error events.
+
+## Installation
+
+`uv add swarmauri_live_model_amazon_nova` or `pip install swarmauri_live_model_amazon_nova`.
+
+## Usage
+
+Inject `AmazonNovaProvider` and an `IBidirectionalStreamClient` into `NovaSonicLiveModel`.

swarmauri_live_model_amazon_nova-0.1.0.dev3/pyproject.toml

@@ -0,0 +1,29 @@
+[project]
+name = "swarmauri_live_model_amazon_nova"
+version = "0.1.0.dev3"
+description = "Amazon Nova Sonic bidirectional HTTP/2 model adapter for Swarmauri."
+readme = "README.md"
+requires-python = ">=3.10,<3.13"
+authors = [{ name = "Jacob Stewart", email = "jacob@swarmauri.com" }]
+license = { file = "LICENSE" }
+dependencies = ["swarmauri_core>=0.11.0.dev5", "swarmauri_base>=0.11.0.dev5", "swarmauri_provider_amazon_nova>=0.1.0.dev3", "swarmauri_transport_aws_eventstream>=0.1.0.dev1"]
+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"]
+keywords = ["swarmauri", "amazon nova", "nova sonic", "audio", "http2"]
+[tool.uv.sources]
+swarmauri_core = { workspace = true }
+swarmauri_base = { workspace = true }
+swarmauri_provider_amazon_nova = { workspace = true }
+swarmauri_transport_aws_eventstream = { workspace = true }
+[project.entry-points."swarmauri.live_models"]
+NovaSonicLiveModel = "swarmauri_live_model_amazon_nova:NovaSonicLiveModel"
+[build-system]
+requires = ["poetry-core>=1.8.0"]
+build-backend = "poetry.core.masonry.api"
+[dependency-groups]
+dev = ["pytest>=8.0", "pytest-asyncio>=0.24", "ruff>=0.9.9", "swarmauri_tests_component"]
+
+[tool.ruff]
+extend = "../../pyproject.toml"
+
+[tool.ruff.lint]
+ignore = ["BLE001", "N999"]

swarmauri_live_model_amazon_nova-0.1.0.dev3/swarmauri_live_model_amazon_nova/NovaSonicCodec.py

@@ -0,0 +1,84 @@
+import base64
+
+from swarmauri_base.live_models.events.AudioDeltaEvent import AudioDeltaEvent
+from swarmauri_base.live_models.events.SessionErrorEvent import (
+    SessionErrorEvent,
+)
+from swarmauri_base.live_models.events.TextDeltaEvent import TextDeltaEvent
+from swarmauri_base.live_models.events.ToolCallEvent import ToolCallEvent
+from swarmauri_base.live_models.events.TurnCompleteEvent import (
+    TurnCompleteEvent,
+)
+from swarmauri_base.live_models.events.UsageEvent import UsageEvent
+
+
+class NovaSonicCodec:
+    def encode(self, input_data):
+        if input_data.kind == "audio":
+            return {
+                "event": {
+                    "audioInput": {
+                        "content": base64.b64encode(
+                            input_data.payload
+                        ).decode()
+                    }
+                }
+            }
+        if input_data.kind == "text":
+            return {"event": {"textInput": {"content": input_data.payload}}}
+        if input_data.kind == "tool_result":
+            return {
+                "event": {
+                    "toolResult": {
+                        "toolUseId": input_data.call_id,
+                        "content": [{"json": input_data.payload}],
+                        "status": "error"
+                        if input_data.is_error
+                        else "success",
+                    }
+                }
+            }
+        if input_data.kind == "commit_turn":
+            return {"event": {"contentEnd": {}}}
+        if input_data.kind == "interrupt":
+            return {"event": {"playbackCompletion": {}}}
+        raise ValueError(f"Unsupported Nova Sonic input: {input_data.kind}")
+
+    def decode(self, data):
+        event = data.get("event", data)
+        if "audioOutput" in event:
+            return [
+                AudioDeltaEvent(
+                    payload=base64.b64decode(
+                        event["audioOutput"].get("content", "")
+                    )
+                )
+            ]
+        if "textOutput" in event:
+            return [
+                TextDeltaEvent(payload=event["textOutput"].get("content", ""))
+            ]
+        if "toolUse" in event:
+            use = event["toolUse"]
+            return [
+                ToolCallEvent(
+                    call_id=use.get("toolUseId", ""),
+                    tool_name=use.get("toolName", ""),
+                    arguments=use.get("content", {}),
+                )
+            ]
+        if "completionEnd" in event or "contentEnd" in event:
+            return [TurnCompleteEvent()]
+        if "usageEvent" in event:
+            return [
+                UsageEvent(
+                    payload={
+                        key: int(value)
+                        for key, value in event["usageEvent"].items()
+                        if isinstance(value, int)
+                    }
+                )
+            ]
+        if "error" in event:
+            return [SessionErrorEvent(message=str(event["error"]))]
+        return []

swarmauri_live_model_amazon_nova-0.1.0.dev3/swarmauri_live_model_amazon_nova/NovaSonicConfigMapper.py

@@ -0,0 +1,38 @@
+from swarmauri_base.live_models.LiveConfigBase import LiveConfigBase
+
+
+class NovaSonicConfigMapper:
+    def map(self, model: str, config: LiveConfigBase) -> list[dict]:
+        return [
+            {
+                "event": {
+                    "sessionStart": {
+                        "inferenceConfiguration": config.provider_options.get(
+                            "inference", {}
+                        )
+                    }
+                }
+            },
+            {
+                "event": {
+                    "promptStart": {
+                        "promptName": config.provider_options.get(
+                            "prompt_name", "swarmauri"
+                        ),
+                        "textOutputConfiguration": {"mediaType": "text/plain"},
+                        "audioOutputConfiguration": config.provider_options.get(  # noqa: E501
+                            "audio_output",
+                            {
+                                "mediaType": "audio/lpcm",
+                                "sampleRateHertz": 24000,
+                                "sampleSizeBits": 16,
+                                "channelCount": 1,
+                                "voiceId": "matthew",
+                                "encoding": "base64",
+                                "audioType": "SPEECH",
+                            },
+                        ),
+                    }
+                }
+            },
+        ]

swarmauri_live_model_amazon_nova-0.1.0.dev3/swarmauri_live_model_amazon_nova/NovaSonicLiveModel.py

@@ -0,0 +1,49 @@
+from typing import Literal
+
+from pydantic import ConfigDict, Field
+from swarmauri_base.ComponentBase import ComponentBase
+from swarmauri_base.live_models.LiveCapabilitiesBase import (
+    LiveCapabilitiesBase,
+)
+from swarmauri_base.live_models.LiveConfigBase import LiveConfigBase
+from swarmauri_base.live_models.LiveModelBase import LiveModelBase
+from swarmauri_core.transports.i_bidirectional_stream import (
+    IBidirectionalStreamClient,
+)
+from swarmauri_provider_amazon_nova import AmazonNovaProvider
+from swarmauri_transport_aws_eventstream import AwsEventStreamClient
+
+from .NovaSonicConfigMapper import NovaSonicConfigMapper
+from .NovaSonicLiveSession import NovaSonicLiveSession
+
+
+@ComponentBase.register_type(LiveModelBase, "NovaSonicLiveModel")
+class NovaSonicLiveModel(LiveModelBase):
+    name: str = "amazon.nova-sonic-v1:0"
+    provider: AmazonNovaProvider = Field(default_factory=AmazonNovaProvider)
+    transport: IBidirectionalStreamClient = Field(
+        default_factory=AwsEventStreamClient, exclude=True
+    )
+    capabilities: LiveCapabilitiesBase = Field(
+        default_factory=lambda: LiveCapabilitiesBase(
+            input_modalities=frozenset({"text", "audio"}),
+            output_modalities=frozenset({"text", "audio"}),
+            features=frozenset({"tools"}),
+            session_features=frozenset({"realtime", "bidirectional_http2"}),
+        )
+    )
+    type: Literal["NovaSonicLiveModel"] = "NovaSonicLiveModel"
+    model_config = ConfigDict(arbitrary_types_allowed=True, extra="allow")
+
+    async def aconnect(self, config: LiveConfigBase):
+        connection_config = self.provider.prepare_connection(
+            "live.connect", model=self.name
+        )
+        connection = await self.transport.aconnect(
+            connection_config.resolved_url(),
+            headers=connection_config.headers,
+            **connection_config.transport_options,
+        )
+        return await NovaSonicLiveSession.create(
+            connection, NovaSonicConfigMapper().map(self.name, config)
+        )

swarmauri_live_model_amazon_nova-0.1.0.dev3/swarmauri_live_model_amazon_nova/NovaSonicLiveSession.py

@@ -0,0 +1,53 @@
+import asyncio
+from contextlib import suppress
+from typing import Literal
+
+from pydantic import PrivateAttr
+from swarmauri_base.live_models.events.SessionErrorEvent import (
+    SessionErrorEvent,
+)
+from swarmauri_base.live_models.LiveSessionBase import LiveSessionBase
+from swarmauri_base.ComponentBase import ComponentBase
+
+from .NovaSonicCodec import NovaSonicCodec
+
+
+@ComponentBase.register_type(LiveSessionBase, "NovaSonicLiveSession")
+class NovaSonicLiveSession(LiveSessionBase):
+    type: Literal["NovaSonicLiveSession"] = "NovaSonicLiveSession"
+    _connection = PrivateAttr()
+    _codec = PrivateAttr(default_factory=NovaSonicCodec)
+    _receiver = PrivateAttr(default=None)
+
+    @classmethod
+    async def create(cls, connection, setup_events):
+        session = cls()
+        session._connection = connection
+        for event in setup_events:
+            await connection.asend(event)
+        session.mark_open()
+        session._receiver = asyncio.create_task(session._receive_loop())
+        return session
+
+    async def _receive_loop(self) -> None:
+        try:
+            async for data in self._connection.aiter_events():
+                for event in self._codec.decode(data):
+                    await self.emit(event)
+        except asyncio.CancelledError:
+            raise
+        except Exception as exc:
+            await self.emit(SessionErrorEvent(message=str(exc)))
+            await self.fail(exc)
+        else:
+            await self.finish()
+
+    async def _asend_impl(self, input_data):
+        await self._connection.asend(self._codec.encode(input_data))
+
+    async def _aclose_impl(self):
+        if self._receiver and self._receiver is not asyncio.current_task():
+            self._receiver.cancel()
+            with suppress(asyncio.CancelledError):
+                await self._receiver
+        await self._connection.aclose()

swarmauri_live_model_amazon_nova-0.1.0.dev3/swarmauri_live_model_amazon_nova/NovaSonicToolAdapter.py

@@ -0,0 +1,8 @@
+from swarmauri_base.live_models.inputs.ToolResultInput import ToolResultInput
+
+
+class NovaSonicToolAdapter:
+    def result(self, call_id, value, *, is_error=False):
+        return ToolResultInput(
+            call_id=call_id, payload=value, is_error=is_error
+        )

swarmauri_live_model_amazon_nova-0.1.0.dev3/swarmauri_live_model_amazon_nova/__init__.py

@@ -0,0 +1,13 @@
+from .NovaSonicCodec import NovaSonicCodec
+from .NovaSonicConfigMapper import NovaSonicConfigMapper
+from .NovaSonicLiveModel import NovaSonicLiveModel
+from .NovaSonicLiveSession import NovaSonicLiveSession
+from .NovaSonicToolAdapter import NovaSonicToolAdapter
+
+__all__ = [
+    "NovaSonicCodec",
+    "NovaSonicConfigMapper",
+    "NovaSonicLiveModel",
+    "NovaSonicLiveSession",
+    "NovaSonicToolAdapter",
+]