Update workflow feature with codex

api/app/routers/workflows.py

@@ -2,14 +2,30 @@ from fastapi import APIRouter, Depends, HTTPException
 from sqlalchemy.orm import Session
 import uuid
 from datetime import datetime
+from typing import Any, Dict, List, Tuple
 
 from ..db import get_db
 from ..models import Workflow
-from ..schemas import WorkflowCreate, WorkflowUpdate, WorkflowOut
+from ..schemas import WorkflowCreate, WorkflowUpdate, WorkflowOut, WorkflowNode, WorkflowEdge
 
 router = APIRouter(prefix="/workflows", tags=["Workflows"])
 
 
+def _normalize_graph_payload(nodes: List[Any], edges: List[Any]) -> Tuple[List[Dict[str, Any]], List[Dict[str, Any]]]:
+    """Normalize graph payload to canonical dict structures."""
+    parsed_nodes: List[WorkflowNode] = []
+    for node in nodes:
+        parsed_nodes.append(node if isinstance(node, WorkflowNode) else WorkflowNode.model_validate(node))
+
+    parsed_edges: List[WorkflowEdge] = []
+    for edge in edges:
+        parsed_edges.append(edge if isinstance(edge, WorkflowEdge) else WorkflowEdge.model_validate(edge))
+
+    normalized_nodes = [node.model_dump() for node in parsed_nodes]
+    normalized_edges = [edge.model_dump() for edge in parsed_edges]
+    return normalized_nodes, normalized_edges
+
+
 @router.get("")
 def list_workflows(
     page: int = 1,
@@ -27,16 +43,17 @@ def list_workflows(
 @router.post("", response_model=WorkflowOut)
 def create_workflow(data: WorkflowCreate, db: Session = Depends(get_db)):
     """创建工作流"""
+    nodes, edges = _normalize_graph_payload(data.nodes, data.edges)
     workflow = Workflow(
         id=str(uuid.uuid4())[:8],
         user_id=1,
         name=data.name,
-        node_count=data.nodeCount,
+        node_count=data.nodeCount or len(nodes),
         created_at=data.createdAt or datetime.utcnow().isoformat(),
         updated_at=data.updatedAt or "",
         global_prompt=data.globalPrompt,
-        nodes=data.nodes,
-        edges=data.edges,
+        nodes=nodes,
+        edges=edges,
     )
     db.add(workflow)
     db.commit()
@@ -60,7 +77,7 @@ def update_workflow(id: str, data: WorkflowUpdate, db: Session = Depends(get_db)
     if not workflow:
         raise HTTPException(status_code=404, detail="Workflow not found")
 
-    update_data = data.model_dump(exclude_unset=True)
+    update_data = data.model_dump(exclude_unset=True, exclude={"nodes", "edges"})
     field_map = {
         "nodeCount": "node_count",
         "globalPrompt": "global_prompt",
@@ -68,6 +85,16 @@ def update_workflow(id: str, data: WorkflowUpdate, db: Session = Depends(get_db)
     for field, value in update_data.items():
         setattr(workflow, field_map.get(field, field), value)
 
+    if data.nodes is not None or data.edges is not None:
+        existing_nodes = workflow.nodes if isinstance(workflow.nodes, list) else []
+        existing_edges = workflow.edges if isinstance(workflow.edges, list) else []
+        input_nodes = data.nodes if data.nodes is not None else existing_nodes
+        input_edges = data.edges if data.edges is not None else existing_edges
+        nodes, edges = _normalize_graph_payload(input_nodes, input_edges)
+        workflow.nodes = nodes
+        workflow.edges = edges
+        workflow.node_count = len(nodes)
+
     workflow.updated_at = datetime.utcnow().isoformat()
     db.commit()
     db.refresh(workflow)