@aigne/cli 1.60.0-beta → 1.74.0-beta

package/dist/utils/aigne-hub/credential.mjs

@@ -0,0 +1,175 @@
+import { ACCESS_KEY_SESSION_API, WELLKNOWN_SERVICE_PATH_PREFIX, isTest } from "./constants.mjs";
+import { decrypt, encodeEncryptionKey } from "./crypto.mjs";
+import store_default from "./store/index.mjs";
+import { existsSync, mkdirSync } from "node:fs";
+import { join } from "node:path";
+import { fetch } from "@aigne/core/utils/fetch";
+import { joinURL, withQuery } from "ufo";
+import { AIGNE_HUB_BLOCKLET_DID, AIGNE_HUB_URL, getAIGNEHubMountPoint } from "@aigne/aigne-hub";
+import inquirer from "inquirer";
+import { homedir } from "node:os";
+import open from "open";
+import pWaitFor from "p-wait-for";
+
+//#region src/utils/aigne-hub/credential.ts
+const request = async (config) => {
+	const headers = {};
+	if (config.requestCount !== void 0) headers["X-Request-Count"] = config.requestCount.toString();
+	return { data: await (await fetch(config.url, {
+		method: config.method || "GET",
+		headers
+	})).json() };
+};
+const fetchConfigs = async ({ connectUrl, sessionId, fetchInterval, fetchTimeout }) => {
+	const sessionURL = withQuery(joinURL(connectUrl, ACCESS_KEY_SESSION_API), { sid: sessionId });
+	let requestCount = 0;
+	const condition = async () => {
+		const { data: session$1 } = await request({
+			url: sessionURL,
+			requestCount
+		});
+		requestCount++;
+		return Boolean(session$1.accessKeyId && session$1.accessKeySecret);
+	};
+	await pWaitFor(condition, {
+		interval: fetchInterval,
+		timeout: fetchTimeout
+	});
+	const { data: session } = await request({
+		url: sessionURL,
+		requestCount
+	});
+	await request({
+		url: sessionURL,
+		method: "DELETE"
+	});
+	return {
+		...session,
+		accessKeyId: session.accessKeyId,
+		accessKeySecret: decrypt(session.accessKeySecret, session.accessKeyId, session.challenge)
+	};
+};
+function baseWrapSpinner(_, waiting) {
+	return waiting();
+}
+async function createConnect({ connectUrl, openPage, fetchInterval = 3 * 1e3, retry = 1500, source = "Blocklet CLI", connectAction = "connect-cli", wrapSpinner = baseWrapSpinner, closeOnSuccess, intervalFetchConfig, appName = "AIGNE CLI", appLogo = "https://www.aigne.io/favicon.ico?imageFilter=resize&w=32" }) {
+	const { data: session } = await request({
+		url: joinURL(connectUrl, ACCESS_KEY_SESSION_API),
+		method: "POST"
+	});
+	const token = session.id;
+	const pageUrl = withQuery(joinURL(connectUrl, connectAction), {
+		__token__: encodeEncryptionKey(token),
+		source,
+		closeOnSuccess,
+		cli: true,
+		appName: ` ${appName}`,
+		appLogo
+	});
+	openPage?.(pageUrl);
+	return await wrapSpinner(`Waiting for connection: ${connectUrl}`, async () => {
+		return await (intervalFetchConfig ?? fetchConfigs)({
+			connectUrl,
+			sessionId: token,
+			fetchTimeout: retry * fetchInterval,
+			fetchInterval: retry
+		});
+	});
+}
+async function connectToAIGNEHub(url) {
+	const { origin } = new URL(url);
+	const connectUrl = joinURL(origin, WELLKNOWN_SERVICE_PATH_PREFIX);
+	const apiUrl = await getAIGNEHubMountPoint(url, AIGNE_HUB_BLOCKLET_DID);
+	const openFn = isTest ? () => {} : open;
+	const accessKeyOptions = {
+		apiKey: (await createConnect({
+			connectUrl,
+			connectAction: "gen-simple-access-key",
+			source: `@aigne/cli connect to AIGNE hub`,
+			closeOnSuccess: true,
+			openPage: (pageUrl) => openFn(pageUrl)
+		})).accessKeySecret,
+		url: apiUrl
+	};
+	const secretStore = await store_default();
+	await secretStore.setKey(accessKeyOptions.url, accessKeyOptions.apiKey);
+	await secretStore.setDefault(accessKeyOptions.url);
+	return accessKeyOptions;
+}
+const checkConnectionStatus = async (host) => {
+	const env = await (await store_default()).getKey(host);
+	if (!env?.AIGNE_HUB_API_KEY) throw new Error("AIGNE_HUB_API_KEY key not found, need to login first");
+	return {
+		apiKey: env.AIGNE_HUB_API_KEY,
+		url: env.AIGNE_HUB_API_URL
+	};
+};
+async function loadAIGNEHubCredential(options) {
+	if (process.env.BLOCKLET_AIGNE_API_URL && process.env.BLOCKLET_AIGNE_API_PROVIDER) return void 0;
+	const aigneDir = join(homedir(), ".aigne");
+	if (!existsSync(aigneDir)) mkdirSync(aigneDir, { recursive: true });
+	const defaultHost = await (await store_default()).getDefault();
+	const inquirerPrompt = options?.inquirerPromptFn ?? inquirer.prompt;
+	const configUrl = options?.aigneHubUrl || process.env.AIGNE_HUB_API_URL;
+	const url = configUrl || defaultHost?.AIGNE_HUB_API_URL || AIGNE_HUB_URL;
+	const connectUrl = joinURL(new URL(url).origin, WELLKNOWN_SERVICE_PATH_PREFIX);
+	const { host } = new URL(url);
+	let credential = {};
+	try {
+		if (process.env.AIGNE_HUB_API_KEY) credential = {
+			apiKey: process.env.AIGNE_HUB_API_KEY,
+			url: process.env.AIGNE_HUB_API_URL || url
+		};
+		else credential = await checkConnectionStatus(host);
+	} catch (error) {
+		if (error instanceof Error && error.message.includes("login first")) {
+			let aigneHubUrl = connectUrl;
+			if (!configUrl) {
+				const { subscribe } = await inquirerPrompt({
+					type: "list",
+					name: "subscribe",
+					message: "No LLM API Keys or AIGNE Hub connections found. How would you like to proceed?",
+					choices: [
+						{
+							name: "Connect to the Arcblock official AIGNE Hub (recommended, free credits for new users)",
+							value: "official"
+						},
+						connectUrl.includes(AIGNE_HUB_URL) ? {
+							name: "Connect to your own AIGNE Hub instance (self-hosted)",
+							value: "custom"
+						} : null,
+						{
+							name: "Exit and configure my own LLM API Keys",
+							value: "manual"
+						}
+					].filter(Boolean),
+					default: "official"
+				});
+				if (subscribe === "custom") {
+					const { customUrl } = await inquirerPrompt({
+						type: "input",
+						name: "customUrl",
+						message: "Enter the URL of your AIGNE Hub:",
+						validate(input) {
+							try {
+								return new URL(input).protocol.startsWith("http") ? true : "Must be a valid URL with http or https";
+							} catch {
+								return "Invalid URL";
+							}
+						}
+					});
+					aigneHubUrl = customUrl;
+				} else if (subscribe === "manual") {
+					console.log("You chose to configure your own LLM API Keys. Please follow the instructions in the documentation: https://www.arcblock.io/docs/aigne-framework/models-configuration-9dc03e");
+					process.exit(0);
+				}
+			}
+			credential = await connectToAIGNEHub(aigneHubUrl);
+		} else throw error;
+	}
+	return credential;
+}
+
+//#endregion
+export { connectToAIGNEHub, loadAIGNEHubCredential };
+//# sourceMappingURL=credential.mjs.map

package/dist/utils/aigne-hub/credential.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"credential.mjs","names":["session","getSecretStore"],"sources":["../../../src/utils/aigne-hub/credential.ts"],"sourcesContent":["import { existsSync, mkdirSync } from \"node:fs\";\nimport { homedir } from \"node:os\";\nimport { join } from \"node:path\";\nimport { AIGNE_HUB_BLOCKLET_DID, AIGNE_HUB_URL, getAIGNEHubMountPoint } from \"@aigne/aigne-hub\";\nimport { fetch } from \"@aigne/core/utils/fetch\";\nimport inquirer from \"inquirer\";\nimport open from \"open\";\nimport pWaitFor from \"p-wait-for\";\nimport { joinURL, withQuery } from \"ufo\";\nimport { ACCESS_KEY_SESSION_API, isTest, WELLKNOWN_SERVICE_PATH_PREFIX } from \"./constants.js\";\nimport { decrypt, encodeEncryptionKey } from \"./crypto.js\";\nimport getSecretStore from \"./store/index.js\";\nimport type { CreateConnectOptions, FetchResult, LoadCredentialOptions } from \"./type.js\";\n\nconst request = async (config: { url: string; method?: string; requestCount?: number }) => {\n  const headers: Record<string, string> = {};\n  if (config.requestCount !== undefined) {\n    headers[\"X-Request-Count\"] = config.requestCount.toString();\n  }\n\n  const response = await fetch(config.url, { method: config.method || \"GET\", headers });\n\n  const data = await response.json();\n\n  return { data };\n};\n\nexport const fetchConfigs = async ({\n  connectUrl,\n  sessionId,\n  fetchInterval,\n  fetchTimeout,\n}: {\n  connectUrl: string;\n  sessionId: string;\n  fetchInterval: number;\n  fetchTimeout: number;\n}) => {\n  const sessionURL = withQuery(joinURL(connectUrl, ACCESS_KEY_SESSION_API), { sid: sessionId });\n\n  let requestCount = 0;\n  const condition = async () => {\n    const { data: session } = await request({ url: sessionURL, requestCount });\n    requestCount++;\n    return Boolean(session.accessKeyId && session.accessKeySecret);\n  };\n\n  await pWaitFor(condition, { interval: fetchInterval, timeout: fetchTimeout });\n\n  const { data: session } = await request({ url: sessionURL, requestCount });\n  await request({ url: sessionURL, method: \"DELETE\" });\n\n  return {\n    ...session,\n    accessKeyId: session.accessKeyId,\n    accessKeySecret: decrypt(session.accessKeySecret, session.accessKeyId, session.challenge),\n  };\n};\n\nfunction baseWrapSpinner(_: string, waiting: () => Promise<FetchResult>) {\n  return waiting();\n}\n\nexport async function createConnect({\n  connectUrl,\n  openPage,\n  fetchInterval = 3 * 1000,\n  retry = 1500,\n  source = \"Blocklet CLI\",\n  connectAction = \"connect-cli\",\n  wrapSpinner = baseWrapSpinner,\n  closeOnSuccess,\n  intervalFetchConfig,\n  appName = \"AIGNE CLI\",\n  appLogo = \"https://www.aigne.io/favicon.ico?imageFilter=resize&w=32\",\n}: CreateConnectOptions) {\n  const startSessionURL = joinURL(connectUrl, ACCESS_KEY_SESSION_API);\n  const { data: session } = await request({ url: startSessionURL, method: \"POST\" });\n  const token = session.id;\n\n  const pageUrl = withQuery(joinURL(connectUrl, connectAction), {\n    __token__: encodeEncryptionKey(token),\n    source,\n    closeOnSuccess,\n    cli: true,\n    appName: ` ${appName}`,\n    appLogo,\n  });\n\n  openPage?.(pageUrl);\n\n  return await wrapSpinner(`Waiting for connection: ${connectUrl}`, async () => {\n    const checkAuthorizeStatus = intervalFetchConfig ?? fetchConfigs;\n\n    const authorizeStatus = await checkAuthorizeStatus({\n      connectUrl,\n      sessionId: token,\n      fetchTimeout: retry * fetchInterval,\n      fetchInterval: retry,\n    });\n\n    return authorizeStatus;\n  });\n}\n\nexport async function connectToAIGNEHub(url: string) {\n  const { origin } = new URL(url);\n  const connectUrl = joinURL(origin, WELLKNOWN_SERVICE_PATH_PREFIX);\n  const apiUrl = await getAIGNEHubMountPoint(url, AIGNE_HUB_BLOCKLET_DID);\n\n  const openFn = isTest ? () => {} : open;\n  const result = await createConnect({\n    connectUrl: connectUrl,\n    connectAction: \"gen-simple-access-key\",\n    source: `@aigne/cli connect to AIGNE hub`,\n    closeOnSuccess: true,\n    openPage: (pageUrl) => openFn(pageUrl),\n  });\n\n  const accessKeyOptions = {\n    apiKey: result.accessKeySecret,\n    url: apiUrl,\n  };\n\n  const secretStore = await getSecretStore();\n  await secretStore.setKey(accessKeyOptions.url, accessKeyOptions.apiKey);\n  await secretStore.setDefault(accessKeyOptions.url);\n\n  return accessKeyOptions;\n}\n\nexport const checkConnectionStatus = async (host: string) => {\n  const secretStore = await getSecretStore();\n  const env = await secretStore.getKey(host);\n\n  if (!env?.AIGNE_HUB_API_KEY) {\n    throw new Error(\"AIGNE_HUB_API_KEY key not found, need to login first\");\n  }\n\n  return {\n    apiKey: env.AIGNE_HUB_API_KEY,\n    url: env.AIGNE_HUB_API_URL,\n  };\n};\n\nexport async function loadAIGNEHubCredential(options?: LoadCredentialOptions) {\n  const isBlocklet = process.env.BLOCKLET_AIGNE_API_URL && process.env.BLOCKLET_AIGNE_API_PROVIDER;\n  if (isBlocklet) return undefined;\n\n  const aigneDir = join(homedir(), \".aigne\");\n  if (!existsSync(aigneDir)) {\n    mkdirSync(aigneDir, { recursive: true });\n  }\n\n  const secretStore = await getSecretStore();\n  const defaultHost = await secretStore.getDefault();\n  const inquirerPrompt = (options?.inquirerPromptFn ?? inquirer.prompt) as typeof inquirer.prompt;\n\n  const configUrl = options?.aigneHubUrl || process.env.AIGNE_HUB_API_URL;\n  const url = configUrl || defaultHost?.AIGNE_HUB_API_URL || AIGNE_HUB_URL;\n  const connectUrl = joinURL(new URL(url).origin, WELLKNOWN_SERVICE_PATH_PREFIX);\n  const { host } = new URL(url);\n\n  let credential: { apiKey?: string; url?: string } = {};\n\n  try {\n    if (process.env.AIGNE_HUB_API_KEY) {\n      credential = {\n        apiKey: process.env.AIGNE_HUB_API_KEY,\n        url: process.env.AIGNE_HUB_API_URL || url,\n      };\n    } else {\n      credential = await checkConnectionStatus(host);\n    }\n  } catch (error) {\n    if (error instanceof Error && error.message.includes(\"login first\")) {\n      let aigneHubUrl = connectUrl;\n\n      if (!configUrl) {\n        const { subscribe } = await inquirerPrompt({\n          type: \"list\",\n          name: \"subscribe\",\n          message: \"No LLM API Keys or AIGNE Hub connections found. How would you like to proceed?\",\n          choices: [\n            {\n              name: \"Connect to the Arcblock official AIGNE Hub (recommended, free credits for new users)\",\n              value: \"official\",\n            },\n            connectUrl.includes(AIGNE_HUB_URL)\n              ? {\n                  name: \"Connect to your own AIGNE Hub instance (self-hosted)\",\n                  value: \"custom\",\n                }\n              : null,\n            {\n              name: \"Exit and configure my own LLM API Keys\",\n              value: \"manual\",\n            },\n          ].filter(Boolean) as { name: string; value: string }[],\n          default: \"official\",\n        });\n\n        if (subscribe === \"custom\") {\n          const { customUrl } = await inquirerPrompt({\n            type: \"input\",\n            name: \"customUrl\",\n            message: \"Enter the URL of your AIGNE Hub:\",\n            validate(input) {\n              try {\n                const url = new URL(input);\n                return url.protocol.startsWith(\"http\")\n                  ? true\n                  : \"Must be a valid URL with http or https\";\n              } catch {\n                return \"Invalid URL\";\n              }\n            },\n          });\n          aigneHubUrl = customUrl;\n        } else if (subscribe === \"manual\") {\n          console.log(\n            \"You chose to configure your own LLM API Keys. Please follow the instructions in the documentation: https://www.arcblock.io/docs/aigne-framework/models-configuration-9dc03e\",\n          );\n          process.exit(0);\n        }\n      }\n\n      credential = await connectToAIGNEHub(aigneHubUrl);\n    } else {\n      throw error;\n    }\n  }\n\n  return credential;\n}\n"],"mappings":";;;;;;;;;;;;;;AAcA,MAAM,UAAU,OAAO,WAAoE;CACzF,MAAM,UAAkC,EAAE;AAC1C,KAAI,OAAO,iBAAiB,OAC1B,SAAQ,qBAAqB,OAAO,aAAa,UAAU;AAO7D,QAAO,EAAE,MAFI,OAFI,MAAM,MAAM,OAAO,KAAK;EAAE,QAAQ,OAAO,UAAU;EAAO;EAAS,CAAC,EAEzD,MAAM,EAEnB;;AAGjB,MAAa,eAAe,OAAO,EACjC,YACA,WACA,eACA,mBAMI;CACJ,MAAM,aAAa,UAAU,QAAQ,YAAY,uBAAuB,EAAE,EAAE,KAAK,WAAW,CAAC;CAE7F,IAAI,eAAe;CACnB,MAAM,YAAY,YAAY;EAC5B,MAAM,EAAE,MAAMA,cAAY,MAAM,QAAQ;GAAE,KAAK;GAAY;GAAc,CAAC;AAC1E;AACA,SAAO,QAAQA,UAAQ,eAAeA,UAAQ,gBAAgB;;AAGhE,OAAM,SAAS,WAAW;EAAE,UAAU;EAAe,SAAS;EAAc,CAAC;CAE7E,MAAM,EAAE,MAAM,YAAY,MAAM,QAAQ;EAAE,KAAK;EAAY;EAAc,CAAC;AAC1E,OAAM,QAAQ;EAAE,KAAK;EAAY,QAAQ;EAAU,CAAC;AAEpD,QAAO;EACL,GAAG;EACH,aAAa,QAAQ;EACrB,iBAAiB,QAAQ,QAAQ,iBAAiB,QAAQ,aAAa,QAAQ,UAAU;EAC1F;;AAGH,SAAS,gBAAgB,GAAW,SAAqC;AACvE,QAAO,SAAS;;AAGlB,eAAsB,cAAc,EAClC,YACA,UACA,gBAAgB,IAAI,KACpB,QAAQ,MACR,SAAS,gBACT,gBAAgB,eAChB,cAAc,iBACd,gBACA,qBACA,UAAU,aACV,UAAU,8DACa;CAEvB,MAAM,EAAE,MAAM,YAAY,MAAM,QAAQ;EAAE,KADlB,QAAQ,YAAY,uBAAuB;EACH,QAAQ;EAAQ,CAAC;CACjF,MAAM,QAAQ,QAAQ;CAEtB,MAAM,UAAU,UAAU,QAAQ,YAAY,cAAc,EAAE;EAC5D,WAAW,oBAAoB,MAAM;EACrC;EACA;EACA,KAAK;EACL,SAAS,IAAI;EACb;EACD,CAAC;AAEF,YAAW,QAAQ;AAEnB,QAAO,MAAM,YAAY,2BAA2B,cAAc,YAAY;AAU5E,SAPwB,OAFK,uBAAuB,cAED;GACjD;GACA,WAAW;GACX,cAAc,QAAQ;GACtB,eAAe;GAChB,CAAC;GAGF;;AAGJ,eAAsB,kBAAkB,KAAa;CACnD,MAAM,EAAE,WAAW,IAAI,IAAI,IAAI;CAC/B,MAAM,aAAa,QAAQ,QAAQ,8BAA8B;CACjE,MAAM,SAAS,MAAM,sBAAsB,KAAK,uBAAuB;CAEvE,MAAM,SAAS,eAAe,KAAK;CASnC,MAAM,mBAAmB;EACvB,SATa,MAAM,cAAc;GACrB;GACZ,eAAe;GACf,QAAQ;GACR,gBAAgB;GAChB,WAAW,YAAY,OAAO,QAAQ;GACvC,CAAC,EAGe;EACf,KAAK;EACN;CAED,MAAM,cAAc,MAAMC,eAAgB;AAC1C,OAAM,YAAY,OAAO,iBAAiB,KAAK,iBAAiB,OAAO;AACvE,OAAM,YAAY,WAAW,iBAAiB,IAAI;AAElD,QAAO;;AAGT,MAAa,wBAAwB,OAAO,SAAiB;CAE3D,MAAM,MAAM,OADQ,MAAMA,eAAgB,EACZ,OAAO,KAAK;AAE1C,KAAI,CAAC,KAAK,kBACR,OAAM,IAAI,MAAM,uDAAuD;AAGzE,QAAO;EACL,QAAQ,IAAI;EACZ,KAAK,IAAI;EACV;;AAGH,eAAsB,uBAAuB,SAAiC;AAE5E,KADmB,QAAQ,IAAI,0BAA0B,QAAQ,IAAI,4BACrD,QAAO;CAEvB,MAAM,WAAW,KAAK,SAAS,EAAE,SAAS;AAC1C,KAAI,CAAC,WAAW,SAAS,CACvB,WAAU,UAAU,EAAE,WAAW,MAAM,CAAC;CAI1C,MAAM,cAAc,OADA,MAAMA,eAAgB,EACJ,YAAY;CAClD,MAAM,iBAAkB,SAAS,oBAAoB,SAAS;CAE9D,MAAM,YAAY,SAAS,eAAe,QAAQ,IAAI;CACtD,MAAM,MAAM,aAAa,aAAa,qBAAqB;CAC3D,MAAM,aAAa,QAAQ,IAAI,IAAI,IAAI,CAAC,QAAQ,8BAA8B;CAC9E,MAAM,EAAE,SAAS,IAAI,IAAI,IAAI;CAE7B,IAAI,aAAgD,EAAE;AAEtD,KAAI;AACF,MAAI,QAAQ,IAAI,kBACd,cAAa;GACX,QAAQ,QAAQ,IAAI;GACpB,KAAK,QAAQ,IAAI,qBAAqB;GACvC;MAED,cAAa,MAAM,sBAAsB,KAAK;UAEzC,OAAO;AACd,MAAI,iBAAiB,SAAS,MAAM,QAAQ,SAAS,cAAc,EAAE;GACnE,IAAI,cAAc;AAElB,OAAI,CAAC,WAAW;IACd,MAAM,EAAE,cAAc,MAAM,eAAe;KACzC,MAAM;KACN,MAAM;KACN,SAAS;KACT,SAAS;MACP;OACE,MAAM;OACN,OAAO;OACR;MACD,WAAW,SAAS,cAAc,GAC9B;OACE,MAAM;OACN,OAAO;OACR,GACD;MACJ;OACE,MAAM;OACN,OAAO;OACR;MACF,CAAC,OAAO,QAAQ;KACjB,SAAS;KACV,CAAC;AAEF,QAAI,cAAc,UAAU;KAC1B,MAAM,EAAE,cAAc,MAAM,eAAe;MACzC,MAAM;MACN,MAAM;MACN,SAAS;MACT,SAAS,OAAO;AACd,WAAI;AAEF,eADY,IAAI,IAAI,MAAM,CACf,SAAS,WAAW,OAAO,GAClC,OACA;eACE;AACN,eAAO;;;MAGZ,CAAC;AACF,mBAAc;eACL,cAAc,UAAU;AACjC,aAAQ,IACN,8KACD;AACD,aAAQ,KAAK,EAAE;;;AAInB,gBAAa,MAAM,kBAAkB,YAAY;QAEjD,OAAM;;AAIV,QAAO"}

package/dist/utils/aigne-hub/crypto.cjs

@@ -0,0 +1,41 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+let node_crypto = require("node:crypto");
+node_crypto = require_rolldown_runtime.__toESM(node_crypto);
+let crypto_js_aes_js = require("crypto-js/aes.js");
+crypto_js_aes_js = require_rolldown_runtime.__toESM(crypto_js_aes_js);
+let crypto_js_enc_base64_js = require("crypto-js/enc-base64.js");
+crypto_js_enc_base64_js = require_rolldown_runtime.__toESM(crypto_js_enc_base64_js);
+let crypto_js_enc_hex_js = require("crypto-js/enc-hex.js");
+crypto_js_enc_hex_js = require_rolldown_runtime.__toESM(crypto_js_enc_hex_js);
+let crypto_js_enc_latin1_js = require("crypto-js/enc-latin1.js");
+crypto_js_enc_latin1_js = require_rolldown_runtime.__toESM(crypto_js_enc_latin1_js);
+let crypto_js_enc_utf8_js = require("crypto-js/enc-utf8.js");
+crypto_js_enc_utf8_js = require_rolldown_runtime.__toESM(crypto_js_enc_utf8_js);
+let crypto_js_enc_utf16_js = require("crypto-js/enc-utf16.js");
+crypto_js_enc_utf16_js = require_rolldown_runtime.__toESM(crypto_js_enc_utf16_js);
+
+//#region src/utils/aigne-hub/crypto.ts
+const encoders = {
+	latin1: crypto_js_enc_latin1_js.default,
+	utf8: crypto_js_enc_utf8_js.default,
+	hex: crypto_js_enc_hex_js.default,
+	utf16: crypto_js_enc_utf16_js.default,
+	base64: crypto_js_enc_base64_js.default
+};
+var AesCrypter = class {
+	encrypt(message, secret) {
+		const text = typeof message === "string" ? message : JSON.stringify(message);
+		return crypto_js_aes_js.default.encrypt(text, secret).toString();
+	}
+	decrypt(cipher, secret, outputEncoding = "utf8") {
+		return crypto_js_aes_js.default.decrypt(cipher, secret).toString(encoders[outputEncoding]);
+	}
+};
+const aes = new AesCrypter();
+const decrypt = (m, s, i) => aes.decrypt(m, node_crypto.default.pbkdf2Sync(i, s, 256, 32, "sha512").toString("hex"));
+const escapeFn = (str) => str.replace(/\+/g, "-").replace(/\//g, "_").replace(/=/g, "");
+const encodeEncryptionKey = (key) => escapeFn(Buffer.from(key).toString("base64"));
+
+//#endregion
+exports.decrypt = decrypt;
+exports.encodeEncryptionKey = encodeEncryptionKey;

package/dist/utils/aigne-hub/crypto.mjs

@@ -0,0 +1,33 @@
+import crypto from "node:crypto";
+import AES from "crypto-js/aes.js";
+import encBase64 from "crypto-js/enc-base64.js";
+import encHex from "crypto-js/enc-hex.js";
+import encLatin1 from "crypto-js/enc-latin1.js";
+import encUtf8 from "crypto-js/enc-utf8.js";
+import encUtf16 from "crypto-js/enc-utf16.js";
+
+//#region src/utils/aigne-hub/crypto.ts
+const encoders = {
+	latin1: encLatin1,
+	utf8: encUtf8,
+	hex: encHex,
+	utf16: encUtf16,
+	base64: encBase64
+};
+var AesCrypter = class {
+	encrypt(message, secret) {
+		const text = typeof message === "string" ? message : JSON.stringify(message);
+		return AES.encrypt(text, secret).toString();
+	}
+	decrypt(cipher, secret, outputEncoding = "utf8") {
+		return AES.decrypt(cipher, secret).toString(encoders[outputEncoding]);
+	}
+};
+const aes = new AesCrypter();
+const decrypt = (m, s, i) => aes.decrypt(m, crypto.pbkdf2Sync(i, s, 256, 32, "sha512").toString("hex"));
+const escapeFn = (str) => str.replace(/\+/g, "-").replace(/\//g, "_").replace(/=/g, "");
+const encodeEncryptionKey = (key) => escapeFn(Buffer.from(key).toString("base64"));
+
+//#endregion
+export { decrypt, encodeEncryptionKey };
+//# sourceMappingURL=crypto.mjs.map

package/dist/utils/aigne-hub/crypto.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"crypto.mjs","names":[],"sources":["../../../src/utils/aigne-hub/crypto.ts"],"sourcesContent":["import crypto from \"node:crypto\";\nimport AES from \"crypto-js/aes.js\";\nimport encBase64 from \"crypto-js/enc-base64.js\";\nimport encHex from \"crypto-js/enc-hex.js\";\nimport encLatin1 from \"crypto-js/enc-latin1.js\";\nimport encUtf8 from \"crypto-js/enc-utf8.js\";\nimport encUtf16 from \"crypto-js/enc-utf16.js\";\n\nconst encoders: { [index: string]: typeof encLatin1 } = {\n  latin1: encLatin1,\n  utf8: encUtf8,\n  hex: encHex,\n  utf16: encUtf16,\n  base64: encBase64,\n};\n\nclass AesCrypter {\n  encrypt(message: string | object, secret: string) {\n    const text = typeof message === \"string\" ? message : JSON.stringify(message);\n    return AES.encrypt(text, secret).toString();\n  }\n\n  decrypt(cipher: string, secret: string, outputEncoding = \"utf8\") {\n    return AES.decrypt(cipher, secret).toString(encoders[outputEncoding]);\n  }\n}\n\nconst aes = new AesCrypter();\n\nexport const decrypt = (m: string, s: string, i: string) =>\n  aes.decrypt(m, crypto.pbkdf2Sync(i, s, 256, 32, \"sha512\").toString(\"hex\"));\nexport const encrypt = (m: string, s: string, i: string) =>\n  aes.encrypt(m, crypto.pbkdf2Sync(i, s, 256, 32, \"sha512\").toString(\"hex\"));\n\nconst escapeFn = (str: string) => str.replace(/\\+/g, \"-\").replace(/\\//g, \"_\").replace(/=/g, \"\");\nconst unescapeFn = (str: string) =>\n  (str + \"===\".slice((str.length + 3) % 4)).replace(/-/g, \"+\").replace(/_/g, \"/\");\nexport const encodeEncryptionKey = (key: string) => escapeFn(Buffer.from(key).toString(\"base64\"));\nexport const decodeEncryptionKey = (str: string) =>\n  new Uint8Array(Buffer.from(unescapeFn(str), \"base64\"));\n"],"mappings":";;;;;;;;;AAQA,MAAM,WAAkD;CACtD,QAAQ;CACR,MAAM;CACN,KAAK;CACL,OAAO;CACP,QAAQ;CACT;AAED,IAAM,aAAN,MAAiB;CACf,QAAQ,SAA0B,QAAgB;EAChD,MAAM,OAAO,OAAO,YAAY,WAAW,UAAU,KAAK,UAAU,QAAQ;AAC5E,SAAO,IAAI,QAAQ,MAAM,OAAO,CAAC,UAAU;;CAG7C,QAAQ,QAAgB,QAAgB,iBAAiB,QAAQ;AAC/D,SAAO,IAAI,QAAQ,QAAQ,OAAO,CAAC,SAAS,SAAS,gBAAgB;;;AAIzE,MAAM,MAAM,IAAI,YAAY;AAE5B,MAAa,WAAW,GAAW,GAAW,MAC5C,IAAI,QAAQ,GAAG,OAAO,WAAW,GAAG,GAAG,KAAK,IAAI,SAAS,CAAC,SAAS,MAAM,CAAC;AAI5E,MAAM,YAAY,QAAgB,IAAI,QAAQ,OAAO,IAAI,CAAC,QAAQ,OAAO,IAAI,CAAC,QAAQ,MAAM,GAAG;AAG/F,MAAa,uBAAuB,QAAgB,SAAS,OAAO,KAAK,IAAI,CAAC,SAAS,SAAS,CAAC"}

package/dist/utils/aigne-hub/model.cjs

@@ -0,0 +1,112 @@
+const require_rolldown_runtime = require('../../_virtual/rolldown_runtime.cjs');
+const require_constants = require('./constants.cjs');
+const require_index = require('./store/index.cjs');
+const require_credential = require('./credential.cjs');
+let chalk = require("chalk");
+chalk = require_rolldown_runtime.__toESM(chalk);
+let _aigne_core_utils_type_utils = require("@aigne/core/utils/type-utils");
+let _aigne_aigne_hub = require("@aigne/aigne-hub");
+let inquirer = require("inquirer");
+inquirer = require_rolldown_runtime.__toESM(inquirer);
+
+//#region src/utils/aigne-hub/model.ts
+function maskApiKey(apiKey) {
+	if (!apiKey || apiKey.length <= 8) return apiKey;
+	const start = apiKey.slice(0, 4);
+	const end = apiKey.slice(-4);
+	return `${start}${"*".repeat(8)}${end}`;
+}
+function findConfiguredProvider(provider, name) {
+	if (provider && provider.trim().toLowerCase() !== require_constants.AIGNE_HUB_PROVIDER.trim().toLowerCase()) return;
+	if (!name) return void 0;
+	const supportedProviders = (0, _aigne_aigne_hub.getSupportedProviders)(name);
+	for (const supportedProvider of supportedProviders) {
+		const { match } = (0, _aigne_aigne_hub.findModel)(supportedProvider);
+		if (match) {
+			if ((0, _aigne_core_utils_type_utils.flat)(match.apiKeyEnvName).some((name$1) => name$1 && process.env[name$1])) return {
+				provider: supportedProvider,
+				model: (0, _aigne_aigne_hub.resolveProviderModelId)(supportedProvider, name)
+			};
+		}
+	}
+}
+const formatModelName = async (model, inquirerPrompt) => {
+	let { provider, model: name } = (0, _aigne_aigne_hub.parseModel)(model);
+	const configuredEnvProvider = findConfiguredProvider(provider, name);
+	if (configuredEnvProvider) return configuredEnvProvider;
+	provider ||= require_constants.AIGNE_HUB_PROVIDER;
+	const { match, all } = (0, _aigne_aigne_hub.findModel)(provider);
+	if (!match) throw new Error(`Unsupported model: ${provider}/${name}, available providers: ${all.map((m) => m.name).join(", ")}`);
+	if (provider.includes(require_constants.AIGNE_HUB_PROVIDER)) return {
+		provider,
+		model: name || _aigne_aigne_hub.AIGNE_HUB_DEFAULT_MODEL
+	};
+	const requireEnvs = (0, _aigne_core_utils_type_utils.flat)(match.apiKeyEnvName);
+	if (requireEnvs.some((name$1) => name$1 && process.env[name$1])) return {
+		provider,
+		model: name
+	};
+	const secretStore = await require_index.default();
+	const defaultHost = await secretStore.getDefault();
+	if (process.env.AIGNE_HUB_API_KEY || defaultHost?.AIGNE_HUB_API_URL) return {
+		provider: require_constants.AIGNE_HUB_PROVIDER,
+		model: `${provider}/${name}`
+	};
+	if (!(await inquirerPrompt({
+		type: "list",
+		name: "useAigneHub",
+		message: `No API key is configured for ${provider}/${name}, How would you like to proceed?`,
+		choices: [{
+			name: `Connect to AIGNE Hub to use ${name} (recommended — includes free credits)`,
+			value: true
+		}, {
+			name: `Exit and use my own API key (set ${requireEnvs.join(" or ")})`,
+			value: false
+		}],
+		default: true
+	})).useAigneHub) {
+		console.log(chalk.default.yellow(`You can use command "export ${requireEnvs[0]}=xxx" to set API Key in your shell. Or you can set environment variables in .env file.`));
+		process.exit(0);
+	}
+	const envs = await secretStore.listHostsMap();
+	if (envs && Object.keys(envs).length > 0 && !defaultHost?.AIGNE_HUB_API_URL) {
+		const host = new URL(_aigne_aigne_hub.AIGNE_HUB_URL).host;
+		const defaultEnv = envs[host]?.AIGNE_HUB_API_URL ? envs[host] : Object.values(envs)[0] || { AIGNE_HUB_API_URL: "" };
+		await secretStore.setDefault(defaultEnv?.AIGNE_HUB_API_URL);
+	}
+	return {
+		provider: require_constants.AIGNE_HUB_PROVIDER,
+		model: `${provider}/${name}`
+	};
+};
+async function loadChatModel(options) {
+	const { provider, model } = await formatModelName((typeof options?.model === "string" ? options.model : void 0) || process.env.MODEL || "", options?.inquirerPromptFn ?? inquirer.default.prompt);
+	const { match, all } = (0, _aigne_aigne_hub.findModel)(provider);
+	if (!match) throw new Error(`Unsupported model provider ${provider}, available providers: ${all.map((m) => m.name).join(", ")}`);
+	const credential = provider.toLowerCase().includes(require_constants.AIGNE_HUB_PROVIDER) ? await require_credential.loadAIGNEHubCredential(options) : void 0;
+	return match.create({
+		...credential,
+		baseURL: credential?.url,
+		model,
+		modelOptions: options && (0, _aigne_core_utils_type_utils.omit)(options, "model", "aigneHubUrl", "inquirerPromptFn")
+	});
+}
+async function loadImageModel(options) {
+	const { provider, model } = await formatModelName((typeof options?.model === "string" ? options.model : void 0) || process.env.IMAGE_MODEL || "", options?.inquirerPromptFn ?? inquirer.default.prompt);
+	const { match, all } = (0, _aigne_aigne_hub.findImageModel)(provider);
+	if (!match) throw new Error(`Unsupported image model provider ${provider}, available providers: ${all.map((m) => m.name).join(", ")}`);
+	const credential = provider.toLowerCase().includes(require_constants.AIGNE_HUB_PROVIDER) ? await require_credential.loadAIGNEHubCredential(options) : void 0;
+	return match.create({
+		...credential,
+		baseURL: credential?.url,
+		model,
+		modelOptions: options && (0, _aigne_core_utils_type_utils.omit)(options, "model", "aigneHubUrl", "inquirerPromptFn")
+	});
+}
+
+//#endregion
+exports.findConfiguredProvider = findConfiguredProvider;
+exports.formatModelName = formatModelName;
+exports.loadChatModel = loadChatModel;
+exports.loadImageModel = loadImageModel;
+exports.maskApiKey = maskApiKey;

package/dist/utils/aigne-hub/model.d.cts

@@ -0,0 +1,19 @@
+import { LoadCredentialOptions } from "./type.cjs";
+import { ChatModel, ChatModelInputOptionsWithGetter, ImageModel, ImageModelInputOptionsWithGetter } from "@aigne/core";
+import * as _aigne_aigne_hub0 from "@aigne/aigne-hub";
+
+//#region src/utils/aigne-hub/model.d.ts
+declare function maskApiKey(apiKey?: string): string | undefined;
+declare function findConfiguredProvider(provider?: string, name?: string): {
+  provider: _aigne_aigne_hub0.AIProviderType;
+  model: string;
+} | undefined;
+declare const formatModelName: (model: string, inquirerPrompt: NonNullable<LoadCredentialOptions["inquirerPromptFn"]>) => Promise<{
+  provider: string;
+  model?: string;
+}>;
+declare function loadChatModel(options?: ChatModelInputOptionsWithGetter & LoadCredentialOptions): Promise<ChatModel>;
+declare function loadImageModel(options?: ImageModelInputOptionsWithGetter & LoadCredentialOptions): Promise<ImageModel>;
+//#endregion
+export { findConfiguredProvider, formatModelName, loadChatModel, loadImageModel, maskApiKey };
+//# sourceMappingURL=model.d.cts.map

package/dist/utils/aigne-hub/model.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.cts","names":[],"sources":["../../../src/utils/aigne-hub/model.ts"],"mappings":";;;;;iBAuBgB,UAAA,CAAA,MAAA;AAAA,iBAOA,sBAAA,CAAA,QAAA,WAAA,IAAA;EAAA,QAAA,EAAsB,iBAAA,CAAA,cAAA;EAAA,KAAA;AAAA;AAAA,cAqBzB,eAAA,GAAA,KAAA,UAAA,cAAA,EAEK,WAAA,CAAY,qBAAA,0BAC3B,OAAA;EAAA,QAAA;EAAA,KAAA;AAAA;AAAA,iBAsEmB,aAAA,CAAA,OAAA,GACV,+BAAA,GAAkC,qBAAA,GAC3C,OAAA,CAAQ,SAAA;AAAA,iBA0BW,cAAA,CAAA,OAAA,GACV,gCAAA,GAAmC,qBAAA,GAC5C,OAAA,CAAQ,UAAA"}

package/dist/utils/aigne-hub/model.d.mts

@@ -0,0 +1,19 @@
+import { LoadCredentialOptions } from "./type.mjs";
+import { ChatModel, ChatModelInputOptionsWithGetter, ImageModel, ImageModelInputOptionsWithGetter } from "@aigne/core";
+import * as _aigne_aigne_hub0 from "@aigne/aigne-hub";
+
+//#region src/utils/aigne-hub/model.d.ts
+declare function maskApiKey(apiKey?: string): string | undefined;
+declare function findConfiguredProvider(provider?: string, name?: string): {
+  provider: _aigne_aigne_hub0.AIProviderType;
+  model: string;
+} | undefined;
+declare const formatModelName: (model: string, inquirerPrompt: NonNullable<LoadCredentialOptions["inquirerPromptFn"]>) => Promise<{
+  provider: string;
+  model?: string;
+}>;
+declare function loadChatModel(options?: ChatModelInputOptionsWithGetter & LoadCredentialOptions): Promise<ChatModel>;
+declare function loadImageModel(options?: ImageModelInputOptionsWithGetter & LoadCredentialOptions): Promise<ImageModel>;
+//#endregion
+export { findConfiguredProvider, formatModelName, loadChatModel, loadImageModel, maskApiKey };
+//# sourceMappingURL=model.d.mts.map

package/dist/utils/aigne-hub/model.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.d.mts","names":[],"sources":["../../../src/utils/aigne-hub/model.ts"],"mappings":";;;;;iBAuBgB,UAAA,CAAA,MAAA;AAAA,iBAOA,sBAAA,CAAA,QAAA,WAAA,IAAA;EAAA,QAAA,EAAsB,iBAAA,CAAA,cAAA;EAAA,KAAA;AAAA;AAAA,cAqBzB,eAAA,GAAA,KAAA,UAAA,cAAA,EAEK,WAAA,CAAY,qBAAA,0BAC3B,OAAA;EAAA,QAAA;EAAA,KAAA;AAAA;AAAA,iBAsEmB,aAAA,CAAA,OAAA,GACV,+BAAA,GAAkC,qBAAA,GAC3C,OAAA,CAAQ,SAAA;AAAA,iBA0BW,cAAA,CAAA,OAAA,GACV,gCAAA,GAAmC,qBAAA,GAC5C,OAAA,CAAQ,UAAA"}

package/dist/utils/aigne-hub/model.mjs

@@ -0,0 +1,106 @@
+import { AIGNE_HUB_PROVIDER } from "./constants.mjs";
+import store_default from "./store/index.mjs";
+import { loadAIGNEHubCredential } from "./credential.mjs";
+import chalk from "chalk";
+import { flat, omit } from "@aigne/core/utils/type-utils";
+import { AIGNE_HUB_DEFAULT_MODEL, AIGNE_HUB_URL, findImageModel, findModel, getSupportedProviders, parseModel, resolveProviderModelId } from "@aigne/aigne-hub";
+import inquirer from "inquirer";
+
+//#region src/utils/aigne-hub/model.ts
+function maskApiKey(apiKey) {
+	if (!apiKey || apiKey.length <= 8) return apiKey;
+	const start = apiKey.slice(0, 4);
+	const end = apiKey.slice(-4);
+	return `${start}${"*".repeat(8)}${end}`;
+}
+function findConfiguredProvider(provider, name) {
+	if (provider && provider.trim().toLowerCase() !== AIGNE_HUB_PROVIDER.trim().toLowerCase()) return;
+	if (!name) return void 0;
+	const supportedProviders = getSupportedProviders(name);
+	for (const supportedProvider of supportedProviders) {
+		const { match } = findModel(supportedProvider);
+		if (match) {
+			if (flat(match.apiKeyEnvName).some((name$1) => name$1 && process.env[name$1])) return {
+				provider: supportedProvider,
+				model: resolveProviderModelId(supportedProvider, name)
+			};
+		}
+	}
+}
+const formatModelName = async (model, inquirerPrompt) => {
+	let { provider, model: name } = parseModel(model);
+	const configuredEnvProvider = findConfiguredProvider(provider, name);
+	if (configuredEnvProvider) return configuredEnvProvider;
+	provider ||= AIGNE_HUB_PROVIDER;
+	const { match, all } = findModel(provider);
+	if (!match) throw new Error(`Unsupported model: ${provider}/${name}, available providers: ${all.map((m) => m.name).join(", ")}`);
+	if (provider.includes(AIGNE_HUB_PROVIDER)) return {
+		provider,
+		model: name || AIGNE_HUB_DEFAULT_MODEL
+	};
+	const requireEnvs = flat(match.apiKeyEnvName);
+	if (requireEnvs.some((name$1) => name$1 && process.env[name$1])) return {
+		provider,
+		model: name
+	};
+	const secretStore = await store_default();
+	const defaultHost = await secretStore.getDefault();
+	if (process.env.AIGNE_HUB_API_KEY || defaultHost?.AIGNE_HUB_API_URL) return {
+		provider: AIGNE_HUB_PROVIDER,
+		model: `${provider}/${name}`
+	};
+	if (!(await inquirerPrompt({
+		type: "list",
+		name: "useAigneHub",
+		message: `No API key is configured for ${provider}/${name}, How would you like to proceed?`,
+		choices: [{
+			name: `Connect to AIGNE Hub to use ${name} (recommended — includes free credits)`,
+			value: true
+		}, {
+			name: `Exit and use my own API key (set ${requireEnvs.join(" or ")})`,
+			value: false
+		}],
+		default: true
+	})).useAigneHub) {
+		console.log(chalk.yellow(`You can use command "export ${requireEnvs[0]}=xxx" to set API Key in your shell. Or you can set environment variables in .env file.`));
+		process.exit(0);
+	}
+	const envs = await secretStore.listHostsMap();
+	if (envs && Object.keys(envs).length > 0 && !defaultHost?.AIGNE_HUB_API_URL) {
+		const host = new URL(AIGNE_HUB_URL).host;
+		const defaultEnv = envs[host]?.AIGNE_HUB_API_URL ? envs[host] : Object.values(envs)[0] || { AIGNE_HUB_API_URL: "" };
+		await secretStore.setDefault(defaultEnv?.AIGNE_HUB_API_URL);
+	}
+	return {
+		provider: AIGNE_HUB_PROVIDER,
+		model: `${provider}/${name}`
+	};
+};
+async function loadChatModel(options) {
+	const { provider, model } = await formatModelName((typeof options?.model === "string" ? options.model : void 0) || process.env.MODEL || "", options?.inquirerPromptFn ?? inquirer.prompt);
+	const { match, all } = findModel(provider);
+	if (!match) throw new Error(`Unsupported model provider ${provider}, available providers: ${all.map((m) => m.name).join(", ")}`);
+	const credential = provider.toLowerCase().includes(AIGNE_HUB_PROVIDER) ? await loadAIGNEHubCredential(options) : void 0;
+	return match.create({
+		...credential,
+		baseURL: credential?.url,
+		model,
+		modelOptions: options && omit(options, "model", "aigneHubUrl", "inquirerPromptFn")
+	});
+}
+async function loadImageModel(options) {
+	const { provider, model } = await formatModelName((typeof options?.model === "string" ? options.model : void 0) || process.env.IMAGE_MODEL || "", options?.inquirerPromptFn ?? inquirer.prompt);
+	const { match, all } = findImageModel(provider);
+	if (!match) throw new Error(`Unsupported image model provider ${provider}, available providers: ${all.map((m) => m.name).join(", ")}`);
+	const credential = provider.toLowerCase().includes(AIGNE_HUB_PROVIDER) ? await loadAIGNEHubCredential(options) : void 0;
+	return match.create({
+		...credential,
+		baseURL: credential?.url,
+		model,
+		modelOptions: options && omit(options, "model", "aigneHubUrl", "inquirerPromptFn")
+	});
+}
+
+//#endregion
+export { findConfiguredProvider, formatModelName, loadChatModel, loadImageModel, maskApiKey };
+//# sourceMappingURL=model.mjs.map

package/dist/utils/aigne-hub/model.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"model.mjs","names":["name","getSecretStore"],"sources":["../../../src/utils/aigne-hub/model.ts"],"sourcesContent":["import {\n  AIGNE_HUB_DEFAULT_MODEL,\n  AIGNE_HUB_URL,\n  findImageModel,\n  findModel,\n  getSupportedProviders,\n  parseModel,\n  resolveProviderModelId,\n} from \"@aigne/aigne-hub\";\nimport type {\n  ChatModel,\n  ChatModelInputOptionsWithGetter,\n  ImageModel,\n  ImageModelInputOptionsWithGetter,\n} from \"@aigne/core\";\nimport { flat, omit } from \"@aigne/core/utils/type-utils\";\nimport chalk from \"chalk\";\nimport inquirer from \"inquirer\";\nimport { AIGNE_HUB_PROVIDER } from \"./constants.js\";\nimport { loadAIGNEHubCredential } from \"./credential.js\";\nimport getSecretStore from \"./store/index.js\";\nimport type { LoadCredentialOptions } from \"./type.js\";\n\nexport function maskApiKey(apiKey?: string) {\n  if (!apiKey || apiKey.length <= 8) return apiKey;\n  const start = apiKey.slice(0, 4);\n  const end = apiKey.slice(-4);\n  return `${start}${\"*\".repeat(8)}${end}`;\n}\n\nexport function findConfiguredProvider(provider?: string, name?: string) {\n  if (provider && provider.trim().toLowerCase() !== AIGNE_HUB_PROVIDER.trim().toLowerCase()) {\n    return undefined;\n  }\n  if (!name) return undefined;\n\n  const supportedProviders = getSupportedProviders(name);\n  for (const supportedProvider of supportedProviders) {\n    const { match } = findModel(supportedProvider);\n    if (match) {\n      const requireEnvs = flat(match.apiKeyEnvName);\n      if (requireEnvs.some((name) => name && process.env[name])) {\n        return {\n          provider: supportedProvider,\n          model: resolveProviderModelId(supportedProvider, name),\n        };\n      }\n    }\n  }\n}\n\nexport const formatModelName = async (\n  model: string,\n  inquirerPrompt: NonNullable<LoadCredentialOptions[\"inquirerPromptFn\"]>,\n): Promise<{ provider: string; model?: string }> => {\n  let { provider, model: name } = parseModel(model);\n\n  const configuredEnvProvider = findConfiguredProvider(provider, name);\n  if (configuredEnvProvider) return configuredEnvProvider;\n\n  provider ||= AIGNE_HUB_PROVIDER;\n  const { match, all } = findModel(provider);\n  if (!match) {\n    throw new Error(\n      `Unsupported model: ${provider}/${name}, available providers: ${all.map((m) => m.name).join(\", \")}`,\n    );\n  }\n\n  if (provider.includes(AIGNE_HUB_PROVIDER)) {\n    return { provider, model: name || AIGNE_HUB_DEFAULT_MODEL };\n  }\n\n  const requireEnvs = flat(match.apiKeyEnvName);\n  if (requireEnvs.some((name) => name && process.env[name])) {\n    return { provider, model: name };\n  }\n\n  const secretStore = await getSecretStore();\n  const defaultHost = await secretStore.getDefault();\n\n  if (process.env.AIGNE_HUB_API_KEY || defaultHost?.AIGNE_HUB_API_URL) {\n    return { provider: AIGNE_HUB_PROVIDER, model: `${provider}/${name}` };\n  }\n\n  const result = await inquirerPrompt({\n    type: \"list\",\n    name: \"useAigneHub\",\n    message: `No API key is configured for ${provider}/${name}, How would you like to proceed?`,\n    choices: [\n      {\n        name: `Connect to AIGNE Hub to use ${name} (recommended — includes free credits)`,\n        value: true,\n      },\n      {\n        name: `Exit and use my own API key (set ${requireEnvs.join(\" or \")})`,\n        value: false,\n      },\n    ],\n    default: true,\n  });\n\n  if (!result.useAigneHub) {\n    console.log(\n      chalk.yellow(\n        `You can use command \"export ${requireEnvs[0]}=xxx\" to set API Key in your shell. Or you can set environment variables in .env file.`,\n      ),\n    );\n    process.exit(0);\n  }\n\n  const envs = await secretStore.listHostsMap();\n  if (envs && Object.keys(envs).length > 0 && !defaultHost?.AIGNE_HUB_API_URL) {\n    const host = new URL(AIGNE_HUB_URL).host;\n\n    const defaultEnv = envs[host]?.AIGNE_HUB_API_URL\n      ? envs[host]\n      : Object.values(envs)[0] || { AIGNE_HUB_API_URL: \"\" };\n\n    await secretStore.setDefault(defaultEnv?.AIGNE_HUB_API_URL);\n  }\n\n  return { provider: AIGNE_HUB_PROVIDER, model: `${provider}/${name}` };\n};\n\nexport async function loadChatModel(\n  options?: ChatModelInputOptionsWithGetter & LoadCredentialOptions,\n): Promise<ChatModel> {\n  const { provider, model } = await formatModelName(\n    (typeof options?.model === \"string\" ? options.model : undefined) || process.env.MODEL || \"\",\n    options?.inquirerPromptFn ??\n      (inquirer.prompt as NonNullable<LoadCredentialOptions[\"inquirerPromptFn\"]>),\n  );\n\n  const { match, all } = findModel(provider);\n  if (!match) {\n    throw new Error(\n      `Unsupported model provider ${provider}, available providers: ${all.map((m) => m.name).join(\", \")}`,\n    );\n  }\n\n  const credential = provider.toLowerCase().includes(AIGNE_HUB_PROVIDER)\n    ? await loadAIGNEHubCredential(options)\n    : undefined;\n\n  return match.create({\n    ...credential,\n    baseURL: credential?.url,\n    model,\n    modelOptions: options && omit(options, \"model\", \"aigneHubUrl\", \"inquirerPromptFn\"),\n  });\n}\n\nexport async function loadImageModel(\n  options?: ImageModelInputOptionsWithGetter & LoadCredentialOptions,\n): Promise<ImageModel> {\n  const { provider, model } = await formatModelName(\n    (typeof options?.model === \"string\" ? options.model : undefined) ||\n      process.env.IMAGE_MODEL ||\n      \"\",\n    options?.inquirerPromptFn ??\n      (inquirer.prompt as NonNullable<LoadCredentialOptions[\"inquirerPromptFn\"]>),\n  );\n\n  const { match, all } = findImageModel(provider);\n  if (!match) {\n    throw new Error(\n      `Unsupported image model provider ${provider}, available providers: ${all.map((m) => m.name).join(\", \")}`,\n    );\n  }\n\n  const credential = provider.toLowerCase().includes(AIGNE_HUB_PROVIDER)\n    ? await loadAIGNEHubCredential(options)\n    : undefined;\n\n  return match.create({\n    ...credential,\n    baseURL: credential?.url,\n    model,\n    modelOptions: options && omit(options, \"model\", \"aigneHubUrl\", \"inquirerPromptFn\"),\n  });\n}\n"],"mappings":";;;;;;;;;AAuBA,SAAgB,WAAW,QAAiB;AAC1C,KAAI,CAAC,UAAU,OAAO,UAAU,EAAG,QAAO;CAC1C,MAAM,QAAQ,OAAO,MAAM,GAAG,EAAE;CAChC,MAAM,MAAM,OAAO,MAAM,GAAG;AAC5B,QAAO,GAAG,QAAQ,IAAI,OAAO,EAAE,GAAG;;AAGpC,SAAgB,uBAAuB,UAAmB,MAAe;AACvE,KAAI,YAAY,SAAS,MAAM,CAAC,aAAa,KAAK,mBAAmB,MAAM,CAAC,aAAa,CACvF;AAEF,KAAI,CAAC,KAAM,QAAO;CAElB,MAAM,qBAAqB,sBAAsB,KAAK;AACtD,MAAK,MAAM,qBAAqB,oBAAoB;EAClD,MAAM,EAAE,UAAU,UAAU,kBAAkB;AAC9C,MAAI,OAEF;OADoB,KAAK,MAAM,cAAc,CAC7B,MAAM,WAASA,UAAQ,QAAQ,IAAIA,QAAM,CACvD,QAAO;IACL,UAAU;IACV,OAAO,uBAAuB,mBAAmB,KAAK;IACvD;;;;AAMT,MAAa,kBAAkB,OAC7B,OACA,mBACkD;CAClD,IAAI,EAAE,UAAU,OAAO,SAAS,WAAW,MAAM;CAEjD,MAAM,wBAAwB,uBAAuB,UAAU,KAAK;AACpE,KAAI,sBAAuB,QAAO;AAElC,cAAa;CACb,MAAM,EAAE,OAAO,QAAQ,UAAU,SAAS;AAC1C,KAAI,CAAC,MACH,OAAM,IAAI,MACR,sBAAsB,SAAS,GAAG,KAAK,yBAAyB,IAAI,KAAK,MAAM,EAAE,KAAK,CAAC,KAAK,KAAK,GAClG;AAGH,KAAI,SAAS,SAAS,mBAAmB,CACvC,QAAO;EAAE;EAAU,OAAO,QAAQ;EAAyB;CAG7D,MAAM,cAAc,KAAK,MAAM,cAAc;AAC7C,KAAI,YAAY,MAAM,WAASA,UAAQ,QAAQ,IAAIA,QAAM,CACvD,QAAO;EAAE;EAAU,OAAO;EAAM;CAGlC,MAAM,cAAc,MAAMC,eAAgB;CAC1C,MAAM,cAAc,MAAM,YAAY,YAAY;AAElD,KAAI,QAAQ,IAAI,qBAAqB,aAAa,kBAChD,QAAO;EAAE,UAAU;EAAoB,OAAO,GAAG,SAAS,GAAG;EAAQ;AAoBvE,KAAI,EAjBW,MAAM,eAAe;EAClC,MAAM;EACN,MAAM;EACN,SAAS,gCAAgC,SAAS,GAAG,KAAK;EAC1D,SAAS,CACP;GACE,MAAM,+BAA+B,KAAK;GAC1C,OAAO;GACR,EACD;GACE,MAAM,oCAAoC,YAAY,KAAK,OAAO,CAAC;GACnE,OAAO;GACR,CACF;EACD,SAAS;EACV,CAAC,EAEU,aAAa;AACvB,UAAQ,IACN,MAAM,OACJ,+BAA+B,YAAY,GAAG,wFAC/C,CACF;AACD,UAAQ,KAAK,EAAE;;CAGjB,MAAM,OAAO,MAAM,YAAY,cAAc;AAC7C,KAAI,QAAQ,OAAO,KAAK,KAAK,CAAC,SAAS,KAAK,CAAC,aAAa,mBAAmB;EAC3E,MAAM,OAAO,IAAI,IAAI,cAAc,CAAC;EAEpC,MAAM,aAAa,KAAK,OAAO,oBAC3B,KAAK,QACL,OAAO,OAAO,KAAK,CAAC,MAAM,EAAE,mBAAmB,IAAI;AAEvD,QAAM,YAAY,WAAW,YAAY,kBAAkB;;AAG7D,QAAO;EAAE,UAAU;EAAoB,OAAO,GAAG,SAAS,GAAG;EAAQ;;AAGvE,eAAsB,cACpB,SACoB;CACpB,MAAM,EAAE,UAAU,UAAU,MAAM,iBAC/B,OAAO,SAAS,UAAU,WAAW,QAAQ,QAAQ,WAAc,QAAQ,IAAI,SAAS,IACzF,SAAS,oBACN,SAAS,OACb;CAED,MAAM,EAAE,OAAO,QAAQ,UAAU,SAAS;AAC1C,KAAI,CAAC,MACH,OAAM,IAAI,MACR,8BAA8B,SAAS,yBAAyB,IAAI,KAAK,MAAM,EAAE,KAAK,CAAC,KAAK,KAAK,GAClG;CAGH,MAAM,aAAa,SAAS,aAAa,CAAC,SAAS,mBAAmB,GAClE,MAAM,uBAAuB,QAAQ,GACrC;AAEJ,QAAO,MAAM,OAAO;EAClB,GAAG;EACH,SAAS,YAAY;EACrB;EACA,cAAc,WAAW,KAAK,SAAS,SAAS,eAAe,mBAAmB;EACnF,CAAC;;AAGJ,eAAsB,eACpB,SACqB;CACrB,MAAM,EAAE,UAAU,UAAU,MAAM,iBAC/B,OAAO,SAAS,UAAU,WAAW,QAAQ,QAAQ,WACpD,QAAQ,IAAI,eACZ,IACF,SAAS,oBACN,SAAS,OACb;CAED,MAAM,EAAE,OAAO,QAAQ,eAAe,SAAS;AAC/C,KAAI,CAAC,MACH,OAAM,IAAI,MACR,oCAAoC,SAAS,yBAAyB,IAAI,KAAK,MAAM,EAAE,KAAK,CAAC,KAAK,KAAK,GACxG;CAGH,MAAM,aAAa,SAAS,aAAa,CAAC,SAAS,mBAAmB,GAClE,MAAM,uBAAuB,QAAQ,GACrC;AAEJ,QAAO,MAAM,OAAO;EAClB,GAAG;EACH,SAAS,YAAY;EACrB;EACA,cAAc,WAAW,KAAK,SAAS,SAAS,eAAe,mBAAmB;EACnF,CAAC"}

package/dist/utils/aigne-hub/store/file.cjs

@@ -0,0 +1,64 @@
+const require_rolldown_runtime = require('../../../_virtual/rolldown_runtime.cjs');
+let _aigne_secrets = require("@aigne/secrets");
+
+//#region src/utils/aigne-hub/store/file.ts
+var FileStore = class extends _aigne_secrets.FileStore {
+	outputConfig;
+	constructor(options) {
+		super(options);
+		this.outputConfig = {
+			url: "AIGNE_HUB_API_URL",
+			key: "AIGNE_HUB_API_KEY"
+		};
+	}
+	async setKey(url, apiKey) {
+		return this.setItem(this.normalizeHostFrom(url), {
+			[this.outputConfig.url]: url,
+			[this.outputConfig.key]: apiKey
+		});
+	}
+	async getKey(url) {
+		const host = this.normalizeHostFrom(url);
+		return this.getItem(host);
+	}
+	async deleteKey(url) {
+		const host = this.normalizeHostFrom(url);
+		return this.deleteItem(host);
+	}
+	async listHosts() {
+		return this.listEntries();
+	}
+	async listHostsMap() {
+		return this.listMap();
+	}
+	async setDefault(url) {
+		return this.setDefaultItem({ [this.outputConfig.url]: url });
+	}
+	async getDefault(options = {}) {
+		const { fallbackToFirst = false, presetIfFallback = false } = options;
+		if (!await this.available()) return null;
+		try {
+			const apiUrl = (await this.getDefaultItem())?.[this.outputConfig.url];
+			if (apiUrl) {
+				const defaultInfo = await this.getKey(apiUrl);
+				if (defaultInfo) return defaultInfo;
+			}
+		} catch {}
+		if (!fallbackToFirst) return null;
+		const hosts = await this.listHosts();
+		if (hosts.length === 0) return null;
+		const firstHost = hosts[0];
+		if (!firstHost) return null;
+		if (presetIfFallback && firstHost[this.outputConfig.url]) try {
+			await this.setDefault(firstHost[this.outputConfig.url]);
+		} catch {}
+		return firstHost;
+	}
+	async deleteDefault() {
+		return this.deleteDefaultItem();
+	}
+};
+var file_default = FileStore;
+
+//#endregion
+exports.default = file_default;