cascade-ai 0.47.0 → 0.48.0

package/dist/index.js

@@ -193,7 +193,7 @@ var init_audit_logger = __esm({
 });
 
 // src/constants.ts
-var CASCADE_VERSION = "0.47.0";
+var CASCADE_VERSION = "0.48.0";
 var CASCADE_CONFIG_DIR = ".cascade";
 var CASCADE_MD_FILE = "CASCADE.md";
 var CASCADE_IGNORE_FILE = ".cascadeignore";
@@ -675,6 +675,55 @@ var BaseProvider = class {
   }
 };
 
+// src/providers/model-filter.ts
+var NON_CHAT_PATTERNS = [
+  "embed",
+  // text-embedding-3, nomic-embed, mxbai-embed, snowflake-arctic-embed
+  "(?:^|[-_/])tts(?:[-_/]|$)",
+  "text-to-speech",
+  "speech",
+  // text-to-speech
+  "whisper",
+  "transcrib",
+  "(?:^|[-_/])stt(?:[-_/]|$)",
+  // speech-to-text
+  "dall[-]?e",
+  "imagen",
+  "image-generation",
+  "gpt-image",
+  "(?:^|[-_/])image(?:[-_/]|$)",
+  "stable-diffusion",
+  "sdxl",
+  "flux",
+  // image gen
+  "moderation",
+  "(?:^|[-_/])guard(?:[-_/]|$)",
+  // moderation / safety
+  "rerank",
+  "(?:^|[-_/])audio(?:[-_/]|$)",
+  "native-audio",
+  "realtime",
+  // audio / realtime
+  "veo",
+  "sora",
+  "(?:^|[-_/])video(?:[-_/]|$)",
+  // video
+  "(?:^|[-_/])aqa(?:[-_/]|$)",
+  // attributed QA
+  "(?:^|[-_/])live(?:[-_/]|$)",
+  // live-audio variants
+  "(?:^|[-_/])(?:davinci|babbage)(?:[-_/]|$)"
+  // legacy completion-only base models
+];
+var NON_CHAT = new RegExp(NON_CHAT_PATTERNS.join("|"), "i");
+function isChatModel(id, methods) {
+  if (methods && methods.length > 0) {
+    const canGenerate = methods.some((m) => /generatecontent|generatemessage|chat|completion/i.test(m));
+    if (!canGenerate) return false;
+  }
+  return !NON_CHAT.test(id);
+}
+
 // src/providers/anthropic.ts
 function anthropicThinkingParam(modelId, maxTokens) {
   if (!/claude-(opus|sonnet)-4/i.test(modelId)) return {};
@@ -788,7 +837,7 @@ var AnthropicProvider = class extends BaseProvider {
       if (!Array.isArray(data?.data)) {
         return Object.values(MODELS).filter((m) => m.provider === "anthropic");
       }
-      return data.data.map((m) => {
+      return data.data.filter((m) => isChatModel(m.id)).map((m) => {
         const known = Object.values(MODELS).find((km) => km.id === m.id && km.provider === "anthropic");
         if (known) return known;
         return {
@@ -1003,7 +1052,7 @@ var OpenAIProvider = class extends BaseProvider {
   async listModels() {
     try {
       const response = await this.client.models.list();
-      return response.data.map((m) => {
+      return response.data.filter((m) => isChatModel(m.id)).map((m) => {
         const known = Object.values(MODELS).find((km) => km.id === m.id && km.provider === "openai");
         if (known) return known;
         return {
@@ -1278,7 +1327,7 @@ var GeminiProvider = class extends BaseProvider {
       if (!Array.isArray(data?.models)) {
         return Object.values(MODELS).filter((m) => m.provider === "gemini");
       }
-      return data.models.map((m) => {
+      return data.models.filter((m) => isChatModel(m.name.replace("models/", ""), m.supportedGenerationMethods)).map((m) => {
         const id = m.name.replace("models/", "");
         const known = Object.values(MODELS).find(
           (km) => km.id === id && km.provider === "gemini"
@@ -1636,10 +1685,10 @@ var OllamaProvider = class extends BaseProvider {
       const response = await fetch(`${this.baseUrl}/api/tags`);
       if (!response.ok) return [];
       const data = await response.json();
-      const supportedKeywords = ["llama3", "llama2", "gemma", "mistral", "mixtral", "qwen", "phi3", "codellama", "deepseek", "llava", "starcoder", "stable-code", "nomic-embed"];
+      const supportedKeywords = ["llama3", "llama2", "gemma", "mistral", "mixtral", "qwen", "phi3", "codellama", "deepseek", "llava", "starcoder", "stable-code"];
       const entries = data.models.filter((m) => {
         const name = m.name.toLowerCase();
-        return supportedKeywords.some((k) => name.includes(k));
+        return supportedKeywords.some((k) => name.includes(k)) && isChatModel(name);
       });
       const shows = await Promise.all(entries.map((m) => this.showModel(m.name)));
       return entries.map((m, i) => {
@@ -1759,7 +1808,8 @@ var OpenAICompatibleProvider = class extends OpenAIProvider {
       return typeof m === "string" ? m : void 0;
     }).filter((x) => typeof x === "string" && x.length > 0);
     if (ids.length === 0) return [this.model];
-    return ids.map((id) => ({
+    const chatIds = ids.filter((id) => isChatModel(id));
+    return (chatIds.length ? chatIds : ids).map((id) => ({
       id,
       name: id,
       provider: "openai-compatible",