modelfusion 0.105.0 → 0.107.0

package/guard/guard.cjs

@@ -1,79 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.guard = void 0;
-const executeFunctionCall_js_1 = require("../core/executeFunctionCall.cjs");
-async function guard(execute, input, guards, options) {
-    const guardList = Array.isArray(guards) ? guards : [guards];
-    const maxAttempts = options?.maxAttempts ?? 2;
-    return (0, executeFunctionCall_js_1.executeFunctionCall)({
-        options,
-        input,
-        functionType: "guard",
-        execute: async (options) => {
-            let attempts = 0;
-            while (attempts < maxAttempts) {
-                let result;
-                try {
-                    result = {
-                        type: "value",
-                        input,
-                        output: await execute(input, options),
-                    };
-                }
-                catch (error) {
-                    result = {
-                        type: "error",
-                        input,
-                        error,
-                    };
-                }
-                let isValid = true;
-                for (const guard of guardList) {
-                    const guardResult = await guard(result);
-                    if (guardResult === undefined) {
-                        continue;
-                    }
-                    switch (guardResult.action) {
-                        case "passThrough": {
-                            break;
-                        }
-                        case "retry": {
-                            input = guardResult.input;
-                            isValid = false;
-                            break;
-                        }
-                        case "return": {
-                            result = {
-                                type: "value",
-                                input,
-                                output: guardResult.output,
-                            };
-                            break;
-                        }
-                        case "throwError": {
-                            result = {
-                                type: "error",
-                                input,
-                                error: guardResult.error,
-                            };
-                            break;
-                        }
-                    }
-                }
-                if (isValid) {
-                    if (result.type === "value") {
-                        return result.output;
-                    }
-                    else {
-                        throw result.error;
-                    }
-                }
-                attempts++;
-            }
-            // TODO dedicated error type
-            throw new Error(`Maximum attempts of ${maxAttempts} reached ` +
-                `without producing a valid output or handling an error.`);
-        },
-    });
-}
-exports.guard = guard;

package/guard/guard.d.ts

@@ -1,29 +0,0 @@
-import { FunctionOptions } from "../core/FunctionOptions.js";
-type OutputResult<INPUT, OUTPUT> = {
-    type: "value";
-    input: INPUT;
-    output: OUTPUT;
-    error?: undefined;
-} | {
-    type: "error";
-    input: INPUT;
-    output?: undefined;
-    error: unknown;
-};
-export type OutputValidator<INPUT, OUTPUT> = ({ type, input, output, error, }: OutputResult<INPUT, OUTPUT>) => PromiseLike<boolean>;
-export type Guard<INPUT, OUTPUT> = ({ type, input, output, error, }: OutputResult<INPUT, OUTPUT>) => PromiseLike<{
-    action: "retry";
-    input: INPUT;
-} | {
-    action: "return";
-    output: OUTPUT;
-} | {
-    action: "throwError";
-    error: unknown;
-} | {
-    action: "passThrough";
-} | undefined>;
-export declare function guard<INPUT, OUTPUT>(execute: (input: INPUT, options?: FunctionOptions) => PromiseLike<OUTPUT>, input: INPUT, guards: Guard<INPUT, OUTPUT> | Array<Guard<INPUT, OUTPUT>>, options?: FunctionOptions & {
-    maxAttempts: number;
-}): Promise<OUTPUT | undefined>;
-export {};

package/guard/guard.js

@@ -1,75 +0,0 @@
-import { executeFunctionCall } from "../core/executeFunctionCall.js";
-export async function guard(execute, input, guards, options) {
-    const guardList = Array.isArray(guards) ? guards : [guards];
-    const maxAttempts = options?.maxAttempts ?? 2;
-    return executeFunctionCall({
-        options,
-        input,
-        functionType: "guard",
-        execute: async (options) => {
-            let attempts = 0;
-            while (attempts < maxAttempts) {
-                let result;
-                try {
-                    result = {
-                        type: "value",
-                        input,
-                        output: await execute(input, options),
-                    };
-                }
-                catch (error) {
-                    result = {
-                        type: "error",
-                        input,
-                        error,
-                    };
-                }
-                let isValid = true;
-                for (const guard of guardList) {
-                    const guardResult = await guard(result);
-                    if (guardResult === undefined) {
-                        continue;
-                    }
-                    switch (guardResult.action) {
-                        case "passThrough": {
-                            break;
-                        }
-                        case "retry": {
-                            input = guardResult.input;
-                            isValid = false;
-                            break;
-                        }
-                        case "return": {
-                            result = {
-                                type: "value",
-                                input,
-                                output: guardResult.output,
-                            };
-                            break;
-                        }
-                        case "throwError": {
-                            result = {
-                                type: "error",
-                                input,
-                                error: guardResult.error,
-                            };
-                            break;
-                        }
-                    }
-                }
-                if (isValid) {
-                    if (result.type === "value") {
-                        return result.output;
-                    }
-                    else {
-                        throw result.error;
-                    }
-                }
-                attempts++;
-            }
-            // TODO dedicated error type
-            throw new Error(`Maximum attempts of ${maxAttempts} reached ` +
-                `without producing a valid output or handling an error.`);
-        },
-    });
-}

package/guard/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("./GuardEvent.cjs"), exports);
-__exportStar(require("./fixStructure.cjs"), exports);
-__exportStar(require("./guard.cjs"), exports);

package/guard/index.d.ts

@@ -1,3 +0,0 @@
-export * from "./GuardEvent.js";
-export * from "./fixStructure.js";
-export * from "./guard.js";

package/guard/index.js

@@ -1,3 +0,0 @@
-export * from "./GuardEvent.js";
-export * from "./fixStructure.js";
-export * from "./guard.js";

package/model-function/SuccessfulModelCall.cjs

@@ -1,10 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.extractSuccessfulModelCalls = void 0;
-function extractSuccessfulModelCalls(runFunctionEvents) {
-    return runFunctionEvents.filter((event) => "model" in event &&
-        "result" in event &&
-        "status" in event.result &&
-        event.result.status === "success");
-}
-exports.extractSuccessfulModelCalls = extractSuccessfulModelCalls;

package/model-function/SuccessfulModelCall.d.ts

@@ -1,12 +0,0 @@
-import { FunctionEvent } from "../core/FunctionEvent.js";
-import { ModelCallFinishedEvent } from "./ModelCallEvent.js";
-export type SuccessfulModelCall = ModelCallFinishedEvent & {
-    result: {
-        status: "success";
-    };
-};
-export declare function extractSuccessfulModelCalls(runFunctionEvents: FunctionEvent[]): (ModelCallFinishedEvent & {
-    result: {
-        status: "success";
-    };
-})[];

package/model-function/SuccessfulModelCall.js

@@ -1,6 +0,0 @@
-export function extractSuccessfulModelCalls(runFunctionEvents) {
-    return runFunctionEvents.filter((event) => "model" in event &&
-        "result" in event &&
-        "status" in event.result &&
-        event.result.status === "success");
-}

package/model-provider/openai/OpenAICostCalculator.cjs

@@ -1,89 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.OpenAICostCalculator = void 0;
-const OpenAICompletionModel_js_1 = require("./OpenAICompletionModel.cjs");
-const OpenAIImageGenerationModel_js_1 = require("./OpenAIImageGenerationModel.cjs");
-const OpenAISpeechModel_js_1 = require("./OpenAISpeechModel.cjs");
-const OpenAITextEmbeddingModel_js_1 = require("./OpenAITextEmbeddingModel.cjs");
-const OpenAITranscriptionModel_js_1 = require("./OpenAITranscriptionModel.cjs");
-const OpenAIChatModel_js_1 = require("./OpenAIChatModel.cjs");
-class OpenAICostCalculator {
-    constructor() {
-        Object.defineProperty(this, "provider", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "openai"
-        });
-    }
-    async calculateCostInMillicents(call) {
-        const type = call.functionType;
-        const model = call.model.modelName;
-        switch (type) {
-            case "generate-image": {
-                if (model == null) {
-                    return null;
-                }
-                return (0, OpenAIImageGenerationModel_js_1.calculateOpenAIImageGenerationCostInMillicents)({
-                    model: model,
-                    settings: call.settings,
-                });
-            }
-            case "embed": {
-                if (model == null) {
-                    return null;
-                }
-                if ((0, OpenAITextEmbeddingModel_js_1.isOpenAIEmbeddingModel)(model)) {
-                    const responses = Array.isArray(call.result.response)
-                        ? call.result.response
-                        : [call.result.response];
-                    return (0, OpenAITextEmbeddingModel_js_1.calculateOpenAIEmbeddingCostInMillicents)({
-                        model,
-                        responses,
-                    });
-                }
-                break;
-            }
-            case "generate-structure":
-            case "generate-text": {
-                if (model == null) {
-                    return null;
-                }
-                if ((0, OpenAIChatModel_js_1.isOpenAIChatModel)(model)) {
-                    return (0, OpenAIChatModel_js_1.calculateOpenAIChatCostInMillicents)({
-                        model,
-                        response: call.result.response,
-                    });
-                }
-                if ((0, OpenAICompletionModel_js_1.isOpenAICompletionModel)(model)) {
-                    return (0, OpenAICompletionModel_js_1.calculateOpenAICompletionCostInMillicents)({
-                        model,
-                        response: call.result.response,
-                    });
-                }
-                break;
-            }
-            case "generate-transcription": {
-                if (model == null) {
-                    return null;
-                }
-                return (0, OpenAITranscriptionModel_js_1.calculateOpenAITranscriptionCostInMillicents)({
-                    model: model,
-                    response: call.result
-                        .response,
-                });
-            }
-            case "generate-speech": {
-                if (model == null) {
-                    return null;
-                }
-                return (0, OpenAISpeechModel_js_1.calculateOpenAISpeechCostInMillicents)({
-                    model: model,
-                    input: call.input,
-                });
-            }
-        }
-        return null;
-    }
-}
-exports.OpenAICostCalculator = OpenAICostCalculator;

package/model-provider/openai/OpenAICostCalculator.d.ts

@@ -1,6 +0,0 @@
-import { CostCalculator } from "../../cost/CostCalculator.js";
-import { SuccessfulModelCall } from "../../model-function/SuccessfulModelCall.js";
-export declare class OpenAICostCalculator implements CostCalculator {
-    readonly provider = "openai";
-    calculateCostInMillicents(call: SuccessfulModelCall): Promise<number | null>;
-}

package/model-provider/openai/OpenAICostCalculator.js

@@ -1,85 +0,0 @@
-import { calculateOpenAICompletionCostInMillicents, isOpenAICompletionModel, } from "./OpenAICompletionModel.js";
-import { calculateOpenAIImageGenerationCostInMillicents, } from "./OpenAIImageGenerationModel.js";
-import { calculateOpenAISpeechCostInMillicents, } from "./OpenAISpeechModel.js";
-import { calculateOpenAIEmbeddingCostInMillicents, isOpenAIEmbeddingModel, } from "./OpenAITextEmbeddingModel.js";
-import { calculateOpenAITranscriptionCostInMillicents, } from "./OpenAITranscriptionModel.js";
-import { calculateOpenAIChatCostInMillicents, isOpenAIChatModel, } from "./OpenAIChatModel.js";
-export class OpenAICostCalculator {
-    constructor() {
-        Object.defineProperty(this, "provider", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "openai"
-        });
-    }
-    async calculateCostInMillicents(call) {
-        const type = call.functionType;
-        const model = call.model.modelName;
-        switch (type) {
-            case "generate-image": {
-                if (model == null) {
-                    return null;
-                }
-                return calculateOpenAIImageGenerationCostInMillicents({
-                    model: model,
-                    settings: call.settings,
-                });
-            }
-            case "embed": {
-                if (model == null) {
-                    return null;
-                }
-                if (isOpenAIEmbeddingModel(model)) {
-                    const responses = Array.isArray(call.result.response)
-                        ? call.result.response
-                        : [call.result.response];
-                    return calculateOpenAIEmbeddingCostInMillicents({
-                        model,
-                        responses,
-                    });
-                }
-                break;
-            }
-            case "generate-structure":
-            case "generate-text": {
-                if (model == null) {
-                    return null;
-                }
-                if (isOpenAIChatModel(model)) {
-                    return calculateOpenAIChatCostInMillicents({
-                        model,
-                        response: call.result.response,
-                    });
-                }
-                if (isOpenAICompletionModel(model)) {
-                    return calculateOpenAICompletionCostInMillicents({
-                        model,
-                        response: call.result.response,
-                    });
-                }
-                break;
-            }
-            case "generate-transcription": {
-                if (model == null) {
-                    return null;
-                }
-                return calculateOpenAITranscriptionCostInMillicents({
-                    model: model,
-                    response: call.result
-                        .response,
-                });
-            }
-            case "generate-speech": {
-                if (model == null) {
-                    return null;
-                }
-                return calculateOpenAISpeechCostInMillicents({
-                    model: model,
-                    input: call.input,
-                });
-            }
-        }
-        return null;
-    }
-}

package/server/fastify/AssetStorage.cjs

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

package/server/fastify/AssetStorage.d.ts

@@ -1,17 +0,0 @@
-/// <reference types="node" />
-import { FlowRun } from "./FlowRun.js";
-export type Asset = {
-    data: Buffer;
-    contentType: string;
-    name: string;
-};
-export interface AssetStorage {
-    storeAsset(options: {
-        run: FlowRun<unknown>;
-        asset: Asset;
-    }): Promise<void>;
-    readAsset(options: {
-        run: FlowRun<unknown>;
-        assetName: string;
-    }): Promise<Asset | null>;
-}

package/server/fastify/AssetStorage.js

@@ -1 +0,0 @@
-export {};

package/server/fastify/DefaultFlow.cjs

@@ -1,22 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.DefaultFlow = void 0;
-class DefaultFlow {
-    constructor({ schema, process, }) {
-        Object.defineProperty(this, "schema", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "process", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.schema = schema;
-        this.process = process;
-    }
-}
-exports.DefaultFlow = DefaultFlow;

package/server/fastify/DefaultFlow.d.ts

@@ -1,16 +0,0 @@
-import { FlowRun } from "./FlowRun.js";
-import { FlowSchema } from "./FlowSchema.js";
-export declare class DefaultFlow<INPUT, EVENT> {
-    readonly schema: FlowSchema<INPUT, EVENT>;
-    constructor({ schema, process, }: {
-        schema: FlowSchema<INPUT, EVENT>;
-        process: (options: {
-            input: INPUT;
-            run: FlowRun<EVENT>;
-        }) => Promise<void>;
-    });
-    process: (options: {
-        input: INPUT;
-        run: FlowRun<EVENT>;
-    }) => Promise<void>;
-}

package/server/fastify/DefaultFlow.js

@@ -1,18 +0,0 @@
-export class DefaultFlow {
-    constructor({ schema, process, }) {
-        Object.defineProperty(this, "schema", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "process", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.schema = schema;
-        this.process = process;
-    }
-}

package/server/fastify/FileSystemAssetStorage.cjs

@@ -1,60 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.FileSystemAssetStorage = void 0;
-const node_fs_1 = require("node:fs");
-const node_path_1 = require("node:path");
-class FileSystemAssetStorage {
-    constructor({ path, logger, }) {
-        Object.defineProperty(this, "path", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "logger", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.path = path;
-        this.logger = logger;
-    }
-    async storeAsset({ run, asset, }) {
-        try {
-            const assetPath = this.path(run);
-            await node_fs_1.promises.mkdir(assetPath, { recursive: true });
-            await node_fs_1.promises.writeFile((0, node_path_1.join)(assetPath, asset.name), asset.data);
-            await node_fs_1.promises.writeFile((0, node_path_1.join)(assetPath, `${asset.name}.meta.json`), JSON.stringify({
-                name: asset.name,
-                contentType: asset.contentType,
-            }));
-        }
-        catch (error) {
-            this.logger.logError({
-                run,
-                message: `Failed to store asset ${asset.name}`,
-                error,
-            });
-            throw error;
-        }
-    }
-    async readAsset(options) {
-        try {
-            const assetPath = this.path(options.run);
-            const data = await node_fs_1.promises.readFile((0, node_path_1.join)(assetPath, options.assetName));
-            const meta = await node_fs_1.promises.readFile((0, node_path_1.join)(assetPath, `${options.assetName}.meta.json`));
-            const { name, contentType } = JSON.parse(meta.toString());
-            return { data, name, contentType };
-        }
-        catch (error) {
-            this.logger.logError({
-                run: options.run,
-                message: `Failed to read asset ${options.assetName}`,
-                error,
-            });
-            throw error;
-        }
-    }
-}
-exports.FileSystemAssetStorage = FileSystemAssetStorage;

package/server/fastify/FileSystemAssetStorage.d.ts

@@ -1,19 +0,0 @@
-import type { Asset, AssetStorage } from "./AssetStorage.js";
-import { FlowRun } from "./FlowRun.js";
-import { Logger } from "./Logger.js";
-export declare class FileSystemAssetStorage implements AssetStorage {
-    private readonly path;
-    private readonly logger;
-    constructor({ path, logger, }: {
-        path: (run: FlowRun<unknown>) => string;
-        logger: Logger;
-    });
-    storeAsset({ run, asset, }: {
-        run: FlowRun<unknown>;
-        asset: Asset;
-    }): Promise<void>;
-    readAsset(options: {
-        run: FlowRun<unknown>;
-        assetName: string;
-    }): Promise<Asset | null>;
-}

package/server/fastify/FileSystemAssetStorage.js

@@ -1,56 +0,0 @@
-import { promises as fs } from "node:fs";
-import { join } from "node:path";
-export class FileSystemAssetStorage {
-    constructor({ path, logger, }) {
-        Object.defineProperty(this, "path", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "logger", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        this.path = path;
-        this.logger = logger;
-    }
-    async storeAsset({ run, asset, }) {
-        try {
-            const assetPath = this.path(run);
-            await fs.mkdir(assetPath, { recursive: true });
-            await fs.writeFile(join(assetPath, asset.name), asset.data);
-            await fs.writeFile(join(assetPath, `${asset.name}.meta.json`), JSON.stringify({
-                name: asset.name,
-                contentType: asset.contentType,
-            }));
-        }
-        catch (error) {
-            this.logger.logError({
-                run,
-                message: `Failed to store asset ${asset.name}`,
-                error,
-            });
-            throw error;
-        }
-    }
-    async readAsset(options) {
-        try {
-            const assetPath = this.path(options.run);
-            const data = await fs.readFile(join(assetPath, options.assetName));
-            const meta = await fs.readFile(join(assetPath, `${options.assetName}.meta.json`));
-            const { name, contentType } = JSON.parse(meta.toString());
-            return { data, name, contentType };
-        }
-        catch (error) {
-            this.logger.logError({
-                run: options.run,
-                message: `Failed to read asset ${options.assetName}`,
-                error,
-            });
-            throw error;
-        }
-    }
-}