livekit-plugins-aws 1.1.6__py3-none-any.whl → 1.1.7__py3-none-any.whl

livekit/plugins/aws/experimental/realtime/realtime_model.py

@@ -355,6 +355,7 @@ class RealtimeSession(  # noqa: F811
             aws_credentials_identity_resolver=Boto3CredentialsResolver(),
             http_auth_scheme_resolver=HTTPAuthSchemeResolver(),
             http_auth_schemes={"aws.auth#sigv4": SigV4AuthScheme()},
+            user_agent_extra="x-client-framework:livekit-plugins-aws[realtime]",
         )
         self._bedrock_client = BedrockRuntimeClient(config=config)
 
@@ -516,7 +517,7 @@ class RealtimeSession(  # noqa: F811
                 self._chat_ctx.truncate(max_items=MAX_MESSAGES)
             init_events = self._event_builder.create_prompt_start_block(
                 voice_id=self._realtime_model._opts.voice,
-                sample_rate=DEFAULT_OUTPUT_SAMPLE_RATE,
+                sample_rate=DEFAULT_OUTPUT_SAMPLE_RATE,  # type: ignore
                 system_content=self._instructions,
                 chat_ctx=self.chat_ctx,
                 tool_configuration=self._serialize_tool_config(),
@@ -823,12 +824,14 @@ class RealtimeSession(  # noqa: F811
                     curr_gen.audio_ch.close()
                 if not curr_gen.text_ch.closed:
                     curr_gen.text_ch.close()
-            if self._current_generation.response_id in self._current_generation.tool_messages:
-                curr_gen = self._current_generation.tool_messages[
-                    self._current_generation.response_id
-                ]
-                if not curr_gen.function_ch.closed:
-                    curr_gen.function_ch.close()
+
+            # TODO: seems not needed, tool_messages[id] is a str, function_ch is closed below?
+            # if self._current_generation.response_id in self._current_generation.tool_messages:
+            #     curr_gen = self._current_generation.tool_messages[
+            #         self._current_generation.response_id
+            #     ]
+            #     if not curr_gen.function_ch.closed:
+            #         curr_gen.function_ch.close()
 
             if not self._current_generation.message_ch.closed:
                 self._current_generation.message_ch.close()
@@ -1055,7 +1058,7 @@ class RealtimeSession(  # noqa: F811
                     "tool_use_id": item.call_id,
                     "tool_result": item.output
                     if not item.is_error
-                    else f"{{'error': '{item.error}'}}",
+                    else f"{{'error': '{item.output}'}}",
                 }
             )
 

livekit/plugins/aws/llm.py

@@ -19,6 +19,7 @@ from dataclasses import dataclass
 from typing import Any, cast
 
 import aioboto3  # type: ignore
+from botocore.config import Config
 
 from livekit.agents import APIConnectionError, APIStatusError, llm
 from livekit.agents.llm import (
@@ -205,7 +206,8 @@ class LLMStream(llm.LLMStream):
     async def _run(self) -> None:
         retryable = True
         try:
-            async with self._session.client("bedrock-runtime") as client:
+            config = Config(user_agent_extra="x-client-framework:livekit-plugins-aws")
+            async with self._session.client("bedrock-runtime", config=config) as client:
                 response = await client.converse_stream(**self._opts)
                 request_id = response["ResponseMetadata"]["RequestId"]
                 if response["ResponseMetadata"]["HTTPStatusCode"] != 200:

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.1.6"
+__version__ = "1.1.7"

{livekit_plugins_aws-1.1.6.dist-info → livekit_plugins_aws-1.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: livekit-plugins-aws
-Version: 1.1.6
+Version: 1.1.7
 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.2
-Requires-Dist: livekit-agents>=1.1.6
+Requires-Dist: livekit-agents>=1.1.7
 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.1.6.dist-info → livekit_plugins_aws-1.1.7.dist-info}/RECORD

@@ -1,17 +1,17 @@
 livekit/plugins/aws/__init__.py,sha256=dCZISj1yZG0WZTojk3sU-Ub4PK1ThCVhamrl9k_NbBw,2047
-livekit/plugins/aws/llm.py,sha256=pSbO7SaqYZYJ-3JGOmyMTmTLLcjoyrZy-j5BSkFOHU8,11922
+livekit/plugins/aws/llm.py,sha256=CVbOASZhd0QiUcZ_p6cL97fx4Dr1VSj25GwJPnfX2-s,12059
 livekit/plugins/aws/log.py,sha256=S5ICcsnwshZhMG0HPmc_lI3mtHmcY4oQMJBsnnho-bM,289
 livekit/plugins/aws/models.py,sha256=ztI9n3r8OHGlQewWXiyQP3SaavFKHkw-VVuK9Xz4K1I,696
 livekit/plugins/aws/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 livekit/plugins/aws/stt.py,sha256=PSR89aN28wm4i83yEdhkDJ9xzM0CsNIKrc3v3EbPndQ,9018
 livekit/plugins/aws/tts.py,sha256=T5dVpTuIuzQimYNnkfXi5dRLmRldWySL4IcbkXjmJLM,6083
 livekit/plugins/aws/utils.py,sha256=nA5Ua1f4T-25Loar6EvlrKTXI9N-zpTIH7cdQkwGyGI,1518
-livekit/plugins/aws/version.py,sha256=-bNd31cMcYCdhZCIKJ1-jtY4NgZvppVgKyzXAIzQtqM,600
+livekit/plugins/aws/version.py,sha256=EcBB23XE8aEiF7xHMivcb9wptFeYkGB1WNGSn1bIV3A,600
 livekit/plugins/aws/experimental/realtime/__init__.py,sha256=mm_TGZc9QAWSO-VOO3PdE8Y5R6xlWckXRZuiFUIHa-Q,287
 livekit/plugins/aws/experimental/realtime/events.py,sha256=-pJrwVrH5AZFxa1eDbX5nDdnJMz4BNucNZlYUYLsP-Y,15853
 livekit/plugins/aws/experimental/realtime/pretty_printer.py,sha256=KN7KPrfQu8cU7ff34vFAtfrd1umUSTVNKXQU7D8AMiM,1442
-livekit/plugins/aws/experimental/realtime/realtime_model.py,sha256=em_3Fbp1qefF7cIIHc6ib1FLdD1MOGes2Lwq61o2wlk,59464
+livekit/plugins/aws/experimental/realtime/realtime_model.py,sha256=qLrZdCxx4_JEk4-Z39nSfbqMcETMCmXm12Xhg9Sg1aQ,59670
 livekit/plugins/aws/experimental/realtime/turn_tracker.py,sha256=bcufaap-coeIYuK3ct1Is9W_UoefGYRmnJu7Mn5DCYU,6002
-livekit_plugins_aws-1.1.6.dist-info/METADATA,sha256=ST8uYsoqQgHUVRCLC3BdkYdwALh3joYGRjblVKQgDrE,1989
-livekit_plugins_aws-1.1.6.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-livekit_plugins_aws-1.1.6.dist-info/RECORD,,
+livekit_plugins_aws-1.1.7.dist-info/METADATA,sha256=Ct7BINBZCzj9FfLmvJA7IIrrZIm1LZ9Zd0boYSu-atQ,1989
+livekit_plugins_aws-1.1.7.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+livekit_plugins_aws-1.1.7.dist-info/RECORD,,