@dexto/agent-management 1.3.0 → 1.5.0

package/dist/installation.d.ts

@@ -0,0 +1,72 @@
+import type { AgentMetadata } from './AgentManager.js';
+export interface InstallOptions {
+    /** Directory where agents are stored (default: ~/.dexto/agents) */
+    agentsDir?: string;
+}
+/**
+ * Install agent from bundled registry to local directory
+ *
+ * @param agentId ID of the agent to install from bundled registry
+ * @param options Installation options
+ * @returns Path to the installed agent's main config file
+ *
+ * @throws {DextoRuntimeError} If agent not found in bundled registry or installation fails
+ *
+ * @example
+ * ```typescript
+ * await installBundledAgent('coding-agent');
+ * console.log('Agent installed to ~/.dexto/agents/coding-agent');
+ * ```
+ */
+export declare function installBundledAgent(agentId: string, options?: InstallOptions): Promise<string>;
+/**
+ * Install custom agent from local path
+ *
+ * @param agentId Unique ID for the custom agent
+ * @param sourcePath Absolute path to agent YAML file or directory
+ * @param metadata Agent metadata (name, description, author, tags)
+ * @param options Installation options
+ * @returns Path to the installed agent's main config file
+ *
+ * @throws {DextoRuntimeError} If agent ID already exists or installation fails
+ *
+ * @example
+ * ```typescript
+ * await installCustomAgent('my-agent', '/path/to/agent.yml', {
+ *   name: 'My Agent',
+ *   description: 'Custom agent for my use case',
+ *   author: 'John Doe',
+ *   tags: ['custom']
+ * });
+ * ```
+ */
+export declare function installCustomAgent(agentId: string, sourcePath: string, metadata: Pick<AgentMetadata, 'name' | 'description' | 'author' | 'tags'>, options?: InstallOptions): Promise<string>;
+/**
+ * Uninstall agent by removing it from disk and user registry
+ *
+ * @param agentId ID of the agent to uninstall
+ * @param options Installation options
+ *
+ * @throws {DextoRuntimeError} If agent not installed
+ *
+ * @example
+ * ```typescript
+ * await uninstallAgent('my-custom-agent');
+ * console.log('Agent uninstalled');
+ * ```
+ */
+export declare function uninstallAgent(agentId: string, options?: InstallOptions): Promise<void>;
+/**
+ * List installed agents
+ *
+ * @param options Installation options
+ * @returns Array of installed agent IDs
+ *
+ * @example
+ * ```typescript
+ * const installed = await listInstalledAgents();
+ * console.log(installed); // ['coding-agent', 'my-custom-agent']
+ * ```
+ */
+export declare function listInstalledAgents(options?: InstallOptions): Promise<string[]>;
+//# sourceMappingURL=installation.d.ts.map

package/dist/installation.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"installation.d.ts","sourceRoot":"","sources":["../src/installation.ts"],"names":[],"mappings":"AAaA,OAAO,KAAK,EAAE,aAAa,EAAE,MAAM,mBAAmB,CAAC;AAEvD,MAAM,WAAW,cAAc;IAC3B,mEAAmE;IACnE,SAAS,CAAC,EAAE,MAAM,CAAC;CACtB;AAuCD;;;;;;;;;;;;;;GAcG;AACH,wBAAsB,mBAAmB,CACrC,OAAO,EAAE,MAAM,EACf,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,MAAM,CAAC,CA6FjB;AAED;;;;;;;;;;;;;;;;;;;;GAoBG;AACH,wBAAsB,kBAAkB,CACpC,OAAO,EAAE,MAAM,EACf,UAAU,EAAE,MAAM,EAClB,QAAQ,EAAE,IAAI,CAAC,aAAa,EAAE,MAAM,GAAG,aAAa,GAAG,QAAQ,GAAG,MAAM,CAAC,EACzE,OAAO,CAAC,EAAE,cAAc,GACzB,OAAO,CAAC,MAAM,CAAC,CA2FjB;AAED;;;;;;;;;;;;;GAaG;AACH,wBAAsB,cAAc,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,IAAI,CAAC,CA6B7F;AAED;;;;;;;;;;;GAWG;AACH,wBAAsB,mBAAmB,CAAC,OAAO,CAAC,EAAE,cAAc,GAAG,OAAO,CAAC,MAAM,EAAE,CAAC,CAYrF"}

package/dist/installation.js

@@ -0,0 +1,202 @@
+import { promises as fs } from "fs";
+import path from "path";
+import { logger } from "@dexto/core";
+import { getDextoGlobalPath, resolveBundledScript, copyDirectory } from "./utils/path.js";
+import { RegistryError } from "./registry/errors.js";
+import { ConfigError } from "./config/errors.js";
+function getAgentsDir(options) {
+  return options?.agentsDir ?? getDextoGlobalPath("agents");
+}
+function getUserRegistryPath(agentsDir) {
+  return path.join(agentsDir, "registry.json");
+}
+async function loadUserRegistry(registryPath) {
+  try {
+    const content = await fs.readFile(registryPath, "utf-8");
+    return JSON.parse(content);
+  } catch (error) {
+    if (error.code === "ENOENT") {
+      return { agents: [] };
+    }
+    throw error;
+  }
+}
+async function saveUserRegistry(registryPath, registry) {
+  await fs.mkdir(path.dirname(registryPath), { recursive: true });
+  await fs.writeFile(registryPath, JSON.stringify(registry, null, 2));
+}
+async function installBundledAgent(agentId, options) {
+  const agentsDir = getAgentsDir(options);
+  const bundledRegistryPath = resolveBundledScript("agents/agent-registry.json");
+  logger.info(`Installing agent: ${agentId}`);
+  let bundledRegistry;
+  try {
+    const content = await fs.readFile(bundledRegistryPath, "utf-8");
+    bundledRegistry = JSON.parse(content);
+  } catch (error) {
+    throw RegistryError.registryParseError(
+      bundledRegistryPath,
+      error instanceof Error ? error.message : String(error)
+    );
+  }
+  const agentEntry = bundledRegistry.agents[agentId];
+  if (!agentEntry) {
+    const available = Object.keys(bundledRegistry.agents);
+    throw RegistryError.agentNotFound(agentId, available);
+  }
+  const targetDir = path.join(agentsDir, agentId);
+  try {
+    await fs.access(targetDir);
+    logger.info(`Agent '${agentId}' already installed`);
+    const mainFile = agentEntry.main || path.basename(agentEntry.source);
+    return path.join(targetDir, mainFile);
+  } catch {
+  }
+  await fs.mkdir(agentsDir, { recursive: true });
+  const sourcePath = resolveBundledScript(`agents/${agentEntry.source}`);
+  const tempDir = `${targetDir}.tmp.${Date.now()}`;
+  try {
+    if (agentEntry.source.endsWith("/")) {
+      await copyDirectory(sourcePath, tempDir);
+    } else {
+      await fs.mkdir(tempDir, { recursive: true });
+      const targetFile = path.join(tempDir, path.basename(sourcePath));
+      await fs.copyFile(sourcePath, targetFile);
+    }
+    await fs.rename(tempDir, targetDir);
+    logger.info(`\u2713 Installed agent '${agentId}' to ${targetDir}`);
+    const userRegistryPath = getUserRegistryPath(agentsDir);
+    const userRegistry = await loadUserRegistry(userRegistryPath);
+    if (!userRegistry.agents.some((a) => a.id === agentId)) {
+      const mainFile = agentEntry.main || path.basename(agentEntry.source);
+      userRegistry.agents.push({
+        id: agentId,
+        name: agentEntry.name,
+        description: agentEntry.description,
+        configPath: `./${agentId}/${mainFile}`,
+        author: agentEntry.author,
+        tags: agentEntry.tags
+      });
+      await saveUserRegistry(userRegistryPath, userRegistry);
+    }
+    return path.join(targetDir, agentEntry.main || path.basename(agentEntry.source));
+  } catch (error) {
+    try {
+      await fs.rm(tempDir, { recursive: true, force: true });
+    } catch {
+    }
+    throw RegistryError.installationFailed(
+      agentId,
+      error instanceof Error ? error.message : String(error)
+    );
+  }
+}
+async function installCustomAgent(agentId, sourcePath, metadata, options) {
+  const agentsDir = getAgentsDir(options);
+  const targetDir = path.join(agentsDir, agentId);
+  logger.info(`Installing custom agent: ${agentId}`);
+  try {
+    const bundledRegistryPath = resolveBundledScript("agents/agent-registry.json");
+    const bundledContent = await fs.readFile(bundledRegistryPath, "utf-8");
+    const bundledRegistry = JSON.parse(bundledContent);
+    if (agentId in bundledRegistry.agents) {
+      throw RegistryError.customAgentNameConflict(agentId);
+    }
+  } catch (error) {
+    if (error instanceof Error && error.message.includes("conflicts with builtin")) {
+      throw error;
+    }
+    logger.debug(
+      `Could not validate against bundled registry: ${error instanceof Error ? error.message : String(error)}`
+    );
+  }
+  try {
+    await fs.access(targetDir);
+    throw RegistryError.agentAlreadyExists(agentId);
+  } catch (error) {
+    if (error.code !== "ENOENT") {
+      throw error;
+    }
+  }
+  const resolvedSource = path.resolve(sourcePath);
+  let stat;
+  try {
+    stat = await fs.stat(resolvedSource);
+  } catch (_error) {
+    throw ConfigError.fileNotFound(resolvedSource);
+  }
+  await fs.mkdir(agentsDir, { recursive: true });
+  try {
+    if (stat.isDirectory()) {
+      await copyDirectory(resolvedSource, targetDir);
+    } else {
+      await fs.mkdir(targetDir, { recursive: true });
+      const filename = path.basename(resolvedSource);
+      await fs.copyFile(resolvedSource, path.join(targetDir, filename));
+    }
+    logger.info(`\u2713 Installed custom agent '${agentId}' to ${targetDir}`);
+    const userRegistryPath = getUserRegistryPath(agentsDir);
+    const userRegistry = await loadUserRegistry(userRegistryPath);
+    const configFile = stat.isDirectory() ? "agent.yml" : path.basename(resolvedSource);
+    userRegistry.agents.push({
+      id: agentId,
+      name: metadata.name || agentId,
+      description: metadata.description,
+      configPath: `./${agentId}/${configFile}`,
+      author: metadata.author,
+      tags: metadata.tags || []
+    });
+    await saveUserRegistry(userRegistryPath, userRegistry);
+    return path.join(targetDir, configFile);
+  } catch (error) {
+    try {
+      await fs.rm(targetDir, { recursive: true, force: true });
+    } catch {
+    }
+    throw RegistryError.installationFailed(
+      agentId,
+      error instanceof Error ? error.message : String(error)
+    );
+  }
+}
+async function uninstallAgent(agentId, options) {
+  const agentsDir = getAgentsDir(options);
+  const targetDir = path.join(agentsDir, agentId);
+  logger.info(`Uninstalling agent: ${agentId}`);
+  try {
+    await fs.access(targetDir);
+  } catch (_error) {
+    throw RegistryError.agentNotInstalled(agentId);
+  }
+  await fs.rm(targetDir, { recursive: true, force: true });
+  logger.info(`\u2713 Removed agent directory: ${targetDir}`);
+  const userRegistryPath = getUserRegistryPath(agentsDir);
+  try {
+    const userRegistry = await loadUserRegistry(userRegistryPath);
+    userRegistry.agents = userRegistry.agents.filter((a) => a.id !== agentId);
+    await saveUserRegistry(userRegistryPath, userRegistry);
+    logger.info(`\u2713 Removed '${agentId}' from user registry`);
+  } catch (error) {
+    logger.warn(
+      `Failed to update user registry: ${error instanceof Error ? error.message : String(error)}`
+    );
+  }
+}
+async function listInstalledAgents(options) {
+  const agentsDir = getAgentsDir(options);
+  try {
+    const entries = await fs.readdir(agentsDir, { withFileTypes: true });
+    return entries.filter((e) => e.isDirectory()).map((e) => e.name);
+  } catch (error) {
+    if (error.code === "ENOENT") {
+      return [];
+    }
+    throw error;
+  }
+}
+export {
+  installBundledAgent,
+  installCustomAgent,
+  listInstalledAgents,
+  uninstallAgent
+};

package/dist/models/custom-models.cjs

@@ -0,0 +1,157 @@
+"use strict";
+var __create = Object.create;
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __getProtoOf = Object.getPrototypeOf;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toESM = (mod, isNodeMode, target) => (target = mod != null ? __create(__getProtoOf(mod)) : {}, __copyProps(
+  // If the importer is in node compatibility mode or this is not an ESM
+  // file that has been converted to a CommonJS file using a Babel-
+  // compatible transform (i.e. "__esModule" has not been set), then set
+  // "default" to the CommonJS "module.exports" for node compatibility.
+  isNodeMode || !mod || !mod.__esModule ? __defProp(target, "default", { value: mod, enumerable: true }) : target,
+  mod
+));
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+var custom_models_exports = {};
+__export(custom_models_exports, {
+  CUSTOM_MODEL_PROVIDERS: () => CUSTOM_MODEL_PROVIDERS,
+  CustomModelSchema: () => CustomModelSchema,
+  deleteCustomModel: () => deleteCustomModel,
+  getCustomModel: () => getCustomModel,
+  getCustomModelsPath: () => getCustomModelsPath,
+  loadCustomModels: () => loadCustomModels,
+  saveCustomModel: () => saveCustomModel
+});
+module.exports = __toCommonJS(custom_models_exports);
+var import_zod = require("zod");
+var import_fs = require("fs");
+var path = __toESM(require("path"), 1);
+var import_path = require("../utils/path.js");
+const CUSTOM_MODEL_PROVIDERS = [
+  "openai-compatible",
+  "openrouter",
+  "litellm",
+  "glama",
+  "bedrock",
+  "ollama",
+  "local",
+  "vertex"
+];
+const CustomModelSchema = import_zod.z.object({
+  name: import_zod.z.string().min(1),
+  provider: import_zod.z.enum(CUSTOM_MODEL_PROVIDERS).default("openai-compatible"),
+  baseURL: import_zod.z.string().url().optional(),
+  displayName: import_zod.z.string().optional(),
+  maxInputTokens: import_zod.z.number().int().positive().optional(),
+  maxOutputTokens: import_zod.z.number().int().positive().optional(),
+  // Optional per-model API key. For openai-compatible this is the primary key source.
+  // For litellm/glama/openrouter this overrides the provider-level env var key.
+  apiKey: import_zod.z.string().optional(),
+  // File path for local GGUF models. Required when provider is 'local'.
+  // Stores the absolute path to the .gguf file on disk.
+  filePath: import_zod.z.string().optional()
+}).superRefine((data, ctx) => {
+  if ((data.provider === "openai-compatible" || data.provider === "litellm") && !data.baseURL) {
+    ctx.addIssue({
+      code: import_zod.z.ZodIssueCode.custom,
+      path: ["baseURL"],
+      message: `Base URL is required for ${data.provider} provider`
+    });
+  }
+  if (data.provider === "local" && !data.filePath) {
+    ctx.addIssue({
+      code: import_zod.z.ZodIssueCode.custom,
+      path: ["filePath"],
+      message: "File path is required for local provider"
+    });
+  }
+  if (data.provider === "local" && data.filePath && !data.filePath.endsWith(".gguf")) {
+    ctx.addIssue({
+      code: import_zod.z.ZodIssueCode.custom,
+      path: ["filePath"],
+      message: "File path must be a .gguf file"
+    });
+  }
+});
+const StorageSchema = import_zod.z.object({
+  version: import_zod.z.literal(1),
+  models: import_zod.z.array(CustomModelSchema)
+});
+function getCustomModelsPath() {
+  return (0, import_path.getDextoGlobalPath)("models", "custom-models.json");
+}
+async function loadCustomModels() {
+  const filePath = getCustomModelsPath();
+  try {
+    const content = await import_fs.promises.readFile(filePath, "utf-8");
+    const parsed = StorageSchema.safeParse(JSON.parse(content));
+    if (!parsed.success) {
+      console.warn(
+        `[custom-models] Failed to parse ${filePath}: ${parsed.error.issues.map((i) => i.message).join(", ")}`
+      );
+      return [];
+    }
+    return parsed.data.models;
+  } catch (error) {
+    if (error.code === "ENOENT") {
+      return [];
+    }
+    throw error;
+  }
+}
+async function saveCustomModel(model) {
+  const parsed = CustomModelSchema.safeParse(model);
+  if (!parsed.success) {
+    throw new Error(`Invalid model: ${parsed.error.issues.map((i) => i.message).join(", ")}`);
+  }
+  const models = await loadCustomModels();
+  const existingIndex = models.findIndex((m) => m.name === parsed.data.name);
+  if (existingIndex >= 0) {
+    models[existingIndex] = parsed.data;
+  } else {
+    models.push(parsed.data);
+  }
+  await writeCustomModels(models);
+}
+async function deleteCustomModel(name) {
+  const models = await loadCustomModels();
+  const filtered = models.filter((m) => m.name !== name);
+  if (filtered.length === models.length) {
+    return false;
+  }
+  await writeCustomModels(filtered);
+  return true;
+}
+async function getCustomModel(name) {
+  const models = await loadCustomModels();
+  return models.find((m) => m.name === name) ?? null;
+}
+async function writeCustomModels(models) {
+  const filePath = getCustomModelsPath();
+  await import_fs.promises.mkdir(path.dirname(filePath), { recursive: true });
+  await import_fs.promises.writeFile(filePath, JSON.stringify({ version: 1, models }, null, 2), "utf-8");
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  CUSTOM_MODEL_PROVIDERS,
+  CustomModelSchema,
+  deleteCustomModel,
+  getCustomModel,
+  getCustomModelsPath,
+  loadCustomModels,
+  saveCustomModel
+});

package/dist/models/custom-models.d.ts

@@ -0,0 +1,94 @@
+/**
+ * Custom Models Persistence
+ *
+ * Manages saved custom model configurations for openai-compatible and openrouter providers.
+ * Stored in ~/.dexto/models/custom-models.json
+ */
+import { z } from 'zod';
+/** Providers that support custom models */
+export declare const CUSTOM_MODEL_PROVIDERS: readonly ["openai-compatible", "openrouter", "litellm", "glama", "bedrock", "ollama", "local", "vertex"];
+export type CustomModelProvider = (typeof CUSTOM_MODEL_PROVIDERS)[number];
+/**
+ * Schema for a saved custom model configuration.
+ * - openai-compatible: requires baseURL, optional per-model apiKey
+ * - openrouter: baseURL is auto-injected, maxInputTokens from registry
+ * - litellm: requires baseURL, uses LITELLM_API_KEY or per-model override
+ * - glama: fixed baseURL, uses GLAMA_API_KEY or per-model override
+ * - bedrock: no baseURL, uses AWS credentials from environment
+ * - ollama: optional baseURL (defaults to http://localhost:11434)
+ * - local: no baseURL, uses local GGUF files via node-llama-cpp
+ * - vertex: no baseURL, uses Google Cloud ADC
+ *
+ * TODO: For hosted deployments, API keys should be stored in a secure
+ * key management service (e.g., AWS Secrets Manager, HashiCorp Vault)
+ * rather than in the local JSON file. Current approach is suitable for
+ * local CLI usage where the file is in ~/.dexto/ (user-private).
+ */
+export declare const CustomModelSchema: z.ZodEffects<z.ZodObject<{
+    name: z.ZodString;
+    provider: z.ZodDefault<z.ZodEnum<["openai-compatible", "openrouter", "litellm", "glama", "bedrock", "ollama", "local", "vertex"]>>;
+    baseURL: z.ZodOptional<z.ZodString>;
+    displayName: z.ZodOptional<z.ZodString>;
+    maxInputTokens: z.ZodOptional<z.ZodNumber>;
+    maxOutputTokens: z.ZodOptional<z.ZodNumber>;
+    apiKey: z.ZodOptional<z.ZodString>;
+    filePath: z.ZodOptional<z.ZodString>;
+}, "strip", z.ZodTypeAny, {
+    name: string;
+    provider: "openai-compatible" | "openrouter" | "litellm" | "glama" | "vertex" | "bedrock" | "local" | "ollama";
+    apiKey?: string | undefined;
+    baseURL?: string | undefined;
+    filePath?: string | undefined;
+    displayName?: string | undefined;
+    maxInputTokens?: number | undefined;
+    maxOutputTokens?: number | undefined;
+}, {
+    name: string;
+    provider?: "openai-compatible" | "openrouter" | "litellm" | "glama" | "vertex" | "bedrock" | "local" | "ollama" | undefined;
+    apiKey?: string | undefined;
+    baseURL?: string | undefined;
+    filePath?: string | undefined;
+    displayName?: string | undefined;
+    maxInputTokens?: number | undefined;
+    maxOutputTokens?: number | undefined;
+}>, {
+    name: string;
+    provider: "openai-compatible" | "openrouter" | "litellm" | "glama" | "vertex" | "bedrock" | "local" | "ollama";
+    apiKey?: string | undefined;
+    baseURL?: string | undefined;
+    filePath?: string | undefined;
+    displayName?: string | undefined;
+    maxInputTokens?: number | undefined;
+    maxOutputTokens?: number | undefined;
+}, {
+    name: string;
+    provider?: "openai-compatible" | "openrouter" | "litellm" | "glama" | "vertex" | "bedrock" | "local" | "ollama" | undefined;
+    apiKey?: string | undefined;
+    baseURL?: string | undefined;
+    filePath?: string | undefined;
+    displayName?: string | undefined;
+    maxInputTokens?: number | undefined;
+    maxOutputTokens?: number | undefined;
+}>;
+export type CustomModel = z.output<typeof CustomModelSchema>;
+/**
+ * Get the path to the custom models storage file.
+ */
+export declare function getCustomModelsPath(): string;
+/**
+ * Load custom models from storage.
+ */
+export declare function loadCustomModels(): Promise<CustomModel[]>;
+/**
+ * Save a custom model to storage.
+ */
+export declare function saveCustomModel(model: CustomModel): Promise<void>;
+/**
+ * Delete a custom model by name.
+ */
+export declare function deleteCustomModel(name: string): Promise<boolean>;
+/**
+ * Get a specific custom model by name.
+ */
+export declare function getCustomModel(name: string): Promise<CustomModel | null>;
+//# sourceMappingURL=custom-models.d.ts.map

package/dist/models/custom-models.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"custom-models.d.ts","sourceRoot":"","sources":["../../src/models/custom-models.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAKxB,2CAA2C;AAC3C,eAAO,MAAM,sBAAsB,0GASzB,CAAC;AACX,MAAM,MAAM,mBAAmB,GAAG,CAAC,OAAO,sBAAsB,CAAC,CAAC,MAAM,CAAC,CAAC;AAE1E;;;;;;;;;;;;;;;GAeG;AACH,eAAO,MAAM,iBAAiB;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2CxB,CAAC;AAEP,MAAM,MAAM,WAAW,GAAG,CAAC,CAAC,MAAM,CAAC,OAAO,iBAAiB,CAAC,CAAC;AAO7D;;GAEG;AACH,wBAAgB,mBAAmB,IAAI,MAAM,CAE5C;AAED;;GAEG;AACH,wBAAsB,gBAAgB,IAAI,OAAO,CAAC,WAAW,EAAE,CAAC,CAmB/D;AAED;;GAEG;AACH,wBAAsB,eAAe,CAAC,KAAK,EAAE,WAAW,GAAG,OAAO,CAAC,IAAI,CAAC,CAgBvE;AAED;;GAEG;AACH,wBAAsB,iBAAiB,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,OAAO,CAAC,CAUtE;AAED;;GAEG;AACH,wBAAsB,cAAc,CAAC,IAAI,EAAE,MAAM,GAAG,OAAO,CAAC,WAAW,GAAG,IAAI,CAAC,CAG9E"}

package/dist/models/custom-models.js

@@ -0,0 +1,117 @@
+import { z } from "zod";
+import { promises as fs } from "fs";
+import * as path from "path";
+import { getDextoGlobalPath } from "../utils/path.js";
+const CUSTOM_MODEL_PROVIDERS = [
+  "openai-compatible",
+  "openrouter",
+  "litellm",
+  "glama",
+  "bedrock",
+  "ollama",
+  "local",
+  "vertex"
+];
+const CustomModelSchema = z.object({
+  name: z.string().min(1),
+  provider: z.enum(CUSTOM_MODEL_PROVIDERS).default("openai-compatible"),
+  baseURL: z.string().url().optional(),
+  displayName: z.string().optional(),
+  maxInputTokens: z.number().int().positive().optional(),
+  maxOutputTokens: z.number().int().positive().optional(),
+  // Optional per-model API key. For openai-compatible this is the primary key source.
+  // For litellm/glama/openrouter this overrides the provider-level env var key.
+  apiKey: z.string().optional(),
+  // File path for local GGUF models. Required when provider is 'local'.
+  // Stores the absolute path to the .gguf file on disk.
+  filePath: z.string().optional()
+}).superRefine((data, ctx) => {
+  if ((data.provider === "openai-compatible" || data.provider === "litellm") && !data.baseURL) {
+    ctx.addIssue({
+      code: z.ZodIssueCode.custom,
+      path: ["baseURL"],
+      message: `Base URL is required for ${data.provider} provider`
+    });
+  }
+  if (data.provider === "local" && !data.filePath) {
+    ctx.addIssue({
+      code: z.ZodIssueCode.custom,
+      path: ["filePath"],
+      message: "File path is required for local provider"
+    });
+  }
+  if (data.provider === "local" && data.filePath && !data.filePath.endsWith(".gguf")) {
+    ctx.addIssue({
+      code: z.ZodIssueCode.custom,
+      path: ["filePath"],
+      message: "File path must be a .gguf file"
+    });
+  }
+});
+const StorageSchema = z.object({
+  version: z.literal(1),
+  models: z.array(CustomModelSchema)
+});
+function getCustomModelsPath() {
+  return getDextoGlobalPath("models", "custom-models.json");
+}
+async function loadCustomModels() {
+  const filePath = getCustomModelsPath();
+  try {
+    const content = await fs.readFile(filePath, "utf-8");
+    const parsed = StorageSchema.safeParse(JSON.parse(content));
+    if (!parsed.success) {
+      console.warn(
+        `[custom-models] Failed to parse ${filePath}: ${parsed.error.issues.map((i) => i.message).join(", ")}`
+      );
+      return [];
+    }
+    return parsed.data.models;
+  } catch (error) {
+    if (error.code === "ENOENT") {
+      return [];
+    }
+    throw error;
+  }
+}
+async function saveCustomModel(model) {
+  const parsed = CustomModelSchema.safeParse(model);
+  if (!parsed.success) {
+    throw new Error(`Invalid model: ${parsed.error.issues.map((i) => i.message).join(", ")}`);
+  }
+  const models = await loadCustomModels();
+  const existingIndex = models.findIndex((m) => m.name === parsed.data.name);
+  if (existingIndex >= 0) {
+    models[existingIndex] = parsed.data;
+  } else {
+    models.push(parsed.data);
+  }
+  await writeCustomModels(models);
+}
+async function deleteCustomModel(name) {
+  const models = await loadCustomModels();
+  const filtered = models.filter((m) => m.name !== name);
+  if (filtered.length === models.length) {
+    return false;
+  }
+  await writeCustomModels(filtered);
+  return true;
+}
+async function getCustomModel(name) {
+  const models = await loadCustomModels();
+  return models.find((m) => m.name === name) ?? null;
+}
+async function writeCustomModels(models) {
+  const filePath = getCustomModelsPath();
+  await fs.mkdir(path.dirname(filePath), { recursive: true });
+  await fs.writeFile(filePath, JSON.stringify({ version: 1, models }, null, 2), "utf-8");
+}
+export {
+  CUSTOM_MODEL_PROVIDERS,
+  CustomModelSchema,
+  deleteCustomModel,
+  getCustomModel,
+  getCustomModelsPath,
+  loadCustomModels,
+  saveCustomModel
+};