优化agent

backend/app/services/storage.py

@@ -88,11 +88,26 @@ class WorkspaceStore:
         write_json(path, payload)
         return (Path(conversation) / relative).as_posix()
 
+    def write_cdsl_attempt(self, conversation_id: str, cdsl: Any, iteration: int) -> str:
+        """Retain a parsed model candidate before validation or execution."""
+        conversation = safe_conversation_id(conversation_id)
+        relative = Path("diagnostics") / f"cdsl_attempt_{iteration:02d}_{secrets.token_hex(8)}.json"
+        path = self.conversation_dir(conversation) / relative
+        write_json(path, cdsl)
+        return (Path(conversation) / relative).as_posix()
+
+    def write_cdsl_validation_diagnostic(self, conversation_id: str, payload: dict[str, Any]) -> str:
+        """Persist the reason a retained CDSL candidate was rejected."""
+        conversation = safe_conversation_id(conversation_id)
+        relative = Path("diagnostics") / f"cdsl_validation_{secrets.token_hex(8)}.json"
+        path = self.conversation_dir(conversation) / relative
+        write_json(path, payload)
+        return (Path(conversation) / relative).as_posix()
+
     def ensure_conversation(
         self,
         conversation_id: str | None,
         current_task_id: str | None = None,
-        attachments: list[dict[str, Any]] | None = None,
     ) -> dict[str, Any]:
         cid = safe_conversation_id(conversation_id) if conversation_id else new_id("conv")
         path = self.conversation_path(cid)
@@ -102,21 +117,18 @@ class WorkspaceStore:
             if current_task_id:
                 current["current_task_id"] = safe_task_id(current_task_id)
                 changed = True
-            if attachments is not None:
-                current["attachments"] = attachments
-                changed = True
             if changed:
                 current["updated_at"] = now_iso()
                 write_json(path, current)
             return current
         record = {
-            "schema_version": "1.1",
+            "schema_version": "1.2",
             "conversation_id": cid,
             "created_at": now_iso(),
             "updated_at": now_iso(),
             "current_task_id": safe_task_id(current_task_id) if current_task_id else "",
             "messages": [],
-            "attachments": attachments or [],
+            "attachments": [],
         }
         write_json(path, record)
         return record
@@ -135,14 +147,34 @@ class WorkspaceStore:
         write_json(self.conversation_path(record["conversation_id"]), record)
         return record
 
-    def write_upload(self, task_id: str, filename: str, data: bytes) -> tuple[str, Path]:
+    def write_conversation_upload(self, conversation_id: str, filename: str, data: bytes) -> tuple[str, Path]:
+        conversation = safe_conversation_id(conversation_id)
         safe_name = re.sub(r"[^a-zA-Z0-9._-]+", "_", Path(filename).name).strip("._") or "attachment"
         relative = Path("uploads") / f"upload_{secrets.token_hex(6)}_{safe_name}"
-        target = self.artifact_path(task_id, relative.as_posix())
+        target = self.conversation_attachment_path(conversation, relative.as_posix())
         target.parent.mkdir(parents=True, exist_ok=True)
         target.write_bytes(data)
         return relative.as_posix(), target
 
+    def add_conversation_attachment(self, conversation_id: str, attachment: dict[str, Any]) -> dict[str, Any]:
+        conversation = safe_conversation_id(conversation_id)
+        record = self.read_conversation(conversation)
+        if record is None:
+            raise ValueError("Conversation not found")
+        if str(attachment.get("conversation_id") or "") != conversation:
+            raise ValueError("Attachment does not belong to this conversation")
+        attachment_id = str(attachment.get("id") or "")
+        if not attachment_id:
+            raise ValueError("Attachment id is required")
+        self.conversation_attachment_path(conversation, str(attachment.get("path") or ""))
+        attachments = record.setdefault("attachments", [])
+        if any(str(item.get("id") or "") == attachment_id for item in attachments if isinstance(item, dict)):
+            raise ValueError("Attachment already exists")
+        attachments.append(attachment)
+        record["updated_at"] = now_iso()
+        write_json(self.conversation_path(conversation), record)
+        return record
+
     def ensure_task(self, task_id: str | None, request: str) -> dict[str, Any]:
         tid = safe_task_id(task_id) if task_id else new_id("cad")
         path = self.task_path(tid)
@@ -274,3 +306,11 @@ class WorkspaceStore:
         if root != target and root not in target.parents:
             raise ValueError("Artifact path escapes task directory")
         return target
+
+    def conversation_attachment_path(self, conversation_id: str, relative_path: str) -> Path:
+        safe = safe_relative_path(relative_path)
+        root = self.conversation_dir(conversation_id).resolve()
+        target = (root / safe).resolve()
+        if root != target and root not in target.parents:
+            raise ValueError("Attachment path escapes conversation directory")
+        return target