livekit-plugins-aws 1.2.11__tar.gz → 1.2.12__tar.gz

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-aws
-Version: 1.2.11
+Version: 1.2.12
 Summary: LiveKit Agents Plugin for services from AWS
 Project-URL: Documentation, https://docs.livekit.io
 Project-URL: Website, https://livekit.io/
@@ -20,7 +20,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Requires-Python: >=3.9.0
 Requires-Dist: aioboto3>=14.1.0
 Requires-Dist: amazon-transcribe>=0.6.4
-Requires-Dist: livekit-agents>=1.2.11
+Requires-Dist: livekit-agents>=1.2.12
 Provides-Extra: realtime
 Requires-Dist: aws-sdk-bedrock-runtime==0.0.2; (python_version >= '3.12') and extra == 'realtime'
 Requires-Dist: boto3>1.35.10; extra == 'realtime'

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/livekit/plugins/aws/experimental/realtime/realtime_model.py

@@ -40,6 +40,7 @@ from livekit.agents import (
     utils,
 )
 from livekit.agents.metrics import RealtimeModelMetrics
+from livekit.agents.metrics.base import Metadata
 from livekit.agents.types import NOT_GIVEN, NotGivenOr
 from livekit.agents.utils import is_given
 from livekit.plugins.aws.experimental.realtime.turn_tracker import _TurnTracker
@@ -259,6 +260,14 @@ class RealtimeModel(llm.RealtimeModel):
         )
         self._sessions = weakref.WeakSet[RealtimeSession]()
 
+    @property
+    def model(self) -> str:
+        return self.model_id
+
+    @property
+    def provider(self) -> str:
+        return "Amazon"
+
     def session(self) -> RealtimeSession:
         """Return a new RealtimeSession bound to this model instance."""
         sess = RealtimeSession(self)
@@ -273,10 +282,6 @@ class RealtimeModel(llm.RealtimeModel):
         async def aclose(self) -> None:
             pass
 
-    @property
-    def model(self) -> str:
-        return self.model_id
-
 
 class RealtimeSession(  # noqa: F811
     llm.RealtimeSession[Literal["bedrock_server_event_received", "bedrock_client_event_queued"]]
@@ -871,7 +876,6 @@ class RealtimeSession(  # noqa: F811
         # Q: should we be counting per turn or utterance?
         metrics = RealtimeModelMetrics(
             label=self._realtime_model.label,
-            model=self._realtime_model.model,
             # TODO: pass in the correct request_id
             request_id=event_data["event"]["usageEvent"]["completionId"],
             timestamp=time.monotonic(),
@@ -898,6 +902,9 @@ class RealtimeSession(  # noqa: F811
                 audio_tokens=output_tokens["speechTokens"],
                 image_tokens=0,
             ),
+            metadata=Metadata(
+                model_name=self._realtime_model.model, model_provider=self._realtime_model.provider
+            ),
         )
         self.emit("metrics_collected", metrics)
 

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/livekit/plugins/aws/llm.py

@@ -125,6 +125,10 @@ class LLM(llm.LLM):
     def model(self) -> str:
         return self._opts.model
 
+    @property
+    def provider(self) -> str:
+        return "AWS Bedrock"
+
     def chat(
         self,
         *,

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/livekit/plugins/aws/stt.py

@@ -94,6 +94,18 @@ class STT(stt.STT):
             region=region,
         )
 
+    @property
+    def model(self) -> str:
+        return (
+            self._config.language_model_name
+            if is_given(self._config.language_model_name)
+            else "unknown"
+        )
+
+    @property
+    def provider(self) -> str:
+        return "Amazon Transcribe"
+
     async def aclose(self) -> None:
         await super().aclose()
 

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/livekit/plugins/aws/tts.py

@@ -107,6 +107,14 @@ class TTS(tts.TTS):
             sample_rate=sample_rate,
         )
 
+    @property
+    def model(self) -> str:
+        return self._opts.speech_engine
+
+    @property
+    def provider(self) -> str:
+        return "Amazon Polly"
+
     def synthesize(
         self, text: str, *, conn_options: APIConnectOptions = DEFAULT_API_CONNECT_OPTIONS
     ) -> ChunkedStream:

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/livekit/plugins/aws/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-__version__ = "1.2.11"
+__version__ = "1.2.12"

{livekit_plugins_aws-1.2.11 → livekit_plugins_aws-1.2.12}/pyproject.toml

@@ -23,7 +23,7 @@ classifiers = [
     "Programming Language :: Python :: 3 :: Only",
 ]
 dependencies = [
-    "livekit-agents>=1.2.11",
+    "livekit-agents>=1.2.12",
     "aioboto3>=14.1.0",
     "amazon-transcribe>=0.6.4",
 ]