@aigne/core 1.60.3 → 1.61.0-beta.1

package/CHANGELOG.md

@@ -1,5 +1,31 @@
 # Changelog
 
+## [1.61.0-beta.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.61.0-beta...core-v1.61.0-beta.1) (2025-09-23)
+
+
+### Bug Fixes
+
+* standardize file parameter naming across models ([#534](https://github.com/AIGNE-io/aigne-framework/issues/534)) ([f159a9d](https://github.com/AIGNE-io/aigne-framework/commit/f159a9d6af21ec0e99641996b150560929845845))
+
+## [1.61.0-beta](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.60.3...core-v1.61.0-beta) (2025-09-22)
+
+
+### Features
+
+* improve image model architecture and file handling ([#527](https://github.com/AIGNE-io/aigne-framework/issues/527)) ([4db50aa](https://github.com/AIGNE-io/aigne-framework/commit/4db50aa0387a1a0f045ca11aaa61613e36ca7597))
+
+
+### Bug Fixes
+
+* resolve Windows file import URI issues ([#528](https://github.com/AIGNE-io/aigne-framework/issues/528)) ([bf807c5](https://github.com/AIGNE-io/aigne-framework/commit/bf807c5a3563c4423dc82fddff7fba280ef57957))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/platform-helpers bumped to 0.6.3-beta
+
 ## [1.60.3](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.60.2...core-v1.60.3) (2025-09-18)
 
 

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

@@ -1,10 +1,11 @@
 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, FileOutputType } from "./chat-model.js";
+import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
+import type { FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
-export declare const DEFAULT_FILE_OUTPUT_KEY = "files";
+export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
  * Configuration options for an AI Agent
  *
@@ -26,15 +27,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;
+    inputFileKey?: string;
     /**
      * Custom key to use for text output in the response
      *
      * Defaults to `message` if not specified
      */
     outputKey?: string;
-    fileOutputKey?: string;
-    fileOutputType?: FileOutputType;
+    outputFileKey?: string;
+    outputFileType?: FileType;
     /**
      * Controls how the agent uses tools during execution
      *
@@ -209,7 +210,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;
+    inputFileKey?: string;
     /**
      * Custom key to use for text output in the response
      *
@@ -218,8 +219,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;
+    outputFileKey: string;
+    outputFileType?: FileType;
     /**
      * Controls how the agent uses tools during execution
      *

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

@@ -1,6 +1,6 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AIAgent = exports.aiAgentOptionsSchema = exports.aiAgentToolChoiceSchema = exports.AIAgentToolChoice = exports.DEFAULT_FILE_OUTPUT_KEY = exports.DEFAULT_OUTPUT_KEY = void 0;
+exports.AIAgent = exports.aiAgentOptionsSchema = exports.aiAgentToolChoiceSchema = exports.AIAgentToolChoice = exports.DEFAULT_OUTPUT_FILE_KEY = exports.DEFAULT_OUTPUT_KEY = void 0;
 const zod_1 = require("zod");
 const prompt_builder_js_1 = require("../prompt/prompt-builder.js");
 const structured_stream_instructions_js_1 = require("../prompt/prompts/structured-stream-instructions.js");
@@ -10,7 +10,7 @@ const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
 const types_js_1 = require("./types.js");
 exports.DEFAULT_OUTPUT_KEY = "message";
-exports.DEFAULT_FILE_OUTPUT_KEY = "files";
+exports.DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
  * Tool choice options for AI agents
  *
@@ -111,10 +111,10 @@ class AIAgent extends agent_js_1.Agent {
                 ? prompt_builder_js_1.PromptBuilder.from(options.instructions)
                 : (options.instructions ?? new prompt_builder_js_1.PromptBuilder());
         this.inputKey = options.inputKey;
-        this.fileInputKey = options.fileInputKey;
+        this.inputFileKey = options.inputFileKey;
         this.outputKey = options.outputKey || exports.DEFAULT_OUTPUT_KEY;
-        this.fileOutputKey = options.fileOutputKey || exports.DEFAULT_FILE_OUTPUT_KEY;
-        this.fileOutputType = options.fileOutputType;
+        this.outputFileKey = options.outputFileKey || exports.DEFAULT_OUTPUT_FILE_KEY;
+        this.outputFileType = options.outputFileType;
         this.toolChoice = options.toolChoice;
         this.memoryAgentsAsTools = options.memoryAgentsAsTools;
         this.memoryPromptTemplate = options.memoryPromptTemplate;
@@ -148,7 +148,7 @@ class AIAgent extends agent_js_1.Agent {
      * Pick a message from input to use as the user's message
      */
     inputKey;
-    fileInputKey;
+    inputFileKey;
     /**
      * Custom key to use for text output in the response
      *
@@ -157,8 +157,8 @@ class AIAgent extends agent_js_1.Agent {
      * {@includeCode ../../test/agents/ai-agent.test.ts#example-ai-agent-custom-output-key}
      */
     outputKey;
-    fileOutputKey;
-    fileOutputType;
+    outputFileKey;
+    outputFileType;
     /**
      * Controls how the agent uses tools during execution
      *
@@ -309,7 +309,7 @@ class AIAgent extends agent_js_1.Agent {
                 Object.assign(result, { [outputKey]: text });
             }
             if (files) {
-                Object.assign(result, { [this.fileOutputKey]: files });
+                Object.assign(result, { [this.outputFileKey]: files });
             }
             if (!(0, type_utils_js_1.isEmpty)(result)) {
                 yield { delta: { json: result } };

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

@@ -1,11 +1,12 @@
 import { z } from "zod";
 import { type PromiseOrValue } from "../utils/type-utils.js";
-import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type AgentResponse, type AgentResponseStream, type Message } from "./agent.js";
+import { type AgentInvokeOptions, type AgentOptions, type AgentProcessResult, type AgentResponse, type AgentResponseStream, type Message } from "./agent.js";
+import { type FileType, type FileUnionContent, Model } from "./model.js";
 export declare class StructuredOutputError extends Error {
 }
 export interface ChatModelOptions extends Omit<AgentOptions<ChatModelInput, ChatModelOutput>, "model" | "inputSchema" | "outputSchema"> {
     model?: string;
-    modelOptions?: Omit<ModelOptions, "model">;
+    modelOptions?: Omit<ChatModelInputOptions, "model">;
 }
 /**
  * ChatModel is an abstract base class for interacting with Large Language Models (LLMs).
@@ -30,7 +31,7 @@ export interface ChatModelOptions extends Omit<AgentOptions<ChatModelInput, Chat
  * Here's an example with tool calls:
  * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
-export declare abstract class ChatModel extends Agent<ChatModelInput, ChatModelOutput> {
+export declare abstract class ChatModel extends Model<ChatModelInput, ChatModelOutput> {
     options?: ChatModelOptions | undefined;
     tag: string;
     constructor(options?: ChatModelOptions | undefined);
@@ -110,10 +111,6 @@ export declare abstract class ChatModel extends Agent<ChatModelInput, ChatModelO
      */
     abstract process(input: ChatModelInput, options: AgentInvokeOptions): PromiseOrValue<AgentProcessResult<ChatModelOutput>>;
     protected processAgentOutput(input: ChatModelInput, output: Exclude<AgentResponse<ChatModelOutput>, AgentResponseStream<ChatModelOutput>>, options: AgentInvokeOptions): Promise<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
@@ -138,7 +135,7 @@ export interface ChatModelInput extends Message {
      * Specifies the expected response format
      */
     responseFormat?: ChatModelInputResponseFormat;
-    fileOutputType?: FileOutputType;
+    outputFileType?: FileType;
     /**
      * List of tools available for the model to use
      */
@@ -150,7 +147,7 @@ export interface ChatModelInput extends Message {
     /**
      * Model-specific configuration options
      */
-    modelOptions?: ModelOptions;
+    modelOptions?: ChatModelInputOptions;
 }
 /**
  * Message role types
@@ -221,235 +218,6 @@ export declare const textContentSchema: z.ZodObject<{
     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, {
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}>;
-/**
- * Image URL content type
- *
- * Used for image parts of message content, referencing images via 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "url";
-    url: string;
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "file";
-    data: string;
-    mimeType?: string | undefined;
-    filename?: 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";
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    path: string;
-    type: "local";
-    mimeType?: string | undefined;
-    filename?: 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";
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    path: string;
-    type: "local";
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "url";
-    url: string;
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "file";
-    data: string;
-    mimeType?: string | undefined;
-    filename?: 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";
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    path: string;
-    type: "local";
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "url";
-    url: string;
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "file";
-    data: string;
-    mimeType?: string | undefined;
-    filename?: 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";
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    path: string;
-    type: "local";
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "url";
-    url: string;
-    mimeType?: string | undefined;
-    filename?: 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;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}, {
-    type: "file";
-    data: string;
-    mimeType?: string | undefined;
-    filename?: string | undefined;
-}>]>, "many">]>;
 export type UnionContent = TextContent | FileUnionContent;
 export declare const unionContentSchema: z.ZodDiscriminatedUnion<"type", [z.ZodObject<{
     type: z.ZodLiteral<"text">;
@@ -469,13 +237,13 @@ export declare const unionContentSchema: z.ZodDiscriminatedUnion<"type", [z.ZodO
 }, "strip", z.ZodTypeAny, {
     path: string;
     type: "local";
-    mimeType?: string | undefined;
     filename?: string | undefined;
+    mimeType?: string | undefined;
 }, {
     path: string;
     type: "local";
-    mimeType?: string | undefined;
     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>;
@@ -485,13 +253,13 @@ export declare const unionContentSchema: z.ZodDiscriminatedUnion<"type", [z.ZodO
 }, "strip", z.ZodTypeAny, {
     type: "url";
     url: string;
-    mimeType?: string | undefined;
     filename?: string | undefined;
+    mimeType?: string | undefined;
 }, {
     type: "url";
     url: string;
-    mimeType?: string | undefined;
     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>;
@@ -501,13 +269,13 @@ export declare const unionContentSchema: z.ZodDiscriminatedUnion<"type", [z.ZodO
 }, "strip", z.ZodTypeAny, {
     type: "file";
     data: string;
-    mimeType?: string | undefined;
     filename?: string | undefined;
+    mimeType?: string | undefined;
 }, {
     type: "file";
     data: string;
-    mimeType?: string | undefined;
     filename?: string | undefined;
+    mimeType?: string | undefined;
 }>]>;
 /**
  * Model response format settings
@@ -583,7 +351,7 @@ export type Modality = "text" | "image" | "audio";
  *
  * Contains various parameters for controlling model behavior, such as model name, temperature, etc.
  */
-export interface ModelOptions {
+export interface ChatModelInputOptions {
     /**
      * Model name or version
      */
@@ -609,7 +377,7 @@ export interface ModelOptions {
      */
     parallelToolCalls?: boolean;
     modalities?: Modality[];
-    preferFileInputType?: "file" | "url";
+    preferInputFileType?: "file" | "url";
 }
 /**
  * Output message format for ChatModel
@@ -647,10 +415,6 @@ export interface ChatModelOutput extends Message {
     model?: string;
     files?: FileUnionContent[];
 }
-export declare enum FileOutputType {
-    local = "local",
-    file = "file"
-}
 /**
  * Tool call information in model output
  *

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

@@ -32,19 +32,14 @@ var __importStar = (this && this.__importStar) || (function () {
         return result;
     };
 })();
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.chatModelOutputUsageSchema = exports.FileOutputType = exports.unionContentSchema = exports.fileUnionContentsSchema = exports.fileUnionContentSchema = exports.localContentSchema = exports.fileContentSchema = exports.urlContentSchema = exports.fileContentBaseSchema = exports.textContentSchema = exports.ChatModel = exports.StructuredOutputError = void 0;
+exports.chatModelOutputUsageSchema = exports.unionContentSchema = exports.textContentSchema = exports.ChatModel = exports.StructuredOutputError = void 0;
 const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
 const ajv_1 = require("ajv");
-const mime_1 = __importDefault(require("mime"));
-const uuid_1 = require("uuid");
 const zod_1 = require("zod");
-const schema_js_1 = require("../loader/schema.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
+const model_js_1 = require("./model.js");
 const CHAT_MODEL_DEFAULT_RETRY_OPTIONS = {
     retries: 3,
     shouldRetry: async (error) => error instanceof StructuredOutputError || (await Promise.resolve().then(() => __importStar(require("is-network-error")))).default(error),
@@ -75,7 +70,7 @@ exports.StructuredOutputError = StructuredOutputError;
  * Here's an example with tool calls:
  * {@includeCode ../../test/agents/chat-model.test.ts#example-chat-model-tools}
  */
-class ChatModel extends agent_js_1.Agent {
+class ChatModel extends model_js_1.Model {
     options;
     tag = "ChatModelAgent";
     constructor(options) {
@@ -191,8 +186,8 @@ class ChatModel extends agent_js_1.Agent {
                             mimeType: item.mimeType || ChatModel.getMimeType(item.filename || item.path),
                         };
                     }
-                    if ((input.modelOptions?.preferFileInputType ||
-                        this.options?.modelOptions?.preferFileInputType) !== "url") {
+                    if ((input.modelOptions?.preferInputFileType ||
+                        this.options?.modelOptions?.preferInputFileType) !== "url") {
                         if (item.type === "url") {
                             return {
                                 ...item,
@@ -242,10 +237,10 @@ class ChatModel extends agent_js_1.Agent {
     }
     async processAgentOutput(input, output, options) {
         if (output.files) {
-            const files = zod_1.z.array(exports.fileUnionContentSchema).parse(output.files);
+            const files = zod_1.z.array(model_js_1.fileUnionContentSchema).parse(output.files);
             output = {
                 ...output,
-                files: await Promise.all(files.map((file) => this.transformFileOutput(input, file, options))),
+                files: await Promise.all(files.map((file) => this.transformFileType(input.outputFileType, file, options))),
             };
         }
         // Remove fields with `null` value for validation
@@ -260,97 +255,17 @@ class ChatModel extends agent_js_1.Agent {
         }
         return super.processAgentOutput(input, output, options);
     }
-    async transformFileOutput(input, data, options) {
-        const fileOutputType = input.fileOutputType || FileOutputType.local;
-        if (fileOutputType === data.type)
-            return data;
-        const common = (0, type_utils_js_1.pick)(data, "filename", "mimeType");
-        switch (fileOutputType) {
-            case FileOutputType.local: {
-                const dir = index_js_1.nodejs.path.join(index_js_1.nodejs.os.tmpdir(), options.context.id);
-                await index_js_1.nodejs.fs.mkdir(dir, { recursive: true });
-                const ext = ChatModel.getFileExtension(data.mimeType || data.filename || "");
-                const id = (0, uuid_1.v7)();
-                const filename = ext ? `${id}.${ext}` : id;
-                const path = index_js_1.nodejs.path.join(dir, filename);
-                if (data.type === "file") {
-                    await index_js_1.nodejs.fs.writeFile(path, data.data, "base64");
-                }
-                else if (data.type === "url") {
-                    await this.downloadFile(data.url)
-                        .then((res) => res.body)
-                        .then((body) => body && index_js_1.nodejs.fs.writeFile(path, body));
-                }
-                else {
-                    throw new Error(`Unexpected file type: ${data.type}`);
-                }
-                return { ...common, type: "local", path };
-            }
-            case FileOutputType.file: {
-                let base64;
-                if (data.type === "local") {
-                    base64 = await index_js_1.nodejs.fs.readFile(data.path, "base64");
-                }
-                else if (data.type === "url") {
-                    base64 = Buffer.from(await (await this.downloadFile(data.url)).arrayBuffer()).toString("base64");
-                }
-                else {
-                    throw new Error(`Unexpected file type: ${data.type}`);
-                }
-                return { ...common, type: "file", data: base64 };
-            }
-        }
-    }
-    static getFileExtension(type) {
-        return mime_1.default.getExtension(type) || undefined;
-    }
-    static getMimeType(filename) {
-        return mime_1.default.getType(filename) || undefined;
-    }
-    async downloadFile(url) {
-        const response = await fetch(url);
-        if (!response.ok) {
-            const text = await response.text().catch(() => null);
-            throw new Error(`Failed to download content from ${url}, ${response.status} ${response.statusText} ${text}`);
-        }
-        return response;
-    }
 }
 exports.ChatModel = ChatModel;
 exports.textContentSchema = zod_1.z.object({
     type: zod_1.z.literal("text"),
     text: zod_1.z.string(),
 });
-exports.fileContentBaseSchema = zod_1.z.object({
-    filename: (0, schema_js_1.optionalize)(zod_1.z.string()),
-    mimeType: (0, schema_js_1.optionalize)(zod_1.z.string()),
-});
-exports.urlContentSchema = exports.fileContentBaseSchema.extend({
-    type: zod_1.z.literal("url"),
-    url: zod_1.z.string(),
-});
-exports.fileContentSchema = exports.fileContentBaseSchema.extend({
-    type: zod_1.z.literal("file"),
-    data: zod_1.z.string(),
-});
-exports.localContentSchema = exports.fileContentBaseSchema.extend({
-    type: zod_1.z.literal("local"),
-    path: zod_1.z.string(),
-});
-exports.fileUnionContentSchema = zod_1.z.discriminatedUnion("type", [
-    exports.localContentSchema,
-    exports.urlContentSchema,
-    exports.fileContentSchema,
-]);
-exports.fileUnionContentsSchema = zod_1.z.union([
-    exports.fileUnionContentSchema,
-    zod_1.z.array(exports.fileUnionContentSchema),
-]);
 exports.unionContentSchema = zod_1.z.discriminatedUnion("type", [
     exports.textContentSchema,
-    exports.localContentSchema,
-    exports.urlContentSchema,
-    exports.fileContentSchema,
+    model_js_1.localContentSchema,
+    model_js_1.urlContentSchema,
+    model_js_1.fileContentSchema,
 ]);
 const chatModelInputMessageSchema = zod_1.z.object({
     role: zod_1.z.union([zod_1.z.literal("system"), zod_1.z.literal("user"), zod_1.z.literal("agent"), zod_1.z.literal("tool")]),
@@ -414,11 +329,6 @@ const chatModelInputSchema = zod_1.z.object({
     toolChoice: chatModelInputToolChoiceSchema.optional(),
     modelOptions: modelOptionsSchema.optional(),
 });
-var FileOutputType;
-(function (FileOutputType) {
-    FileOutputType["local"] = "local";
-    FileOutputType["file"] = "file";
-})(FileOutputType || (exports.FileOutputType = FileOutputType = {}));
 const chatModelOutputToolCallSchema = zod_1.z.object({
     id: zod_1.z.string(),
     type: zod_1.z.literal("function"),
@@ -438,5 +348,5 @@ const chatModelOutputSchema = zod_1.z.object({
     toolCalls: zod_1.z.array(chatModelOutputToolCallSchema).optional(),
     usage: exports.chatModelOutputUsageSchema.optional(),
     model: zod_1.z.string().optional(),
-    files: zod_1.z.array(exports.fileUnionContentSchema).optional(),
+    files: zod_1.z.array(model_js_1.fileUnionContentSchema).optional(),
 });