Fix type errors in scripts and add to pyright checked set

This commit is contained in:
Mark Backman
2026-04-21 16:17:49 -04:00
parent 609a0a14e7
commit 10e58d6e42
6 changed files with 25 additions and 19 deletions

View File

@@ -198,7 +198,7 @@ class EvalRunner:
async def run_example_pipeline(script_path: Path, eval_config: EvalConfig):
room_url = os.getenv("DAILY_ROOM_URL")
room_url = os.environ["DAILY_ROOM_URL"]
module = load_module_from_path(script_path)
@@ -227,7 +227,7 @@ async def run_eval_pipeline(
):
logger.info(f"Starting eval bot")
room_url = os.getenv("DAILY_ROOM_URL")
room_url = os.environ["DAILY_ROOM_URL"]
transport = DailyTransport(
room_url,
@@ -243,7 +243,7 @@ async def run_eval_pipeline(
# We disable smart formatting because some times if the user says "3 + 2 is
# 5" (in audio) this can be converted to "32 is 5".
stt = DeepgramSTTService(
api_key=os.getenv("DEEPGRAM_API_KEY"),
api_key=os.environ["DEEPGRAM_API_KEY"],
settings=DeepgramSTTService.Settings(
language="multi",
smart_format=False,
@@ -251,7 +251,7 @@ async def run_eval_pipeline(
)
tts = CartesiaTTSService(
api_key=os.getenv("CARTESIA_API_KEY"),
api_key=os.environ["CARTESIA_API_KEY"],
settings=CartesiaTTSService.Settings(
voice="97f4b8fb-f2fe-444b-bb9a-c109783a857a", # Nathan
),
@@ -375,7 +375,7 @@ async def run_eval_pipeline(
@task.event_handler("on_pipeline_finished")
async def on_pipeline_finished(task, frame):
if isinstance(frame, EndFrame):
await eval_runner.assert_eval(frame.reason)
await eval_runner.assert_eval(bool(frame.reason))
elif isinstance(frame, CancelFrame):
await eval_runner.assert_eval(False)

View File

@@ -11,7 +11,7 @@ import sys
from pathlib import Path
from dotenv import load_dotenv
from eval import EvalRunner
from eval import EvalConfig, EvalRunner
from loguru import logger
from utils import check_env_variables
@@ -33,7 +33,8 @@ async def main(args: argparse.Namespace):
runner = EvalRunner(examples_dir=script_path, record_audio=args.audio, log_level=log_level)
await runner.run_eval(script_file, args.prompt, args.eval)
eval_config = EvalConfig(prompt=args.prompt, eval=args.eval)
await runner.run_eval(script_file, eval_config)
runner.print_results()

View File

@@ -79,6 +79,8 @@ def load_module_from_path(path: str | Path):
module_name = path.stem
spec = importlib.util.spec_from_file_location(module_name, str(path))
if spec is None or spec.loader is None:
raise ImportError(f"Could not load module spec from {path}")
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)
return module