Files
helia/scripts/verify_turns.py

40 lines
1.0 KiB
Python

import asyncio
import logging
from helia.configuration import HeliaConfig
from helia.db import init_db
from helia.models.transcript import Transcript
logging.basicConfig(level=logging.INFO)
logger = logging.getLogger(__name__)
async def verify() -> None:
config = HeliaConfig() # type: ignore[call-arg]
await init_db(config)
transcript = await Transcript.find_one(Transcript.transcript_id == "300")
if not transcript:
logger.error("Transcript 300 not found.")
return
logger.info("Transcript 300 found with %d utterances.", len(transcript.utterances))
turns = transcript.turns
logger.info("Aggregated into %d turns.", len(turns))
for i, turn in enumerate(turns[:5]):
logger.info(
"Turn %d [%s] (%s - %s): %s... (Merged %d utterances)",
i + 1,
turn.speaker,
turn.start_time,
turn.end_time,
turn.value[:50],
turn.utterance_count,
)
if __name__ == "__main__":
asyncio.run(verify())