@aigne/doubao 1.3.0 → 1.74.0-beta

package/README.md

@@ -2,19 +2,19 @@
 
 <p align="center">
   <picture>
-    <source srcset="https://raw.githubusercontent.com/AIGNE-io/aigne-framework/main/logo-dark.svg" media="(prefers-color-scheme: dark)">
-    <source srcset="https://raw.githubusercontent.com/AIGNE-io/aigne-framework/main/logo.svg" media="(prefers-color-scheme: light)">
-    <img src="https://raw.githubusercontent.com/AIGNE-io/aigne-framework/main/logo.svg" alt="AIGNE Logo" width="400" />
+    <source srcset="https://raw.githubusercontent.com/ArcBlock/aigne-framework/main/logo-dark.svg" media="(prefers-color-scheme: dark)">
+    <source srcset="https://raw.githubusercontent.com/ArcBlock/aigne-framework/main/logo.svg" media="(prefers-color-scheme: light)">
+    <img src="https://raw.githubusercontent.com/ArcBlock/aigne-framework/main/logo.svg" alt="AIGNE Logo" width="400" />
   </picture>
 </p>
 
-[![GitHub star chart](https://img.shields.io/github/stars/AIGNE-io/aigne-framework?style=flat-square)](https://star-history.com/#AIGNE-io/aigne-framework)
-[![Open Issues](https://img.shields.io/github/issues-raw/AIGNE-io/aigne-framework?style=flat-square)](https://github.com/AIGNE-io/aigne-framework/issues)
-[![codecov](https://codecov.io/gh/AIGNE-io/aigne-framework/graph/badge.svg?token=DO07834RQL)](https://codecov.io/gh/AIGNE-io/aigne-framework)
+[![GitHub star chart](https://img.shields.io/github/stars/ArcBlock/aigne-framework?style=flat-square)](https://star-history.com/#ArcBlock/aigne-framework)
+[![Open Issues](https://img.shields.io/github/issues-raw/ArcBlock/aigne-framework?style=flat-square)](https://github.com/ArcBlock/aigne-framework/issues)
+[![codecov](https://codecov.io/gh/ArcBlock/aigne-framework/graph/badge.svg?token=DO07834RQL)](https://codecov.io/gh/ArcBlock/aigne-framework)
 [![NPM Version](https://img.shields.io/npm/v/@aigne/doubao)](https://www.npmjs.com/package/@aigne/doubao)
-[![Elastic-2.0 licensed](https://img.shields.io/npm/l/@aigne/doubao)](https://github.com/AIGNE-io/aigne-framework/blob/main/LICENSE.md)
+[![Elastic-2.0 licensed](https://img.shields.io/npm/l/@aigne/doubao)](https://github.com/ArcBlock/aigne-framework/blob/main/LICENSE.md)
 
-AIGNE Doubao SDK for integrating with Doubao AI models within the [AIGNE Framework](https://github.com/AIGNE-io/aigne-framework).
+AIGNE Doubao SDK for integrating with Doubao AI models within the [AIGNE Framework](https://github.com/ArcBlock/aigne-framework).
 
 ## Introduction
 

package/dist/doubao-chat-model.cjs

@@ -0,0 +1,34 @@
+let _aigne_openai = require("@aigne/openai");
+
+//#region src/doubao-chat-model.ts
+const DOUBAO_DEFAULT_CHAT_MODEL = "doubao-seed-1-6-250615";
+const DOUBAO_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3";
+/**
+* Implementation of the ChatModel interface for Doubao's API
+*
+* This model uses OpenAI-compatible API format to interact with Doubao's models,
+* but with specific configuration and capabilities for Doubao.
+*
+* Default model: 'doubao-seed-1-6-250615'
+*
+* @example
+* Here's how to create and use a Doubao chat model:
+* {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model}
+*
+* @example
+* Here's an example with streaming response:
+* {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model-streaming}
+*/
+var DoubaoChatModel = class extends _aigne_openai.OpenAIChatModel {
+	constructor(options) {
+		super({
+			...options,
+			model: options?.model || DOUBAO_DEFAULT_CHAT_MODEL,
+			baseURL: options?.baseURL || DOUBAO_BASE_URL
+		});
+	}
+	apiKeyEnvName = "DOUBAO_API_KEY";
+};
+
+//#endregion
+exports.DoubaoChatModel = DoubaoChatModel;

package/{lib/dts/doubao-chat-model.d.ts → dist/doubao-chat-model.d.cts}

@@ -1,4 +1,6 @@
-import { OpenAIChatModel, type OpenAIChatModelOptions } from "@aigne/openai";
+import { OpenAIChatModel, OpenAIChatModelOptions } from "@aigne/openai";
+
+//#region src/doubao-chat-model.d.ts
 /**
  * Implementation of the ChatModel interface for Doubao's API
  *
@@ -15,7 +17,10 @@ import { OpenAIChatModel, type OpenAIChatModelOptions } from "@aigne/openai";
  * Here's an example with streaming response:
  * {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model-streaming}
  */
-export declare class DoubaoChatModel extends OpenAIChatModel {
-    constructor(options?: OpenAIChatModelOptions);
-    protected apiKeyEnvName: string;
+declare class DoubaoChatModel extends OpenAIChatModel {
+  constructor(options?: OpenAIChatModelOptions);
+  protected apiKeyEnvName: string;
 }
+//#endregion
+export { DoubaoChatModel };
+//# sourceMappingURL=doubao-chat-model.d.cts.map

package/dist/doubao-chat-model.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"doubao-chat-model.d.cts","names":[],"sources":["../src/doubao-chat-model.ts"],"mappings":";;;;AAqBA;;;;;;;;;;;;;;;cAAa,eAAA,SAAwB,eAAA;EAAA,YAAA,OAAA,GACb,sBAAA;EAAA,UAAA,aAAA;AAAA"}

package/{lib/cjs/doubao-chat-model.d.ts → dist/doubao-chat-model.d.mts}

@@ -1,4 +1,6 @@
-import { OpenAIChatModel, type OpenAIChatModelOptions } from "@aigne/openai";
+import { OpenAIChatModel, OpenAIChatModelOptions } from "@aigne/openai";
+
+//#region src/doubao-chat-model.d.ts
 /**
  * Implementation of the ChatModel interface for Doubao's API
  *
@@ -15,7 +17,10 @@ import { OpenAIChatModel, type OpenAIChatModelOptions } from "@aigne/openai";
  * Here's an example with streaming response:
  * {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model-streaming}
  */
-export declare class DoubaoChatModel extends OpenAIChatModel {
-    constructor(options?: OpenAIChatModelOptions);
-    protected apiKeyEnvName: string;
+declare class DoubaoChatModel extends OpenAIChatModel {
+  constructor(options?: OpenAIChatModelOptions);
+  protected apiKeyEnvName: string;
 }
+//#endregion
+export { DoubaoChatModel };
+//# sourceMappingURL=doubao-chat-model.d.mts.map

package/dist/doubao-chat-model.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"doubao-chat-model.d.mts","names":[],"sources":["../src/doubao-chat-model.ts"],"mappings":";;;;AAqBA;;;;;;;;;;;;;;;cAAa,eAAA,SAAwB,eAAA;EAAA,YAAA,OAAA,GACb,sBAAA;EAAA,UAAA,aAAA;AAAA"}

package/dist/doubao-chat-model.mjs

@@ -0,0 +1,35 @@
+import { OpenAIChatModel } from "@aigne/openai";
+
+//#region src/doubao-chat-model.ts
+const DOUBAO_DEFAULT_CHAT_MODEL = "doubao-seed-1-6-250615";
+const DOUBAO_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3";
+/**
+* Implementation of the ChatModel interface for Doubao's API
+*
+* This model uses OpenAI-compatible API format to interact with Doubao's models,
+* but with specific configuration and capabilities for Doubao.
+*
+* Default model: 'doubao-seed-1-6-250615'
+*
+* @example
+* Here's how to create and use a Doubao chat model:
+* {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model}
+*
+* @example
+* Here's an example with streaming response:
+* {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model-streaming}
+*/
+var DoubaoChatModel = class extends OpenAIChatModel {
+	constructor(options) {
+		super({
+			...options,
+			model: options?.model || DOUBAO_DEFAULT_CHAT_MODEL,
+			baseURL: options?.baseURL || DOUBAO_BASE_URL
+		});
+	}
+	apiKeyEnvName = "DOUBAO_API_KEY";
+};
+
+//#endregion
+export { DoubaoChatModel };
+//# sourceMappingURL=doubao-chat-model.mjs.map

package/dist/doubao-chat-model.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"doubao-chat-model.mjs","names":[],"sources":["../src/doubao-chat-model.ts"],"sourcesContent":["import { OpenAIChatModel, type OpenAIChatModelOptions } from \"@aigne/openai\";\n\nconst DOUBAO_DEFAULT_CHAT_MODEL = \"doubao-seed-1-6-250615\";\nconst DOUBAO_BASE_URL = \"https://ark.cn-beijing.volces.com/api/v3\";\n\n/**\n * Implementation of the ChatModel interface for Doubao's API\n *\n * This model uses OpenAI-compatible API format to interact with Doubao's models,\n * but with specific configuration and capabilities for Doubao.\n *\n * Default model: 'doubao-seed-1-6-250615'\n *\n * @example\n * Here's how to create and use a Doubao chat model:\n * {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model}\n *\n * @example\n * Here's an example with streaming response:\n * {@includeCode ../test/doubao-chat-model.test.ts#example-doubao-chat-model-streaming}\n */\nexport class DoubaoChatModel extends OpenAIChatModel {\n  constructor(options?: OpenAIChatModelOptions) {\n    super({\n      ...options,\n      model: options?.model || DOUBAO_DEFAULT_CHAT_MODEL,\n      baseURL: options?.baseURL || DOUBAO_BASE_URL,\n    });\n  }\n\n  protected override apiKeyEnvName = \"DOUBAO_API_KEY\";\n}\n"],"mappings":";;;AAEA,MAAM,4BAA4B;AAClC,MAAM,kBAAkB;;;;;;;;;;;;;;;;;AAkBxB,IAAa,kBAAb,cAAqC,gBAAgB;CACnD,YAAY,SAAkC;AAC5C,QAAM;GACJ,GAAG;GACH,OAAO,SAAS,SAAS;GACzB,SAAS,SAAS,WAAW;GAC9B,CAAC;;CAGJ,AAAmB,gBAAgB"}

package/dist/doubao-image-model.cjs

@@ -0,0 +1,180 @@
+let _aigne_core = require("@aigne/core");
+let _aigne_core_utils_camelize = require("@aigne/core/utils/camelize");
+let _aigne_core_utils_fetch = require("@aigne/core/utils/fetch");
+let _aigne_core_utils_type_utils = require("@aigne/core/utils/type-utils");
+let ufo = require("ufo");
+let zod = require("zod");
+
+//#region src/doubao-image-model.ts
+const DOUBAO_DEFAULT_IMAGE_MODEL = "doubao-seedream-4-0-250828";
+const DOUBAO_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3";
+const OUTPUT_MIME_TYPE = "image/jpeg";
+const doubaoImageModelInputSchema = _aigne_core.imageModelInputSchema.extend({});
+const doubaoImageModelOptionsSchema = zod.z.object({
+	apiKey: zod.z.string().optional(),
+	baseURL: zod.z.string().optional(),
+	model: zod.z.string().optional(),
+	modelOptions: zod.z.object({}).optional(),
+	clientOptions: zod.z.object({}).optional()
+});
+var DoubaoImageModel = class extends _aigne_core.ImageModel {
+	constructor(options) {
+		super({
+			...options,
+			inputSchema: doubaoImageModelInputSchema,
+			description: options?.description ?? "Draw or edit image by Doubao image models"
+		});
+		this.options = options;
+		if (options) (0, _aigne_core_utils_type_utils.checkArguments)(this.name, doubaoImageModelOptionsSchema, options);
+	}
+	apiKeyEnvName = "DOUBAO_API_KEY";
+	get credential() {
+		return {
+			url: this.options?.baseURL || process.env.DOUBAO_BASE_URL || DOUBAO_BASE_URL,
+			apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],
+			model: this.options?.model || DOUBAO_DEFAULT_IMAGE_MODEL
+		};
+	}
+	get modelOptions() {
+		return this.options?.modelOptions;
+	}
+	extractDataObjects(text) {
+		const dataObjects = [];
+		const lines = text.split("\n");
+		for (const line of lines) {
+			const trimmed = line.trim();
+			if (trimmed.startsWith("data:")) {
+				const jsonPart = trimmed.slice(5).trim();
+				if (jsonPart === "[DONE]") continue;
+				try {
+					const obj = JSON.parse(jsonPart);
+					dataObjects.push(obj);
+				} catch (e) {
+					console.warn("Failed to parse JSON data object:", jsonPart, e);
+				}
+			}
+		}
+		return dataObjects;
+	}
+	async process(input, _options) {
+		const model = input.modelOptions?.model || this.credential.model;
+		const { url, apiKey } = this.credential;
+		if (!apiKey) throw new Error(`${this.name} requires an API key. Please provide it via \`options.apiKey\`, or set the \`${this.apiKeyEnvName}\` environment variable`);
+		const map = {
+			"doubao-seedream-4": [
+				"model",
+				"prompt",
+				"size",
+				"sequentialImageGeneration",
+				"sequentialImageGenerationOptions",
+				"stream",
+				"responseFormat",
+				"watermark"
+			],
+			"doubao-seedream-3-0-t2i": [
+				"model",
+				"prompt",
+				"size",
+				"seed",
+				"guidanceScale",
+				"responseFormat",
+				"watermark"
+			],
+			"doubao-seededit-3-0-i2i": [
+				"model",
+				"prompt",
+				"size",
+				"seed",
+				"guidanceScale",
+				"responseFormat",
+				"watermark"
+			]
+		};
+		const key = Object.keys(map).find((key$1) => model.includes(key$1));
+		if (!key) throw new Error(`${this.name} only support ${Object.keys(map).join(", ")}`);
+		if (!map[key]) throw new Error(`${this.name} only support ${Object.keys(map).join(", ")}`);
+		const mergeInput = {
+			...this.modelOptions,
+			...input.modelOptions,
+			...input
+		};
+		const image = await Promise.all((0, _aigne_core_utils_type_utils.flat)(input.image).map((image$1) => this.transformFileType("file", image$1).then((file) => `data:${file.mimeType || "image/png"};base64,${file.data}`)));
+		const body = {
+			...(0, _aigne_core_utils_camelize.snakelize)((0, _aigne_core_utils_type_utils.pick)(mergeInput, map[key])),
+			model,
+			response_format: "b64_json",
+			watermark: mergeInput.watermark ?? false,
+			image: image.length ? image : void 0
+		};
+		const response = await (0, _aigne_core_utils_fetch.fetch)((0, ufo.joinURL)(url, `/images/generations`), {
+			method: "POST",
+			headers: {
+				Authorization: `Bearer ${apiKey}`,
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify(body),
+			timeout: this.options?.clientOptions?.timeout ?? 60 * 1e3
+		});
+		if (body.stream) {
+			if (!response.body) throw new Error("Streaming not supported in this environment");
+			const reader = response.body.getReader();
+			const decoder = new TextDecoder();
+			let done = false;
+			let buffer = "";
+			while (!done) {
+				const { value, done: doneReading } = await reader.read();
+				done = doneReading;
+				if (value) buffer += decoder.decode(value);
+			}
+			const dataObjects = this.extractDataObjects(buffer);
+			const error = dataObjects.find((i) => i.type === "image_generation.partial_failed");
+			if (error) throw new Error(`Doubao API error: ${error.error.message}`);
+			const completed = dataObjects.find((i) => i.type === "image_generation.completed");
+			return {
+				images: dataObjects.filter((i) => i.type === "image_generation.partial_succeeded").map((i) => {
+					if (typeof i.url === "string") return {
+						type: "url",
+						url: i.url,
+						mimeType: OUTPUT_MIME_TYPE
+					};
+					if (typeof i.b64_json === "string") return {
+						type: "file",
+						data: i.b64_json,
+						mimeType: OUTPUT_MIME_TYPE
+					};
+					throw new Error("Image response does not contain a valid URL or base64 data");
+				}),
+				usage: {
+					inputTokens: 0,
+					outputTokens: completed?.usage.output_tokens || 0
+				},
+				model
+			};
+		}
+		const data = await response.json();
+		if (data.error) throw new Error(`Doubao API error: ${data.error.message}`);
+		return {
+			images: data.data.map((item) => {
+				if (item.url) return {
+					type: "url",
+					url: item.url,
+					mimeType: OUTPUT_MIME_TYPE
+				};
+				if (item.b64_json) return {
+					type: "file",
+					data: item.b64_json,
+					mimeType: OUTPUT_MIME_TYPE
+				};
+				throw new Error("Image response does not contain a valid URL or base64 data");
+			}),
+			usage: {
+				inputTokens: 0,
+				outputTokens: data?.usage?.output_tokens || 0
+			},
+			model: data.model
+		};
+	}
+};
+
+//#endregion
+exports.DoubaoImageModel = DoubaoImageModel;

package/dist/doubao-image-model.d.cts

@@ -0,0 +1,38 @@
+import { AgentInvokeOptions, ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput } from "@aigne/core";
+
+//#region src/doubao-image-model.d.ts
+interface DoubaoImageModelInput extends ImageModelInput {
+  size?: string;
+  seed?: number;
+  sequentialImageGeneration?: boolean;
+  sequentialImageGenerationOptions?: {
+    maxImages: number;
+  };
+  stream?: boolean;
+  guidanceScale?: number;
+  watermark?: boolean;
+}
+interface DoubaoImageModelOutput extends ImageModelOutput {}
+interface DoubaoImageModelOptions extends ImageModelOptions<DoubaoImageModelInput, DoubaoImageModelOutput> {
+  apiKey?: string;
+  baseURL?: string;
+  model?: string;
+  modelOptions?: Omit<Partial<DoubaoImageModelInput>, "model">;
+  clientOptions?: Record<string, any>;
+}
+declare class DoubaoImageModel extends ImageModel<DoubaoImageModelInput, DoubaoImageModelOutput> {
+  options?: DoubaoImageModelOptions | undefined;
+  constructor(options?: DoubaoImageModelOptions | undefined);
+  protected apiKeyEnvName: string;
+  get credential(): {
+    url: string;
+    apiKey: string | undefined;
+    model: string;
+  };
+  get modelOptions(): Omit<Partial<DoubaoImageModelInput>, "model"> | undefined;
+  private extractDataObjects;
+  process(input: DoubaoImageModelInput, _options: AgentInvokeOptions): Promise<ImageModelOutput>;
+}
+//#endregion
+export { DoubaoImageModel, DoubaoImageModelInput, DoubaoImageModelOptions, DoubaoImageModelOutput };
+//# sourceMappingURL=doubao-image-model.d.cts.map

package/dist/doubao-image-model.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"doubao-image-model.d.cts","names":[],"sources":["../src/doubao-image-model.ts"],"mappings":";;;UAmBiB,qBAAA,SAA8B,eAAA;EAAA,IAAA;EAAA,IAAA;EAAA,yBAAA;EAAA,gCAAA;IAAA,SAAA;EAAA;EAAA,MAAA;EAAA,aAAA;EAAA,SAAA;AAAA;AAAA,UAY9B,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EAAA,MAAA;EAAA,OAAA;EAAA,KAAA;EAAA,YAAA,GAIlC,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAAA,aAAA,GACZ,MAAA;AAAA;AAAA,cAaL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA,UAAA,aAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GAoBtB,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA,QAAA,kBAAA;EAAA,QAAA,KAAA,EA4BP,qBAAA,EAAA,QAAA,EACG,kBAAA,GACT,OAAA,CAAQ,gBAAA;AAAA"}

package/dist/doubao-image-model.d.mts

@@ -0,0 +1,38 @@
+import { AgentInvokeOptions, ImageModel, ImageModelInput, ImageModelOptions, ImageModelOutput } from "@aigne/core";
+
+//#region src/doubao-image-model.d.ts
+interface DoubaoImageModelInput extends ImageModelInput {
+  size?: string;
+  seed?: number;
+  sequentialImageGeneration?: boolean;
+  sequentialImageGenerationOptions?: {
+    maxImages: number;
+  };
+  stream?: boolean;
+  guidanceScale?: number;
+  watermark?: boolean;
+}
+interface DoubaoImageModelOutput extends ImageModelOutput {}
+interface DoubaoImageModelOptions extends ImageModelOptions<DoubaoImageModelInput, DoubaoImageModelOutput> {
+  apiKey?: string;
+  baseURL?: string;
+  model?: string;
+  modelOptions?: Omit<Partial<DoubaoImageModelInput>, "model">;
+  clientOptions?: Record<string, any>;
+}
+declare class DoubaoImageModel extends ImageModel<DoubaoImageModelInput, DoubaoImageModelOutput> {
+  options?: DoubaoImageModelOptions | undefined;
+  constructor(options?: DoubaoImageModelOptions | undefined);
+  protected apiKeyEnvName: string;
+  get credential(): {
+    url: string;
+    apiKey: string | undefined;
+    model: string;
+  };
+  get modelOptions(): Omit<Partial<DoubaoImageModelInput>, "model"> | undefined;
+  private extractDataObjects;
+  process(input: DoubaoImageModelInput, _options: AgentInvokeOptions): Promise<ImageModelOutput>;
+}
+//#endregion
+export { DoubaoImageModel, DoubaoImageModelInput, DoubaoImageModelOptions, DoubaoImageModelOutput };
+//# sourceMappingURL=doubao-image-model.d.mts.map

package/dist/doubao-image-model.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"doubao-image-model.d.mts","names":[],"sources":["../src/doubao-image-model.ts"],"mappings":";;;UAmBiB,qBAAA,SAA8B,eAAA;EAAA,IAAA;EAAA,IAAA;EAAA,yBAAA;EAAA,gCAAA;IAAA,SAAA;EAAA;EAAA,MAAA;EAAA,aAAA;EAAA,SAAA;AAAA;AAAA,UAY9B,sBAAA,SAA+B,gBAAA;AAAA,UAE/B,uBAAA,SACP,iBAAA,CAAkB,qBAAA,EAAuB,sBAAA;EAAA,MAAA;EAAA,OAAA;EAAA,KAAA;EAAA,YAAA,GAIlC,IAAA,CAAK,OAAA,CAAQ,qBAAA;EAAA,aAAA,GACZ,MAAA;AAAA;AAAA,cAaL,gBAAA,SAAyB,UAAA,CAAW,qBAAA,EAAuB,sBAAA;EAAA,OAAA,GAChC,uBAAA;EAAA,YAAA,OAAA,GAAA,uBAAA;EAAA,UAAA,aAAA;EAAA,IAAA,WAAA;IAAA,GAAA;IAAA,MAAA;IAAA,KAAA;EAAA;EAAA,IAAA,aAAA,GAoBtB,IAAA,CAAA,OAAA,CAAA,qBAAA;EAAA,QAAA,kBAAA;EAAA,QAAA,KAAA,EA4BP,qBAAA,EAAA,QAAA,EACG,kBAAA,GACT,OAAA,CAAQ,gBAAA;AAAA"}

package/dist/doubao-image-model.mjs

@@ -0,0 +1,181 @@
+import { ImageModel, imageModelInputSchema } from "@aigne/core";
+import { snakelize } from "@aigne/core/utils/camelize";
+import { fetch } from "@aigne/core/utils/fetch";
+import { checkArguments, flat, pick } from "@aigne/core/utils/type-utils";
+import { joinURL } from "ufo";
+import { z } from "zod";
+
+//#region src/doubao-image-model.ts
+const DOUBAO_DEFAULT_IMAGE_MODEL = "doubao-seedream-4-0-250828";
+const DOUBAO_BASE_URL = "https://ark.cn-beijing.volces.com/api/v3";
+const OUTPUT_MIME_TYPE = "image/jpeg";
+const doubaoImageModelInputSchema = imageModelInputSchema.extend({});
+const doubaoImageModelOptionsSchema = z.object({
+	apiKey: z.string().optional(),
+	baseURL: z.string().optional(),
+	model: z.string().optional(),
+	modelOptions: z.object({}).optional(),
+	clientOptions: z.object({}).optional()
+});
+var DoubaoImageModel = class extends ImageModel {
+	constructor(options) {
+		super({
+			...options,
+			inputSchema: doubaoImageModelInputSchema,
+			description: options?.description ?? "Draw or edit image by Doubao image models"
+		});
+		this.options = options;
+		if (options) checkArguments(this.name, doubaoImageModelOptionsSchema, options);
+	}
+	apiKeyEnvName = "DOUBAO_API_KEY";
+	get credential() {
+		return {
+			url: this.options?.baseURL || process.env.DOUBAO_BASE_URL || DOUBAO_BASE_URL,
+			apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],
+			model: this.options?.model || DOUBAO_DEFAULT_IMAGE_MODEL
+		};
+	}
+	get modelOptions() {
+		return this.options?.modelOptions;
+	}
+	extractDataObjects(text) {
+		const dataObjects = [];
+		const lines = text.split("\n");
+		for (const line of lines) {
+			const trimmed = line.trim();
+			if (trimmed.startsWith("data:")) {
+				const jsonPart = trimmed.slice(5).trim();
+				if (jsonPart === "[DONE]") continue;
+				try {
+					const obj = JSON.parse(jsonPart);
+					dataObjects.push(obj);
+				} catch (e) {
+					console.warn("Failed to parse JSON data object:", jsonPart, e);
+				}
+			}
+		}
+		return dataObjects;
+	}
+	async process(input, _options) {
+		const model = input.modelOptions?.model || this.credential.model;
+		const { url, apiKey } = this.credential;
+		if (!apiKey) throw new Error(`${this.name} requires an API key. Please provide it via \`options.apiKey\`, or set the \`${this.apiKeyEnvName}\` environment variable`);
+		const map = {
+			"doubao-seedream-4": [
+				"model",
+				"prompt",
+				"size",
+				"sequentialImageGeneration",
+				"sequentialImageGenerationOptions",
+				"stream",
+				"responseFormat",
+				"watermark"
+			],
+			"doubao-seedream-3-0-t2i": [
+				"model",
+				"prompt",
+				"size",
+				"seed",
+				"guidanceScale",
+				"responseFormat",
+				"watermark"
+			],
+			"doubao-seededit-3-0-i2i": [
+				"model",
+				"prompt",
+				"size",
+				"seed",
+				"guidanceScale",
+				"responseFormat",
+				"watermark"
+			]
+		};
+		const key = Object.keys(map).find((key$1) => model.includes(key$1));
+		if (!key) throw new Error(`${this.name} only support ${Object.keys(map).join(", ")}`);
+		if (!map[key]) throw new Error(`${this.name} only support ${Object.keys(map).join(", ")}`);
+		const mergeInput = {
+			...this.modelOptions,
+			...input.modelOptions,
+			...input
+		};
+		const image = await Promise.all(flat(input.image).map((image$1) => this.transformFileType("file", image$1).then((file) => `data:${file.mimeType || "image/png"};base64,${file.data}`)));
+		const body = {
+			...snakelize(pick(mergeInput, map[key])),
+			model,
+			response_format: "b64_json",
+			watermark: mergeInput.watermark ?? false,
+			image: image.length ? image : void 0
+		};
+		const response = await fetch(joinURL(url, `/images/generations`), {
+			method: "POST",
+			headers: {
+				Authorization: `Bearer ${apiKey}`,
+				"Content-Type": "application/json"
+			},
+			body: JSON.stringify(body),
+			timeout: this.options?.clientOptions?.timeout ?? 60 * 1e3
+		});
+		if (body.stream) {
+			if (!response.body) throw new Error("Streaming not supported in this environment");
+			const reader = response.body.getReader();
+			const decoder = new TextDecoder();
+			let done = false;
+			let buffer = "";
+			while (!done) {
+				const { value, done: doneReading } = await reader.read();
+				done = doneReading;
+				if (value) buffer += decoder.decode(value);
+			}
+			const dataObjects = this.extractDataObjects(buffer);
+			const error = dataObjects.find((i) => i.type === "image_generation.partial_failed");
+			if (error) throw new Error(`Doubao API error: ${error.error.message}`);
+			const completed = dataObjects.find((i) => i.type === "image_generation.completed");
+			return {
+				images: dataObjects.filter((i) => i.type === "image_generation.partial_succeeded").map((i) => {
+					if (typeof i.url === "string") return {
+						type: "url",
+						url: i.url,
+						mimeType: OUTPUT_MIME_TYPE
+					};
+					if (typeof i.b64_json === "string") return {
+						type: "file",
+						data: i.b64_json,
+						mimeType: OUTPUT_MIME_TYPE
+					};
+					throw new Error("Image response does not contain a valid URL or base64 data");
+				}),
+				usage: {
+					inputTokens: 0,
+					outputTokens: completed?.usage.output_tokens || 0
+				},
+				model
+			};
+		}
+		const data = await response.json();
+		if (data.error) throw new Error(`Doubao API error: ${data.error.message}`);
+		return {
+			images: data.data.map((item) => {
+				if (item.url) return {
+					type: "url",
+					url: item.url,
+					mimeType: OUTPUT_MIME_TYPE
+				};
+				if (item.b64_json) return {
+					type: "file",
+					data: item.b64_json,
+					mimeType: OUTPUT_MIME_TYPE
+				};
+				throw new Error("Image response does not contain a valid URL or base64 data");
+			}),
+			usage: {
+				inputTokens: 0,
+				outputTokens: data?.usage?.output_tokens || 0
+			},
+			model: data.model
+		};
+	}
+};
+
+//#endregion
+export { DoubaoImageModel };
+//# sourceMappingURL=doubao-image-model.mjs.map

package/dist/doubao-image-model.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"doubao-image-model.mjs","names":["key","image"],"sources":["../src/doubao-image-model.ts"],"sourcesContent":["import {\n  type AgentInvokeOptions,\n  type FileUnionContent,\n  ImageModel,\n  type ImageModelInput,\n  type ImageModelOptions,\n  type ImageModelOutput,\n  imageModelInputSchema,\n} from \"@aigne/core\";\nimport { snakelize } from \"@aigne/core/utils/camelize\";\nimport { fetch } from \"@aigne/core/utils/fetch\";\nimport { checkArguments, flat, pick } from \"@aigne/core/utils/type-utils\";\nimport { joinURL } from \"ufo\";\nimport { z } from \"zod\";\n\nconst DOUBAO_DEFAULT_IMAGE_MODEL = \"doubao-seedream-4-0-250828\";\nconst DOUBAO_BASE_URL = \"https://ark.cn-beijing.volces.com/api/v3\";\nconst OUTPUT_MIME_TYPE = \"image/jpeg\";\n\nexport interface DoubaoImageModelInput extends ImageModelInput {\n  size?: string;\n  seed?: number;\n  sequentialImageGeneration?: boolean;\n  sequentialImageGenerationOptions?: {\n    maxImages: number;\n  };\n  stream?: boolean;\n  guidanceScale?: number;\n  watermark?: boolean;\n}\n\nexport interface DoubaoImageModelOutput extends ImageModelOutput {}\n\nexport interface DoubaoImageModelOptions\n  extends ImageModelOptions<DoubaoImageModelInput, DoubaoImageModelOutput> {\n  apiKey?: string;\n  baseURL?: string;\n  model?: string;\n  modelOptions?: Omit<Partial<DoubaoImageModelInput>, \"model\">;\n  clientOptions?: Record<string, any>;\n}\n\nconst doubaoImageModelInputSchema = imageModelInputSchema.extend({});\n\nconst doubaoImageModelOptionsSchema = z.object({\n  apiKey: z.string().optional(),\n  baseURL: z.string().optional(),\n  model: z.string().optional(),\n  modelOptions: z.object({}).optional(),\n  clientOptions: z.object({}).optional(),\n});\n\nexport class DoubaoImageModel extends ImageModel<DoubaoImageModelInput, DoubaoImageModelOutput> {\n  constructor(public override options?: DoubaoImageModelOptions) {\n    super({\n      ...options,\n      inputSchema: doubaoImageModelInputSchema,\n      description: options?.description ?? \"Draw or edit image by Doubao image models\",\n    });\n\n    if (options) checkArguments(this.name, doubaoImageModelOptionsSchema, options);\n  }\n\n  protected apiKeyEnvName = \"DOUBAO_API_KEY\";\n\n  override get credential() {\n    return {\n      url: this.options?.baseURL || process.env.DOUBAO_BASE_URL || DOUBAO_BASE_URL,\n      apiKey: this.options?.apiKey || process.env[this.apiKeyEnvName],\n      model: this.options?.model || DOUBAO_DEFAULT_IMAGE_MODEL,\n    };\n  }\n\n  get modelOptions() {\n    return this.options?.modelOptions;\n  }\n\n  private extractDataObjects(text: string) {\n    const dataObjects = [];\n    const lines = text.split(\"\\n\");\n\n    for (const line of lines) {\n      const trimmed = line.trim();\n\n      if (trimmed.startsWith(\"data:\")) {\n        const jsonPart = trimmed.slice(5).trim();\n        if (jsonPart === \"[DONE]\") continue;\n\n        try {\n          const obj = JSON.parse(jsonPart);\n          dataObjects.push(obj);\n        } catch (e) {\n          console.warn(\"Failed to parse JSON data object:\", jsonPart, e);\n        }\n      }\n    }\n\n    return dataObjects;\n  }\n\n  override async process(\n    input: DoubaoImageModelInput,\n    _options: AgentInvokeOptions,\n  ): Promise<ImageModelOutput> {\n    const model = input.modelOptions?.model || this.credential.model;\n    const { url, apiKey } = this.credential;\n    if (!apiKey) {\n      throw new Error(\n        `${this.name} requires an API key. Please provide it via \\`options.apiKey\\`, or set the \\`${this.apiKeyEnvName}\\` environment variable`,\n      );\n    }\n\n    const map: { [key: string]: string[] } = {\n      \"doubao-seedream-4\": [\n        \"model\",\n        \"prompt\",\n        \"size\",\n        \"sequentialImageGeneration\",\n        \"sequentialImageGenerationOptions\",\n        \"stream\",\n        \"responseFormat\",\n        \"watermark\",\n      ],\n      \"doubao-seedream-3-0-t2i\": [\n        \"model\",\n        \"prompt\",\n        \"size\",\n        \"seed\",\n        \"guidanceScale\",\n        \"responseFormat\",\n        \"watermark\",\n      ],\n      \"doubao-seededit-3-0-i2i\": [\n        \"model\",\n        \"prompt\",\n        \"size\",\n        \"seed\",\n        \"guidanceScale\",\n        \"responseFormat\",\n        \"watermark\",\n      ],\n    };\n\n    const key = Object.keys(map).find((key) => model.includes(key));\n    if (!key) {\n      throw new Error(`${this.name} only support ${Object.keys(map).join(\", \")}`);\n    }\n\n    if (!map[key]) {\n      throw new Error(`${this.name} only support ${Object.keys(map).join(\", \")}`);\n    }\n\n    const mergeInput = { ...this.modelOptions, ...input.modelOptions, ...input };\n\n    const image = await Promise.all(\n      flat(input.image).map((image) =>\n        this.transformFileType(\"file\", image).then(\n          (file) => `data:${file.mimeType || \"image/png\"};base64,${file.data}`,\n        ),\n      ),\n    );\n\n    const body = {\n      ...snakelize(pick(mergeInput, map[key])),\n      model,\n      response_format: \"b64_json\",\n      watermark: mergeInput.watermark ?? false,\n      image: image.length ? image : undefined,\n    };\n\n    const response = await fetch(joinURL(url, `/images/generations`), {\n      method: \"POST\",\n      headers: { Authorization: `Bearer ${apiKey}`, \"Content-Type\": \"application/json\" },\n      body: JSON.stringify(body),\n      timeout: this.options?.clientOptions?.timeout ?? 60 * 1000,\n    });\n\n    if (body.stream) {\n      if (!response.body) throw new Error(\"Streaming not supported in this environment\");\n      const reader = response.body.getReader();\n      const decoder = new TextDecoder();\n      let done = false;\n      let buffer = \"\";\n\n      while (!done) {\n        const { value, done: doneReading } = await reader.read();\n        done = doneReading;\n        if (value) {\n          buffer += decoder.decode(value);\n        }\n      }\n\n      const dataObjects = this.extractDataObjects(buffer);\n      const error = dataObjects.find((i) => i.type === \"image_generation.partial_failed\");\n      if (error) {\n        throw new Error(`Doubao API error: ${error.error.message}`);\n      }\n\n      const completed = dataObjects.find((i) => i.type === \"image_generation.completed\");\n\n      return {\n        images: dataObjects\n          .filter((i) => i.type === \"image_generation.partial_succeeded\")\n          .map<FileUnionContent>((i) => {\n            if (typeof i.url === \"string\")\n              return { type: \"url\", url: i.url, mimeType: OUTPUT_MIME_TYPE };\n            if (typeof i.b64_json === \"string\")\n              return { type: \"file\", data: i.b64_json, mimeType: OUTPUT_MIME_TYPE };\n            throw new Error(\"Image response does not contain a valid URL or base64 data\");\n          }),\n        usage: { inputTokens: 0, outputTokens: completed?.usage.output_tokens || 0 },\n        model: model,\n      };\n    }\n\n    const data: {\n      model: string;\n      usage?: { output_tokens?: number };\n      data: { url?: string; b64_json?: string }[];\n      error?: { message: string };\n    } = await response.json();\n\n    if (data.error) {\n      throw new Error(`Doubao API error: ${data.error.message}`);\n    }\n\n    return {\n      images: data.data.map<FileUnionContent>((item) => {\n        if (item.url) return { type: \"url\", url: item.url, mimeType: OUTPUT_MIME_TYPE };\n        if (item.b64_json) return { type: \"file\", data: item.b64_json, mimeType: OUTPUT_MIME_TYPE };\n        throw new Error(\"Image response does not contain a valid URL or base64 data\");\n      }),\n      usage: {\n        inputTokens: 0,\n        outputTokens: data?.usage?.output_tokens || 0,\n      },\n      model: data.model,\n    };\n  }\n}\n"],"mappings":";;;;;;;;AAeA,MAAM,6BAA6B;AACnC,MAAM,kBAAkB;AACxB,MAAM,mBAAmB;AAyBzB,MAAM,8BAA8B,sBAAsB,OAAO,EAAE,CAAC;AAEpE,MAAM,gCAAgC,EAAE,OAAO;CAC7C,QAAQ,EAAE,QAAQ,CAAC,UAAU;CAC7B,SAAS,EAAE,QAAQ,CAAC,UAAU;CAC9B,OAAO,EAAE,QAAQ,CAAC,UAAU;CAC5B,cAAc,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACrC,eAAe,EAAE,OAAO,EAAE,CAAC,CAAC,UAAU;CACvC,CAAC;AAEF,IAAa,mBAAb,cAAsC,WAA0D;CAC9F,YAAY,AAAgB,SAAmC;AAC7D,QAAM;GACJ,GAAG;GACH,aAAa;GACb,aAAa,SAAS,eAAe;GACtC,CAAC;EALwB;AAO1B,MAAI,QAAS,gBAAe,KAAK,MAAM,+BAA+B,QAAQ;;CAGhF,AAAU,gBAAgB;CAE1B,IAAa,aAAa;AACxB,SAAO;GACL,KAAK,KAAK,SAAS,WAAW,QAAQ,IAAI,mBAAmB;GAC7D,QAAQ,KAAK,SAAS,UAAU,QAAQ,IAAI,KAAK;GACjD,OAAO,KAAK,SAAS,SAAS;GAC/B;;CAGH,IAAI,eAAe;AACjB,SAAO,KAAK,SAAS;;CAGvB,AAAQ,mBAAmB,MAAc;EACvC,MAAM,cAAc,EAAE;EACtB,MAAM,QAAQ,KAAK,MAAM,KAAK;AAE9B,OAAK,MAAM,QAAQ,OAAO;GACxB,MAAM,UAAU,KAAK,MAAM;AAE3B,OAAI,QAAQ,WAAW,QAAQ,EAAE;IAC/B,MAAM,WAAW,QAAQ,MAAM,EAAE,CAAC,MAAM;AACxC,QAAI,aAAa,SAAU;AAE3B,QAAI;KACF,MAAM,MAAM,KAAK,MAAM,SAAS;AAChC,iBAAY,KAAK,IAAI;aACd,GAAG;AACV,aAAQ,KAAK,qCAAqC,UAAU,EAAE;;;;AAKpE,SAAO;;CAGT,MAAe,QACb,OACA,UAC2B;EAC3B,MAAM,QAAQ,MAAM,cAAc,SAAS,KAAK,WAAW;EAC3D,MAAM,EAAE,KAAK,WAAW,KAAK;AAC7B,MAAI,CAAC,OACH,OAAM,IAAI,MACR,GAAG,KAAK,KAAK,+EAA+E,KAAK,cAAc,yBAChH;EAGH,MAAM,MAAmC;GACvC,qBAAqB;IACnB;IACA;IACA;IACA;IACA;IACA;IACA;IACA;IACD;GACD,2BAA2B;IACzB;IACA;IACA;IACA;IACA;IACA;IACA;IACD;GACD,2BAA2B;IACzB;IACA;IACA;IACA;IACA;IACA;IACA;IACD;GACF;EAED,MAAM,MAAM,OAAO,KAAK,IAAI,CAAC,MAAM,UAAQ,MAAM,SAASA,MAAI,CAAC;AAC/D,MAAI,CAAC,IACH,OAAM,IAAI,MAAM,GAAG,KAAK,KAAK,gBAAgB,OAAO,KAAK,IAAI,CAAC,KAAK,KAAK,GAAG;AAG7E,MAAI,CAAC,IAAI,KACP,OAAM,IAAI,MAAM,GAAG,KAAK,KAAK,gBAAgB,OAAO,KAAK,IAAI,CAAC,KAAK,KAAK,GAAG;EAG7E,MAAM,aAAa;GAAE,GAAG,KAAK;GAAc,GAAG,MAAM;GAAc,GAAG;GAAO;EAE5E,MAAM,QAAQ,MAAM,QAAQ,IAC1B,KAAK,MAAM,MAAM,CAAC,KAAK,YACrB,KAAK,kBAAkB,QAAQC,QAAM,CAAC,MACnC,SAAS,QAAQ,KAAK,YAAY,YAAY,UAAU,KAAK,OAC/D,CACF,CACF;EAED,MAAM,OAAO;GACX,GAAG,UAAU,KAAK,YAAY,IAAI,KAAK,CAAC;GACxC;GACA,iBAAiB;GACjB,WAAW,WAAW,aAAa;GACnC,OAAO,MAAM,SAAS,QAAQ;GAC/B;EAED,MAAM,WAAW,MAAM,MAAM,QAAQ,KAAK,sBAAsB,EAAE;GAChE,QAAQ;GACR,SAAS;IAAE,eAAe,UAAU;IAAU,gBAAgB;IAAoB;GAClF,MAAM,KAAK,UAAU,KAAK;GAC1B,SAAS,KAAK,SAAS,eAAe,WAAW,KAAK;GACvD,CAAC;AAEF,MAAI,KAAK,QAAQ;AACf,OAAI,CAAC,SAAS,KAAM,OAAM,IAAI,MAAM,8CAA8C;GAClF,MAAM,SAAS,SAAS,KAAK,WAAW;GACxC,MAAM,UAAU,IAAI,aAAa;GACjC,IAAI,OAAO;GACX,IAAI,SAAS;AAEb,UAAO,CAAC,MAAM;IACZ,MAAM,EAAE,OAAO,MAAM,gBAAgB,MAAM,OAAO,MAAM;AACxD,WAAO;AACP,QAAI,MACF,WAAU,QAAQ,OAAO,MAAM;;GAInC,MAAM,cAAc,KAAK,mBAAmB,OAAO;GACnD,MAAM,QAAQ,YAAY,MAAM,MAAM,EAAE,SAAS,kCAAkC;AACnF,OAAI,MACF,OAAM,IAAI,MAAM,qBAAqB,MAAM,MAAM,UAAU;GAG7D,MAAM,YAAY,YAAY,MAAM,MAAM,EAAE,SAAS,6BAA6B;AAElF,UAAO;IACL,QAAQ,YACL,QAAQ,MAAM,EAAE,SAAS,qCAAqC,CAC9D,KAAuB,MAAM;AAC5B,SAAI,OAAO,EAAE,QAAQ,SACnB,QAAO;MAAE,MAAM;MAAO,KAAK,EAAE;MAAK,UAAU;MAAkB;AAChE,SAAI,OAAO,EAAE,aAAa,SACxB,QAAO;MAAE,MAAM;MAAQ,MAAM,EAAE;MAAU,UAAU;MAAkB;AACvE,WAAM,IAAI,MAAM,6DAA6D;MAC7E;IACJ,OAAO;KAAE,aAAa;KAAG,cAAc,WAAW,MAAM,iBAAiB;KAAG;IACrE;IACR;;EAGH,MAAM,OAKF,MAAM,SAAS,MAAM;AAEzB,MAAI,KAAK,MACP,OAAM,IAAI,MAAM,qBAAqB,KAAK,MAAM,UAAU;AAG5D,SAAO;GACL,QAAQ,KAAK,KAAK,KAAuB,SAAS;AAChD,QAAI,KAAK,IAAK,QAAO;KAAE,MAAM;KAAO,KAAK,KAAK;KAAK,UAAU;KAAkB;AAC/E,QAAI,KAAK,SAAU,QAAO;KAAE,MAAM;KAAQ,MAAM,KAAK;KAAU,UAAU;KAAkB;AAC3F,UAAM,IAAI,MAAM,6DAA6D;KAC7E;GACF,OAAO;IACL,aAAa;IACb,cAAc,MAAM,OAAO,iBAAiB;IAC7C;GACD,OAAO,KAAK;GACb"}

package/dist/index.cjs

@@ -0,0 +1,5 @@
+const require_doubao_chat_model = require('./doubao-chat-model.cjs');
+const require_doubao_image_model = require('./doubao-image-model.cjs');
+
+exports.DoubaoChatModel = require_doubao_chat_model.DoubaoChatModel;
+exports.DoubaoImageModel = require_doubao_image_model.DoubaoImageModel;

package/dist/index.d.cts

@@ -0,0 +1,3 @@
+import { DoubaoChatModel } from "./doubao-chat-model.cjs";
+import { DoubaoImageModel, DoubaoImageModelInput, DoubaoImageModelOptions, DoubaoImageModelOutput } from "./doubao-image-model.cjs";
+export { DoubaoChatModel, DoubaoImageModel, DoubaoImageModelInput, DoubaoImageModelOptions, DoubaoImageModelOutput };

package/dist/index.d.mts

@@ -0,0 +1,3 @@
+import { DoubaoChatModel } from "./doubao-chat-model.mjs";
+import { DoubaoImageModel, DoubaoImageModelInput, DoubaoImageModelOptions, DoubaoImageModelOutput } from "./doubao-image-model.mjs";
+export { DoubaoChatModel, DoubaoImageModel, DoubaoImageModelInput, DoubaoImageModelOptions, DoubaoImageModelOutput };

package/dist/index.mjs

@@ -0,0 +1,4 @@
+import { DoubaoChatModel } from "./doubao-chat-model.mjs";
+import { DoubaoImageModel } from "./doubao-image-model.mjs";
+
+export { DoubaoChatModel, DoubaoImageModel };