examples: updated llamaindex-rag examples to use VoicePipelineAgent (#…

…913)
livekit · Oct 14, 2024 · cce8e08 · cce8e08
1 parent 6b2aa77
commit cce8e08
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 7 deletions.
diff --git a/...e-assistant/llamaindex-rag/chat_engine.py → ...eline-agent/llamaindex-rag/chat_engine.py b/...e-assistant/llamaindex-rag/chat_engine.py → ...eline-agent/llamaindex-rag/chat_engine.py
@@ -2,7 +2,7 @@
 
 from dotenv import load_dotenv
 from livekit.agents import AutoSubscribe, JobContext, WorkerOptions, cli, llm
-from livekit.agents.voice_assistant import VoiceAssistant
+from livekit.agents.pipeline import VoicePipelineAgent
 from livekit.plugins import deepgram, llama_index, openai, silero
 from llama_index.core import (
     SimpleDirectoryReader,
@@ -41,7 +41,7 @@ async def entrypoint(ctx: JobContext):
 
     await ctx.connect(auto_subscribe=AutoSubscribe.AUDIO_ONLY)
 
-    assistant = VoiceAssistant(
+    assistant = VoicePipelineAgent(
         vad=silero.VAD.load(),
         stt=deepgram.STT(),
         llm=llama_index.LLM(chat_engine=chat_engine),

diff --git a/...ssistant/llamaindex-rag/data/raw_data.txt → ...ne-agent/llamaindex-rag/data/raw_data.txt b/...ssistant/llamaindex-rag/data/raw_data.txt → ...ne-agent/llamaindex-rag/data/raw_data.txt
diff --git a/...-assistant/llamaindex-rag/query_engine.py → ...line-agent/llamaindex-rag/query_engine.py b/...-assistant/llamaindex-rag/query_engine.py → ...line-agent/llamaindex-rag/query_engine.py
@@ -2,7 +2,7 @@
 
 from dotenv import load_dotenv
 from livekit.agents import AutoSubscribe, JobContext, WorkerOptions, cli, llm
-from livekit.agents.voice_assistant import VoiceAssistant
+from livekit.agents.pipeline import VoicePipelineAgent
 from livekit.plugins import deepgram, openai, silero
 from llama_index.core import (
     SimpleDirectoryReader,
@@ -47,7 +47,7 @@ async def query_info(query: str) -> str:
         print("Query result:", res)
         return str(res)
 
-    assistant = VoiceAssistant(
+    assistant = VoicePipelineAgent(
         vad=silero.VAD.load(),
         stt=deepgram.STT(),
         llm=openai.LLM(),

diff --git a/...ice-assistant/llamaindex-rag/retrieval.py → ...ipeline-agent/llamaindex-rag/retrieval.py b/...ice-assistant/llamaindex-rag/retrieval.py → ...ipeline-agent/llamaindex-rag/retrieval.py
@@ -2,7 +2,7 @@
 
 from dotenv import load_dotenv
 from livekit.agents import AutoSubscribe, JobContext, WorkerOptions, cli, llm
-from livekit.agents.voice_assistant import VoiceAssistant
+from livekit.agents.pipeline import VoicePipelineAgent
 from livekit.plugins import deepgram, openai, silero
 from llama_index.core import (
     SimpleDirectoryReader,
@@ -40,7 +40,7 @@ async def entrypoint(ctx: JobContext):
     initial_ctx.messages.append(system_msg)
 
     async def _will_synthesize_assistant_reply(
-        assistant: VoiceAssistant, chat_ctx: llm.ChatContext
+        assistant: VoicePipelineAgent, chat_ctx: llm.ChatContext
     ):
         ctx_msg = system_msg.copy()
         user_msg = chat_ctx.messages[-1]
@@ -57,7 +57,7 @@ async def _will_synthesize_assistant_reply(
 
     await ctx.connect(auto_subscribe=AutoSubscribe.AUDIO_ONLY)
 
-    assistant = VoiceAssistant(
+    assistant = VoicePipelineAgent(
         vad=silero.VAD.load(),
         stt=deepgram.STT(),
         llm=openai.LLM(),