@aigne/core 1.73.0-beta → 1.74.0-beta

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

@@ -1,270 +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 ImageModelOptions<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Omit<AgentOptions<I, O>, "model"> {
-    model?: string;
-    modelOptions?: ImageModelInputOptionsWithGetter;
-}
-export declare abstract class ImageModel<I extends ImageModelInput = ImageModelInput, O extends ImageModelOutput = ImageModelOutput> extends Model<I, O> {
-    options?: ImageModelOptions<I, O> | undefined;
-    tag: string;
-    constructor(options?: ImageModelOptions<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 type ImageModelInputImage = FileUnionContent[];
-export interface ImageModelInput extends Message {
-    prompt: string;
-    image?: ImageModelInputImage;
-    n?: number;
-    outputFileType?: FileType;
-    modelOptions?: ImageModelInputOptions;
-}
-export interface ImageModelInputOptions extends Record<string, unknown> {
-    model?: string;
-    preferInputFileType?: "file" | "url";
-}
-export type ImageModelInputOptionsWithGetter = GetterSchema<ImageModelInputOptions>;
-export declare const imageModelInputSchema: z.ZodObject<{
-    prompt: z.ZodString;
-    image: z.ZodOptional<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">>;
-    n: z.ZodOptional<z.ZodNumber>;
-    outputFileType: z.ZodOptional<z.ZodEnum<["local", "file", "url"]>>;
-    modelOptions: z.ZodOptional<z.ZodRecord<z.ZodString, z.ZodUnknown>>;
-}, "strip", z.ZodTypeAny, {
-    prompt: string;
-    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;
-    n?: number | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
-}, {
-    prompt: string;
-    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;
-    n?: number | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
-}>;
-export interface ImageModelOutput extends Message {
-    images: FileUnionContent[];
-    /**
-     * Token usage statistics
-     */
-    usage?: ChatModelOutputUsage;
-    /**
-     * Model name or version used
-     */
-    model?: string;
-}
-export declare const imageModelOutputSchema: z.ZodObject<{
-    images: 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>;
-}, "strip", z.ZodTypeAny, {
-    images: ({
-        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;
-}, {
-    images: ({
-        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;
-}>;

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

@@ -1,91 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.imageModelOutputSchema = exports.imageModelInputSchema = exports.ImageModel = void 0;
-const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
-const zod_1 = require("zod");
-const chat_model_js_1 = require("./chat-model.js");
-const model_js_1 = require("./model.js");
-class ImageModel extends model_js_1.Model {
-    options;
-    tag = "ImageModelAgent";
-    constructor(options) {
-        super({
-            inputSchema: exports.imageModelInputSchema,
-            outputSchema: exports.imageModelOutputSchema,
-            ...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}`);
-        }
-        if (input.image) {
-            input.image = await Promise.all(input.image.map(async (item) => {
-                if (item.type === "local") {
-                    return {
-                        ...item,
-                        type: "file",
-                        data: await index_js_1.nodejs.fs.readFile(item.path, "base64"),
-                        path: undefined,
-                        mimeType: item.mimeType || (await ImageModel.getMimeType(item.filename || item.path)),
-                    };
-                }
-                if ((input.modelOptions?.preferInputFileType ||
-                    this.options?.modelOptions?.preferInputFileType) !== "url") {
-                    if (item.type === "url") {
-                        return {
-                            ...item,
-                            type: "file",
-                            data: Buffer.from(await (await this.downloadFile(item.url)).arrayBuffer()).toString("base64"),
-                            url: undefined,
-                            mimeType: item.mimeType || (await ImageModel.getMimeType(item.filename || item.url)),
-                        };
-                    }
-                }
-                return item;
-            }));
-        }
-    }
-    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.images) {
-            const images = zod_1.z.array(model_js_1.fileUnionContentSchema).parse(output.images);
-            output = {
-                ...output,
-                images: await Promise.all(images.map((image) => this.transformFileType(input.outputFileType, image))),
-            };
-        }
-        return super.processAgentOutput(input, output, options);
-    }
-}
-exports.ImageModel = ImageModel;
-exports.imageModelInputSchema = zod_1.z.object({
-    prompt: zod_1.z.string(),
-    image: zod_1.z.array(model_js_1.fileUnionContentSchema).optional().describe("Images used for editing"),
-    n: zod_1.z.number().int().min(1).optional(),
-    outputFileType: model_js_1.fileTypeSchema.optional(),
-    modelOptions: zod_1.z.record(zod_1.z.unknown()).optional(),
-});
-exports.imageModelOutputSchema = zod_1.z.object({
-    images: zod_1.z.array(model_js_1.fileUnionContentSchema),
-    usage: chat_model_js_1.chatModelOutputUsageSchema.optional(),
-    model: zod_1.z.string().optional(),
-});

package/lib/cjs/agents/mcp-agent.d.ts

@@ -1,231 +0,0 @@
-import { Client, type ClientOptions } from "@modelcontextprotocol/sdk/client/index.js";
-import { type SSEClientTransportOptions } from "@modelcontextprotocol/sdk/client/sse.js";
-import type { StdioServerParameters } from "@modelcontextprotocol/sdk/client/stdio.js";
-import { type StreamableHTTPClientTransportOptions } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
-import type { RequestOptions } from "@modelcontextprotocol/sdk/shared/protocol.js";
-import type { Transport } from "@modelcontextprotocol/sdk/shared/transport.js";
-import type { CallToolResult, GetPromptResult, Implementation, ReadResourceResult, Request } from "@modelcontextprotocol/sdk/types.js";
-import { type ZodType, z } from "zod";
-import { type PromiseOrValue } from "../utils/type-utils.js";
-import { Agent, type AgentInvokeOptions, type AgentOptions, type Message } from "./agent.js";
-export interface MCPAgentOptions extends AgentOptions {
-    client: Client;
-    prompts?: MCPPrompt[];
-    resources?: MCPResource[];
-}
-export type MCPServerOptions = SSEServerParameters | StdioServerParameters;
-export type SSEServerParameters = {
-    url: string;
-    /**
-     * Whether to use the StreamableHTTPClientTransport instead of the SSEClientTransport.
-     * @default "sse"
-     */
-    transport?: "sse" | "streamableHttp";
-    /**
-     * Additional options to pass to the SSEClientTransport or StreamableHTTPClientTransport.
-     */
-    opts?: SSEClientTransportOptions | StreamableHTTPClientTransportOptions;
-    /**
-     * The timeout for requests to the server, in milliseconds.
-     * @default 60000
-     */
-    timeout?: number;
-    /**
-     * Whether to automatically reconnect to the server if the connection is lost.
-     * @default 10 set to 0 to disable automatic reconnection
-     */
-    maxReconnects?: number;
-    /**
-     * A function that determines whether to reconnect to the server based on the error.
-     * default to reconnect on all errors.
-     */
-    shouldReconnect?: (error: Error) => boolean;
-};
-/**
- * MCPAgent is a specialized agent for interacting with MCP (Model Context Protocol) servers.
- * It provides the ability to connect to remote MCP servers using different transport methods,
- * and access their tools, prompts, and resources.
- *
- * MCPAgent serves as a bridge between your application and MCP servers, allowing you to:
- * - Connect to MCP servers over HTTP/SSE or stdio
- * - Access server tools as agent skills
- * - Utilize server prompts and resources
- * - Manage server connections with automatic reconnection
- *
- * @example
- * Here's an example of creating an MCPAgent with SSE transport:
- * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-from-sse}
- */
-export declare class MCPAgent extends Agent {
-    tag: string;
-    static schema(): z.ZodObject<{
-        url: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        command: ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
-        args: ZodType<string[] | undefined, z.ZodTypeDef, string[] | undefined>;
-    }, "strip", z.ZodTypeAny, {
-        url?: string | undefined;
-        command?: string | undefined;
-        args?: string[] | undefined;
-    }, {
-        url?: string | undefined;
-        command?: string | undefined;
-        args?: string[] | undefined;
-    }>;
-    static load<I extends Message = any, O extends Message = any>(options: {
-        filepath: string;
-        parsed: object;
-    }): Promise<Agent<I, O>>;
-    /**
-     * Create an MCPAgent from a connection to an SSE server.
-     *
-     * This overload establishes a Server-Sent Events connection to an MCP server
-     * and automatically discovers its available tools, prompts, and resources.
-     *
-     * @param options SSE server connection parameters
-     * @returns Promise resolving to a new MCPAgent instance
-     *
-     * @example
-     * Here's an example of creating an MCPAgent with StreamableHTTP transport:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-from-streamable-http}
-     *
-     * @example
-     * Here's an example of creating an MCPAgent with SSE transport:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-from-sse}
-     *
-     * @example
-     * Here's an example of creating an MCPAgent with Stdio transport:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-from-stdio}
-     */
-    static from(options: MCPServerOptions): Promise<MCPAgent>;
-    /**
-     * Create an MCPAgent from a pre-configured MCP client.
-     *
-     * This overload uses an existing MCP client instance and optionally
-     * pre-defined prompts and resources.
-     *
-     * @param options MCPAgent configuration with client instance
-     * @returns A new MCPAgent instance
-     *
-     * @example
-     * Here's an example of creating an MCPAgent with a client instance:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-direct}
-     */
-    static from(options: MCPAgentOptions): MCPAgent;
-    private static fromTransport;
-    /**
-     * Create an MCPAgent instance directly with a configured client.
-     *
-     * @param options MCPAgent configuration options, including client instance
-     *
-     * @example
-     * Here's an example of creating an MCPAgent with an existing client:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-direct}
-     */
-    constructor(options: MCPAgentOptions);
-    /**
-     * The MCP client instance used for communication with the MCP server.
-     *
-     * This client manages the connection to the MCP server and provides
-     * methods for interacting with server-provided functionality.
-     */
-    client: Client;
-    /**
-     * Array of MCP prompts available from the connected server.
-     *
-     * Prompts can be accessed by index or by name.
-     *
-     * @example
-     * Here's an example of accessing prompts:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-prompts}
-     */
-    readonly prompts: MCPPrompt[] & {
-        [key: string]: MCPPrompt;
-    };
-    /**
-     * Array of MCP resources available from the connected server.
-     *
-     * Resources can be accessed by index or by name.
-     *
-     * @example
-     * Here's an example of accessing resources:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-resources}
-     */
-    readonly resources: MCPResource[] & {
-        [key: string]: MCPResource;
-    };
-    /**
-     * Check if the agent is invokable.
-     *
-     * MCPAgent itself is not directly invokable as it acts as a container
-     * for tools, prompts, and resources. Always returns false.
-     */
-    get isInvokable(): boolean;
-    /**
-     * Process method required by Agent interface.
-     *
-     * Since MCPAgent itself is not directly invokable, this method
-     * throws an error if called.
-     *
-     * @param _input Input message (unused)
-     * @param _options AgentInvokeOptions (unused)
-     * @throws Error This method always throws an error since MCPAgent is not directly invokable
-     */
-    process(_input: Message, _options: AgentInvokeOptions): Promise<Message>;
-    /**
-     * Shut down the agent and close the MCP connection.
-     *
-     * This method cleans up resources and closes the connection
-     * to the MCP server.
-     *
-     * @example
-     * Here's an example of shutting down an MCPAgent:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-shutdown}
-     *
-     * @example
-     * Here's an example of shutting down an MCPAgent by using statement:
-     * {@includeCode ../../test/agents/mcp-agent.test.ts#example-mcp-agent-shutdown-by-using}
-     */
-    shutdown(): Promise<void>;
-}
-export interface ClientWithReconnectOptions {
-    transportCreator?: () => PromiseOrValue<Transport>;
-    timeout?: number;
-    maxReconnects?: number;
-    shouldReconnect?: (error: Error) => boolean;
-}
-declare class ClientWithReconnect extends Client {
-    private reconnectOptions?;
-    constructor(info: Implementation, options?: ClientOptions, reconnectOptions?: ClientWithReconnectOptions | undefined);
-    private shouldReconnect;
-    private reconnect;
-    request<T extends ZodType<object>>(request: Request, resultSchema: T, options?: RequestOptions): Promise<z.infer<T>>;
-}
-export interface MCPBaseOptions<I extends Message = any, O extends Message = any> extends AgentOptions<I, O> {
-    client: ClientWithReconnect;
-}
-export declare abstract class MCPBase<I extends Message, O extends Message> extends Agent<I, O> {
-    tag: string;
-    constructor(options: MCPBaseOptions<I, O>);
-    protected client: ClientWithReconnect;
-}
-export declare class MCPTool extends MCPBase<Message, CallToolResult> {
-    tag: string;
-    process(input: Message): Promise<CallToolResult>;
-}
-export interface MCPPromptInput extends Record<string, unknown> {
-    [key: string]: string;
-}
-export declare class MCPPrompt extends MCPBase<MCPPromptInput, GetPromptResult> {
-    tag: string;
-    process(input: MCPPromptInput): Promise<GetPromptResult>;
-}
-export interface MCPResourceOptions extends MCPBaseOptions<MCPPromptInput, ReadResourceResult> {
-    uri: string;
-}
-export declare class MCPResource extends MCPBase<MCPPromptInput, ReadResourceResult> {
-    tag: string;
-    constructor(options: MCPResourceOptions);
-    uri: string;
-    process(input: MCPPromptInput): Promise<ReadResourceResult>;
-}
-export {};