@stackfactor/agent-utils 1.1.0 → 1.1.1

package/dist/cjs/langChain.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"langChain.d.ts","sourceRoot":"","sources":["../../src/langChain.ts"],"names":[],"mappings":";0CAiD6C,GAAG,KAAG,IAAI;wBAke/C,MAAM,aACD,MAAM,gBACH,MAAM,SACb,GAAG,EAAE,kBACI,GAAG,UACX,GAAG,KACV,GAAG;sBAyBG,GAAG,UACF,MAAM,UACN,GAAG,eACC,QAAQ,GAAG,IAAI,KAC1B,OAAO,CAAC,GAAG,CAAC;oCAuXF,MAAM,UACT,GAAG,UACH,GAAG,oBACO,GAAG,eACT,MAAM,eACN,MAAM,wBACG,OAAO,WACpB,GAAG,cACA,MAAM,UACV,GAAG,EAAE,KACX,OAAO,CAAC,GAAG,CAAC;sDAwsBF,MAAM,UACT,GAAG,UACH,MAAM,YACL,GAAG,KACX,OAAO,CAAC,GAAG,CAAC;0CAv+B8B,GAAG,KAAG,MAAM;mCAoBxC,MAAM,mBACJ,MAAM,EAAE,KACxB,MAAM;;AA4gCT,wBAQE"}
+{"version":3,"file":"langChain.d.ts","sourceRoot":"","sources":["../../src/langChain.ts"],"names":[],"mappings":";0CAiD6C,GAAG,KAAG,IAAI;wBAqgB/C,MAAM,aACD,MAAM,gBACH,MAAM,SACb,GAAG,EAAE,kBACI,GAAG,UACX,GAAG,KACV,GAAG;sBAyBG,GAAG,UACF,MAAM,UACN,GAAG,eACC,QAAQ,GAAG,IAAI,KAC1B,OAAO,CAAC,GAAG,CAAC;oCAqXF,MAAM,UACT,GAAG,UACH,GAAG,oBACO,GAAG,eACT,MAAM,eACN,MAAM,wBACG,OAAO,WACpB,GAAG,cACA,MAAM,UACV,GAAG,EAAE,KACX,OAAO,CAAC,GAAG,CAAC;sDAosBF,MAAM,UACT,GAAG,UACH,MAAM,YACL,GAAG,KACX,OAAO,CAAC,GAAG,CAAC;0CAn+B8B,GAAG,KAAG,MAAM;mCAoBxC,MAAM,mBACJ,MAAM,EAAE,KACxB,MAAM;;AAwgCT,wBAQE"}

package/dist/cjs/langChain.js

@@ -68,40 +68,68 @@ const recordCost = (cost) => {
     }
 };
 /**
- * Looks up the per-model pricing entry from `config.modelPricing`. Returns `null`
- * when pricing is not configured for the model, in which case cost recording
- * becomes a no-op (the call still succeeds — pricing data is the host's
- * responsibility, not the agent's).
+ * Reads a per-model rate from the flat cost constants the integration defines
+ * in its `config.yaml` (exposed on the config object), e.g.
+ * `claude-opus-4-7-input-token-costs`. Returns `null` when the constant is
+ * absent or not numeric, in which case cost recording becomes a no-op (the
+ * call still succeeds — pricing is the integration's configuration concern).
  */
-const getModelPrice = (modelName, config) => {
-    const pricing = config?.modelPricing;
-    if (!pricing)
+const getModelRate = (modelName, config, kind) => {
+    if (!config || !modelName)
         return null;
-    return pricing[modelName] || null;
+    const rate = Number(config[`${modelName}-${kind}-costs`]);
+    return Number.isFinite(rate) ? rate : null;
 };
 /**
- * Computes USD cost for a text LLM call. `config.modelPricing[modelName]` is
- * expected to provide `input` and `output` rates in dollars per million tokens.
+ * Accumulates the per-model cost breakdown on `global.quota.byModel` so the
+ * agent can report input vs output spend per model to the UI. No-ops when the
+ * quota global is absent.
  */
-const calculateTextCost = (modelName, usage, config) => {
+const recordModelBreakdown = (modelName, inputCost, outputCost, imageCost) => {
+    const quota = globalThis.quota;
+    if (!quota || !modelName)
+        return;
+    if (!quota.byModel || typeof quota.byModel !== "object")
+        quota.byModel = {};
+    const entry = (quota.byModel[modelName] ||= {
+        inputCost: 0,
+        outputCost: 0,
+        imageCost: 0,
+    });
+    if (Number.isFinite(inputCost) && inputCost > 0)
+        entry.inputCost += inputCost;
+    if (Number.isFinite(outputCost) && outputCost > 0)
+        entry.outputCost += outputCost;
+    if (Number.isFinite(imageCost) && imageCost > 0)
+        entry.imageCost += imageCost;
+};
+/**
+ * Computes and records the USD cost of a text LLM call from the
+ * `<model>-input-token-costs` / `<model>-output-token-costs` constants
+ * (dollars per million tokens). Updates both the session total
+ * (`quota.usedThisSession` / `quota.remaining`) and the per-model breakdown
+ * (`quota.byModel`).
+ */
+const recordTextCost = (modelName, usage, config) => {
     if (!usage)
-        return 0;
-    const price = getModelPrice(modelName, config);
-    if (!price)
-        return 0;
-    const inputCost = ((usage.input_tokens || 0) / 1_000_000) * (price.input || 0);
-    const outputCost = ((usage.output_tokens || 0) / 1_000_000) * (price.output || 0);
-    return inputCost + outputCost;
+        return;
+    const inputRate = getModelRate(modelName, config, "input-token") || 0;
+    const outputRate = getModelRate(modelName, config, "output-token") || 0;
+    const inputCost = ((usage.input_tokens || 0) / 1_000_000) * inputRate;
+    const outputCost = ((usage.output_tokens || 0) / 1_000_000) * outputRate;
+    recordCost(inputCost + outputCost);
+    recordModelBreakdown(modelName, inputCost, outputCost, 0);
 };
 /**
- * Computes USD cost for an image generation call. `config.modelPricing[modelName]`
- * is expected to provide a `perImage` rate in dollars.
+ * Computes and records the USD cost of an image generation call from the
+ * `<model>-image-costs` constant (dollars per generated image). Updates both
+ * the session total and the per-model breakdown.
  */
-const calculateImageCost = (modelName, numImages, config) => {
-    const price = getModelPrice(modelName, config);
-    if (!price)
-        return 0;
-    return (numImages || 0) * (price.perImage || 0);
+const recordImageCost = (modelName, numImages, config) => {
+    const rate = getModelRate(modelName, config, "image") || 0;
+    const imageCost = (numImages || 0) * rate;
+    recordCost(imageCost);
+    recordModelBreakdown(modelName, 0, 0, imageCost);
 };
 /**
  * Extracts a normalized token-usage object from a single LangChain `invoke()`
@@ -569,7 +597,7 @@ const runAgent = async (agent, prompt, config, onProgress = null) => {
             throw err;
         }
     }
-    recordCost(calculateTextCost(modelName, sumAgentResponseUsage(response), config));
+    recordTextCost(modelName, sumAgentResponseUsage(response), config);
     const endTime = Date.now();
     const duration = endTime - startTime;
     logger_js_1.default.log(null, logger_js_1.default.levels.info, `Agent "${agent.options?.name}" completed in ${Math.round(duration / 1000)} seconds.`);
@@ -1044,7 +1072,7 @@ const runPromptWithModel = async (modelName, config, prompt, onProgressReport, m
                     throw err;
                 }
             }
-            recordCost(calculateTextCost(modelName, streamUsage, config));
+            recordTextCost(modelName, streamUsage, config);
             if (!rawContent) {
                 throw errorHandling_js_1.default.create(const_js_1.default.HTTP_CODES.INTERNAL_SERVER_ERROR, "LLM returned no content");
             }
@@ -1144,7 +1172,7 @@ const runPromptWithModel = async (modelName, config, prompt, onProgressReport, m
                     throw err;
                 }
             }
-            recordCost(calculateTextCost(modelName, extractUsageFromInvoke(response), config));
+            recordTextCost(modelName, extractUsageFromInvoke(response), config);
             const rawContent = response?.content || response;
             // If not expecting JSON, return raw content directly
             if (!expectsJsonResponse) {
@@ -1252,7 +1280,7 @@ const generateImageWithOpenAI = async (modelName, config, prompt, options) => {
     }
     assertQuotaAvailable();
     const response = await openai.images.generate(requestParams);
-    recordCost(calculateImageCost(modelName, response.data?.length || n, config));
+    recordImageCost(modelName, response.data?.length || n, config);
     // Format response based on number of images
     if (n === 1) {
         const imageData = response.data[0];
@@ -1369,7 +1397,7 @@ const generateImageWithGoogle = async (modelName, config, prompt, options) => {
     if (images.length === 0) {
         throw errorHandling_js_1.default.create(const_js_1.default.HTTP_CODES.INTERNAL_SERVER_ERROR, `No images were generated by ${modelName}`);
     }
-    recordCost(calculateImageCost(modelName, images.length, config));
+    recordImageCost(modelName, images.length, config);
     if (numberOfImages === 1 || images.length === 1) {
         return images[0];
     }

package/dist/esm/langChain.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"langChain.d.ts","sourceRoot":"","sources":["../../src/langChain.ts"],"names":[],"mappings":";0CAiD6C,GAAG,KAAG,IAAI;wBAke/C,MAAM,aACD,MAAM,gBACH,MAAM,SACb,GAAG,EAAE,kBACI,GAAG,UACX,GAAG,KACV,GAAG;sBAyBG,GAAG,UACF,MAAM,UACN,GAAG,eACC,QAAQ,GAAG,IAAI,KAC1B,OAAO,CAAC,GAAG,CAAC;oCAuXF,MAAM,UACT,GAAG,UACH,GAAG,oBACO,GAAG,eACT,MAAM,eACN,MAAM,wBACG,OAAO,WACpB,GAAG,cACA,MAAM,UACV,GAAG,EAAE,KACX,OAAO,CAAC,GAAG,CAAC;sDAwsBF,MAAM,UACT,GAAG,UACH,MAAM,YACL,GAAG,KACX,OAAO,CAAC,GAAG,CAAC;0CAv+B8B,GAAG,KAAG,MAAM;mCAoBxC,MAAM,mBACJ,MAAM,EAAE,KACxB,MAAM;;AA4gCT,wBAQE"}
+{"version":3,"file":"langChain.d.ts","sourceRoot":"","sources":["../../src/langChain.ts"],"names":[],"mappings":";0CAiD6C,GAAG,KAAG,IAAI;wBAqgB/C,MAAM,aACD,MAAM,gBACH,MAAM,SACb,GAAG,EAAE,kBACI,GAAG,UACX,GAAG,KACV,GAAG;sBAyBG,GAAG,UACF,MAAM,UACN,GAAG,eACC,QAAQ,GAAG,IAAI,KAC1B,OAAO,CAAC,GAAG,CAAC;oCAqXF,MAAM,UACT,GAAG,UACH,GAAG,oBACO,GAAG,eACT,MAAM,eACN,MAAM,wBACG,OAAO,WACpB,GAAG,cACA,MAAM,UACV,GAAG,EAAE,KACX,OAAO,CAAC,GAAG,CAAC;sDAosBF,MAAM,UACT,GAAG,UACH,MAAM,YACL,GAAG,KACX,OAAO,CAAC,GAAG,CAAC;0CAn+B8B,GAAG,KAAG,MAAM;mCAoBxC,MAAM,mBACJ,MAAM,EAAE,KACxB,MAAM;;AAwgCT,wBAQE"}

package/dist/esm/langChain.js

@@ -63,40 +63,68 @@ const recordCost = (cost) => {
     }
 };
 /**
- * Looks up the per-model pricing entry from `config.modelPricing`. Returns `null`
- * when pricing is not configured for the model, in which case cost recording
- * becomes a no-op (the call still succeeds — pricing data is the host's
- * responsibility, not the agent's).
+ * Reads a per-model rate from the flat cost constants the integration defines
+ * in its `config.yaml` (exposed on the config object), e.g.
+ * `claude-opus-4-7-input-token-costs`. Returns `null` when the constant is
+ * absent or not numeric, in which case cost recording becomes a no-op (the
+ * call still succeeds — pricing is the integration's configuration concern).
  */
-const getModelPrice = (modelName, config) => {
-    const pricing = config?.modelPricing;
-    if (!pricing)
+const getModelRate = (modelName, config, kind) => {
+    if (!config || !modelName)
         return null;
-    return pricing[modelName] || null;
+    const rate = Number(config[`${modelName}-${kind}-costs`]);
+    return Number.isFinite(rate) ? rate : null;
 };
 /**
- * Computes USD cost for a text LLM call. `config.modelPricing[modelName]` is
- * expected to provide `input` and `output` rates in dollars per million tokens.
+ * Accumulates the per-model cost breakdown on `global.quota.byModel` so the
+ * agent can report input vs output spend per model to the UI. No-ops when the
+ * quota global is absent.
  */
-const calculateTextCost = (modelName, usage, config) => {
+const recordModelBreakdown = (modelName, inputCost, outputCost, imageCost) => {
+    const quota = globalThis.quota;
+    if (!quota || !modelName)
+        return;
+    if (!quota.byModel || typeof quota.byModel !== "object")
+        quota.byModel = {};
+    const entry = (quota.byModel[modelName] ||= {
+        inputCost: 0,
+        outputCost: 0,
+        imageCost: 0,
+    });
+    if (Number.isFinite(inputCost) && inputCost > 0)
+        entry.inputCost += inputCost;
+    if (Number.isFinite(outputCost) && outputCost > 0)
+        entry.outputCost += outputCost;
+    if (Number.isFinite(imageCost) && imageCost > 0)
+        entry.imageCost += imageCost;
+};
+/**
+ * Computes and records the USD cost of a text LLM call from the
+ * `<model>-input-token-costs` / `<model>-output-token-costs` constants
+ * (dollars per million tokens). Updates both the session total
+ * (`quota.usedThisSession` / `quota.remaining`) and the per-model breakdown
+ * (`quota.byModel`).
+ */
+const recordTextCost = (modelName, usage, config) => {
     if (!usage)
-        return 0;
-    const price = getModelPrice(modelName, config);
-    if (!price)
-        return 0;
-    const inputCost = ((usage.input_tokens || 0) / 1_000_000) * (price.input || 0);
-    const outputCost = ((usage.output_tokens || 0) / 1_000_000) * (price.output || 0);
-    return inputCost + outputCost;
+        return;
+    const inputRate = getModelRate(modelName, config, "input-token") || 0;
+    const outputRate = getModelRate(modelName, config, "output-token") || 0;
+    const inputCost = ((usage.input_tokens || 0) / 1_000_000) * inputRate;
+    const outputCost = ((usage.output_tokens || 0) / 1_000_000) * outputRate;
+    recordCost(inputCost + outputCost);
+    recordModelBreakdown(modelName, inputCost, outputCost, 0);
 };
 /**
- * Computes USD cost for an image generation call. `config.modelPricing[modelName]`
- * is expected to provide a `perImage` rate in dollars.
+ * Computes and records the USD cost of an image generation call from the
+ * `<model>-image-costs` constant (dollars per generated image). Updates both
+ * the session total and the per-model breakdown.
  */
-const calculateImageCost = (modelName, numImages, config) => {
-    const price = getModelPrice(modelName, config);
-    if (!price)
-        return 0;
-    return (numImages || 0) * (price.perImage || 0);
+const recordImageCost = (modelName, numImages, config) => {
+    const rate = getModelRate(modelName, config, "image") || 0;
+    const imageCost = (numImages || 0) * rate;
+    recordCost(imageCost);
+    recordModelBreakdown(modelName, 0, 0, imageCost);
 };
 /**
  * Extracts a normalized token-usage object from a single LangChain `invoke()`
@@ -564,7 +592,7 @@ const runAgent = async (agent, prompt, config, onProgress = null) => {
             throw err;
         }
     }
-    recordCost(calculateTextCost(modelName, sumAgentResponseUsage(response), config));
+    recordTextCost(modelName, sumAgentResponseUsage(response), config);
     const endTime = Date.now();
     const duration = endTime - startTime;
     logger.log(null, logger.levels.info, `Agent "${agent.options?.name}" completed in ${Math.round(duration / 1000)} seconds.`);
@@ -1039,7 +1067,7 @@ const runPromptWithModel = async (modelName, config, prompt, onProgressReport, m
                     throw err;
                 }
             }
-            recordCost(calculateTextCost(modelName, streamUsage, config));
+            recordTextCost(modelName, streamUsage, config);
             if (!rawContent) {
                 throw errorHandlingHelper.create(constants.HTTP_CODES.INTERNAL_SERVER_ERROR, "LLM returned no content");
             }
@@ -1139,7 +1167,7 @@ const runPromptWithModel = async (modelName, config, prompt, onProgressReport, m
                     throw err;
                 }
             }
-            recordCost(calculateTextCost(modelName, extractUsageFromInvoke(response), config));
+            recordTextCost(modelName, extractUsageFromInvoke(response), config);
             const rawContent = response?.content || response;
             // If not expecting JSON, return raw content directly
             if (!expectsJsonResponse) {
@@ -1247,7 +1275,7 @@ const generateImageWithOpenAI = async (modelName, config, prompt, options) => {
     }
     assertQuotaAvailable();
     const response = await openai.images.generate(requestParams);
-    recordCost(calculateImageCost(modelName, response.data?.length || n, config));
+    recordImageCost(modelName, response.data?.length || n, config);
     // Format response based on number of images
     if (n === 1) {
         const imageData = response.data[0];
@@ -1364,7 +1392,7 @@ const generateImageWithGoogle = async (modelName, config, prompt, options) => {
     if (images.length === 0) {
         throw errorHandlingHelper.create(constants.HTTP_CODES.INTERNAL_SERVER_ERROR, `No images were generated by ${modelName}`);
     }
-    recordCost(calculateImageCost(modelName, images.length, config));
+    recordImageCost(modelName, images.length, config);
     if (numberOfImages === 1 || images.length === 1) {
         return images[0];
     }

package/package.json

@@ -3,7 +3,7 @@
   "publishConfig": {
     "access": "restricted"
   },
-  "version": "1.1.0",
+  "version": "1.1.1",
   "description": "",
   "main": "dist/cjs/index.js",
   "module": "dist/esm/index.js",