diff --git a/cypher-executor/src/lib/schemas.ts b/cypher-executor/src/lib/schemas.ts
index 0fe5391..9ebbd0a 100644
--- a/cypher-executor/src/lib/schemas.ts
+++ b/cypher-executor/src/lib/schemas.ts
@@ -8,12 +8,13 @@ export const graphSchema = z.object({
     id: z.string(),
     type: z.enum(['Input', 'Component', 'Output']),
     componentId: z.string().optional(),
+    label: z.string().optional(),
     data: z.record(z.unknown()).optional(),
   })),
   edges: z.array(z.object({
     from: z.string(),
     to: z.string(),
-    type: z.enum(['PIPE', 'IF', 'FOREACH', 'CONTINUE']),
+    type: z.enum(['PIPE', 'IF', 'FOREACH', 'CONTINUE', 'IS_A', 'ON_SUCCESS', 'ON_FAIL', 'ON_CLICK', 'CALLS_SUBFLOW', 'CONTAINS', 'HAS_STYLE', 'HAS_BEHAVIOR']),
     condition: z.string().optional(),
     iterator: z.string().optional(),
   })),