aimodels 0.1.1 → 0.2.0

package/dist/index.js

@@ -20,91 +20,602 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  creators: () => creators_default,
   models: () => models
 });
 module.exports = __toCommonJS(index_exports);
 
-// src/data/models.json
-var models_default = [
-  {
-    id: "gpt-4",
-    name: "GPT-4",
-    provider: "openai",
-    contextWindow: 8192,
-    capabilities: ["chat", "completion"],
-    pricing: {
-      input: 0.03,
-      output: 0.06
-    },
-    released: "2023-03-14",
-    parameters: 11e11
-  },
-  {
-    id: "gpt-3.5-turbo",
-    name: "GPT-3.5 Turbo",
-    provider: "openai",
-    contextWindow: 4096,
-    capabilities: ["chat", "completion"],
-    pricing: {
-      input: 15e-4,
-      output: 2e-3
-    },
-    released: "2022-11-30"
-  },
-  {
-    id: "claude-2.1",
-    name: "Claude 2.1",
-    provider: "anthropic",
-    contextWindow: 2e5,
-    capabilities: ["chat", "completion"],
-    pricing: {
-      input: 8e-3,
-      output: 0.024
-    },
-    released: "2023-11-21"
-  },
-  {
-    id: "llama-2-70b",
-    name: "Llama 2 70B",
-    provider: "meta",
-    contextWindow: 4096,
-    capabilities: ["chat", "completion"],
-    pricing: {
-      input: 0,
-      output: 0
-    },
-    released: "2023-07-18",
-    license: "Llama 2 Community License",
-    parameters: 7e10
-  }
-];
+// src/types/models.ts
+var ModelCollection = class _ModelCollection extends Array {
+  constructor(models2) {
+    super(...models2);
+    Object.setPrototypeOf(this, _ModelCollection.prototype);
+  }
+  can(...capabilities) {
+    return new _ModelCollection(
+      this.filter((model) => capabilities.every((cap) => model.can.includes(cap)))
+    );
+  }
+  know(...languages) {
+    return new _ModelCollection(
+      this.filter((model) => languages.every((lang) => model.languages?.includes(lang)))
+    );
+  }
+  // Override array methods to return ModelCollection
+  filter(predicate) {
+    return new _ModelCollection(super.filter(predicate));
+  }
+  slice(start, end) {
+    return new _ModelCollection(super.slice(start, end));
+  }
+};
+
+// src/data/models/openai.json
+var openai_default = {
+  models: [
+    {
+      id: "whisper-1",
+      name: "Whisper-1",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "sound-in",
+        "text-out"
+      ],
+      context: {
+        total: null,
+        maxOutput: null
+      }
+    },
+    {
+      id: "tts-1",
+      name: "TTS-1",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "text-in",
+        "sound-out"
+      ],
+      context: {
+        total: null,
+        maxOutput: null
+      }
+    },
+    {
+      id: "tts-1-hd",
+      name: "TTS-1 HD",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "text-in",
+        "sound-out"
+      ],
+      context: {
+        total: null,
+        maxOutput: null
+      }
+    },
+    {
+      id: "gpt-4o",
+      name: "GPT-4O",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        total: 128e3,
+        maxOutput: 16384
+      }
+    },
+    {
+      id: "gpt-4o-mini",
+      name: "GPT-4O Mini",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "img-in",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        total: 128e3,
+        maxOutput: 16384
+      }
+    },
+    {
+      id: "o1",
+      name: "OpenAI O1",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "img-in",
+        "json-out",
+        "function-out",
+        "reason"
+      ],
+      context: {
+        total: 2e5,
+        maxOutput: 1e5
+      }
+    },
+    {
+      id: "o1-mini",
+      name: "OpenAI O1 Mini",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "json-out",
+        "function-out",
+        "reason"
+      ],
+      context: {
+        total: 128e3,
+        maxOutput: 65536
+      }
+    },
+    {
+      id: "o3-mini",
+      name: "OpenAI O3 Mini",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "json-out",
+        "function-out",
+        "reason"
+      ],
+      context: {
+        total: 2e5,
+        maxOutput: 1e5
+      }
+    },
+    {
+      id: "gpt-4o-audio-preview",
+      name: "GPT-4O Audio",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "sound-in",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        total: 128e3,
+        maxOutput: 16384
+      }
+    },
+    {
+      id: "gpt-4o-realtime-preview",
+      name: "GPT-4O Realtime",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "chat",
+        "sound-in",
+        "json-out",
+        "function-out"
+      ],
+      context: {
+        total: 128e3,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "dall-e-3",
+      name: "DALL-E 3",
+      license: "proprietary",
+      providers: [
+        "openai",
+        "azure"
+      ],
+      can: [
+        "img-out"
+      ],
+      context: {
+        maxOutput: 1,
+        sizes: [
+          "1024x1024",
+          "1024x1792",
+          "1792x1024"
+        ],
+        qualities: [
+          "standard",
+          "hd"
+        ]
+      }
+    }
+  ]
+};
+
+// src/data/models/anthropic.json
+var anthropic_default = {
+  models: [
+    {
+      id: "claude-3-opus",
+      name: "Claude 3 Opus",
+      license: "proprietary",
+      providers: ["anthropic", "aws", "google"],
+      can: ["chat", "img-in", "json-out", "function-out"],
+      context: {
+        total: 2e5,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "claude-3-sonnet",
+      name: "Claude 3 Sonnet",
+      license: "proprietary",
+      providers: ["anthropic", "aws", "google"],
+      can: ["chat", "img-in", "json-out", "function-out"],
+      context: {
+        total: 2e5,
+        maxOutput: 4096
+      }
+    },
+    {
+      id: "claude-3-haiku",
+      name: "Claude 3 Haiku",
+      license: "proprietary",
+      providers: ["anthropic", "aws", "google"],
+      can: ["chat", "img-in", "json-out", "function-out"],
+      context: {
+        total: 2e5,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
+// src/data/models/meta.json
+var meta_default = {
+  models: [
+    {
+      id: "llama2-70b-4096",
+      name: "Llama 2 70B",
+      license: "llama-2-community",
+      providers: ["groq", "ollama"],
+      can: ["chat", "json-out", "function-out"],
+      context: {
+        total: 4096,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
+// src/data/models/mistral.json
+var mistral_default = {
+  models: [
+    {
+      id: "mixtral-8x7b-32768",
+      name: "Mixtral 8x7B",
+      license: "apache-2.0",
+      providers: ["groq"],
+      can: ["chat", "json-out", "function-out"],
+      context: {
+        total: 32768,
+        maxOutput: 4096
+      }
+    }
+  ]
+};
+
+// src/builders/models.ts
+function buildAllModels() {
+  return [
+    ...openai_default.models,
+    ...anthropic_default.models,
+    ...meta_default.models,
+    ...mistral_default.models
+  ];
+}
+
+// src/data/providers/openai.json
+var openai_default2 = {
+  id: "openai",
+  name: "OpenAI",
+  websiteUrl: "https://openai.com/",
+  apiUrl: "https://api.openai.com/v1",
+  defaultModel: "gpt-4o",
+  models: {
+    "gpt-4o": {
+      type: "token",
+      input: 20,
+      output: 80
+    },
+    "gpt-3.5-turbo": {
+      type: "token",
+      input: 5,
+      output: 20
+    },
+    "gpt-4o-vision": {
+      type: "token",
+      input: 15,
+      input_cached: 7.5,
+      output: 60
+    },
+    "o1-mini": {
+      type: "token",
+      input: 3,
+      input_cached: 1.5,
+      output: 15
+    },
+    "o3-mini": {
+      type: "token",
+      input: 3,
+      input_cached: 1.5,
+      output: 15
+    },
+    "gpt-4o-audio-preview": {
+      type: "token",
+      input: 20,
+      output: 80
+    },
+    "gpt-4o-vision-preview": {
+      type: "token",
+      input: 20,
+      output: 80
+    },
+    "whisper-1": {
+      type: "minute",
+      price: 6e-3
+    },
+    "tts-1": {
+      type: "character",
+      price: 15e-6
+    },
+    "tts-1-hd": {
+      type: "character",
+      price: 3e-5
+    },
+    "dall-e-2": {
+      type: "image",
+      price: 0.016,
+      size: "256x256",
+      unit: "per_image"
+    },
+    "dall-e-2-512": {
+      type: "image",
+      price: 0.018,
+      size: "512x512",
+      unit: "per_image"
+    },
+    "dall-e-2-1024": {
+      type: "image",
+      price: 0.02,
+      size: "1024x1024",
+      unit: "per_image"
+    },
+    "dall-e-3": {
+      type: "image",
+      price: 0.04,
+      size: "1024x1024",
+      unit: "per_image"
+    },
+    "dall-e-3-hd": {
+      type: "image",
+      price: 0.08,
+      size: "1024x1024",
+      unit: "per_image"
+    }
+  }
+};
+
+// src/data/providers/anthropic.json
+var anthropic_default2 = {
+  id: "anthropic",
+  name: "Anthropic",
+  websiteUrl: "https://www.anthropic.com/",
+  apiUrl: "https://api.anthropic.com/v1",
+  defaultModel: "claude-3-sonnet",
+  models: {
+    "claude-3-opus": {
+      type: "token",
+      input: 15,
+      output: 75
+    },
+    "claude-3-sonnet": {
+      type: "token",
+      input: 3,
+      output: 15
+    },
+    "claude-3-haiku": {
+      type: "token",
+      input: 2.5,
+      output: 12.5
+    }
+  }
+};
+
+// src/data/providers/mistral.json
+var mistral_default2 = {
+  id: "mistral",
+  name: "Mistral",
+  websiteUrl: "https://mistral.ai/",
+  apiUrl: "https://api.mistral.ai/v1",
+  defaultModel: "mixtral-8x7b-32768",
+  models: {
+    "mixtral-8x7b-32768": {
+      type: "token",
+      input: 0.7,
+      output: 0.7
+    },
+    "mistral-large-latest": {
+      type: "token",
+      input: 0.7,
+      output: 0.7
+    }
+  }
+};
+
+// src/builders/providers.ts
+function isTokenPrice(price) {
+  return typeof price === "object" && price !== null && "type" in price && price.type === "token" && "input" in price && typeof price.input === "number" && "output" in price && typeof price.output === "number";
+}
+function isImagePrice(price) {
+  return typeof price === "object" && price !== null && "type" in price && price.type === "image" && "price" in price && typeof price.price === "number" && "size" in price && typeof price.size === "string" && "unit" in price && price.unit === "per_image";
+}
+function isCharacterPrice(price) {
+  return typeof price === "object" && price !== null && "type" in price && price.type === "character" && "price" in price && typeof price.price === "number";
+}
+function isMinutePrice(price) {
+  return typeof price === "object" && price !== null && "type" in price && price.type === "minute" && "price" in price && typeof price.price === "number";
+}
+function validateProvider(raw) {
+  if (typeof raw !== "object" || raw === null) {
+    throw new Error("Provider data must be an object");
+  }
+  const provider = raw;
+  if (typeof provider.id !== "string") {
+    throw new Error("Provider id must be a string");
+  }
+  if (typeof provider.name !== "string") {
+    throw new Error("Provider name must be a string");
+  }
+  if (typeof provider.websiteUrl !== "string") {
+    throw new Error("Provider websiteUrl must be a string");
+  }
+  if (typeof provider.apiUrl !== "string") {
+    throw new Error("Provider apiUrl must be a string");
+  }
+  if (typeof provider.models !== "object" || provider.models === null) {
+    throw new Error("Provider models must be an object");
+  }
+  const models2 = provider.models;
+  Object.values(models2).forEach((price) => {
+    if (!isTokenPrice(price) && !isImagePrice(price) && !isCharacterPrice(price) && !isMinutePrice(price)) {
+      throw new Error(`Invalid price data: ${JSON.stringify(price)}`);
+    }
+  });
+  return {
+    id: provider.id,
+    name: provider.name,
+    websiteUrl: provider.websiteUrl,
+    apiUrl: provider.apiUrl,
+    models: provider.models
+  };
+}
+function buildAllProviders() {
+  return [
+    validateProvider(openai_default2),
+    validateProvider(anthropic_default2),
+    validateProvider(mistral_default2)
+  ];
+}
+function buildProvidersData() {
+  return {
+    providers: buildAllProviders()
+  };
+}
+
+// src/data/creators.json
+var creators_default = {
+  creators: {
+    openai: {
+      name: "OpenAI",
+      website: "https://openai.com"
+    },
+    anthropic: {
+      name: "Anthropic",
+      website: "https://anthropic.com"
+    },
+    meta: {
+      name: "Meta",
+      website: "https://ai.meta.com"
+    },
+    mistral: {
+      name: "Mistral AI",
+      website: "https://mistral.ai"
+    }
+  }
+};
 
 // src/index.ts
-var aimodels = models_default;
+var allModels = buildAllModels();
+var providersData = buildProvidersData();
 var models = {
-  all: aimodels,
+  all: new ModelCollection(allModels),
+  get creators() {
+    return Object.keys(creators_default.creators);
+  },
   get providers() {
-    return [...new Set(aimodels.map((model) => model.provider))];
+    return providersData.providers.map((p) => p.id);
   },
-  from(provider) {
-    return aimodels.filter((model) => model.provider === provider);
+  fromCreator(creator) {
+    return new ModelCollection(
+      allModels.filter(
+        (model) => model.license.startsWith(creator) || // For open source models
+        providersData.providers.find((p) => p.id === creator)?.models[model.id]
+        // For proprietary models
+      )
+    );
+  },
+  fromProvider(provider) {
+    return new ModelCollection(
+      allModels.filter((model) => model.providers.includes(provider))
+    );
   },
   find(id) {
-    return aimodels.find((model) => model.id === id);
+    return allModels.find((model) => model.id === id);
   },
-  withCapability(capability) {
-    return aimodels.filter((model) => model.capabilities.includes(capability));
+  can(...capabilities) {
+    return new ModelCollection(
+      allModels.filter(
+        (model) => capabilities.every((capability) => model.can.includes(capability))
+      )
+    );
   },
   withMinContext(tokens) {
-    return aimodels.filter((model) => model.contextWindow >= tokens);
-  },
-  withMaxPrice(price) {
-    return aimodels.filter(
-      (model) => model.pricing.input <= price && model.pricing.output <= price
+    return new ModelCollection(
+      allModels.filter((model) => model.context.total >= tokens)
     );
+  },
+  getPrice(modelId, provider) {
+    const providerData = providersData.providers.find((p) => p.id === provider);
+    const price = providerData?.models[modelId];
+    return price?.type === "token" ? price : void 0;
   }
 };
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  creators,
   models
 });