Update kitt.py

kitt.py

@@ -12,7 +12,7 @@ from livekit.agents.llm import (
     ChatMessage,
     ChatRole,
 )
-from livekit.agents.voice_assistant import AssistantContext, VoiceAssistant
+from livekit.agents.voice_assistant import AssistantCallContext, VoiceAssistant
 from livekit.plugins import deepgram, openai, silero
 
 dotenv.load_dotenv()
@@ -35,7 +35,7 @@ class AssistantFnc(agents.llm.FunctionContext):
                 agents.llm.TypeInfo(desc="The user message that triggered this function"),
             ],
     ):
-        ctx = AssistantContext.get_current()
+        ctx = AssistantCallContext.get_current()
         ctx.store_metadata("user_msg", user_msg)
 
 
@@ -133,7 +133,7 @@ async def entrypoint(ctx: JobContext):
         await assistant.say(stream, allow_interruptions=True)
 
     @assistant.on("function_calls_finished")
-    def _function_calls_done(ctx: AssistantContext):
+    def _function_calls_done(ctx: AssistantCallContext):
         user_msg = ctx.get_metadata("user_msg")
         if not user_msg:
             return