livekit-plugins-aws 0.1.1__tar.gz → 0.1.2__tar.gz

{livekit_plugins_aws-0.1.1 → livekit_plugins_aws-0.1.2}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: livekit-plugins-aws
-Version: 0.1.1
+Version: 0.1.2
 Summary: LiveKit Agents Plugin for services from AWS
 Home-page: https://github.com/livekit/agents
 License: Apache-2.0
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: >=3.9.0
 Description-Content-Type: text/markdown
-Requires-Dist: livekit-agents[codecs]<1.0.0,>=0.12.16
+Requires-Dist: livekit-agents[codecs]<1.0.0,>=0.12.20
 Requires-Dist: aiobotocore==2.19.0
 Requires-Dist: boto3==1.36.3
 Requires-Dist: amazon-transcribe>=0.6.2

{livekit_plugins_aws-0.1.1 → livekit_plugins_aws-0.1.2}/livekit/plugins/aws/llm.py

@@ -258,12 +258,14 @@ class LLMStream(llm.LLMStream):
             self._tool_call_id = tool_use["toolUseId"]
             self._fnc_name = tool_use["name"]
             self._fnc_raw_arguments = ""
+
         elif "contentBlockDelta" in chunk:
             delta = chunk["contentBlockDelta"]["delta"]
             if "toolUse" in delta:
                 self._fnc_raw_arguments += delta["toolUse"]["input"]
             elif "text" in delta:
                 self._text += delta["text"]
+
         elif "contentBlockStop" in chunk:
             if self._text:
                 chat_chunk = llm.ChatChunk(
@@ -280,6 +282,17 @@ class LLMStream(llm.LLMStream):
             elif self._tool_call_id:
                 return self._try_build_function(request_id, chunk)
 
+        elif "metadata" in chunk:
+            metadata = chunk["metadata"]
+            return llm.ChatChunk(
+                request_id=request_id,
+                usage=llm.CompletionUsage(
+                    completion_tokens=metadata["usage"]["outputTokens"],
+                    prompt_tokens=metadata["usage"]["inputTokens"],
+                    total_tokens=metadata["usage"]["totalTokens"],
+                ),
+            )
+
         return None
 
     def _try_build_function(self, request_id: str, chunk: dict) -> llm.ChatChunk | None:

{livekit_plugins_aws-0.1.1 → livekit_plugins_aws-0.1.2}/livekit/plugins/aws/version.py

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

{livekit_plugins_aws-0.1.1 → livekit_plugins_aws-0.1.2}/livekit_plugins_aws.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.2
+Metadata-Version: 2.4
 Name: livekit-plugins-aws
-Version: 0.1.1
+Version: 0.1.2
 Summary: LiveKit Agents Plugin for services from AWS
 Home-page: https://github.com/livekit/agents
 License: Apache-2.0
@@ -19,7 +19,7 @@ Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: >=3.9.0
 Description-Content-Type: text/markdown
-Requires-Dist: livekit-agents[codecs]<1.0.0,>=0.12.16
+Requires-Dist: livekit-agents[codecs]<1.0.0,>=0.12.20
 Requires-Dist: aiobotocore==2.19.0
 Requires-Dist: boto3==1.36.3
 Requires-Dist: amazon-transcribe>=0.6.2

{livekit_plugins_aws-0.1.1 → livekit_plugins_aws-0.1.2}/livekit_plugins_aws.egg-info/requires.txt

@@ -1,4 +1,4 @@
-livekit-agents[codecs]<1.0.0,>=0.12.16
+livekit-agents[codecs]<1.0.0,>=0.12.20
 aiobotocore==2.19.0
 boto3==1.36.3
 amazon-transcribe>=0.6.2

{livekit_plugins_aws-0.1.1 → livekit_plugins_aws-0.1.2}/setup.py

@@ -48,7 +48,7 @@ setuptools.setup(
     packages=setuptools.find_namespace_packages(include=["livekit.*"]),
     python_requires=">=3.9.0",
     install_requires=[
-        "livekit-agents[codecs]>=0.12.16,<1.0.0",
+        "livekit-agents[codecs]>=0.12.20,<1.0.0",
         "aiobotocore==2.19.0",
         "boto3==1.36.3",
         "amazon-transcribe>=0.6.2",