noosphere 0.2.1 → 0.2.2

package/dist/index.cjs

@@ -279,7 +279,6 @@ var UsageTracker = class {
 // src/providers/pi-ai.ts
 var import_pi_ai = require("@mariozechner/pi-ai");
 var KNOWN_PROVIDERS = ["anthropic", "google", "openai", "xai", "groq", "cerebras", "openrouter", "zai"];
-var LOCAL_PROVIDERS = /* @__PURE__ */ new Set(["ollama"]);
 var FETCH_TIMEOUT_MS = 8e3;
 var OPENAI_CHAT_PREFIXES = ["gpt-", "o1", "o3", "o4", "chatgpt-", "codex-"];
 var OPENAI_REASONING_PREFIXES = ["o1", "o3", "o4"];
@@ -420,35 +419,7 @@ var PiAiProvider = class {
     if (modality && modality !== "llm") return [];
     await this.ensureDynamicModels();
     const models = [];
-    const seenIds = /* @__PURE__ */ new Set();
-    for (const provider of KNOWN_PROVIDERS) {
-      try {
-        const providerModels = (0, import_pi_ai.getModels)(provider);
-        for (const m of providerModels) {
-          seenIds.add(m.id);
-          models.push({
-            id: m.id,
-            provider: "pi-ai",
-            name: m.name || m.id,
-            modality: "llm",
-            local: LOCAL_PROVIDERS.has(String(m.provider)),
-            cost: {
-              price: m.cost.input ?? 0,
-              unit: m.cost.input > 0 ? "per_1m_tokens" : "free"
-            },
-            capabilities: {
-              contextWindow: m.contextWindow,
-              maxTokens: m.maxTokens,
-              supportsVision: m.input.includes("image"),
-              supportsStreaming: true
-            }
-          });
-        }
-      } catch {
-      }
-    }
-    for (const [id, m] of this.dynamicModels) {
-      if (seenIds.has(id)) continue;
+    for (const [, m] of this.dynamicModels) {
       models.push({
         id: m.id,
         provider: "pi-ai",
@@ -583,24 +554,15 @@ var PiAiProvider = class {
   async ensureDynamicModels() {
     if (this.dynamicModelsFetched) return;
     this.dynamicModelsFetched = true;
-    const staticIds = /* @__PURE__ */ new Set();
-    for (const provider of KNOWN_PROVIDERS) {
-      try {
-        for (const m of (0, import_pi_ai.getModels)(provider)) {
-          staticIds.add(m.id);
-        }
-      } catch {
-      }
-    }
     const fetchPromises = [];
     for (const [providerKey, configFactory] of Object.entries(PROVIDER_APIS)) {
       const apiKey = this.keys[providerKey];
       if (!apiKey) continue;
-      fetchPromises.push(this.fetchProviderModels(configFactory(apiKey), apiKey, staticIds));
+      fetchPromises.push(this.fetchProviderModels(configFactory(apiKey), apiKey));
     }
     await Promise.allSettled(fetchPromises);
   }
-  async fetchProviderModels(config, apiKey, staticIds) {
+  async fetchProviderModels(config, apiKey) {
     try {
       const controller = new AbortController();
       const timer = setTimeout(() => controller.abort(), FETCH_TIMEOUT_MS);
@@ -613,11 +575,11 @@ var PiAiProvider = class {
         if (!res.ok) return;
         const data = await res.json();
         const entries = config.extractEntries(data);
-        const templateModel = this.findStaticTemplate(config.providerName);
+        const staticTemplate = this.findStaticTemplate(config.providerName);
         for (const entry of entries) {
           const id = entry.id;
           if (!config.filterChat(id)) continue;
-          if (staticIds.has(id)) continue;
+          const staticMatch = this.findStaticModel(config.providerName, id);
           this.dynamicModels.set(id, {
             id,
             name: entry.name ?? id,
@@ -625,10 +587,10 @@ var PiAiProvider = class {
             provider: config.providerName,
             baseUrl: config.piBaseUrl,
             reasoning: config.isReasoning(id),
-            input: ["text", "image"],
-            cost: templateModel?.cost ?? { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
-            contextWindow: entry.contextWindow ?? templateModel?.contextWindow ?? 128e3,
-            maxTokens: entry.maxTokens ?? templateModel?.maxTokens ?? 16384
+            input: staticMatch?.input ?? ["text", "image"],
+            cost: staticMatch?.cost ?? staticTemplate?.cost ?? { input: 0, output: 0, cacheRead: 0, cacheWrite: 0 },
+            contextWindow: entry.contextWindow ?? staticMatch?.contextWindow ?? staticTemplate?.contextWindow ?? 128e3,
+            maxTokens: entry.maxTokens ?? staticMatch?.maxTokens ?? staticTemplate?.maxTokens ?? 16384
           });
         }
       } finally {
@@ -645,6 +607,14 @@ var PiAiProvider = class {
       return null;
     }
   }
+  findStaticModel(providerName, modelId) {
+    try {
+      const models = (0, import_pi_ai.getModels)(providerName);
+      return models.find((m) => m.id === modelId) ?? null;
+    } catch {
+      return null;
+    }
+  }
   /** Force re-fetch of dynamic models from provider APIs */
   async refreshDynamicModels() {
     this.dynamicModelsFetched = false;
@@ -652,18 +622,24 @@ var PiAiProvider = class {
     await this.ensureDynamicModels();
   }
   findModel(modelId) {
+    if (modelId) {
+      const dynamic = this.dynamicModels.get(modelId);
+      if (dynamic) return { model: dynamic, provider: String(dynamic.provider) };
+    }
+    if (!modelId) {
+      const first = this.dynamicModels.values().next();
+      if (!first.done && first.value) {
+        return { model: first.value, provider: String(first.value.provider) };
+      }
+    }
     for (const provider of KNOWN_PROVIDERS) {
       try {
         const models = (0, import_pi_ai.getModels)(provider);
-        const found = modelId ? models.find((m) => m.id === modelId) : models[0];
+        const found = modelId ? models.find((m) => m.id === modelId) : void 0;
         if (found) return { model: found, provider };
       } catch {
       }
     }
-    if (modelId) {
-      const dynamic = this.dynamicModels.get(modelId);
-      if (dynamic) return { model: dynamic, provider: String(dynamic.provider) };
-    }
     return { model: null, provider: null };
   }
 };
@@ -1044,11 +1020,6 @@ var PIPELINE_TAG_MAP = {
   "text-to-image": { modality: "image", limit: 50 },
   "text-to-speech": { modality: "tts", limit: 30 }
 };
-var DEFAULT_MODELS = [
-  { id: "stabilityai/stable-diffusion-xl-base-1.0", provider: "huggingface", name: "SDXL Base", modality: "image", local: false, cost: { price: 0, unit: "free" } },
-  { id: "facebook/mms-tts-eng", provider: "huggingface", name: "MMS TTS English", modality: "tts", local: false, cost: { price: 0, unit: "free" } },
-  { id: "meta-llama/Llama-3.1-8B-Instruct", provider: "huggingface", name: "Llama 3.1 8B", modality: "llm", local: false, cost: { price: 0, unit: "free" } }
-];
 var HuggingFaceProvider = class {
   id = "huggingface";
   name = "HuggingFace Inference";
@@ -1068,17 +1039,13 @@ var HuggingFaceProvider = class {
     if (!this.dynamicModels) {
       await this.fetchHubModels();
     }
-    const all = this.dynamicModels ?? DEFAULT_MODELS;
+    const all = this.dynamicModels ?? [];
     if (modality) return all.filter((m) => m.modality === modality);
     return all;
   }
   async fetchHubModels() {
     const seenIds = /* @__PURE__ */ new Set();
     const models = [];
-    for (const d of DEFAULT_MODELS) {
-      seenIds.add(d.id);
-      models.push(d);
-    }
     const fetches = Object.entries(PIPELINE_TAG_MAP).map(
       ([tag, { modality, limit }]) => this.fetchByPipelineTag(tag, modality, limit)
     );