@cjhyy/code-shell 0.1.0-alpha.3 → 0.1.6

package/dist/{chunk-I57ZD5IQ.js → chunk-VIAGTYYW.js}

@@ -1,6 +1,6 @@
 import {
   init_esm_shims
-} from "./chunk-DI7RDLOS.js";
+} from "./chunk-MD32R22I.js";
 
 // src/session/memory.ts
 init_esm_shims();

package/dist/chunk-WASS5YWB.js

@@ -0,0 +1,215 @@
+import {
+  isStale,
+  readCache,
+  writeCache
+} from "./chunk-3OA5G4TM.js";
+import {
+  init_esm_shims
+} from "./chunk-MD32R22I.js";
+
+// src/llm/model-fetcher.ts
+init_esm_shims();
+import { readFileSync, existsSync } from "fs";
+import { fileURLToPath } from "url";
+import { dirname, join } from "path";
+
+// src/llm/provider-kinds.ts
+init_esm_shims();
+var NON_CHAT_PATTERNS = /(?:^|[-_/])(?:embed(?:ding)?|whisper|tts|audio|image|dall-?e|moderation|rerank|guard|vision-only)(?:$|[-_/])/i;
+var isChatLike = (id) => !NON_CHAT_PATTERNS.test(id);
+var bearer = (k) => k ? { Authorization: `Bearer ${k}` } : {};
+var PROVIDER_KINDS = {
+  openai: {
+    label: "OpenAI",
+    defaultBaseUrl: "https://api.openai.com/v1",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  },
+  anthropic: {
+    label: "Anthropic",
+    defaultBaseUrl: "https://api.anthropic.com/v1",
+    modelsPath: "/models",
+    protocol: "anthropic-style",
+    authHeader: (k) => k ? { "x-api-key": k, "anthropic-version": "2023-06-01" } : { "anthropic-version": "2023-06-01" },
+    chatFilter: isChatLike
+  },
+  deepseek: {
+    label: "DeepSeek",
+    defaultBaseUrl: "https://api.deepseek.com/v1",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  },
+  xai: {
+    label: "xAI (Grok)",
+    defaultBaseUrl: "https://api.x.ai/v1",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  },
+  mistral: {
+    label: "Mistral",
+    defaultBaseUrl: "https://api.mistral.ai/v1",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  },
+  groq: {
+    label: "Groq",
+    defaultBaseUrl: "https://api.groq.com/openai/v1",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  },
+  google: {
+    label: "Google Gemini",
+    defaultBaseUrl: "https://generativelanguage.googleapis.com/v1beta",
+    modelsPath: "/models",
+    protocol: "gemini",
+    authHeader: () => ({}),
+    authQuery: (k) => k ? { key: k } : {},
+    chatFilter: (id) => isChatLike(id) && /gemini/i.test(id)
+  },
+  openrouter: {
+    label: "OpenRouter",
+    defaultBaseUrl: "https://openrouter.ai/api/v1",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  },
+  ollama: {
+    label: "Ollama (local)",
+    defaultBaseUrl: "http://localhost:11434",
+    modelsPath: "/api/tags",
+    protocol: "ollama",
+    authHeader: () => ({}),
+    chatFilter: isChatLike
+  },
+  custom: {
+    label: "Custom",
+    defaultBaseUrl: "",
+    modelsPath: "/models",
+    protocol: "openai-compat",
+    authHeader: bearer,
+    chatFilter: isChatLike
+  }
+};
+function getKindMeta(kind) {
+  return PROVIDER_KINDS[kind] ?? PROVIDER_KINDS.custom;
+}
+
+// src/llm/model-fetcher.ts
+var HERE = dirname(fileURLToPath(import.meta.url));
+var OPENROUTER_SNAPSHOT = join(HERE, "..", "data", "openrouter-models.json");
+async function fetchModelList(provider, opts) {
+  const meta = getKindMeta(provider.kind);
+  if (!opts.refresh) {
+    const cached = readCache(opts.cacheDir, provider.key);
+    if (cached && !isStale(cached)) return { ...cached, fromCache: true };
+  }
+  if (provider.kind === "openrouter") {
+    return loadOpenRouterSnapshot(provider.key, opts.cacheDir);
+  }
+  try {
+    const path = provider.modelsPath ?? meta.modelsPath;
+    const url = new URL(joinUrl(provider.baseUrl, path));
+    const headers = meta.authHeader(provider.apiKey ?? "");
+    if (meta.authQuery) {
+      const q = meta.authQuery(provider.apiKey ?? "");
+      for (const [k, v] of Object.entries(q)) url.searchParams.set(k, v);
+    }
+    const res = await fetch(url.toString(), {
+      headers: { Accept: "application/json", ...headers },
+      signal: AbortSignal.timeout(opts.timeoutMs ?? 2e4)
+    });
+    if (!res.ok) {
+      return errorResult(provider.key, opts.cacheDir, `HTTP ${res.status}`);
+    }
+    const payload = await res.json();
+    const raw = normalize(payload, meta.protocol);
+    const filtered = raw.filter((m) => meta.chatFilter(m.id));
+    const file = writeCache(opts.cacheDir, provider.key, filtered);
+    return file;
+  } catch (err) {
+    const msg = err.message || String(err);
+    return errorResult(provider.key, opts.cacheDir, msg);
+  }
+}
+function normalize(payload, protocol) {
+  switch (protocol) {
+    case "ollama": {
+      const p = payload;
+      return (p.models ?? []).map((m) => ({ id: m.name, contextLength: 0, maxOutputTokens: 0 }));
+    }
+    case "anthropic-style": {
+      const p = payload;
+      return (p.data ?? []).map((m) => ({
+        id: m.id,
+        contextLength: m.context_window ?? 2e5,
+        maxOutputTokens: m.max_output_tokens ?? 0
+      }));
+    }
+    case "gemini": {
+      const p = payload;
+      return (p.models ?? []).map((m) => ({
+        id: m.name.replace(/^models\//, ""),
+        contextLength: m.inputTokenLimit ?? 0,
+        maxOutputTokens: m.outputTokenLimit ?? 0
+      }));
+    }
+    case "openai-compat": {
+      const p = payload;
+      return (p.data ?? []).map((m) => ({
+        id: m.id,
+        contextLength: m.context_window ?? m.context_length ?? 0,
+        maxOutputTokens: m.max_completion_tokens ?? 0
+      }));
+    }
+  }
+  const _exhaustive = protocol;
+  return _exhaustive;
+}
+function loadOpenRouterSnapshot(providerKey, cacheDir) {
+  if (!existsSync(OPENROUTER_SNAPSHOT)) {
+    return errorResult(providerKey, cacheDir, "openrouter snapshot missing");
+  }
+  try {
+    const snapshot = JSON.parse(
+      readFileSync(OPENROUTER_SNAPSHOT, "utf-8")
+    );
+    const models = (snapshot.models ?? []).map((m) => ({
+      id: m.id,
+      contextLength: m.contextLength ?? 0,
+      maxOutputTokens: m.maxOutputTokens ?? 0
+    }));
+    return writeCache(cacheDir, providerKey, models);
+  } catch (err) {
+    return errorResult(providerKey, cacheDir, err.message);
+  }
+}
+function errorResult(providerKey, cacheDir, error) {
+  const stale = readCache(cacheDir, providerKey);
+  if (stale) return { ...stale, error, fromCache: true };
+  return {
+    fetchedAt: (/* @__PURE__ */ new Date()).toISOString(),
+    providerKey,
+    models: [],
+    error
+  };
+}
+function joinUrl(base, path) {
+  if (!base) return path;
+  return `${base.replace(/\/$/, "")}${path.startsWith("/") ? "" : "/"}${path}`;
+}
+
+export {
+  PROVIDER_KINDS,
+  fetchModelList
+};