@sriinnu/kosha-discovery 0.5.0

package/dist/discovery/mistral.js

@@ -0,0 +1,82 @@
+/**
+ * kosha-discovery — Mistral AI provider discoverer.
+ *
+ * Queries the Mistral AI `/v1/models` endpoint, filters relevant models,
+ * and maps them into {@link ModelCard} objects.
+ *
+ * Mistral's catalog is first-party only — all models are Mistral's own,
+ * so `originProvider` is always `"mistral"`.
+ * @module
+ */
+import { OpenAICompatibleDiscoverer } from "./openai-compatible.js";
+/**
+ * Discovers models available through the Mistral AI API.
+ *
+ * Mistral's catalog is first-party — model IDs are flat strings such as
+ * `mistral-large-latest`, `codestral-latest`, and `mistral-embed`. The
+ * discoverer infers mode and capabilities from patterns in the model ID.
+ */
+export class MistralDiscoverer extends OpenAICompatibleDiscoverer {
+    providerId = "mistral";
+    providerName = "Mistral AI";
+    baseUrl = "https://api.mistral.ai";
+    /**
+     * Determine whether a model ID represents a model we want to track.
+     *
+     * We keep all standard Mistral models and filter out:
+     * - Fine-tuned model snapshots (id starts with "ft:")
+     */
+    isRelevantModel(model) {
+        // Skip fine-tuned model snapshots
+        if (model.id.startsWith("ft:"))
+            return false;
+        return true;
+    }
+    /**
+     * Classify a Mistral model: determine origin, mode, and capabilities.
+     *
+     * Key heuristics:
+     * - Origin is always "mistral"
+     * - Embedding models get mode "embedding"
+     * - Chat models get mode "chat"
+     * - Models containing "pixtral" get "vision" capability
+     * - Models containing "codestral" get "code" capability
+     * - All chat models get "function_calling" (Mistral supports it across their lineup)
+     * - Large models get "nlu"
+     */
+    classifyModel(model) {
+        const id = model.id;
+        const lower = id.toLowerCase();
+        // Determine mode
+        const mode = lower.includes("embed") ? "embedding" : "chat";
+        // Determine capabilities
+        const capabilities = [];
+        if (mode === "embedding") {
+            capabilities.push("embedding");
+        }
+        else {
+            // Chat models get function_calling
+            capabilities.push("chat");
+            capabilities.push("function_calling");
+            // Vision models
+            if (this.looksLikeVision(lower)) {
+                capabilities.push("vision");
+            }
+            // Code-specialized models
+            if (this.looksLikeCode(lower)) {
+                capabilities.push("code");
+            }
+            // Large models get NLU capability
+            if (lower.includes("large")) {
+                capabilities.push("nlu");
+            }
+        }
+        return {
+            originProvider: "mistral",
+            mode,
+            capabilities,
+            contextWindow: model.context_window,
+        };
+    }
+}
+//# sourceMappingURL=mistral.js.map

package/dist/discovery/mistral.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mistral.js","sourceRoot":"","sources":["../../src/discovery/mistral.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AAEpE;;;;;;GAMG;AACH,MAAM,OAAO,iBAAkB,SAAQ,0BAA0B;IACvD,UAAU,GAAG,SAAS,CAAC;IACvB,YAAY,GAAG,YAAY,CAAC;IAC5B,OAAO,GAAG,wBAAwB,CAAC;IAE5C;;;;;OAKG;IACO,eAAe,CAAC,KAA4B;QACrD,kCAAkC;QAClC,IAAI,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC;YAAE,OAAO,KAAK,CAAC;QAE7C,OAAO,IAAI,CAAC;IACb,CAAC;IAED;;;;;;;;;;;OAWG;IACO,aAAa,CAAC,KAA4B;QACnD,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC;QAE/B,iBAAiB;QACjB,MAAM,IAAI,GAAG,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC,CAAC,CAAC,WAAW,CAAC,CAAC,CAAC,MAAM,CAAC;QAE5D,yBAAyB;QACzB,MAAM,YAAY,GAAa,EAAE,CAAC;QAElC,IAAI,IAAI,KAAK,WAAW,EAAE,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC;QAChC,CAAC;aAAM,CAAC;YACP,mCAAmC;YACnC,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC1B,YAAY,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;YAEtC,gBAAgB;YAChB,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,EAAE,CAAC;gBACjC,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;YAC7B,CAAC;YAED,0BAA0B;YAC1B,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC,EAAE,CAAC;gBAC/B,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;YAC3B,CAAC;YAED,kCAAkC;YAClC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;gBAC7B,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;YAC1B,CAAC;QACF,CAAC;QAED,OAAO;YACN,cAAc,EAAE,SAAS;YACzB,IAAI;YACJ,YAAY;YACZ,aAAa,EAAE,KAAK,CAAC,cAAc;SACnC,CAAC;IACH,CAAC;CACD"}

package/dist/discovery/moonshot.d.ts

@@ -0,0 +1,16 @@
+/**
+ * kosha-discovery — Moonshot/Kimi provider discoverer.
+ *
+ * Moonshot (Kimi) exposes an OpenAI-compatible API.
+ * @module
+ */
+import type { OpenAICompatibleModel, ModelClassification } from "./openai-compatible.js";
+import { OpenAICompatibleDiscoverer } from "./openai-compatible.js";
+export declare class MoonshotDiscoverer extends OpenAICompatibleDiscoverer {
+    readonly providerId = "moonshot";
+    readonly providerName = "Moonshot (Kimi)";
+    readonly baseUrl = "https://api.moonshot.cn";
+    protected isRelevantModel(model: OpenAICompatibleModel): boolean;
+    protected classifyModel(model: OpenAICompatibleModel): ModelClassification;
+}
+//# sourceMappingURL=moonshot.d.ts.map

package/dist/discovery/moonshot.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"moonshot.d.ts","sourceRoot":"","sources":["../../src/discovery/moonshot.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,KAAK,EAAE,qBAAqB,EAAE,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AACzF,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AAEpE,qBAAa,kBAAmB,SAAQ,0BAA0B;IACjE,QAAQ,CAAC,UAAU,cAAc;IACjC,QAAQ,CAAC,YAAY,qBAAqB;IAC1C,QAAQ,CAAC,OAAO,6BAA6B;IAE7C,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO;IAIhE,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,qBAAqB,GAAG,mBAAmB;CA8B1E"}

package/dist/discovery/moonshot.js

@@ -0,0 +1,46 @@
+/**
+ * kosha-discovery — Moonshot/Kimi provider discoverer.
+ *
+ * Moonshot (Kimi) exposes an OpenAI-compatible API.
+ * @module
+ */
+import { OpenAICompatibleDiscoverer } from "./openai-compatible.js";
+export class MoonshotDiscoverer extends OpenAICompatibleDiscoverer {
+    providerId = "moonshot";
+    providerName = "Moonshot (Kimi)";
+    baseUrl = "https://api.moonshot.cn";
+    isRelevantModel(model) {
+        return !model.id.startsWith("ft:");
+    }
+    classifyModel(model) {
+        const lower = model.id.toLowerCase();
+        if (this.looksLikeEmbedding(lower) || lower.includes("embedding")) {
+            return {
+                originProvider: "moonshot",
+                mode: "embedding",
+                capabilities: ["embedding"],
+                contextWindow: model.context_window,
+                maxOutputTokens: model.max_output_tokens,
+                maxInputTokens: model.max_input_tokens,
+                dimensions: model.output_vector_size,
+            };
+        }
+        const capabilities = ["chat", "function_calling"];
+        // Kimi's k2.5 family includes multimodal support.
+        if (this.looksLikeVision(lower) || lower.includes("k2.5"))
+            capabilities.push("vision");
+        if (this.looksLikeCode(lower))
+            capabilities.push("code");
+        if (lower.includes("thinking") || lower.includes("reason"))
+            capabilities.push("nlu");
+        return {
+            originProvider: "moonshot",
+            mode: "chat",
+            capabilities,
+            contextWindow: model.context_window,
+            maxOutputTokens: model.max_output_tokens,
+            maxInputTokens: model.max_input_tokens,
+        };
+    }
+}
+//# sourceMappingURL=moonshot.js.map

package/dist/discovery/moonshot.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"moonshot.js","sourceRoot":"","sources":["../../src/discovery/moonshot.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAGH,OAAO,EAAE,0BAA0B,EAAE,MAAM,wBAAwB,CAAC;AAEpE,MAAM,OAAO,kBAAmB,SAAQ,0BAA0B;IACxD,UAAU,GAAG,UAAU,CAAC;IACxB,YAAY,GAAG,iBAAiB,CAAC;IACjC,OAAO,GAAG,yBAAyB,CAAC;IAEnC,eAAe,CAAC,KAA4B;QACrD,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC;IACpC,CAAC;IAES,aAAa,CAAC,KAA4B;QACnD,MAAM,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,IAAI,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACnE,OAAO;gBACN,cAAc,EAAE,UAAU;gBAC1B,IAAI,EAAE,WAAW;gBACjB,YAAY,EAAE,CAAC,WAAW,CAAC;gBAC3B,aAAa,EAAE,KAAK,CAAC,cAAc;gBACnC,eAAe,EAAE,KAAK,CAAC,iBAAiB;gBACxC,cAAc,EAAE,KAAK,CAAC,gBAAgB;gBACtC,UAAU,EAAE,KAAK,CAAC,kBAAkB;aACpC,CAAC;QACH,CAAC;QAED,MAAM,YAAY,GAAG,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;QAClD,kDAAkD;QAClD,IAAI,IAAI,CAAC,eAAe,CAAC,KAAK,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,MAAM,CAAC;YAAE,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACvF,IAAI,IAAI,CAAC,aAAa,CAAC,KAAK,CAAC;YAAE,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACzD,IAAI,KAAK,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,YAAY,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAErF,OAAO;YACN,cAAc,EAAE,UAAU;YAC1B,IAAI,EAAE,MAAM;YACZ,YAAY;YACZ,aAAa,EAAE,KAAK,CAAC,cAAc;YACnC,eAAe,EAAE,KAAK,CAAC,iBAAiB;YACxC,cAAc,EAAE,KAAK,CAAC,gBAAgB;SACtC,CAAC;IACH,CAAC;CACD"}

package/dist/discovery/nvidia.d.ts

@@ -0,0 +1,60 @@
+/**
+ * kosha-discovery — NVIDIA provider discoverer.
+ *
+ * Queries the NVIDIA build.nvidia.com `/v1/models` endpoint (OpenAI-compatible),
+ * filters relevant models, and maps them into {@link ModelCard} objects.
+ *
+ * NVIDIA's API catalog is multi-vendor, hosting models from nvidia, meta,
+ * mistral, google, microsoft, deepseek, qwen, and others.
+ * @module
+ */
+import { OpenAICompatibleDiscoverer, type OpenAICompatibleModel, type ModelClassification } from "./openai-compatible.js";
+/**
+ * Discovers models available through the NVIDIA API (build.nvidia.com).
+ *
+ * The NVIDIA catalog serves models from multiple vendors using an
+ * OpenAI-compatible API. Model IDs are namespaced (e.g. `nvidia/llama-3.1-nemotron-70b-instruct`,
+ * `meta/llama-3.1-405b-instruct`), allowing origin provider extraction from the prefix.
+ */
+export declare class NvidiaDiscoverer extends OpenAICompatibleDiscoverer {
+    readonly providerId = "nvidia";
+    readonly providerName = "NVIDIA";
+    readonly baseUrl = "https://integrate.api.nvidia.com";
+    /**
+     * Determine whether a model ID represents a model we want to track.
+     *
+     * We keep chat/instruct models and filter out:
+     * - Embedding-only models (handled separately if needed)
+     * - Reward models (used for RLHF, not inference)
+     * - Guard/safety models (used for content filtering pipelines)
+     */
+    protected isRelevantModel(model: OpenAICompatibleModel): boolean;
+    /**
+     * Classify an NVIDIA model: extract origin provider, infer mode and capabilities.
+     */
+    protected classifyModel(model: OpenAICompatibleModel): ModelClassification;
+    /**
+     * Extract the origin provider from a namespaced model ID.
+     *
+     * NVIDIA model IDs use `vendor/model-name` format. The prefix before
+     * the first `/` identifies the origin provider.
+     *
+     * @example
+     * extractOriginProvider("nvidia/llama-3.1-nemotron-70b-instruct") // "nvidia"
+     * extractOriginProvider("meta/llama-3.1-405b-instruct")           // "meta"
+     */
+    private extractOriginProvider;
+    /** Map a model ID to its primary {@link ModelMode}. */
+    private inferMode;
+    /**
+     * Infer capability flags from the model ID.
+     *
+     * Key heuristics:
+     * - Embedding models get "embedding" only
+     * - Models with "vlm", "vision", or known vision model names get "vision"
+     * - Instruct/chat models get "function_calling" (NVIDIA API supports it)
+     * - Code-specialized models get "code"
+     */
+    private inferCapabilities;
+}
+//# sourceMappingURL=nvidia.d.ts.map

package/dist/discovery/nvidia.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"nvidia.d.ts","sourceRoot":"","sources":["../../src/discovery/nvidia.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAAE,0BAA0B,EAAE,KAAK,qBAAqB,EAAE,KAAK,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAqB1H;;;;;;GAMG;AACH,qBAAa,gBAAiB,SAAQ,0BAA0B;IAC/D,QAAQ,CAAC,UAAU,YAAY;IAC/B,QAAQ,CAAC,YAAY,YAAY;IACjC,QAAQ,CAAC,OAAO,sCAAsC;IAEtD;;;;;;;OAOG;IACH,SAAS,CAAC,eAAe,CAAC,KAAK,EAAE,qBAAqB,GAAG,OAAO;IAYhE;;OAEG;IACH,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,qBAAqB,GAAG,mBAAmB;IAc1E;;;;;;;;;OASG;IACH,OAAO,CAAC,qBAAqB;IAgB7B,uDAAuD;IACvD,OAAO,CAAC,SAAS;IAMjB;;;;;;;;OAQG;IACH,OAAO,CAAC,iBAAiB;CAsCzB"}

package/dist/discovery/nvidia.js

@@ -0,0 +1,145 @@
+/**
+ * kosha-discovery — NVIDIA provider discoverer.
+ *
+ * Queries the NVIDIA build.nvidia.com `/v1/models` endpoint (OpenAI-compatible),
+ * filters relevant models, and maps them into {@link ModelCard} objects.
+ *
+ * NVIDIA's API catalog is multi-vendor, hosting models from nvidia, meta,
+ * mistral, google, microsoft, deepseek, qwen, and others.
+ * @module
+ */
+import { OpenAICompatibleDiscoverer } from "./openai-compatible.js";
+/** Known origin provider prefixes in NVIDIA's model catalog. */
+const KNOWN_ORIGIN_PREFIXES = [
+    "nvidia",
+    "meta",
+    "mistralai",
+    "mistral",
+    "google",
+    "microsoft",
+    "deepseek",
+    "qwen",
+    "ibm",
+    "snowflake",
+    "databricks",
+    "writer",
+    "mediatek",
+    "rakuten",
+    "nv-mistralai",
+];
+/**
+ * Discovers models available through the NVIDIA API (build.nvidia.com).
+ *
+ * The NVIDIA catalog serves models from multiple vendors using an
+ * OpenAI-compatible API. Model IDs are namespaced (e.g. `nvidia/llama-3.1-nemotron-70b-instruct`,
+ * `meta/llama-3.1-405b-instruct`), allowing origin provider extraction from the prefix.
+ */
+export class NvidiaDiscoverer extends OpenAICompatibleDiscoverer {
+    providerId = "nvidia";
+    providerName = "NVIDIA";
+    baseUrl = "https://integrate.api.nvidia.com";
+    /**
+     * Determine whether a model ID represents a model we want to track.
+     *
+     * We keep chat/instruct models and filter out:
+     * - Embedding-only models (handled separately if needed)
+     * - Reward models (used for RLHF, not inference)
+     * - Guard/safety models (used for content filtering pipelines)
+     */
+    isRelevantModel(model) {
+        const lower = model.id.toLowerCase();
+        // Skip reward models — used for RLHF, not direct inference
+        if (lower.includes("reward"))
+            return false;
+        // Skip guard/safety filter models (but keep llama-guard)
+        if (lower.includes("guard") && !lower.includes("llama-guard"))
+            return false;
+        return true;
+    }
+    /**
+     * Classify an NVIDIA model: extract origin provider, infer mode and capabilities.
+     */
+    classifyModel(model) {
+        const id = model.id;
+        const lower = id.toLowerCase();
+        const originProvider = this.extractOriginProvider(id);
+        const mode = this.inferMode(lower);
+        const capabilities = this.inferCapabilities(lower);
+        return {
+            originProvider,
+            mode,
+            capabilities,
+        };
+    }
+    /**
+     * Extract the origin provider from a namespaced model ID.
+     *
+     * NVIDIA model IDs use `vendor/model-name` format. The prefix before
+     * the first `/` identifies the origin provider.
+     *
+     * @example
+     * extractOriginProvider("nvidia/llama-3.1-nemotron-70b-instruct") // "nvidia"
+     * extractOriginProvider("meta/llama-3.1-405b-instruct")           // "meta"
+     */
+    extractOriginProvider(id) {
+        const slashIdx = id.indexOf("/");
+        if (slashIdx === -1)
+            return "nvidia";
+        const prefix = id.slice(0, slashIdx).toLowerCase();
+        // Normalize known prefixes
+        if (prefix === "mistralai" || prefix === "nv-mistralai")
+            return "mistral";
+        for (const known of KNOWN_ORIGIN_PREFIXES) {
+            if (prefix === known)
+                return known;
+        }
+        return prefix;
+    }
+    /** Map a model ID to its primary {@link ModelMode}. */
+    inferMode(id) {
+        if (id.includes("embed"))
+            return "embedding";
+        if (id.includes("rerank"))
+            return "embedding";
+        return "chat";
+    }
+    /**
+     * Infer capability flags from the model ID.
+     *
+     * Key heuristics:
+     * - Embedding models get "embedding" only
+     * - Models with "vlm", "vision", or known vision model names get "vision"
+     * - Instruct/chat models get "function_calling" (NVIDIA API supports it)
+     * - Code-specialized models get "code"
+     */
+    inferCapabilities(id) {
+        // Embedding models
+        if (id.includes("embed") || id.includes("rerank")) {
+            return ["embedding"];
+        }
+        const caps = ["chat"];
+        // Vision models
+        if (id.includes("vlm") ||
+            id.includes("vision") ||
+            id.includes("llava") ||
+            id.includes("cogvlm") ||
+            id.includes("fuyu") ||
+            id.includes("neva") ||
+            id.includes("vila")) {
+            caps.push("vision");
+        }
+        // Code-specialized models
+        if (id.includes("code") ||
+            id.includes("codestral") ||
+            id.includes("starcoder") ||
+            id.includes("codellama")) {
+            caps.push("code");
+        }
+        // Function calling — most instruct models on NVIDIA support it
+        if (id.includes("instruct") || id.includes("chat") || id.includes("nemotron")) {
+            caps.push("function_calling");
+        }
+        return caps;
+    }
+}
+//# sourceMappingURL=nvidia.js.map

package/dist/discovery/nvidia.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"nvidia.js","sourceRoot":"","sources":["../../src/discovery/nvidia.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAAE,0BAA0B,EAAwD,MAAM,wBAAwB,CAAC;AAE1H,gEAAgE;AAChE,MAAM,qBAAqB,GAAG;IAC7B,QAAQ;IACR,MAAM;IACN,WAAW;IACX,SAAS;IACT,QAAQ;IACR,WAAW;IACX,UAAU;IACV,MAAM;IACN,KAAK;IACL,WAAW;IACX,YAAY;IACZ,QAAQ;IACR,UAAU;IACV,SAAS;IACT,cAAc;CACL,CAAC;AAEX;;;;;;GAMG;AACH,MAAM,OAAO,gBAAiB,SAAQ,0BAA0B;IACtD,UAAU,GAAG,QAAQ,CAAC;IACtB,YAAY,GAAG,QAAQ,CAAC;IACxB,OAAO,GAAG,kCAAkC,CAAC;IAEtD;;;;;;;OAOG;IACO,eAAe,CAAC,KAA4B;QACrD,MAAM,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QAErC,2DAA2D;QAC3D,IAAI,KAAK,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,KAAK,CAAC;QAE3C,yDAAyD;QACzD,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC;YAAE,OAAO,KAAK,CAAC;QAE5E,OAAO,IAAI,CAAC;IACb,CAAC;IAED;;OAEG;IACO,aAAa,CAAC,KAA4B;QACnD,MAAM,EAAE,GAAG,KAAK,CAAC,EAAE,CAAC;QACpB,MAAM,KAAK,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC;QAC/B,MAAM,cAAc,GAAG,IAAI,CAAC,qBAAqB,CAAC,EAAE,CAAC,CAAC;QACtD,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QAEnD,OAAO;YACN,cAAc;YACd,IAAI;YACJ,YAAY;SACZ,CAAC;IACH,CAAC;IAED;;;;;;;;;OASG;IACK,qBAAqB,CAAC,EAAU;QACvC,MAAM,QAAQ,GAAG,EAAE,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;QACjC,IAAI,QAAQ,KAAK,CAAC,CAAC;YAAE,OAAO,QAAQ,CAAC;QAErC,MAAM,MAAM,GAAG,EAAE,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;QAEnD,2BAA2B;QAC3B,IAAI,MAAM,KAAK,WAAW,IAAI,MAAM,KAAK,cAAc;YAAE,OAAO,SAAS,CAAC;QAE1E,KAAK,MAAM,KAAK,IAAI,qBAAqB,EAAE,CAAC;YAC3C,IAAI,MAAM,KAAK,KAAK;gBAAE,OAAO,KAAK,CAAC;QACpC,CAAC;QAED,OAAO,MAAM,CAAC;IACf,CAAC;IAED,uDAAuD;IAC/C,SAAS,CAAC,EAAU;QAC3B,IAAI,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,OAAO,WAAW,CAAC;QAC7C,IAAI,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;YAAE,OAAO,WAAW,CAAC;QAC9C,OAAO,MAAM,CAAC;IACf,CAAC;IAED;;;;;;;;OAQG;IACK,iBAAiB,CAAC,EAAU;QACnC,mBAAmB;QACnB,IAAI,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACnD,OAAO,CAAC,WAAW,CAAC,CAAC;QACtB,CAAC;QAED,MAAM,IAAI,GAAa,CAAC,MAAM,CAAC,CAAC;QAEhC,gBAAgB;QAChB,IACC,EAAE,CAAC,QAAQ,CAAC,KAAK,CAAC;YAClB,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACrB,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC;YACpB,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC;YACrB,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,EAClB,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QACrB,CAAC;QAED,0BAA0B;QAC1B,IACC,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC;YACnB,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;YACxB,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC;YACxB,EAAE,CAAC,QAAQ,CAAC,WAAW,CAAC,EACvB,CAAC;YACF,IAAI,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QACnB,CAAC;QAED,+DAA+D;QAC/D,IAAI,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,UAAU,CAAC,EAAE,CAAC;YAC/E,IAAI,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QAC/B,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;CACD"}

package/dist/discovery/ollama.d.ts

@@ -0,0 +1,57 @@
+/**
+ * kosha-discovery — Ollama (local) provider discoverer.
+ *
+ * Queries the Ollama REST API (`/api/tags`) on localhost to list locally
+ * pulled models.  Because Ollama runs without authentication and may not
+ * be installed at all, this discoverer fails gracefully on connection errors.
+ * @module
+ */
+import type { CredentialResult, ModelCard } from "../types.js";
+import { BaseDiscoverer } from "./base.js";
+/**
+ * Discovers locally-pulled models from an Ollama instance.
+ *
+ * Ollama requires no authentication, so the credential is ignored.
+ * If the Ollama daemon is not running, discovery returns an empty array
+ * instead of throwing, making it safe to include in every scan.
+ */
+export declare class OllamaDiscoverer extends BaseDiscoverer {
+    readonly providerId = "ollama";
+    readonly providerName = "Ollama (Local)";
+    readonly baseUrl: string;
+    /** Set of model names currently loaded in VRAM (warm). */
+    private runningModels;
+    /**
+     * @param baseUrl - Override the default Ollama URL (default: `http://localhost:11434`).
+     *                  Useful when Ollama is running on a non-standard port or remote host.
+     */
+    constructor(baseUrl?: string);
+    /**
+     * List all locally-pulled models from Ollama.
+     *
+     * The `_credential` parameter is ignored — Ollama never requires auth.
+     * Uses a shorter default timeout (5 s) since the server is local.
+     */
+    discover(_credential: CredentialResult, options?: {
+        timeout?: number;
+    }): Promise<ModelCard[]>;
+    /**
+     * Fetch the list of models currently loaded in VRAM.
+     * Failures are silently ignored since this is purely informational.
+     */
+    private fetchRunningModels;
+    /** Convert an Ollama model into a normalized {@link ModelCard}. */
+    private toModelCard;
+    /**
+     * Infer capabilities from the model name and Ollama-provided family metadata.
+     */
+    private inferCapabilities;
+    /**
+     * Detect embedding models by common naming conventions.
+     * Patterns: "embed", "nomic-embed", "mxbai-embed".
+     */
+    private isEmbeddingModel;
+    /** Check whether an error represents a network connectivity failure. */
+    private isConnectionError;
+}
+//# sourceMappingURL=ollama.d.ts.map

package/dist/discovery/ollama.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.d.ts","sourceRoot":"","sources":["../../src/discovery/ollama.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAEH,OAAO,KAAK,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAsC3C;;;;;;GAMG;AACH,qBAAa,gBAAiB,SAAQ,cAAc;IACnD,QAAQ,CAAC,UAAU,YAAY;IAC/B,QAAQ,CAAC,YAAY,oBAAoB;IACzC,QAAQ,CAAC,OAAO,EAAE,MAAM,CAAC;IAEzB,0DAA0D;IAC1D,OAAO,CAAC,aAAa,CAAqB;IAE1C;;;OAGG;gBACS,OAAO,CAAC,EAAE,MAAM;IAK5B;;;;;OAKG;IACG,QAAQ,CAAC,WAAW,EAAE,gBAAgB,EAAE,OAAO,CAAC,EAAE;QAAE,OAAO,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,SAAS,EAAE,CAAC;IAyBnG;;;OAGG;YACW,kBAAkB;IAahC,mEAAmE;IACnE,OAAO,CAAC,WAAW;IA+BnB;;OAEG;IACH,OAAO,CAAC,iBAAiB;IA6BzB;;;OAGG;IACH,OAAO,CAAC,gBAAgB;IAKxB,wEAAwE;IACxE,OAAO,CAAC,iBAAiB;CAWzB"}

package/dist/discovery/ollama.js

@@ -0,0 +1,151 @@
+/**
+ * kosha-discovery — Ollama (local) provider discoverer.
+ *
+ * Queries the Ollama REST API (`/api/tags`) on localhost to list locally
+ * pulled models.  Because Ollama runs without authentication and may not
+ * be installed at all, this discoverer fails gracefully on connection errors.
+ * @module
+ */
+import { BaseDiscoverer } from "./base.js";
+/**
+ * Discovers locally-pulled models from an Ollama instance.
+ *
+ * Ollama requires no authentication, so the credential is ignored.
+ * If the Ollama daemon is not running, discovery returns an empty array
+ * instead of throwing, making it safe to include in every scan.
+ */
+export class OllamaDiscoverer extends BaseDiscoverer {
+    providerId = "ollama";
+    providerName = "Ollama (Local)";
+    baseUrl;
+    /** Set of model names currently loaded in VRAM (warm). */
+    runningModels = new Set();
+    /**
+     * @param baseUrl - Override the default Ollama URL (default: `http://localhost:11434`).
+     *                  Useful when Ollama is running on a non-standard port or remote host.
+     */
+    constructor(baseUrl) {
+        super();
+        this.baseUrl = baseUrl ?? "http://localhost:11434";
+    }
+    /**
+     * List all locally-pulled models from Ollama.
+     *
+     * The `_credential` parameter is ignored — Ollama never requires auth.
+     * Uses a shorter default timeout (5 s) since the server is local.
+     */
+    async discover(_credential, options) {
+        const timeoutMs = this.validateTimeout(options?.timeout, 5_000);
+        // Best-effort fetch of currently loaded (warm) models — non-critical
+        await this.fetchRunningModels(timeoutMs);
+        let response;
+        try {
+            response = await this.fetchJSON(`${this.baseUrl}/api/tags`, undefined, timeoutMs);
+        }
+        catch (error) {
+            // Gracefully return empty when Ollama is not installed or not running,
+            // rather than propagating connection errors up to the registry
+            if (this.isConnectionError(error)) {
+                return [];
+            }
+            throw error;
+        }
+        if (!response.models || !Array.isArray(response.models)) {
+            return [];
+        }
+        return response.models.map((model) => this.toModelCard(model));
+    }
+    /**
+     * Fetch the list of models currently loaded in VRAM.
+     * Failures are silently ignored since this is purely informational.
+     */
+    async fetchRunningModels(timeoutMs) {
+        try {
+            const response = await this.fetchJSON(`${this.baseUrl}/api/ps`, undefined, timeoutMs);
+            if (response.models && Array.isArray(response.models)) {
+                for (const model of response.models) {
+                    this.runningModels.add(model.name);
+                }
+            }
+        }
+        catch {
+            // Non-critical — running model info is nice-to-have only
+        }
+    }
+    /** Convert an Ollama model into a normalized {@link ModelCard}. */
+    toModelCard(model) {
+        const capabilities = this.inferCapabilities(model);
+        const mode = this.isEmbeddingModel(model.name) ? "embedding" : "chat";
+        const families = Array.isArray(model.details?.families) ? model.details.families : [];
+        return this.makeCard({
+            id: model.name,
+            name: model.name,
+            provider: this.providerId,
+            mode,
+            capabilities,
+            rawCapabilities: [...capabilities],
+            // Ollama's /api/tags does not return context window info;
+            // the litellm enricher will fill these in later
+            contextWindow: 0,
+            maxOutputTokens: 0,
+            source: "local",
+            localRuntime: {
+                runtimeFamily: "ollama",
+                transport: "native-http",
+                tokenizerFamily: families[0] ?? undefined,
+                quantization: model.details?.quantization_level,
+                memoryFootprintBytes: model.size,
+                computeTarget: this.runningModels.has(model.name) ? "gpu" : "unknown",
+                // Ollama supports streaming and structured JSON at the runtime layer.
+                supportsStructuredOutput: true,
+                supportsStreaming: true,
+            },
+        });
+    }
+    /**
+     * Infer capabilities from the model name and Ollama-provided family metadata.
+     */
+    inferCapabilities(model) {
+        const name = model.name.toLowerCase();
+        const families = Array.isArray(model.details?.families) ? model.details.families : [];
+        // Embedding models — detected by common naming patterns:
+        // "nomic-embed-text", "mxbai-embed-large", "all-minilm", etc.
+        if (this.isEmbeddingModel(name)) {
+            return ["embedding"];
+        }
+        const capabilities = ["chat"];
+        // Code-focused models (codestral, deepseek-coder, starcoder, codellama)
+        if (name.includes("code") || name.includes("coder") || name.includes("starcoder") || name.includes("codellama")) {
+            capabilities.push("code");
+        }
+        else {
+            // Most general-purpose local models (llama, qwen, mistral) handle code too
+            capabilities.push("code");
+        }
+        // Vision support: Ollama reports model families (e.g. "clip") in metadata;
+        // also detect by well-known multimodal model names
+        if (families.includes("clip") || name.includes("llava") || name.includes("vision")) {
+            capabilities.push("vision");
+        }
+        return capabilities;
+    }
+    /**
+     * Detect embedding models by common naming conventions.
+     * Patterns: "embed", "nomic-embed", "mxbai-embed".
+     */
+    isEmbeddingModel(name) {
+        const lower = name.toLowerCase();
+        return lower.includes("embed") || lower.includes("nomic-embed") || lower.includes("mxbai-embed");
+    }
+    /** Check whether an error represents a network connectivity failure. */
+    isConnectionError(error) {
+        const message = (error instanceof Error ? error.message : String(error)).toLowerCase();
+        return (message.includes("econnrefused") ||
+            message.includes("enotfound") ||
+            message.includes("econnreset") ||
+            message.includes("fetch failed") ||
+            message.includes("network") ||
+            message.includes("timed out"));
+    }
+}
+//# sourceMappingURL=ollama.js.map

package/dist/discovery/ollama.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"ollama.js","sourceRoot":"","sources":["../../src/discovery/ollama.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAGH,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAsC3C;;;;;;GAMG;AACH,MAAM,OAAO,gBAAiB,SAAQ,cAAc;IAC1C,UAAU,GAAG,QAAQ,CAAC;IACtB,YAAY,GAAG,gBAAgB,CAAC;IAChC,OAAO,CAAS;IAEzB,0DAA0D;IAClD,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;IAE1C;;;OAGG;IACH,YAAY,OAAgB;QAC3B,KAAK,EAAE,CAAC;QACR,IAAI,CAAC,OAAO,GAAG,OAAO,IAAI,wBAAwB,CAAC;IACpD,CAAC;IAED;;;;;OAKG;IACH,KAAK,CAAC,QAAQ,CAAC,WAA6B,EAAE,OAA8B;QAC3E,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,EAAE,KAAK,CAAC,CAAC;QAEhE,qEAAqE;QACrE,MAAM,IAAI,CAAC,kBAAkB,CAAC,SAAS,CAAC,CAAC;QAEzC,IAAI,QAA4B,CAAC;QACjC,IAAI,CAAC;YACJ,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAqB,GAAG,IAAI,CAAC,OAAO,WAAW,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;QACvG,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,uEAAuE;YACvE,+DAA+D;YAC/D,IAAI,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,EAAE,CAAC;gBACnC,OAAO,EAAE,CAAC;YACX,CAAC;YACD,MAAM,KAAK,CAAC;QACb,CAAC;QAED,IAAI,CAAC,QAAQ,CAAC,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YACzD,OAAO,EAAE,CAAC;QACX,CAAC;QAED,OAAO,QAAQ,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,CAAC,CAAC,CAAC;IAChE,CAAC;IAED;;;OAGG;IACK,KAAK,CAAC,kBAAkB,CAAC,SAAiB;QACjD,IAAI,CAAC;YACJ,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAmB,GAAG,IAAI,CAAC,OAAO,SAAS,EAAE,SAAS,EAAE,SAAS,CAAC,CAAC;YACxG,IAAI,QAAQ,CAAC,MAAM,IAAI,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;gBACvD,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,MAAM,EAAE,CAAC;oBACrC,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;gBACpC,CAAC;YACF,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,yDAAyD;QAC1D,CAAC;IACF,CAAC;IAED,mEAAmE;IAC3D,WAAW,CAAC,KAAkB;QACrC,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,IAAI,GAAG,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,WAAoB,CAAC,CAAC,CAAC,MAAe,CAAC;QACxF,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAEtF,OAAO,IAAI,CAAC,QAAQ,CAAC;YACpB,EAAE,EAAE,KAAK,CAAC,IAAI;YACd,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,IAAI,CAAC,UAAU;YACzB,IAAI;YACJ,YAAY;YACZ,eAAe,EAAE,CAAC,GAAG,YAAY,CAAC;YAClC,0DAA0D;YAC1D,gDAAgD;YAChD,aAAa,EAAE,CAAC;YAChB,eAAe,EAAE,CAAC;YAClB,MAAM,EAAE,OAAO;YACf,YAAY,EAAE;gBACb,aAAa,EAAE,QAAQ;gBACvB,SAAS,EAAE,aAAa;gBACxB,eAAe,EAAE,QAAQ,CAAC,CAAC,CAAC,IAAI,SAAS;gBACzC,YAAY,EAAE,KAAK,CAAC,OAAO,EAAE,kBAAkB;gBAC/C,oBAAoB,EAAE,KAAK,CAAC,IAAI;gBAChC,aAAa,EAAE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,SAAS;gBACrE,sEAAsE;gBACtE,wBAAwB,EAAE,IAAI;gBAC9B,iBAAiB,EAAE,IAAI;aACvB;SACD,CAAC,CAAC;IACJ,CAAC;IAED;;OAEG;IACK,iBAAiB,CAAC,KAAkB;QAC3C,MAAM,IAAI,GAAG,KAAK,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACtC,MAAM,QAAQ,GAAG,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC,CAAC,EAAE,CAAC;QAEtF,yDAAyD;QACzD,8DAA8D;QAC9D,IAAI,IAAI,CAAC,gBAAgB,CAAC,IAAI,CAAC,EAAE,CAAC;YACjC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtB,CAAC;QAED,MAAM,YAAY,GAAa,CAAC,MAAM,CAAC,CAAC;QAExC,wEAAwE;QACxE,IAAI,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC,EAAE,CAAC;YACjH,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;aAAM,CAAC;YACP,2EAA2E;YAC3E,YAAY,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC;QAC3B,CAAC;QAED,2EAA2E;QAC3E,mDAAmD;QACnD,IAAI,QAAQ,CAAC,QAAQ,CAAC,MAAM,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YACpF,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC;QAED,OAAO,YAAY,CAAC;IACrB,CAAC;IAED;;;OAGG;IACK,gBAAgB,CAAC,IAAY;QACpC,MAAM,KAAK,GAAG,IAAI,CAAC,WAAW,EAAE,CAAC;QACjC,OAAO,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,IAAI,KAAK,CAAC,QAAQ,CAAC,aAAa,CAAC,CAAC;IAClG,CAAC;IAED,wEAAwE;IAChE,iBAAiB,CAAC,KAAc;QACvC,MAAM,OAAO,GAAG,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC;QACvF,OAAO,CACN,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;YAChC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC;YAC7B,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAC;YAC9B,OAAO,CAAC,QAAQ,CAAC,cAAc,CAAC;YAChC,OAAO,CAAC,QAAQ,CAAC,SAAS,CAAC;YAC3B,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAC,CAC7B,CAAC;IACH,CAAC;CACD"}