Update agent-voice example

Author: leawn

agent_voice/livekit/agent/pyproject.toml

@@ -10,8 +10,8 @@ dependencies = [
     "watchfiles>=1.0.4",
     "python-dotenv==1.0.1",
     "openai>=1.61.0",
-    "restack-ai>=0.0.77",
     "livekit-api>=0.8.2",
+    "restack-ai>=0.0.78",
 ]
 
 [project.scripts]

agent_voice/livekit/agent/src/agents/agent.py

@@ -1,7 +1,7 @@
 from datetime import timedelta
 
 from pydantic import BaseModel, Field
-from restack_ai.agent import agent, import_functions, log
+from restack_ai.agent import agent, import_functions, log, NonRetryableError
 
 with import_functions():
     from src.functions.livekit_dispatch import LivekitDispatchInput, livekit_dispatch
@@ -29,14 +29,18 @@ def __init__(self) -> None:
     async def messages(self, messages_event: MessagesEvent) -> list[Message]:
         log.info(f"Received message: {messages_event.messages}")
         self.messages.extend(messages_event.messages)
-
-        assistant_message = await agent.step(
-            function=llm_chat,
-            function_input=LlmChatInput(messages=self.messages),
-            start_to_close_timeout=timedelta(minutes=2),
-        )
-        self.messages.append(Message(role="assistant", content=str(assistant_message)))
-        return self.messages
+        try:
+            assistant_message = await agent.step(
+                function=llm_chat,
+                function_input=LlmChatInput(messages=self.messages),
+                start_to_close_timeout=timedelta(minutes=2),
+            )
+        except Exception as e:
+            error_message = f"Error during llm_chat: {e}"
+            raise NonRetryableError(error_message) from e
+        else:
+            self.messages.append(Message(role="assistant", content=str(assistant_message)))
+            return self.messages
 
     @agent.event
     async def end(self, end: EndEvent) -> EndEvent:
@@ -48,9 +52,13 @@ async def end(self, end: EndEvent) -> EndEvent:
     async def run(self, agent_input: AgentVoiceInput) -> None:
         log.info("Run", agent_input=agent_input)
         room_id = agent_input.room_id
-
-        await agent.step(
-            function=livekit_dispatch,
-            function_input=LivekitDispatchInput(room_id=room_id),
-        )
-        await agent.condition(lambda: self.end)
+        try:
+            await agent.step(
+                function=livekit_dispatch,
+                function_input=LivekitDispatchInput(room_id=room_id),
+            )
+        except Exception as e:
+            error_message = f"Error during livekit_dispatch: {e}"
+            raise NonRetryableError(error_message) from e
+        else:
+            await agent.condition(lambda: self.end)

agent_voice/livekit/agent/src/functions/livekit_dispatch.py

@@ -3,7 +3,7 @@
 
 from livekit import api
 from livekit.protocol.agent_dispatch import AgentDispatch
-from restack_ai.function import function, function_info, log
+from restack_ai.function import function, function_info, log, NonRetryableError
 
 
 @dataclass
@@ -38,7 +38,7 @@ async def livekit_dispatch(function_input: LivekitDispatchInput) -> AgentDispatc
 
     except Exception as e:
         log.error("livekit_dispatch function failed", error=str(e))
-        raise
+        raise NonRetryableError(f"Livekit dispatch failed: {e}") from e
 
     else:
         return dispatch

agent_voice/livekit/agent/src/functions/llm_chat.py

@@ -3,7 +3,7 @@
 
 from openai import OpenAI
 from pydantic import BaseModel, Field
-from restack_ai.function import function, log, stream_to_websocket
+from restack_ai.function import function, log, stream_to_websocket, NonRetryableError
 
 from src.client import api_address
 
@@ -49,4 +49,4 @@ async def llm_chat(function_input: LlmChatInput) -> str:
 
     except Exception as e:
         log.error("llm_chat function failed", error=str(e))
-        raise
+        raise NonRetryableError(f"LLM chat failed: {e}") from e