aimodels 0.4.0 → 0.4.1

package/dist/index.d.ts

@@ -31,6 +31,12 @@ interface Organization {
     founded: number;
 }
 
+interface ProviderSource {
+    id: string;
+    apiUrl: string;
+    apiDocsUrl: string;
+    pricing: Record<string, TokenBasedPricePerMillionTokens | ImagePrice>;
+}
 /** Provider-specific data */
 interface Provider extends Organization {
     /** Provider's API endpoint URL */
@@ -191,7 +197,7 @@ declare class Model {
 }
 
 declare class ModelCollection extends Array<Model> {
-    static providersData: Record<string, Provider>;
+    static providersData: Record<string, ProviderSource>;
     static orgsData: Record<string, Organization>;
     static modelSources: Record<string, ModelSource>;
     /** Create a new ModelCollection from an array of models */
@@ -277,7 +283,7 @@ declare class AIModels extends ModelCollection {
    */
     static addStaticData({ models, providers, orgs }: {
         models?: Record<string, ModelSource>;
-        providers?: Record<string, Provider>;
+        providers?: Record<string, ProviderSource>;
         orgs?: Record<string, Organization>;
     }): void;
     static get instance(): AIModels;
@@ -294,4 +300,4 @@ declare class AIModels extends ModelCollection {
 }
 declare const models: AIModels;
 
-export { AIModels, type AudioInputContext, type AudioOutputContext, type BaseContext, type Capability, type CharacterContext, type EmbeddingContext, type ImageContext, Model, ModelCollection, type ModelContext, type ModelSource, type Organization, type Provider, type TokenContext, models };
+export { AIModels, type AudioInputContext, type AudioOutputContext, type BaseContext, type Capability, type CharacterContext, type EmbeddingContext, type ImageContext, Model, ModelCollection, type ModelContext, type ModelSource, type Organization, type Provider, type ProviderSource, type TokenContext, models };

package/dist/index.js

@@ -1628,7 +1628,7 @@ var providers = {
   },
   "azure": {
     "id": "azure",
-    "name": "Azure OpenAI",
+    "name": "Azure",
     "apiUrl": "",
     "apiDocsUrl": "https://learn.microsoft.com/en-us/azure/ai-services/openai/reference",
     "pricing": {
@@ -1870,6 +1870,10 @@ var providers = {
       }
     }
   },
+  "ollama": {
+    "id": "ollama",
+    "name": "Ollama"
+  },
   "openai": {
     "id": "openai",
     "name": "OpenAI",
@@ -2321,7 +2325,16 @@ var ModelCollection = class _ModelCollection extends Array {
   /** Get all providers from all models in the collection deduplicated */
   get providers() {
     const providerIds = [...new Set(this.flatMap((model) => model.providerIds))];
-    return providerIds.map((id) => _ModelCollection.providersData[id]).filter((p) => p !== void 0);
+    const providers2 = [];
+    for (const id of providerIds) {
+      const provider = _ModelCollection.providersData[id];
+      const organization = _ModelCollection.orgsData[id];
+      providers2.push({
+        ...organization,
+        ...provider
+      });
+    }
+    return providers2;
   }
   /** Get all orgs from all models in the collection deduplicated */
   get orgs() {
@@ -2330,7 +2343,12 @@ var ModelCollection = class _ModelCollection extends Array {
   }
   /** Get a specific provider by ID */
   getProvider(id) {
-    return _ModelCollection.providersData[id];
+    const provider = _ModelCollection.providersData[id];
+    const organization = _ModelCollection.orgsData[id];
+    return {
+      ...organization,
+      ...provider
+    };
   }
   /** Get a specific creator by ID */
   getCreator(id) {
@@ -2409,7 +2427,16 @@ var Model = class _Model {
     return this.resolveProperty("providerIds") || [];
   }
   get providers() {
-    return this.providerIds.map((id) => ModelCollection.providersData[id]).filter(Boolean);
+    const providers2 = [];
+    for (const id of this.providerIds) {
+      const provider = ModelCollection.providersData[id];
+      const organization = ModelCollection.orgsData[id];
+      providers2.push({
+        ...organization,
+        ...provider
+      });
+    }
+    return providers2;
   }
   get creatorId() {
     return this.resolveProperty("creatorId");
@@ -2513,7 +2540,10 @@ var AIModels = class _AIModels extends ModelCollection {
    * We want to return all known providers here.
    */
   get providers() {
-    return Object.values(ModelCollection.providersData);
+    return Object.values(ModelCollection.providersData).map((provider) => ({
+      ...ModelCollection.orgsData[provider.id],
+      ...provider
+    }));
   }
   /**
    * Override to return all creators directly without filtering through models.

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "aimodels",
-  "version": "0.4.0",
+  "version": "0.4.1",
   "description": "A collection of AI model specifications across different providers",
   "type": "module",
   "main": "dist/index.js",