@pydantic/genai-prices 0.0.6 → 0.0.7

package/README.md

@@ -174,7 +174,7 @@ src/
 ### Common Issues
 
 - **No price found (returns null)**:
-  - Make sure you specify the correct `providerId` (e.g., `openai`)
+  - Make sure you specify the correct `providerId` (e.g., `openai`, `google`, `anthropic`)
   - Try using `provider:model` format in CLI
   - Use `--auto-update` flag to fetch latest data
   - Check that the model name is correct and supported by the provider

package/dist/__tests__/matcher.test.d.ts

@@ -0,0 +1 @@
+export {};

package/dist/data.js

@@ -1292,6 +1292,16 @@ export const data = [
         model_match: {
             contains: 'gemini',
         },
+        provider_match: {
+            or: [
+                {
+                    contains: 'google',
+                },
+                {
+                    equals: 'gemini',
+                },
+            ],
+        },
         models: [
             {
                 id: 'claude-3-5-haiku',
@@ -1973,6 +1983,9 @@ export const data = [
         model_match: {
             regex: '(?:mi|code|dev|magi|mini)stral',
         },
+        provider_match: {
+            starts_with: 'mistral',
+        },
         models: [
             {
                 id: 'codestral',

package/dist/index.umd.cjs

@@ -1298,6 +1298,16 @@
       model_match: {
         contains: "gemini"
       },
+      provider_match: {
+        or: [
+          {
+            contains: "google"
+          },
+          {
+            equals: "gemini"
+          }
+        ]
+      },
       models: [
         {
           id: "claude-3-5-haiku",
@@ -1979,6 +1989,9 @@
       model_match: {
         regex: "(?:mi|code|dev|magi|mini)stral"
       },
+      provider_match: {
+        starts_with: "mistral"
+      },
       models: [
         {
           id: "codestral",
@@ -10573,9 +10586,15 @@
     }
     return false;
   }
+  function findProviderById(providers, providerId) {
+    const normalizedProviderId = providerId.toLowerCase().trim();
+    const exactMatch = providers.find((p) => p.id === normalizedProviderId);
+    if (exactMatch) return exactMatch;
+    return providers.find((p) => p.provider_match && matchLogic(p.provider_match, normalizedProviderId));
+  }
   function matchProvider(providers, modelRef, providerId, providerApiUrl) {
     if (providerId) {
-      return providers.find((p) => p.id === providerId);
+      return findProviderById(providers, providerId);
     }
     if (providerApiUrl) {
       return providers.find((p) => new RegExp(p.api_pattern).test(providerApiUrl));

package/dist/matcher.js

@@ -22,14 +22,20 @@ function matchLogic(logic, text) {
     }
     return false;
 }
+function findProviderById(providers, providerId) {
+    const normalizedProviderId = providerId.toLowerCase().trim();
+    const exactMatch = providers.find((p) => p.id === normalizedProviderId);
+    if (exactMatch)
+        return exactMatch;
+    return providers.find((p) => p.provider_match && matchLogic(p.provider_match, normalizedProviderId));
+}
 export function matchProvider(providers, modelRef, providerId, providerApiUrl) {
     if (providerId) {
-        return providers.find((p) => p.id === providerId);
+        return findProviderById(providers, providerId);
     }
     if (providerApiUrl) {
         return providers.find((p) => new RegExp(p.api_pattern).test(providerApiUrl));
     }
-    // Try model_match logic
     return providers.find((p) => p.model_match && matchLogic(p.model_match, modelRef));
 }
 export function matchModel(models, modelRef) {

package/dist/types.d.ts

@@ -71,6 +71,7 @@ export interface Provider {
     description?: string;
     price_comments?: string;
     model_match?: MatchLogic;
+    provider_match?: MatchLogic;
     models: ModelInfo[];
 }
 export interface PriceCalculation {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@pydantic/genai-prices",
-  "version": "0.0.6",
+  "version": "0.0.7",
   "description": "Calculate prices for calling LLM inference APIs (JS/TS version of genai-prices)",
   "main": "./dist/index.cjs",
   "module": "./dist/index.js",