debug_tools.py

"""Run one MTSamples record through the coder and print every chunk type."""
import csv
import daimon_client as daimon

with open("data/mtsamples.csv", encoding="utf-8") as f:
    row = next(r for r in csv.DictReader(f) if (r.get("transcription") or "").strip())

text = row["transcription"].strip()[:2000]  # truncate for speed
print(f"Sample: {row.get('sample_name', '').strip()}\n{'─'*60}")

with daimon.Client(base_url="http://localhost:3500") as client:
    for chunk in client.llm("coder").converse(
        messages=[{"role": "user", "content": text}]
    ):
        if chunk.type == "tool_call":
            tc = chunk.tool_call
            print(f"\n→ TOOL: {tc.name}")
            print(f"  input: {tc.input}")
        elif chunk.type == "text":
            print(chunk.text, end="", flush=True)
        elif chunk.type == "error":
            print(f"\n[ERROR] {chunk.error}")

print("\n")