@aigne/core 1.57.4 → 1.58.0

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

@@ -32,13 +32,13 @@ export declare const imageModelInputSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     model?: string | undefined;
-    responseFormat?: "url" | "base64" | undefined;
+    responseFormat?: "base64" | "url" | undefined;
     image?: string | string[] | undefined;
     n?: number | undefined;
 }, {
     prompt: string;
     model?: string | undefined;
-    responseFormat?: "url" | "base64" | undefined;
+    responseFormat?: "base64" | "url" | undefined;
     image?: string | string[] | undefined;
     n?: number | undefined;
 }>;

package/lib/cjs/aigne/context.js

@@ -202,6 +202,8 @@ class AIGNEContext {
         return this.internal.events.emit(eventName, ...newArgs);
     }
     async trace(eventName, args, b) {
+        if (process.env.AIGNE_OBSERVABILITY_DISABLED)
+            return;
         const span = this.span;
         if (!span)
             return;

package/lib/cjs/prompt/prompt-builder.d.ts

@@ -1,7 +1,7 @@
 import type { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
 import { Agent, type AgentInvokeOptions, type Message } from "../agents/agent.js";
-import type { AIAgent } from "../agents/ai-agent.js";
-import type { ChatModel, ChatModelInput } from "../agents/chat-model.js";
+import { type AIAgent } from "../agents/ai-agent.js";
+import { type ChatModel, type ChatModelInput } from "../agents/chat-model.js";
 import { ChatMessagesTemplate } from "./template.js";
 export interface PromptBuilderOptions {
     instructions?: string | ChatMessagesTemplate;

package/lib/cjs/prompt/prompt-builder.js

@@ -6,6 +6,9 @@ const yaml_1 = require("yaml");
 const zod_1 = require("zod");
 const zod_to_json_schema_1 = require("zod-to-json-schema");
 const agent_js_1 = require("../agents/agent.js");
+const ai_agent_js_1 = require("../agents/ai-agent.js");
+const chat_model_js_1 = require("../agents/chat-model.js");
+const schema_js_1 = require("../loader/schema.js");
 const json_schema_js_1 = require("../utils/json-schema.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const memory_message_template_js_1 = require("./prompts/memory-message-template.js");
@@ -37,11 +40,11 @@ class PromptBuilder {
                         content = resource.text;
                     }
                     else if (typeof resource.blob === "string") {
-                        content = [{ type: "image_url", url: resource.blob }];
+                        content = [{ type: "url", url: resource.blob }];
                     }
                 }
                 else if (i.content.type === "image") {
-                    content = [{ type: "image_url", url: i.content.data }];
+                    content = [{ type: "url", url: i.content.data }];
                 }
                 if (!content)
                     throw new Error(`Unsupported content type ${i.content.type}`);
@@ -65,6 +68,7 @@ class PromptBuilder {
             responseFormat: options.agent?.structuredStreamMode
                 ? undefined
                 : this.buildResponseFormat(options),
+            fileOutputType: options.agent?.fileOutputType,
             ...this.buildTools(options),
         };
     }
@@ -83,6 +87,10 @@ class PromptBuilder {
         const messages = (await (typeof this.instructions === "string"
             ? template_js_1.ChatMessagesTemplate.from([template_js_1.SystemMessageTemplate.from(this.instructions)])
             : this.instructions)?.format(options.input, { workingDir: this.workingDir })) ?? [];
+        const fileInputKey = options.agent?.fileInputKey;
+        const files = (0, type_utils_js_1.flat)(fileInputKey
+            ? (0, type_utils_js_1.checkArguments)("Check input files", (0, schema_js_1.optionalize)(chat_model_js_1.fileUnionContentsSchema), input?.[fileInputKey])
+            : null);
         const memories = [];
         if (options.agent && options.context) {
             memories.push(...(await options.agent.retrieveMemories({ search: message }, { context: options.context })));
@@ -104,26 +112,68 @@ class PromptBuilder {
                 },
             })));
         }
-        if (message) {
-            messages.push({
-                role: "user",
-                content: message,
-            });
+        if (message || files.length) {
+            const content = [];
+            if (message)
+                content.push({ type: "text", text: message });
+            if (files.length)
+                content.push(...files);
+            messages.push({ role: "user", content });
         }
         return messages;
     }
     async convertMemoriesToMessages(memories, options) {
         const messages = [];
         const other = [];
+        const inputKey = options.agent?.inputKey;
+        const fileInputKey = options.agent?.fileInputKey;
+        const outputKey = options.agent?.outputKey || ai_agent_js_1.DEFAULT_OUTPUT_KEY;
+        const fileOutputKey = options.agent?.fileOutputKey || ai_agent_js_1.DEFAULT_FILE_OUTPUT_KEY;
         const stringOrStringify = (value) => typeof value === "string" ? value : (0, yaml_1.stringify)(value);
-        for (const { content } of memories) {
-            if ((0, type_utils_js_1.isRecord)(content) && "input" in content && "output" in content) {
-                if (!(0, type_utils_js_1.isNil)(content.input) && content.input !== "") {
-                    messages.push({ role: "user", content: stringOrStringify(content.input) });
+        const convertMemoryToMessage = async (content) => {
+            const { input, output } = content;
+            if (!input || !output)
+                return [];
+            const result = [];
+            const userMessageContent = [];
+            if (typeof input === "object") {
+                const inputMessage = inputKey ? Reflect.get(input, inputKey) : undefined;
+                const inputFiles = fileInputKey ? Reflect.get(input, fileInputKey) : undefined;
+                if (inputMessage) {
+                    userMessageContent.push({ type: "text", text: stringOrStringify(inputMessage) });
                 }
-                if (!(0, type_utils_js_1.isNil)(content.output) && content.output !== "") {
-                    messages.push({ role: "agent", content: stringOrStringify(content.output) });
+                if (inputFiles) {
+                    userMessageContent.push(...(0, type_utils_js_1.flat)((0, type_utils_js_1.checkArguments)("Check memory input files", (0, schema_js_1.optionalize)(chat_model_js_1.fileUnionContentsSchema), inputFiles)));
+                }
+            }
+            if (!userMessageContent.length) {
+                userMessageContent.push({ type: "text", text: stringOrStringify(input) });
+            }
+            result.push({ role: "user", content: userMessageContent });
+            const agentMessageContent = [];
+            if (typeof output === "object") {
+                const outputMessage = Reflect.get(output, outputKey);
+                const outputFiles = Reflect.get(output, fileOutputKey);
+                if (outputMessage) {
+                    agentMessageContent.push({ type: "text", text: stringOrStringify(outputMessage) });
                 }
+                if (outputFiles) {
+                    agentMessageContent.push(...(0, type_utils_js_1.flat)((0, type_utils_js_1.checkArguments)("Check memory output files", (0, schema_js_1.optionalize)(chat_model_js_1.fileUnionContentsSchema), outputFiles)));
+                }
+            }
+            if (!agentMessageContent.length) {
+                agentMessageContent.push({ type: "text", text: stringOrStringify(output) });
+            }
+            result.push({ role: "agent", content: agentMessageContent });
+            return result;
+        };
+        for (const { content } of memories) {
+            if ((0, type_utils_js_1.isRecord)(content) &&
+                "input" in content &&
+                content.input &&
+                "output" in content &&
+                content.output) {
+                messages.push(...(await convertMemoryToMessage(content)));
             }
             else {
                 other.push(content);

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

@@ -1,9 +1,10 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
-import type { ChatModel, ChatModelInput } from "./chat-model.js";
+import type { ChatModel, ChatModelInput, FileOutputType } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
+export declare const DEFAULT_FILE_OUTPUT_KEY = "files";
 /**
  * Configuration options for an AI Agent
  *
@@ -31,12 +32,15 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * Pick a message from input to use as the user's message
      */
     inputKey?: string;
+    fileInputKey?: string;
     /**
      * Custom key to use for text output in the response
      *
      * Defaults to `message` if not specified
      */
     outputKey?: string;
+    fileOutputKey?: string;
+    fileOutputType?: FileOutputType;
     /**
      * Controls how the agent uses tools during execution
      *
@@ -217,6 +221,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * Pick a message from input to use as the user's message
      */
     inputKey?: string;
+    fileInputKey?: string;
     /**
      * Custom key to use for text output in the response
      *
@@ -225,6 +230,8 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-custom-output-key}
      */
     outputKey: string;
+    fileOutputKey: string;
+    fileOutputType?: FileOutputType;
     /**
      * Controls how the agent uses tools during execution
      *

package/lib/dts/agents/chat-model.d.ts

@@ -104,6 +104,10 @@ export declare abstract class ChatModel extends Agent<ChatModelInput, ChatModelO
      * @returns A promise or direct value containing the model's response
      */
     abstract process(input: ChatModelInput, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<ChatModelOutput>>;
+    transformFileOutput(input: ChatModelInput, data: FileUnionContent, options: AgentInvokeOptions): Promise<FileUnionContent>;
+    static getFileExtension(type: string): string | undefined;
+    static getMimeType(filename: string): string | undefined;
+    protected downloadFile(url: string): Promise<Response>;
 }
 /**
  * Input message format for ChatModel
@@ -128,6 +132,7 @@ export interface ChatModelInput extends Message {
      * Specifies the expected response format
      */
     responseFormat?: ChatModelInputResponseFormat;
+    fileOutputType?: FileOutputType;
     /**
      * List of tools available for the model to use
      */
@@ -190,7 +195,7 @@ export interface ChatModelInputMessage {
  *
  * Can be a simple string, or a mixed array of text and image content
  */
-export type ChatModelInputMessageContent = string | (TextContent | ImageUrlContent)[];
+export type ChatModelInputMessageContent = string | UnionContent[];
 /**
  * Text content type
  *
@@ -200,15 +205,304 @@ export type TextContent = {
     type: "text";
     text: string;
 };
+export declare const textContentSchema: z.ZodObject<{
+    type: z.ZodLiteral<"text">;
+    text: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    type: "text";
+    text: string;
+}, {
+    type: "text";
+    text: string;
+}>;
+export interface FileContentBase {
+    filename?: string;
+    mimeType?: string;
+}
+export declare const fileContentBaseSchema: z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+}, "strip", z.ZodTypeAny, {
+    filename?: string | undefined;
+    mimeType?: string | undefined;
+}, {
+    filename?: string | undefined;
+    mimeType?: string | undefined;
+}>;
 /**
  * Image URL content type
  *
  * Used for image parts of message content, referencing images via URL
  */
-export type ImageUrlContent = {
-    type: "image_url";
+export interface UrlContent extends FileContentBase {
+    type: "url";
     url: string;
-};
+}
+export declare const urlContentSchema: z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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;
+}>;
+export interface FileContent extends FileContentBase {
+    type: "file";
+    data: string;
+}
+export declare const fileContentSchema: z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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;
+}>;
+export interface LocalContent extends FileContentBase {
+    type: "local";
+    path: string;
+}
+export declare const localContentSchema: z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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;
+}>;
+export type FileUnionContent = LocalContent | UrlContent | FileContent;
+export declare const fileUnionContentSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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;
+}>]>;
+export declare const fileUnionContentsSchema: z.ZodUnion<[z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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;
+}>]>, z.ZodArray<z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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">]>;
+export type UnionContent = TextContent | FileUnionContent;
+export declare const unionContentSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
+    type: z.ZodLiteral<"text">;
+    text: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    type: "text";
+    text: string;
+}, {
+    type: "text";
+    text: string;
+}>, z.ZodObject<{
+    filename: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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: z.ZodType<string | undefined, z.ZodTypeDef, string | undefined>;
+    mimeType: z.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;
+}>]>;
 /**
  * Model response format settings
  *
@@ -277,6 +571,7 @@ export type ChatModelInputToolChoice = "auto" | "none" | "required" | {
         description?: string;
     };
 };
+export type Modality = "text" | "image" | "audio";
 /**
  * Model-specific configuration options
  *
@@ -307,6 +602,7 @@ export interface ChatModelOptions {
      * Whether to allow parallel tool calls
      */
     parallelToolCalls?: boolean;
+    modalities?: Modality[];
 }
 /**
  * Output message format for ChatModel
@@ -342,6 +638,11 @@ export interface ChatModelOutput extends Message {
      * Model name or version used
      */
     model?: string;
+    files?: FileUnionContent[];
+}
+export declare enum FileOutputType {
+    local = "local",
+    file = "file"
 }
 /**
  * Tool call information in model output

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

@@ -32,13 +32,13 @@ export declare const imageModelInputSchema: z.ZodObject<{
 }, "strip", z.ZodTypeAny, {
     prompt: string;
     model?: string | undefined;
-    responseFormat?: "url" | "base64" | undefined;
+    responseFormat?: "base64" | "url" | undefined;
     image?: string | string[] | undefined;
     n?: number | undefined;
 }, {
     prompt: string;
     model?: string | undefined;
-    responseFormat?: "url" | "base64" | undefined;
+    responseFormat?: "base64" | "url" | undefined;
     image?: string | string[] | undefined;
     n?: number | undefined;
 }>;

package/lib/dts/prompt/prompt-builder.d.ts

@@ -1,7 +1,7 @@
 import type { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
 import { Agent, type AgentInvokeOptions, type Message } from "../agents/agent.js";
-import type { AIAgent } from "../agents/ai-agent.js";
-import type { ChatModel, ChatModelInput } from "../agents/chat-model.js";
+import { type AIAgent } from "../agents/ai-agent.js";
+import { type ChatModel, type ChatModelInput } from "../agents/chat-model.js";
 import { ChatMessagesTemplate } from "./template.js";
 export interface PromptBuilderOptions {
     instructions?: string | ChatMessagesTemplate;

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

@@ -1,9 +1,10 @@
 import { type ZodObject, type ZodType, z } from "zod";
 import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type Message } from "./agent.js";
-import type { ChatModel, ChatModelInput } from "./chat-model.js";
+import type { ChatModel, ChatModelInput, FileOutputType } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
+export declare const DEFAULT_FILE_OUTPUT_KEY = "files";
 /**
  * Configuration options for an AI Agent
  *
@@ -31,12 +32,15 @@ export interface AIAgentOptions<I extends Message = Message, O extends Message =
      * Pick a message from input to use as the user's message
      */
     inputKey?: string;
+    fileInputKey?: string;
     /**
      * Custom key to use for text output in the response
      *
      * Defaults to `message` if not specified
      */
     outputKey?: string;
+    fileOutputKey?: string;
+    fileOutputType?: FileOutputType;
     /**
      * Controls how the agent uses tools during execution
      *
@@ -217,6 +221,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * Pick a message from input to use as the user's message
      */
     inputKey?: string;
+    fileInputKey?: string;
     /**
      * Custom key to use for text output in the response
      *
@@ -225,6 +230,8 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-custom-output-key}
      */
     outputKey: string;
+    fileOutputKey: string;
+    fileOutputType?: FileOutputType;
     /**
      * Controls how the agent uses tools during execution
      *