from agent.CADAgent import CADAgent


def test_cadagent_prompt_includes_artifact_tags() -> None:
    prompt = CADAgent.chat_impl.__doc__ or ""

    assert "<|code_file|>" in prompt
    assert "<|output_file|>" in prompt
    assert "Repeat the final `<|code_file|>` and `<|output_file|>` tags" in prompt


def test_cadagent_prompt_prefers_step_for_visual_feedback() -> None:
    prompt = CADAgent.chat_impl.__doc__ or ""

    assert (
        "prefer passing the exported `.step`/`.stp` file into `get_visual_feedback`"
        in prompt
    )
    assert "visual feedback should use the STEP/STP file" in prompt