trickle-observe 0.2.121 → 0.2.123

package/dist/llm-observer.js

@@ -54,6 +54,12 @@ let llmFile = null;
 let eventCount = 0;
 const MAX_LLM_EVENTS = 500;
 const TRUNCATE_LEN = 500;
+// Token budget enforcement
+let cumulativeTokens = 0;
+let cumulativeCost = 0;
+let budgetWarned = false;
+const TOKEN_BUDGET = parseInt(process.env.TRICKLE_TOKEN_BUDGET || '0', 10);
+const COST_BUDGET = parseFloat(process.env.TRICKLE_COST_BUDGET || '0');
 // Approximate pricing per 1M tokens (USD) — used for cost estimation
 const PRICING = {
     'gpt-4o': { input: 2.5, output: 10 },
@@ -88,6 +94,19 @@ function writeLlmEvent(event) {
     if (eventCount >= MAX_LLM_EVENTS)
         return;
     eventCount++;
+    // Track cumulative usage for budget enforcement
+    cumulativeTokens += event.totalTokens || 0;
+    cumulativeCost += event.estimatedCostUsd || 0;
+    if (!budgetWarned) {
+        if (TOKEN_BUDGET > 0 && cumulativeTokens > TOKEN_BUDGET) {
+            console.warn(`[trickle] ⚠ Token budget exceeded: ${cumulativeTokens} tokens used (budget: ${TOKEN_BUDGET}). Set TRICKLE_TOKEN_BUDGET=0 to disable.`);
+            budgetWarned = true;
+        }
+        if (COST_BUDGET > 0 && cumulativeCost > COST_BUDGET) {
+            console.warn(`[trickle] ⚠ Cost budget exceeded: $${cumulativeCost.toFixed(4)} spent (budget: $${COST_BUDGET.toFixed(4)}). Set TRICKLE_COST_BUDGET=0 to disable.`);
+            budgetWarned = true;
+        }
+    }
     try {
         fs.appendFileSync(getLlmFile(), JSON.stringify(event) + '\n');
     }

package/dist/observe-register.js

@@ -1572,6 +1572,17 @@ if (enabled) {
             }
             catch { /* not critical */ }
         }
+        // Claude Agent SDK
+        if (request.includes('claude-agent-sdk') && !expressPatched.has('claude-agent-sdk')) {
+            expressPatched.add('claude-agent-sdk');
+            try {
+                // The Claude Agent SDK uses hooks config — we can't easily patch
+                // from Module._load since it's ESM-only. Log detection for now.
+                if (debug)
+                    console.log('[trickle] Claude Agent SDK detected — use hooks for tracing');
+            }
+            catch { /* not critical */ }
+        }
         // Google Gemini SDK
         if (request === '@google/genai' && !expressPatched.has('@google/genai')) {
             expressPatched.add('@google/genai');

package/package.json

@@ -1,7 +1,7 @@
 {
   "name": "trickle-observe",
-  "version": "0.2.121",
-  "description": "Runtime type observability for JavaScript applications",
+  "version": "0.2.123",
+  "description": "Zero-code runtime observability for JavaScript/TypeScript. Auto-instruments Express, OpenAI, Anthropic, Gemini, MCP. Captures functions, variables, LLM calls, agent workflows.",
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
   "exports": {

package/src/llm-observer.ts

@@ -18,6 +18,13 @@ let eventCount = 0;
 const MAX_LLM_EVENTS = 500;
 const TRUNCATE_LEN = 500;
 
+// Token budget enforcement
+let cumulativeTokens = 0;
+let cumulativeCost = 0;
+let budgetWarned = false;
+const TOKEN_BUDGET = parseInt(process.env.TRICKLE_TOKEN_BUDGET || '0', 10);
+const COST_BUDGET = parseFloat(process.env.TRICKLE_COST_BUDGET || '0');
+
 // Approximate pricing per 1M tokens (USD) — used for cost estimation
 const PRICING: Record<string, { input: number; output: number }> = {
   'gpt-4o': { input: 2.5, output: 10 },
@@ -71,6 +78,21 @@ interface LlmEvent {
 function writeLlmEvent(event: LlmEvent): void {
   if (eventCount >= MAX_LLM_EVENTS) return;
   eventCount++;
+
+  // Track cumulative usage for budget enforcement
+  cumulativeTokens += event.totalTokens || 0;
+  cumulativeCost += event.estimatedCostUsd || 0;
+
+  if (!budgetWarned) {
+    if (TOKEN_BUDGET > 0 && cumulativeTokens > TOKEN_BUDGET) {
+      console.warn(`[trickle] ⚠ Token budget exceeded: ${cumulativeTokens} tokens used (budget: ${TOKEN_BUDGET}). Set TRICKLE_TOKEN_BUDGET=0 to disable.`);
+      budgetWarned = true;
+    }
+    if (COST_BUDGET > 0 && cumulativeCost > COST_BUDGET) {
+      console.warn(`[trickle] ⚠ Cost budget exceeded: $${cumulativeCost.toFixed(4)} spent (budget: $${COST_BUDGET.toFixed(4)}). Set TRICKLE_COST_BUDGET=0 to disable.`);
+      budgetWarned = true;
+    }
+  }
   try {
     fs.appendFileSync(getLlmFile(), JSON.stringify(event) + '\n');
   } catch {}

package/src/observe-register.ts

@@ -1557,6 +1557,16 @@ if (enabled) {
       } catch { /* not critical */ }
     }
 
+    // Claude Agent SDK
+    if (request.includes('claude-agent-sdk') && !expressPatched.has('claude-agent-sdk')) {
+      expressPatched.add('claude-agent-sdk');
+      try {
+        // The Claude Agent SDK uses hooks config — we can't easily patch
+        // from Module._load since it's ESM-only. Log detection for now.
+        if (debug) console.log('[trickle] Claude Agent SDK detected — use hooks for tracing');
+      } catch { /* not critical */ }
+    }
+
     // Google Gemini SDK
     if (request === '@google/genai' && !expressPatched.has('@google/genai')) {
       expressPatched.add('@google/genai');