@krmxd/onegpt 2.0.0-beta.1

package/src/catalog.js

@@ -0,0 +1,108 @@
+"use strict";
+
+// OGPT model catalog - curated models with the oGPT-(number)(alphabet) naming scheme.
+// class digit = capability tier, lowercase letter = variant within the class.
+
+const MODEL_CATALOG = {
+  // --- Local models ---
+  "qwen2.5-coder:0.5b":    { ogpt: "oGPT-0a", tier: "fast",     ram_gb: 4,  size: "400 MB", desc: "Ultra-light coder" },
+  "qwen2.5-coder:1.5b":    { ogpt: "oGPT-1a", tier: "fast",     ram_gb: 6,  size: "1 GB",   desc: "Fast coding model" },
+  "qwen2.5-coder:3b":      { ogpt: "oGPT-1b", tier: "fast",     ram_gb: 8,  size: "1.9 GB", desc: "Light coder, better quality" },
+  "llama3.2:3b":           { ogpt: "oGPT-1c", tier: "fast",     ram_gb: 8,  size: "2 GB",   desc: "Fast general chat" },
+  "qwen2.5-coder:7b":      { ogpt: "oGPT-2a", tier: "balanced", ram_gb: 12, size: "4.7 GB", desc: "Best coding value" },
+  "llama3.1:8b":           { ogpt: "oGPT-2c", tier: "balanced", ram_gb: 12, size: "4.9 GB", desc: "Solid general purpose" },
+  "qwen2.5-coder:14b":     { ogpt: "oGPT-2b", tier: "balanced", ram_gb: 16, size: "9 GB",   desc: "Heavy-duty coder" },
+  "deepseek-coder-v2:16b": { ogpt: "oGPT-2d", tier: "balanced", ram_gb: 20, size: "8.9 GB", desc: "MoE code + chat" },
+  "qwen2.5-coder:32b":     { ogpt: "oGPT-3a", tier: "pro",      ram_gb: 32, size: "20 GB",  desc: "Flagship local coder" },
+  // --- Cloud providers ---
+  "gpt-4o-mini":               { ogpt: "oGPT-4b", tier: "cloud", provider: "openai",    desc: "Cheap & quick" },
+  "gpt-4o":                    { ogpt: "oGPT-4a", tier: "cloud", provider: "openai",    desc: "OpenAI flagship" },
+  "claude-3-5-haiku-20241022": { ogpt: "oGPT-5b", tier: "cloud", provider: "anthropic", desc: "Fast Claude" },
+  "claude-sonnet-4-20250514":  { ogpt: "oGPT-5a", tier: "cloud", provider: "anthropic", desc: "Anthropic flagship" },
+  "gemini-2.0-flash":          { ogpt: "oGPT-6a", tier: "cloud", provider: "google",    desc: "Fast Gemini" },
+  "gemini-1.5-pro":            { ogpt: "oGPT-6b", tier: "cloud", provider: "google",    desc: "Long-context Gemini" },
+};
+
+const OGPT_RE = /^ogpt-(\d+)([a-z])$/i;
+
+const BY_OGPT = {};
+for (const [id, e] of Object.entries(MODEL_CATALOG)) {
+  BY_OGPT[e.ogpt.toLowerCase()] = id;
+}
+
+function displayName(modelId) {
+  const e = MODEL_CATALOG[modelId];
+  return e ? e.ogpt : modelId;
+}
+
+function resolveModel(name) {
+  if (!name) return "";
+  name = String(name).trim();
+  if (MODEL_CATALOG[name]) return name;
+  const m = OGPT_RE.exec(name);
+  if (m) return BY_OGPT[`ogpt-${m[1]}${m[2].toLowerCase()}`] || "";
+  const stripped = name.toLowerCase().replace(/^(o|og|g)?pt-?/, "");
+  if (OGPT_RE.test("ogpt-" + stripped)) return BY_OGPT["ogpt-" + stripped] || "";
+  return "";
+}
+
+function isOgptName(name) {
+  return OGPT_RE.test(name || "");
+}
+
+function recommendForRam(ramGB) {
+  let bestId = "";
+  let bestKey = -1;
+  for (const [id, e] of Object.entries(MODEL_CATALOG)) {
+    if (e.provider || e.tier === "cloud") continue;
+    if (ramGB + 0.01 >= e.ram_gb) {
+      const bonus = id.includes("coder") ? 0.5 : 0;
+      const key = e.ram_gb + bonus;
+      if (key >= bestKey) {
+        bestKey = key;
+        bestId = id;
+      }
+    }
+  }
+  return bestId || "qwen2.5-coder:1.5b";
+}
+
+function tierRouting() {
+  const caps = { fast: 6, balanced: 14, pro: 64 };
+  const out = {};
+  const sorted = Object.entries(MODEL_CATALOG).sort((a, b) => (a[1].ram_gb || 0) - (b[1].ram_gb || 0));
+  for (const [id, e] of sorted) {
+    if (e.provider || e.tier === "cloud" || !(e.tier in caps)) continue;
+    if (e.ram_gb <= caps[e.tier] && (id.includes("coder") || !out[e.tier])) {
+      out[e.tier] = id;
+    }
+  }
+  return out;
+}
+
+function entries(provider) {
+  return Object.entries(MODEL_CATALOG)
+    .filter(([id, e]) => {
+      if (!provider) return true;
+      if (provider === "ollama") return !e.provider;
+      return e.provider === provider;
+    })
+    .sort((a, b) => a[1].ogpt.localeCompare(b[1].ogpt));
+}
+
+function ogptNamesMap() {
+  const out = {};
+  for (const [id, e] of Object.entries(MODEL_CATALOG)) out[id] = e.ogpt;
+  return out;
+}
+
+module.exports = {
+  MODEL_CATALOG,
+  displayName,
+  resolveModel,
+  isOgptName,
+  recommendForRam,
+  tierRouting,
+  entries,
+  ogptNamesMap,
+};