wx44wx/pipecat
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update SmallWebRTCTransport text examples with new run_example

Browse Source
This commit is contained in:
vipyne
2025-05-28 10:16:27 -05:00
parent 575b97ba60
commit 821da723c0
2 changed files with 159 additions and 149 deletions
Download Patch File Download Diff File Expand all files Collapse all files

25
examples/foundational/41a-text-only-webrtc.py
View File

@@ -7,6 +7,7 @@
import argparse import argparse
import os import os
import aiohttp
from dotenv import load_dotenv from dotenv import load_dotenv
from loguru import logger from loguru import logger
@@ -28,9 +29,8 @@ from pipecat.processors.frameworks.rtvi import (
) )
from pipecat.services.openai import OpenAIContextAggregatorPair from pipecat.services.openai import OpenAIContextAggregatorPair
from pipecat.services.openai.llm import OpenAILLMService from pipecat.services.openai.llm import OpenAILLMService
from pipecat.transports.base_transport import TransportParams from pipecat.transports.base_transport import BaseTransport, TransportParams
from pipecat.transports.network.small_webrtc import SmallWebRTCTransport from pipecat.transports.services.daily import DailyParams
from pipecat.transports.network.webrtc_connection import SmallWebRTCConnection
load_dotenv(override=True) load_dotenv(override=True)
@@ -71,14 +71,19 @@ def create_action_llm_append_to_messages(context_aggregator: OpenAIContextAggreg
return action_llm_append_to_messages return action_llm_append_to_messages
async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespace): # We store functions so objects (e.g. SileroVADAnalyzer) don't get
# instantiated. The function will be called when the desired transport gets
# selected.
transport_params = {
"webrtc": lambda: TransportParams(),
}
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot") logger.info(f"Starting bot")
transport = SmallWebRTCTransport( # Create an HTTP session for API calls
webrtc_connection=webrtc_connection, async with aiohttp.ClientSession() as session:
params=TransportParams(),
)
llm = OpenAILLMService(api_key=os.getenv("OPENAI_API_KEY")) llm = OpenAILLMService(api_key=os.getenv("OPENAI_API_KEY"))
messages = [ messages = [
@@ -154,4 +159,4 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespac
if __name__ == "__main__": if __name__ == "__main__":
from run import main from run import main
main() main(run_example, transport_params=transport_params)

27
examples/foundational/41b-text-and-audio-webrtc.py
View File

@@ -7,6 +7,7 @@
import argparse import argparse
import os import os
import aiohttp
from dotenv import load_dotenv from dotenv import load_dotenv
from loguru import logger from loguru import logger
@@ -31,9 +32,8 @@ from pipecat.services.cartesia.tts import CartesiaTTSService
from pipecat.services.deepgram.stt import DeepgramSTTService from pipecat.services.deepgram.stt import DeepgramSTTService
from pipecat.services.openai import OpenAIContextAggregatorPair from pipecat.services.openai import OpenAIContextAggregatorPair
from pipecat.services.openai.llm import OpenAILLMService from pipecat.services.openai.llm import OpenAILLMService
from pipecat.transports.base_transport import TransportParams from pipecat.transports.base_transport import BaseTransport, TransportParams
from pipecat.transports.network.small_webrtc import SmallWebRTCTransport from pipecat.transports.services.daily import DailyParams
from pipecat.transports.network.webrtc_connection import SmallWebRTCConnection
load_dotenv(override=True) load_dotenv(override=True)
@@ -77,18 +77,23 @@ def create_action_llm_append_to_messages(context_aggregator: OpenAIContextAggreg
return action_llm_append_to_messages return action_llm_append_to_messages
async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespace): # We store functions so objects (e.g. SileroVADAnalyzer) don't get
logger.info(f"Starting bot") # instantiated. The function will be called when the desired transport gets
# selected.
transport = SmallWebRTCTransport( transport_params = {
webrtc_connection=webrtc_connection, "webrtc": lambda: TransportParams(
params=TransportParams(
audio_in_enabled=True, audio_in_enabled=True,
audio_out_enabled=True, audio_out_enabled=True,
vad_analyzer=SileroVADAnalyzer(), vad_analyzer=SileroVADAnalyzer(),
), ),
) }
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session for API calls
async with aiohttp.ClientSession() as session:
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY")) stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
llm = OpenAILLMService(api_key=os.getenv("OPENAI_API_KEY")) llm = OpenAILLMService(api_key=os.getenv("OPENAI_API_KEY"))
@@ -171,4 +176,4 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespac
if __name__ == "__main__": if __name__ == "__main__":
from run import main from run import main
main() main(run_example, transport_params=transport_params)