wehub-resource-sync
119 lines
3.7 KiB
TypeScript
119 lines
3.7 KiB
TypeScript
import { WorkflowTool, WorkflowAgent, WorkflowPrompt, WorkflowPipeline } from "./types/workflow_types";
|
|
import { Message } from "./types/types";
|
|
import { z } from "zod";
|
|
|
|
const ZFallbackSchema = z.object({}).passthrough();
|
|
|
|
export function validateConfigChanges(configType: string, configChanges: Record<string, unknown>, name: string) {
|
|
let testObject: any;
|
|
let schema: z.ZodType<any> = ZFallbackSchema;
|
|
|
|
switch (configType) {
|
|
case 'tool': {
|
|
testObject = {
|
|
name: 'test',
|
|
description: 'test',
|
|
parameters: {
|
|
type: 'object',
|
|
properties: {},
|
|
required: [],
|
|
},
|
|
} as z.infer<typeof WorkflowTool>;
|
|
schema = WorkflowTool;
|
|
break;
|
|
}
|
|
case 'agent': {
|
|
testObject = {
|
|
name: 'test',
|
|
description: 'test',
|
|
type: 'conversation',
|
|
instructions: 'test',
|
|
prompts: [],
|
|
tools: [],
|
|
model: 'gpt-4.1',
|
|
ragReturnType: 'chunks',
|
|
ragK: 10,
|
|
connectedAgents: [],
|
|
controlType: 'retain',
|
|
outputVisibility: 'user_facing',
|
|
maxCallsPerParentAgent: 3,
|
|
} as z.infer<typeof WorkflowAgent>;
|
|
schema = WorkflowAgent;
|
|
break;
|
|
}
|
|
case 'prompt': {
|
|
testObject = {
|
|
name: 'test',
|
|
type: 'base_prompt',
|
|
prompt: "test",
|
|
} as z.infer<typeof WorkflowPrompt>;
|
|
schema = WorkflowPrompt;
|
|
break;
|
|
}
|
|
case 'pipeline': {
|
|
testObject = {
|
|
name: 'test',
|
|
description: 'test',
|
|
agents: [],
|
|
} as z.infer<typeof WorkflowPipeline>;
|
|
schema = WorkflowPipeline;
|
|
break;
|
|
}
|
|
case 'start_agent': {
|
|
testObject = {};
|
|
break;
|
|
}
|
|
case 'one_time_trigger': {
|
|
testObject = {
|
|
scheduledTime: new Date(0).toISOString(),
|
|
input: {
|
|
messages: [],
|
|
},
|
|
};
|
|
schema = z.object({
|
|
scheduledTime: z.string().min(1),
|
|
input: z.object({
|
|
messages: z.array(Message),
|
|
}),
|
|
}).passthrough();
|
|
break;
|
|
}
|
|
case 'recurring_trigger': {
|
|
testObject = {
|
|
cron: '* * * * *',
|
|
input: {
|
|
messages: [],
|
|
},
|
|
};
|
|
schema = z.object({
|
|
cron: z.string().min(1),
|
|
input: z.object({
|
|
messages: z.array(Message),
|
|
}),
|
|
}).passthrough();
|
|
break;
|
|
}
|
|
case 'external_trigger': {
|
|
// External triggers have flexible schemas per provider; do not strip any config.
|
|
return { changes: configChanges };
|
|
}
|
|
default:
|
|
return { error: `Unknown config type: ${configType}` };
|
|
}
|
|
|
|
// Validate each field and remove invalid ones
|
|
const validatedChanges = { ...configChanges };
|
|
for (const [key, value] of Object.entries(configChanges)) {
|
|
const result = schema.safeParse({
|
|
...testObject,
|
|
[key]: value,
|
|
});
|
|
if (!result.success) {
|
|
console.log(`discarding field ${key} from ${configType}: ${name}`, result.error.message);
|
|
delete validatedChanges[key];
|
|
}
|
|
}
|
|
|
|
return { changes: validatedChanges };
|
|
}
|