@dexcost/sdk 0.2.0

package/dist/instruments/mcp.js

@@ -0,0 +1,459 @@
+/**
+ * MCP (Model Context Protocol) auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `Client.prototype.callTool` to automatically record cost
+ * events for every MCP tool invocation within an active task context.
+ */
+import { randomUUID } from "node:crypto";
+import { createCostEvent } from "../core/models.js";
+import { getCurrentTask } from "../core/context.js";
+import { createAutoTask } from "../core/auto-task.js";
+import { registerInstrument } from "./index.js";
+/* eslint-disable @typescript-eslint/no-explicit-any */
+let _patched = false;
+// eslint-disable-next-line @typescript-eslint/no-unsafe-function-type
+let _original = null;
+let _clientClass = null;
+let _buffer = null;
+/** Test helper: inject a mock Client class so tests avoid importing @modelcontextprotocol/sdk. */
+export function _setClientClass(cls) {
+    _clientClass = cls;
+}
+/** Test helper: reset to real module resolution. */
+export function _resetClientClass() {
+    _clientClass = null;
+}
+// ---------------------------------------------------------------------------
+// MCP tool name → service catalog key mapping
+//
+// Comprehensive mapping covering all 163 services in service_prices.json.
+// ---------------------------------------------------------------------------
+const MCP_TOOL_MAP = {
+    // Search
+    tavily_search: "tavily_search",
+    tavily_extract: "tavily_search",
+    tavily_crawl: "tavily_search",
+    tavily_map: "tavily_search",
+    tavily_research: "tavily_search",
+    serper_search: "serper_search",
+    serper_google_search: "serper_search",
+    exa_search: "exa_search",
+    exa_find_similar: "exa_search",
+    exa_get_contents: "exa_search",
+    perplexity_search: "perplexity_search",
+    sonar_search: "perplexity_search",
+    brave_web_search: "brave_search",
+    brave_local_search: "brave_search",
+    brave_search: "brave_search",
+    bing_search: "bing_search",
+    bing_web_search: "bing_search",
+    serpapi_search: "serpapi",
+    serpapi_google_search: "serpapi",
+    // Scraping
+    firecrawl_scrape: "firecrawl_scrape",
+    firecrawl_crawl: "firecrawl_scrape",
+    firecrawl_map: "firecrawl_scrape",
+    firecrawl_search: "firecrawl_scrape",
+    firecrawl_extract: "firecrawl_scrape",
+    browserbase_create_session: "browserbase",
+    browserbase_navigate: "browserbase",
+    browserbase_screenshot: "browserbase",
+    browserbase_get_content: "browserbase",
+    scrapingbee_scrape: "scrapingbee",
+    apify_run_actor: "apify",
+    apify_get_dataset: "apify",
+    scrapingdog_scrape: "scrapingdog",
+    diffbot_analyze: "diffbot",
+    diffbot_extract: "diffbot",
+    jina_read: "jina_reader",
+    jina_reader: "jina_reader",
+    jina_search: "jina_reader",
+    // Vector DBs
+    pinecone_query: "pinecone_query",
+    pinecone_upsert: "pinecone_query",
+    pinecone_delete: "pinecone_query",
+    pinecone_fetch: "pinecone_query",
+    weaviate_query: "weaviate_cloud",
+    weaviate_search: "weaviate_cloud",
+    qdrant_search: "qdrant_cloud",
+    qdrant_query: "qdrant_cloud",
+    qdrant_upsert: "qdrant_cloud",
+    milvus_search: "milvus_zilliz",
+    milvus_query: "milvus_zilliz",
+    astra_query: "astra_db",
+    astra_search: "astra_db",
+    // Compute / Sandbox
+    e2b_run_code: "e2b_sandbox",
+    e2b_execute: "e2b_sandbox",
+    e2b_create_sandbox: "e2b_sandbox",
+    code_execute: "e2b_sandbox",
+    modal_run: "modal_compute",
+    modal_deploy: "modal_compute",
+    lambda_invoke: "aws_lambda",
+    // Communication
+    twilio_send_sms: "twilio_sms",
+    twilio_send_message: "twilio_sms",
+    sendgrid_send_email: "sendgrid_email",
+    send_email: "sendgrid_email",
+    resend_send_email: "resend_email",
+    mailgun_send: "mailgun",
+    postmark_send: "postmark",
+    // Maps
+    google_maps_geocode: "google_maps_geocode",
+    google_maps_search: "google_maps_places",
+    google_maps_directions: "google_maps_directions",
+    google_maps_places: "google_maps_places",
+    mapbox_geocode: "mapbox_geocoding",
+    mapbox_directions: "mapbox_geocoding",
+    // Data Enrichment
+    people_search: "people_data_labs",
+    pdl_search: "people_data_labs",
+    pdl_enrich: "people_data_labs",
+    clearbit_enrich: "clearbit_enrichment",
+    clearbit_lookup: "clearbit_enrichment",
+    hunter_search: "hunter_io",
+    hunter_email_finder: "hunter_io",
+    crunchbase_search: "crunchbase",
+    crunchbase_lookup: "crunchbase",
+    // Payments
+    stripe_create_charge: "stripe_payment",
+    stripe_create_payment: "stripe_payment",
+    stripe_create_payment_link: "stripe_payment",
+    stripe_list_charges: "stripe_payment",
+    // Speech / Audio
+    elevenlabs_tts: "eleven_labs",
+    elevenlabs_text_to_speech: "eleven_labs",
+    deepgram_transcribe: "deepgram_transcription",
+    assemblyai_transcribe: "assemblyai",
+    whisper_transcribe: "openai_whisper",
+    // Image Generation
+    dalle_generate: "openai_dalle",
+    openai_generate_image: "openai_dalle",
+    stability_generate: "stability_ai",
+    replicate_run: "replicate",
+    replicate_predict: "replicate",
+    // Document Processing
+    textract_analyze: "amazon_textract",
+    textract_detect: "amazon_textract",
+    document_ai_process: "google_document_ai",
+    unstructured_parse: "unstructured_io",
+    llamaparse_parse: "llamaparse",
+    llamaparse_upload: "llamaparse",
+    // Financial Data
+    twelve_data_quote: "twelve_data",
+    twelve_data_time_series: "twelve_data",
+    alpha_vantage_quote: "alpha_vantage",
+    polygon_quote: "polygon_io",
+    polygon_aggs: "polygon_io",
+    coinapi_exchange_rate: "coinapi",
+    // Cloud Storage
+    s3_get_object: "aws_s3",
+    s3_put_object: "aws_s3",
+    supabase_query: "supabase",
+    supabase_insert: "supabase",
+    supabase_select: "supabase",
+    // Agent Tools
+    composio_execute: "composio",
+    composio_action: "composio",
+    wolfram_query: "wolfram_alpha",
+    wolfram_alpha_query: "wolfram_alpha",
+    // Embeddings
+    openai_embed: "openai_embeddings",
+    cohere_embed: "cohere_embed",
+    voyage_embed: "voyage_ai",
+    jina_embed: "jina_embeddings",
+    mixedbread_embed: "mixedbread_embed",
+    nomic_embed: "nomic_embed",
+    // Translation
+    deepl_translate: "deepl_translate",
+    translate_text: "deepl_translate",
+    google_translate: "google_translate",
+    aws_translate: "aws_translate",
+    azure_translate: "azure_translator",
+    // Vision / OCR
+    google_vision_annotate: "google_vision",
+    google_vision_ocr: "google_vision",
+    azure_vision_analyze: "azure_computer_vision",
+    azure_ocr: "azure_computer_vision",
+    aws_rekognition_detect: "aws_rekognition",
+    aws_rekognition_labels: "aws_rekognition",
+    mathpix_process: "mathpix_ocr",
+    // Video AI
+    runway_generate: "runway_video",
+    heygen_create_video: "heygen_video",
+    luma_generate: "luma_video",
+    mux_upload: "mux_video",
+    mux_create_asset: "mux_video",
+    // Messaging / Notification
+    slack_post_message: "slack_api",
+    slack_send_message: "slack_api",
+    slack_chat_post: "slack_api",
+    discord_send_message: "discord_api",
+    discord_create_message: "discord_api",
+    telegram_send_message: "telegram_api",
+    telegram_send: "telegram_api",
+    twilio_call: "twilio_voice",
+    twilio_make_call: "twilio_voice",
+    onesignal_send: "onesignal",
+    pusher_trigger: "pusher",
+    novu_trigger: "novu",
+    // Database
+    neon_query: "neon_postgres",
+    neon_sql: "neon_postgres",
+    planetscale_query: "planetscale",
+    turso_query: "turso_db",
+    turso_execute: "turso_db",
+    mongodb_find: "mongodb_atlas",
+    mongodb_query: "mongodb_atlas",
+    mongodb_insert: "mongodb_atlas",
+    upstash_redis_get: "upstash_redis",
+    upstash_redis_set: "upstash_redis",
+    upstash_redis_command: "upstash_redis",
+    fauna_query: "fauna_db",
+    cockroach_query: "cockroachdb",
+    // Project Management
+    jira_create_issue: "jira_api",
+    jira_search: "jira_api",
+    jira_update_issue: "jira_api",
+    jira_get_issue: "jira_api",
+    linear_create_issue: "linear_api",
+    linear_search: "linear_api",
+    linear_update_issue: "linear_api",
+    asana_create_task: "asana_api",
+    asana_search: "asana_api",
+    clickup_create_task: "clickup_api",
+    clickup_search: "clickup_api",
+    notion_query: "notion_api",
+    notion_create_page: "notion_api",
+    notion_search: "notion_api",
+    notion_update_page: "notion_api",
+    // CRM
+    salesforce_query: "salesforce_api",
+    salesforce_create: "salesforce_api",
+    salesforce_soql: "salesforce_api",
+    hubspot_create_contact: "hubspot_api",
+    hubspot_search: "hubspot_api",
+    hubspot_get_contacts: "hubspot_api",
+    // DevOps / Code
+    github_create_issue: "github_api",
+    github_search: "github_api",
+    github_create_pr: "github_api",
+    github_get_file: "github_api",
+    github_list_repos: "github_api",
+    gitlab_create_issue: "gitlab_api",
+    gitlab_search: "gitlab_api",
+    vercel_deploy: "vercel_api",
+    vercel_list_deployments: "vercel_api",
+    // Social Media
+    twitter_post: "twitter_api",
+    twitter_search: "twitter_api",
+    x_post_tweet: "twitter_api",
+    linkedin_post: "linkedin_api",
+    linkedin_share: "linkedin_api",
+    reddit_submit: "reddit_api",
+    reddit_search: "reddit_api",
+    youtube_search: "youtube_data_api",
+    youtube_list_videos: "youtube_data_api",
+    // Weather / News / Geo
+    openweathermap_current: "openweathermap",
+    openweathermap_forecast: "openweathermap",
+    get_weather: "openweathermap",
+    weather_forecast: "weatherapi",
+    get_news: "newsapi",
+    news_search: "newsapi",
+    ip_lookup: "ipinfo",
+    geolocate: "maxmind_geoip",
+    // Workflow Automation
+    zapier_nla: "zapier",
+    zapier_action: "zapier",
+    make_webhook: "make_integromat",
+    pipedream_trigger: "pipedream",
+    // Payment / E-commerce
+    paypal_create_payment: "paypal",
+    paypal_capture: "paypal",
+    razorpay_create_order: "razorpay",
+    razorpay_capture_payment: "razorpay",
+    shopify_create_product: "shopify_api",
+    shopify_get_orders: "shopify_api",
+    // Scraping (new)
+    scraperapi_scrape: "scraperapi",
+    zenrows_scrape: "zenrows",
+    oxylabs_scrape: "oxylabs_scraper",
+    // AI/ML Platforms
+    huggingface_inference: "huggingface_inference",
+    hf_generate: "huggingface_inference",
+    roboflow_detect: "roboflow",
+    scale_create_task: "scale_ai",
+    together_generate: "together_ai",
+    groq_chat: "groq_inference",
+    fireworks_generate: "fireworks_ai",
+    cohere_rerank: "cohere_rerank",
+    jina_rerank: "jina_reranker",
+    rerank: "cohere_rerank",
+    // Cloud Storage (new)
+    gcs_upload: "gcs",
+    gcs_download: "gcs",
+    r2_put_object: "r2_cloudflare",
+    r2_get_object: "r2_cloudflare",
+    azure_blob_upload: "azure_blob",
+    // Document (new)
+    adobe_pdf_extract: "adobe_pdf_services",
+    docraptor_convert: "docraptor",
+    // Speech (new)
+    openai_tts: "openai_tts",
+    openai_text_to_speech: "openai_tts",
+    resemble_generate: "resemble_ai",
+    playht_generate: "playht",
+    // Image (new)
+    midjourney_generate: "midjourney_api",
+    fal_generate: "flux_fal",
+    flux_generate: "flux_fal",
+    leonardo_generate: "leonardo_ai",
+    // Data Enrichment (new)
+    apollo_search: "apollo_io",
+    apollo_enrich: "apollo_io",
+    zoominfo_search: "zoominfo_api",
+    // E-signature / Calendar
+    docusign_send: "docusign",
+    docusign_create_envelope: "docusign",
+    calendly_create_event: "calendly_api",
+    // Monitoring
+    sentry_capture: "sentry",
+    datadog_submit: "datadog",
+    posthog_capture: "posthog",
+    mixpanel_track: "mixpanel",
+    segment_track: "segment",
+    // CDN / Media
+    cloudinary_upload: "cloudinary",
+    cloudinary_transform: "cloudinary",
+    // Compute (new)
+    fly_machine_run: "fly_machines",
+    cloudflare_worker_run: "cloudflare_workers",
+    // Vector DB (new)
+    chroma_query: "chroma_cloud",
+    chroma_add: "chroma_cloud",
+    turbopuffer_query: "turbopuffer",
+    turbopuffer_upsert: "turbopuffer",
+    // Agent Tools (new)
+    browserless_navigate: "browserless",
+    browserless_screenshot: "browserless",
+    browserless_scrape: "browserless",
+};
+// ---------------------------------------------------------------------------
+// Instrument functions
+// ---------------------------------------------------------------------------
+/**
+ * Patch `Client.prototype.callTool` to record cost events.
+ *
+ * If `@modelcontextprotocol/sdk` is not installed and no mock class is
+ * injected, the dynamic import will throw and the function will reject.
+ */
+export async function instrumentMcp(_pricing, buffer) {
+    if (_patched)
+        return;
+    let ClientProto;
+    if (_clientClass) {
+        ClientProto = _clientClass.prototype;
+    }
+    else {
+        // @ts-expect-error -- @modelcontextprotocol/sdk types are not bundled
+        const mcpSdk = await import("@modelcontextprotocol/sdk/client/index.js");
+        const Client = mcpSdk.Client;
+        ClientProto = Client.prototype;
+    }
+    _original = ClientProto.callTool;
+    _buffer = buffer;
+    ClientProto.callTool = async function (params, ...rest) {
+        let task = getCurrentTask();
+        if (!task) {
+            task = createAutoTask("mcp.tool_call");
+            _buffer?.upsertTask(task);
+        }
+        const toolName = params?.name ?? "unknown";
+        const startTime = performance.now();
+        let isError = false;
+        try {
+            const result = await _original.call(this, params, ...rest);
+            const latencyMs = Math.round(performance.now() - startTime);
+            // Check if MCP result itself signals an error
+            if (result?.isError) {
+                isError = true;
+            }
+            try {
+                recordMcpEvent(toolName, task, latencyMs, isError, this);
+            }
+            catch {
+                // dexcost errors must never crash user code
+            }
+            return result;
+        }
+        catch (err) {
+            isError = true;
+            const latencyMs = Math.round(performance.now() - startTime);
+            try {
+                recordMcpEvent(toolName, task, latencyMs, isError, this);
+            }
+            catch {
+                // dexcost errors must never crash user code
+            }
+            throw err;
+        }
+    };
+    _patched = true;
+}
+/**
+ * Remove the monkey-patch and restore the original `callTool` method.
+ */
+export function uninstrumentMcp() {
+    if (!_patched || !_original)
+        return;
+    if (_clientClass) {
+        _clientClass.prototype.callTool = _original;
+    }
+    _original = null;
+    _buffer = null;
+    _patched = false;
+}
+// ---------------------------------------------------------------------------
+// Internal helpers
+// ---------------------------------------------------------------------------
+function recordMcpEvent(toolName, task, latencyMs, isError, instance) {
+    if (!_buffer)
+        return;
+    const { costUsd, costConfidence, pricingSource } = resolveCost(toolName);
+    const mcpServer = instance?._serverName ?? "unknown";
+    const event = createCostEvent({
+        eventId: randomUUID(),
+        taskId: task.taskId,
+        eventType: "external_cost",
+        costUsd,
+        costConfidence,
+        pricingSource,
+        serviceName: `mcp:${toolName}`,
+        latencyMs,
+        isRetry: false,
+        details: {
+            mcp_tool: toolName,
+            mcp_server: mcpServer,
+            latency_ms: latencyMs,
+            is_error: isError,
+        },
+    });
+    _buffer.addEvent(event);
+    task.externalCostUsd += costUsd;
+    task.totalCostUsd += costUsd;
+    _buffer.upsertTask(task);
+}
+function resolveCost(toolName) {
+    // Service catalog mapping — recognised tools get "estimated" confidence
+    // since we know the service but not exact usage-based cost from this call.
+    const catalogKey = MCP_TOOL_MAP[toolName];
+    if (catalogKey) {
+        return { costUsd: 0, costConfidence: "estimated", pricingSource: "service_catalog" };
+    }
+    // Unknown tool
+    return { costUsd: 0, costConfidence: "unknown", pricingSource: "unknown" };
+}
+// Self-register so the instrument registry can discover us.
+registerInstrument("mcp", instrumentMcp, uninstrumentMcp);
+//# sourceMappingURL=mcp.js.map

package/dist/instruments/mcp.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mcp.js","sourceRoot":"","sources":["../../src/instruments/mcp.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,EAAE,UAAU,EAAE,MAAM,aAAa,CAAC;AACzC,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,sBAAsB,CAAC;AAGtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAEhD,uDAAuD;AAEvD,IAAI,QAAQ,GAAG,KAAK,CAAC;AACrB,sEAAsE;AACtE,IAAI,SAAS,GAAoB,IAAI,CAAC;AACtC,IAAI,YAAY,GAAQ,IAAI,CAAC;AAC7B,IAAI,OAAO,GAAuB,IAAI,CAAC;AAEvC,kGAAkG;AAClG,MAAM,UAAU,eAAe,CAAC,GAAQ;IACtC,YAAY,GAAG,GAAG,CAAC;AACrB,CAAC;AAED,oDAAoD;AACpD,MAAM,UAAU,iBAAiB;IAC/B,YAAY,GAAG,IAAI,CAAC;AACtB,CAAC;AAED,8EAA8E;AAC9E,8CAA8C;AAC9C,EAAE;AACF,0EAA0E;AAC1E,8EAA8E;AAE9E,MAAM,YAAY,GAA2B;IAC3C,SAAS;IACT,aAAa,EAAE,eAAe;IAC9B,cAAc,EAAE,eAAe;IAC/B,YAAY,EAAE,eAAe;IAC7B,UAAU,EAAE,eAAe;IAC3B,eAAe,EAAE,eAAe;IAChC,aAAa,EAAE,eAAe;IAC9B,oBAAoB,EAAE,eAAe;IACrC,UAAU,EAAE,YAAY;IACxB,gBAAgB,EAAE,YAAY;IAC9B,gBAAgB,EAAE,YAAY;IAC9B,iBAAiB,EAAE,mBAAmB;IACtC,YAAY,EAAE,mBAAmB;IACjC,gBAAgB,EAAE,cAAc;IAChC,kBAAkB,EAAE,cAAc;IAClC,YAAY,EAAE,cAAc;IAC5B,WAAW,EAAE,aAAa;IAC1B,eAAe,EAAE,aAAa;IAC9B,cAAc,EAAE,SAAS;IACzB,qBAAqB,EAAE,SAAS;IAChC,WAAW;IACX,gBAAgB,EAAE,kBAAkB;IACpC,eAAe,EAAE,kBAAkB;IACnC,aAAa,EAAE,kBAAkB;IACjC,gBAAgB,EAAE,kBAAkB;IACpC,iBAAiB,EAAE,kBAAkB;IACrC,0BAA0B,EAAE,aAAa;IACzC,oBAAoB,EAAE,aAAa;IACnC,sBAAsB,EAAE,aAAa;IACrC,uBAAuB,EAAE,aAAa;IACtC,kBAAkB,EAAE,aAAa;IACjC,eAAe,EAAE,OAAO;IACxB,iBAAiB,EAAE,OAAO;IAC1B,kBAAkB,EAAE,aAAa;IACjC,eAAe,EAAE,SAAS;IAC1B,eAAe,EAAE,SAAS;IAC1B,SAAS,EAAE,aAAa;IACxB,WAAW,EAAE,aAAa;IAC1B,WAAW,EAAE,aAAa;IAC1B,aAAa;IACb,cAAc,EAAE,gBAAgB;IAChC,eAAe,EAAE,gBAAgB;IACjC,eAAe,EAAE,gBAAgB;IACjC,cAAc,EAAE,gBAAgB;IAChC,cAAc,EAAE,gBAAgB;IAChC,eAAe,EAAE,gBAAgB;IACjC,aAAa,EAAE,cAAc;IAC7B,YAAY,EAAE,cAAc;IAC5B,aAAa,EAAE,cAAc;IAC7B,aAAa,EAAE,eAAe;IAC9B,YAAY,EAAE,eAAe;IAC7B,WAAW,EAAE,UAAU;IACvB,YAAY,EAAE,UAAU;IACxB,oBAAoB;IACpB,YAAY,EAAE,aAAa;IAC3B,WAAW,EAAE,aAAa;IAC1B,kBAAkB,EAAE,aAAa;IACjC,YAAY,EAAE,aAAa;IAC3B,SAAS,EAAE,eAAe;IAC1B,YAAY,EAAE,eAAe;IAC7B,aAAa,EAAE,YAAY;IAC3B,gBAAgB;IAChB,eAAe,EAAE,YAAY;IAC7B,mBAAmB,EAAE,YAAY;IACjC,mBAAmB,EAAE,gBAAgB;IACrC,UAAU,EAAE,gBAAgB;IAC5B,iBAAiB,EAAE,cAAc;IACjC,YAAY,EAAE,SAAS;IACvB,aAAa,EAAE,UAAU;IACzB,OAAO;IACP,mBAAmB,EAAE,qBAAqB;IAC1C,kBAAkB,EAAE,oBAAoB;IACxC,sBAAsB,EAAE,wBAAwB;IAChD,kBAAkB,EAAE,oBAAoB;IACxC,cAAc,EAAE,kBAAkB;IAClC,iBAAiB,EAAE,kBAAkB;IACrC,kBAAkB;IAClB,aAAa,EAAE,kBAAkB;IACjC,UAAU,EAAE,kBAAkB;IAC9B,UAAU,EAAE,kBAAkB;IAC9B,eAAe,EAAE,qBAAqB;IACtC,eAAe,EAAE,qBAAqB;IACtC,aAAa,EAAE,WAAW;IAC1B,mBAAmB,EAAE,WAAW;IAChC,iBAAiB,EAAE,YAAY;IAC/B,iBAAiB,EAAE,YAAY;IAC/B,WAAW;IACX,oBAAoB,EAAE,gBAAgB;IACtC,qBAAqB,EAAE,gBAAgB;IACvC,0BAA0B,EAAE,gBAAgB;IAC5C,mBAAmB,EAAE,gBAAgB;IACrC,iBAAiB;IACjB,cAAc,EAAE,aAAa;IAC7B,yBAAyB,EAAE,aAAa;IACxC,mBAAmB,EAAE,wBAAwB;IAC7C,qBAAqB,EAAE,YAAY;IACnC,kBAAkB,EAAE,gBAAgB;IACpC,mBAAmB;IACnB,cAAc,EAAE,cAAc;IAC9B,qBAAqB,EAAE,cAAc;IACrC,kBAAkB,EAAE,cAAc;IAClC,aAAa,EAAE,WAAW;IAC1B,iBAAiB,EAAE,WAAW;IAC9B,sBAAsB;IACtB,gBAAgB,EAAE,iBAAiB;IACnC,eAAe,EAAE,iBAAiB;IAClC,mBAAmB,EAAE,oBAAoB;IACzC,kBAAkB,EAAE,iBAAiB;IACrC,gBAAgB,EAAE,YAAY;IAC9B,iBAAiB,EAAE,YAAY;IAC/B,iBAAiB;IACjB,iBAAiB,EAAE,aAAa;IAChC,uBAAuB,EAAE,aAAa;IACtC,mBAAmB,EAAE,eAAe;IACpC,aAAa,EAAE,YAAY;IAC3B,YAAY,EAAE,YAAY;IAC1B,qBAAqB,EAAE,SAAS;IAChC,gBAAgB;IAChB,aAAa,EAAE,QAAQ;IACvB,aAAa,EAAE,QAAQ;IACvB,cAAc,EAAE,UAAU;IAC1B,eAAe,EAAE,UAAU;IAC3B,eAAe,EAAE,UAAU;IAC3B,cAAc;IACd,gBAAgB,EAAE,UAAU;IAC5B,eAAe,EAAE,UAAU;IAC3B,aAAa,EAAE,eAAe;IAC9B,mBAAmB,EAAE,eAAe;IACpC,aAAa;IACb,YAAY,EAAE,mBAAmB;IACjC,YAAY,EAAE,cAAc;IAC5B,YAAY,EAAE,WAAW;IACzB,UAAU,EAAE,iBAAiB;IAC7B,gBAAgB,EAAE,kBAAkB;IACpC,WAAW,EAAE,aAAa;IAC1B,cAAc;IACd,eAAe,EAAE,iBAAiB;IAClC,cAAc,EAAE,iBAAiB;IACjC,gBAAgB,EAAE,kBAAkB;IACpC,aAAa,EAAE,eAAe;IAC9B,eAAe,EAAE,kBAAkB;IACnC,eAAe;IACf,sBAAsB,EAAE,eAAe;IACvC,iBAAiB,EAAE,eAAe;IAClC,oBAAoB,EAAE,uBAAuB;IAC7C,SAAS,EAAE,uBAAuB;IAClC,sBAAsB,EAAE,iBAAiB;IACzC,sBAAsB,EAAE,iBAAiB;IACzC,eAAe,EAAE,aAAa;IAC9B,WAAW;IACX,eAAe,EAAE,cAAc;IAC/B,mBAAmB,EAAE,cAAc;IACnC,aAAa,EAAE,YAAY;IAC3B,UAAU,EAAE,WAAW;IACvB,gBAAgB,EAAE,WAAW;IAC7B,2BAA2B;IAC3B,kBAAkB,EAAE,WAAW;IAC/B,kBAAkB,EAAE,WAAW;IAC/B,eAAe,EAAE,WAAW;IAC5B,oBAAoB,EAAE,aAAa;IACnC,sBAAsB,EAAE,aAAa;IACrC,qBAAqB,EAAE,cAAc;IACrC,aAAa,EAAE,cAAc;IAC7B,WAAW,EAAE,cAAc;IAC3B,gBAAgB,EAAE,cAAc;IAChC,cAAc,EAAE,WAAW;IAC3B,cAAc,EAAE,QAAQ;IACxB,YAAY,EAAE,MAAM;IACpB,WAAW;IACX,UAAU,EAAE,eAAe;IAC3B,QAAQ,EAAE,eAAe;IACzB,iBAAiB,EAAE,aAAa;IAChC,WAAW,EAAE,UAAU;IACvB,aAAa,EAAE,UAAU;IACzB,YAAY,EAAE,eAAe;IAC7B,aAAa,EAAE,eAAe;IAC9B,cAAc,EAAE,eAAe;IAC/B,iBAAiB,EAAE,eAAe;IAClC,iBAAiB,EAAE,eAAe;IAClC,qBAAqB,EAAE,eAAe;IACtC,WAAW,EAAE,UAAU;IACvB,eAAe,EAAE,aAAa;IAC9B,qBAAqB;IACrB,iBAAiB,EAAE,UAAU;IAC7B,WAAW,EAAE,UAAU;IACvB,iBAAiB,EAAE,UAAU;IAC7B,cAAc,EAAE,UAAU;IAC1B,mBAAmB,EAAE,YAAY;IACjC,aAAa,EAAE,YAAY;IAC3B,mBAAmB,EAAE,YAAY;IACjC,iBAAiB,EAAE,WAAW;IAC9B,YAAY,EAAE,WAAW;IACzB,mBAAmB,EAAE,aAAa;IAClC,cAAc,EAAE,aAAa;IAC7B,YAAY,EAAE,YAAY;IAC1B,kBAAkB,EAAE,YAAY;IAChC,aAAa,EAAE,YAAY;IAC3B,kBAAkB,EAAE,YAAY;IAChC,MAAM;IACN,gBAAgB,EAAE,gBAAgB;IAClC,iBAAiB,EAAE,gBAAgB;IACnC,eAAe,EAAE,gBAAgB;IACjC,sBAAsB,EAAE,aAAa;IACrC,cAAc,EAAE,aAAa;IAC7B,oBAAoB,EAAE,aAAa;IACnC,gBAAgB;IAChB,mBAAmB,EAAE,YAAY;IACjC,aAAa,EAAE,YAAY;IAC3B,gBAAgB,EAAE,YAAY;IAC9B,eAAe,EAAE,YAAY;IAC7B,iBAAiB,EAAE,YAAY;IAC/B,mBAAmB,EAAE,YAAY;IACjC,aAAa,EAAE,YAAY;IAC3B,aAAa,EAAE,YAAY;IAC3B,uBAAuB,EAAE,YAAY;IACrC,eAAe;IACf,YAAY,EAAE,aAAa;IAC3B,cAAc,EAAE,aAAa;IAC7B,YAAY,EAAE,aAAa;IAC3B,aAAa,EAAE,cAAc;IAC7B,cAAc,EAAE,cAAc;IAC9B,aAAa,EAAE,YAAY;IAC3B,aAAa,EAAE,YAAY;IAC3B,cAAc,EAAE,kBAAkB;IAClC,mBAAmB,EAAE,kBAAkB;IACvC,uBAAuB;IACvB,sBAAsB,EAAE,gBAAgB;IACxC,uBAAuB,EAAE,gBAAgB;IACzC,WAAW,EAAE,gBAAgB;IAC7B,gBAAgB,EAAE,YAAY;IAC9B,QAAQ,EAAE,SAAS;IACnB,WAAW,EAAE,SAAS;IACtB,SAAS,EAAE,QAAQ;IACnB,SAAS,EAAE,eAAe;IAC1B,sBAAsB;IACtB,UAAU,EAAE,QAAQ;IACpB,aAAa,EAAE,QAAQ;IACvB,YAAY,EAAE,iBAAiB;IAC/B,iBAAiB,EAAE,WAAW;IAC9B,uBAAuB;IACvB,qBAAqB,EAAE,QAAQ;IAC/B,cAAc,EAAE,QAAQ;IACxB,qBAAqB,EAAE,UAAU;IACjC,wBAAwB,EAAE,UAAU;IACpC,sBAAsB,EAAE,aAAa;IACrC,kBAAkB,EAAE,aAAa;IACjC,iBAAiB;IACjB,iBAAiB,EAAE,YAAY;IAC/B,cAAc,EAAE,SAAS;IACzB,cAAc,EAAE,iBAAiB;IACjC,kBAAkB;IAClB,qBAAqB,EAAE,uBAAuB;IAC9C,WAAW,EAAE,uBAAuB;IACpC,eAAe,EAAE,UAAU;IAC3B,iBAAiB,EAAE,UAAU;IAC7B,iBAAiB,EAAE,aAAa;IAChC,SAAS,EAAE,gBAAgB;IAC3B,kBAAkB,EAAE,cAAc;IAClC,aAAa,EAAE,eAAe;IAC9B,WAAW,EAAE,eAAe;IAC5B,MAAM,EAAE,eAAe;IACvB,sBAAsB;IACtB,UAAU,EAAE,KAAK;IACjB,YAAY,EAAE,KAAK;IACnB,aAAa,EAAE,eAAe;IAC9B,aAAa,EAAE,eAAe;IAC9B,iBAAiB,EAAE,YAAY;IAC/B,iBAAiB;IACjB,iBAAiB,EAAE,oBAAoB;IACvC,iBAAiB,EAAE,WAAW;IAC9B,eAAe;IACf,UAAU,EAAE,YAAY;IACxB,qBAAqB,EAAE,YAAY;IACnC,iBAAiB,EAAE,aAAa;IAChC,eAAe,EAAE,QAAQ;IACzB,cAAc;IACd,mBAAmB,EAAE,gBAAgB;IACrC,YAAY,EAAE,UAAU;IACxB,aAAa,EAAE,UAAU;IACzB,iBAAiB,EAAE,aAAa;IAChC,wBAAwB;IACxB,aAAa,EAAE,WAAW;IAC1B,aAAa,EAAE,WAAW;IAC1B,eAAe,EAAE,cAAc;IAC/B,yBAAyB;IACzB,aAAa,EAAE,UAAU;IACzB,wBAAwB,EAAE,UAAU;IACpC,qBAAqB,EAAE,cAAc;IACrC,aAAa;IACb,cAAc,EAAE,QAAQ;IACxB,cAAc,EAAE,SAAS;IACzB,eAAe,EAAE,SAAS;IAC1B,cAAc,EAAE,UAAU;IAC1B,aAAa,EAAE,SAAS;IACxB,cAAc;IACd,iBAAiB,EAAE,YAAY;IAC/B,oBAAoB,EAAE,YAAY;IAClC,gBAAgB;IAChB,eAAe,EAAE,cAAc;IAC/B,qBAAqB,EAAE,oBAAoB;IAC3C,kBAAkB;IAClB,YAAY,EAAE,cAAc;IAC5B,UAAU,EAAE,cAAc;IAC1B,iBAAiB,EAAE,aAAa;IAChC,kBAAkB,EAAE,aAAa;IACjC,oBAAoB;IACpB,oBAAoB,EAAE,aAAa;IACnC,sBAAsB,EAAE,aAAa;IACrC,kBAAkB,EAAE,aAAa;CAClC,CAAC;AAEF,8EAA8E;AAC9E,uBAAuB;AACvB,8EAA8E;AAE9E;;;;;GAKG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,QAAuB,EACvB,MAAmB;IAEnB,IAAI,QAAQ;QAAE,OAAO;IAErB,IAAI,WAAgB,CAAC;IACrB,IAAI,YAAY,EAAE,CAAC;QACjB,WAAW,GAAG,YAAY,CAAC,SAAS,CAAC;IACvC,CAAC;SAAM,CAAC;QACN,sEAAsE;QACtE,MAAM,MAAM,GAAG,MAAM,MAAM,CAAC,2CAA2C,CAAC,CAAC;QACzE,MAAM,MAAM,GAAG,MAAM,CAAC,MAAM,CAAC;QAC7B,WAAW,GAAG,MAAM,CAAC,SAAS,CAAC;IACjC,CAAC;IAED,SAAS,GAAG,WAAW,CAAC,QAAQ,CAAC;IACjC,OAAO,GAAG,MAAM,CAAC;IAEjB,WAAW,CAAC,QAAQ,GAAG,KAAK,WAE1B,MAAW,EACX,GAAG,IAAW;QAEd,IAAI,IAAI,GAAG,cAAc,EAAE,CAAC;QAE5B,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,IAAI,GAAG,cAAc,CAAC,eAAe,CAAC,CAAC;YACvC,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC;QAC5B,CAAC;QAED,MAAM,QAAQ,GAAW,MAAM,EAAE,IAAI,IAAI,SAAS,CAAC;QACnD,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,OAAO,GAAG,KAAK,CAAC;QAEpB,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,SAAU,CAAC,IAAI,CAAC,IAAI,EAAE,MAAM,EAAE,GAAG,IAAI,CAAC,CAAC;YAC5D,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAE5D,8CAA8C;YAC9C,IAAI,MAAM,EAAE,OAAO,EAAE,CAAC;gBACpB,OAAO,GAAG,IAAI,CAAC;YACjB,CAAC;YAED,IAAI,CAAC;gBACH,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YAC3D,CAAC;YAAC,MAAM,CAAC;gBACP,4CAA4C;YAC9C,CAAC;YACD,OAAO,MAAM,CAAC;QAChB,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,OAAO,GAAG,IAAI,CAAC;YACf,MAAM,SAAS,GAAG,IAAI,CAAC,KAAK,CAAC,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,CAAC;YAC5D,IAAI,CAAC;gBACH,cAAc,CAAC,QAAQ,EAAE,IAAI,EAAE,SAAS,EAAE,OAAO,EAAE,IAAI,CAAC,CAAC;YAC3D,CAAC;YAAC,MAAM,CAAC;gBACP,4CAA4C;YAC9C,CAAC;YACD,MAAM,GAAG,CAAC;QACZ,CAAC;IACH,CAAC,CAAC;IAEF,QAAQ,GAAG,IAAI,CAAC;AAClB,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,eAAe;IAC7B,IAAI,CAAC,QAAQ,IAAI,CAAC,SAAS;QAAE,OAAO;IAEpC,IAAI,YAAY,EAAE,CAAC;QACjB,YAAY,CAAC,SAAS,CAAC,QAAQ,GAAG,SAAS,CAAC;IAC9C,CAAC;IAED,SAAS,GAAG,IAAI,CAAC;IACjB,OAAO,GAAG,IAAI,CAAC;IACf,QAAQ,GAAG,KAAK,CAAC;AACnB,CAAC;AAED,8EAA8E;AAC9E,mBAAmB;AACnB,8EAA8E;AAE9E,SAAS,cAAc,CACrB,QAAgB,EAChB,IAAU,EACV,SAAiB,EACjB,OAAgB,EAChB,QAAa;IAEb,IAAI,CAAC,OAAO;QAAE,OAAO;IAErB,MAAM,EAAE,OAAO,EAAE,cAAc,EAAE,aAAa,EAAE,GAAG,WAAW,CAAC,QAAQ,CAAC,CAAC;IACzE,MAAM,SAAS,GAAW,QAAQ,EAAE,WAAW,IAAI,SAAS,CAAC;IAE7D,MAAM,KAAK,GAAG,eAAe,CAAC;QAC5B,OAAO,EAAE,UAAU,EAAE;QACrB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,SAAS,EAAE,eAAe;QAC1B,OAAO;QACP,cAAc;QACd,aAAa;QACb,WAAW,EAAE,OAAO,QAAQ,EAAE;QAC9B,SAAS;QACT,OAAO,EAAE,KAAK;QACd,OAAO,EAAE;YACP,QAAQ,EAAE,QAAQ;YAClB,UAAU,EAAE,SAAS;YACrB,UAAU,EAAE,SAAS;YACrB,QAAQ,EAAE,OAAO;SAClB;KACF,CAAC,CAAC;IAEH,OAAO,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;IAExB,IAAI,CAAC,eAAe,IAAI,OAAO,CAAC;IAChC,IAAI,CAAC,YAAY,IAAI,OAAO,CAAC;IAC7B,OAAO,CAAC,UAAU,CAAC,IAAI,CAAC,CAAC;AAC3B,CAAC;AAED,SAAS,WAAW,CAAC,QAAgB;IAKnC,wEAAwE;IACxE,2EAA2E;IAC3E,MAAM,UAAU,GAAG,YAAY,CAAC,QAAQ,CAAC,CAAC;IAC1C,IAAI,UAAU,EAAE,CAAC;QACf,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,cAAc,EAAE,WAAW,EAAE,aAAa,EAAE,iBAAiB,EAAE,CAAC;IACvF,CAAC;IAED,eAAe;IACf,OAAO,EAAE,OAAO,EAAE,CAAC,EAAE,cAAc,EAAE,SAAS,EAAE,aAAa,EAAE,SAAS,EAAE,CAAC;AAC7E,CAAC;AAED,4DAA4D;AAC5D,kBAAkB,CAAC,KAAK,EAAE,aAAa,EAAE,eAAe,CAAC,CAAC"}

package/dist/instruments/openai.d.ts

@@ -0,0 +1,26 @@
+/**
+ * OpenAI auto-instrumentation for dexcost TypeScript SDK.
+ *
+ * Monkey-patches `OpenAI.Chat.Completions.prototype.create` to automatically
+ * record cost events and aggregate token usage on the active task context.
+ *
+ * Supports both non-streaming and streaming responses.
+ */
+import type { EventBuffer } from "../transport/buffer.js";
+import type { PricingEngine } from "../pricing/engine.js";
+/** Test helper: inject a mock Completions class so tests avoid importing openai. */
+export declare function _setCompletionsClass(cls: any): void;
+/** Test helper: reset to real module resolution. */
+export declare function _resetCompletionsClass(): void;
+/**
+ * Patch `OpenAI.Chat.Completions.prototype.create` to record cost events.
+ *
+ * If `openai` is not installed and no mock class is injected, the dynamic
+ * import will throw and the function will reject.
+ */
+export declare function instrumentOpenai(pricing: PricingEngine, buffer: EventBuffer): Promise<void>;
+/**
+ * Remove the monkey-patch and restore the original `create` method.
+ */
+export declare function uninstrumentOpenai(): void;
+//# sourceMappingURL=openai.d.ts.map

package/dist/instruments/openai.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.d.ts","sourceRoot":"","sources":["../../src/instruments/openai.ts"],"names":[],"mappings":"AAAA;;;;;;;GAOG;AAOH,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,wBAAwB,CAAC;AAC1D,OAAO,KAAK,EAAE,aAAa,EAAc,MAAM,sBAAsB,CAAC;AAYtE,oFAAoF;AACpF,wBAAgB,oBAAoB,CAAC,GAAG,EAAE,GAAG,GAAG,IAAI,CAEnD;AAED,oDAAoD;AACpD,wBAAgB,sBAAsB,IAAI,IAAI,CAE7C;AAED;;;;;GAKG;AACH,wBAAsB,gBAAgB,CACpC,OAAO,EAAE,aAAa,EACtB,MAAM,EAAE,WAAW,GAClB,OAAO,CAAC,IAAI,CAAC,CAmDf;AAED;;GAEG;AACH,wBAAgB,kBAAkB,IAAI,IAAI,CAWzC"}