examples(foundational): update handle_signint depending on transport

This commit is contained in:
Aleix Conchillo Flaqué
2025-05-27 08:02:16 -07:00
parent 884268fce3
commit 6393e89022
123 changed files with 290 additions and 269 deletions

View File

@@ -33,7 +33,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -49,7 +49,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def on_client_connected(transport, client):
await task.queue_frames([TTSSpeakFrame(f"Hello there!"), EndFrame()])
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -32,7 +32,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -50,7 +50,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def on_client_connected(transport, client):
await task.queue_frames([TTSSpeakFrame(f"Hello there!"), EndFrame()])
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -32,7 +32,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tts = CartesiaTTSService(
@@ -47,7 +47,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def on_client_connected(transport, client):
await task.queue_frames([TTSSpeakFrame(f"Hello there!"), EndFrame()])
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -32,7 +32,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tts = FastPitchTTSService(api_key=os.getenv("NVIDIA_API_KEY"))
@@ -44,7 +44,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def on_client_connected(transport, client):
await task.queue_frames([TTSSpeakFrame(f"Hello there!"), EndFrame()])
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -33,7 +33,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tts = CartesiaTTSService(
@@ -57,7 +57,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def on_client_connected(transport, client):
await task.queue_frames([LLMMessagesFrame(messages), EndFrame()])
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -39,7 +39,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -67,7 +67,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -38,7 +38,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
imagegen = GoogleImageGenService(
@@ -67,7 +67,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -82,7 +82,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
"""Run the Calendar Month Narration bot using WebRTC transport.
Args:
@@ -174,7 +174,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
await task.cancel()
# Run the pipeline
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -75,7 +75,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -137,7 +137,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -90,7 +90,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -156,7 +156,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -107,7 +107,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -65,7 +65,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -140,7 +140,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -50,7 +50,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(
@@ -119,7 +119,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -105,7 +105,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -48,7 +48,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -112,7 +112,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -110,7 +110,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = AzureSTTService(
@@ -114,7 +114,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = OpenAISTTService(
@@ -109,7 +109,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -113,7 +113,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -111,7 +111,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -48,7 +48,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = GladiaSTTService(
@@ -116,7 +116,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -104,7 +104,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = GroqSTTService(api_key=os.getenv("GROQ_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -45,7 +45,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = AWSTranscribeSTTService()
@@ -111,7 +111,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = GoogleSTTService(
@@ -112,7 +112,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = AssemblyAISTTService(
@@ -110,7 +110,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -50,7 +50,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -48,7 +48,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -113,7 +113,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -107,7 +107,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = RivaSTTService(api_key=os.getenv("NVIDIA_API_KEY"))
@@ -104,7 +104,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -213,7 +213,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
llm = GoogleLLMService(api_key=os.getenv("GOOGLE_API_KEY"), model="gemini-2.0-flash-001")
@@ -281,7 +281,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -57,7 +57,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tts = CartesiaTTSService(
@@ -96,7 +96,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -108,7 +108,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -107,7 +107,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -47,7 +47,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = FalSTTService(
@@ -110,7 +110,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -49,7 +49,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -114,7 +114,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -49,7 +49,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create an HTTP session
@@ -113,7 +113,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -71,7 +71,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
pipeline = Pipeline([transport.input(), MirrorProcessor(), transport.output()])
@@ -95,7 +95,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -22,10 +22,9 @@ from pipecat.pipeline.pipeline import Pipeline
from pipecat.pipeline.runner import PipelineRunner
from pipecat.pipeline.task import PipelineParams, PipelineTask
from pipecat.processors.frame_processor import FrameDirection, FrameProcessor
from pipecat.transports.base_transport import TransportParams
from pipecat.transports.base_transport import BaseTransport, TransportParams
from pipecat.transports.local.tk import TkLocalTransport, TkTransportParams
from pipecat.transports.network.small_webrtc import SmallWebRTCTransport
from pipecat.transports.network.webrtc_connection import SmallWebRTCConnection
from pipecat.transports.services.daily import DailyParams
load_dotenv(override=True)
@@ -50,21 +49,33 @@ class MirrorProcessor(FrameProcessor):
await self.push_frame(frame, direction)
async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespace):
logger.info(f"Starting bot")
# 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 = {
"daily": lambda: DailyParams(
audio_in_enabled=True,
audio_out_enabled=True,
video_in_enabled=True,
video_out_enabled=True,
video_out_is_live=True,
video_out_width=1280,
video_out_height=720,
),
"webrtc": lambda: TransportParams(
audio_in_enabled=True,
audio_out_enabled=True,
video_in_enabled=True,
video_out_enabled=True,
video_out_is_live=True,
video_out_width=1280,
video_out_height=720,
),
}
p2p_transport = SmallWebRTCTransport(
webrtc_connection=webrtc_connection,
params=TransportParams(
audio_in_enabled=True,
audio_out_enabled=True,
video_in_enabled=True,
video_out_enabled=True,
video_out_is_live=True,
video_out_width=1280,
video_out_height=720,
),
)
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tk_root = tk.Tk()
tk_root.title("Local Mirror")
@@ -80,11 +91,7 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespac
),
)
@p2p_transport.event_handler("on_client_connected")
async def on_client_connected(transport, client):
logger.info(f"Client connected")
pipeline = Pipeline([p2p_transport.input(), MirrorProcessor(), tk_transport.output()])
pipeline = Pipeline([transport.input(), MirrorProcessor(), tk_transport.output()])
task = PipelineTask(
pipeline,
@@ -97,7 +104,21 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespac
tk_root.update_idletasks()
await asyncio.sleep(0.1)
runner = PipelineRunner(handle_sigint=False)
@transport.event_handler("on_client_connected")
async def on_client_connected(transport, client):
logger.info(f"Client connected")
@transport.event_handler("on_client_disconnected")
async def on_client_disconnected(transport, client):
logger.info(f"Client disconnected")
await task.cancel()
@transport.event_handler("on_client_closed")
async def on_client_closed(transport, client):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=handle_sigint)
await asyncio.gather(runner.run(task), run_tk())
@@ -105,4 +126,4 @@ async def run_bot(webrtc_connection: SmallWebRTCConnection, _: argparse.Namespac
if __name__ == "__main__":
from run import main
main()
main(run_example, transport_params=transport_params)

View File

@@ -48,7 +48,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -103,7 +103,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -100,7 +100,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -161,7 +161,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -66,7 +66,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
user_response = UserResponseAggregator()
@@ -124,7 +124,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -66,7 +66,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
user_response = UserResponseAggregator()
@@ -127,7 +127,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -66,7 +66,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
user_response = UserResponseAggregator()
@@ -127,7 +127,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -66,7 +66,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
user_response = UserResponseAggregator()
@@ -127,7 +127,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -50,7 +50,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = WhisperSTTService()
@@ -71,7 +71,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -41,7 +41,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(
@@ -65,7 +65,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -41,7 +41,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = GladiaSTTService(
@@ -65,7 +65,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -50,7 +50,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = GladiaSTTService(
@@ -86,7 +86,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -41,7 +41,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = AssemblyAISTTService(
@@ -64,7 +64,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -71,7 +71,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = WhisperSTTServiceMLX(model=MLXModel.LARGE_V3_TURBO)
@@ -98,7 +98,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -138,7 +138,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -55,7 +55,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -131,7 +131,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -78,7 +78,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -199,7 +199,7 @@ If you need to use a tool, simply use the tool. Do not tell the user the tool yo
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -131,7 +131,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -78,7 +78,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -182,7 +182,7 @@ indicate you should use the get_image tool are:
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -80,7 +80,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -192,7 +192,7 @@ indicate you should use the get_image tool are:
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -57,7 +57,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = GroqSTTService(api_key=os.getenv("GROQ_API_KEY"), model="distil-whisper-large-v3-en")
@@ -140,7 +140,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -54,7 +54,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -133,7 +133,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -139,7 +139,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -138,7 +138,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -136,7 +136,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -146,7 +146,7 @@ Start by asking me for my location. Then, use 'get_weather_current' to give me a
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -146,7 +146,7 @@ Start by asking me for my location. Then, use 'get_weather_current' to give me a
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -140,7 +140,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -53,7 +53,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -113,7 +113,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -135,7 +135,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -141,7 +141,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -56,7 +56,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -138,7 +138,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -54,7 +54,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = AWSTranscribeSTTService()
@@ -142,7 +142,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -76,7 +76,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -171,7 +171,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -71,7 +71,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(
@@ -159,7 +159,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -46,7 +46,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -142,7 +142,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -48,7 +48,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -140,7 +140,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -40,7 +40,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, args: argparse.Namespace):
async def run_example(transport: BaseTransport, args: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot with video input: {args.input}")
gst = GStreamerPipelineSource(
@@ -60,7 +60,7 @@ async def run_example(transport: BaseTransport, args: argparse.Namespace):
task = PipelineTask(pipeline)
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -37,7 +37,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot with video test source")
gst = GStreamerPipelineSource(
@@ -56,7 +56,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
task = PipelineTask(pipeline)
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -88,7 +88,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
session_properties = SessionProperties(
@@ -182,7 +182,7 @@ Remember, your responses should be short. Just one or two sentences, usually."""
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -87,7 +87,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
session_properties = SessionProperties(
@@ -181,7 +181,7 @@ Remember, your responses should be short. Just one or two sentences, usually."""
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -186,7 +186,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
global tts
@@ -246,7 +246,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -174,7 +174,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -256,7 +256,7 @@ Remember, your responses should be short. Just one or two sentences, usually."""
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -181,7 +181,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
global tts
@@ -245,7 +245,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -237,7 +237,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -304,7 +304,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -191,7 +191,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Specify initial system instruction.
@@ -269,7 +269,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -103,7 +103,7 @@ async def main():
logger.info(f"Client disconnected")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner()
await runner.run(task)

View File

@@ -50,7 +50,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
async with aiohttp.ClientSession() as session:
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -125,7 +125,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -54,7 +54,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -171,7 +171,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -224,7 +224,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -396,7 +396,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -428,7 +428,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -603,7 +603,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -649,7 +649,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tts = CartesiaTTSService(
@@ -782,7 +782,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -68,7 +68,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
tts = CartesiaTTSService(
@@ -137,7 +137,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner()
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -60,7 +60,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
stt = DeepgramSTTService(api_key=os.getenv("DEEPGRAM_API_KEY"))
@@ -145,7 +145,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -291,7 +291,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
tts = CartesiaTTSService(
@@ -379,7 +379,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -53,7 +53,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
# Create the Gemini Multimodal Live LLM service
@@ -119,7 +119,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
await task.cancel()
# Run the pipeline
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -60,7 +60,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
llm = GeminiMultimodalLiveLLMService(
@@ -138,7 +138,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
line = f"{timestamp}{msg.role}: {msg.content}"
logger.info(f"Transcript: {line}")
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -82,7 +82,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot")
weather_function = FunctionSchema(
@@ -154,7 +154,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner(handle_sigint=False)
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

View File

@@ -51,7 +51,7 @@ transport_params = {
}
async def run_example(transport: BaseTransport, _: argparse.Namespace):
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
llm = GeminiMultimodalLiveLLMService(
api_key=os.getenv("GOOGLE_API_KEY"),
voice_id="Aoede", # Puck, Charon, Kore, Fenrir, Aoede
@@ -110,7 +110,7 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace):
logger.info(f"Client closed connection")
await task.cancel()
runner = PipelineRunner()
runner = PipelineRunner(handle_sigint=handle_sigint)
await runner.run(task)

Some files were not shown because too many files have changed in this diff Show More