modelfusion 0.104.0 → 0.106.0

package/browser/readEventSourceStream.cjs

@@ -1,35 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.readEventSourceStream = void 0;
-const parseJSON_js_1 = require("../core/schema/parseJSON.cjs");
-const AsyncQueue_js_1 = require("../util/AsyncQueue.cjs");
-const parseEventSourceStream_js_1 = require("../util/streaming/parseEventSourceStream.cjs");
-function readEventSourceStream({ stream, schema, errorHandler, }) {
-    const queue = new AsyncQueue_js_1.AsyncQueue();
-    // run async (no await on purpose):
-    (0, parseEventSourceStream_js_1.parseEventSourceStream)({ stream })
-        .then(async (events) => {
-        try {
-            for await (const event of events) {
-                const validationResult = (0, parseJSON_js_1.safeParseJSON)({ text: event.data, schema });
-                if (!validationResult.success) {
-                    errorHandler?.(validationResult.error);
-                    continue;
-                }
-                queue.push(validationResult.data);
-            }
-        }
-        catch (error) {
-            errorHandler?.(error);
-        }
-        finally {
-            queue.close();
-        }
-    })
-        .catch((error) => {
-        errorHandler?.(error);
-        queue.close();
-    });
-    return queue;
-}
-exports.readEventSourceStream = readEventSourceStream;

package/browser/readEventSourceStream.d.ts

@@ -1,7 +0,0 @@
-import { Schema } from "../core/schema/Schema.js";
-import { ErrorHandler } from "../util/ErrorHandler.js";
-export declare function readEventSourceStream<T>({ stream, schema, errorHandler, }: {
-    stream: ReadableStream<Uint8Array>;
-    schema: Schema<T>;
-    errorHandler?: ErrorHandler;
-}): AsyncIterable<T>;

package/browser/readEventSourceStream.js

@@ -1,31 +0,0 @@
-import { safeParseJSON } from "../core/schema/parseJSON.js";
-import { AsyncQueue } from "../util/AsyncQueue.js";
-import { parseEventSourceStream } from "../util/streaming/parseEventSourceStream.js";
-export function readEventSourceStream({ stream, schema, errorHandler, }) {
-    const queue = new AsyncQueue();
-    // run async (no await on purpose):
-    parseEventSourceStream({ stream })
-        .then(async (events) => {
-        try {
-            for await (const event of events) {
-                const validationResult = safeParseJSON({ text: event.data, schema });
-                if (!validationResult.success) {
-                    errorHandler?.(validationResult.error);
-                    continue;
-                }
-                queue.push(validationResult.data);
-            }
-        }
-        catch (error) {
-            errorHandler?.(error);
-        }
-        finally {
-            queue.close();
-        }
-    })
-        .catch((error) => {
-        errorHandler?.(error);
-        queue.close();
-    });
-    return queue;
-}

package/composed-function/index.cjs

@@ -1,19 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./summarize/SummarizationFunction.cjs"), exports);
-__exportStar(require("./summarize/summarizeRecursively.cjs"), exports);
-__exportStar(require("./summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.cjs"), exports);

package/composed-function/index.d.ts

@@ -1,3 +0,0 @@
-export * from "./summarize/SummarizationFunction.js";
-export * from "./summarize/summarizeRecursively.js";
-export * from "./summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.js";

package/composed-function/index.js

@@ -1,3 +0,0 @@
-export * from "./summarize/SummarizationFunction.js";
-export * from "./summarize/summarizeRecursively.js";
-export * from "./summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.js";

package/composed-function/summarize/SummarizationFunction.d.ts

@@ -1,4 +0,0 @@
-import { FunctionOptions } from "../../core/FunctionOptions.js";
-export type SummarizationFunction = (input: {
-    text: string;
-}, options?: FunctionOptions) => PromiseLike<string>;

package/composed-function/summarize/summarizeRecursively.cjs

@@ -1,19 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.summarizeRecursively = void 0;
-async function summarizeRecursively({ summarize, split, join = (texts) => texts.join("\n\n"), text, }, options) {
-    const chunks = await split({ text });
-    const summarizedTexts = await Promise.all(chunks.map((chunk) => summarize({ text: chunk }, options)));
-    if (summarizedTexts.length === 1) {
-        return summarizedTexts[0];
-    }
-    // recursive mapping: will split joined results as needed to stay
-    // within the allowed size limit of the splitter.
-    return summarizeRecursively({
-        text: join(summarizedTexts),
-        summarize,
-        split,
-        join,
-    }, options);
-}
-exports.summarizeRecursively = summarizeRecursively;

package/composed-function/summarize/summarizeRecursively.d.ts

@@ -1,11 +0,0 @@
-import { Run } from "../../core/Run.js";
-import { SplitFunction } from "../../text-chunk/split/SplitFunction.js";
-import { SummarizationFunction } from "./SummarizationFunction.js";
-export declare function summarizeRecursively({ summarize, split, join, text, }: {
-    summarize: SummarizationFunction;
-    split: SplitFunction;
-    join?: (texts: Array<string>) => string;
-    text: string;
-}, options?: {
-    run?: Run;
-}): Promise<string>;

package/composed-function/summarize/summarizeRecursively.js

@@ -1,15 +0,0 @@
-export async function summarizeRecursively({ summarize, split, join = (texts) => texts.join("\n\n"), text, }, options) {
-    const chunks = await split({ text });
-    const summarizedTexts = await Promise.all(chunks.map((chunk) => summarize({ text: chunk }, options)));
-    if (summarizedTexts.length === 1) {
-        return summarizedTexts[0];
-    }
-    // recursive mapping: will split joined results as needed to stay
-    // within the allowed size limit of the splitter.
-    return summarizeRecursively({
-        text: join(summarizedTexts),
-        summarize,
-        split,
-        join,
-    }, options);
-}

package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.cjs

@@ -1,25 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.summarizeRecursivelyWithTextGenerationAndTokenSplitting = void 0;
-const generateText_js_1 = require("../../model-function/generate-text/generateText.cjs");
-const splitRecursively_js_1 = require("../../text-chunk/split/splitRecursively.cjs");
-const summarizeRecursively_js_1 = require("./summarizeRecursively.cjs");
-/**
- * Recursively summarizes a text using a text generation model, e.g. for summarization or text extraction.
- * It automatically splits the text into optimal chunks that are small enough to be processed by the model,
- * while leaving enough space for the model to generate text.
- */
-async function summarizeRecursivelyWithTextGenerationAndTokenSplitting({ text, model, prompt, tokenLimit = model.contextWindowSize -
-    (model.settings.maxGenerationTokens ?? model.contextWindowSize / 4), join, }, options) {
-    const emptyPromptTokens = await model.countPromptTokens(await prompt({ text: "" }));
-    return (0, summarizeRecursively_js_1.summarizeRecursively)({
-        split: (0, splitRecursively_js_1.splitAtToken)({
-            tokenizer: model.tokenizer,
-            maxTokensPerChunk: tokenLimit - emptyPromptTokens,
-        }),
-        summarize: async (input) => (0, generateText_js_1.generateText)(model, await prompt(input), options),
-        join,
-        text,
-    }, options);
-}
-exports.summarizeRecursivelyWithTextGenerationAndTokenSplitting = summarizeRecursivelyWithTextGenerationAndTokenSplitting;

package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.d.ts

@@ -1,24 +0,0 @@
-import { Run } from "../../core/Run.js";
-import { TextGenerationModel } from "../../model-function/generate-text/TextGenerationModel.js";
-import { FullTokenizer } from "../../model-function/tokenize-text/Tokenizer.js";
-/**
- * Recursively summarizes a text using a text generation model, e.g. for summarization or text extraction.
- * It automatically splits the text into optimal chunks that are small enough to be processed by the model,
- * while leaving enough space for the model to generate text.
- */
-export declare function summarizeRecursivelyWithTextGenerationAndTokenSplitting<PROMPT>({ text, model, prompt, tokenLimit, join, }: {
-    text: string;
-    model: TextGenerationModel<PROMPT> & {
-        contextWindowSize: number;
-        tokenizer: FullTokenizer;
-        countPromptTokens: (prompt: PROMPT) => PromiseLike<number>;
-    };
-    prompt: (input: {
-        text: string;
-    }) => Promise<PROMPT>;
-    tokenLimit?: number;
-    join?: (texts: Array<string>) => string;
-}, options?: {
-    functionId?: string;
-    run?: Run;
-}): Promise<string>;

package/composed-function/summarize/summarizeRecursivelyWithTextGenerationAndTokenSplitting.js

@@ -1,21 +0,0 @@
-import { generateText } from "../../model-function/generate-text/generateText.js";
-import { splitAtToken } from "../../text-chunk/split/splitRecursively.js";
-import { summarizeRecursively } from "./summarizeRecursively.js";
-/**
- * Recursively summarizes a text using a text generation model, e.g. for summarization or text extraction.
- * It automatically splits the text into optimal chunks that are small enough to be processed by the model,
- * while leaving enough space for the model to generate text.
- */
-export async function summarizeRecursivelyWithTextGenerationAndTokenSplitting({ text, model, prompt, tokenLimit = model.contextWindowSize -
-    (model.settings.maxGenerationTokens ?? model.contextWindowSize / 4), join, }, options) {
-    const emptyPromptTokens = await model.countPromptTokens(await prompt({ text: "" }));
-    return summarizeRecursively({
-        split: splitAtToken({
-            tokenizer: model.tokenizer,
-            maxTokensPerChunk: tokenLimit - emptyPromptTokens,
-        }),
-        summarize: async (input) => generateText(model, await prompt(input), options),
-        join,
-        text,
-    }, options);
-}

package/cost/Cost.cjs

@@ -1,38 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.Cost = void 0;
-class Cost {
-    constructor({ costInMillicents, hasUnknownCost, callsWithUnknownCost, }) {
-        Object.defineProperty(this, "costInMillicents", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "hasUnknownCost", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "callsWithUnknownCost", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.costInMillicents = costInMillicents;
-        this.hasUnknownCost = hasUnknownCost;
-        this.callsWithUnknownCost = callsWithUnknownCost;
-    }
-    get costInCent() {
-        return this.costInMillicents / 1000;
-    }
-    get costInDollar() {
-        return this.costInCent / 100;
-    }
-    formatAsDollarAmount({ decimals = 2 } = {}) {
-        return `$${this.costInDollar.toFixed(decimals)}`;
-    }
-}
-exports.Cost = Cost;

package/cost/Cost.d.ts

@@ -1,16 +0,0 @@
-import { SuccessfulModelCall } from "../model-function/SuccessfulModelCall.js";
-export declare class Cost {
-    readonly costInMillicents: number;
-    readonly hasUnknownCost: boolean;
-    readonly callsWithUnknownCost: SuccessfulModelCall[];
-    constructor({ costInMillicents, hasUnknownCost, callsWithUnknownCost, }: {
-        costInMillicents: number;
-        hasUnknownCost: boolean;
-        callsWithUnknownCost: SuccessfulModelCall[];
-    });
-    get costInCent(): number;
-    get costInDollar(): number;
-    formatAsDollarAmount({ decimals }?: {
-        decimals?: number;
-    }): string;
-}

package/cost/Cost.js

@@ -1,34 +0,0 @@
-export class Cost {
-    constructor({ costInMillicents, hasUnknownCost, callsWithUnknownCost, }) {
-        Object.defineProperty(this, "costInMillicents", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "hasUnknownCost", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "callsWithUnknownCost", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.costInMillicents = costInMillicents;
-        this.hasUnknownCost = hasUnknownCost;
-        this.callsWithUnknownCost = callsWithUnknownCost;
-    }
-    get costInCent() {
-        return this.costInMillicents / 1000;
-    }
-    get costInDollar() {
-        return this.costInCent / 100;
-    }
-    formatAsDollarAmount({ decimals = 2 } = {}) {
-        return `$${this.costInDollar.toFixed(decimals)}`;
-    }
-}

package/cost/CostCalculator.d.ts

@@ -1,8 +0,0 @@
-import { SuccessfulModelCall } from "../model-function/SuccessfulModelCall.js";
-export interface CostCalculator {
-    readonly provider: string;
-    /**
-     * @return null if the cost is unknown, otherwise the cost in Millicents (0 if free)
-     */
-    calculateCostInMillicents(call: SuccessfulModelCall): PromiseLike<number | null>;
-}

package/cost/calculateCost.cjs

@@ -1,28 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.calculateCost = void 0;
-const Cost_js_1 = require("./Cost.cjs");
-async function calculateCost({ calls, costCalculators, }) {
-    let costInMillicents = 0;
-    const callsWithUnknownCost = [];
-    for (const call of calls) {
-        const model = call.model;
-        const providerCostCalculator = costCalculators.find((providerCostCalculator) => providerCostCalculator.provider === model.provider);
-        if (!providerCostCalculator) {
-            callsWithUnknownCost.push(call);
-            continue;
-        }
-        const cost = await providerCostCalculator.calculateCostInMillicents(call);
-        if (cost === null) {
-            callsWithUnknownCost.push(call);
-            continue;
-        }
-        costInMillicents += cost;
-    }
-    return new Cost_js_1.Cost({
-        costInMillicents,
-        hasUnknownCost: callsWithUnknownCost.length > 0,
-        callsWithUnknownCost,
-    });
-}
-exports.calculateCost = calculateCost;

package/cost/calculateCost.d.ts

@@ -1,7 +0,0 @@
-import { Cost } from "./Cost.js";
-import { CostCalculator } from "./CostCalculator.js";
-import { SuccessfulModelCall } from "../model-function/SuccessfulModelCall.js";
-export declare function calculateCost({ calls, costCalculators, }: {
-    calls: SuccessfulModelCall[];
-    costCalculators: CostCalculator[];
-}): Promise<Cost>;

package/cost/calculateCost.js

@@ -1,24 +0,0 @@
-import { Cost } from "./Cost.js";
-export async function calculateCost({ calls, costCalculators, }) {
-    let costInMillicents = 0;
-    const callsWithUnknownCost = [];
-    for (const call of calls) {
-        const model = call.model;
-        const providerCostCalculator = costCalculators.find((providerCostCalculator) => providerCostCalculator.provider === model.provider);
-        if (!providerCostCalculator) {
-            callsWithUnknownCost.push(call);
-            continue;
-        }
-        const cost = await providerCostCalculator.calculateCostInMillicents(call);
-        if (cost === null) {
-            callsWithUnknownCost.push(call);
-            continue;
-        }
-        costInMillicents += cost;
-    }
-    return new Cost({
-        costInMillicents,
-        hasUnknownCost: callsWithUnknownCost.length > 0,
-        callsWithUnknownCost,
-    });
-}

package/cost/index.cjs

@@ -1,19 +0,0 @@
-"use strict";
-var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    var desc = Object.getOwnPropertyDescriptor(m, k);
-    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
-      desc = { enumerable: true, get: function() { return m[k]; } };
-    }
-    Object.defineProperty(o, k2, desc);
-}) : (function(o, m, k, k2) {
-    if (k2 === undefined) k2 = k;
-    o[k2] = m[k];
-}));
-var __exportStar = (this && this.__exportStar) || function(m, exports) {
-    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
-};
-Object.defineProperty(exports, "__esModule", { value: true });
-__exportStar(require("./Cost.cjs"), exports);
-__exportStar(require("./calculateCost.cjs"), exports);
-__exportStar(require("./CostCalculator.cjs"), exports);

package/cost/index.d.ts

@@ -1,3 +0,0 @@
-export * from "./Cost.js";
-export * from "./calculateCost.js";
-export * from "./CostCalculator.js";

package/cost/index.js

@@ -1,3 +0,0 @@
-export * from "./Cost.js";
-export * from "./calculateCost.js";
-export * from "./CostCalculator.js";

package/guard/GuardEvent.cjs

@@ -1,2 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });

package/guard/GuardEvent.d.ts

@@ -1,7 +0,0 @@
-import { BaseFunctionFinishedEvent, BaseFunctionStartedEvent } from "../core/FunctionEvent.js";
-export interface GuardStartedEvent extends BaseFunctionStartedEvent {
-    functionType: "guard";
-}
-export interface GuardFinishedEvent extends BaseFunctionFinishedEvent {
-    functionType: "guard";
-}

package/guard/fixStructure.cjs

@@ -1,75 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.fixStructure = void 0;
-const StructureParseError_js_1 = require("../model-function/generate-structure/StructureParseError.cjs");
-const StructureValidationError_js_1 = require("../model-function/generate-structure/StructureValidationError.cjs");
-/**
- * Attempts to correct and retry structure generation when encountering parsing or validation errors.
- *
- * This function acts as a guard within the structure generation process. If the generation results in
- * an error, identified as either a `StructureParseError` or `StructureValidationError`, the function
- * triggers a retry mechanism. It uses the `modifyInputForRetry` method, provided via options, to adjust
- * the input parameters, aiming to resolve the issue that caused the initial error. The process continues
- * until a valid structure is generated, or it exhausts the predefined retry limits.
- *
- * @template INPUT - The expected format/type of the input data used for structure generation.
- * @template OUTPUT - The expected format/type of the output data, i.e., the successfully generated structure.
- *
- * @param {Object} options - Configuration options for modifying the input before retrying structure generation.
- * @param {function} options.modifyInputForRetry - A function that takes the error type, original input, and error object.
- *        It modifies the input data based on the error information, aiming to correct the issue for the retry attempt.
- *        This function must return a promise that resolves with the modified input.
- *
- * @returns {Guard<INPUT, OUTPUT>} A guard function that intercepts the structure generation process, checking for
- *          errors, and initiating retries by modifying the input parameters. The guard can trigger multiple retries
- *          if the issues persist. If the process succeeds, it returns the valid structure; otherwise, it returns
- *          undefined, indicating the exhaustion of retry attempts or a non-recoverable error.
- *
- * @example
- * const result = await guard(
- *  (input) =>
- *    generateStructure(
- *      openai
- *        .ChatTextGenerator(/* ... * /)
- *        .asFunctionCallStructureGenerationModel(/* ... * /),
- *
- *      new ZodSchema({
- *        // ...
- *      }),
- *
- *      input
- *    ),
- *  [
- *    // ...
- *  ],
- *  fixStructure({
- *    modifyInputForRetry: async ({ input, error }) => [
- *      ...input,
- *      openai.ChatMessage.assistant(null, {
- *        functionCall: {
- *          name: "sentiment",
- *          arguments: JSON.stringify(error.valueText),
- *        },
- *      }),
- *      openai.ChatMessage.user(error.message),
- *      openai.ChatMessage.user("Please fix the error and try again."),
- *    ],
- *  })
- * );
- */
-const fixStructure = ({ modifyInputForRetry }) => async (result) => {
-    if (result.type === "error" &&
-        (result.error instanceof StructureValidationError_js_1.StructureValidationError ||
-            result.error instanceof StructureParseError_js_1.StructureParseError)) {
-        return {
-            action: "retry",
-            input: await modifyInputForRetry({
-                type: "error",
-                input: result.input,
-                error: result.error,
-            }),
-        };
-    }
-    return undefined;
-};
-exports.fixStructure = fixStructure;

package/guard/fixStructure.d.ts

@@ -1,64 +0,0 @@
-import { StructureParseError } from "../model-function/generate-structure/StructureParseError.js";
-import { StructureValidationError } from "../model-function/generate-structure/StructureValidationError.js";
-import { Guard } from "./guard.js";
-/**
- * Attempts to correct and retry structure generation when encountering parsing or validation errors.
- *
- * This function acts as a guard within the structure generation process. If the generation results in
- * an error, identified as either a `StructureParseError` or `StructureValidationError`, the function
- * triggers a retry mechanism. It uses the `modifyInputForRetry` method, provided via options, to adjust
- * the input parameters, aiming to resolve the issue that caused the initial error. The process continues
- * until a valid structure is generated, or it exhausts the predefined retry limits.
- *
- * @template INPUT - The expected format/type of the input data used for structure generation.
- * @template OUTPUT - The expected format/type of the output data, i.e., the successfully generated structure.
- *
- * @param {Object} options - Configuration options for modifying the input before retrying structure generation.
- * @param {function} options.modifyInputForRetry - A function that takes the error type, original input, and error object.
- *        It modifies the input data based on the error information, aiming to correct the issue for the retry attempt.
- *        This function must return a promise that resolves with the modified input.
- *
- * @returns {Guard<INPUT, OUTPUT>} A guard function that intercepts the structure generation process, checking for
- *          errors, and initiating retries by modifying the input parameters. The guard can trigger multiple retries
- *          if the issues persist. If the process succeeds, it returns the valid structure; otherwise, it returns
- *          undefined, indicating the exhaustion of retry attempts or a non-recoverable error.
- *
- * @example
- * const result = await guard(
- *  (input) =>
- *    generateStructure(
- *      openai
- *        .ChatTextGenerator(/* ... * /)
- *        .asFunctionCallStructureGenerationModel(/* ... * /),
- *
- *      new ZodSchema({
- *        // ...
- *      }),
- *
- *      input
- *    ),
- *  [
- *    // ...
- *  ],
- *  fixStructure({
- *    modifyInputForRetry: async ({ input, error }) => [
- *      ...input,
- *      openai.ChatMessage.assistant(null, {
- *        functionCall: {
- *          name: "sentiment",
- *          arguments: JSON.stringify(error.valueText),
- *        },
- *      }),
- *      openai.ChatMessage.user(error.message),
- *      openai.ChatMessage.user("Please fix the error and try again."),
- *    ],
- *  })
- * );
- */
-export declare const fixStructure: <INPUT, OUTPUT>(options: {
-    modifyInputForRetry: (options: {
-        type: "error";
-        input: INPUT;
-        error: StructureValidationError | StructureParseError;
-    }) => PromiseLike<INPUT>;
-}) => Guard<INPUT, OUTPUT>;

package/guard/fixStructure.js

@@ -1,71 +0,0 @@
-import { StructureParseError } from "../model-function/generate-structure/StructureParseError.js";
-import { StructureValidationError } from "../model-function/generate-structure/StructureValidationError.js";
-/**
- * Attempts to correct and retry structure generation when encountering parsing or validation errors.
- *
- * This function acts as a guard within the structure generation process. If the generation results in
- * an error, identified as either a `StructureParseError` or `StructureValidationError`, the function
- * triggers a retry mechanism. It uses the `modifyInputForRetry` method, provided via options, to adjust
- * the input parameters, aiming to resolve the issue that caused the initial error. The process continues
- * until a valid structure is generated, or it exhausts the predefined retry limits.
- *
- * @template INPUT - The expected format/type of the input data used for structure generation.
- * @template OUTPUT - The expected format/type of the output data, i.e., the successfully generated structure.
- *
- * @param {Object} options - Configuration options for modifying the input before retrying structure generation.
- * @param {function} options.modifyInputForRetry - A function that takes the error type, original input, and error object.
- *        It modifies the input data based on the error information, aiming to correct the issue for the retry attempt.
- *        This function must return a promise that resolves with the modified input.
- *
- * @returns {Guard<INPUT, OUTPUT>} A guard function that intercepts the structure generation process, checking for
- *          errors, and initiating retries by modifying the input parameters. The guard can trigger multiple retries
- *          if the issues persist. If the process succeeds, it returns the valid structure; otherwise, it returns
- *          undefined, indicating the exhaustion of retry attempts or a non-recoverable error.
- *
- * @example
- * const result = await guard(
- *  (input) =>
- *    generateStructure(
- *      openai
- *        .ChatTextGenerator(/* ... * /)
- *        .asFunctionCallStructureGenerationModel(/* ... * /),
- *
- *      new ZodSchema({
- *        // ...
- *      }),
- *
- *      input
- *    ),
- *  [
- *    // ...
- *  ],
- *  fixStructure({
- *    modifyInputForRetry: async ({ input, error }) => [
- *      ...input,
- *      openai.ChatMessage.assistant(null, {
- *        functionCall: {
- *          name: "sentiment",
- *          arguments: JSON.stringify(error.valueText),
- *        },
- *      }),
- *      openai.ChatMessage.user(error.message),
- *      openai.ChatMessage.user("Please fix the error and try again."),
- *    ],
- *  })
- * );
- */
-export const fixStructure = ({ modifyInputForRetry }) => async (result) => {
-    if (result.type === "error" &&
-        (result.error instanceof StructureValidationError ||
-            result.error instanceof StructureParseError)) {
-        return {
-            action: "retry",
-            input: await modifyInputForRetry({
-                type: "error",
-                input: result.input,
-                error: result.error,
-            }),
-        };
-    }
-    return undefined;
-};