@aigne/aigne-hub 0.4.7 → 0.4.9

package/CHANGELOG.md

@@ -1,5 +1,55 @@
 # Changelog
 
+## [0.4.9](https://github.com/AIGNE-io/aigne-framework/compare/aigne-hub-v0.4.8...aigne-hub-v0.4.9) (2025-08-12)
+
+
+### Bug Fixes
+
+* **core:** examples cases that failed when using aigne-hub ([#337](https://github.com/AIGNE-io/aigne-framework/issues/337)) ([0d4a31c](https://github.com/AIGNE-io/aigne-framework/commit/0d4a31c24d9e7d26f00d1accb80719d9ad79a4c6))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/anthropic bumped to 0.11.0
+    * @aigne/bedrock bumped to 0.9.0
+    * @aigne/core bumped to 1.49.0
+    * @aigne/deepseek bumped to 0.7.18
+    * @aigne/default-memory bumped to 1.1.0
+    * @aigne/gemini bumped to 0.9.0
+    * @aigne/ollama bumped to 0.7.18
+    * @aigne/open-router bumped to 0.7.18
+    * @aigne/openai bumped to 0.11.0
+    * @aigne/platform-helpers bumped to 0.6.1
+    * @aigne/transport bumped to 0.13.0
+    * @aigne/xai bumped to 0.7.18
+  * devDependencies
+    * @aigne/openai bumped to 0.11.0
+    * @aigne/test-utils bumped to 0.5.26
+
+## [0.4.8](https://github.com/AIGNE-io/aigne-framework/compare/aigne-hub-v0.4.7...aigne-hub-v0.4.8) (2025-08-12)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/anthropic bumped to 0.10.13
+    * @aigne/bedrock bumped to 0.8.17
+    * @aigne/core bumped to 1.48.0
+    * @aigne/deepseek bumped to 0.7.17
+    * @aigne/default-memory bumped to 1.0.17
+    * @aigne/gemini bumped to 0.8.17
+    * @aigne/ollama bumped to 0.7.17
+    * @aigne/open-router bumped to 0.7.17
+    * @aigne/openai bumped to 0.10.17
+    * @aigne/transport bumped to 0.12.5
+    * @aigne/xai bumped to 0.7.17
+  * devDependencies
+    * @aigne/openai bumped to 0.10.17
+    * @aigne/test-utils bumped to 0.5.25
+
 ## [0.4.7](https://github.com/AIGNE-io/aigne-framework/compare/aigne-hub-v0.4.6...aigne-hub-v0.4.7) (2025-08-11)
 
 

package/lib/cjs/blocklet-aigne-hub-model.js

@@ -2,17 +2,18 @@
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.BlockletAIGNEHubChatModel = void 0;
 const core_1 = require("@aigne/core");
-const constants_js_1 = require("./constants.js");
+const constants_js_1 = require("./util/constants.js");
+const model_js_1 = require("./util/model.js");
 class BlockletAIGNEHubChatModel extends core_1.ChatModel {
     options;
     client;
     constructor(options) {
         super();
         this.options = options;
-        const models = (0, constants_js_1.availableModels)();
+        const models = (0, model_js_1.availableModels)();
         const rawProvider = process.env.BLOCKLET_AIGNE_API_PROVIDER ?? "";
         const providerKey = rawProvider.toLowerCase().replace(/-/g, "");
-        const modelEntry = (0, constants_js_1.findModel)(models, providerKey);
+        const modelEntry = (0, model_js_1.findModel)(models, providerKey);
         if (!modelEntry) {
             const available = models.map((m) => m.name).join(", ");
             throw new Error(`Unsupported model provider: ${rawProvider} ${process.env.BLOCKLET_AIGNE_API_MODEL}. Available providers: ${available}`);

package/lib/cjs/cli-aigne-hub-model.js

@@ -6,7 +6,7 @@ const type_utils_js_1 = require("@aigne/core/utils/type-utils.js");
 const base_client_js_1 = require("@aigne/transport/http-client/base-client.js");
 const ufo_1 = require("ufo");
 const zod_1 = require("zod");
-const constants_js_1 = require("./constants.js");
+const constants_js_1 = require("./util/constants.js");
 const DEFAULT_CHAT_MODEL = "openai/gpt-4o";
 const aigneHubChatModelOptionsSchema = zod_1.z.object({
     url: zod_1.z.string().optional(),

package/lib/cjs/index.d.ts

@@ -2,7 +2,11 @@ import { type AgentInvokeOptions, type AgentProcessResult, ChatModel, type ChatM
 import type { PromiseOrValue } from "@aigne/core/utils/type-utils.js";
 import { type HubChatModelOptions } from "./blocklet-aigne-hub-model.js";
 import { type AIGNEHubChatModelOptions } from "./cli-aigne-hub-model.js";
-export * from "./constants.js";
+export * from "./util/constants.js";
+export * from "./util/credential.js";
+export * from "./util/crypto.js";
+export * from "./util/model.js";
+export * from "./util/type.js";
 export declare class AIGNEHubChatModel extends ChatModel {
     options: HubChatModelOptions;
     private client;

package/lib/cjs/index.js

@@ -18,8 +18,13 @@ exports.AIGNEHubChatModel = void 0;
 const core_1 = require("@aigne/core");
 const blocklet_aigne_hub_model_js_1 = require("./blocklet-aigne-hub-model.js");
 const cli_aigne_hub_model_js_1 = require("./cli-aigne-hub-model.js");
-const constants_js_1 = require("./constants.js");
-__exportStar(require("./constants.js"), exports);
+const constants_js_1 = require("./util/constants.js");
+const credential_js_1 = require("./util/credential.js");
+__exportStar(require("./util/constants.js"), exports);
+__exportStar(require("./util/credential.js"), exports);
+__exportStar(require("./util/crypto.js"), exports);
+__exportStar(require("./util/model.js"), exports);
+__exportStar(require("./util/type.js"), exports);
 class AIGNEHubChatModel extends core_1.ChatModel {
     options;
     client;
@@ -27,7 +32,7 @@ class AIGNEHubChatModel extends core_1.ChatModel {
         const u = process.env.BLOCKLET_AIGNE_API_URL ?? process.env.AIGNE_HUB_API_URL ?? constants_js_1.AIGNE_HUB_URL;
         let url = options.url ?? u;
         if ((process.env.BLOCKLET_AIGNE_API_PROVIDER || "").toLocaleLowerCase().includes("aignehub")) {
-            url = await (0, constants_js_1.getAIGNEHubMountPoint)(url);
+            url = await (0, credential_js_1.getAIGNEHubMountPoint)(url);
         }
         return new AIGNEHubChatModel({ ...options, url });
     }

package/lib/cjs/util/constants.d.ts

@@ -0,0 +1,13 @@
+declare const WELLKNOWN_SERVICE_PATH_PREFIX = "/.well-known/service";
+declare const ACCESS_KEY_PREFIX = "/api/access-key";
+declare const ACCESS_KEY_SESSION_API = "/api/access-key/session";
+declare const AIGNE_HUB_DID = "z8ia3xzq2tMq8CRHfaXj1BTYJyYnEcHbqP8cJ";
+declare const AGENT_HUB_PROVIDER = "aignehub";
+declare const AIGNE_HUB_URL = "https://hub.aigne.io/";
+declare const DEFAULT_AIGNE_HUB_PROVIDER_MODEL = "aignehub:openai/gpt-5-mini";
+declare const DEFAULT_MODEL_PROVIDER = "openai";
+declare const isTest: string | boolean;
+declare const TEST_AIGNE_ENV_FILE: string;
+declare const PROD_AIGNE_ENV_FILE: string;
+declare const AIGNE_ENV_FILE: string;
+export { DEFAULT_AIGNE_HUB_PROVIDER_MODEL, DEFAULT_MODEL_PROVIDER, AIGNE_HUB_DID, AIGNE_HUB_URL, isTest, WELLKNOWN_SERVICE_PATH_PREFIX, ACCESS_KEY_PREFIX, ACCESS_KEY_SESSION_API, AGENT_HUB_PROVIDER, TEST_AIGNE_ENV_FILE, PROD_AIGNE_ENV_FILE, AIGNE_ENV_FILE, };

package/lib/cjs/util/constants.js

@@ -0,0 +1,29 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIGNE_ENV_FILE = exports.PROD_AIGNE_ENV_FILE = exports.TEST_AIGNE_ENV_FILE = exports.AGENT_HUB_PROVIDER = exports.ACCESS_KEY_SESSION_API = exports.ACCESS_KEY_PREFIX = exports.WELLKNOWN_SERVICE_PATH_PREFIX = exports.isTest = exports.AIGNE_HUB_URL = exports.AIGNE_HUB_DID = exports.DEFAULT_MODEL_PROVIDER = exports.DEFAULT_AIGNE_HUB_PROVIDER_MODEL = void 0;
+const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
+const WELLKNOWN_SERVICE_PATH_PREFIX = "/.well-known/service";
+exports.WELLKNOWN_SERVICE_PATH_PREFIX = WELLKNOWN_SERVICE_PATH_PREFIX;
+const ACCESS_KEY_PREFIX = "/api/access-key";
+exports.ACCESS_KEY_PREFIX = ACCESS_KEY_PREFIX;
+const ACCESS_KEY_SESSION_API = `${ACCESS_KEY_PREFIX}/session`;
+exports.ACCESS_KEY_SESSION_API = ACCESS_KEY_SESSION_API;
+const AIGNE_HUB_DID = "z8ia3xzq2tMq8CRHfaXj1BTYJyYnEcHbqP8cJ";
+exports.AIGNE_HUB_DID = AIGNE_HUB_DID;
+const AGENT_HUB_PROVIDER = "aignehub";
+exports.AGENT_HUB_PROVIDER = AGENT_HUB_PROVIDER;
+const AIGNE_HUB_URL = "https://hub.aigne.io/";
+exports.AIGNE_HUB_URL = AIGNE_HUB_URL;
+const DEFAULT_AIGNE_HUB_MODEL = "openai/gpt-5-mini";
+const DEFAULT_AIGNE_HUB_PROVIDER_MODEL = `${AGENT_HUB_PROVIDER}:${DEFAULT_AIGNE_HUB_MODEL}`;
+exports.DEFAULT_AIGNE_HUB_PROVIDER_MODEL = DEFAULT_AIGNE_HUB_PROVIDER_MODEL;
+const DEFAULT_MODEL_PROVIDER = "openai";
+exports.DEFAULT_MODEL_PROVIDER = DEFAULT_MODEL_PROVIDER;
+const isTest = process.env.CI || process.env.NODE_ENV === "test";
+exports.isTest = isTest;
+const TEST_AIGNE_ENV_FILE = index_js_1.nodejs.path.join(index_js_1.nodejs.os.homedir(), ".aigne", "test-aigne-hub-connected.yaml");
+exports.TEST_AIGNE_ENV_FILE = TEST_AIGNE_ENV_FILE;
+const PROD_AIGNE_ENV_FILE = index_js_1.nodejs.path.join(index_js_1.nodejs.os.homedir(), ".aigne", "aigne-hub-connected.yaml");
+exports.PROD_AIGNE_ENV_FILE = PROD_AIGNE_ENV_FILE;
+const AIGNE_ENV_FILE = isTest ? TEST_AIGNE_ENV_FILE : PROD_AIGNE_ENV_FILE;
+exports.AIGNE_ENV_FILE = AIGNE_ENV_FILE;

package/lib/cjs/util/credential.d.ts

@@ -0,0 +1,24 @@
+import type { CreateConnectOptions, FetchResult, LoadCredentialOptions } from "./type.js";
+export declare const fetchConfigs: ({ connectUrl, sessionId, fetchInterval, fetchTimeout, }: {
+    connectUrl: string;
+    sessionId: string;
+    fetchInterval: number;
+    fetchTimeout: number;
+}) => Promise<any>;
+export declare function createConnect({ connectUrl, openPage, fetchInterval, retry, source, connectAction, wrapSpinner, closeOnSuccess, intervalFetchConfig, appName, appLogo, }: CreateConnectOptions): Promise<FetchResult>;
+export declare function getAIGNEHubMountPoint(url: string): Promise<string>;
+export declare function connectToAIGNEHub(url: string): Promise<{
+    apiKey: string;
+    url: string;
+} | {
+    apiKey: undefined;
+    url: undefined;
+}>;
+export declare const checkConnectionStatus: (host: string) => Promise<{
+    apiKey: any;
+    url: string;
+}>;
+export declare function loadCredential(options?: LoadCredentialOptions): Promise<{
+    apiKey?: string;
+    url?: string;
+} | undefined>;

package/lib/cjs/util/credential.js

@@ -0,0 +1,223 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.checkConnectionStatus = exports.fetchConfigs = void 0;
+exports.createConnect = createConnect;
+exports.getAIGNEHubMountPoint = getAIGNEHubMountPoint;
+exports.connectToAIGNEHub = connectToAIGNEHub;
+exports.loadCredential = loadCredential;
+const logger_js_1 = require("@aigne/core/utils/logger.js");
+const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
+const inquirer_1 = __importDefault(require("inquirer"));
+const open_1 = __importDefault(require("open"));
+const p_wait_for_1 = __importDefault(require("p-wait-for"));
+const ufo_1 = require("ufo");
+const yaml_1 = require("yaml");
+const constants_js_1 = require("./constants.js");
+const crypto_js_1 = require("./crypto.js");
+const request = async (config) => {
+    const headers = {};
+    if (config.requestCount !== undefined) {
+        headers["X-Request-Count"] = config.requestCount.toString();
+    }
+    const response = await fetch(config.url, { method: config.method || "GET", headers });
+    if (!response.ok)
+        throw new Error(`HTTP error! status: ${response.status}`);
+    const data = await response.json();
+    return { data };
+};
+const fetchConfigs = async ({ connectUrl, sessionId, fetchInterval, fetchTimeout, }) => {
+    const sessionURL = (0, ufo_1.withQuery)((0, ufo_1.joinURL)(connectUrl, constants_js_1.ACCESS_KEY_SESSION_API), { sid: sessionId });
+    let requestCount = 0;
+    const condition = async () => {
+        const { data: session } = await request({ url: sessionURL, requestCount });
+        requestCount++;
+        return Boolean(session.accessKeyId && session.accessKeySecret);
+    };
+    await (0, p_wait_for_1.default)(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: (0, crypto_js_1.decrypt)(session.accessKeySecret, session.accessKeyId, session.challenge),
+    };
+};
+exports.fetchConfigs = fetchConfigs;
+function baseWrapSpinner(_, waiting) {
+    return Promise.resolve(waiting());
+}
+async function createConnect({ connectUrl, openPage, fetchInterval = 3 * 1000, 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 startSessionURL = (0, ufo_1.joinURL)(connectUrl, constants_js_1.ACCESS_KEY_SESSION_API);
+    const { data: session } = await request({ url: startSessionURL, method: "POST" });
+    const token = session.id;
+    const pageUrl = (0, ufo_1.withQuery)((0, ufo_1.joinURL)(connectUrl, connectAction), {
+        __token__: (0, crypto_js_1.encodeEncryptionKey)(token),
+        source,
+        closeOnSuccess,
+        cli: true,
+        appName: ` ${appName}`,
+        appLogo,
+    });
+    openPage?.(pageUrl);
+    return await wrapSpinner(`Waiting for connection: ${connectUrl}`, async () => {
+        const checkAuthorizeStatus = intervalFetchConfig ?? exports.fetchConfigs;
+        const authorizeStatus = await checkAuthorizeStatus({
+            connectUrl,
+            sessionId: token,
+            fetchTimeout: retry * fetchInterval,
+            fetchInterval: retry,
+        });
+        return authorizeStatus;
+    });
+}
+async function getAIGNEHubMountPoint(url) {
+    const { origin } = new URL(url);
+    const BLOCKLET_JSON_PATH = "__blocklet__.js?type=json";
+    const blockletInfo = await fetch((0, ufo_1.joinURL)(origin, BLOCKLET_JSON_PATH));
+    const blocklet = await blockletInfo.json();
+    const aigneHubMount = (blocklet?.componentMountPoints || []).find((m) => m.did === constants_js_1.AIGNE_HUB_DID);
+    return (0, ufo_1.joinURL)(origin, aigneHubMount?.mountPoint || "");
+}
+async function connectToAIGNEHub(url) {
+    const { origin, host } = new URL(url);
+    const connectUrl = (0, ufo_1.joinURL)(origin, constants_js_1.WELLKNOWN_SERVICE_PATH_PREFIX);
+    const urlWithAIGNEHubMountPoint = await getAIGNEHubMountPoint(url);
+    try {
+        const openFn = constants_js_1.isTest ? () => { } : open_1.default;
+        const result = await createConnect({
+            connectUrl: connectUrl,
+            connectAction: "gen-simple-access-key",
+            source: `@aigne/cli connect to AIGNE hub`,
+            closeOnSuccess: true,
+            openPage: (pageUrl) => openFn(pageUrl),
+        });
+        const accessKeyOptions = {
+            apiKey: result.accessKeySecret,
+            url: urlWithAIGNEHubMountPoint,
+        };
+        // After redirection, write the AIGNE Hub access token
+        const aigneDir = index_js_1.nodejs.path.join(index_js_1.nodejs.os.homedir(), ".aigne");
+        if (!index_js_1.nodejs.fsSync.existsSync(aigneDir)) {
+            index_js_1.nodejs.fsSync.mkdirSync(aigneDir, { recursive: true });
+        }
+        const envs = (0, yaml_1.parse)(await index_js_1.nodejs.fs.readFile(constants_js_1.AIGNE_ENV_FILE, "utf8").catch(() => (0, yaml_1.stringify)({})));
+        await index_js_1.nodejs.fs.writeFile(constants_js_1.AIGNE_ENV_FILE, (0, yaml_1.stringify)({
+            ...envs,
+            [host]: {
+                AIGNE_HUB_API_KEY: accessKeyOptions.apiKey,
+                AIGNE_HUB_API_URL: accessKeyOptions.url,
+            },
+            default: {
+                AIGNE_HUB_API_URL: accessKeyOptions.url,
+            },
+        }));
+        return accessKeyOptions;
+    }
+    catch (error) {
+        logger_js_1.logger.error("Failed to connect to AIGNE Hub", error.message);
+        return { apiKey: undefined, url: undefined };
+    }
+}
+const checkConnectionStatus = async (host) => {
+    // aigne-hub access token
+    if (!index_js_1.nodejs.fsSync.existsSync(constants_js_1.AIGNE_ENV_FILE)) {
+        throw new Error("AIGNE_HUB_API_KEY file not found, need to login first");
+    }
+    const data = await index_js_1.nodejs.fs.readFile(constants_js_1.AIGNE_ENV_FILE, "utf8");
+    if (!data.includes("AIGNE_HUB_API_KEY")) {
+        throw new Error("AIGNE_HUB_API_KEY key not found, need to login first");
+    }
+    const envs = (0, yaml_1.parse)(data);
+    if (!envs[host]) {
+        throw new Error("AIGNE_HUB_API_KEY host not found, need to login first");
+    }
+    const env = envs[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: (0, ufo_1.joinURL)(env.AIGNE_HUB_API_URL),
+    };
+};
+exports.checkConnectionStatus = checkConnectionStatus;
+async function loadCredential(options) {
+    const isBlocklet = process.env.BLOCKLET_AIGNE_API_URL && process.env.BLOCKLET_AIGNE_API_PROVIDER;
+    if (isBlocklet)
+        return undefined;
+    const aigneDir = index_js_1.nodejs.path.join(index_js_1.nodejs.os.homedir(), ".aigne");
+    if (!index_js_1.nodejs.fsSync.existsSync(aigneDir)) {
+        index_js_1.nodejs.fsSync.mkdirSync(aigneDir, { recursive: true });
+    }
+    const envs = (0, yaml_1.parse)(await index_js_1.nodejs.fs.readFile(constants_js_1.AIGNE_ENV_FILE, "utf8").catch(() => (0, yaml_1.stringify)({})));
+    const inquirerPrompt = (options?.inquirerPromptFn ?? inquirer_1.default.prompt);
+    const configUrl = options?.aigneHubUrl || process.env.AIGNE_HUB_API_URL;
+    const AIGNE_HUB_URL = configUrl || envs?.default?.AIGNE_HUB_API_URL || constants_js_1.AIGNE_HUB_URL;
+    const connectUrl = (0, ufo_1.joinURL)(new URL(AIGNE_HUB_URL).origin, constants_js_1.WELLKNOWN_SERVICE_PATH_PREFIX);
+    const { host } = new URL(AIGNE_HUB_URL);
+    const modelName = options?.model || "";
+    const isAIGNEHubModel = (modelName.toLocaleLowerCase() || "").includes(constants_js_1.AGENT_HUB_PROVIDER);
+    let credential = {};
+    if (isAIGNEHubModel) {
+        try {
+            credential = await (0, exports.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(constants_js_1.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 {
+                                    const url = new URL(input);
+                                    return url.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. Exiting...");
+                        process.exit(0);
+                    }
+                }
+                credential = await connectToAIGNEHub(aigneHubUrl);
+            }
+        }
+    }
+    return credential;
+}

package/lib/cjs/util/crypto.d.ts

@@ -0,0 +1,4 @@
+export declare const decrypt: (m: string, s: string, i: string) => string;
+export declare const encrypt: (m: string, s: string, i: string) => string;
+export declare const encodeEncryptionKey: (key: string) => string;
+export declare const decodeEncryptionKey: (str: string) => Uint8Array<ArrayBuffer>;

package/lib/cjs/util/crypto.js

@@ -0,0 +1,19 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.decodeEncryptionKey = exports.encodeEncryptionKey = exports.encrypt = exports.decrypt = void 0;
+const node_crypto_1 = __importDefault(require("node:crypto"));
+const aes_legacy_js_1 = require("@ocap/mcrypto/lib/crypter/aes-legacy.js");
+const aes = new aes_legacy_js_1.AesCrypter();
+const decrypt = (m, s, i) => aes.decrypt(m, node_crypto_1.default.pbkdf2Sync(i, s, 256, 32, "sha512").toString("hex"));
+exports.decrypt = decrypt;
+const encrypt = (m, s, i) => aes.encrypt(m, node_crypto_1.default.pbkdf2Sync(i, s, 256, 32, "sha512").toString("hex"));
+exports.encrypt = encrypt;
+const escapeFn = (str) => str.replace(/\+/g, "-").replace(/\//g, "_").replace(/=/g, "");
+const unescapeFn = (str) => (str + "===".slice((str.length + 3) % 4)).replace(/-/g, "+").replace(/_/g, "/");
+const encodeEncryptionKey = (key) => escapeFn(Buffer.from(key).toString("base64"));
+exports.encodeEncryptionKey = encodeEncryptionKey;
+const decodeEncryptionKey = (str) => new Uint8Array(Buffer.from(unescapeFn(str), "base64"));
+exports.decodeEncryptionKey = decodeEncryptionKey;

package/lib/cjs/util/model.d.ts

@@ -0,0 +1,12 @@
+import type { ChatModel, ChatModelOptions } from "@aigne/core/agents/chat-model.js";
+import type inquirer from "inquirer";
+import type { LoadableModel, LoadCredentialOptions, Model } from "./type.js";
+export declare function availableModels(): LoadableModel[];
+export declare function findModel(models: LoadableModel[], provider: string): LoadableModel | undefined;
+export declare const parseModelOption: (model?: string) => {
+    provider: string | undefined;
+    name: string | undefined;
+};
+export declare const formatModelName: (model: string, inquirerPrompt: typeof inquirer.prompt) => Promise<string>;
+export declare function maskApiKey(apiKey?: string): string | undefined;
+export declare function loadModel(model?: Model, modelOptions?: ChatModelOptions, options?: LoadCredentialOptions): Promise<ChatModel | undefined>;

package/lib/cjs/util/model.js

@@ -0,0 +1,201 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.formatModelName = exports.parseModelOption = void 0;
+exports.availableModels = availableModels;
+exports.findModel = findModel;
+exports.maskApiKey = maskApiKey;
+exports.loadModel = loadModel;
+const anthropic_1 = require("@aigne/anthropic");
+const bedrock_1 = require("@aigne/bedrock");
+const deepseek_1 = require("@aigne/deepseek");
+const gemini_1 = require("@aigne/gemini");
+const ollama_1 = require("@aigne/ollama");
+const open_router_1 = require("@aigne/open-router");
+const openai_1 = require("@aigne/openai");
+const index_js_1 = require("@aigne/platform-helpers/nodejs/index.js");
+const xai_1 = require("@aigne/xai");
+const node_http_handler_1 = require("@smithy/node-http-handler");
+const boxen_1 = __importDefault(require("boxen"));
+const chalk_1 = __importDefault(require("chalk"));
+const https_proxy_agent_1 = require("https-proxy-agent");
+const index_js_2 = require("../index.js");
+const constants_js_1 = require("./constants.js");
+const credential_js_1 = require("./credential.js");
+const { MODEL_PROVIDER, MODEL_NAME } = index_js_1.nodejs.env;
+function availableModels() {
+    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_1.HttpsProxyAgent(proxy) : undefined;
+    const clientOptions = {
+        fetchOptions: {
+            // @ts-ignore
+            agent: httpAgent,
+        },
+    };
+    return [
+        {
+            name: openai_1.OpenAIChatModel.name,
+            apiKeyEnvName: "OPENAI_API_KEY",
+            create: (params) => new openai_1.OpenAIChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: anthropic_1.AnthropicChatModel.name,
+            apiKeyEnvName: "ANTHROPIC_API_KEY",
+            create: (params) => new anthropic_1.AnthropicChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: bedrock_1.BedrockChatModel.name,
+            apiKeyEnvName: "AWS_ACCESS_KEY_ID",
+            create: (params) => new bedrock_1.BedrockChatModel({
+                ...params,
+                clientOptions: {
+                    requestHandler: node_http_handler_1.NodeHttpHandler.create({ httpAgent, httpsAgent: httpAgent }),
+                    streamCollector: node_http_handler_1.streamCollector,
+                },
+            }),
+        },
+        {
+            name: deepseek_1.DeepSeekChatModel.name,
+            apiKeyEnvName: "DEEPSEEK_API_KEY",
+            create: (params) => new deepseek_1.DeepSeekChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: [gemini_1.GeminiChatModel.name, "google"],
+            apiKeyEnvName: ["GEMINI_API_KEY", "GOOGLE_API_KEY"],
+            create: (params) => new gemini_1.GeminiChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: ollama_1.OllamaChatModel.name,
+            apiKeyEnvName: "OLLAMA_API_KEY",
+            create: (params) => new ollama_1.OllamaChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: open_router_1.OpenRouterChatModel.name,
+            apiKeyEnvName: "OPEN_ROUTER_API_KEY",
+            create: (params) => new open_router_1.OpenRouterChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: xai_1.XAIChatModel.name,
+            apiKeyEnvName: "XAI_API_KEY",
+            create: (params) => new xai_1.XAIChatModel({ ...params, clientOptions }),
+        },
+        {
+            name: index_js_2.AIGNEHubChatModel.name,
+            apiKeyEnvName: "AIGNE_HUB_API_KEY",
+            create: (params) => new index_js_2.AIGNEHubChatModel({ ...params, clientOptions }),
+        },
+    ];
+}
+function findModel(models, provider) {
+    return models.find((m) => {
+        if (typeof m.name === "string") {
+            return m.name.toLowerCase().includes(provider.toLowerCase());
+        }
+        return m.name.some((n) => n.toLowerCase().includes(provider.toLowerCase()));
+    });
+}
+const parseModelOption = (model) => {
+    const { provider, name } = (model || process.env.MODEL)?.match(/(?<provider>[^:]*)(:(?<name>(\S+)))?/)?.groups ?? {};
+    return { provider, name };
+};
+exports.parseModelOption = parseModelOption;
+const formatModelName = async (model, inquirerPrompt) => {
+    const models = availableModels();
+    if (!model)
+        return constants_js_1.DEFAULT_AIGNE_HUB_PROVIDER_MODEL;
+    const { provider, name } = (0, exports.parseModelOption)(model);
+    if (!provider) {
+        return constants_js_1.DEFAULT_AIGNE_HUB_PROVIDER_MODEL;
+    }
+    const providerName = provider.replace(/-/g, "");
+    if (providerName.includes(constants_js_1.AGENT_HUB_PROVIDER)) {
+        return model;
+    }
+    const m = findModel(models, providerName);
+    if (!m)
+        throw new Error(`Unsupported model: ${provider} ${name}`);
+    const apiKeyEnvName = Array.isArray(m.apiKeyEnvName) ? m.apiKeyEnvName : [m.apiKeyEnvName];
+    if (apiKeyEnvName.some((name) => name && process.env[name])) {
+        return model;
+    }
+    const result = await inquirerPrompt({
+        type: "list",
+        name: "useAigneHub",
+        message: `Seems no API Key configured for ${provider}/${name}, select your preferred way to continue:`,
+        choices: [
+            {
+                name: `Connect to AIGNE Hub to use ${name} (Recommended since free credits available)`,
+                value: true,
+            },
+            {
+                name: `Exit and bring my owner API Key by set ${apiKeyEnvName.join(", ")}`,
+                value: false,
+            },
+        ],
+        default: true,
+    });
+    if (!result.useAigneHub) {
+        console.log(chalk_1.default.yellow(`You can use command "export ${apiKeyEnvName[0]}=xxx" to set API Key in your shell. Or you can set environment variables in .env file.`));
+        process.exit(0);
+    }
+    return `${constants_js_1.AGENT_HUB_PROVIDER}:${provider}/${name}`;
+};
+exports.formatModelName = formatModelName;
+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 printChatModelInfoBox({ provider, model, credential, m, }) {
+    const lines = [
+        `${chalk_1.default.cyan("Provider")}: ${chalk_1.default.green(provider)}`,
+        `${chalk_1.default.cyan("Model")}: ${chalk_1.default.green(model)}`,
+    ];
+    if (provider.includes(constants_js_1.AGENT_HUB_PROVIDER)) {
+        lines.push(`${chalk_1.default.cyan("API URL")}: ${chalk_1.default.green(credential?.url || "N/A")}`, `${chalk_1.default.cyan("API Key")}: ${chalk_1.default.green(maskApiKey(credential?.apiKey))}`);
+    }
+    else {
+        const apiKeyEnvName = Array.isArray(m.apiKeyEnvName) ? m.apiKeyEnvName : [m.apiKeyEnvName];
+        const envKeyName = apiKeyEnvName.find((name) => name && process.env[name]);
+        if (envKeyName) {
+            lines.push(`${chalk_1.default.cyan("API Key")}: ${chalk_1.default.green(maskApiKey(process.env[envKeyName]))}`);
+        }
+        else {
+            lines.push(`${chalk_1.default.cyan("API Key")}: ${chalk_1.default.yellow("Not found")}`);
+        }
+    }
+    console.log("\n");
+    console.log((0, boxen_1.default)(lines.join("\n"), {
+        padding: 1,
+        borderStyle: "classic",
+        borderColor: "cyan",
+    }));
+    console.log("\n");
+}
+async function loadModel(model, modelOptions, options) {
+    const params = {
+        model: MODEL_NAME ?? model?.name ?? undefined,
+        temperature: model?.temperature ?? undefined,
+        topP: model?.topP ?? undefined,
+        frequencyPenalty: model?.frequencyPenalty ?? undefined,
+        presencePenalty: model?.presencePenalty ?? undefined,
+    };
+    const provider = (MODEL_PROVIDER ?? model?.provider ?? constants_js_1.DEFAULT_MODEL_PROVIDER).replace(/-/g, "");
+    const models = availableModels();
+    const m = findModel(models, provider);
+    if (!m)
+        throw new Error(`Unsupported model: ${model?.provider} ${model?.name}`);
+    const credential = await (0, credential_js_1.loadCredential)({ ...options, model: `${provider}:${params.model}` });
+    printChatModelInfoBox({ provider, model: params.model || "", credential, m });
+    return m.create({
+        ...(credential || {}),
+        model: params.model,
+        modelOptions: { ...params, ...modelOptions },
+    });
+}