aimodels 0.5.2 → 0.6.0

package/dist/index.d.ts

@@ -68,7 +68,7 @@ interface Provider extends Organization {
 /**
  * Defines all possible model capabilities
  */
-type Capability = "chat" | "reason" | "txt-in" | "txt-out" | "img-in" | "img-out" | "audio-in" | "audio-out" | "json-out" | "fn-out" | "vec-out";
+type Capability = "chat" | "reason" | "txt-in" | "txt-out" | "img-in" | "img-out" | "audio-in" | "audio-out" | "video-in" | "video-out" | "json-out" | "fn-out" | "vec-out";
 
 interface BaseContext {
     /** The type discriminator */

package/dist/index.js

@@ -619,13 +619,13 @@ var models = {
     "context": {
       "type": "token",
       "total": 131072,
-      "maxOutput": 8192
+      "maxOutput": 65536
     },
     "creatorId": "deepseek"
   },
-  "gemini-2.5-pro-preview-03-25": {
-    "id": "gemini-2.5-pro-preview-03-25",
-    "name": "Gemini 2.5 Pro (Preview)",
+  "gemini-3-pro-preview": {
+    "id": "gemini-3-pro-preview",
+    "name": "Gemini 3 Pro Preview",
     "license": "proprietary",
     "capabilities": [
       "chat",
@@ -635,21 +635,19 @@ var models = {
       "fn-out",
       "img-in",
       "audio-in",
+      "video-in",
       "reason"
     ],
     "context": {
       "type": "token",
-      "total": 1114112,
+      "total": 1048576,
       "maxOutput": 65536
     },
-    "aliases": [
-      "gemini-2.5"
-    ],
     "creatorId": "google"
   },
-  "gemini-2.5-pro-exp-03-25": {
-    "id": "gemini-2.5-pro-exp-03-25",
-    "name": "Gemini 2.5 Pro (Experimental)",
+  "gemini-3-flash-preview": {
+    "id": "gemini-3-flash-preview",
+    "name": "Gemini 3 Flash Preview",
     "license": "proprietary",
     "capabilities": [
       "chat",
@@ -659,240 +657,228 @@ var models = {
       "fn-out",
       "img-in",
       "audio-in",
+      "video-in",
       "reason"
     ],
     "context": {
       "type": "token",
-      "total": 1114112,
+      "total": 1048576,
       "maxOutput": 65536
     },
-    "aliases": [
-      "gemini-2.5-exp"
-    ],
     "creatorId": "google"
   },
-  "gemini-2.5-flash-preview": {
-    "id": "gemini-2.5-flash-preview",
-    "name": "Gemini 2.5 Flash Preview",
+  "gemini-2.5-pro": {
+    "id": "gemini-2.5-pro",
+    "name": "Gemini 2.5 Pro",
     "license": "proprietary",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
+      "img-in",
+      "audio-in",
+      "video-in",
       "json-out",
       "fn-out",
-      "img-in",
       "reason"
     ],
     "context": {
       "type": "token",
       "total": 1048576,
-      "maxOutput": 32768
+      "maxOutput": 65536
     },
+    "aliases": [
+      "gemini-2.5"
+    ],
     "creatorId": "google"
   },
-  "gemma-3-1b": {
-    "id": "gemma-3-1b",
-    "name": "Gemma 3 1B",
-    "license": "apache-2.0",
-    "capabilities": [
-      "chat",
-      "txt-in",
-      "txt-out",
-      "json-out",
-      "fn-out"
-    ],
-    "context": {
-      "type": "token",
-      "total": 40960,
-      "maxOutput": 8192
+  "gemini-2.5-pro-preview-06-05": {
+    "id": "gemini-2.5-pro-preview-06-05",
+    "extends": "gemini-2.5-pro",
+    "overrides": {
+      "name": "Gemini 2.5 Pro Preview 06-05"
     },
     "creatorId": "google"
   },
-  "gemma-3-4b": {
-    "id": "gemma-3-4b",
-    "name": "Gemma 3 4B",
-    "license": "apache-2.0",
+  "gemini-2.5-flash": {
+    "id": "gemini-2.5-flash",
+    "name": "Gemini 2.5 Flash",
+    "license": "proprietary",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
+      "img-in",
+      "audio-in",
+      "video-in",
       "json-out",
       "fn-out",
-      "img-in"
+      "reason"
     ],
     "context": {
       "type": "token",
-      "total": 40960,
-      "maxOutput": 8192
+      "total": 1048576,
+      "maxOutput": 65536
     },
+    "aliases": [
+      "gemini-flash-latest"
+    ],
     "creatorId": "google"
   },
-  "gemma-3-12b": {
-    "id": "gemma-3-12b",
-    "name": "Gemma 3 12B",
-    "license": "apache-2.0",
+  "gemini-2.5-flash-preview-09-2025": {
+    "id": "gemini-2.5-flash-preview-09-2025",
+    "extends": "gemini-2.5-flash",
+    "overrides": {
+      "name": "Gemini 2.5 Flash Preview 09-25"
+    },
+    "creatorId": "google"
+  },
+  "gemini-2.5-flash-lite": {
+    "id": "gemini-2.5-flash-lite",
+    "name": "Gemini 2.5 Flash Lite",
+    "license": "proprietary",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
+      "img-in",
+      "audio-in",
+      "video-in",
       "json-out",
       "fn-out",
-      "img-in"
+      "reason"
     ],
     "context": {
       "type": "token",
-      "total": 40960,
-      "maxOutput": 8192
+      "total": 1048576,
+      "maxOutput": 65536
     },
+    "aliases": [
+      "gemini-flash-lite-latest"
+    ],
     "creatorId": "google"
   },
-  "gemma-3-27b": {
-    "id": "gemma-3-27b",
-    "name": "Gemma 3 27B",
-    "license": "apache-2.0",
+  "gemini-2.5-flash-lite-preview-09-2025": {
+    "id": "gemini-2.5-flash-lite-preview-09-2025",
+    "extends": "gemini-2.5-flash-lite",
+    "overrides": {
+      "name": "Gemini 2.5 Flash Lite Preview 09-25"
+    },
+    "creatorId": "google"
+  },
+  "gemini-2.0-flash": {
+    "id": "gemini-2.0-flash",
+    "name": "Gemini 2.0 Flash",
+    "license": "proprietary",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
+      "img-in",
+      "audio-in",
+      "video-in",
       "json-out",
-      "fn-out",
-      "img-in"
+      "fn-out"
     ],
     "context": {
       "type": "token",
-      "total": 40960,
+      "total": 1048576,
       "maxOutput": 8192
     },
     "creatorId": "google"
   },
-  "text-embedding-004": {
-    "id": "text-embedding-004",
-    "name": "Text Embedding 004",
+  "gemini-2.0-flash-lite": {
+    "id": "gemini-2.0-flash-lite",
+    "name": "Gemini 2.0 Flash Lite",
     "license": "proprietary",
     "capabilities": [
+      "chat",
       "txt-in",
-      "vec-out"
+      "txt-out",
+      "img-in",
+      "audio-in",
+      "video-in",
+      "json-out",
+      "fn-out"
     ],
     "context": {
       "type": "token",
-      "total": 2048,
-      "maxOutput": 768
+      "total": 1048576,
+      "maxOutput": 8192
     },
     "creatorId": "google"
   },
-  "aqa": {
-    "id": "aqa",
-    "name": "Attributed Question Answering",
-    "license": "proprietary",
+  "gemma-3-1b": {
+    "id": "gemma-3-1b",
+    "name": "Gemma 3 1B",
+    "license": "apache-2.0",
     "capabilities": [
+      "chat",
       "txt-in",
-      "txt-out"
+      "txt-out",
+      "json-out",
+      "fn-out"
     ],
     "context": {
       "type": "token",
-      "total": 7168,
-      "maxOutput": 1024
+      "total": 40960,
+      "maxOutput": 8192
     },
     "creatorId": "google"
   },
-  "gemini-2.5-pro-preview-06-05": {
-    "id": "gemini-2.5-pro-preview-06-05",
-    "name": "Gemini 2.5 Pro Preview",
-    "license": "proprietary",
+  "gemma-3-4b": {
+    "id": "gemma-3-4b",
+    "name": "Gemma 3 4B",
+    "license": "apache-2.0",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
-      "img-in",
       "json-out",
       "fn-out",
-      "reason"
+      "img-in"
     ],
     "context": {
       "type": "token",
-      "total": 1048576,
-      "maxOutput": 65536
-    },
-    "creatorId": "google"
-  },
-  "gemini-2.5-pro-preview-05-06": {
-    "id": "gemini-2.5-pro-preview-05-06",
-    "extends": "gemini-2.5-pro-preview-06-05",
-    "overrides": {
-      "name": "Gemini 2.5 Pro Preview 05-06"
-    },
-    "creatorId": "google"
-  },
-  "gemini-2.5-flash-preview-05-20": {
-    "id": "gemini-2.5-flash-preview-05-20",
-    "extends": "gemini-2.5-flash-preview",
-    "overrides": {
-      "name": "Gemini 2.5 Flash Preview 05-20"
-    },
-    "creatorId": "google"
-  },
-  "gemini-2.5-flash-preview-04-17-thinking": {
-    "id": "gemini-2.5-flash-preview-04-17-thinking",
-    "extends": "gemini-2.5-flash-preview",
-    "overrides": {
-      "name": "Gemini 2.5 Flash Preview (Thinking)",
-      "capabilities": [
-        "chat",
-        "txt-in",
-        "txt-out",
-        "img-in",
-        "json-out",
-        "fn-out",
-        "reason"
-      ],
-      "context": {
-        "type": "token",
-        "total": 1048576,
-        "maxOutput": 65536
-      }
+      "total": 40960,
+      "maxOutput": 8192
     },
     "creatorId": "google"
   },
-  "gemini-2.5-flash-preview-tts": {
-    "id": "gemini-2.5-flash-preview-tts",
-    "name": "Gemini 2.5 Flash Preview TTS",
-    "license": "proprietary",
+  "gemma-3-12b": {
+    "id": "gemma-3-12b",
+    "name": "Gemma 3 12B",
+    "license": "apache-2.0",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
-      "audio-out"
+      "json-out",
+      "fn-out",
+      "img-in"
     ],
     "context": {
       "type": "token",
-      "total": 8192,
-      "maxOutput": 16384
-    },
-    "creatorId": "google"
-  },
-  "gemini-2.5-pro-preview-tts": {
-    "id": "gemini-2.5-pro-preview-tts",
-    "extends": "gemini-2.5-flash-preview-tts",
-    "overrides": {
-      "name": "Gemini 2.5 Pro Preview TTS"
+      "total": 40960,
+      "maxOutput": 8192
     },
     "creatorId": "google"
   },
-  "gemini-2.5-flash-preview-native-audio-dialog": {
-    "id": "gemini-2.5-flash-preview-native-audio-dialog",
-    "name": "Gemini 2.5 Flash Preview Native Audio Dialog",
-    "license": "proprietary",
+  "gemma-3-27b": {
+    "id": "gemma-3-27b",
+    "name": "Gemma 3 27B",
+    "license": "apache-2.0",
     "capabilities": [
       "chat",
       "txt-in",
       "txt-out",
-      "audio-in",
-      "audio-out"
+      "json-out",
+      "fn-out",
+      "img-in"
     ],
     "context": {
       "type": "token",
-      "total": 131072,
+      "total": 40960,
       "maxOutput": 8192
     },
     "creatorId": "google"
@@ -959,6 +945,53 @@ var models = {
     },
     "creatorId": "google"
   },
+  "text-embedding-004": {
+    "id": "text-embedding-004",
+    "name": "Text Embedding 004",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "vec-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 2048,
+      "maxOutput": 768
+    },
+    "creatorId": "google"
+  },
+  "gemini-embedding-001": {
+    "id": "gemini-embedding-001",
+    "name": "Gemini Embedding 001",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "vec-out"
+    ],
+    "context": {
+      "type": "embedding",
+      "total": 2048,
+      "unit": "tokens",
+      "dimensions": 3072,
+      "embeddingType": "text"
+    },
+    "creatorId": "google"
+  },
+  "aqa": {
+    "id": "aqa",
+    "name": "Attributed Question Answering",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "txt-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 7168,
+      "maxOutput": 1024
+    },
+    "creatorId": "google"
+  },
   "gemini-2.5-flash-image": {
     "id": "gemini-2.5-flash-image",
     "name": "Gemini 2.5 Flash Image (Nano Banana)",
@@ -998,12 +1031,110 @@ var models = {
     },
     "creatorId": "google"
   },
-  "veo-2.0-generate-001": {
-    "id": "veo-2.0-generate-001",
-    "name": "Veo 2",
+  "imagen-4.0-generate-001": {
+    "id": "imagen-4.0-generate-001",
+    "name": "Imagen 4.0",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "img-out"
+    ],
+    "context": {
+      "maxOutput": 4,
+      "sizes": [
+        "1024x1024",
+        "896x1280",
+        "1280x896",
+        "768x1408",
+        "1408x768",
+        "2048x2048",
+        "1792x2560",
+        "2560x1792",
+        "1536x2816",
+        "2816x1536"
+      ]
+    },
+    "creatorId": "google"
+  },
+  "imagen-4.0-ultra-generate-001": {
+    "id": "imagen-4.0-ultra-generate-001",
+    "name": "Imagen 4.0 Ultra",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "img-out"
+    ],
+    "context": {
+      "maxOutput": 4,
+      "sizes": [
+        "1024x1024",
+        "896x1280",
+        "1280x896",
+        "768x1408",
+        "1408x768",
+        "2048x2048",
+        "1792x2560",
+        "2560x1792",
+        "1536x2816",
+        "2816x1536"
+      ]
+    },
+    "creatorId": "google"
+  },
+  "veo-2.0-generate-001": {
+    "id": "veo-2.0-generate-001",
+    "name": "Veo 2",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "video-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 480,
+      "maxOutput": 8192
+    },
+    "creatorId": "google"
+  },
+  "veo-3.0-generate-preview": {
+    "id": "veo-3.0-generate-preview",
+    "name": "Veo 3.0 Generate",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "img-in",
+      "video-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 480,
+      "maxOutput": 8192
+    },
+    "creatorId": "google"
+  },
+  "veo-3.1-generate-preview": {
+    "id": "veo-3.1-generate-preview",
+    "name": "Veo 3.1 Generate",
+    "license": "proprietary",
+    "capabilities": [
+      "txt-in",
+      "img-in",
+      "video-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 480,
+      "maxOutput": 8192
+    },
+    "creatorId": "google"
+  },
+  "veo-3.1-fast-generate-preview": {
+    "id": "veo-3.1-fast-generate-preview",
+    "name": "Veo 3.1 Fast Generate",
     "license": "proprietary",
     "capabilities": [
       "txt-in",
+      "img-in",
       "video-out"
     ],
     "context": {
@@ -1147,6 +1278,151 @@ var models = {
     },
     "creatorId": "meta"
   },
+  "llama-3.1-8b-instant": {
+    "id": "llama-3.1-8b-instant",
+    "name": "Llama 3.1 8B",
+    "license": "llama-3-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "json-out",
+      "fn-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 131072
+    },
+    "aliases": [
+      "llama-3.1-8b"
+    ],
+    "creatorId": "meta"
+  },
+  "llama-3.3-70b-versatile": {
+    "id": "llama-3.3-70b-versatile",
+    "name": "Llama 3.3 70B",
+    "license": "llama-3-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "json-out",
+      "fn-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 32768
+    },
+    "aliases": [
+      "llama-3.3-70b"
+    ],
+    "creatorId": "meta"
+  },
+  "meta-llama/llama-guard-4-12b": {
+    "id": "meta-llama/llama-guard-4-12b",
+    "name": "LlamaGuard 4 12B",
+    "license": "llama-4-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 1024
+    },
+    "aliases": [
+      "llama-guard-4-12b"
+    ],
+    "creatorId": "meta"
+  },
+  "meta-llama/llama-4-maverick-17b-128e-instruct": {
+    "id": "meta-llama/llama-4-maverick-17b-128e-instruct",
+    "name": "Llama 4 Maverick 17B 128E",
+    "license": "llama-4-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "json-out",
+      "fn-out",
+      "img-in",
+      "reason"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 8192
+    },
+    "aliases": [
+      "llama-4-maverick-17b-128e"
+    ],
+    "creatorId": "meta"
+  },
+  "meta-llama/llama-4-scout-17b-16e-instruct": {
+    "id": "meta-llama/llama-4-scout-17b-16e-instruct",
+    "name": "Llama 4 Scout 17B 16E",
+    "license": "llama-4-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "json-out",
+      "fn-out",
+      "img-in",
+      "reason"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 8192
+    },
+    "aliases": [
+      "llama-4-scout-17b-16e"
+    ],
+    "creatorId": "meta"
+  },
+  "meta-llama/llama-prompt-guard-2-22m": {
+    "id": "meta-llama/llama-prompt-guard-2-22m",
+    "name": "Llama Prompt Guard 2 22M",
+    "license": "llama-3-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 512,
+      "maxOutput": 512
+    },
+    "aliases": [
+      "llama-prompt-guard-2-22m"
+    ],
+    "creatorId": "meta"
+  },
+  "meta-llama/llama-prompt-guard-2-86m": {
+    "id": "meta-llama/llama-prompt-guard-2-86m",
+    "name": "Llama Prompt Guard 2 86M",
+    "license": "llama-3-community",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 512,
+      "maxOutput": 512
+    },
+    "aliases": [
+      "llama-prompt-guard-2-86m"
+    ],
+    "creatorId": "meta"
+  },
   "mistral-medium-3-2025-05-07": {
     "id": "mistral-medium-3-2025-05-07",
     "name": "Mistral Medium 3",
@@ -1381,6 +1657,42 @@ var models = {
     ],
     "creatorId": "openai"
   },
+  "gpt-oss-120b": {
+    "id": "gpt-oss-120b",
+    "name": "GPT OSS 120B",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "fn-out",
+      "reason"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 65536
+    },
+    "creatorId": "openai"
+  },
+  "gpt-oss-20b": {
+    "id": "gpt-oss-20b",
+    "name": "GPT OSS 20B",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "fn-out",
+      "reason"
+    ],
+    "context": {
+      "type": "token",
+      "total": 131072,
+      "maxOutput": 65536
+    },
+    "creatorId": "openai"
+  },
   "whisper-1": {
     "id": "whisper-1",
     "name": "Whisper",
@@ -1939,6 +2251,113 @@ var models = {
       "grok-4-heavy-latest"
     ],
     "creatorId": "xai"
+  },
+  "grok-code-fast-1": {
+    "id": "grok-code-fast-1",
+    "name": "Grok Code Fast 1",
+    "creator": "xai",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "reason",
+      "fn-out",
+      "json-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 256e3,
+      "maxOutput": 1e4
+    },
+    "releasedAt": "2025-08-28",
+    "creatorId": "xai"
+  },
+  "grok-4-fast": {
+    "id": "grok-4-fast",
+    "name": "Grok 4 Fast",
+    "creator": "xai",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "img-in",
+      "reason",
+      "fn-out",
+      "json-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 2e6,
+      "maxOutput": 3e4
+    },
+    "releasedAt": "2025-09-19",
+    "creatorId": "xai"
+  },
+  "grok-4-fast-non-reasoning": {
+    "id": "grok-4-fast-non-reasoning",
+    "name": "Grok 4 Fast (Non-Reasoning)",
+    "creator": "xai",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "img-in",
+      "fn-out",
+      "json-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 2e6,
+      "maxOutput": 3e4
+    },
+    "releasedAt": "2025-09-19",
+    "creatorId": "xai"
+  },
+  "grok-4-1-fast": {
+    "id": "grok-4-1-fast",
+    "name": "Grok 4.1 Fast",
+    "creator": "xai",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "img-in",
+      "reason",
+      "fn-out",
+      "json-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 2e6,
+      "maxOutput": 3e4
+    },
+    "releasedAt": "2025-11-19",
+    "creatorId": "xai"
+  },
+  "grok-4-1-fast-non-reasoning": {
+    "id": "grok-4-1-fast-non-reasoning",
+    "name": "Grok 4.1 Fast (Non-Reasoning)",
+    "creator": "xai",
+    "license": "proprietary",
+    "capabilities": [
+      "chat",
+      "txt-in",
+      "txt-out",
+      "img-in",
+      "fn-out",
+      "json-out"
+    ],
+    "context": {
+      "type": "token",
+      "total": 2e6,
+      "maxOutput": 3e4
+    },
+    "releasedAt": "2025-11-19",
+    "creatorId": "xai"
   }
 };
 var providers = {
@@ -1996,7 +2415,32 @@ var providers = {
     "name": "Groq",
     "apiUrl": "https://api.groq.com/openai/v1",
     "apiDocsUrl": "https://console.groq.com/docs/api-reference",
-    "pricing": {}
+    "pricing": {},
+    "models": [
+      {
+        "creator": "meta",
+        "include": [
+          "llama3-8b-8192",
+          "llama3-70b-8192",
+          "llama-guard-3-8b",
+          "llama-3.1-8b-instant",
+          "llama-3.3-70b-versatile",
+          "meta-llama/llama-guard-4-12b",
+          "meta-llama/llama-4-maverick-17b-128e-instruct",
+          "meta-llama/llama-4-scout-17b-16e-instruct",
+          "meta-llama/llama-prompt-guard-2-22m",
+          "meta-llama/llama-prompt-guard-2-86m"
+        ]
+      },
+      {
+        "creator": "openai",
+        "include": [
+          "gpt-oss-120b",
+          "gpt-oss-20b",
+          "whisper-1"
+        ]
+      }
+    ]
   },
   "mistral": {
     "id": "mistral",
@@ -2014,7 +2458,17 @@ var providers = {
     "name": "OpenAI",
     "apiUrl": "https://api.openai.com/v1",
     "apiDocsUrl": "https://platform.openai.com/docs/api-reference",
-    "pricing": {}
+    "pricing": {},
+    "models": [
+      {
+        "creator": "openai",
+        "include": "all",
+        "exclude": [
+          "gpt-oss-120b",
+          "gpt-oss-20b"
+        ]
+      }
+    ]
   },
   "openrouter": {
     "id": "openrouter",
@@ -2366,21 +2820,33 @@ var Model = class _Model {
   get providerIds() {
     const ids = /* @__PURE__ */ new Set();
     const creatorId = this.creatorId;
-    if (creatorId && ModelCollection.providersData[creatorId]) {
-      ids.add(creatorId);
-    }
+    if (!creatorId) return [];
+    const includesModel = (entry) => {
+      if (entry.creator !== creatorId) return false;
+      if (entry.include === "all") {
+        return !entry.exclude || !entry.exclude.includes(this.id);
+      }
+      return Array.isArray(entry.include) && entry.include.includes(this.id);
+    };
     for (const [providerId, provider] of Object.entries(ModelCollection.providersData)) {
+      const isNative = providerId === creatorId;
       const entries = provider.models;
-      if (!entries) continue;
-      for (const entry of entries) {
-        if (entry.creator !== creatorId) continue;
-        if (entry.include === "all") {
-          if (!entry.exclude || !entry.exclude.includes(this.id)) {
-            ids.add(providerId);
-          }
-        } else if (Array.isArray(entry.include) && entry.include.includes(this.id)) {
+      if (isNative) {
+        if (!entries || entries.length === 0) {
+          ids.add(providerId);
+          continue;
+        }
+        const creatorEntry = entries.find((e) => e.creator === creatorId);
+        if (!creatorEntry) {
+          ids.add(providerId);
+        } else if (includesModel(creatorEntry)) {
           ids.add(providerId);
         }
+        continue;
+      }
+      if (!entries) continue;
+      if (entries.some(includesModel)) {
+        ids.add(providerId);
       }
     }
     return Array.from(ids);

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimodels",
-  "version": "0.5.2",
+  "version": "0.6.0",
   "description": "A collection of AI model specifications across different providers",
   "type": "module",
   "main": "dist/index.js",
@@ -23,8 +23,9 @@
     "validate:data": "tsx src/schemas/validate-data.ts",
     "prebuild": "npm run validate:data && npm run clean && npm run gen-data-js-file",
     "build:ts": "tsup",
-    "build": "npm run build:ts",
+    "build": "npm run build:ts && npm test",
     "test": "vitest run",
+    "test:all": "vitest run --include tests/**/*.test.ts",
     "test:watch": "vitest watch",
     "test:coverage": "vitest run --coverage",
     "typecheck": "tsc --noEmit",