@aigne/core 1.72.0-beta.3 → 1.72.0-beta.5

package/lib/esm/agents/ai-agent.d.ts

@@ -188,6 +188,13 @@ export declare const aiAgentOptionsSchema: ZodObject<{
  */
 export declare class AIAgent<I extends Message = any, O extends Message = any> extends Agent<I, O> {
     tag: string;
+    static schema<T>({ filepath }: {
+        filepath: string;
+    }): ZodType<T>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+    }): Promise<Agent<I, O>>;
     /**
      * Create an AIAgent with the specified options
      *

package/lib/esm/agents/ai-agent.js

@@ -1,8 +1,10 @@
 import { z } from "zod";
+import { camelizeSchema, getInstructionsSchema, instructionsToPromptBuilder, optionalize, } from "../loader/schema.js";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { STRUCTURED_STREAM_INSTRUCTIONS } from "../prompt/prompts/structured-stream-instructions.js";
 import { AgentMessageTemplate, ToolMessageTemplate } from "../prompt/template.js";
 import * as fastq from "../utils/queue.js";
+import { mergeAgentResponseChunk } from "../utils/stream-utils.js";
 import { ExtractMetadataTransform } from "../utils/structured-stream-extractor.js";
 import { checkArguments, isEmpty } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, isAgentResponseDelta, } from "./agent.js";
@@ -84,6 +86,32 @@ export const aiAgentOptionsSchema = agentOptionsSchema.extend({
  */
 export class AIAgent extends Agent {
     tag = "AIAgent";
+    static schema({ filepath }) {
+        const instructionsSchema = getInstructionsSchema({ filepath });
+        return camelizeSchema(z.object({
+            instructions: optionalize(instructionsSchema),
+            autoReorderSystemMessages: optionalize(z.boolean()),
+            autoMergeSystemMessages: optionalize(z.boolean()),
+            inputKey: optionalize(z.string()),
+            outputKey: optionalize(z.string()),
+            inputFileKey: optionalize(z.string()),
+            outputFileKey: optionalize(z.string()),
+            toolChoice: optionalize(z.nativeEnum(AIAgentToolChoice)),
+            toolCallsConcurrency: optionalize(z.number().int().min(0)),
+            keepTextInToolUses: optionalize(z.boolean()),
+            catchToolsError: optionalize(z.boolean()),
+            structuredStreamMode: optionalize(z.boolean()),
+        }));
+    }
+    static async load(options) {
+        const schema = AIAgent.schema(options);
+        const valid = await schema.parseAsync(options.parsed);
+        return new AIAgent({
+            ...options.parsed,
+            ...valid,
+            instructions: valid.instructions && instructionsToPromptBuilder(valid.instructions),
+        });
+    }
     /**
      * Create an AIAgent with the specified options
      *
@@ -280,6 +308,16 @@ export class AIAgent extends Agent {
         }
         const toolCallMessages = [];
         const outputKey = this.outputKey;
+        const inputMessage = this.inputKey ? input[this.inputKey] : undefined;
+        if (inputMessage) {
+            yield {
+                progress: {
+                    event: "message",
+                    role: "user",
+                    message: [{ type: "text", content: inputMessage }],
+                },
+            };
+        }
         for (;;) {
             const modelOutput = {};
             let stream = await this.invokeChildAgent(model, { ...modelInput, messages: modelInput.messages.concat(toolCallMessages) }, { ...options, streaming: true });
@@ -289,12 +327,12 @@ export class AIAgent extends Agent {
             }
             let isTextIgnored = false;
             for await (const value of stream) {
+                mergeAgentResponseChunk(modelOutput, value);
                 if (isAgentResponseDelta(value)) {
                     if (!isTextIgnored && value.delta.text?.text) {
                         yield { delta: { text: { [outputKey]: value.delta.text.text } } };
                     }
                     if (value.delta.json) {
-                        Object.assign(modelOutput, value.delta.json);
                         if (this.structuredStreamMode) {
                             yield { delta: { json: value.delta.json.json } };
                             if (!isTextIgnored && modelOutput.json && this.ignoreTextOfStructuredStreamMode) {
@@ -304,13 +342,23 @@ export class AIAgent extends Agent {
                     }
                 }
             }
-            const { toolCalls, json, text, files } = modelOutput;
+            const { toolCalls, json, text, thoughts, files } = modelOutput;
+            if (text) {
+                yield {
+                    progress: { event: "message", role: "agent", message: [{ type: "text", content: text }] },
+                };
+            }
+            if (thoughts) {
+                yield {
+                    progress: { event: "message", role: "agent", message: [{ type: "thinking", thoughts }] },
+                };
+            }
             if (toolCalls?.length) {
                 if (this.keepTextInToolUses !== true) {
                     yield { delta: { json: { [outputKey]: "" } } };
                 }
                 else {
-                    yield { delta: { text: { [outputKey]: "\n" } } };
+                    yield { delta: { text: { [outputKey]: "\n\n" } } };
                 }
                 const executedToolCalls = [];
                 let error;
@@ -340,6 +388,20 @@ export class AIAgent extends Agent {
                     const tool = toolsMap.get(call.function.name);
                     if (!tool)
                         throw new Error(`Tool not found: ${call.function.name}`);
+                    yield {
+                        progress: {
+                            event: "message",
+                            role: "agent",
+                            message: [
+                                {
+                                    type: "tool_use",
+                                    name: call.function.name,
+                                    input: call.function.arguments,
+                                    toolUseId: call.id,
+                                },
+                            ],
+                        },
+                    };
                     queue.push({ tool, call });
                 }
                 await queue.drained();
@@ -347,6 +409,21 @@ export class AIAgent extends Agent {
                     throw error;
                 // Continue LLM function calling loop if any tools were executed
                 if (executedToolCalls.length) {
+                    for (const { call, output } of executedToolCalls) {
+                        yield {
+                            progress: {
+                                event: "message",
+                                role: "agent",
+                                message: [
+                                    {
+                                        type: "tool_result",
+                                        toolUseId: call.id,
+                                        content: output,
+                                    },
+                                ],
+                            },
+                        };
+                    }
                     const transferOutput = executedToolCalls.find((i) => isTransferAgentOutput(i.output))?.output;
                     if (transferOutput)
                         return transferOutput;
@@ -367,6 +444,11 @@ export class AIAgent extends Agent {
             if (!isEmpty(result)) {
                 yield { delta: { json: result } };
             }
+            if (text) {
+                yield {
+                    progress: { event: "message", role: "agent", message: [{ type: "text", content: text }] },
+                };
+            }
             return;
         }
     }

package/lib/esm/agents/image-agent.d.ts

@@ -1,4 +1,4 @@
-import { type ZodObject, type ZodType } from "zod";
+import z, { type ZodObject, type ZodType } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type Message } from "./agent.js";
 import { type ImageModelOutput } from "./image-model.js";
@@ -13,6 +13,22 @@ export declare const imageAgentOptionsSchema: ZodObject<{
 }>;
 export declare class ImageAgent<I extends Message = any, O extends ImageModelOutput = any> extends Agent<I, O> {
     tag: string;
+    static schema({ filepath }: {
+        filepath: string;
+    }): z.ZodObject<{
+        instructions: z.ZodType<import("../loader/schema.js").Instructions, z.ZodTypeDef, import("../loader/schema.js").Instructions>;
+        inputFileKey: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        instructions: import("../loader/schema.js").Instructions;
+        inputFileKey?: string | undefined;
+    }, {
+        instructions: import("../loader/schema.js").Instructions;
+        inputFileKey?: string | undefined;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+    }): Promise<Agent<I, O>>;
     static from<I extends Message = any, O extends ImageModelOutput = any>(options: ImageAgentOptions<I, O>): ImageAgent<I, O>;
     constructor(options: ImageAgentOptions<I, O>);
     instructions: PromptBuilder;

package/lib/esm/agents/image-agent.js

@@ -1,4 +1,5 @@
 import z from "zod";
+import { camelizeSchema, getInstructionsSchema, instructionsToPromptBuilder, optionalize, } from "../loader/schema.js";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { checkArguments } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, } from "./agent.js";
@@ -10,6 +11,21 @@ export const imageAgentOptionsSchema = agentOptionsSchema.extend({
 });
 export class ImageAgent extends Agent {
     tag = "ImageAgent";
+    static schema({ filepath }) {
+        const instructionsSchema = getInstructionsSchema({ filepath });
+        return camelizeSchema(z.object({
+            instructions: instructionsSchema,
+            inputFileKey: optionalize(z.string()),
+        }));
+    }
+    static async load(options) {
+        const valid = await ImageAgent.schema({ filepath: options.filepath }).parseAsync(options.parsed);
+        return new ImageAgent({
+            ...options.parsed,
+            ...valid,
+            instructions: instructionsToPromptBuilder(valid.instructions),
+        });
+    }
     static from(options) {
         return new ImageAgent(options);
     }

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

@@ -91,6 +91,7 @@ export declare const imageModelInputSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     modelOptions?: Record<string, unknown> | undefined;
+    outputFileType?: "local" | "url" | "file" | undefined;
     image?: ({
         type: "url";
         url: string;
@@ -107,11 +108,11 @@ export declare const imageModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[] | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     n?: number | undefined;
 }, {
     prompt: string;
     modelOptions?: Record<string, unknown> | undefined;
+    outputFileType?: "local" | "url" | "file" | undefined;
     image?: ({
         type: "url";
         url: string;
@@ -128,7 +129,6 @@ export declare const imageModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[] | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     n?: number | undefined;
 }>;
 export interface ImageModelOutput extends Message {
@@ -232,7 +232,6 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[];
-    model?: string | undefined;
     usage?: {
         inputTokens: number;
         outputTokens: number;
@@ -241,6 +240,7 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
+    model?: string | undefined;
 }, {
     images: ({
         type: "url";
@@ -258,7 +258,6 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[];
-    model?: string | undefined;
     usage?: {
         inputTokens: number;
         outputTokens: number;
@@ -267,4 +266,5 @@ export declare const imageModelOutputSchema: z.ZodObject<{
         cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
+    model?: string | undefined;
 }>;

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

@@ -58,6 +58,23 @@ export type SSEServerParameters = {
  */
 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;
+        args?: string[] | undefined;
+        command?: string | undefined;
+    }, {
+        url?: string | undefined;
+        args?: string[] | undefined;
+        command?: 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.
      *

package/lib/esm/agents/mcp-agent.js

@@ -5,6 +5,7 @@ import { SSEClientTransport, } from "@modelcontextprotocol/sdk/client/sse.js";
 import { StreamableHTTPClientTransport, } from "@modelcontextprotocol/sdk/client/streamableHttp.js";
 import { UriTemplate } from "@modelcontextprotocol/sdk/shared/uriTemplate.js";
 import { z } from "zod";
+import { optionalize } from "../loader/schema.js";
 import { logger } from "../utils/logger.js";
 import { promptFromMCPPrompt, resourceFromMCPResource, toolFromMCPTool, } from "../utils/mcp-utils.js";
 import { checkArguments, createAccessorArray } from "../utils/type-utils.js";
@@ -49,6 +50,23 @@ function getMCPServerString(options) {
  */
 export class MCPAgent extends Agent {
     tag = "MCPAgent";
+    static schema() {
+        return z.object({
+            url: optionalize(z.string()),
+            command: optionalize(z.string()),
+            args: optionalize(z.array(z.string())),
+        });
+    }
+    static async load(options) {
+        const valid = await MCPAgent.schema().parseAsync(options.parsed);
+        if (!valid.url && !valid.command) {
+            throw new Error(`Missing url or command in mcp agent: ${options.filepath}`);
+        }
+        return MCPAgent.from({
+            ...options.parsed,
+            ...valid,
+        });
+    }
     static from(options) {
         checkArguments("MCPAgent.from", mcpAgentOptionsSchema, options);
         if (isSSEServerParameters(options)) {

package/lib/esm/agents/team-agent.d.ts

@@ -1,3 +1,5 @@
+import { z } from "zod";
+import type { AgentLoadOptions } from "../loader/index.js";
 import { type PromiseOrValue } from "../utils/type-utils.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type Message } from "./agent.js";
 /**
@@ -200,6 +202,59 @@ export interface TeamAgentOptions<I extends Message, O extends Message> extends
  */
 export declare class TeamAgent<I extends Message, O extends Message> extends Agent<I, O> {
     tag: string;
+    static schema({ filepath }: {
+        filepath: string;
+    }): z.ZodObject<{
+        mode: z.ZodType<ProcessMode | undefined, z.ZodTypeDef, ProcessMode | undefined>;
+        iterateOn: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+        concurrency: z.ZodType<number | undefined, z.ZodTypeDef, number | undefined>;
+        iterateWithPreviousOutput: z.ZodType<boolean | undefined, z.ZodTypeDef, boolean | undefined>;
+        includeAllStepsOutput: z.ZodType<boolean | undefined, z.ZodTypeDef, boolean | undefined>;
+        reflection: z.ZodType<{
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined, z.ZodTypeDef, {
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        mode?: ProcessMode | undefined;
+        reflection?: {
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined;
+        iterateOn?: string | undefined;
+        concurrency?: number | undefined;
+        iterateWithPreviousOutput?: boolean | undefined;
+        includeAllStepsOutput?: boolean | undefined;
+    }, {
+        mode?: ProcessMode | undefined;
+        reflection?: {
+            reviewer: import("../loader/agent-yaml.js").NestAgentSchema;
+            isApproved: string;
+            maxIterations?: number | undefined;
+            returnLastOnMaxIterations?: boolean | undefined;
+            customErrorMessage?: string | undefined;
+        } | undefined;
+        iterateOn?: string | undefined;
+        concurrency?: number | undefined;
+        iterateWithPreviousOutput?: boolean | undefined;
+        includeAllStepsOutput?: boolean | undefined;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+        options: AgentLoadOptions;
+    }): Promise<Agent<I, O>>;
     /**
      * Create a TeamAgent from the provided options.
      *

package/lib/esm/agents/team-agent.js

@@ -1,5 +1,8 @@
 import assert from "node:assert";
 import { produce } from "immer";
+import { z } from "zod";
+import { getNestAgentSchema } from "../loader/agent-yaml.js";
+import { camelizeSchema, optionalize } from "../loader/schema.js";
 import * as fastq from "../utils/queue.js";
 import { mergeAgentResponseChunk } from "../utils/stream-utils.js";
 import { isEmpty, isNil, isRecord, omit } from "../utils/type-utils.js";
@@ -47,6 +50,34 @@ export const DEFAULT_REFLECTION_MAX_ITERATIONS = 3;
  */
 export class TeamAgent extends Agent {
     tag = "TeamAgent";
+    static schema({ filepath }) {
+        const nestAgentSchema = getNestAgentSchema({ filepath });
+        return z.object({
+            mode: optionalize(z.nativeEnum(ProcessMode)),
+            iterateOn: optionalize(z.string()),
+            concurrency: optionalize(z.number().int().min(1)),
+            iterateWithPreviousOutput: optionalize(z.boolean()),
+            includeAllStepsOutput: optionalize(z.boolean()),
+            reflection: camelizeSchema(optionalize(z.object({
+                reviewer: nestAgentSchema,
+                isApproved: z.string(),
+                maxIterations: optionalize(z.number().int().min(1)),
+                returnLastOnMaxIterations: optionalize(z.boolean()),
+                customErrorMessage: optionalize(z.string()),
+            }))),
+        });
+    }
+    static async load(options) {
+        const valid = await TeamAgent.schema({ filepath: options.filepath }).parseAsync(options.parsed);
+        return TeamAgent.from({
+            ...options.parsed,
+            ...valid,
+            reflection: valid.reflection && {
+                ...valid.reflection,
+                reviewer: await options.options.loadNestAgent(options.filepath, valid.reflection.reviewer, options.options),
+            },
+        });
+    }
     /**
      * Create a TeamAgent from the provided options.
      *

package/lib/esm/agents/transform-agent.d.ts

@@ -1,3 +1,4 @@
+import { z } from "zod";
 import { Agent, type AgentOptions, type Message } from "./agent.js";
 /**
  * Configuration options for TransformAgent
@@ -47,6 +48,17 @@ export interface TransformAgentOptions<I extends Message, O extends Message> ext
  */
 export declare class TransformAgent<I extends Message = Message, O extends Message = Message> extends Agent<I, O> {
     static type: string;
+    static schema(): z.ZodObject<{
+        jsonata: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        jsonata: string;
+    }, {
+        jsonata: string;
+    }>;
+    static load<I extends Message = any, O extends Message = any>(options: {
+        filepath: string;
+        parsed: object;
+    }): Promise<Agent<I, O>>;
     /**
      * Factory method to create a new TransformAgent instance
      *

package/lib/esm/agents/transform-agent.js

@@ -1,4 +1,5 @@
 import jsonata from "jsonata";
+import { z } from "zod";
 import { Agent } from "./agent.js";
 /**
  * TransformAgent - A specialized agent for data transformation using JSONata expressions
@@ -17,6 +18,18 @@ import { Agent } from "./agent.js";
  */
 export class TransformAgent extends Agent {
     static type = "TransformAgent";
+    static schema() {
+        return z.object({
+            jsonata: z.string(),
+        });
+    }
+    static async load(options) {
+        const parsed = await TransformAgent.schema().parseAsync(options.parsed);
+        return TransformAgent.from({
+            ...options.parsed,
+            ...parsed,
+        });
+    }
     /**
      * Factory method to create a new TransformAgent instance
      *

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

@@ -94,6 +94,7 @@ export declare const videoModelInputSchema: z.ZodObject<{
     prompt: string;
     model?: string | undefined;
     modelOptions?: Record<string, unknown> | undefined;
+    outputFileType?: "local" | "url" | "file" | undefined;
     image?: {
         type: "url";
         url: string;
@@ -110,13 +111,13 @@ export declare const videoModelInputSchema: z.ZodObject<{
         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;
+    outputFileType?: "local" | "url" | "file" | undefined;
     image?: {
         type: "url";
         url: string;
@@ -133,7 +134,6 @@ export declare const videoModelInputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     } | undefined;
-    outputFileType?: "local" | "file" | "url" | undefined;
     size?: string | undefined;
     seconds?: string | undefined;
 }>;
@@ -240,7 +240,6 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[];
-    model?: string | undefined;
     usage?: {
         inputTokens: number;
         outputTokens: number;
@@ -249,6 +248,7 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
+    model?: string | undefined;
     seconds?: number | undefined;
 }, {
     videos: ({
@@ -267,7 +267,6 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         filename?: string | undefined;
         mimeType?: string | undefined;
     })[];
-    model?: string | undefined;
     usage?: {
         inputTokens: number;
         outputTokens: number;
@@ -276,5 +275,6 @@ export declare const videoModelOutputSchema: z.ZodObject<{
         cacheReadInputTokens?: number | undefined;
         creditPrefix?: "$" | "€" | "¥" | undefined;
     } | undefined;
+    model?: string | undefined;
     seconds?: number | undefined;
 }>;

package/lib/esm/aigne/context.d.ts

@@ -159,7 +159,7 @@ export declare class AIGNEContext implements Context {
     readonly internal: AIGNEContextShared;
     get messageQueue(): MessageQueue;
     get model(): ChatModel | undefined;
-    get imageModel(): ImageModel<import("../agents/image-model.js").ImageModelInput, import("../agents/image-model.js").ImageModelOutput> | undefined;
+    get imageModel(): ImageModel<import("../index.js").ImageModelInput, import("../index.js").ImageModelOutput> | undefined;
     get skills(): Agent<any, any>[] | undefined;
     get agents(): Agent<any, any>[];
     get observer(): AIGNEObserver | undefined;
@@ -197,7 +197,7 @@ declare class AIGNEContextShared {
     readonly messageQueue: MessageQueue;
     readonly events: Emitter<any>;
     get model(): ChatModel | undefined;
-    get imageModel(): ImageModel<import("../agents/image-model.js").ImageModelInput, import("../agents/image-model.js").ImageModelOutput> | undefined;
+    get imageModel(): ImageModel<import("../index.js").ImageModelInput, import("../index.js").ImageModelOutput> | undefined;
     get skills(): Agent<any, any>[] | undefined;
     get agents(): Agent<any, any>[];
     get observer(): AIGNEObserver | undefined;

package/lib/esm/loader/agent-yaml.d.ts

@@ -1,9 +1,6 @@
 import type { AFSOptions } from "@aigne/afs";
 import { type ZodType, z } from "zod";
-import type { AgentClass, AgentHooks, FunctionAgentFn, TaskRenderMode } from "../agents/agent.js";
-import { AIAgentToolChoice } from "../agents/ai-agent.js";
-import { type Role } from "../agents/chat-model.js";
-import { ProcessMode, type ReflectionMode } from "../agents/team-agent.js";
+import type { AgentHooks, TaskRenderMode } from "../agents/agent.js";
 import type { LoadOptions } from "./index.js";
 import { chatModelSchema, imageModelSchema } from "./schema.js";
 export interface HooksSchema {
@@ -45,7 +42,8 @@ export interface AFSContextSchema {
         presets?: Record<string, AFSContextPresetSchema>;
     };
 }
-export interface BaseAgentSchema {
+export interface AgentSchema {
+    type: string;
     name?: string;
     description?: string;
     model?: z.infer<typeof chatModelSchema>;
@@ -67,71 +65,11 @@ export interface BaseAgentSchema {
         context?: AFSContextSchema;
     });
     shareAFS?: boolean;
+    [key: string]: unknown;
 }
-export type Instructions = {
-    role: Exclude<Role, "tool">;
-    content: string;
-    path: string;
-    cacheControl?: {
-        type: "ephemeral";
-        ttl?: "5m" | "1h";
-    };
-}[];
-export interface AIAgentSchema extends BaseAgentSchema {
-    type: "ai";
-    instructions?: Instructions;
-    autoReorderSystemMessages?: boolean;
-    autoMergeSystemMessages?: boolean;
-    inputKey?: string;
-    inputFileKey?: string;
-    outputKey?: string;
-    outputFileKey?: string;
-    toolChoice?: AIAgentToolChoice;
-    toolCallsConcurrency?: number;
-    keepTextInToolUses?: boolean;
-}
-export interface ImageAgentSchema extends BaseAgentSchema {
-    type: "image";
-    instructions: Instructions;
-    inputFileKey?: string;
-}
-export interface MCPAgentSchema extends BaseAgentSchema {
-    type: "mcp";
-    url?: string;
-    command?: string;
-    args?: string[];
-}
-export interface TeamAgentSchema extends BaseAgentSchema {
-    type: "team";
-    mode?: ProcessMode;
-    iterateOn?: string;
-    concurrency?: number;
-    iterateWithPreviousOutput?: boolean;
-    includeAllStepsOutput?: boolean;
-    reflection?: Omit<ReflectionMode, "reviewer"> & {
-        reviewer: NestAgentSchema;
-    };
-}
-export interface TransformAgentSchema extends BaseAgentSchema {
-    type: "transform";
-    jsonata: string;
-}
-export interface FunctionAgentSchema extends BaseAgentSchema {
-    type: "function";
-    process: FunctionAgentFn;
-}
-export interface ThirdAgentSchema extends BaseAgentSchema {
-    agentClass?: AgentClass;
-    type: "";
-    [key: string]: any;
-}
-export type AgentSchema = AIAgentSchema | ImageAgentSchema | MCPAgentSchema | TeamAgentSchema | TransformAgentSchema | FunctionAgentSchema | ThirdAgentSchema;
 export declare function parseAgentFile(path: string, data: any, options: LoadOptions): Promise<AgentSchema>;
 export declare function loadAgentFromYamlFile(path: string, options: LoadOptions): Promise<AgentSchema>;
-export declare const getInstructionsSchema: ({ filepath }: {
-    filepath: string;
-}) => ZodType<Instructions>;
-export declare const getAgentSchema: ({ filepath, options, }: {
+export declare const getAgentSchema: ({ filepath }: {
     filepath: string;
     options?: LoadOptions;
 }) => ZodType<AgentSchema, z.ZodTypeDef, AgentSchema>;