pybao-cli 1.3.3

package/dist/chunk-Z3HMXDXP.js

@@ -0,0 +1,654 @@
+import { createRequire as __pybCreateRequire } from "node:module";
+const require = __pybCreateRequire(import.meta.url);
+import {
+  getGlobalConfig,
+  saveGlobalConfig
+} from "./chunk-JZDE77EH.js";
+import {
+  debug
+} from "./chunk-3KNGJX7Q.js";
+import {
+  logError
+} from "./chunk-TXFCNQDE.js";
+
+// src/utils/model/index.ts
+import { memoize } from "lodash-es";
+var USE_BEDROCK = !!(process.env.KODE_USE_BEDROCK ?? process.env.CLAUDE_CODE_USE_BEDROCK);
+var USE_VERTEX = !!(process.env.KODE_USE_VERTEX ?? process.env.CLAUDE_CODE_USE_VERTEX);
+var DEFAULT_MODEL_CONFIG = {
+  bedrock: "us.anthropic.claude-3-7-sonnet-20250219-v1:0",
+  vertex: "claude-3-7-sonnet@20250219",
+  firstParty: "claude-sonnet-4-20250514"
+};
+async function getModelConfig() {
+  return DEFAULT_MODEL_CONFIG;
+}
+var getSlowAndCapableModel = memoize(async () => {
+  const config = await getGlobalConfig();
+  const modelManager = new ModelManager(config);
+  const model = modelManager.getMainAgentModel();
+  if (model) {
+    return model;
+  }
+  const modelConfig = await getModelConfig();
+  if (USE_BEDROCK) return modelConfig.bedrock;
+  if (USE_VERTEX) return modelConfig.vertex;
+  return modelConfig.firstParty;
+});
+async function isDefaultSlowAndCapableModel() {
+  return !process.env.ANTHROPIC_MODEL || process.env.ANTHROPIC_MODEL === await getSlowAndCapableModel();
+}
+function getVertexRegionForModel(model) {
+  if (model?.startsWith("claude-3-5-haiku")) {
+    return process.env.VERTEX_REGION_CLAUDE_3_5_HAIKU;
+  } else if (model?.startsWith("claude-3-5-sonnet")) {
+    return process.env.VERTEX_REGION_CLAUDE_3_5_SONNET;
+  } else if (model?.startsWith("claude-3-7-sonnet")) {
+    return process.env.VERTEX_REGION_CLAUDE_3_7_SONNET;
+  }
+}
+var ModelManager = class {
+  config;
+  modelProfiles;
+  constructor(config) {
+    this.config = config;
+    this.modelProfiles = config.modelProfiles || [];
+  }
+  getCurrentModel() {
+    const mainModelName = this.config.modelPointers?.main;
+    if (mainModelName) {
+      const profile = this.findModelProfile(mainModelName);
+      if (profile && profile.isActive) {
+        return profile.modelName;
+      }
+    }
+    return this.getMainAgentModel();
+  }
+  getMainAgentModel() {
+    const mainModelName = this.config.modelPointers?.main;
+    if (mainModelName) {
+      const profile = this.findModelProfile(mainModelName);
+      if (profile && profile.isActive) {
+        return profile.modelName;
+      }
+    }
+    const activeProfile = this.modelProfiles.find((p) => p.isActive);
+    if (activeProfile) {
+      return activeProfile.modelName;
+    }
+    return null;
+  }
+  getTaskToolModel() {
+    const taskModelName = this.config.modelPointers?.task;
+    if (taskModelName) {
+      const profile = this.findModelProfile(taskModelName);
+      if (profile && profile.isActive) {
+        return profile.modelName;
+      }
+    }
+    return this.getMainAgentModel();
+  }
+  switchToNextModelWithContextCheck(currentContextTokens = 0) {
+    const allProfiles = this.getAllConfiguredModels();
+    if (allProfiles.length === 0) {
+      return {
+        success: false,
+        modelName: null,
+        previousModelName: null,
+        contextOverflow: false,
+        usagePercentage: 0,
+        currentContextTokens
+      };
+    }
+    allProfiles.sort((a, b) => a.createdAt - b.createdAt);
+    const currentMainModelName = this.config.modelPointers?.main;
+    const currentModel = currentMainModelName ? this.findModelProfile(currentMainModelName) : null;
+    const previousModelName = currentModel?.name || null;
+    const budgetForModel = (model) => {
+      const contextLength = Number(model.contextLength);
+      if (!Number.isFinite(contextLength) || contextLength <= 0) {
+        return { budgetTokens: null, usagePercentage: 0, compatible: true };
+      }
+      const budgetTokens = Math.floor(contextLength * 0.9);
+      const usagePercentage = budgetTokens > 0 ? currentContextTokens / budgetTokens * 100 : 0;
+      return {
+        budgetTokens,
+        usagePercentage,
+        compatible: budgetTokens > 0 ? currentContextTokens <= budgetTokens : true
+      };
+    };
+    const currentIndex = currentMainModelName ? allProfiles.findIndex((p) => p.modelName === currentMainModelName) : -1;
+    const startIndex = currentIndex >= 0 ? currentIndex : -1;
+    if (allProfiles.length === 1) {
+      return {
+        success: false,
+        modelName: null,
+        previousModelName,
+        contextOverflow: false,
+        usagePercentage: 0,
+        currentContextTokens
+      };
+    }
+    const maxOffsets = startIndex === -1 ? allProfiles.length : allProfiles.length - 1;
+    const skippedModels = [];
+    let selected = null;
+    let selectedUsagePercentage = 0;
+    for (let offset = 1; offset <= maxOffsets; offset++) {
+      const candidateIndex = (startIndex + offset + allProfiles.length) % allProfiles.length;
+      const candidate = allProfiles[candidateIndex];
+      if (!candidate) continue;
+      const { budgetTokens, usagePercentage, compatible } = budgetForModel(candidate);
+      if (compatible) {
+        selected = candidate;
+        selectedUsagePercentage = usagePercentage;
+        break;
+      }
+      skippedModels.push({
+        name: candidate.name,
+        provider: candidate.provider,
+        contextLength: candidate.contextLength,
+        budgetTokens,
+        usagePercentage
+      });
+    }
+    if (!selected) {
+      const firstSkipped = skippedModels[0];
+      return {
+        success: false,
+        modelName: null,
+        previousModelName,
+        contextOverflow: true,
+        usagePercentage: firstSkipped?.usagePercentage ?? 0,
+        currentContextTokens,
+        skippedModels
+      };
+    }
+    if (!selected.isActive) {
+      selected.isActive = true;
+    }
+    this.setPointer("main", selected.modelName);
+    this.updateLastUsed(selected.modelName);
+    return {
+      success: true,
+      modelName: selected.name,
+      previousModelName,
+      contextOverflow: false,
+      usagePercentage: selectedUsagePercentage,
+      currentContextTokens,
+      skippedModels
+    };
+  }
+  switchToNextModel(currentContextTokens = 0) {
+    const result = this.switchToNextModelWithContextCheck(currentContextTokens);
+    const formatTokens = (tokens) => {
+      if (!Number.isFinite(tokens)) return "unknown";
+      if (tokens >= 1e3) return `${Math.round(tokens / 1e3)}k`;
+      return String(Math.round(tokens));
+    };
+    const allModels = this.getAllConfiguredModels();
+    if (allModels.length === 0) {
+      return {
+        success: false,
+        modelName: null,
+        blocked: false,
+        message: "\u274C No models configured. Use /model to add models."
+      };
+    }
+    if (allModels.length === 1) {
+      return {
+        success: false,
+        modelName: null,
+        blocked: false,
+        message: `\u26A0\uFE0F Only one model configured (${allModels[0].modelName}). Use /model to add more models for switching.`
+      };
+    }
+    const currentModel = this.findModelProfile(this.config.modelPointers?.main);
+    const modelsSorted = [...allModels].sort(
+      (a, b) => a.createdAt - b.createdAt
+    );
+    const currentIndex = modelsSorted.findIndex(
+      (m) => m.modelName === currentModel?.modelName
+    );
+    const totalModels = modelsSorted.length;
+    if (result.success && result.modelName) {
+      const skippedCount = result.skippedModels?.length ?? 0;
+      const skippedSuffix = skippedCount > 0 ? ` \xB7 skipped ${skippedCount} incompatible` : "";
+      const contextSuffix = currentModel?.contextLength && result.currentContextTokens ? ` \xB7 context ~${formatTokens(result.currentContextTokens)}/${formatTokens(currentModel.contextLength)}` : "";
+      return {
+        success: true,
+        modelName: result.modelName,
+        blocked: false,
+        message: `\u2705 Switched to ${result.modelName} (${currentIndex + 1}/${totalModels})${currentModel?.provider ? ` [${currentModel.provider}]` : ""}${skippedSuffix}${contextSuffix}`
+      };
+    }
+    if (result.contextOverflow) {
+      const attempted = result.skippedModels?.[0];
+      const attemptedContext = attempted?.contextLength;
+      const attemptedBudget = attempted?.budgetTokens;
+      const currentLabel = currentModel?.name || currentModel?.modelName || "current model";
+      const attemptedText = attempted ? `Can't switch to ${attempted.name}: current ~${formatTokens(result.currentContextTokens)} tokens exceeds safe budget (~${formatTokens(attemptedBudget ?? 0)} tokens, 90% of ${formatTokens(attemptedContext ?? 0)}).` : `Can't switch models due to context size (~${formatTokens(result.currentContextTokens)} tokens).`;
+      return {
+        success: false,
+        modelName: null,
+        blocked: true,
+        message: `\u26A0\uFE0F ${attemptedText} Keeping ${currentLabel}.`
+      };
+    }
+    return {
+      success: false,
+      modelName: null,
+      blocked: false,
+      message: "\u274C Failed to switch models"
+    };
+  }
+  revertToPreviousModel(previousModelName) {
+    const previousModel = this.modelProfiles.find(
+      (p) => p.name === previousModelName && p.isActive
+    );
+    if (!previousModel) {
+      return false;
+    }
+    this.setPointer("main", previousModel.modelName);
+    this.updateLastUsed(previousModel.modelName);
+    return true;
+  }
+  analyzeContextCompatibility(model, contextTokens) {
+    const usableContext = Math.floor(model.contextLength * 0.8);
+    const usagePercentage = contextTokens / usableContext * 100;
+    if (usagePercentage <= 70) {
+      return {
+        compatible: true,
+        severity: "safe",
+        usagePercentage,
+        recommendation: "Full context preserved"
+      };
+    } else if (usagePercentage <= 90) {
+      return {
+        compatible: true,
+        severity: "warning",
+        usagePercentage,
+        recommendation: "Context usage high, consider compression"
+      };
+    } else {
+      return {
+        compatible: false,
+        severity: "critical",
+        usagePercentage,
+        recommendation: "Auto-compression or message truncation required"
+      };
+    }
+  }
+  switchToNextModelWithAnalysis(currentContextTokens = 0) {
+    const result = this.switchToNextModel(currentContextTokens);
+    if (!result.success || !result.modelName) {
+      return {
+        modelName: null,
+        contextAnalysis: null,
+        requiresCompression: false,
+        estimatedTokensAfterSwitch: 0
+      };
+    }
+    const newModel = this.getModel("main");
+    if (!newModel) {
+      return {
+        modelName: result.modelName,
+        contextAnalysis: null,
+        requiresCompression: false,
+        estimatedTokensAfterSwitch: currentContextTokens
+      };
+    }
+    const analysis = this.analyzeContextCompatibility(
+      newModel,
+      currentContextTokens
+    );
+    return {
+      modelName: result.modelName,
+      contextAnalysis: analysis,
+      requiresCompression: analysis.severity === "critical",
+      estimatedTokensAfterSwitch: currentContextTokens
+    };
+  }
+  canModelHandleContext(model, contextTokens) {
+    const analysis = this.analyzeContextCompatibility(model, contextTokens);
+    return analysis.compatible;
+  }
+  findModelWithSufficientContext(models, contextTokens) {
+    return models.find((model) => this.canModelHandleContext(model, contextTokens)) || null;
+  }
+  getModelForContext(contextType) {
+    switch (contextType) {
+      case "terminal":
+        return this.getCurrentModel();
+      case "main-agent":
+        return this.getMainAgentModel();
+      case "task-tool":
+        return this.getTaskToolModel();
+      default:
+        return this.getMainAgentModel();
+    }
+  }
+  getActiveModelProfiles() {
+    return this.modelProfiles.filter((p) => p.isActive);
+  }
+  hasConfiguredModels() {
+    return this.getActiveModelProfiles().length > 0;
+  }
+  getModel(pointer) {
+    const pointerId = this.config.modelPointers?.[pointer];
+    if (!pointerId) {
+      return this.getDefaultModel();
+    }
+    const profile = this.findModelProfile(pointerId);
+    return profile && profile.isActive ? profile : this.getDefaultModel();
+  }
+  getModelName(pointer) {
+    const profile = this.getModel(pointer);
+    return profile ? profile.modelName : null;
+  }
+  getCompactModel() {
+    return this.getModelName("compact") || this.getModelName("main");
+  }
+  getQuickModel() {
+    return this.getModelName("quick") || this.getModelName("task") || this.getModelName("main");
+  }
+  async addModel(config) {
+    const existingByModelName = this.modelProfiles.find(
+      (p) => p.modelName === config.modelName
+    );
+    if (existingByModelName) {
+      throw new Error(
+        `Model with modelName '${config.modelName}' already exists: ${existingByModelName.name}`
+      );
+    }
+    const existingByName = this.modelProfiles.find((p) => p.name === config.name);
+    if (existingByName) {
+      throw new Error(`Model with name '${config.name}' already exists`);
+    }
+    const newModel = {
+      ...config,
+      createdAt: Date.now(),
+      isActive: true
+    };
+    this.modelProfiles.push(newModel);
+    if (this.modelProfiles.length === 1) {
+      this.config.modelPointers = {
+        main: config.modelName,
+        task: config.modelName,
+        compact: config.modelName,
+        quick: config.modelName
+      };
+      this.config.defaultModelName = config.modelName;
+    } else {
+      if (!this.config.modelPointers) {
+        this.config.modelPointers = {
+          main: config.modelName,
+          task: "",
+          compact: "",
+          quick: ""
+        };
+      } else {
+        this.config.modelPointers.main = config.modelName;
+      }
+    }
+    this.saveConfig();
+    return config.modelName;
+  }
+  setPointer(pointer, modelName) {
+    if (!this.findModelProfile(modelName)) {
+      throw new Error(`Model '${modelName}' not found`);
+    }
+    if (!this.config.modelPointers) {
+      this.config.modelPointers = {
+        main: "",
+        task: "",
+        compact: "",
+        quick: ""
+      };
+    }
+    this.config.modelPointers[pointer] = modelName;
+    this.saveConfig();
+  }
+  getAvailableModels() {
+    return this.modelProfiles.filter((p) => p.isActive);
+  }
+  getAllConfiguredModels() {
+    return this.modelProfiles;
+  }
+  getAllAvailableModelNames() {
+    return this.getAvailableModels().map((p) => p.modelName);
+  }
+  getAllConfiguredModelNames() {
+    return this.getAllConfiguredModels().map((p) => p.modelName);
+  }
+  getModelSwitchingDebugInfo() {
+    const availableModels = this.getAvailableModels();
+    const currentMainModelName = this.config.modelPointers?.main;
+    return {
+      totalModels: this.modelProfiles.length,
+      activeModels: availableModels.length,
+      inactiveModels: this.modelProfiles.length - availableModels.length,
+      currentMainModel: currentMainModelName || null,
+      availableModels: this.modelProfiles.map((p) => ({
+        name: p.name,
+        modelName: p.modelName,
+        provider: p.provider,
+        isActive: p.isActive,
+        lastUsed: p.lastUsed
+      })),
+      modelPointers: this.config.modelPointers || {}
+    };
+  }
+  removeModel(modelName) {
+    this.modelProfiles = this.modelProfiles.filter(
+      (p) => p.modelName !== modelName
+    );
+    if (this.config.modelPointers) {
+      Object.keys(this.config.modelPointers).forEach((pointer) => {
+        if (this.config.modelPointers[pointer] === modelName) {
+          this.config.modelPointers[pointer] = this.config.defaultModelName || "";
+        }
+      });
+    }
+    this.saveConfig();
+  }
+  getDefaultModel() {
+    if (this.config.defaultModelId) {
+      const profile = this.findModelProfile(this.config.defaultModelId);
+      if (profile && profile.isActive) {
+        return profile;
+      }
+    }
+    return this.modelProfiles.find((p) => p.isActive) || null;
+  }
+  saveConfig() {
+    const updatedConfig = {
+      ...this.config,
+      modelProfiles: this.modelProfiles
+    };
+    saveGlobalConfig(updatedConfig);
+  }
+  async getFallbackModel() {
+    const modelConfig = await getModelConfig();
+    if (USE_BEDROCK) return modelConfig.bedrock;
+    if (USE_VERTEX) return modelConfig.vertex;
+    return modelConfig.firstParty;
+  }
+  resolveModel(modelParam) {
+    if (["main", "task", "compact", "quick"].includes(modelParam)) {
+      const pointerId = this.config.modelPointers?.[modelParam];
+      if (pointerId) {
+        let profile2 = this.findModelProfile(pointerId);
+        if (!profile2) {
+          profile2 = this.findModelProfileByModelName(pointerId);
+        }
+        if (profile2 && profile2.isActive) {
+          return profile2;
+        }
+      }
+      return this.getDefaultModel();
+    }
+    let profile = this.findModelProfile(modelParam);
+    if (profile && profile.isActive) {
+      return profile;
+    }
+    profile = this.findModelProfileByModelName(modelParam);
+    if (profile && profile.isActive) {
+      return profile;
+    }
+    profile = this.findModelProfileByName(modelParam);
+    if (profile && profile.isActive) {
+      return profile;
+    }
+    if (typeof modelParam === "string") {
+      const qualified = this.resolveProviderQualifiedModel(modelParam);
+      if (qualified && qualified.isActive) {
+        return qualified;
+      }
+    }
+    return this.getDefaultModel();
+  }
+  resolveModelWithInfo(modelParam) {
+    const isPointer = ["main", "task", "compact", "quick"].includes(modelParam);
+    if (isPointer) {
+      const pointerId = this.config.modelPointers?.[modelParam];
+      if (!pointerId) {
+        return {
+          success: false,
+          profile: null,
+          error: `Model pointer '${modelParam}' is not configured. Use /model to set up models.`
+        };
+      }
+      let profile = this.findModelProfile(pointerId);
+      if (!profile) {
+        profile = this.findModelProfileByModelName(pointerId);
+      }
+      if (!profile) {
+        return {
+          success: false,
+          profile: null,
+          error: `Model pointer '${modelParam}' points to invalid model '${pointerId}'. Use /model to reconfigure.`
+        };
+      }
+      if (!profile.isActive) {
+        return {
+          success: false,
+          profile: null,
+          error: `Model '${profile.name}' (pointed by '${modelParam}') is inactive. Use /model to activate it.`
+        };
+      }
+      return {
+        success: true,
+        profile
+      };
+    } else {
+      let profile = this.findModelProfile(modelParam);
+      if (!profile) {
+        profile = this.findModelProfileByModelName(modelParam);
+      }
+      if (!profile) {
+        profile = this.findModelProfileByName(modelParam);
+      }
+      if (!profile && typeof modelParam === "string") {
+        profile = this.resolveProviderQualifiedModel(modelParam);
+      }
+      if (!profile) {
+        return {
+          success: false,
+          profile: null,
+          error: `Model '${modelParam}' not found. Use /model to add models, or run 'kode models list' to see configured profiles.`
+        };
+      }
+      if (!profile.isActive) {
+        return {
+          success: false,
+          profile: null,
+          error: `Model '${profile.name}' is inactive. Use /model to activate it.`
+        };
+      }
+      return {
+        success: true,
+        profile
+      };
+    }
+  }
+  resolveProviderQualifiedModel(input) {
+    const trimmed = input.trim();
+    const colonIndex = trimmed.indexOf(":");
+    if (colonIndex <= 0 || colonIndex >= trimmed.length - 1) return null;
+    const provider = trimmed.slice(0, colonIndex).trim().toLowerCase();
+    const modelOrName = trimmed.slice(colonIndex + 1).trim();
+    if (!provider || !modelOrName) return null;
+    const providerProfiles = this.modelProfiles.filter(
+      (p) => String(p.provider).trim().toLowerCase() === provider
+    );
+    if (providerProfiles.length === 0) return null;
+    const byModelName = providerProfiles.find((p) => p.modelName === modelOrName);
+    if (byModelName) return byModelName;
+    const byName = providerProfiles.find((p) => p.name === modelOrName);
+    if (byName) return byName;
+    return null;
+  }
+  findModelProfile(modelName) {
+    return this.modelProfiles.find((p) => p.modelName === modelName) || null;
+  }
+  findModelProfileByModelName(modelName) {
+    return this.modelProfiles.find((p) => p.modelName === modelName) || null;
+  }
+  findModelProfileByName(name) {
+    return this.modelProfiles.find((p) => p.name === name) || null;
+  }
+  updateLastUsed(modelName) {
+    const profile = this.findModelProfile(modelName);
+    if (profile) {
+      profile.lastUsed = Date.now();
+    }
+  }
+};
+var globalModelManager = null;
+var getModelManager = () => {
+  try {
+    if (!globalModelManager) {
+      const config = getGlobalConfig();
+      if (!config) {
+        debug.warn("MODEL_MANAGER_GLOBAL_CONFIG_MISSING", {});
+        globalModelManager = new ModelManager({
+          modelProfiles: [],
+          modelPointers: { main: "", task: "", compact: "", quick: "" }
+        });
+      } else {
+        globalModelManager = new ModelManager(config);
+      }
+    }
+    return globalModelManager;
+  } catch (error) {
+    logError(error);
+    debug.error("MODEL_MANAGER_CREATE_FAILED", {
+      error: error instanceof Error ? error.message : String(error)
+    });
+    return new ModelManager({
+      modelProfiles: [],
+      modelPointers: { main: "", task: "", compact: "", quick: "" }
+    });
+  }
+};
+var reloadModelManager = () => {
+  globalModelManager = null;
+  getModelManager();
+};
+var getQuickModel = () => {
+  const manager = getModelManager();
+  const quickModel = manager.getModel("quick");
+  return quickModel?.modelName || "quick";
+};
+
+export {
+  USE_BEDROCK,
+  USE_VERTEX,
+  getSlowAndCapableModel,
+  isDefaultSlowAndCapableModel,
+  getVertexRegionForModel,
+  ModelManager,
+  getModelManager,
+  reloadModelManager,
+  getQuickModel
+};