codegen output token stream prints to console in testing

2026-09-24 03:48:03 -05:00
parent bb97a09caa
commit 8c10a94c32
5 changed files with 133 additions and 1 deletions
@@ -163,6 +163,7 @@ def test_generate_skill_body_codegen(tmp_path):
        base_url=codegen_cfg.get("base_url", "http://localhost:11434/v1"),
        model=codegen_cfg.get("model", "qwen38-iq3s"),
        timeout=float(codegen_cfg.get("timeout", 1200.0)),
+        stream=True,
    )
    tree = build_tree(build_skills({"skills": {}}))
    draft = SkillDraft(
@@ -223,6 +224,7 @@ def test_create_category_chain_runs_new_skill(tmp_path):
    config["trace"] = str(tmp_path / "runs.jsonl")
    config["category_registry"] = str(tmp_path / "categories.json")
    config["skill_bodies"] = str(tmp_path / "skills")
+    config["codegen"] = {**config.get("codegen", {}), "stream": True}
    scheduler, config = build_scheduler(config)
    scheduler.tree = {}