first commit

CADDesigner-Code-main/test/test_cadagent_prompt_contract.py

@@ -0,0 +1,19 @@
+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