Function configureStructuredOutput

packages/components/src/utils.ts:2123–2173

(llmNodeInstance: BaseChatModel, structuredOutput: any[])

 * @returns {BaseChatModel} - The configured LLM instance
 */
export const configureStructuredOutput = (llmNodeInstance: BaseChatModel, structuredOutput: any[]): BaseChatModel => {
    try {
        const zodObj: ICommonObject = {}
        for (const sch of structuredOutput) {
            if (sch.type === 'string') {
                zodObj[sch.key] = z.string().describe(sch.description || '')
            } else if (sch.type === 'stringArray') {
                zodObj[sch.key] = z.array(z.string()).describe(sch.description || '')
            } else if (sch.type === 'number') {
                zodObj[sch.key] = z.number().describe(sch.description || '')
            } else if (sch.type === 'boolean') {
                zodObj[sch.key] = z.boolean().describe(sch.description || '')
            } else if (sch.type === 'enum') {
                const enumValues = sch.enumValues?.split(',').map((item: string) => item.trim()) || []
                zodObj[sch.key] = z
                    .enum(enumValues.length ? (enumValues as [string, ...string[]]) : ['default'])
                    .describe(sch.description || '')
            } else if (sch.type === 'jsonArray') {
                const jsonSchema = sch.jsonSchema
                if (jsonSchema) {
                    try {
                        // Parse the JSON schema
                        const schemaObj = JSON.parse(jsonSchema)

                        // Create a Zod schema from the JSON schema
                        const itemSchema = createZodSchemaFromJSON(schemaObj)

                        // Create an array schema of the item schema
                        zodObj[sch.key] = z.array(itemSchema).describe(sch.description || '')
                    } catch (err) {
                        console.error(`Error parsing JSON schema for ${sch.key}:`, err)
                        // Fallback to generic array of records
                        zodObj[sch.key] = z.array(z.record(z.any())).describe(sch.description || '')
                    }
                } else {
                    // If no schema provided, use generic array of records
                    zodObj[sch.key] = z.array(z.record(z.any())).describe(sch.description || '')
                }
            }
        }
        const structuredOutputSchema = z.object(zodObj)

        // @ts-ignore
        return llmNodeInstance.withStructuredOutput(structuredOutputSchema, {
            method: 'functionCalling'
        })
    } catch (exception) {
        console.error(exception)
        return llmNodeInstance
    }
}

/**
 * Creates a Zod schema from a JSON schema object