优化流程

backend/app/cad_agent/domain/operation_contract.py

@@ -131,7 +131,7 @@ def validate_operation_contract(contract: dict[str, Any]) -> None:
         raise OperationContractError("Operation candidate verifiers are duplicated")
 
 
-def fragment_schema(contract: dict[str, Any], *, selector_tokens: list[str], reference_tokens: list[str] | None = None) -> dict[str, Any]:
+def fragment_schema(contract: dict[str, Any], *, selector_tokens: list[str], reference_tokens: list[str] | None = None, root_xy_datum: bool = False) -> dict[str, Any]:
     """Build the one-operation schema exposed for one pending action."""
     validate_operation_contract(contract)
     shape = contract["fragment_shape"]
@@ -174,15 +174,15 @@ def fragment_schema(contract: dict[str, Any], *, selector_tokens: list[str], ref
     properties: dict[str, Any] = {"feature": feature}
     required = ["feature"]
     if shape["sketch"] == "required":
-        properties["sketch"] = _sketch_schema()
+        properties["sketch"] = _sketch_schema(root_xy_datum=root_xy_datum and contract["atomic_id"] in {"extrude_add_blind", "extrude_add_two_sided"})
         required.insert(0, "sketch")
     schema = {"$schema": "https://json-schema.org/draft/2020-12/schema", "type": "object", "properties": properties, "required": required, "additionalProperties": False}
     Draft202012Validator.check_schema(schema)
     return schema
 
 
-def validate_fragment(contract: dict[str, Any], fragment: Any, *, selector_tokens: list[str], reference_tokens: list[str] | None = None) -> list[dict[str, str]]:
-    schema = fragment_schema(contract, selector_tokens=selector_tokens, reference_tokens=reference_tokens)
+def validate_fragment(contract: dict[str, Any], fragment: Any, *, selector_tokens: list[str], reference_tokens: list[str] | None = None, root_xy_datum: bool = False) -> list[dict[str, str]]:
+    schema = fragment_schema(contract, selector_tokens=selector_tokens, reference_tokens=reference_tokens, root_xy_datum=root_xy_datum)
     return [
         {"path": "/" + "/".join(str(part) for part in error.absolute_path), "message": error.message}
         for error in sorted(Draft202012Validator(schema).iter_errors(fragment), key=lambda item: (list(item.absolute_path), item.message))
@@ -193,10 +193,23 @@ def _point(size: int) -> dict[str, Any]:
     return {"type": "array", "items": {"type": "number"}, "minItems": size, "maxItems": size}
 
 
-def _sketch_schema() -> dict[str, Any]:
+def _sketch_schema(*, root_xy_datum: bool = False) -> dict[str, Any]:
     point2 = _point(2)
     point3 = _point(3)
-    workplane = {"type": "object", "properties": {"origin_mm": point3, "x_dir": deepcopy(point3), "normal": deepcopy(point3)}, "required": ["origin_mm", "x_dir", "normal"], "additionalProperties": False}
+    workplane = {
+        "type": "object",
+        "description": "origin_mm is the world position of sketch local (0,0); profile coordinates are local to this plane. normal is positive extrusion direction and x_dir is local +X in world coordinates.",
+        "properties": {"origin_mm": point3, "x_dir": deepcopy(point3), "normal": deepcopy(point3)},
+        "required": ["origin_mm", "x_dir", "normal"],
+        "additionalProperties": False,
+    }
+    if root_xy_datum:
+        workplane["description"] += " Root extrusion uses fixed world XY datum: origin X/Y are 0, normal is +Z, x_dir is +X. Only origin Z is task-defined."
+        workplane["properties"] = {
+            "origin_mm": {"type": "array", "prefixItems": [{"const": 0}, {"const": 0}, {"type": "number"}], "items": False, "minItems": 3, "maxItems": 3},
+            "x_dir": {"const": [1, 0, 0]},
+            "normal": {"const": [0, 0, 1]},
+        }
     profile = {
         "oneOf": [
             {"type": "object", "properties": {"type": {"const": "circle"}, "center": deepcopy(point2), "radius_mm": {"type": "number", "exclusiveMinimum": 0}}, "required": ["type", "radius_mm"], "additionalProperties": False},