@aigne/aigne-hub 0.10.17-beta.1 → 1.74.0-beta

package/dist/utils/model.cjs

@@ -0,0 +1,260 @@
+const require_constants = require('./constants.cjs');
+const require_aigne_hub_image_model = require('../aigne-hub-image-model.cjs');
+const require_aigne_hub_model = require('../aigne-hub-model.cjs');
+const require_aigne_hub_video_model = require('../aigne-hub-video-model.cjs');
+let _aigne_anthropic = require("@aigne/anthropic");
+let _aigne_bedrock = require("@aigne/bedrock");
+let _aigne_deepseek = require("@aigne/deepseek");
+let _aigne_doubao = require("@aigne/doubao");
+let _aigne_gemini = require("@aigne/gemini");
+let _aigne_ideogram = require("@aigne/ideogram");
+let _aigne_ollama = require("@aigne/ollama");
+let _aigne_open_router = require("@aigne/open-router");
+let _aigne_openai = require("@aigne/openai");
+let _aigne_poe = require("@aigne/poe");
+let _aigne_xai = require("@aigne/xai");
+let _smithy_node_http_handler = require("@smithy/node-http-handler");
+let https_proxy_agent = require("https-proxy-agent");
+
+//#region src/utils/model.ts
+const getClientOptions = () => {
+	const proxy = [
+		"HTTPS_PROXY",
+		"https_proxy",
+		"HTTP_PROXY",
+		"http_proxy",
+		"ALL_PROXY",
+		"all_proxy"
+	].map((i) => process.env[i]).filter(Boolean)[0];
+	const httpAgent = proxy ? new https_proxy_agent.HttpsProxyAgent(proxy) : void 0;
+	return {
+		clientOptions: { fetchOptions: { agent: httpAgent } },
+		httpAgent
+	};
+};
+const GOOGLE = "google";
+function availableModels() {
+	const { clientOptions, httpAgent } = getClientOptions();
+	return [
+		{
+			name: _aigne_openai.OpenAIChatModel.name,
+			apiKeyEnvName: "OPENAI_API_KEY",
+			create: (params) => new _aigne_openai.OpenAIChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_anthropic.AnthropicChatModel.name,
+			apiKeyEnvName: "ANTHROPIC_API_KEY",
+			create: (params) => new _aigne_anthropic.AnthropicChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_bedrock.BedrockChatModel.name,
+			apiKeyEnvName: "AWS_ACCESS_KEY_ID",
+			create: (params) => new _aigne_bedrock.BedrockChatModel({
+				...params,
+				clientOptions: {
+					requestHandler: _smithy_node_http_handler.NodeHttpHandler.create({
+						httpAgent,
+						httpsAgent: httpAgent
+					}),
+					streamCollector: _smithy_node_http_handler.streamCollector
+				}
+			})
+		},
+		{
+			name: _aigne_deepseek.DeepSeekChatModel.name,
+			apiKeyEnvName: "DEEPSEEK_API_KEY",
+			create: (params) => new _aigne_deepseek.DeepSeekChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: [_aigne_gemini.GeminiChatModel.name, GOOGLE],
+			apiKeyEnvName: ["GEMINI_API_KEY", "GOOGLE_API_KEY"],
+			create: (params) => new _aigne_gemini.GeminiChatModel({
+				...params,
+				clientOptions: { httpOptions: clientOptions.fetchOptions }
+			})
+		},
+		{
+			name: _aigne_ollama.OllamaChatModel.name,
+			apiKeyEnvName: "OLLAMA_API_KEY",
+			create: (params) => new _aigne_ollama.OllamaChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_open_router.OpenRouterChatModel.name,
+			apiKeyEnvName: "OPEN_ROUTER_API_KEY",
+			create: (params) => new _aigne_open_router.OpenRouterChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_xai.XAIChatModel.name,
+			apiKeyEnvName: "XAI_API_KEY",
+			create: (params) => new _aigne_xai.XAIChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_doubao.DoubaoChatModel.name,
+			apiKeyEnvName: "DOUBAO_API_KEY",
+			create: (params) => new _aigne_doubao.DoubaoChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_poe.PoeChatModel.name,
+			apiKeyEnvName: "POE_API_KEY",
+			create: (params) => new _aigne_poe.PoeChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: require_aigne_hub_model.AIGNEHubChatModel.name,
+			apiKeyEnvName: "AIGNE_HUB_API_KEY",
+			create: (params) => new require_aigne_hub_model.AIGNEHubChatModel({
+				...params,
+				clientOptions
+			})
+		}
+	];
+}
+function availableImageModels() {
+	const { clientOptions } = getClientOptions();
+	return [
+		{
+			name: _aigne_openai.OpenAIImageModel.name,
+			apiKeyEnvName: "OPENAI_API_KEY",
+			create: (params) => new _aigne_openai.OpenAIImageModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: [_aigne_gemini.GeminiImageModel.name, GOOGLE],
+			apiKeyEnvName: "GEMINI_API_KEY",
+			create: (params) => new _aigne_gemini.GeminiImageModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: _aigne_ideogram.IdeogramImageModel.name,
+			apiKeyEnvName: "IDEOGRAM_API_KEY",
+			create: (params) => new _aigne_ideogram.IdeogramImageModel({ ...params })
+		},
+		{
+			name: _aigne_doubao.DoubaoImageModel.name,
+			apiKeyEnvName: "DOUBAO_API_KEY",
+			create: (params) => new _aigne_doubao.DoubaoImageModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: require_aigne_hub_image_model.AIGNEHubImageModel.name,
+			apiKeyEnvName: "AIGNE_HUB_API_KEY",
+			create: (params) => new require_aigne_hub_image_model.AIGNEHubImageModel({
+				...params,
+				clientOptions
+			})
+		}
+	];
+}
+function availableVideoModels() {
+	const { clientOptions } = getClientOptions();
+	return [
+		{
+			name: _aigne_openai.OpenAIVideoModel.name,
+			apiKeyEnvName: "OPENAI_API_KEY",
+			create: (params) => new _aigne_openai.OpenAIVideoModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: [_aigne_gemini.GeminiVideoModel.name, GOOGLE],
+			apiKeyEnvName: "GEMINI_API_KEY",
+			create: (params) => new _aigne_gemini.GeminiVideoModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: require_aigne_hub_video_model.AIGNEHubVideoModel.name,
+			apiKeyEnvName: "AIGNE_HUB_API_KEY",
+			create: (params) => new require_aigne_hub_video_model.AIGNEHubVideoModel({
+				...params,
+				clientOptions
+			})
+		}
+	];
+}
+function findModel(provider) {
+	provider = provider.toLowerCase().replace(/-/g, "");
+	const all = availableModels();
+	return {
+		all,
+		match: all.find((m) => {
+			if (typeof m.name === "string") return m.name.toLowerCase().includes(provider);
+			return m.name.some((n) => n.toLowerCase().includes(provider));
+		})
+	};
+}
+function findImageModel(provider) {
+	provider = provider.toLowerCase().replace(/-/g, "");
+	const all = availableImageModels();
+	return {
+		all,
+		match: all.find((m) => {
+			if (typeof m.name === "string") return m.name.toLowerCase().includes(provider);
+			return m.name.some((n) => n.toLowerCase().includes(provider));
+		})
+	};
+}
+function findVideoModel(provider) {
+	provider = provider.toLowerCase().replace(/-/g, "");
+	const all = availableVideoModels();
+	return {
+		all,
+		match: all.find((m) => {
+			if (typeof m.name === "string") return m.name.toLowerCase().includes(provider);
+			return m.name.some((n) => n.toLowerCase().includes(provider));
+		})
+	};
+}
+const parseModel = (model) => {
+	model = model.replace(/^([\w-]+):/, "$1/");
+	let { provider, name } = model.match(/(?<provider>[^/]*)(\/(?<name>.*))?/)?.groups ?? {};
+	provider = provider?.replace(/-/g, "");
+	if (provider ? findModel(provider)?.match : void 0) return {
+		provider,
+		model: name
+	};
+	return {
+		provider: require_constants.AIGNE_HUB_PROVIDER,
+		model
+	};
+};
+
+//#endregion
+exports.availableImageModels = availableImageModels;
+exports.availableModels = availableModels;
+exports.availableVideoModels = availableVideoModels;
+exports.findImageModel = findImageModel;
+exports.findModel = findModel;
+exports.findVideoModel = findVideoModel;
+exports.parseModel = parseModel;

package/dist/utils/model.d.cts

@@ -0,0 +1,57 @@
+import { ChatModel, ChatModelInputOptionsWithGetter, ImageModel, ImageModelInputOptionsWithGetter, VideoModel } from "@aigne/core";
+
+//#region src/utils/model.d.ts
+interface LoadableModel {
+  name: string | string[];
+  apiKeyEnvName?: string | string[];
+  create: (options: {
+    model?: string;
+    modelOptions?: ChatModelInputOptionsWithGetter;
+    apiKey?: string;
+    baseURL?: string;
+  }) => ChatModel;
+}
+declare function availableModels(): LoadableModel[];
+interface LoadableImageModel {
+  name: string | string[];
+  apiKeyEnvName: string;
+  create: (options: {
+    apiKey?: string;
+    baseURL?: string;
+    model?: string;
+    modelOptions?: ImageModelInputOptionsWithGetter;
+  }) => ImageModel;
+}
+interface LoadableVideoModel {
+  name: string | string[];
+  apiKeyEnvName: string;
+  create: (options: {
+    apiKey?: string;
+    baseURL?: string;
+    model?: string;
+    modelOptions?: {
+      [key: string]: any;
+    };
+  }) => VideoModel;
+}
+declare function availableImageModels(): LoadableImageModel[];
+declare function availableVideoModels(): LoadableVideoModel[];
+declare function findModel(provider: string): {
+  all: LoadableModel[];
+  match: LoadableModel | undefined;
+};
+declare function findImageModel(provider: string): {
+  all: LoadableImageModel[];
+  match: LoadableImageModel | undefined;
+};
+declare function findVideoModel(provider: string): {
+  all: LoadableVideoModel[];
+  match: LoadableVideoModel | undefined;
+};
+declare const parseModel: (model: string) => {
+  provider: string | undefined;
+  model: string | undefined;
+};
+//#endregion
+export { LoadableImageModel, LoadableModel, LoadableVideoModel, availableImageModels, availableModels, availableVideoModels, findImageModel, findModel, findVideoModel, parseModel };
+//# sourceMappingURL=model.d.cts.map

package/dist/utils/model.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.cts","names":[],"sources":["../../src/utils/model.ts"],"mappings":";;;UAuDiB,aAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA,GAAA,OAAA;IAAA,KAAA;IAAA,YAAA,GAKE,+BAAA;IAAA,MAAA;IAAA,OAAA;EAAA,MAGX,SAAA;AAAA;AAAA,iBAGQ,eAAA,CAAA,GAAmB,aAAA;AAAA,UAyElB,kBAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA,GAAA,OAAA;IAAA,MAAA;IAAA,OAAA;IAAA,KAAA;IAAA,YAAA,GAOE,gCAAA;EAAA,MACX,UAAA;AAAA;AAAA,UAGS,kBAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA,GAAA,OAAA;IAAA,MAAA;IAAA,OAAA;IAAA,KAAA;IAAA,YAAA;MAAA,CAAA,GAAA;IAAA;EAAA,MAQT,UAAA;AAAA;AAAA,iBAGQ,oBAAA,CAAA,GAAwB,kBAAA;AAAA,iBAgCxB,oBAAA,CAAA,GAAwB,kBAAA;AAAA,iBAsBxB,SAAA,CAAA,QAAA;EAAA,GAAA,EACT,aAAA;EAAA,KAAA,EACE,aAAA;AAAA;AAAA,iBAiBO,cAAA,CAAA,QAAA;EAAA,GAAA,EACT,kBAAA;EAAA,KAAA,EACE,kBAAA;AAAA;AAAA,iBAiBO,cAAA,CAAA,QAAA;EAAA,GAAA,EACT,kBAAA;EAAA,KAAA,EACE,kBAAA;AAAA;AAAA,cAiBI,UAAA,GAAA,KAAA;EAAA,QAAA;EAAA,KAAA;AAAA"}

package/dist/utils/model.d.mts

@@ -0,0 +1,57 @@
+import { ChatModel, ChatModelInputOptionsWithGetter, ImageModel, ImageModelInputOptionsWithGetter, VideoModel } from "@aigne/core";
+
+//#region src/utils/model.d.ts
+interface LoadableModel {
+  name: string | string[];
+  apiKeyEnvName?: string | string[];
+  create: (options: {
+    model?: string;
+    modelOptions?: ChatModelInputOptionsWithGetter;
+    apiKey?: string;
+    baseURL?: string;
+  }) => ChatModel;
+}
+declare function availableModels(): LoadableModel[];
+interface LoadableImageModel {
+  name: string | string[];
+  apiKeyEnvName: string;
+  create: (options: {
+    apiKey?: string;
+    baseURL?: string;
+    model?: string;
+    modelOptions?: ImageModelInputOptionsWithGetter;
+  }) => ImageModel;
+}
+interface LoadableVideoModel {
+  name: string | string[];
+  apiKeyEnvName: string;
+  create: (options: {
+    apiKey?: string;
+    baseURL?: string;
+    model?: string;
+    modelOptions?: {
+      [key: string]: any;
+    };
+  }) => VideoModel;
+}
+declare function availableImageModels(): LoadableImageModel[];
+declare function availableVideoModels(): LoadableVideoModel[];
+declare function findModel(provider: string): {
+  all: LoadableModel[];
+  match: LoadableModel | undefined;
+};
+declare function findImageModel(provider: string): {
+  all: LoadableImageModel[];
+  match: LoadableImageModel | undefined;
+};
+declare function findVideoModel(provider: string): {
+  all: LoadableVideoModel[];
+  match: LoadableVideoModel | undefined;
+};
+declare const parseModel: (model: string) => {
+  provider: string | undefined;
+  model: string | undefined;
+};
+//#endregion
+export { LoadableImageModel, LoadableModel, LoadableVideoModel, availableImageModels, availableModels, availableVideoModels, findImageModel, findModel, findVideoModel, parseModel };
+//# sourceMappingURL=model.d.mts.map

package/dist/utils/model.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.mts","names":[],"sources":["../../src/utils/model.ts"],"mappings":";;;UAuDiB,aAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA,GAAA,OAAA;IAAA,KAAA;IAAA,YAAA,GAKE,+BAAA;IAAA,MAAA;IAAA,OAAA;EAAA,MAGX,SAAA;AAAA;AAAA,iBAGQ,eAAA,CAAA,GAAmB,aAAA;AAAA,UAyElB,kBAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA,GAAA,OAAA;IAAA,MAAA;IAAA,OAAA;IAAA,KAAA;IAAA,YAAA,GAOE,gCAAA;EAAA,MACX,UAAA;AAAA;AAAA,UAGS,kBAAA;EAAA,IAAA;EAAA,aAAA;EAAA,MAAA,GAAA,OAAA;IAAA,MAAA;IAAA,OAAA;IAAA,KAAA;IAAA,YAAA;MAAA,CAAA,GAAA;IAAA;EAAA,MAQT,UAAA;AAAA;AAAA,iBAGQ,oBAAA,CAAA,GAAwB,kBAAA;AAAA,iBAgCxB,oBAAA,CAAA,GAAwB,kBAAA;AAAA,iBAsBxB,SAAA,CAAA,QAAA;EAAA,GAAA,EACT,aAAA;EAAA,KAAA,EACE,aAAA;AAAA;AAAA,iBAiBO,cAAA,CAAA,QAAA;EAAA,GAAA,EACT,kBAAA;EAAA,KAAA,EACE,kBAAA;AAAA;AAAA,iBAiBO,cAAA,CAAA,QAAA;EAAA,GAAA,EACT,kBAAA;EAAA,KAAA,EACE,kBAAA;AAAA;AAAA,cAiBI,UAAA,GAAA,KAAA;EAAA,QAAA;EAAA,KAAA;AAAA"}

package/dist/utils/model.mjs

@@ -0,0 +1,255 @@
+import { AIGNE_HUB_PROVIDER } from "./constants.mjs";
+import { AIGNEHubImageModel } from "../aigne-hub-image-model.mjs";
+import { AIGNEHubChatModel } from "../aigne-hub-model.mjs";
+import { AIGNEHubVideoModel } from "../aigne-hub-video-model.mjs";
+import { AnthropicChatModel } from "@aigne/anthropic";
+import { BedrockChatModel } from "@aigne/bedrock";
+import { DeepSeekChatModel } from "@aigne/deepseek";
+import { DoubaoChatModel, DoubaoImageModel } from "@aigne/doubao";
+import { GeminiChatModel, GeminiImageModel, GeminiVideoModel } from "@aigne/gemini";
+import { IdeogramImageModel } from "@aigne/ideogram";
+import { OllamaChatModel } from "@aigne/ollama";
+import { OpenRouterChatModel } from "@aigne/open-router";
+import { OpenAIChatModel, OpenAIImageModel, OpenAIVideoModel } from "@aigne/openai";
+import { PoeChatModel } from "@aigne/poe";
+import { XAIChatModel } from "@aigne/xai";
+import { NodeHttpHandler, streamCollector } from "@smithy/node-http-handler";
+import { HttpsProxyAgent } from "https-proxy-agent";
+
+//#region src/utils/model.ts
+const getClientOptions = () => {
+	const proxy = [
+		"HTTPS_PROXY",
+		"https_proxy",
+		"HTTP_PROXY",
+		"http_proxy",
+		"ALL_PROXY",
+		"all_proxy"
+	].map((i) => process.env[i]).filter(Boolean)[0];
+	const httpAgent = proxy ? new HttpsProxyAgent(proxy) : void 0;
+	return {
+		clientOptions: { fetchOptions: { agent: httpAgent } },
+		httpAgent
+	};
+};
+const GOOGLE = "google";
+function availableModels() {
+	const { clientOptions, httpAgent } = getClientOptions();
+	return [
+		{
+			name: OpenAIChatModel.name,
+			apiKeyEnvName: "OPENAI_API_KEY",
+			create: (params) => new OpenAIChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: AnthropicChatModel.name,
+			apiKeyEnvName: "ANTHROPIC_API_KEY",
+			create: (params) => new AnthropicChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: BedrockChatModel.name,
+			apiKeyEnvName: "AWS_ACCESS_KEY_ID",
+			create: (params) => new BedrockChatModel({
+				...params,
+				clientOptions: {
+					requestHandler: NodeHttpHandler.create({
+						httpAgent,
+						httpsAgent: httpAgent
+					}),
+					streamCollector
+				}
+			})
+		},
+		{
+			name: DeepSeekChatModel.name,
+			apiKeyEnvName: "DEEPSEEK_API_KEY",
+			create: (params) => new DeepSeekChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: [GeminiChatModel.name, GOOGLE],
+			apiKeyEnvName: ["GEMINI_API_KEY", "GOOGLE_API_KEY"],
+			create: (params) => new GeminiChatModel({
+				...params,
+				clientOptions: { httpOptions: clientOptions.fetchOptions }
+			})
+		},
+		{
+			name: OllamaChatModel.name,
+			apiKeyEnvName: "OLLAMA_API_KEY",
+			create: (params) => new OllamaChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: OpenRouterChatModel.name,
+			apiKeyEnvName: "OPEN_ROUTER_API_KEY",
+			create: (params) => new OpenRouterChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: XAIChatModel.name,
+			apiKeyEnvName: "XAI_API_KEY",
+			create: (params) => new XAIChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: DoubaoChatModel.name,
+			apiKeyEnvName: "DOUBAO_API_KEY",
+			create: (params) => new DoubaoChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: PoeChatModel.name,
+			apiKeyEnvName: "POE_API_KEY",
+			create: (params) => new PoeChatModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: AIGNEHubChatModel.name,
+			apiKeyEnvName: "AIGNE_HUB_API_KEY",
+			create: (params) => new AIGNEHubChatModel({
+				...params,
+				clientOptions
+			})
+		}
+	];
+}
+function availableImageModels() {
+	const { clientOptions } = getClientOptions();
+	return [
+		{
+			name: OpenAIImageModel.name,
+			apiKeyEnvName: "OPENAI_API_KEY",
+			create: (params) => new OpenAIImageModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: [GeminiImageModel.name, GOOGLE],
+			apiKeyEnvName: "GEMINI_API_KEY",
+			create: (params) => new GeminiImageModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: IdeogramImageModel.name,
+			apiKeyEnvName: "IDEOGRAM_API_KEY",
+			create: (params) => new IdeogramImageModel({ ...params })
+		},
+		{
+			name: DoubaoImageModel.name,
+			apiKeyEnvName: "DOUBAO_API_KEY",
+			create: (params) => new DoubaoImageModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: AIGNEHubImageModel.name,
+			apiKeyEnvName: "AIGNE_HUB_API_KEY",
+			create: (params) => new AIGNEHubImageModel({
+				...params,
+				clientOptions
+			})
+		}
+	];
+}
+function availableVideoModels() {
+	const { clientOptions } = getClientOptions();
+	return [
+		{
+			name: OpenAIVideoModel.name,
+			apiKeyEnvName: "OPENAI_API_KEY",
+			create: (params) => new OpenAIVideoModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: [GeminiVideoModel.name, GOOGLE],
+			apiKeyEnvName: "GEMINI_API_KEY",
+			create: (params) => new GeminiVideoModel({
+				...params,
+				clientOptions
+			})
+		},
+		{
+			name: AIGNEHubVideoModel.name,
+			apiKeyEnvName: "AIGNE_HUB_API_KEY",
+			create: (params) => new AIGNEHubVideoModel({
+				...params,
+				clientOptions
+			})
+		}
+	];
+}
+function findModel(provider) {
+	provider = provider.toLowerCase().replace(/-/g, "");
+	const all = availableModels();
+	return {
+		all,
+		match: all.find((m) => {
+			if (typeof m.name === "string") return m.name.toLowerCase().includes(provider);
+			return m.name.some((n) => n.toLowerCase().includes(provider));
+		})
+	};
+}
+function findImageModel(provider) {
+	provider = provider.toLowerCase().replace(/-/g, "");
+	const all = availableImageModels();
+	return {
+		all,
+		match: all.find((m) => {
+			if (typeof m.name === "string") return m.name.toLowerCase().includes(provider);
+			return m.name.some((n) => n.toLowerCase().includes(provider));
+		})
+	};
+}
+function findVideoModel(provider) {
+	provider = provider.toLowerCase().replace(/-/g, "");
+	const all = availableVideoModels();
+	return {
+		all,
+		match: all.find((m) => {
+			if (typeof m.name === "string") return m.name.toLowerCase().includes(provider);
+			return m.name.some((n) => n.toLowerCase().includes(provider));
+		})
+	};
+}
+const parseModel = (model) => {
+	model = model.replace(/^([\w-]+):/, "$1/");
+	let { provider, name } = model.match(/(?<provider>[^/]*)(\/(?<name>.*))?/)?.groups ?? {};
+	provider = provider?.replace(/-/g, "");
+	if (provider ? findModel(provider)?.match : void 0) return {
+		provider,
+		model: name
+	};
+	return {
+		provider: AIGNE_HUB_PROVIDER,
+		model
+	};
+};
+
+//#endregion
+export { availableImageModels, availableModels, availableVideoModels, findImageModel, findModel, findVideoModel, parseModel };
+//# sourceMappingURL=model.mjs.map

package/dist/utils/model.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.mjs","names":[],"sources":["../../src/utils/model.ts"],"sourcesContent":["import type { Agent } from \"node:https\";\nimport { AnthropicChatModel } from \"@aigne/anthropic\";\nimport { BedrockChatModel } from \"@aigne/bedrock\";\nimport type {\n  ChatModel,\n  ChatModelInputOptionsWithGetter,\n  ImageModel,\n  ImageModelInputOptionsWithGetter,\n  VideoModel,\n} from \"@aigne/core\";\nimport { DeepSeekChatModel } from \"@aigne/deepseek\";\nimport { DoubaoChatModel, DoubaoImageModel } from \"@aigne/doubao\";\nimport { GeminiChatModel, GeminiImageModel, GeminiVideoModel } from \"@aigne/gemini\";\nimport { IdeogramImageModel } from \"@aigne/ideogram\";\nimport { OllamaChatModel } from \"@aigne/ollama\";\nimport { OpenRouterChatModel } from \"@aigne/open-router\";\nimport {\n  OpenAIChatModel,\n  type OpenAIChatModelOptions,\n  OpenAIImageModel,\n  OpenAIVideoModel,\n} from \"@aigne/openai\";\nimport { PoeChatModel } from \"@aigne/poe\";\nimport { XAIChatModel } from \"@aigne/xai\";\nimport { NodeHttpHandler, streamCollector } from \"@smithy/node-http-handler\";\nimport { HttpsProxyAgent } from \"https-proxy-agent\";\nimport { AIGNEHubImageModel } from \"../aigne-hub-image-model.js\";\nimport { AIGNEHubChatModel } from \"../aigne-hub-model.js\";\nimport { AIGNEHubVideoModel } from \"../aigne-hub-video-model.js\";\nimport { AIGNE_HUB_PROVIDER } from \"./constants.js\";\n\nconst getClientOptions = () => {\n  const proxy = [\"HTTPS_PROXY\", \"https_proxy\", \"HTTP_PROXY\", \"http_proxy\", \"ALL_PROXY\", \"all_proxy\"]\n    .map((i) => process.env[i])\n    .filter(Boolean)[0];\n\n  const httpAgent = proxy ? (new HttpsProxyAgent(proxy) as Agent) : undefined;\n  const clientOptions: Pick<\n    NonNullable<OpenAIChatModelOptions[\"clientOptions\"]>,\n    \"fetchOptions\"\n  > = {\n    fetchOptions: {\n      // @ts-ignore\n      agent: httpAgent,\n    },\n  };\n\n  return {\n    clientOptions,\n    httpAgent,\n  };\n};\n\nconst GOOGLE = \"google\";\n\nexport interface LoadableModel {\n  name: string | string[];\n  apiKeyEnvName?: string | string[];\n  create: (options: {\n    model?: string;\n    modelOptions?: ChatModelInputOptionsWithGetter;\n    apiKey?: string;\n    baseURL?: string;\n  }) => ChatModel;\n}\n\nexport function availableModels(): LoadableModel[] {\n  const { clientOptions, httpAgent } = getClientOptions();\n\n  return [\n    {\n      name: OpenAIChatModel.name,\n      apiKeyEnvName: \"OPENAI_API_KEY\",\n      create: (params) => new OpenAIChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: AnthropicChatModel.name,\n      apiKeyEnvName: \"ANTHROPIC_API_KEY\",\n      create: (params) => new AnthropicChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: BedrockChatModel.name,\n      apiKeyEnvName: \"AWS_ACCESS_KEY_ID\",\n      create: (params) =>\n        new BedrockChatModel({\n          ...params,\n          clientOptions: {\n            requestHandler: NodeHttpHandler.create({ httpAgent, httpsAgent: httpAgent }),\n            streamCollector,\n          },\n        }),\n    },\n    {\n      name: DeepSeekChatModel.name,\n      apiKeyEnvName: \"DEEPSEEK_API_KEY\",\n      create: (params) => new DeepSeekChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: [GeminiChatModel.name, GOOGLE],\n      apiKeyEnvName: [\"GEMINI_API_KEY\", \"GOOGLE_API_KEY\"],\n      create: (params) =>\n        new GeminiChatModel({\n          ...params,\n          clientOptions: { httpOptions: clientOptions.fetchOptions },\n        }),\n    },\n    {\n      name: OllamaChatModel.name,\n      apiKeyEnvName: \"OLLAMA_API_KEY\",\n      create: (params) => new OllamaChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: OpenRouterChatModel.name,\n      apiKeyEnvName: \"OPEN_ROUTER_API_KEY\",\n      create: (params) => new OpenRouterChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: XAIChatModel.name,\n      apiKeyEnvName: \"XAI_API_KEY\",\n      create: (params) => new XAIChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: DoubaoChatModel.name,\n      apiKeyEnvName: \"DOUBAO_API_KEY\",\n      create: (params) => new DoubaoChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: PoeChatModel.name,\n      apiKeyEnvName: \"POE_API_KEY\",\n      create: (params) => new PoeChatModel({ ...params, clientOptions }),\n    },\n    {\n      name: AIGNEHubChatModel.name,\n      apiKeyEnvName: \"AIGNE_HUB_API_KEY\",\n      create: (params) => new AIGNEHubChatModel({ ...params, clientOptions }),\n    },\n  ];\n}\n\nexport interface LoadableImageModel {\n  name: string | string[];\n  apiKeyEnvName: string;\n  create: (options: {\n    apiKey?: string;\n    baseURL?: string;\n    model?: string;\n    modelOptions?: ImageModelInputOptionsWithGetter;\n  }) => ImageModel;\n}\n\nexport interface LoadableVideoModel {\n  name: string | string[];\n  apiKeyEnvName: string;\n  create: (options: {\n    apiKey?: string;\n    baseURL?: string;\n    model?: string;\n    modelOptions?: { [key: string]: any };\n  }) => VideoModel;\n}\n\nexport function availableImageModels(): LoadableImageModel[] {\n  const { clientOptions } = getClientOptions();\n\n  return [\n    {\n      name: OpenAIImageModel.name,\n      apiKeyEnvName: \"OPENAI_API_KEY\",\n      create: (params) => new OpenAIImageModel({ ...params, clientOptions }),\n    },\n    {\n      name: [GeminiImageModel.name, GOOGLE],\n      apiKeyEnvName: \"GEMINI_API_KEY\",\n      create: (params) => new GeminiImageModel({ ...params, clientOptions }),\n    },\n    {\n      name: IdeogramImageModel.name,\n      apiKeyEnvName: \"IDEOGRAM_API_KEY\",\n      create: (params) => new IdeogramImageModel({ ...params }),\n    },\n    {\n      name: DoubaoImageModel.name,\n      apiKeyEnvName: \"DOUBAO_API_KEY\",\n      create: (params) => new DoubaoImageModel({ ...params, clientOptions }),\n    },\n    {\n      name: AIGNEHubImageModel.name,\n      apiKeyEnvName: \"AIGNE_HUB_API_KEY\",\n      create: (params) => new AIGNEHubImageModel({ ...params, clientOptions }),\n    },\n  ];\n}\n\nexport function availableVideoModels(): LoadableVideoModel[] {\n  const { clientOptions } = getClientOptions();\n\n  return [\n    {\n      name: OpenAIVideoModel.name,\n      apiKeyEnvName: \"OPENAI_API_KEY\",\n      create: (params) => new OpenAIVideoModel({ ...params, clientOptions }) as VideoModel,\n    },\n    {\n      name: [GeminiVideoModel.name, GOOGLE],\n      apiKeyEnvName: \"GEMINI_API_KEY\",\n      create: (params) => new GeminiVideoModel({ ...params, clientOptions }) as VideoModel,\n    },\n    {\n      name: AIGNEHubVideoModel.name,\n      apiKeyEnvName: \"AIGNE_HUB_API_KEY\",\n      create: (params) => new AIGNEHubVideoModel({ ...params, clientOptions }) as VideoModel,\n    },\n  ];\n}\n\nexport function findModel(provider: string): {\n  all: LoadableModel[];\n  match: LoadableModel | undefined;\n} {\n  provider = provider.toLowerCase().replace(/-/g, \"\");\n\n  const all = availableModels();\n\n  const match = all.find((m) => {\n    if (typeof m.name === \"string\") {\n      return m.name.toLowerCase().includes(provider);\n    }\n\n    return m.name.some((n) => n.toLowerCase().includes(provider));\n  });\n\n  return { all, match };\n}\n\nexport function findImageModel(provider: string): {\n  all: LoadableImageModel[];\n  match: LoadableImageModel | undefined;\n} {\n  provider = provider.toLowerCase().replace(/-/g, \"\");\n\n  const all = availableImageModels();\n\n  const match = all.find((m) => {\n    if (typeof m.name === \"string\") {\n      return m.name.toLowerCase().includes(provider);\n    }\n\n    return m.name.some((n) => n.toLowerCase().includes(provider));\n  });\n\n  return { all, match };\n}\n\nexport function findVideoModel(provider: string): {\n  all: LoadableVideoModel[];\n  match: LoadableVideoModel | undefined;\n} {\n  provider = provider.toLowerCase().replace(/-/g, \"\");\n\n  const all = availableVideoModels();\n\n  const match = all.find((m) => {\n    if (typeof m.name === \"string\") {\n      return m.name.toLowerCase().includes(provider);\n    }\n\n    return m.name.some((n) => n.toLowerCase().includes(provider));\n  });\n\n  return { all, match };\n}\n\nexport const parseModel = (model: string) => {\n  // replace first ':' with '/' to compatible with `provider:model-name` format\n  model = model.replace(/^([\\w-]+):/, \"$1/\");\n  let { provider, name } = model.match(/(?<provider>[^/]*)(\\/(?<name>.*))?/)?.groups ?? {};\n  provider = provider?.replace(/-/g, \"\");\n  const match = provider ? findModel(provider)?.match : undefined;\n\n  if (match) {\n    return { provider, model: name };\n  }\n\n  return { provider: AIGNE_HUB_PROVIDER, model };\n};\n"],"mappings":";;;;;;;;;;;;;;;;;;;AA+BA,MAAM,yBAAyB;CAC7B,MAAM,QAAQ;EAAC;EAAe;EAAe;EAAc;EAAc;EAAa;EAAY,CAC/F,KAAK,MAAM,QAAQ,IAAI,GAAG,CAC1B,OAAO,QAAQ,CAAC;CAEnB,MAAM,YAAY,QAAS,IAAI,gBAAgB,MAAM,GAAa;AAWlE,QAAO;EACL,eARE,EACF,cAAc,EAEZ,OAAO,WACR,EACF;EAIC;EACD;;AAGH,MAAM,SAAS;AAaf,SAAgB,kBAAmC;CACjD,MAAM,EAAE,eAAe,cAAc,kBAAkB;AAEvD,QAAO;EACL;GACE,MAAM,gBAAgB;GACtB,eAAe;GACf,SAAS,WAAW,IAAI,gBAAgB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACtE;EACD;GACE,MAAM,mBAAmB;GACzB,eAAe;GACf,SAAS,WAAW,IAAI,mBAAmB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACzE;EACD;GACE,MAAM,iBAAiB;GACvB,eAAe;GACf,SAAS,WACP,IAAI,iBAAiB;IACnB,GAAG;IACH,eAAe;KACb,gBAAgB,gBAAgB,OAAO;MAAE;MAAW,YAAY;MAAW,CAAC;KAC5E;KACD;IACF,CAAC;GACL;EACD;GACE,MAAM,kBAAkB;GACxB,eAAe;GACf,SAAS,WAAW,IAAI,kBAAkB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACxE;EACD;GACE,MAAM,CAAC,gBAAgB,MAAM,OAAO;GACpC,eAAe,CAAC,kBAAkB,iBAAiB;GACnD,SAAS,WACP,IAAI,gBAAgB;IAClB,GAAG;IACH,eAAe,EAAE,aAAa,cAAc,cAAc;IAC3D,CAAC;GACL;EACD;GACE,MAAM,gBAAgB;GACtB,eAAe;GACf,SAAS,WAAW,IAAI,gBAAgB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACtE;EACD;GACE,MAAM,oBAAoB;GAC1B,eAAe;GACf,SAAS,WAAW,IAAI,oBAAoB;IAAE,GAAG;IAAQ;IAAe,CAAC;GAC1E;EACD;GACE,MAAM,aAAa;GACnB,eAAe;GACf,SAAS,WAAW,IAAI,aAAa;IAAE,GAAG;IAAQ;IAAe,CAAC;GACnE;EACD;GACE,MAAM,gBAAgB;GACtB,eAAe;GACf,SAAS,WAAW,IAAI,gBAAgB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACtE;EACD;GACE,MAAM,aAAa;GACnB,eAAe;GACf,SAAS,WAAW,IAAI,aAAa;IAAE,GAAG;IAAQ;IAAe,CAAC;GACnE;EACD;GACE,MAAM,kBAAkB;GACxB,eAAe;GACf,SAAS,WAAW,IAAI,kBAAkB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACxE;EACF;;AAyBH,SAAgB,uBAA6C;CAC3D,MAAM,EAAE,kBAAkB,kBAAkB;AAE5C,QAAO;EACL;GACE,MAAM,iBAAiB;GACvB,eAAe;GACf,SAAS,WAAW,IAAI,iBAAiB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACvE;EACD;GACE,MAAM,CAAC,iBAAiB,MAAM,OAAO;GACrC,eAAe;GACf,SAAS,WAAW,IAAI,iBAAiB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACvE;EACD;GACE,MAAM,mBAAmB;GACzB,eAAe;GACf,SAAS,WAAW,IAAI,mBAAmB,EAAE,GAAG,QAAQ,CAAC;GAC1D;EACD;GACE,MAAM,iBAAiB;GACvB,eAAe;GACf,SAAS,WAAW,IAAI,iBAAiB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACvE;EACD;GACE,MAAM,mBAAmB;GACzB,eAAe;GACf,SAAS,WAAW,IAAI,mBAAmB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACzE;EACF;;AAGH,SAAgB,uBAA6C;CAC3D,MAAM,EAAE,kBAAkB,kBAAkB;AAE5C,QAAO;EACL;GACE,MAAM,iBAAiB;GACvB,eAAe;GACf,SAAS,WAAW,IAAI,iBAAiB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACvE;EACD;GACE,MAAM,CAAC,iBAAiB,MAAM,OAAO;GACrC,eAAe;GACf,SAAS,WAAW,IAAI,iBAAiB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACvE;EACD;GACE,MAAM,mBAAmB;GACzB,eAAe;GACf,SAAS,WAAW,IAAI,mBAAmB;IAAE,GAAG;IAAQ;IAAe,CAAC;GACzE;EACF;;AAGH,SAAgB,UAAU,UAGxB;AACA,YAAW,SAAS,aAAa,CAAC,QAAQ,MAAM,GAAG;CAEnD,MAAM,MAAM,iBAAiB;AAU7B,QAAO;EAAE;EAAK,OARA,IAAI,MAAM,MAAM;AAC5B,OAAI,OAAO,EAAE,SAAS,SACpB,QAAO,EAAE,KAAK,aAAa,CAAC,SAAS,SAAS;AAGhD,UAAO,EAAE,KAAK,MAAM,MAAM,EAAE,aAAa,CAAC,SAAS,SAAS,CAAC;IAC7D;EAEmB;;AAGvB,SAAgB,eAAe,UAG7B;AACA,YAAW,SAAS,aAAa,CAAC,QAAQ,MAAM,GAAG;CAEnD,MAAM,MAAM,sBAAsB;AAUlC,QAAO;EAAE;EAAK,OARA,IAAI,MAAM,MAAM;AAC5B,OAAI,OAAO,EAAE,SAAS,SACpB,QAAO,EAAE,KAAK,aAAa,CAAC,SAAS,SAAS;AAGhD,UAAO,EAAE,KAAK,MAAM,MAAM,EAAE,aAAa,CAAC,SAAS,SAAS,CAAC;IAC7D;EAEmB;;AAGvB,SAAgB,eAAe,UAG7B;AACA,YAAW,SAAS,aAAa,CAAC,QAAQ,MAAM,GAAG;CAEnD,MAAM,MAAM,sBAAsB;AAUlC,QAAO;EAAE;EAAK,OARA,IAAI,MAAM,MAAM;AAC5B,OAAI,OAAO,EAAE,SAAS,SACpB,QAAO,EAAE,KAAK,aAAa,CAAC,SAAS,SAAS;AAGhD,UAAO,EAAE,KAAK,MAAM,MAAM,EAAE,aAAa,CAAC,SAAS,SAAS,CAAC;IAC7D;EAEmB;;AAGvB,MAAa,cAAc,UAAkB;AAE3C,SAAQ,MAAM,QAAQ,cAAc,MAAM;CAC1C,IAAI,EAAE,UAAU,SAAS,MAAM,MAAM,qCAAqC,EAAE,UAAU,EAAE;AACxF,YAAW,UAAU,QAAQ,MAAM,GAAG;AAGtC,KAFc,WAAW,UAAU,SAAS,EAAE,QAAQ,OAGpD,QAAO;EAAE;EAAU,OAAO;EAAM;AAGlC,QAAO;EAAE,UAAU;EAAoB;EAAO"}

package/dist/utils/type.cjs

@@ -0,0 +1,20 @@
+let zod = require("zod");
+
+//#region src/utils/type.ts
+const aigneHubModelOptionsSchema = zod.z.object({
+	baseURL: zod.z.string().optional(),
+	apiKey: zod.z.string().optional(),
+	model: zod.z.string().optional(),
+	modelOptions: zod.z.object({
+		model: zod.z.string().optional(),
+		temperature: zod.z.number().optional(),
+		topP: zod.z.number().optional(),
+		frequencyPenalty: zod.z.number().optional(),
+		presencePenalty: zod.z.number().optional(),
+		parallelToolCalls: zod.z.boolean().optional().default(true)
+	}).optional(),
+	clientOptions: zod.z.object({}).optional()
+});
+
+//#endregion
+exports.aigneHubModelOptionsSchema = aigneHubModelOptionsSchema;