fourmis-agents-sdk 0.2.4 → 0.2.6

package/dist/utils/cost.js

@@ -163,13 +163,79 @@ function findOpenAIPricing(model) {
   }
   return bestPricing;
 }
+var GEMINI_PRICING = {
+  "gemini-2.5-pro": {
+    inputPerMillion: 1.25,
+    outputPerMillion: 10,
+    cacheReadPerMillion: 0.315
+  },
+  "gemini-2.5-flash": {
+    inputPerMillion: 0.15,
+    outputPerMillion: 0.6,
+    cacheReadPerMillion: 0.0375
+  },
+  "gemini-2.5-flash-lite": {
+    inputPerMillion: 0.075,
+    outputPerMillion: 0.3
+  },
+  "gemini-2.0-flash": {
+    inputPerMillion: 0.1,
+    outputPerMillion: 0.4,
+    cacheReadPerMillion: 0.025
+  },
+  "gemini-2.0-flash-lite": {
+    inputPerMillion: 0.075,
+    outputPerMillion: 0.3
+  }
+};
+var GEMINI_CONTEXT_WINDOWS = {
+  "gemini-2.5-pro": 1048576,
+  "gemini-2.5-flash": 1048576,
+  "gemini-2.5-flash-lite": 1048576,
+  "gemini-2.0-flash": 1048576,
+  "gemini-2.0-flash-lite": 1048576
+};
+var GEMINI_MAX_OUTPUT = {
+  "gemini-2.5-pro": 65536,
+  "gemini-2.5-flash": 65536,
+  "gemini-2.5-flash-lite": 65536,
+  "gemini-2.0-flash": 8192,
+  "gemini-2.0-flash-lite": 8192
+};
+function calculateGeminiCost(model, usage) {
+  const pricing = findGeminiPricing(model);
+  if (!pricing)
+    return 0;
+  const inputCost = usage.inputTokens / 1e6 * pricing.inputPerMillion;
+  const outputCost = usage.outputTokens / 1e6 * pricing.outputPerMillion;
+  const cacheReadCost = usage.cacheReadInputTokens / 1e6 * (pricing.cacheReadPerMillion ?? pricing.inputPerMillion);
+  return inputCost + outputCost + cacheReadCost;
+}
+function findGeminiPricing(model) {
+  if (GEMINI_PRICING[model])
+    return GEMINI_PRICING[model];
+  let bestKey = "";
+  let bestPricing;
+  for (const [key, pricing] of Object.entries(GEMINI_PRICING)) {
+    if (model.startsWith(key) && key.length > bestKey.length) {
+      bestKey = key;
+      bestPricing = pricing;
+    }
+  }
+  return bestPricing;
+}
 export {
   findOpenAIPricing,
+  findGeminiPricing,
   calculateOpenAICost,
+  calculateGeminiCost,
   calculateAnthropicCost,
   OPENAI_PRICING,
   OPENAI_MAX_OUTPUT,
   OPENAI_CONTEXT_WINDOWS,
+  GEMINI_PRICING,
+  GEMINI_MAX_OUTPUT,
+  GEMINI_CONTEXT_WINDOWS,
   ANTHROPIC_PRICING,
   ANTHROPIC_MAX_OUTPUT,
   ANTHROPIC_CONTEXT_WINDOWS

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "fourmis-agents-sdk",
-  "version": "0.2.4",
+  "version": "0.2.6",
   "description": "Multi-provider AI agent SDK with direct API access and in-process tool execution",
   "type": "module",
   "main": "dist/index.js",
@@ -9,6 +9,14 @@
     ".": {
       "types": "./dist/index.d.ts",
       "import": "./dist/index.js"
+    },
+    "./auth/openai": {
+      "types": "./dist/auth/openai-oauth.d.ts",
+      "import": "./dist/auth/openai-oauth.js"
+    },
+    "./auth/gemini": {
+      "types": "./dist/auth/gemini-oauth.d.ts",
+      "import": "./dist/auth/gemini-oauth.js"
     }
   },
   "files": [
@@ -54,6 +62,7 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.39.0",
+    "@google/genai": "^1.40.0",
     "@modelcontextprotocol/sdk": "^1.26.0",
     "openai": "^6.21.0",
     "zod": "^3.24.0"