feat(Information Extractor Node): Add new simplified AI-node for information extraction (#10149)

This commit is contained in:
Eugene
2024-07-25 14:47:18 +02:00
committed by GitHub
parent 49c7306feb
commit 3d235b0b2d
5 changed files with 566 additions and 0 deletions

View File

@@ -0,0 +1,33 @@
import { z } from 'zod';
import type { AttributeDefinition } from './types';
function makeAttributeSchema(attributeDefinition: AttributeDefinition, required: boolean = true) {
let schema: z.ZodTypeAny;
if (attributeDefinition.type === 'string') {
schema = z.string();
} else if (attributeDefinition.type === 'number') {
schema = z.number();
} else if (attributeDefinition.type === 'boolean') {
schema = z.boolean();
} else if (attributeDefinition.type === 'date') {
schema = z.string().date();
} else {
schema = z.unknown();
}
if (!required) {
schema = schema.optional();
}
return schema.describe(attributeDefinition.description);
}
export function makeZodSchemaFromAttributes(attributes: AttributeDefinition[]) {
const schemaEntries = attributes.map((attr) => [
attr.name,
makeAttributeSchema(attr, attr.required),
]);
return z.object(Object.fromEntries(schemaEntries));
}