@sriinnu/kosha-discovery 0.5.0

package/dist/discovery/bedrock.js

@@ -0,0 +1,356 @@
+/**
+ * kosha-discovery — AWS Bedrock model discoverer.
+ *
+ * Resolution strategy (in order):
+ * 1. AWS SDK (`@aws-sdk/client-bedrock`) — if installed in the host project.
+ * 2. AWS CLI fallback (`aws bedrock list-foundation-models --output json`).
+ * 3. Static fallback list of well-known Bedrock foundation models (Feb 2026).
+ *
+ * Credentials use the standard AWS credential chain:
+ * environment variables (`AWS_ACCESS_KEY_ID` / `AWS_SECRET_ACCESS_KEY`),
+ * `~/.aws/credentials`, EC2 / ECS instance-metadata IAM roles, etc.
+ * No explicit credential is required by this discoverer — the SDK and CLI
+ * both pick up the ambient credential automatically.
+ *
+ * @module
+ */
+import { execFileSync } from "node:child_process";
+import { assertSafeShellArg } from "../shell-safe.js";
+import { assertCleanPayload } from "../security.js";
+import { BaseDiscoverer } from "./base.js";
+// ---------------------------------------------------------------------------
+// Vendor prefix → originProvider mapping
+// ---------------------------------------------------------------------------
+/**
+ * Maps the Bedrock model-ID vendor prefix to the canonical origin provider.
+ *
+ * Bedrock model IDs follow the pattern `{vendor}.{model-family}-v{n}:{variant}`,
+ * e.g. `anthropic.claude-opus-4-6-v1:0`, `amazon.titan-text-premier-v2:0`.
+ */
+const VENDOR_TO_ORIGIN = {
+    anthropic: "anthropic",
+    amazon: "amazon",
+    meta: "meta",
+    mistral: "mistral",
+    cohere: "cohere",
+    ai21: "ai21",
+    stability: "stability",
+    "amazon-bedrock-preview": "amazon",
+};
+// ---------------------------------------------------------------------------
+// Static fallback catalogue (Feb 2026)
+// ---------------------------------------------------------------------------
+/**
+ * A minimal set of well-known Bedrock foundation models used when neither
+ * the SDK nor the CLI is available.  Context windows and output limits are
+ * intentionally set to 0 so that the litellm enrichment pass can fill them in.
+ */
+const STATIC_MODELS = [
+    {
+        id: "anthropic.claude-opus-4-6-v1:0",
+        name: "Claude Opus 4.6 (Bedrock)",
+        mode: "chat",
+        capabilities: ["chat", "vision", "code", "nlu", "function_calling"],
+        originProvider: "anthropic",
+    },
+    {
+        id: "anthropic.claude-sonnet-4-6-v1:0",
+        name: "Claude Sonnet 4.6 (Bedrock)",
+        mode: "chat",
+        capabilities: ["chat", "vision", "code", "nlu", "function_calling"],
+        originProvider: "anthropic",
+    },
+    {
+        id: "anthropic.claude-haiku-4-5-v1:0",
+        name: "Claude Haiku 4.5 (Bedrock)",
+        mode: "chat",
+        capabilities: ["chat", "vision", "code", "nlu", "function_calling"],
+        originProvider: "anthropic",
+    },
+    {
+        id: "amazon.titan-text-premier-v2:0",
+        name: "Titan Text Premier v2 (Bedrock)",
+        mode: "chat",
+        capabilities: ["chat", "nlu"],
+        originProvider: "amazon",
+    },
+    {
+        id: "amazon.titan-embed-text-v2:0",
+        name: "Titan Embed Text v2 (Bedrock)",
+        mode: "embedding",
+        capabilities: ["embedding"],
+        originProvider: "amazon",
+    },
+    {
+        id: "meta.llama3-3-70b-instruct-v1:0",
+        name: "Llama 3.3 70B Instruct (Bedrock)",
+        mode: "chat",
+        capabilities: ["chat", "code", "nlu"],
+        originProvider: "meta",
+    },
+    {
+        id: "mistral.mistral-large-2411-v1:0",
+        name: "Mistral Large 2411 (Bedrock)",
+        mode: "chat",
+        capabilities: ["chat", "code", "nlu", "function_calling"],
+        originProvider: "mistral",
+    },
+];
+// ---------------------------------------------------------------------------
+// BedrockDiscoverer
+// ---------------------------------------------------------------------------
+/**
+ * Discovers models available through AWS Bedrock.
+ *
+ * Tries three resolution strategies in order:
+ * 1. `@aws-sdk/client-bedrock` (if installed in the host project).
+ * 2. AWS CLI (`aws bedrock list-foundation-models`).
+ * 3. Static fallback catalogue.
+ *
+ * The `credential` argument is optional in the sense that the AWS SDK and CLI
+ * both rely on the ambient credential chain.  However, the `credential.metadata`
+ * bag may carry an explicit `{ region }` override.
+ */
+export class BedrockDiscoverer extends BaseDiscoverer {
+    providerId = "bedrock";
+    providerName = "AWS Bedrock";
+    baseUrl = "https://bedrock.us-east-1.amazonaws.com";
+    /**
+     * Discover all foundation models available on AWS Bedrock.
+     *
+     * Resolution order: SDK → CLI → static fallback.
+     *
+     * @param credential - Optional credential; `metadata.region` overrides the
+     *                     `AWS_DEFAULT_REGION` environment variable.
+     * @param options    - Optional timeout in milliseconds (used for CLI exec).
+     */
+    async discover(credential, options) {
+        // 1. Try the AWS SDK first (zero-config, respects credential chain)
+        try {
+            const models = await this.discoverViaSdk(credential, options);
+            if (models.length > 0) {
+                return models;
+            }
+        }
+        catch {
+            // SDK not installed or threw — fall through to CLI
+        }
+        // 2. Fall back to the AWS CLI
+        try {
+            const models = await this.discoverViaCli(credential, options);
+            if (models.length > 0) {
+                return models;
+            }
+        }
+        catch {
+            // CLI not installed or threw — fall through to static list
+        }
+        // 3. Last resort: static catalogue
+        return this.staticFallback();
+    }
+    // -------------------------------------------------------------------------
+    // Strategy 1 — AWS SDK
+    // -------------------------------------------------------------------------
+    /**
+     * Use `@aws-sdk/client-bedrock` to list foundation models.
+     *
+     * Dynamic import prevents a hard dependency: the SDK is optional and must
+     * be installed separately by the host project.  If the import fails we
+     * throw so the caller can fall through to the next strategy.
+     *
+     * @throws When `@aws-sdk/client-bedrock` is not installed, or the API call fails.
+     */
+    async discoverViaSdk(credential, _options) {
+        // Dynamic import: will throw MODULE_NOT_FOUND if the SDK is absent.
+        // The SDK is an optional peer dependency — not installed in kosha itself.
+        // @ts-expect-error — optional peer dep, resolved at runtime by host project
+        const { BedrockClient, ListFoundationModelsCommand } = await import("@aws-sdk/client-bedrock");
+        const region = this.resolveRegion(credential);
+        const client = new BedrockClient({ region });
+        const response = await client.send(new ListFoundationModelsCommand({}));
+        const summaries = response.modelSummaries ?? [];
+        return summaries.map((m) => this.toModelCard(m, "api", credential));
+    }
+    // -------------------------------------------------------------------------
+    // Strategy 2 — AWS CLI
+    // -------------------------------------------------------------------------
+    /**
+     * Use the `aws` CLI to list foundation models.
+     *
+     * Runs `aws bedrock list-foundation-models --region {region} --output json`
+     * synchronously.  Wrapped in try/catch so failures are surfaced to the caller
+     * as thrown errors, letting `discover()` fall through gracefully.
+     *
+     * @throws When the `aws` CLI is not installed or returns a non-zero exit code.
+     */
+    async discoverViaCli(credential, options) {
+        const region = this.resolveRegion(credential);
+        // Default CLI timeout: 15 s (slower than SDK due to subprocess overhead)
+        const timeoutMs = this.validateTimeout(options?.timeout, 15_000);
+        let stdout;
+        try {
+            assertSafeShellArg(region, "region");
+            stdout = execFileSync("aws", ["bedrock", "list-foundation-models", "--region", region, "--output", "json"], {
+                timeout: timeoutMs,
+                encoding: "utf8",
+                // Suppress stderr — credential errors print there, not on stdout
+                stdio: ["ignore", "pipe", "ignore"],
+            });
+        }
+        catch (error) {
+            // Re-throw so discover() can fall through to the static list
+            throw new Error(`AWS CLI failed: ${error instanceof Error ? error.message : String(error)}`);
+        }
+        let parsed;
+        try {
+            parsed = JSON.parse(stdout);
+        }
+        catch {
+            throw new Error("AWS CLI returned invalid JSON");
+        }
+        assertCleanPayload(parsed, "Bedrock AWS CLI");
+        return (parsed.modelSummaries ?? []).map((m) => this.toModelCard(m, "api", credential));
+    }
+    // -------------------------------------------------------------------------
+    // Strategy 3 — Static fallback
+    // -------------------------------------------------------------------------
+    /**
+     * Return a hardcoded catalogue of well-known Bedrock foundation models.
+     *
+     * Used when neither the SDK nor the CLI is available.  All entries have
+     * `source: "manual"` and zero context/output token limits — the litellm
+     * enrichment pass is expected to populate those fields later.
+     */
+    staticFallback() {
+        return STATIC_MODELS.map((entry) => this.makeCard({
+            id: entry.id,
+            name: entry.name,
+            provider: this.providerId,
+            originProvider: entry.originProvider,
+            mode: entry.mode,
+            capabilities: [...entry.capabilities],
+            contextWindow: 0,
+            maxOutputTokens: 0,
+            source: "manual",
+        }));
+    }
+    // -------------------------------------------------------------------------
+    // Shared mapping helpers
+    // -------------------------------------------------------------------------
+    /**
+     * Convert a raw Bedrock foundation model summary into a {@link ModelCard}.
+     *
+     * @param model  - Raw model summary from SDK or CLI.
+     * @param source - Whether the data came from a live API call or static data.
+     * @param credential - Used to attach the resolved AWS region.
+     */
+    toModelCard(model, source, credential) {
+        const mode = this.inferMode(model);
+        const capabilities = this.inferCapabilities(model);
+        const originProvider = inferOriginFromBedrockId(model.modelId);
+        const region = this.resolveRegion(credential);
+        return this.makeCard({
+            id: model.modelId,
+            name: model.modelName || model.modelId,
+            provider: this.providerId,
+            originProvider,
+            mode,
+            capabilities,
+            contextWindow: 0,
+            maxOutputTokens: 0,
+            source,
+            region,
+        });
+    }
+    /**
+     * Infer the primary {@link ModelMode} from Bedrock output modalities.
+     *
+     * Bedrock reports modalities as uppercase strings: "TEXT", "IMAGE", "EMBEDDING".
+     * We pick the most specific non-TEXT modality first; TEXT-only → "chat".
+     */
+    inferMode(model) {
+        const output = (model.outputModalities ?? []).map((m) => m.toUpperCase());
+        const id = model.modelId.toLowerCase();
+        // Embedding: explicit modality or keyword in ID
+        if (output.includes("EMBEDDING") || id.includes("embed")) {
+            return "embedding";
+        }
+        // Image generation: output includes IMAGE but not TEXT
+        if (output.includes("IMAGE") && !output.includes("TEXT")) {
+            return "image";
+        }
+        return "chat";
+    }
+    /**
+     * Infer capability flags from Bedrock input/output modalities and inference types.
+     *
+     * - "embedding" models → ["embedding"]
+     * - "image" output-only models → ["image"]
+     * - All chat models get ["chat"] plus optional vision/code/function_calling
+     */
+    inferCapabilities(model) {
+        const input = (model.inputModalities ?? []).map((m) => m.toUpperCase());
+        const output = (model.outputModalities ?? []).map((m) => m.toUpperCase());
+        const id = model.modelId.toLowerCase();
+        // Embedding models have a single capability
+        if (output.includes("EMBEDDING") || id.includes("embed")) {
+            return ["embedding"];
+        }
+        // Image-generation-only models
+        if (output.includes("IMAGE") && !output.includes("TEXT")) {
+            return ["image"];
+        }
+        const capabilities = ["chat"];
+        // Vision: the model accepts IMAGE input
+        if (input.includes("IMAGE")) {
+            capabilities.push("vision");
+        }
+        // Claude models on Bedrock always support code, NLU, and function calling
+        if (id.includes("claude")) {
+            capabilities.push("code", "nlu", "function_calling");
+        }
+        // Mistral large variants support function calling
+        if (id.includes("mistral") && id.includes("large")) {
+            capabilities.push("function_calling");
+        }
+        // Streaming inference type is informational — not a capability flag
+        return capabilities;
+    }
+    /**
+     * Resolve the AWS region to use for API calls.
+     *
+     * Priority: `credential.metadata.region` > `AWS_DEFAULT_REGION` env var > `"us-east-1"`.
+     */
+    resolveRegion(credential) {
+        return credential.metadata?.region ?? process.env.AWS_DEFAULT_REGION ?? "us-east-1";
+    }
+}
+// ---------------------------------------------------------------------------
+// Exported helper (also used in tests)
+// ---------------------------------------------------------------------------
+/**
+ * Extract the origin provider from a Bedrock model ID.
+ *
+ * Bedrock model IDs follow the pattern `{vendor}.{model-name}-v{n}:{variant}`,
+ * e.g. `anthropic.claude-opus-4-6-v1:0` → `"anthropic"`.
+ *
+ * Falls back to `"unknown"` for unrecognised vendor prefixes.
+ *
+ * @param modelId - A Bedrock foundation model ID.
+ * @returns The canonical origin provider string.
+ *
+ * @example
+ * inferOriginFromBedrockId("anthropic.claude-sonnet-4-6-v1:0") // "anthropic"
+ * inferOriginFromBedrockId("meta.llama3-3-70b-instruct-v1:0")  // "meta"
+ * inferOriginFromBedrockId("amazon.titan-text-premier-v2:0")   // "amazon"
+ */
+export function inferOriginFromBedrockId(modelId) {
+    // The vendor prefix is everything before the first dot
+    const dotIndex = modelId.indexOf(".");
+    if (dotIndex === -1) {
+        return "unknown";
+    }
+    const vendor = modelId.slice(0, dotIndex).toLowerCase();
+    return VENDOR_TO_ORIGIN[vendor] ?? "unknown";
+}
+//# sourceMappingURL=bedrock.js.map

package/dist/discovery/bedrock.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"bedrock.js","sourceRoot":"","sources":["../../src/discovery/bedrock.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;;GAeG;AAEH,OAAO,EAAE,YAAY,EAAE,MAAM,oBAAoB,CAAC;AAClD,OAAO,EAAE,kBAAkB,EAAE,MAAM,kBAAkB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,gBAAgB,CAAC;AAEpD,OAAO,EAAE,cAAc,EAAE,MAAM,WAAW,CAAC;AAgC3C,8EAA8E;AAC9E,yCAAyC;AACzC,8EAA8E;AAE9E;;;;;GAKG;AACH,MAAM,gBAAgB,GAA2B;IAChD,SAAS,EAAE,WAAW;IACtB,MAAM,EAAE,QAAQ;IAChB,IAAI,EAAE,MAAM;IACZ,OAAO,EAAE,SAAS;IAClB,MAAM,EAAE,QAAQ;IAChB,IAAI,EAAE,MAAM;IACZ,SAAS,EAAE,WAAW;IACtB,wBAAwB,EAAE,QAAQ;CAClC,CAAC;AAEF,8EAA8E;AAC9E,uCAAuC;AACvC,8EAA8E;AAE9E;;;;GAIG;AACH,MAAM,aAAa,GAEf;IACH;QACC,EAAE,EAAE,gCAAgC;QACpC,IAAI,EAAE,2BAA2B;QACjC,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,CAAC;QACnE,cAAc,EAAE,WAAW;KAC3B;IACD;QACC,EAAE,EAAE,kCAAkC;QACtC,IAAI,EAAE,6BAA6B;QACnC,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,CAAC;QACnE,cAAc,EAAE,WAAW;KAC3B;IACD;QACC,EAAE,EAAE,iCAAiC;QACrC,IAAI,EAAE,4BAA4B;QAClC,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,CAAC,MAAM,EAAE,QAAQ,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,CAAC;QACnE,cAAc,EAAE,WAAW;KAC3B;IACD;QACC,EAAE,EAAE,gCAAgC;QACpC,IAAI,EAAE,iCAAiC;QACvC,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,CAAC,MAAM,EAAE,KAAK,CAAC;QAC7B,cAAc,EAAE,QAAQ;KACxB;IACD;QACC,EAAE,EAAE,8BAA8B;QAClC,IAAI,EAAE,+BAA+B;QACrC,IAAI,EAAE,WAAW;QACjB,YAAY,EAAE,CAAC,WAAW,CAAC;QAC3B,cAAc,EAAE,QAAQ;KACxB;IACD;QACC,EAAE,EAAE,iCAAiC;QACrC,IAAI,EAAE,kCAAkC;QACxC,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,CAAC;QACrC,cAAc,EAAE,MAAM;KACtB;IACD;QACC,EAAE,EAAE,iCAAiC;QACrC,IAAI,EAAE,8BAA8B;QACpC,IAAI,EAAE,MAAM;QACZ,YAAY,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,KAAK,EAAE,kBAAkB,CAAC;QACzD,cAAc,EAAE,SAAS;KACzB;CACD,CAAC;AAEF,8EAA8E;AAC9E,oBAAoB;AACpB,8EAA8E;AAE9E;;;;;;;;;;;GAWG;AACH,MAAM,OAAO,iBAAkB,SAAQ,cAAc;IAC3C,UAAU,GAAG,SAAS,CAAC;IACvB,YAAY,GAAG,aAAa,CAAC;IAC7B,OAAO,GAAG,yCAAyC,CAAC;IAE7D;;;;;;;;OAQG;IACH,KAAK,CAAC,QAAQ,CAAC,UAA4B,EAAE,OAA8B;QAC1E,oEAAoE;QACpE,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,OAAO,MAAM,CAAC;YACf,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,mDAAmD;QACpD,CAAC;QAED,8BAA8B;QAC9B,IAAI,CAAC;YACJ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,UAAU,EAAE,OAAO,CAAC,CAAC;YAC9D,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;gBACvB,OAAO,MAAM,CAAC;YACf,CAAC;QACF,CAAC;QAAC,MAAM,CAAC;YACR,2DAA2D;QAC5D,CAAC;QAED,mCAAmC;QACnC,OAAO,IAAI,CAAC,cAAc,EAAE,CAAC;IAC9B,CAAC;IAED,4EAA4E;IAC5E,uBAAuB;IACvB,4EAA4E;IAE5E;;;;;;;;OAQG;IACK,KAAK,CAAC,cAAc,CAAC,UAA4B,EAAE,QAA+B;QACzF,oEAAoE;QACpE,0EAA0E;QAC1E,4EAA4E;QAC5E,MAAM,EAAE,aAAa,EAAE,2BAA2B,EAAE,GAAG,MAAM,MAAM,CAAC,yBAAyB,CAAC,CAAC;QAE/F,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAC9C,MAAM,MAAM,GAAG,IAAI,aAAa,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC;QAE7C,MAAM,QAAQ,GAAwB,MAAM,MAAM,CAAC,IAAI,CAAC,IAAI,2BAA2B,CAAC,EAAE,CAAC,CAAC,CAAC;QAC7F,MAAM,SAAS,GAAG,QAAQ,CAAC,cAAc,IAAI,EAAE,CAAC;QAEhD,OAAO,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IACrE,CAAC;IAED,4EAA4E;IAC5E,uBAAuB;IACvB,4EAA4E;IAE5E;;;;;;;;OAQG;IACK,KAAK,CAAC,cAAc,CAAC,UAA4B,EAAE,OAA8B;QACxF,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAC9C,yEAAyE;QACzE,MAAM,SAAS,GAAG,IAAI,CAAC,eAAe,CAAC,OAAO,EAAE,OAAO,EAAE,MAAM,CAAC,CAAC;QAEjE,IAAI,MAAc,CAAC;QACnB,IAAI,CAAC;YACJ,kBAAkB,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;YAErC,MAAM,GAAG,YAAY,CACpB,KAAK,EACL,CAAC,SAAS,EAAE,wBAAwB,EAAE,UAAU,EAAE,MAAM,EAAE,UAAU,EAAE,MAAM,CAAC,EAC7E;gBACC,OAAO,EAAE,SAAS;gBAClB,QAAQ,EAAE,MAAM;gBAChB,iEAAiE;gBACjE,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC;aACnC,CACD,CAAC;QACH,CAAC;QAAC,OAAO,KAAc,EAAE,CAAC;YACzB,6DAA6D;YAC7D,MAAM,IAAI,KAAK,CAAC,mBAAmB,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;QAC9F,CAAC;QAED,IAAI,MAA0B,CAAC;QAC/B,IAAI,CAAC;YACJ,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,MAAM,CAAuB,CAAC;QACnD,CAAC;QAAC,MAAM,CAAC;YACR,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;QAClD,CAAC;QAED,kBAAkB,CAAC,MAAM,EAAE,iBAAiB,CAAC,CAAC;QAC9C,OAAO,CAAC,MAAM,CAAC,cAAc,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,EAAE,KAAK,EAAE,UAAU,CAAC,CAAC,CAAC;IACzF,CAAC;IAED,4EAA4E;IAC5E,+BAA+B;IAC/B,4EAA4E;IAE5E;;;;;;OAMG;IACK,cAAc;QACrB,OAAO,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAClC,IAAI,CAAC,QAAQ,CAAC;YACb,EAAE,EAAE,KAAK,CAAC,EAAE;YACZ,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,QAAQ,EAAE,IAAI,CAAC,UAAU;YACzB,cAAc,EAAE,KAAK,CAAC,cAAc;YACpC,IAAI,EAAE,KAAK,CAAC,IAAI;YAChB,YAAY,EAAE,CAAC,GAAG,KAAK,CAAC,YAAY,CAAC;YACrC,aAAa,EAAE,CAAC;YAChB,eAAe,EAAE,CAAC;YAClB,MAAM,EAAE,QAAQ;SAChB,CAAC,CACF,CAAC;IACH,CAAC;IAED,4EAA4E;IAC5E,yBAAyB;IACzB,4EAA4E;IAE5E;;;;;;OAMG;IACK,WAAW,CAClB,KAA6B,EAC7B,MAAwB,EACxB,UAA4B;QAE5B,MAAM,IAAI,GAAG,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,CAAC;QACnC,MAAM,YAAY,GAAG,IAAI,CAAC,iBAAiB,CAAC,KAAK,CAAC,CAAC;QACnD,MAAM,cAAc,GAAG,wBAAwB,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC;QAC/D,MAAM,MAAM,GAAG,IAAI,CAAC,aAAa,CAAC,UAAU,CAAC,CAAC;QAE9C,OAAO,IAAI,CAAC,QAAQ,CAAC;YACpB,EAAE,EAAE,KAAK,CAAC,OAAO;YACjB,IAAI,EAAE,KAAK,CAAC,SAAS,IAAI,KAAK,CAAC,OAAO;YACtC,QAAQ,EAAE,IAAI,CAAC,UAAU;YACzB,cAAc;YACd,IAAI;YACJ,YAAY;YACZ,aAAa,EAAE,CAAC;YAChB,eAAe,EAAE,CAAC;YAClB,MAAM;YACN,MAAM;SACN,CAAC,CAAC;IACJ,CAAC;IAED;;;;;OAKG;IACK,SAAS,CAAC,KAA6B;QAC9C,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QAC1E,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAEvC,gDAAgD;QAChD,IAAI,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1D,OAAO,WAAW,CAAC;QACpB,CAAC;QAED,uDAAuD;QACvD,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1D,OAAO,OAAO,CAAC;QAChB,CAAC;QAED,OAAO,MAAM,CAAC;IACf,CAAC;IAED;;;;;;OAMG;IACK,iBAAiB,CAAC,KAA6B;QACtD,MAAM,KAAK,GAAG,CAAC,KAAK,CAAC,eAAe,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QACxE,MAAM,MAAM,GAAG,CAAC,KAAK,CAAC,gBAAgB,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,WAAW,EAAE,CAAC,CAAC;QAC1E,MAAM,EAAE,GAAG,KAAK,CAAC,OAAO,CAAC,WAAW,EAAE,CAAC;QAEvC,4CAA4C;QAC5C,IAAI,MAAM,CAAC,QAAQ,CAAC,WAAW,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC1D,OAAO,CAAC,WAAW,CAAC,CAAC;QACtB,CAAC;QAED,+BAA+B;QAC/B,IAAI,MAAM,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;YAC1D,OAAO,CAAC,OAAO,CAAC,CAAC;QAClB,CAAC;QAED,MAAM,YAAY,GAAa,CAAC,MAAM,CAAC,CAAC;QAExC,wCAAwC;QACxC,IAAI,KAAK,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YAC7B,YAAY,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC;QAC7B,CAAC;QAED,0EAA0E;QAC1E,IAAI,EAAE,CAAC,QAAQ,CAAC,QAAQ,CAAC,EAAE,CAAC;YAC3B,YAAY,CAAC,IAAI,CAAC,MAAM,EAAE,KAAK,EAAE,kBAAkB,CAAC,CAAC;QACtD,CAAC;QAED,kDAAkD;QAClD,IAAI,EAAE,CAAC,QAAQ,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC,QAAQ,CAAC,OAAO,CAAC,EAAE,CAAC;YACpD,YAAY,CAAC,IAAI,CAAC,kBAAkB,CAAC,CAAC;QACvC,CAAC;QAED,oEAAoE;QACpE,OAAO,YAAY,CAAC;IACrB,CAAC;IAED;;;;OAIG;IACK,aAAa,CAAC,UAA4B;QACjD,OAAO,UAAU,CAAC,QAAQ,EAAE,MAAM,IAAI,OAAO,CAAC,GAAG,CAAC,kBAAkB,IAAI,WAAW,CAAC;IACrF,CAAC;CACD;AAED,8EAA8E;AAC9E,uCAAuC;AACvC,8EAA8E;AAE9E;;;;;;;;;;;;;;;GAeG;AACH,MAAM,UAAU,wBAAwB,CAAC,OAAe;IACvD,uDAAuD;IACvD,MAAM,QAAQ,GAAG,OAAO,CAAC,OAAO,CAAC,GAAG,CAAC,CAAC;IACtC,IAAI,QAAQ,KAAK,CAAC,CAAC,EAAE,CAAC;QACrB,OAAO,SAAS,CAAC;IAClB,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,KAAK,CAAC,CAAC,EAAE,QAAQ,CAAC,CAAC,WAAW,EAAE,CAAC;IACxD,OAAO,gBAAgB,CAAC,MAAM,CAAC,IAAI,SAAS,CAAC;AAC9C,CAAC"}

package/dist/discovery/cerebras.d.ts

@@ -0,0 +1,43 @@
+/**
+ * kosha-discovery — Cerebras provider discoverer.
+ *
+ * Queries the Cerebras `/v1/models` endpoint (OpenAI-compatible),
+ * and maps results into {@link ModelCard} objects.
+ *
+ * Cerebras hosts open-source models (Llama, Mistral, DeepSeek, Qwen) on
+ * their fast inference hardware. Model IDs are flat (no vendor prefix),
+ * e.g. `llama3.1-8b`, `llama-3.3-70b`, `deepseek-r1-distill-llama-70b`.
+ *
+ * @module
+ */
+import { OpenAICompatibleDiscoverer, type OpenAICompatibleModel, type ModelClassification } from "./openai-compatible.js";
+/**
+ * Discovers models available through the Cerebras Inference API.
+ *
+ * Cerebras serves a small, curated catalog of open-source models on their
+ * custom CS-3 silicon. All models are chat/instruct models — no embedding
+ * or vision models are currently offered. Every model supports function
+ * calling via the OpenAI-compatible tool-use API.
+ *
+ * Model IDs use flat naming without a vendor prefix (e.g. `llama3.1-8b`),
+ * so origin provider is inferred from keywords in the ID.
+ */
+export declare class CerebrasDiscoverer extends OpenAICompatibleDiscoverer {
+    readonly providerId = "cerebras";
+    readonly providerName = "Cerebras";
+    readonly baseUrl = "https://api.cerebras.ai";
+    /**
+     * Keep all models — Cerebras maintains a small curated catalog and every
+     * listed model is a usable chat/instruct model. No filtering needed.
+     */
+    protected isRelevantModel(_model: OpenAICompatibleModel): boolean;
+    /**
+     * Classify a Cerebras model: infer origin from flat ID keywords,
+     * set mode to "chat", and assign standard chat capabilities.
+     *
+     * All models on Cerebras are chat/instruct models with function-calling
+     * support. Origin is inferred by matching keywords in the model ID.
+     */
+    protected classifyModel(model: OpenAICompatibleModel): ModelClassification;
+}
+//# sourceMappingURL=cerebras.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cerebras.d.ts","sourceRoot":"","sources":["../../src/discovery/cerebras.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,0BAA0B,EAAE,KAAK,qBAAqB,EAAE,KAAK,mBAAmB,EAAE,MAAM,wBAAwB,CAAC;AAe1H;;;;;;;;;;GAUG;AACH,qBAAa,kBAAmB,SAAQ,0BAA0B;IACjE,QAAQ,CAAC,UAAU,cAAc;IACjC,QAAQ,CAAC,YAAY,cAAc;IACnC,QAAQ,CAAC,OAAO,6BAA6B;IAE7C;;;OAGG;IACH,SAAS,CAAC,eAAe,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO;IAIjE;;;;;;OAMG;IACH,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,qBAAqB,GAAG,mBAAmB;CAU1E"}

package/dist/discovery/cerebras.js

@@ -0,0 +1,65 @@
+/**
+ * kosha-discovery — Cerebras provider discoverer.
+ *
+ * Queries the Cerebras `/v1/models` endpoint (OpenAI-compatible),
+ * and maps results into {@link ModelCard} objects.
+ *
+ * Cerebras hosts open-source models (Llama, Mistral, DeepSeek, Qwen) on
+ * their fast inference hardware. Model IDs are flat (no vendor prefix),
+ * e.g. `llama3.1-8b`, `llama-3.3-70b`, `deepseek-r1-distill-llama-70b`.
+ *
+ * @module
+ */
+import { OpenAICompatibleDiscoverer } from "./openai-compatible.js";
+/**
+ * Keyword rules for inferring origin provider from flat Cerebras model IDs.
+ *
+ * Order matters: put more-specific keywords first so that, e.g.,
+ * `deepseek-r1-distill-llama-70b` resolves to "deepseek" rather than "meta".
+ */
+const ORIGIN_RULES = [
+    ["deepseek", "deepseek"],
+    ["mistral", "mistral"],
+    ["llama", "meta"],
+    ["qwen", "qwen"],
+];
+/**
+ * Discovers models available through the Cerebras Inference API.
+ *
+ * Cerebras serves a small, curated catalog of open-source models on their
+ * custom CS-3 silicon. All models are chat/instruct models — no embedding
+ * or vision models are currently offered. Every model supports function
+ * calling via the OpenAI-compatible tool-use API.
+ *
+ * Model IDs use flat naming without a vendor prefix (e.g. `llama3.1-8b`),
+ * so origin provider is inferred from keywords in the ID.
+ */
+export class CerebrasDiscoverer extends OpenAICompatibleDiscoverer {
+    providerId = "cerebras";
+    providerName = "Cerebras";
+    baseUrl = "https://api.cerebras.ai";
+    /**
+     * Keep all models — Cerebras maintains a small curated catalog and every
+     * listed model is a usable chat/instruct model. No filtering needed.
+     */
+    isRelevantModel(_model) {
+        return true;
+    }
+    /**
+     * Classify a Cerebras model: infer origin from flat ID keywords,
+     * set mode to "chat", and assign standard chat capabilities.
+     *
+     * All models on Cerebras are chat/instruct models with function-calling
+     * support. Origin is inferred by matching keywords in the model ID.
+     */
+    classifyModel(model) {
+        const lower = model.id.toLowerCase();
+        const originProvider = this.inferOriginFromKeywords(lower, ORIGIN_RULES, "cerebras");
+        return {
+            originProvider,
+            mode: "chat",
+            capabilities: ["chat", "function_calling"],
+        };
+    }
+}
+//# sourceMappingURL=cerebras.js.map

package/dist/discovery/cerebras.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cerebras.js","sourceRoot":"","sources":["../../src/discovery/cerebras.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AAEH,OAAO,EAAE,0BAA0B,EAAwD,MAAM,wBAAwB,CAAC;AAE1H;;;;;GAKG;AACH,MAAM,YAAY,GAAuB;IACxC,CAAC,UAAU,EAAE,UAAU,CAAC;IACxB,CAAC,SAAS,EAAE,SAAS,CAAC;IACtB,CAAC,OAAO,EAAE,MAAM,CAAC;IACjB,CAAC,MAAM,EAAE,MAAM,CAAC;CAChB,CAAC;AAEF;;;;;;;;;;GAUG;AACH,MAAM,OAAO,kBAAmB,SAAQ,0BAA0B;IACxD,UAAU,GAAG,UAAU,CAAC;IACxB,YAAY,GAAG,UAAU,CAAC;IAC1B,OAAO,GAAG,yBAAyB,CAAC;IAE7C;;;OAGG;IACO,eAAe,CAAC,MAA6B;QACtD,OAAO,IAAI,CAAC;IACb,CAAC;IAED;;;;;;OAMG;IACO,aAAa,CAAC,KAA4B;QACnD,MAAM,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,MAAM,cAAc,GAAG,IAAI,CAAC,uBAAuB,CAAC,KAAK,EAAE,YAAY,EAAE,UAAU,CAAC,CAAC;QAErF,OAAO;YACN,cAAc;YACd,IAAI,EAAE,MAAM;YACZ,YAAY,EAAE,CAAC,MAAM,EAAE,kBAAkB,CAAC;SAC1C,CAAC;IACH,CAAC;CACD"}

package/dist/discovery/cohere.d.ts

@@ -0,0 +1,57 @@
+/**
+ * kosha-discovery — Cohere provider discoverer.
+ *
+ * Queries the Cohere OpenAI-compatible `/v1/models` endpoint,
+ * maps each model into a normalized {@link ModelCard}, and classifies
+ * the full catalog — chat, embedding, and rerank models alike.
+ *
+ * Auth: `Authorization: Bearer $CO_API_KEY`
+ * @module
+ */
+import { OpenAICompatibleDiscoverer, type ModelClassification, type OpenAICompatibleModel } from "./openai-compatible.js";
+/**
+ * Discovers models available through the Cohere API.
+ *
+ * Cohere exposes an OpenAI-compatible `/v1/models` endpoint at
+ * `https://api.cohere.com/compatibility`. All models in the catalog
+ * are Cohere's own, so `originProvider` is always `"cohere"`.
+ *
+ * Model families:
+ * - **command-r-plus / command-r** — flagship chat models with NLU capabilities
+ * - **command-light** — lighter chat model
+ * - **embed-*** — text embedding models
+ * - **rerank-*** — reranking models (classified as embedding mode)
+ */
+export declare class CohereDiscoverer extends OpenAICompatibleDiscoverer {
+    readonly providerId = "cohere";
+    readonly providerName = "Cohere";
+    readonly baseUrl = "https://api.cohere.com/compatibility";
+    /**
+     * Keep all models in the Cohere catalog.
+     *
+     * Cohere maintains a clean, intentional catalog (no reward models,
+     * no internal snapshots), so no filtering is required.
+     */
+    protected isRelevantModel(_model: OpenAICompatibleModel): boolean;
+    /**
+     * Classify a Cohere model: all are first-party, with mode and
+     * capabilities inferred from the model ID.
+     *
+     * Rules:
+     * - `originProvider` is always `"cohere"`
+     * - Embedding/rerank models → mode `"embedding"`, caps `["embedding"]`
+     * - All chat models → caps include `"function_calling"` (Cohere supports tool use)
+     * - `command-r-plus` and `command-r` additionally get `"nlu"`
+     */
+    protected classifyModel(model: OpenAICompatibleModel): ModelClassification;
+    /** Map a model ID to its primary {@link ModelMode}. */
+    private inferMode;
+    /**
+     * Infer capability flags from the lowercase model ID.
+     *
+     * - Embedding/rerank → `["embedding"]`
+     * - All chat models  → `["chat", "function_calling"]` (+ `"nlu"` for command-r variants)
+     */
+    private inferCapabilities;
+}
+//# sourceMappingURL=cohere.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.d.ts","sourceRoot":"","sources":["../../src/discovery/cohere.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAAE,0BAA0B,EAAE,KAAK,mBAAmB,EAAE,KAAK,qBAAqB,EAAE,MAAM,wBAAwB,CAAC;AAE1H;;;;;;;;;;;;GAYG;AACH,qBAAa,gBAAiB,SAAQ,0BAA0B;IAC/D,QAAQ,CAAC,UAAU,YAAY;IAC/B,QAAQ,CAAC,YAAY,YAAY;IACjC,QAAQ,CAAC,OAAO,0CAA0C;IAE1D;;;;;OAKG;IACH,SAAS,CAAC,eAAe,CAAC,MAAM,EAAE,qBAAqB,GAAG,OAAO;IAIjE;;;;;;;;;OASG;IACH,SAAS,CAAC,aAAa,CAAC,KAAK,EAAE,qBAAqB,GAAG,mBAAmB;IAY1E,uDAAuD;IACvD,OAAO,CAAC,SAAS;IAKjB;;;;;OAKG;IACH,OAAO,CAAC,iBAAiB;CAczB"}

package/dist/discovery/cohere.js

@@ -0,0 +1,82 @@
+/**
+ * kosha-discovery — Cohere provider discoverer.
+ *
+ * Queries the Cohere OpenAI-compatible `/v1/models` endpoint,
+ * maps each model into a normalized {@link ModelCard}, and classifies
+ * the full catalog — chat, embedding, and rerank models alike.
+ *
+ * Auth: `Authorization: Bearer $CO_API_KEY`
+ * @module
+ */
+import { OpenAICompatibleDiscoverer } from "./openai-compatible.js";
+/**
+ * Discovers models available through the Cohere API.
+ *
+ * Cohere exposes an OpenAI-compatible `/v1/models` endpoint at
+ * `https://api.cohere.com/compatibility`. All models in the catalog
+ * are Cohere's own, so `originProvider` is always `"cohere"`.
+ *
+ * Model families:
+ * - **command-r-plus / command-r** — flagship chat models with NLU capabilities
+ * - **command-light** — lighter chat model
+ * - **embed-*** — text embedding models
+ * - **rerank-*** — reranking models (classified as embedding mode)
+ */
+export class CohereDiscoverer extends OpenAICompatibleDiscoverer {
+    providerId = "cohere";
+    providerName = "Cohere";
+    baseUrl = "https://api.cohere.com/compatibility";
+    /**
+     * Keep all models in the Cohere catalog.
+     *
+     * Cohere maintains a clean, intentional catalog (no reward models,
+     * no internal snapshots), so no filtering is required.
+     */
+    isRelevantModel(_model) {
+        return true;
+    }
+    /**
+     * Classify a Cohere model: all are first-party, with mode and
+     * capabilities inferred from the model ID.
+     *
+     * Rules:
+     * - `originProvider` is always `"cohere"`
+     * - Embedding/rerank models → mode `"embedding"`, caps `["embedding"]`
+     * - All chat models → caps include `"function_calling"` (Cohere supports tool use)
+     * - `command-r-plus` and `command-r` additionally get `"nlu"`
+     */
+    classifyModel(model) {
+        const lower = model.id.toLowerCase();
+        const mode = this.inferMode(lower);
+        const capabilities = this.inferCapabilities(lower);
+        return {
+            originProvider: "cohere",
+            mode,
+            capabilities,
+        };
+    }
+    /** Map a model ID to its primary {@link ModelMode}. */
+    inferMode(lower) {
+        if (this.looksLikeEmbedding(lower))
+            return "embedding";
+        return "chat";
+    }
+    /**
+     * Infer capability flags from the lowercase model ID.
+     *
+     * - Embedding/rerank → `["embedding"]`
+     * - All chat models  → `["chat", "function_calling"]` (+ `"nlu"` for command-r variants)
+     */
+    inferCapabilities(lower) {
+        if (this.looksLikeEmbedding(lower)) {
+            return ["embedding"];
+        }
+        const caps = ["chat", "function_calling"];
+        // command-r-plus and command-r have advanced NLU/RAG capabilities
+        if (lower === "command-r-plus" || lower === "command-r") {
+            caps.push("nlu");
+        }
+        return caps;
+    }
+}
+//# sourceMappingURL=cohere.js.map

package/dist/discovery/cohere.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"cohere.js","sourceRoot":"","sources":["../../src/discovery/cohere.ts"],"names":[],"mappings":"AAAA;;;;;;;;;GASG;AAGH,OAAO,EAAE,0BAA0B,EAAwD,MAAM,wBAAwB,CAAC;AAE1H;;;;;;;;;;;;GAYG;AACH,MAAM,OAAO,gBAAiB,SAAQ,0BAA0B;IACtD,UAAU,GAAG,QAAQ,CAAC;IACtB,YAAY,GAAG,QAAQ,CAAC;IACxB,OAAO,GAAG,sCAAsC,CAAC;IAE1D;;;;;OAKG;IACO,eAAe,CAAC,MAA6B;QACtD,OAAO,IAAI,CAAC;IACb,CAAC;IAED;;;;;;;;;OASG;IACO,aAAa,CAAC,KAA4B;QACnD,MAAM,KAAK,GAAG,KAAK,CAAC,EAAE,CAAC,WAAW,EAAE,CAAC;QACrC,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,EAAE,QAAQ;YACxB,IAAI;YACJ,YAAY;SACZ,CAAC;IACH,CAAC;IAED,uDAAuD;IAC/C,SAAS,CAAC,KAAa;QAC9B,IAAI,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC;YAAE,OAAO,WAAW,CAAC;QACvD,OAAO,MAAM,CAAC;IACf,CAAC;IAED;;;;;OAKG;IACK,iBAAiB,CAAC,KAAa;QACtC,IAAI,IAAI,CAAC,kBAAkB,CAAC,KAAK,CAAC,EAAE,CAAC;YACpC,OAAO,CAAC,WAAW,CAAC,CAAC;QACtB,CAAC;QAED,MAAM,IAAI,GAAa,CAAC,MAAM,EAAE,kBAAkB,CAAC,CAAC;QAEpD,kEAAkE;QAClE,IAAI,KAAK,KAAK,gBAAgB,IAAI,KAAK,KAAK,WAAW,EAAE,CAAC;YACzD,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;QAClB,CAAC;QAED,OAAO,IAAI,CAAC;IACb,CAAC;CACD"}