comfyui-deploy/web/src/lib/workflowVersionInputsToZod.tsx
2024-01-14 23:35:25 +08:00

25 lines
604 B
TypeScript

import type { WorkflowVersionType } from "@/db/schema";
import { getInputsFromWorkflow } from "@/lib/getInputsFromWorkflow";
import { z } from "zod";
export function workflowVersionInputsToZod(
workflow_version: WorkflowVersionType
) {
const inputs = getInputsFromWorkflow(workflow_version);
return plainInputsToZod(inputs);
}
export function plainInputsToZod(
inputs: ReturnType<typeof getInputsFromWorkflow>
) {
if (!inputs) return null;
return z.object({
...Object.fromEntries(
inputs?.map((x) => {
return [x?.input_id, z.string().optional()];
})
),
});
}