@aigne/core 1.72.0 → 1.74.0-beta

package/lib/cjs/agents/user-agent.js

@@ -1,90 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.UserAgent = void 0;
-const message_queue_js_1 = require("../aigne/message-queue.js");
-const type_utils_js_1 = require("../utils/type-utils.js");
-const agent_js_1 = require("./agent.js");
-class UserAgent extends agent_js_1.Agent {
-    tag = "UserAgent";
-    static from(options) {
-        return new UserAgent(options);
-    }
-    constructor(options) {
-        super({ ...options, disableEvents: true });
-        this._process = options.process;
-        this.context = options.context;
-        this.activeAgent = options.activeAgent;
-    }
-    context;
-    _process;
-    activeAgent;
-    subscribeToTopics(context) {
-        if (this._process)
-            super.subscribeToTopics(context);
-    }
-    async publishToTopics(output, options) {
-        if (this._process)
-            super.publishToTopics(output, options);
-    }
-    invoke = ((input, options = {}) => {
-        options.context ??= this.context.newContext({ reset: true });
-        return super.invoke(input, options);
-    });
-    async process(input, options) {
-        if (this._process) {
-            return this._process.apply(this, [input, options]);
-        }
-        if (this.activeAgent) {
-            const [output, agent] = await this.invokeChildAgent(this.activeAgent, input, {
-                ...options,
-                returnActiveAgent: true,
-                streaming: true,
-                // Do not create a new context for the nested agent invocation,
-                // We are resetting the context in the override invoke method
-                newContext: false,
-            });
-            agent.then((agent) => {
-                this.activeAgent = agent;
-            });
-            return output;
-        }
-        const publicTopic = typeof this.publishTopic === "function" ? await this.publishTopic(input) : this.publishTopic;
-        if (publicTopic?.length) {
-            options.context.publish(publicTopic, input, { newContext: false });
-            if (this.subscribeTopic) {
-                return this.subscribe(this.subscribeTopic).then((res) => res.message);
-            }
-            return {};
-        }
-        throw new Error("UserAgent must have a process function or a publishTopic");
-    }
-    publish = ((topic, payload) => {
-        return this.context.publish(topic, (0, message_queue_js_1.toMessagePayload)(payload, { role: "user", source: this.name }));
-    });
-    subscribe = ((...args) => {
-        return this.context.subscribe(...args);
-    });
-    unsubscribe = ((...args) => {
-        this.context.unsubscribe(...args);
-    });
-    get stream() {
-        let subscriptions = [];
-        return new ReadableStream({
-            start: (controller) => {
-                const subscribeTopic = (0, type_utils_js_1.flat)(this.subscribeTopic);
-                subscriptions = subscribeTopic.map((topic) => this.subscribe(topic, (message) => {
-                    controller.enqueue({ ...message, topic });
-                }));
-            },
-            cancel: () => {
-                for (const unsubscribe of subscriptions) {
-                    unsubscribe();
-                }
-            },
-        });
-    }
-    checkAgentInvokesUsage(_options) {
-        // ignore calls usage check for UserAgent
-    }
-}
-exports.UserAgent = UserAgent;

package/lib/cjs/agents/video-model.d.ts

@@ -1,280 +0,0 @@
-import { type ZodType, z } from "zod";
-import type { PromiseOrValue } from "../utils/type-utils.js";
-import type { AgentInvokeOptions, AgentOptions, AgentProcessResult, AgentResponse, AgentResponseStream, GetterSchema, Message } from "./agent.js";
-import { type ChatModelOutputUsage } from "./chat-model.js";
-import { type FileType, type FileUnionContent, Model } from "./model.js";
-export interface VideoModelOptions<I extends VideoModelInput = VideoModelInput, O extends VideoModelOutput = VideoModelOutput> extends Omit<AgentOptions<I, O>, "model"> {
-    model?: string;
-    modelOptions?: VideoModelInputOptionsWithGetter;
-}
-export declare abstract class VideoModel<I extends VideoModelInput = VideoModelInput, O extends VideoModelOutput = VideoModelOutput> extends Model<I, O> {
-    options?: VideoModelOptions<I, O> | undefined;
-    tag: string;
-    constructor(options?: VideoModelOptions<I, O> | undefined);
-    get credential(): PromiseOrValue<{
-        url?: string;
-        apiKey?: string;
-        model?: string;
-    }>;
-    protected preprocess(input: I, options: AgentInvokeOptions): Promise<void>;
-    protected postprocess(input: I, output: O, options: AgentInvokeOptions): Promise<void>;
-    abstract process(input: I, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<O>>;
-    protected processAgentOutput(input: I, output: Exclude<AgentResponse<O>, AgentResponseStream<O>>, options: AgentInvokeOptions): Promise<O>;
-}
-export interface VideoModelInput extends Message {
-    prompt: string;
-    model?: string;
-    size?: string;
-    seconds?: string;
-    outputFileType?: FileType;
-    modelOptions?: VideoModelInputOptions;
-    image?: FileUnionContent;
-}
-export interface VideoModelInputOptions extends Record<string, unknown> {
-    model?: string;
-}
-export type VideoModelInputOptionsWithGetter = GetterSchema<VideoModelInputOptions>;
-export declare const videoModelInputSchema: z.ZodObject<{
-    prompt: z.ZodString;
-    model: z.ZodOptional<z.ZodString>;
-    size: z.ZodOptional<z.ZodString>;
-    seconds: z.ZodOptional<z.ZodString>;
-    outputFileType: z.ZodOptional<z.ZodEnum<["local", "file", "url"]>>;
-    modelOptions: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-    image: z.ZodOptional<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
-        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    } & {
-        type: z.ZodLiteral<"local">;
-        path: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }>, z.ZodObject<{
-        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    } & {
-        type: z.ZodLiteral<"url">;
-        url: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }>, z.ZodObject<{
-        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    } & {
-        type: z.ZodLiteral<"file">;
-        data: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }>]>>;
-}, "strip", z.ZodTypeAny, {
-    prompt: string;
-    model?: string | undefined;
-    modelOptions?: Record<string, unknown> | undefined;
-    image?: {
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
-    size?: string | undefined;
-    seconds?: string | undefined;
-}, {
-    prompt: string;
-    model?: string | undefined;
-    modelOptions?: Record<string, unknown> | undefined;
-    image?: {
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
-    size?: string | undefined;
-    seconds?: string | undefined;
-}>;
-export interface VideoModelOutput extends Message {
-    videos: FileUnionContent[];
-    /**
-     * Token usage statistics
-     */
-    usage?: ChatModelOutputUsage;
-    /**
-     * Model name or version used
-     */
-    model?: string;
-    seconds?: number;
-}
-export declare const videoModelOutputSchema: z.ZodObject<{
-    videos: z.ZodArray<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
-        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    } & {
-        type: z.ZodLiteral<"local">;
-        path: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }>, z.ZodObject<{
-        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    } & {
-        type: z.ZodLiteral<"url">;
-        url: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }>, z.ZodObject<{
-        filename: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        mimeType: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-    } & {
-        type: z.ZodLiteral<"file">;
-        data: z.ZodString;
-    }, "strip", z.ZodTypeAny, {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }, {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    }>]>, "many">;
-    usage: z.ZodOptional<z.ZodObject<{
-        inputTokens: z.ZodNumber;
-        outputTokens: z.ZodNumber;
-        aigneHubCredits: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        cacheCreationInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        cacheReadInputTokens: ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
-        creditPrefix: ZodType<"$" | "€" | "¥" | undefined, z.ZodTypeDef, "$" | "€" | "¥" | undefined>;
-    }, "strip", z.ZodTypeAny, {
-        inputTokens: number;
-        outputTokens: number;
-        aigneHubCredits?: number | undefined;
-        cacheCreationInputTokens?: number | undefined;
-        cacheReadInputTokens?: number | undefined;
-        creditPrefix?: "$" | "€" | "¥" | undefined;
-    }, {
-        inputTokens: number;
-        outputTokens: number;
-        aigneHubCredits?: number | undefined;
-        cacheCreationInputTokens?: number | undefined;
-        cacheReadInputTokens?: number | undefined;
-        creditPrefix?: "$" | "€" | "¥" | undefined;
-    }>>;
-    model: z.ZodOptional<z.ZodString>;
-    seconds: z.ZodOptional<z.ZodNumber>;
-}, "strip", z.ZodTypeAny, {
-    videos: ({
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    })[];
-    model?: string | undefined;
-    usage?: {
-        inputTokens: number;
-        outputTokens: number;
-        aigneHubCredits?: number | undefined;
-        cacheCreationInputTokens?: number | undefined;
-        cacheReadInputTokens?: number | undefined;
-        creditPrefix?: "$" | "€" | "¥" | undefined;
-    } | undefined;
-    seconds?: number | undefined;
-}, {
-    videos: ({
-        type: "url";
-        url: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        type: "file";
-        data: string;
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    } | {
-        path: string;
-        type: "local";
-        filename?: string | undefined;
-        mimeType?: string | undefined;
-    })[];
-    model?: string | undefined;
-    usage?: {
-        inputTokens: number;
-        outputTokens: number;
-        aigneHubCredits?: number | undefined;
-        cacheCreationInputTokens?: number | undefined;
-        cacheReadInputTokens?: number | undefined;
-        creditPrefix?: "$" | "€" | "¥" | undefined;
-    } | undefined;
-    seconds?: number | undefined;
-}>;

package/lib/cjs/agents/video-model.js

@@ -1,67 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.videoModelOutputSchema = exports.videoModelInputSchema = exports.VideoModel = void 0;
-const zod_1 = require("zod");
-const chat_model_js_1 = require("./chat-model.js");
-const model_js_1 = require("./model.js");
-class VideoModel extends model_js_1.Model {
-    options;
-    tag = "VideoModelAgent";
-    constructor(options) {
-        super({
-            inputSchema: exports.videoModelInputSchema,
-            outputSchema: exports.videoModelOutputSchema,
-            ...options,
-        });
-        this.options = options;
-    }
-    get credential() {
-        return {};
-    }
-    async preprocess(input, options) {
-        super.preprocess(input, options);
-        const { limits, usage } = options.context;
-        const usedTokens = usage.outputTokens + usage.inputTokens;
-        if (limits?.maxTokens && usedTokens >= limits.maxTokens) {
-            throw new Error(`Exceeded max tokens ${usedTokens}/${limits.maxTokens}`);
-        }
-    }
-    async postprocess(input, output, options) {
-        super.postprocess(input, output, options);
-        const { usage } = output;
-        if (usage) {
-            options.context.usage.outputTokens += usage.outputTokens;
-            options.context.usage.inputTokens += usage.inputTokens;
-            if (usage.aigneHubCredits)
-                options.context.usage.aigneHubCredits += usage.aigneHubCredits;
-            if (usage.creditPrefix)
-                options.context.usage.creditPrefix = usage.creditPrefix;
-        }
-    }
-    async processAgentOutput(input, output, options) {
-        if (output.videos) {
-            const videos = zod_1.z.array(model_js_1.fileUnionContentSchema).parse(output.videos);
-            output = {
-                ...output,
-                videos: await Promise.all(videos.map((video) => this.transformFileType(input.outputFileType, video))),
-            };
-        }
-        return super.processAgentOutput(input, output, options);
-    }
-}
-exports.VideoModel = VideoModel;
-exports.videoModelInputSchema = zod_1.z.object({
-    prompt: zod_1.z.string().describe("Text prompt describing the video to generate"),
-    model: zod_1.z.string().optional().describe("Model to use for video generation"),
-    size: zod_1.z.string().optional().describe("Size/resolution of the video"),
-    seconds: zod_1.z.string().optional().describe("Duration of the video in seconds"),
-    outputFileType: model_js_1.fileTypeSchema.optional(),
-    modelOptions: zod_1.z.record(zod_1.z.unknown()).optional(),
-    image: model_js_1.fileUnionContentSchema.optional(),
-});
-exports.videoModelOutputSchema = zod_1.z.object({
-    videos: zod_1.z.array(model_js_1.fileUnionContentSchema),
-    usage: chat_model_js_1.chatModelOutputUsageSchema.optional(),
-    model: zod_1.z.string().optional(),
-    seconds: zod_1.z.number().optional(),
-});