@purpleschool/gptbot 0.12.70 → 0.12.72

package/build/models/canvas-node.schema.js

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CanvasNodeSchema = exports.CanvasNodeErrorSchema = exports.CanvasNodeOutputSchema = exports.CanvasNodeParamsSchemaByType = exports.CanvasNodeParamsSchema = exports.HtmlGenOutputSchema = exports.VarGenOutputSchema = exports.MotionControlOutputSchema = exports.ColorPaletteOutputSchema = exports.AudioGenSTTOutputSchema = exports.AudioGenTTSOutputSchema = exports.VideoGenOutputSchema = exports.ImageGenOutputSchema = exports.TextGenOutputSchema = exports.VarBlockParamsSchema = exports.NoteParamsSchema = exports.FileBlockParamsSchema = exports.VideoBlockParamsSchema = exports.ImageBlockParamsSchema = exports.TextBlockParamsSchema = exports.HtmlGenParamsSchema = exports.VarGenParamsSchema = exports.MotionControlParamsSchema = exports.ColorPaletteParamsSchema = exports.AudioGenParamsSchema = exports.VideoGenParamsSchema = exports.ImageGenParamsSchema = exports.TextGenParamsSchema = exports.CanvasNodePositionSchema = void 0;
+exports.CanvasNodeSchema = exports.CanvasNodeBaseSchema = exports.validateCanvasNodeOutputByType = exports.validateCanvasNodeParamsByType = exports.CanvasNodeErrorSchema = exports.CanvasNodeOutputSchemaByType = exports.CanvasNodeParamsSchemaByType = exports.HtmlGenOutputSchema = exports.VarGenOutputSchema = exports.MotionControlOutputSchema = exports.ColorPaletteOutputSchema = exports.AudioGenSTTOutputSchema = exports.AudioGenTTSOutputSchema = exports.VideoGenOutputSchema = exports.ImageGenOutputSchema = exports.TextGenOutputSchema = exports.VarBlockParamsSchema = exports.NoteParamsSchema = exports.FileBlockParamsSchema = exports.VideoBlockParamsSchema = exports.ImageBlockParamsSchema = exports.TextBlockParamsSchema = exports.HtmlGenParamsSchema = exports.VarGenParamsSchema = exports.MotionControlParamsSchema = exports.ColorPaletteParamsSchema = exports.AudioGenParamsSchema = exports.VideoGenParamsSchema = exports.ImageGenParamsSchema = exports.TextGenParamsSchema = exports.CanvasNodePositionSchema = void 0;
 const zod_1 = require("zod");
 const constants_1 = require("../constants");
 const strictObject = (shape) => zod_1.z.object(shape).strict();
@@ -49,6 +49,7 @@ exports.AudioGenParamsSchema = zod_1.z.discriminatedUnion('mode', [
 ]);
 exports.ColorPaletteParamsSchema = strictObject({
     prompt: zod_1.z.string(),
+    colors: zod_1.z.array(zod_1.z.string().regex(/^#[0-9A-Fa-f]{6}$/)).optional(),
 });
 exports.MotionControlParamsSchema = strictObject({
     prompt: zod_1.z.string(),
@@ -105,12 +106,7 @@ exports.AudioGenSTTOutputSchema = strictObject({
     transcript: zod_1.z.string(),
 });
 exports.ColorPaletteOutputSchema = strictObject({
-    paletteName: zod_1.z.string(),
-    colors: zod_1.z.array(strictObject({
-        hex: zod_1.z.string(),
-        name: zod_1.z.string(),
-        usage: zod_1.z.string(),
-    })),
+    colors: zod_1.z.array(zod_1.z.string().regex(/^#[0-9A-Fa-f]{6}$/)).max(7),
 });
 exports.MotionControlOutputSchema = strictObject({
     resultUrl: zod_1.z.string(),
@@ -121,22 +117,6 @@ exports.VarGenOutputSchema = strictObject({
 exports.HtmlGenOutputSchema = strictObject({
     html: zod_1.z.string(),
 });
-exports.CanvasNodeParamsSchema = zod_1.z.union([
-    exports.TextGenParamsSchema,
-    exports.ImageGenParamsSchema,
-    exports.VideoGenParamsSchema,
-    exports.AudioGenParamsSchema,
-    exports.ColorPaletteParamsSchema,
-    exports.MotionControlParamsSchema,
-    exports.VarGenParamsSchema,
-    exports.HtmlGenParamsSchema,
-    exports.TextBlockParamsSchema,
-    exports.ImageBlockParamsSchema,
-    exports.VideoBlockParamsSchema,
-    exports.FileBlockParamsSchema,
-    exports.NoteParamsSchema,
-    exports.VarBlockParamsSchema,
-]);
 exports.CanvasNodeParamsSchemaByType = {
     [constants_1.CANVAS_NODE_TYPE.textGen]: exports.TextGenParamsSchema,
     [constants_1.CANVAS_NODE_TYPE.imageGen]: exports.ImageGenParamsSchema,
@@ -156,33 +136,156 @@ exports.CanvasNodeParamsSchemaByType = {
     [constants_1.CANVAS_NODE_TYPE.varBlock]: exports.VarBlockParamsSchema,
     [constants_1.CANVAS_NODE_TYPE.htmlGen]: exports.HtmlGenParamsSchema,
 };
-exports.CanvasNodeOutputSchema = zod_1.z.union([
-    exports.TextGenOutputSchema,
-    exports.ImageGenOutputSchema,
-    exports.VideoGenOutputSchema,
-    exports.AudioGenTTSOutputSchema,
-    exports.AudioGenSTTOutputSchema,
-    exports.ColorPaletteOutputSchema,
-    exports.MotionControlOutputSchema,
-    exports.VarGenOutputSchema,
-    exports.HtmlGenOutputSchema,
-]);
+exports.CanvasNodeOutputSchemaByType = {
+    [constants_1.CANVAS_NODE_TYPE.textGen]: exports.TextGenOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.imageGen]: exports.ImageGenOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.imageEdit]: exports.ImageGenOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.videoGen]: exports.VideoGenOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.videoEdit]: exports.VideoGenOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.ttsGen]: exports.AudioGenTTSOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.sttGen]: exports.AudioGenSTTOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.textBlock]: exports.TextBlockParamsSchema,
+    [constants_1.CANVAS_NODE_TYPE.imageBlock]: exports.ImageBlockParamsSchema,
+    [constants_1.CANVAS_NODE_TYPE.videoBlock]: exports.VideoBlockParamsSchema,
+    [constants_1.CANVAS_NODE_TYPE.fileBlock]: exports.FileBlockParamsSchema,
+    [constants_1.CANVAS_NODE_TYPE.note]: exports.NoteParamsSchema,
+    [constants_1.CANVAS_NODE_TYPE.colorPalette]: exports.ColorPaletteOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.motionControl]: exports.MotionControlOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.varGen]: exports.VarGenOutputSchema,
+    [constants_1.CANVAS_NODE_TYPE.varBlock]: exports.VarBlockParamsSchema,
+    [constants_1.CANVAS_NODE_TYPE.htmlGen]: exports.HtmlGenOutputSchema,
+};
 exports.CanvasNodeErrorSchema = zod_1.z.object({
     code: zod_1.z.string(),
     message: zod_1.z.string(),
     details: zod_1.z.record(zod_1.z.unknown()).optional(),
     retryable: zod_1.z.boolean().optional(),
 });
-exports.CanvasNodeSchema = zod_1.z.object({
+const validateCanvasNodeParamsByType = (nodeDefinitionId, params, ctx) => {
+    if (params === null || params === undefined) {
+        return;
+    }
+    const schema = exports.CanvasNodeParamsSchemaByType[nodeDefinitionId];
+    const result = schema.safeParse(params);
+    if (result.success) {
+        return;
+    }
+    for (const issue of result.error.issues) {
+        ctx.addIssue(Object.assign(Object.assign({}, issue), { path: ['params', ...issue.path] }));
+    }
+};
+exports.validateCanvasNodeParamsByType = validateCanvasNodeParamsByType;
+const validateCanvasNodeOutputByType = (nodeDefinitionId, output, ctx) => {
+    if (output === null || output === undefined) {
+        return;
+    }
+    const schema = exports.CanvasNodeOutputSchemaByType[nodeDefinitionId];
+    const result = schema.safeParse(output);
+    if (result.success) {
+        return;
+    }
+    for (const issue of result.error.issues) {
+        ctx.addIssue(Object.assign(Object.assign({}, issue), { path: ['output', ...issue.path] }));
+    }
+};
+exports.validateCanvasNodeOutputByType = validateCanvasNodeOutputByType;
+exports.CanvasNodeBaseSchema = zod_1.z.object({
     uuid: zod_1.z.string().uuid(),
     canvasId: zod_1.z.string().uuid(),
     nodeDefinitionId: zod_1.z.nativeEnum(constants_1.CANVAS_NODE_TYPE),
     position: exports.CanvasNodePositionSchema,
     status: zod_1.z.nativeEnum(constants_1.NODE_RUN_STATUS),
     toolJobId: zod_1.z.string().uuid().nullable(),
-    params: exports.CanvasNodeParamsSchema.nullable(),
-    output: exports.CanvasNodeOutputSchema.nullable(),
+    params: zod_1.z.unknown().nullable(),
+    output: zod_1.z.unknown().nullable(),
     error: exports.CanvasNodeErrorSchema.nullable(),
     createdAt: zod_1.z.date(),
     updatedAt: zod_1.z.date(),
 });
+exports.CanvasNodeSchema = zod_1.z.discriminatedUnion('nodeDefinitionId', [
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.textGen),
+        params: exports.TextGenParamsSchema.nullable(),
+        output: exports.TextGenOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.imageGen),
+        params: exports.ImageGenParamsSchema.nullable(),
+        output: exports.ImageGenOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.imageEdit),
+        params: exports.ImageGenParamsSchema.nullable(),
+        output: exports.ImageGenOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.videoGen),
+        params: exports.VideoGenParamsSchema.nullable(),
+        output: exports.VideoGenOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.videoEdit),
+        params: exports.VideoGenParamsSchema.nullable(),
+        output: exports.VideoGenOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.ttsGen),
+        params: exports.AudioGenParamsSchema.nullable(),
+        output: exports.AudioGenTTSOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.sttGen),
+        params: exports.AudioGenParamsSchema.nullable(),
+        output: exports.AudioGenSTTOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.textBlock),
+        params: exports.TextBlockParamsSchema.nullable(),
+        output: exports.TextBlockParamsSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.imageBlock),
+        params: exports.ImageBlockParamsSchema.nullable(),
+        output: exports.ImageBlockParamsSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.videoBlock),
+        params: exports.VideoBlockParamsSchema.nullable(),
+        output: exports.VideoBlockParamsSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.fileBlock),
+        params: exports.FileBlockParamsSchema.nullable(),
+        output: exports.FileBlockParamsSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.note),
+        params: exports.NoteParamsSchema.nullable(),
+        output: exports.NoteParamsSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.colorPalette),
+        params: exports.ColorPaletteParamsSchema.nullable(),
+        output: exports.ColorPaletteOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.motionControl),
+        params: exports.MotionControlParamsSchema.nullable(),
+        output: exports.MotionControlOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.varGen),
+        params: exports.VarGenParamsSchema.nullable(),
+        output: exports.VarGenOutputSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.varBlock),
+        params: exports.VarBlockParamsSchema.nullable(),
+        output: exports.VarBlockParamsSchema.nullable(),
+    }),
+    exports.CanvasNodeBaseSchema.extend({
+        nodeDefinitionId: zod_1.z.literal(constants_1.CANVAS_NODE_TYPE.htmlGen),
+        params: exports.HtmlGenParamsSchema.nullable(),
+        output: exports.HtmlGenOutputSchema.nullable(),
+    }),
+]);

package/build/models/canvas-template.schema.js

@@ -4,11 +4,15 @@ exports.CanvasTemplateSchema = exports.CanvasTemplateTagSchema = exports.CanvasT
 const zod_1 = require("zod");
 const constants_1 = require("../constants");
 const canvas_node_schema_1 = require("./canvas-node.schema");
-const CanvasTemplateGraphNodeSchema = zod_1.z.object({
+const CanvasTemplateGraphNodeSchema = zod_1.z
+    .object({
     uuid: zod_1.z.string().trim().min(1),
     nodeDefinitionId: zod_1.z.nativeEnum(constants_1.CANVAS_NODE_TYPE),
     position: canvas_node_schema_1.CanvasNodePositionSchema,
-    params: canvas_node_schema_1.CanvasNodeParamsSchema.nullable().optional(),
+    params: zod_1.z.unknown().nullable().optional(),
+})
+    .superRefine((node, ctx) => {
+    (0, canvas_node_schema_1.validateCanvasNodeParamsByType)(node.nodeDefinitionId, node.params, ctx);
 });
 const CanvasTemplateGraphEdgeSchema = zod_1.z.object({
     sourceId: zod_1.z.string().trim().min(1),

package/commands/{canvas → ai-studio/canvas}/create-canvas.command.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasSchema } from '../../models';
+import { CanvasSchema } from '../../../models';
 
 export namespace CreateCanvasCommand {
     export const RequestSchema = z.object({

package/commands/{canvas → ai-studio/canvas}/delete-canvas.command.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasSchema } from '../../models';
+import { CanvasSchema } from '../../../models';
 
 export namespace DeleteCanvasCommand {
     export const RequestSchema = CanvasSchema.pick({

package/commands/{canvas → ai-studio/canvas}/find-canvas-by-uuid-with-relations.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasEdgeSchema, CanvasNodeSchema, CanvasSchema } from '../../models';
+import { CanvasEdgeSchema, CanvasNodeSchema, CanvasSchema } from '../../../models';
 
 export namespace FindCanvasByUuidWithRelationsQuery {
     export const RequestSchema = CanvasSchema.pick({

package/commands/{canvas → ai-studio/canvas}/find-canvas-by-uuid.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasSchema } from '../../models';
+import { CanvasSchema } from '../../../models';
 
 export namespace FindCanvasByUuidQuery {
     export const RequestSchema = CanvasSchema.pick({

package/commands/{canvas → ai-studio/canvas}/find-canvases-by-user.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasSchema } from '../../models';
+import { CanvasSchema } from '../../../models';
 
 export namespace FindCanvasesByUserQuery {
     export const RequestSchema = z.object({
@@ -11,7 +11,7 @@ export namespace FindCanvasesByUserQuery {
     export type Request = z.infer<typeof RequestSchema>;
 
     export const ResponseSchema = z.object({
-        data: z.array(CanvasSchema),
+        data: z.array(CanvasSchema.extend({ countOfNodes: z.number() })),
         page: z.number(),
         totalPages: z.number(),
     });

package/commands/{canvas → ai-studio/canvas}/run-canvas.command.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasEdgeSchema, CanvasNodeSchema, CanvasSchema } from '../../models';
+import { CanvasEdgeSchema, CanvasNodeSchema, CanvasSchema } from '../../../models';
 import { SaveCanvasCommand } from './save-canvas.command';
 
 export namespace RunCanvasCommand {

package/commands/{canvas → ai-studio/canvas}/save-canvas.command.ts

@@ -1,12 +1,12 @@
 import { z } from 'zod';
-import { CANVAS_NODE_TYPE, INPUT_OUTPUT_TYPE } from '../../constants';
+import { CANVAS_NODE_TYPE, INPUT_OUTPUT_TYPE } from '../../../constants';
 import {
     CanvasEdgeSchema,
-    CanvasNodeParamsSchema,
     CanvasNodePositionSchema,
     CanvasNodeSchema,
     CanvasSchema,
-} from '../../models';
+    validateCanvasNodeParamsByType,
+} from '../../../models';
 
 export namespace SaveCanvasCommand {
     export const RequestParamSchema = CanvasSchema.pick({
@@ -15,12 +15,16 @@ export namespace SaveCanvasCommand {
 
     export type RequestParam = z.infer<typeof RequestParamSchema>;
 
-    export const CanvasNodePayloadSchema = z.object({
-        uuid: z.string().uuid(),
-        nodeDefinitionId: z.nativeEnum(CANVAS_NODE_TYPE),
-        position: CanvasNodePositionSchema,
-        params: CanvasNodeParamsSchema.nullable().optional(),
-    });
+    export const CanvasNodePayloadSchema = z
+        .object({
+            uuid: z.string().uuid(),
+            nodeDefinitionId: z.nativeEnum(CANVAS_NODE_TYPE),
+            position: CanvasNodePositionSchema,
+            params: z.unknown().nullable().optional(),
+        })
+        .superRefine((node, ctx) => {
+            validateCanvasNodeParamsByType(node.nodeDefinitionId, node.params, ctx);
+        });
 
     export const CanvasEdgePayloadSchema = z.object({
         sourceId: z.string().uuid(),

package/commands/{canvas → ai-studio/canvas}/update-canvas.command.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasSchema } from '../../models';
+import { CanvasSchema } from '../../../models';
 
 export namespace UpdateCanvasCommand {
     export const RequestParamSchema = CanvasSchema.pick({

package/commands/{canvas-edge → ai-studio/canvas-edge}/create-canvas-edge.command.ts

@@ -1,6 +1,6 @@
 import { z } from 'zod';
-import { INPUT_OUTPUT_TYPE } from '../../constants';
-import { CanvasEdgeSchema } from '../../models';
+import { INPUT_OUTPUT_TYPE } from '../../../constants';
+import { CanvasEdgeSchema } from '../../../models';
 
 export namespace CreateCanvasEdgeCommand {
     export const RequestParamSchema = z.object({

package/commands/{canvas-edge → ai-studio/canvas-edge}/find-canvas-edge-by-source-and-target.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasEdgeSchema } from '../../models';
+import { CanvasEdgeSchema } from '../../../models';
 
 export namespace FindCanvasEdgeBySourceAndTargetQuery {
     export const RequestParamSchema = z.object({

package/commands/{canvas-edge → ai-studio/canvas-edge}/find-canvas-edges-by-canvas.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasEdgeSchema } from '../../models';
+import { CanvasEdgeSchema } from '../../../models';
 
 export namespace FindCanvasEdgesByCanvasQuery {
     export const RequestParamSchema = z.object({

package/commands/ai-studio/canvas-node/count-canvas-nodes-by-canvas.query.ts

@@ -0,0 +1,20 @@
+import { z } from 'zod';
+
+export namespace CountCanvasNodesByCanvasQuery {
+    export const RequestSchema = z.object({
+        canvasIds: z.array(z.string().uuid()),
+    });
+
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = z.object({
+        data: z.array(
+            z.object({
+                canvasId: z.string().uuid(),
+                countOfNodes: z.number(),
+            }),
+        ),
+    });
+
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/commands/ai-studio/canvas-node/create-canvas-node.command.ts

@@ -0,0 +1,33 @@
+import { z } from 'zod';
+import { CANVAS_NODE_TYPE } from '../../../constants';
+import {
+    CanvasNodePositionSchema,
+    CanvasNodeSchema,
+    validateCanvasNodeParamsByType,
+} from '../../../models';
+
+export namespace CreateCanvasNodeCommand {
+    export const RequestParamSchema = z.object({
+        canvasId: z.string().uuid(),
+    });
+
+    export type RequestParam = z.infer<typeof RequestParamSchema>;
+
+    export const RequestSchema = z
+        .object({
+            nodeDefinitionId: z.nativeEnum(CANVAS_NODE_TYPE),
+            position: CanvasNodePositionSchema,
+            params: z.unknown().nullable().optional(),
+        })
+        .superRefine((request, ctx) => {
+            validateCanvasNodeParamsByType(request.nodeDefinitionId, request.params, ctx);
+        });
+
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = z.object({
+        data: CanvasNodeSchema,
+    });
+
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/commands/{canvas-node → ai-studio/canvas-node}/delete-canvas-node.command.ts

@@ -1,8 +1,8 @@
 import { z } from 'zod';
-import { CanvasNodeSchema } from '../../models';
+import { CanvasNodeBaseSchema } from '../../../models';
 
 export namespace DeleteCanvasNodeCommand {
-    export const RequestParamSchema = CanvasNodeSchema.pick({
+    export const RequestParamSchema = CanvasNodeBaseSchema.pick({
         canvasId: true,
         uuid: true,
     });

package/commands/{canvas-node → ai-studio/canvas-node}/find-canvas-node-by-uuid.query.ts

@@ -1,8 +1,8 @@
 import { z } from 'zod';
-import { CanvasNodeSchema } from '../../models';
+import { CanvasNodeBaseSchema, CanvasNodeSchema } from '../../../models';
 
 export namespace FindCanvasNodeByUuidQuery {
-    export const RequestParamSchema = CanvasNodeSchema.pick({
+    export const RequestParamSchema = CanvasNodeBaseSchema.pick({
         canvasId: true,
         uuid: true,
     });

package/commands/{canvas-node → ai-studio/canvas-node}/find-canvas-node-definitions-with-configs.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasNodeDefinitionsWithConfigsSchema } from '../../models';
+import { CanvasNodeDefinitionsWithConfigsSchema } from '../../../models';
 
 export namespace FindCanvasNodeDefinitionsWithConfigsQuery {
     export const ResponseSchema = z.object({

package/commands/{canvas-node → ai-studio/canvas-node}/find-canvas-nodes-by-canvas.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasNodeSchema } from '../../models';
+import { CanvasNodeSchema } from '../../../models';
 
 export namespace FindCanvasNodesByCanvasQuery {
     export const RequestParamSchema = z.object({

package/commands/{canvas-node → ai-studio/canvas-node}/index.ts

@@ -1,3 +1,4 @@
+export * from './count-canvas-nodes-by-canvas.query';
 export * from './create-canvas-node.command';
 export * from './delete-canvas-node.command';
 export * from './find-canvas-node-by-uuid.query';

package/commands/ai-studio/canvas-node/update-canvas-node.command.ts

@@ -0,0 +1,40 @@
+import { z } from 'zod';
+import { CANVAS_NODE_TYPE } from '../../../constants';
+import {
+    CanvasNodeBaseSchema,
+    CanvasNodePositionSchema,
+    CanvasNodeSchema,
+    validateCanvasNodeOutputByType,
+    validateCanvasNodeParamsByType,
+} from '../../../models';
+
+export namespace UpdateCanvasNodeCommand {
+    export const RequestParamSchema = CanvasNodeBaseSchema.pick({
+        canvasId: true,
+        uuid: true,
+    });
+
+    export type RequestParam = z.infer<typeof RequestParamSchema>;
+
+    export const RequestSchema = z
+        .object({
+            nodeDefinitionId: z.nativeEnum(CANVAS_NODE_TYPE).optional(),
+            position: CanvasNodePositionSchema.optional(),
+            params: z.unknown().nullable().optional(),
+            output: z.unknown().nullable().optional(),
+        })
+        .superRefine((request, ctx) => {
+            if (request.nodeDefinitionId !== undefined) {
+                validateCanvasNodeParamsByType(request.nodeDefinitionId, request.params, ctx);
+                validateCanvasNodeOutputByType(request.nodeDefinitionId, request.output, ctx);
+            }
+        });
+
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = z.object({
+        data: CanvasNodeSchema,
+    });
+
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/commands/{canvas-template → ai-studio/canvas-template}/create-canvas-from-template.command.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasEdgeSchema, CanvasNodeSchema, CanvasSchema } from '../../models';
+import { CanvasEdgeSchema, CanvasNodeSchema, CanvasSchema } from '../../../models';
 
 export namespace CreateCanvasFromTemplateCommand {
     export const RequestParamSchema = z.object({

package/commands/{canvas-template → ai-studio/canvas-template}/create-canvas-template.command.ts

@@ -1,10 +1,6 @@
 import { z } from 'zod';
-import {
-    CanvasTemplateSchema,
-    CanvasTemplateGraphSchema,
-    CanvasTemplateScopeSchema,
-    CanvasTemplateTagSchema,
-} from '../../models';
+import { CanvasTemplateGraphSchema, CanvasTemplateSchema, CanvasTemplateScopeSchema, CanvasTemplateTagSchema } from '../../../models';
+
 
 export namespace CreateCanvasTemplateCommand {
     export const RequestSchema = z.object({

package/commands/{canvas-template → ai-studio/canvas-template}/delete-canvas-template.command.ts

@@ -1,8 +1,9 @@
 import { z } from 'zod';
+import { CanvasTemplateSchema } from '../../../models';
 
 export namespace DeleteCanvasTemplateCommand {
-    export const RequestParamSchema = z.object({
-        uuid: z.string().uuid(),
+    export const RequestParamSchema = CanvasTemplateSchema.pick({
+        uuid: true,
     });
 
     export type RequestParam = z.infer<typeof RequestParamSchema>;

package/commands/{canvas-template → ai-studio/canvas-template}/find-canvas-template-by-uuid.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasTemplateSchema } from '../../models';
+import { CanvasTemplateSchema } from '../../../models';
 
 export namespace FindCanvasTemplateByUuidQuery {
     export const RequestSchema = CanvasTemplateSchema.pick({

package/commands/{canvas-template → ai-studio/canvas-template}/find-canvas-templates.query.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasTemplateSchema, CanvasTemplateScopeSchema } from '../../models';
+import { CanvasTemplateSchema, CanvasTemplateScopeSchema } from '../../../models';
 
 export namespace FindCanvasTemplatesQuery {
     export const RequestSchema = z.object({

package/commands/{canvas-template → ai-studio/canvas-template}/generate-canvas-template-by-prompt.command.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { CanvasTemplateSchema } from '../../models';
+import { CanvasTemplateSchema } from '../../../models';
 
 export namespace GenerateCanvasTemplateByPromptCommand {
     export const RequestSchema = z.object({

package/commands/{canvas-template → ai-studio/canvas-template}/update-canvas-template.command.ts

@@ -4,7 +4,7 @@ import {
     CanvasTemplateGraphSchema,
     CanvasTemplateScopeSchema,
     CanvasTemplateTagSchema,
-} from '../../models';
+} from '../../../models';
 
 export namespace UpdateCanvasTemplateCommand {
     export const RequestParamSchema = CanvasTemplateSchema.pick({

package/commands/ai-studio/canvas-tool/index.ts

@@ -0,0 +1 @@
+export * from './palette';

package/commands/ai-studio/canvas-tool/palette/generate-palette.command.ts

@@ -0,0 +1,23 @@
+import { z } from 'zod';
+
+const HexColorSchema = z.string().regex(/^#[0-9A-Fa-f]{6}$/);
+
+export namespace GeneratePaletteCommand {
+    export const PaletteSchema = z.object({
+        colors: z.array(HexColorSchema).max(7),
+    });
+    export type Palette = z.infer<typeof PaletteSchema>;
+
+    export const RequestSchema = z.object({
+        nodeUuid: z.string().uuid(),
+        canvasId: z.string().uuid(),
+        prompt: z.string().min(1),
+        colors: z.array(HexColorSchema).optional(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = z.object({
+        data: PaletteSchema,
+    });
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/commands/ai-studio/canvas-tool/palette/get-palette-price.command.ts

@@ -0,0 +1,17 @@
+import { z } from 'zod';
+
+export namespace GetPalettePriceCommand {
+    export const RequestSchema = z.object({
+        promptLength: z.coerce.number().positive(),
+    });
+    export type Request = z.infer<typeof RequestSchema>;
+
+    export const ResponseSchema = z.object({
+        data: z.object({
+            price: z.number(),
+            charsUntilNextPriceIncrease: z.number().int().nonnegative().nullable().optional(),
+            charsUntilNextPriceDecrease: z.number().int().nonnegative().nullable().optional(),
+        }),
+    });
+    export type Response = z.infer<typeof ResponseSchema>;
+}

package/commands/ai-studio/canvas-tool/palette/index.ts

@@ -0,0 +1,2 @@
+export * from './generate-palette.command';
+export * from './get-palette-price.command';

package/commands/ai-studio/index.ts

@@ -0,0 +1,5 @@
+export * from './canvas';
+export * from './canvas-edge';
+export * from './canvas-node';
+export * from './canvas-template';
+export * from './canvas-tool';