This commit is contained in:
jhpiedrahitao
2025-06-24 15:59:40 -05:00
parent d07f45132f
commit dfb0da32a9
3 changed files with 8 additions and 10 deletions

View File

@@ -5,8 +5,8 @@
# #
import argparse import argparse
import time
import os import os
import time
from dotenv import load_dotenv from dotenv import load_dotenv
from loguru import logger from loguru import logger
@@ -75,10 +75,9 @@ transport_params = {
async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool): async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_sigint: bool):
logger.info(f"Starting bot") logger.info(f"Starting bot")
stt = SambaNovaSTTService( stt = SambaNovaSTTService(
model='Whisper-Large-v3', model="Whisper-Large-v3",
api_key=os.getenv('SAMBANOVA_API_KEY'), api_key=os.getenv("SAMBANOVA_API_KEY"),
) )
tl = TranscriptionLogger() tl = TranscriptionLogger()

View File

@@ -20,9 +20,9 @@ from pipecat.pipeline.task import PipelineParams, PipelineTask
from pipecat.processors.aggregators.llm_response import LLMUserAggregatorParams from pipecat.processors.aggregators.llm_response import LLMUserAggregatorParams
from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext from pipecat.processors.aggregators.openai_llm_context import OpenAILLMContext
from pipecat.services.cartesia.tts import CartesiaTTSService from pipecat.services.cartesia.tts import CartesiaTTSService
from pipecat.services.llm_service import FunctionCallParams
from pipecat.services.sambanova.llm import SambaNovaLLMService from pipecat.services.sambanova.llm import SambaNovaLLMService
from pipecat.services.sambanova.stt import SambaNovaSTTService from pipecat.services.sambanova.stt import SambaNovaSTTService
from pipecat.services.llm_service import FunctionCallParams
from pipecat.transports.base_transport import BaseTransport, TransportParams from pipecat.transports.base_transport import BaseTransport, TransportParams
from pipecat.transports.network.fastapi_websocket import FastAPIWebsocketParams from pipecat.transports.network.fastapi_websocket import FastAPIWebsocketParams
from pipecat.transports.services.daily import DailyParams from pipecat.transports.services.daily import DailyParams
@@ -60,8 +60,8 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_si
logger.info(f"Starting bot") logger.info(f"Starting bot")
stt = SambaNovaSTTService( stt = SambaNovaSTTService(
model='Whisper-Large-v3', model="Whisper-Large-v3",
api_key=os.getenv('SAMBANOVA_API_KEY'), api_key=os.getenv("SAMBANOVA_API_KEY"),
) )
tts = CartesiaTTSService( tts = CartesiaTTSService(
@@ -70,8 +70,8 @@ async def run_example(transport: BaseTransport, _: argparse.Namespace, handle_si
) )
llm = SambaNovaLLMService( llm = SambaNovaLLMService(
api_key=os.getenv('SAMBANOVA_API_KEY'), api_key=os.getenv("SAMBANOVA_API_KEY"),
model='Llama-4-Maverick-17B-128E-Instruct', model="Llama-4-Maverick-17B-128E-Instruct",
) )
# You can also register a function_name of None to get all functions # You can also register a function_name of None to get all functions
# sent to the same callback with an additional function_name parameter. # sent to the same callback with an additional function_name parameter.

View File

@@ -6,4 +6,3 @@
from .llm import * from .llm import *
from .stt import * from .stt import *