ai-utils.js/model-function/generate-json/generateJson.cjs

Build AI applications, chatbots, and agents with JavaScript and TypeScript.

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.generateJson = void 0;
const executeCall_js_1 = require("../executeCall.cjs");
const SchemaValidationError_js_1 = require("./SchemaValidationError.cjs");
async function generateJson(model, schemaDefinition, prompt, options) {
    const expandedPrompt = prompt(schemaDefinition);
    const result = await (0, executeCall_js_1.executeCall)({
        model,
        options,
        generateResponse: (options) => model.generateJsonResponse(expandedPrompt, options),
        extractOutputValue: (response) => {
            const json = expandedPrompt.extractJson(response);
            const parseResult = schemaDefinition.schema.safeParse(json);
            if (!parseResult.success) {
                throw new SchemaValidationError_js_1.SchemaValidationError({
                    schemaName: schemaDefinition.name,
                    value: json,
                    errors: parseResult.error,
                });
            }
            return parseResult.data;
        },
        getStartEvent: (metadata, settings) => ({
            type: "json-generation-started",
            metadata,
            settings,
            prompt,
        }),
        getAbortEvent: (metadata, settings) => ({
            type: "json-generation-finished",
            status: "abort",
            metadata,
            settings,
            prompt,
        }),
        getFailureEvent: (metadata, settings, error) => ({
            type: "json-generation-finished",
            status: "failure",
            metadata,
            settings,
            prompt,
            error,
        }),
        getSuccessEvent: (metadata, settings, response, output) => ({
            type: "json-generation-finished",
            status: "success",
            metadata,
            settings,
            prompt,
            response,
            generatedJson: output,
        }),
    });
    return {
        value: result.output,
        response: result.response,
        metadata: result.metadata,
    };
}
exports.generateJson = generateJson;