feat(aws-llm): replace raise with logging

Author: itskyf

livekit-agents/livekit/agents/llm/_provider_format/aws.py

@@ -5,6 +5,7 @@
 from dataclasses import dataclass
 
 from livekit.agents import llm
+from livekit.agents.log import logger
 
 from .utils import group_tool_calls
 
@@ -34,7 +35,8 @@ def to_chat_ctx(
         elif msg.type == "function_call_output":
             role = "user"
         else:
-            raise ValueError(f"Unknown message type: {msg.type!r}")
+            logger.warning("Skipping unknown message type %r", msg.type)
+            continue
 
         # if the effective role changed, finalize the previous turn.
         if role != current_role:

livekit-plugins/livekit-plugins-aws/livekit/plugins/aws/llm.py

@@ -321,28 +321,26 @@ def _parse_chunk(
                 ),
             )
         elif "contentBlockStop" in chunk:
-            if self._tool_call_id is None:
-                logger.warning("aws bedrock llm: no tool call id in the response")
-                return None
-            if self._fnc_name is None:
-                logger.warning("aws bedrock llm: no function name in the response")
-                return None
-            if self._fnc_arg_parts is None:
-                logger.warning("aws bedrock llm: no function arguments in the response")
-                return None
-            chat_chunk = llm.ChatChunk(
-                id=request_id,
-                delta=llm.ChoiceDelta(
-                    role="assistant",
-                    tool_calls=[
-                        FunctionToolCall(
-                            arguments="".join(self._fnc_arg_parts),
-                            name=self._fnc_name,
-                            call_id=self._tool_call_id,
-                        ),
-                    ],
-                ),
-            )
-            self._tool_call_id = self._fnc_name = self._fnc_arg_parts = None
-            return chat_chunk
+            if self._tool_call_id:
+                if self._fnc_name is None:
+                    logger.warning("aws bedrock llm: no function name in the response")
+                    return None
+                if self._fnc_arg_parts is None:
+                    logger.warning("aws bedrock llm: no function arguments in the response")
+                    return None
+                chat_chunk = llm.ChatChunk(
+                    id=request_id,
+                    delta=llm.ChoiceDelta(
+                        role="assistant",
+                        tool_calls=[
+                            FunctionToolCall(
+                                arguments="".join(self._fnc_arg_parts),
+                                name=self._fnc_name,
+                                call_id=self._tool_call_id,
+                            ),
+                        ],
+                    ),
+                )
+                self._tool_call_id = self._fnc_name = self._fnc_arg_parts = None
+                return chat_chunk
         return None