@dexcost/sdk 0.2.0

package/dist/cli/scanner.js

@@ -0,0 +1,480 @@
+/**
+ * Code scanner for cost point detection (US-031).
+ *
+ * Regex-based static analysis — no API key needed, runs offline.
+ *
+ * Detection layers:
+ * 1. LLM SDK calls (auto-instrumentable by dexcost or manual)
+ * 2. Agent framework invocations (LangChain, Mastra, CrewAI, etc.)
+ * 3. Direct HTTP calls to known AI API endpoints
+ * 4. SDK imports for paid services (AWS, Stripe, Firebase, etc.)
+ * 5. Service-specific method calls (MongoDB, Elasticsearch, etc.)
+ */
+const PATTERNS = [
+    // ── LLM SDK calls (auto-instrumented by dexcost) ────────────────────
+    {
+        regex: /\.chat\s*\.\s*completions\s*\.\s*create\b/g,
+        provider: "openai",
+        type: "llm_call",
+        autoInstrumentable: true,
+    },
+    {
+        regex: /\.messages\s*\.\s*create\b/g,
+        provider: "anthropic",
+        type: "llm_call",
+        autoInstrumentable: true,
+    },
+    {
+        regex: /\b(generateText|streamText)\s*\(/g,
+        provider: "vercel-ai",
+        type: "llm_call",
+        autoInstrumentable: true,
+    },
+    // ── Additional LLM providers (need recordCost) ──────────────────────
+    // Groq — uses OpenAI-compatible SDK: groq-sdk
+    {
+        regex: /from\s+["']groq-sdk["']|require\s*\(\s*["']groq-sdk["']\s*\)/g,
+        provider: "groq",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Mistral — @mistralai/mistralai, uses client.chat.complete()
+    {
+        regex: /\.chat\s*\.\s*complete\b/g,
+        provider: "mistral",
+        type: "llm_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']@mistralai\/mistralai["']/g,
+        provider: "mistral",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Together AI — together-ai
+    {
+        regex: /from\s+["']together-ai["']|require\s*\(\s*["']together-ai["']\s*\)/g,
+        provider: "together",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Replicate — replicate.run()
+    {
+        regex: /replicate\s*\.\s*run\s*\(/g,
+        provider: "replicate",
+        type: "llm_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']replicate["']|require\s*\(\s*["']replicate["']\s*\)/g,
+        provider: "replicate",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Fireworks AI
+    {
+        regex: /from\s+["']@fireworks-ai\/client["']/g,
+        provider: "fireworks",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Cohere — cohere.chat() / cohere.generate()
+    {
+        regex: /\bcohere\s*\.\s*(chat|generate|embed)\s*\(/g,
+        provider: "cohere",
+        type: "llm_call",
+        autoInstrumentable: true,
+    },
+    {
+        regex: /from\s+["']cohere-ai["']|require\s*\(\s*["']cohere-ai["']\s*\)/g,
+        provider: "cohere",
+        type: "sdk_import",
+        autoInstrumentable: true,
+    },
+    // Google Gemini — @google/genai, generateContent
+    {
+        regex: /\.generateContent\s*\(/g,
+        provider: "google-ai",
+        type: "llm_call",
+        autoInstrumentable: true,
+    },
+    // Ollama
+    {
+        regex: /from\s+["']ollama["']|require\s*\(\s*["']ollama["']\s*\)/g,
+        provider: "ollama",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Embeddings ──────────────────────────────────────────────────────
+    {
+        regex: /\.embeddings\s*\.\s*create\s*\(/g,
+        provider: "openai",
+        type: "embedding",
+        autoInstrumentable: false,
+    },
+    // ── Speech / Audio ──────────────────────────────────────────────────
+    {
+        regex: /\.audio\s*\.\s*transcriptions\s*\.\s*create\s*\(/g,
+        provider: "openai-whisper",
+        type: "speech",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /\.audio\s*\.\s*speech\s*\.\s*create\s*\(/g,
+        provider: "openai-tts",
+        type: "speech",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']@deepgram\/sdk["']/g,
+        provider: "deepgram",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']assemblyai["']/g,
+        provider: "assemblyai",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Image generation ────────────────────────────────────────────────
+    {
+        regex: /\.images\s*\.\s*generate\s*\(/g,
+        provider: "openai-dalle",
+        type: "image",
+        autoInstrumentable: false,
+    },
+    // ── Agent framework calls ───────────────────────────────────────────
+    // LangChain.js — .invoke(), .stream(), .batch()
+    {
+        regex: /from\s+["'](@langchain\/openai|@langchain\/anthropic|@langchain\/core|langchain)["']/g,
+        provider: "langchain",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /\.(invoke|ainvoke|stream|batch)\s*\(/g,
+        provider: "langchain",
+        type: "framework_call",
+        autoInstrumentable: false,
+    },
+    // LangGraph
+    {
+        regex: /from\s+["']@langchain\/langgraph["']/g,
+        provider: "langgraph",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Mastra
+    {
+        regex: /from\s+["']@mastra\/core["']/g,
+        provider: "mastra",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // CrewAI (JS)
+    {
+        regex: /from\s+["']crewai["']/g,
+        provider: "crewai",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Direct HTTP calls to known AI API endpoints ─────────────────────
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.openai\.com/g,
+        provider: "openai",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.anthropic\.com/g,
+        provider: "anthropic",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.cohere\.(ai|com)/g,
+        provider: "cohere",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.groq\.com/g,
+        provider: "groq",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.mistral\.ai/g,
+        provider: "mistral",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.together\.xyz/g,
+        provider: "together",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.replicate\.com/g,
+        provider: "replicate",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /fetch\s*\(\s*["'`]https?:\/\/api\.deepseek\.com/g,
+        provider: "deepseek",
+        type: "http_call",
+        autoInstrumentable: false,
+    },
+    // ── Cloud / Infrastructure SDK imports ──────────────────────────────
+    {
+        regex: /@aws-sdk\/client-bedrock-runtime/g,
+        provider: "aws-bedrock",
+        type: "sdk_import",
+        autoInstrumentable: true,
+    },
+    {
+        regex: /@aws-sdk\/client-/g,
+        provider: "aws",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /@google-ai\/generativelanguage|@google\/generative-ai|@google\/genai/g,
+        provider: "google-ai",
+        type: "sdk_import",
+        autoInstrumentable: true,
+    },
+    {
+        regex: /from\s+["']@google-cloud\//g,
+        provider: "gcp",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Payments ────────────────────────────────────────────────────────
+    {
+        regex: /from\s+["']stripe["']|require\s*\(\s*["']stripe["']\s*\)/g,
+        provider: "stripe",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /stripe\s*\.\s*(paymentIntents|charges|customers|subscriptions|invoices)\s*\.\s*(create|retrieve|update|list)\s*\(/g,
+        provider: "stripe",
+        type: "service_call",
+        autoInstrumentable: false,
+    },
+    // ── Databases ───────────────────────────────────────────────────────
+    // MongoDB
+    {
+        regex: /from\s+["']mongodb["']|require\s*\(\s*["']mongodb["']\s*\)/g,
+        provider: "mongodb",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /\.\s*(find|findOne|insertOne|insertMany|updateOne|updateMany|deleteOne|deleteMany|aggregate)\s*\(/g,
+        provider: "mongodb",
+        type: "service_call",
+        autoInstrumentable: false,
+    },
+    // Supabase
+    {
+        regex: /from\s+["']@supabase\/supabase-js["']/g,
+        provider: "supabase",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // Firebase
+    {
+        regex: /from\s+["']firebase-admin["']|from\s+["']firebase\/firestore["']/g,
+        provider: "firebase",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Search ──────────────────────────────────────────────────────────
+    {
+        regex: /from\s+["']@elastic\/elasticsearch["']/g,
+        provider: "elasticsearch",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Cache ───────────────────────────────────────────────────────────
+    {
+        regex: /from\s+["'](ioredis|redis)["']|require\s*\(\s*["'](ioredis|redis)["']\s*\)/g,
+        provider: "redis",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Vector databases ────────────────────────────────────────────────
+    {
+        regex: /from\s+["']@pinecone-database\/pinecone["']/g,
+        provider: "pinecone",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /\.\s*(query|upsert)\s*\(/g,
+        provider: "vector-db",
+        type: "service_call",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']weaviate-ts-client["']|from\s+["']weaviate-client["']/g,
+        provider: "weaviate",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']chromadb["']/g,
+        provider: "chromadb",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']@qdrant\/js-client-rest["']/g,
+        provider: "qdrant",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Messaging / Communications ──────────────────────────────────────
+    {
+        regex: /from\s+["']twilio["']|require\s*\(\s*["']twilio["']\s*\)/g,
+        provider: "twilio",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']@sendgrid\/mail["']/g,
+        provider: "sendgrid",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']resend["']/g,
+        provider: "resend",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']@slack\/web-api["']/g,
+        provider: "slack",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Geo / Maps ──────────────────────────────────────────────────────
+    {
+        regex: /from\s+["']@googlemaps\/google-maps-services-js["']/g,
+        provider: "google-maps",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    // ── Web scraping / data ─────────────────────────────────────────────
+    {
+        regex: /from\s+["']@mendable\/firecrawl-js["']/g,
+        provider: "firecrawl",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+    {
+        regex: /from\s+["']@tavily\/core["']/g,
+        provider: "tavily",
+        type: "sdk_import",
+        autoInstrumentable: false,
+    },
+];
+/**
+ * Scan a source string for cost-generating patterns.
+ *
+ * @param source - File contents to scan.
+ * @param fileName - Path to include in results.
+ * @returns Array of detected cost points.
+ */
+export function scanSource(source, fileName) {
+    const results = [];
+    const seenLines = new Set(); // "provider:line" dedup key
+    for (const pattern of PATTERNS) {
+        // Reset regex state for each file
+        pattern.regex.lastIndex = 0;
+        let match;
+        while ((match = pattern.regex.exec(source)) !== null) {
+            const beforeMatch = source.slice(0, match.index);
+            const lineNumber = beforeMatch.split("\n").length;
+            // Deduplicate: same provider on same line
+            const key = `${pattern.provider}:${lineNumber}`;
+            if (seenLines.has(key))
+                continue;
+            seenLines.add(key);
+            results.push({
+                file: fileName,
+                line: lineNumber,
+                provider: pattern.provider,
+                type: pattern.type,
+                pattern: match[0].trim(),
+                autoInstrumentable: pattern.autoInstrumentable,
+            });
+        }
+    }
+    return results;
+}
+/**
+ * Generate integration stub code from detected cost points.
+ *
+ * Produces a self-contained template showing:
+ * 1. SDK initialisation
+ * 2. Customer/project context
+ * 3. Task wrapper with manual `recordCost` calls
+ * 4. Summary of auto-instrumented providers
+ */
+export function generateStubs(points, _scanDir) {
+    const auto = points.filter((p) => p.autoInstrumentable);
+    const manual = points.filter((p) => !p.autoInstrumentable);
+    const lines = [
+        "// ============================================================",
+        "// dexcost integration stubs",
+        "// Generated by: dexcost scan --generate-stubs",
+        "// ============================================================",
+        "",
+        "// --- Step 1: Initialize dexcost ---",
+        'import { init, track } from "@dexcost/sdk";',
+        "",
+        'init({ apiKey: "dx_live_..." }); // or set DEXCOST_API_KEY env var',
+        "",
+        "// --- Step 2: Set customer context (in your request handler) ---",
+        'import { setContext } from "@dexcost/sdk";',
+        "",
+        "setContext({",
+        '  customerId: "your_customer_id",',
+        '  projectId: "your_project_id",',
+        "});",
+        "",
+        "// --- Step 3: Track tasks ---",
+        'await track({ taskType: "your_task_type" }, async (task) => {',
+        "  // Your agent code here...",
+        "",
+    ];
+    if (manual.length > 0) {
+        lines.push("  // --- Manual cost tracking (for services not auto-instrumented) ---");
+        for (const p of manual) {
+            lines.push(`  // ${p.file}:${p.line} \u2014 ${p.pattern}`);
+            lines.push(`  task.recordCost("${p.provider}", 0.00); // TODO: set actual cost`);
+            lines.push("");
+        }
+    }
+    lines.push("});");
+    lines.push("");
+    if (auto.length > 0) {
+        lines.push("// --- Auto-instrumented (no code changes needed) ---");
+        const providers = new Map();
+        for (const p of auto) {
+            providers.set(p.provider, (providers.get(p.provider) ?? 0) + 1);
+        }
+        for (const [provider, count] of providers) {
+            lines.push(`// \u2713 ${provider} (${count} call${count > 1 ? "s" : ""} detected)`);
+        }
+    }
+    return lines.join("\n") + "\n";
+}
+//# sourceMappingURL=scanner.js.map

package/dist/cli/scanner.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"scanner.js","sourceRoot":"","sources":["../../src/cli/scanner.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;GAWG;AA0BH,MAAM,QAAQ,GAAkB;IAC9B,uEAAuE;IACvE;QACE,KAAK,EAAE,4CAA4C;QACnD,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,IAAI;KACzB;IACD;QACE,KAAK,EAAE,6BAA6B;QACpC,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,IAAI;KACzB;IACD;QACE,KAAK,EAAE,mCAAmC;QAC1C,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,IAAI;KACzB;IAED,uEAAuE;IACvE,8CAA8C;IAC9C;QACE,KAAK,EAAE,+DAA+D;QACtE,QAAQ,EAAE,MAAM;QAChB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,8DAA8D;IAC9D;QACE,KAAK,EAAE,2BAA2B;QAClC,QAAQ,EAAE,SAAS;QACnB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,uCAAuC;QAC9C,QAAQ,EAAE,SAAS;QACnB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,4BAA4B;IAC5B;QACE,KAAK,EAAE,qEAAqE;QAC5E,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,8BAA8B;IAC9B;QACE,KAAK,EAAE,4BAA4B;QACnC,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,iEAAiE;QACxE,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,eAAe;IACf;QACE,KAAK,EAAE,uCAAuC;QAC9C,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,6CAA6C;IAC7C;QACE,KAAK,EAAE,6CAA6C;QACpD,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,IAAI;KACzB;IACD;QACE,KAAK,EAAE,iEAAiE;QACxE,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,IAAI;KACzB;IACD,iDAAiD;IACjD;QACE,KAAK,EAAE,yBAAyB;QAChC,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,UAAU;QAChB,kBAAkB,EAAE,IAAI;KACzB;IACD,SAAS;IACT;QACE,KAAK,EAAE,2DAA2D;QAClE,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,kCAAkC;QACzC,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,mDAAmD;QAC1D,QAAQ,EAAE,gBAAgB;QAC1B,IAAI,EAAE,QAAQ;QACd,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,2CAA2C;QAClD,QAAQ,EAAE,YAAY;QACtB,IAAI,EAAE,QAAQ;QACd,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,gCAAgC;QACvC,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,4BAA4B;QACnC,QAAQ,EAAE,YAAY;QACtB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,gCAAgC;QACvC,QAAQ,EAAE,cAAc;QACxB,IAAI,EAAE,OAAO;QACb,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE,gDAAgD;IAChD;QACE,KAAK,EAAE,uFAAuF;QAC9F,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,uCAAuC;QAC9C,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,gBAAgB;QACtB,kBAAkB,EAAE,KAAK;KAC1B;IACD,YAAY;IACZ;QACE,KAAK,EAAE,uCAAuC;QAC9C,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,SAAS;IACT;QACE,KAAK,EAAE,+BAA+B;QACtC,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,cAAc;IACd;QACE,KAAK,EAAE,wBAAwB;QAC/B,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,gDAAgD;QACvD,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,mDAAmD;QAC1D,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,qDAAqD;QAC5D,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,8CAA8C;QACrD,QAAQ,EAAE,MAAM;QAChB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,gDAAgD;QACvD,QAAQ,EAAE,SAAS;QACnB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,kDAAkD;QACzD,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,mDAAmD;QAC1D,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,kDAAkD;QACzD,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,WAAW;QACjB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,mCAAmC;QAC1C,QAAQ,EAAE,aAAa;QACvB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,IAAI;KACzB;IACD;QACE,KAAK,EAAE,oBAAoB;QAC3B,QAAQ,EAAE,KAAK;QACf,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,uEAAuE;QAC9E,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,IAAI;KACzB;IACD;QACE,KAAK,EAAE,6BAA6B;QACpC,QAAQ,EAAE,KAAK;QACf,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,2DAA2D;QAClE,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,oHAAoH;QAC3H,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,cAAc;QACpB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE,UAAU;IACV;QACE,KAAK,EAAE,6DAA6D;QACpE,QAAQ,EAAE,SAAS;QACnB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,oGAAoG;QAC3G,QAAQ,EAAE,SAAS;QACnB,IAAI,EAAE,cAAc;QACpB,kBAAkB,EAAE,KAAK;KAC1B;IACD,WAAW;IACX;QACE,KAAK,EAAE,wCAAwC;QAC/C,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD,WAAW;IACX;QACE,KAAK,EAAE,mEAAmE;QAC1E,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,yCAAyC;QAChD,QAAQ,EAAE,eAAe;QACzB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,6EAA6E;QACpF,QAAQ,EAAE,OAAO;QACjB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,8CAA8C;QACrD,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,2BAA2B;QAClC,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,cAAc;QACpB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,mEAAmE;QAC1E,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,0BAA0B;QACjC,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,yCAAyC;QAChD,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,2DAA2D;QAClE,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,iCAAiC;QACxC,QAAQ,EAAE,UAAU;QACpB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,wBAAwB;QAC/B,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,iCAAiC;QACxC,QAAQ,EAAE,OAAO;QACjB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,sDAAsD;QAC7D,QAAQ,EAAE,aAAa;QACvB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IAED,uEAAuE;IACvE;QACE,KAAK,EAAE,yCAAyC;QAChD,QAAQ,EAAE,WAAW;QACrB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;IACD;QACE,KAAK,EAAE,+BAA+B;QACtC,QAAQ,EAAE,QAAQ;QAClB,IAAI,EAAE,YAAY;QAClB,kBAAkB,EAAE,KAAK;KAC1B;CACF,CAAC;AAEF;;;;;;GAMG;AACH,MAAM,UAAU,UAAU,CAAC,MAAc,EAAE,QAAgB;IACzD,MAAM,OAAO,GAAgB,EAAE,CAAC;IAChC,MAAM,SAAS,GAAG,IAAI,GAAG,EAAU,CAAC,CAAC,4BAA4B;IAEjE,KAAK,MAAM,OAAO,IAAI,QAAQ,EAAE,CAAC;QAC/B,kCAAkC;QAClC,OAAO,CAAC,KAAK,CAAC,SAAS,GAAG,CAAC,CAAC;QAC5B,IAAI,KAA6B,CAAC;QAElC,OAAO,CAAC,KAAK,GAAG,OAAO,CAAC,KAAK,CAAC,IAAI,CAAC,MAAM,CAAC,CAAC,KAAK,IAAI,EAAE,CAAC;YACrD,MAAM,WAAW,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC,EAAE,KAAK,CAAC,KAAK,CAAC,CAAC;YACjD,MAAM,UAAU,GAAG,WAAW,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,MAAM,CAAC;YAElD,0CAA0C;YAC1C,MAAM,GAAG,GAAG,GAAG,OAAO,CAAC,QAAQ,IAAI,UAAU,EAAE,CAAC;YAChD,IAAI,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC;gBAAE,SAAS;YACjC,SAAS,CAAC,GAAG,CAAC,GAAG,CAAC,CAAC;YAEnB,OAAO,CAAC,IAAI,CAAC;gBACX,IAAI,EAAE,QAAQ;gBACd,IAAI,EAAE,UAAU;gBAChB,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,IAAI,EAAE,OAAO,CAAC,IAAI;gBAClB,OAAO,EAAE,KAAK,CAAC,CAAC,CAAC,CAAC,IAAI,EAAE;gBACxB,kBAAkB,EAAE,OAAO,CAAC,kBAAkB;aAC/C,CAAC,CAAC;QACL,CAAC;IACH,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED;;;;;;;;GAQG;AACH,MAAM,UAAU,aAAa,CAAC,MAAmB,EAAE,QAAgB;IACjE,MAAM,IAAI,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC;IACxD,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,CAAC,kBAAkB,CAAC,CAAC;IAE3D,MAAM,KAAK,GAAa;QACtB,iEAAiE;QACjE,8BAA8B;QAC9B,gDAAgD;QAChD,iEAAiE;QACjE,EAAE;QACF,uCAAuC;QACvC,6CAA6C;QAC7C,EAAE;QACF,oEAAoE;QACpE,EAAE;QACF,mEAAmE;QACnE,4CAA4C;QAC5C,EAAE;QACF,cAAc;QACd,mCAAmC;QACnC,iCAAiC;QACjC,KAAK;QACL,EAAE;QACF,gCAAgC;QAChC,+DAA+D;QAC/D,8BAA8B;QAC9B,EAAE;KACH,CAAC;IAEF,IAAI,MAAM,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACtB,KAAK,CAAC,IAAI,CACR,wEAAwE,CACzE,CAAC;QACF,KAAK,MAAM,CAAC,IAAI,MAAM,EAAE,CAAC;YACvB,KAAK,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,IAAI,IAAI,CAAC,CAAC,IAAI,WAAW,CAAC,CAAC,OAAO,EAAE,CAAC,CAAC;YAC3D,KAAK,CAAC,IAAI,CACR,sBAAsB,CAAC,CAAC,QAAQ,oCAAoC,CACrE,CAAC;YACF,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjB,CAAC;IACH,CAAC;IAED,KAAK,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC;IAClB,KAAK,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;IAEf,IAAI,IAAI,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QACpB,KAAK,CAAC,IAAI,CAAC,uDAAuD,CAAC,CAAC;QACpE,MAAM,SAAS,GAAG,IAAI,GAAG,EAAkB,CAAC;QAC5C,KAAK,MAAM,CAAC,IAAI,IAAI,EAAE,CAAC;YACrB,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,EAAE,CAAC,SAAS,CAAC,GAAG,CAAC,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC;QAClE,CAAC;QACD,KAAK,MAAM,CAAC,QAAQ,EAAE,KAAK,CAAC,IAAI,SAAS,EAAE,CAAC;YAC1C,KAAK,CAAC,IAAI,CACR,aAAa,QAAQ,KAAK,KAAK,QAAQ,KAAK,GAAG,CAAC,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,YAAY,CACxE,CAAC;QACJ,CAAC;IACH,CAAC;IAED,OAAO,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,GAAG,IAAI,CAAC;AACjC,CAAC"}

package/dist/clients.d.ts

@@ -0,0 +1,54 @@
+/**
+ * Thin wrapper classes around the OpenAI and Anthropic clients that
+ * auto-record LLM cost events.
+ *
+ * Alternative to auto-instrumentation (no monkey-patching). Matching
+ * the Python SDK's clients.py.
+ *
+ * Usage:
+ *
+ *   import { TrackedOpenAI } from "dexcost/clients";
+ *
+ *   const client = new TrackedOpenAI({ tracker });
+ *   // Inside a tracked task, events are auto-recorded:
+ *   const response = await client.chat.completions.create({ model: "gpt-4o", messages: [] });
+ */
+import { PricingEngine } from "./pricing/engine.js";
+import type { CostTracker } from "./core/tracker.js";
+/**
+ * Wraps `openai.OpenAI` and auto-records llm_call events on every
+ * `chat.completions.create` call when running inside a tracked task context.
+ */
+export declare class TrackedOpenAI {
+    private _client;
+    private _pricing;
+    private _buffer;
+    constructor(opts?: {
+        client?: unknown;
+        tracker?: CostTracker;
+        pricing?: PricingEngine;
+    });
+    get chat(): {
+        completions: {
+            create: (...args: unknown[]) => Promise<unknown>;
+        };
+    };
+}
+/**
+ * Wraps `@anthropic-ai/sdk Anthropic` and auto-records llm_call events on
+ * every `messages.create` call when running inside a tracked task context.
+ */
+export declare class TrackedAnthropic {
+    private _client;
+    private _pricing;
+    private _buffer;
+    constructor(opts?: {
+        client?: unknown;
+        tracker?: CostTracker;
+        pricing?: PricingEngine;
+    });
+    get messages(): {
+        create: (...args: unknown[]) => Promise<unknown>;
+    };
+}
+//# sourceMappingURL=clients.d.ts.map

package/dist/clients.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"clients.d.ts","sourceRoot":"","sources":["../src/clients.ts"],"names":[],"mappings":"AAAA;;;;;;;;;;;;;;GAcG;AAOH,OAAO,EAAE,aAAa,EAAE,MAAM,qBAAqB,CAAC;AAEpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,mBAAmB,CAAC;AAMrD;;;GAGG;AACH,qBAAa,aAAa;IACxB,OAAO,CAAC,OAAO,CAAU;IACzB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,OAAO,CAAqB;gBAExB,IAAI,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,WAAW,CAAC;QAAC,OAAO,CAAC,EAAE,aAAa,CAAA;KAAE;IA8BvF,IAAI,IAAI,IAAI;QAAE,WAAW,EAAE;YAAE,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;SAAE,CAAA;KAAE,CAgBhF;CACF;AA6DD;;;GAGG;AACH,qBAAa,gBAAgB;IAC3B,OAAO,CAAC,OAAO,CAAU;IACzB,OAAO,CAAC,QAAQ,CAAgB;IAChC,OAAO,CAAC,OAAO,CAAqB;gBAExB,IAAI,CAAC,EAAE;QAAE,MAAM,CAAC,EAAE,OAAO,CAAC;QAAC,OAAO,CAAC,EAAE,WAAW,CAAC;QAAC,OAAO,CAAC,EAAE,aAAa,CAAA;KAAE;IAiCvF,IAAI,QAAQ,IAAI;QAAE,MAAM,EAAE,CAAC,GAAG,IAAI,EAAE,OAAO,EAAE,KAAK,OAAO,CAAC,OAAO,CAAC,CAAA;KAAE,CAcnE;CACF"}