@aigne/aigne-hub 0.4.8 → 0.4.9

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

@@ -0,0 +1,58 @@
+import type { ChatModel, ChatModelOptions } from "@aigne/core/agents/chat-model.js";
+export type Model = {
+    provider?: string | null;
+    name?: string | null;
+    temperature?: number | null;
+    topP?: number | null;
+    frequencyPenalty?: number | null;
+    presencePenalty?: number | null;
+} | undefined;
+type InquirerPromptFn = (prompt: {
+    type: string;
+    name: string;
+    message: string;
+    choices: {
+        name: string;
+        value: any;
+    }[];
+    default: any;
+}) => Promise<any>;
+export type LoadCredentialOptions = {
+    model?: string;
+    aigneHubUrl?: string;
+    inquirerPromptFn?: InquirerPromptFn;
+};
+export interface LoadableModel {
+    name: string | string[];
+    apiKeyEnvName?: string | string[];
+    create: (options: {
+        model?: string;
+        modelOptions?: ChatModelOptions;
+        apiKey?: string;
+        url?: string;
+    }) => ChatModel;
+}
+export type FetchResult = {
+    accessKeyId: string;
+    accessKeySecret: string;
+};
+export type BaseWrapSpinner = (_: string, waiting: () => Promise<FetchResult>) => Promise<FetchResult>;
+export interface CreateConnectOptions {
+    connectUrl: string;
+    openPage?: (url: string) => void;
+    fetchInterval?: number;
+    retry?: number;
+    source?: string;
+    connectAction?: string;
+    appName?: string;
+    appLogo?: string;
+    wrapSpinner?: BaseWrapSpinner;
+    prettyUrl?: (url: string) => string;
+    closeOnSuccess?: boolean;
+    intervalFetchConfig?: (options: {
+        sessionId: string;
+        fetchInterval: number;
+        fetchTimeout: number;
+    }) => Promise<FetchResult>;
+}
+export {};

package/lib/cjs/util/type.js

@@ -0,0 +1,2 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });

package/lib/dts/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/dts/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/dts/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/dts/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/dts/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/dts/util/type.d.ts

@@ -0,0 +1,58 @@
+import type { ChatModel, ChatModelOptions } from "@aigne/core/agents/chat-model.js";
+export type Model = {
+    provider?: string | null;
+    name?: string | null;
+    temperature?: number | null;
+    topP?: number | null;
+    frequencyPenalty?: number | null;
+    presencePenalty?: number | null;
+} | undefined;
+type InquirerPromptFn = (prompt: {
+    type: string;
+    name: string;
+    message: string;
+    choices: {
+        name: string;
+        value: any;
+    }[];
+    default: any;
+}) => Promise<any>;
+export type LoadCredentialOptions = {
+    model?: string;
+    aigneHubUrl?: string;
+    inquirerPromptFn?: InquirerPromptFn;
+};
+export interface LoadableModel {
+    name: string | string[];
+    apiKeyEnvName?: string | string[];
+    create: (options: {
+        model?: string;
+        modelOptions?: ChatModelOptions;
+        apiKey?: string;
+        url?: string;
+    }) => ChatModel;
+}
+export type FetchResult = {
+    accessKeyId: string;
+    accessKeySecret: string;
+};
+export type BaseWrapSpinner = (_: string, waiting: () => Promise<FetchResult>) => Promise<FetchResult>;
+export interface CreateConnectOptions {
+    connectUrl: string;
+    openPage?: (url: string) => void;
+    fetchInterval?: number;
+    retry?: number;
+    source?: string;
+    connectAction?: string;
+    appName?: string;
+    appLogo?: string;
+    wrapSpinner?: BaseWrapSpinner;
+    prettyUrl?: (url: string) => string;
+    closeOnSuccess?: boolean;
+    intervalFetchConfig?: (options: {
+        sessionId: string;
+        fetchInterval: number;
+        fetchTimeout: number;
+    }) => Promise<FetchResult>;
+}
+export {};

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

@@ -1,5 +1,6 @@
 import { ChatModel, } from "@aigne/core";
-import { AIGNE_HUB_URL, availableModels, findModel } from "./constants.js";
+import { AIGNE_HUB_URL } from "./util/constants.js";
+import { availableModels, findModel } from "./util/model.js";
 export class BlockletAIGNEHubChatModel extends ChatModel {
     options;
     client;

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

@@ -3,7 +3,7 @@ import { checkArguments } from "@aigne/core/utils/type-utils.js";
 import { BaseClient } from "@aigne/transport/http-client/base-client.js";
 import { joinURL } from "ufo";
 import { z } from "zod";
-import { AIGNE_HUB_URL } from "./constants.js";
+import { AIGNE_HUB_URL } from "./util/constants.js";
 const DEFAULT_CHAT_MODEL = "openai/gpt-4o";
 const aigneHubChatModelOptionsSchema = z.object({
     url: z.string().optional(),

package/lib/esm/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/esm/index.js

@@ -1,8 +1,13 @@
 import { ChatModel, } from "@aigne/core";
 import { BlockletAIGNEHubChatModel } from "./blocklet-aigne-hub-model.js";
 import { CliAIGNEHubChatModel } from "./cli-aigne-hub-model.js";
-import { AIGNE_HUB_URL, getAIGNEHubMountPoint } from "./constants.js";
-export * from "./constants.js";
+import { AIGNE_HUB_URL } from "./util/constants.js";
+import { getAIGNEHubMountPoint } from "./util/credential.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 class AIGNEHubChatModel extends ChatModel {
     options;
     client;

package/lib/esm/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/esm/util/constants.js

@@ -0,0 +1,15 @@
+import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+const WELLKNOWN_SERVICE_PATH_PREFIX = "/.well-known/service";
+const ACCESS_KEY_PREFIX = "/api/access-key";
+const ACCESS_KEY_SESSION_API = `${ACCESS_KEY_PREFIX}/session`;
+const AIGNE_HUB_DID = "z8ia3xzq2tMq8CRHfaXj1BTYJyYnEcHbqP8cJ";
+const AGENT_HUB_PROVIDER = "aignehub";
+const AIGNE_HUB_URL = "https://hub.aigne.io/";
+const DEFAULT_AIGNE_HUB_MODEL = "openai/gpt-5-mini";
+const DEFAULT_AIGNE_HUB_PROVIDER_MODEL = `${AGENT_HUB_PROVIDER}:${DEFAULT_AIGNE_HUB_MODEL}`;
+const DEFAULT_MODEL_PROVIDER = "openai";
+const isTest = process.env.CI || process.env.NODE_ENV === "test";
+const TEST_AIGNE_ENV_FILE = nodejs.path.join(nodejs.os.homedir(), ".aigne", "test-aigne-hub-connected.yaml");
+const PROD_AIGNE_ENV_FILE = nodejs.path.join(nodejs.os.homedir(), ".aigne", "aigne-hub-connected.yaml");
+const AIGNE_ENV_FILE = isTest ? TEST_AIGNE_ENV_FILE : PROD_AIGNE_ENV_FILE;
+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/esm/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/esm/util/credential.js

@@ -0,0 +1,211 @@
+import { logger } from "@aigne/core/utils/logger.js";
+import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
+import inquirer from "inquirer";
+import open from "open";
+import pWaitFor from "p-wait-for";
+import { joinURL, withQuery } from "ufo";
+import { parse, stringify } from "yaml";
+import { ACCESS_KEY_SESSION_API, AGENT_HUB_PROVIDER, AIGNE_ENV_FILE, AIGNE_HUB_DID, AIGNE_HUB_URL as DEFAULT_AIGNE_HUB_URL, isTest, WELLKNOWN_SERVICE_PATH_PREFIX, } from "./constants.js";
+import { decrypt, encodeEncryptionKey } from "./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 };
+};
+export 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 } = await request({ url: sessionURL, requestCount });
+        requestCount++;
+        return Boolean(session.accessKeyId && session.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 Promise.resolve(waiting());
+}
+export 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 = joinURL(connectUrl, ACCESS_KEY_SESSION_API);
+    const { data: session } = await request({ url: startSessionURL, 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 () => {
+        const checkAuthorizeStatus = intervalFetchConfig ?? fetchConfigs;
+        const authorizeStatus = await checkAuthorizeStatus({
+            connectUrl,
+            sessionId: token,
+            fetchTimeout: retry * fetchInterval,
+            fetchInterval: retry,
+        });
+        return authorizeStatus;
+    });
+}
+export async function getAIGNEHubMountPoint(url) {
+    const { origin } = new URL(url);
+    const BLOCKLET_JSON_PATH = "__blocklet__.js?type=json";
+    const blockletInfo = await fetch(joinURL(origin, BLOCKLET_JSON_PATH));
+    const blocklet = await blockletInfo.json();
+    const aigneHubMount = (blocklet?.componentMountPoints || []).find((m) => m.did === AIGNE_HUB_DID);
+    return joinURL(origin, aigneHubMount?.mountPoint || "");
+}
+export async function connectToAIGNEHub(url) {
+    const { origin, host } = new URL(url);
+    const connectUrl = joinURL(origin, WELLKNOWN_SERVICE_PATH_PREFIX);
+    const urlWithAIGNEHubMountPoint = await getAIGNEHubMountPoint(url);
+    try {
+        const openFn = isTest ? () => { } : open;
+        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 = nodejs.path.join(nodejs.os.homedir(), ".aigne");
+        if (!nodejs.fsSync.existsSync(aigneDir)) {
+            nodejs.fsSync.mkdirSync(aigneDir, { recursive: true });
+        }
+        const envs = parse(await nodejs.fs.readFile(AIGNE_ENV_FILE, "utf8").catch(() => stringify({})));
+        await nodejs.fs.writeFile(AIGNE_ENV_FILE, 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.error("Failed to connect to AIGNE Hub", error.message);
+        return { apiKey: undefined, url: undefined };
+    }
+}
+export const checkConnectionStatus = async (host) => {
+    // aigne-hub access token
+    if (!nodejs.fsSync.existsSync(AIGNE_ENV_FILE)) {
+        throw new Error("AIGNE_HUB_API_KEY file not found, need to login first");
+    }
+    const data = await nodejs.fs.readFile(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 = 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: joinURL(env.AIGNE_HUB_API_URL),
+    };
+};
+export async function loadCredential(options) {
+    const isBlocklet = process.env.BLOCKLET_AIGNE_API_URL && process.env.BLOCKLET_AIGNE_API_PROVIDER;
+    if (isBlocklet)
+        return undefined;
+    const aigneDir = nodejs.path.join(nodejs.os.homedir(), ".aigne");
+    if (!nodejs.fsSync.existsSync(aigneDir)) {
+        nodejs.fsSync.mkdirSync(aigneDir, { recursive: true });
+    }
+    const envs = parse(await nodejs.fs.readFile(AIGNE_ENV_FILE, "utf8").catch(() => stringify({})));
+    const inquirerPrompt = (options?.inquirerPromptFn ?? inquirer.prompt);
+    const configUrl = options?.aigneHubUrl || process.env.AIGNE_HUB_API_URL;
+    const AIGNE_HUB_URL = configUrl || envs?.default?.AIGNE_HUB_API_URL || DEFAULT_AIGNE_HUB_URL;
+    const connectUrl = joinURL(new URL(AIGNE_HUB_URL).origin, WELLKNOWN_SERVICE_PATH_PREFIX);
+    const { host } = new URL(AIGNE_HUB_URL);
+    const modelName = options?.model || "";
+    const isAIGNEHubModel = (modelName.toLocaleLowerCase() || "").includes(AGENT_HUB_PROVIDER);
+    let credential = {};
+    if (isAIGNEHubModel) {
+        try {
+            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(DEFAULT_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/esm/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/esm/util/crypto.js

@@ -0,0 +1,9 @@
+import crypto from "node:crypto";
+import { AesCrypter } from "@ocap/mcrypto/lib/crypter/aes-legacy.js";
+const aes = new AesCrypter();
+export const decrypt = (m, s, i) => aes.decrypt(m, crypto.pbkdf2Sync(i, s, 256, 32, "sha512").toString("hex"));
+export const encrypt = (m, s, i) => aes.encrypt(m, crypto.pbkdf2Sync(i, s, 256, 32, "sha512").toString("hex"));
+const escapeFn = (str) => str.replace(/\+/g, "-").replace(/\//g, "_").replace(/=/g, "");
+const unescapeFn = (str) => (str + "===".slice((str.length + 3) % 4)).replace(/-/g, "+").replace(/_/g, "/");
+export const encodeEncryptionKey = (key) => escapeFn(Buffer.from(key).toString("base64"));
+export const decodeEncryptionKey = (str) => new Uint8Array(Buffer.from(unescapeFn(str), "base64"));

package/lib/esm/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>;