@granular-software/sdk 0.4.50 → 0.4.51

package/dist/spend.mjs

@@ -50,7 +50,12 @@ async function recordOpenAIUsageSpend(options) {
   const metadata = {
     ...options.metadata || {},
     ...options.usage.rawUsage !== void 0 ? { openaiUsage: options.usage.rawUsage } : {},
-    usageContext: context
+    usageContext: context,
+    pricingContextTier: options.usage.pricingContextTier,
+    cacheWritePricePerMillionMicros: options.usage.cacheWritePricePerMillionMicros,
+    cacheWriteTokens: options.usage.cacheWriteTokens,
+    cacheWriteCostMicros: options.usage.cacheWriteCostMicros,
+    longContextThresholdTokens: options.usage.longContextThresholdTokens
   };
   const response = await fetch(
     `${toGranularHttpBase(options.apiUrl)}/control/spend/events`,

package/dist/spend.mjs.map

@@ -1 +1 @@
-{"version":3,"sources":["../src/spend.ts"],"names":[],"mappings":";AAqCO,SAAS,mBAAmB,MAAA,EAAwB;AACzD,EAAA,MAAM,GAAA,GAAM,IAAI,GAAA,CAAI,MAAM,CAAA;AAC1B,EAAA,IAAI,GAAA,CAAI,aAAa,KAAA,EAAO;AAC1B,IAAA,GAAA,CAAI,QAAA,GAAW,OAAA;AAAA,EACjB,CAAA,MAAA,IAAW,GAAA,CAAI,QAAA,KAAa,MAAA,EAAQ;AAClC,IAAA,GAAA,CAAI,QAAA,GAAW,QAAA;AAAA,EACjB;AAEA,EAAA,GAAA,CAAI,QAAA,GAAW,IAAI,QAAA,CAChB,OAAA,CAAQ,kBAAkB,EAAE,CAAA,CAC5B,OAAA,CAAQ,OAAA,EAAS,EAAE,CAAA;AACtB,EAAA,IAAI,CAAC,GAAA,CAAI,QAAA,IAAY,GAAA,CAAI,aAAa,GAAA,EAAK;AACzC,IAAA,GAAA,CAAI,QAAA,GAAW,WAAA;AAAA,EACjB;AACA,EAAA,GAAA,CAAI,MAAA,GAAS,EAAA;AACb,EAAA,GAAA,CAAI,IAAA,GAAO,EAAA;AACX,EAAA,OAAO,GAAA,CAAI,QAAA,EAAS,CAAE,OAAA,CAAQ,OAAO,EAAE,CAAA;AACzC;AAEA,SAAS,YAAY,KAAA,EAAuB;AAC1C,EAAA,OAAO,MAAM,OAAA,CAAQ,kBAAA,EAAoB,GAAG,CAAA,CAAE,OAAA,CAAQ,YAAY,EAAE,CAAA;AACtE;AAEO,SAAS,uBAAA,CACd,KAAA,EACA,OAAA,GAAgC,EAAC,EACb;AACpB,EAAA,MAAM,SAAA,GAAY,KAAA,CAAM,SAAA,EAAW,IAAA,EAAK;AACxC,EAAA,IAAI,CAAC,WAAW,OAAO,MAAA;AAEvB,EAAA,MAAM,KAAA,GACJ,QAAQ,SAAA,IACR,OAAA,CAAQ,iBACR,OAAA,CAAQ,SAAA,IACR,QAAQ,SAAA,IACR,QAAA;AACF,EAAA,OAAO,CAAC,OAAA,EAAS,QAAA,EAAU,KAAA,EAAO,SAAS,EAAE,GAAA,CAAI,WAAW,CAAA,CAAE,IAAA,CAAK,GAAG,CAAA;AACxE;AAEA,SAAS,0BAA0B,KAAA,EAAkC;AACnE,EAAA,IAAI,CAAC,OAAO,OAAO,IAAA;AACnB,EAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA;AAC/B,EAAA,OAAO,MAAA,CAAO,SAAS,MAAM,CAAA,GAAI,KAAK,KAAA,CAAM,MAAA,GAAS,GAAI,CAAA,GAAI,IAAA;AAC/D;AAEA,SAAS,eAAe,OAAA,EAAqD;AAC3E,EAAA,OAAO,MAAA,CAAO,WAAA;AAAA,IACZ,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,CAAE,MAAA;AAAA,MACtB,CAAC,GAAG,KAAK,CAAA,KAAM,KAAA,IAAS,QAAQ,KAAA,KAAU;AAAA;AAC5C,GACF;AACF;AAEA,SAAS,aAAa,OAAA,EAAqD;AACzE,EAAA,MAAM,aAAA,GAAgB,EAAE,GAAG,OAAA,EAAQ;AACnC,EAAA,OAAO,aAAA,CAAc,QAAA;AACrB,EAAA,OAAO,aAAA;AACT;AAEA,eAAsB,uBACpB,OAAA,EACuC;AACvC,EAAA,MAAM,eAAe,cAAA,CAAe;AAAA,IAClC,GAAI,OAAA,CAAQ,KAAA,CAAM,YAAA,IAAgB,EAAC;AAAA,IACnC,GAAI,OAAA,CAAQ,OAAA,IAAW;AAAC,GACzB,CAAA;AACD,EAAA,MAAM,OAAA,GAAU,aAAa,YAAY,CAAA;AACzC,EAAA,MAAM,eACJ,OAAA,CAAQ,KAAA,CAAM,gBACd,uBAAA,CAAwB,OAAA,CAAQ,OAAO,OAAO,CAAA;AAChD,EAAA,MAAM,QAAA,GAAW;AAAA,IACf,GAAI,OAAA,CAAQ,QAAA,IAAY,EAAC;AAAA,IACzB,GAAI,OAAA,CAAQ,KAAA,CAAM,QAAA,KAAa,MAAA,GAC3B,EAAE,WAAA,EAAa,OAAA,CAAQ,KAAA,CAAM,QAAA,EAAS,GACtC,EAAC;AAAA,IACL,YAAA,EAAc;AAAA,GAChB;AAEA,EAAA,MAAM,WAAW,MAAM,KAAA;AAAA,IACrB,CAAA,EAAG,kBAAA,CAAmB,OAAA,CAAQ,MAAM,CAAC,CAAA,qBAAA,CAAA;AAAA,IACrC;AAAA,MACE,MAAA,EAAQ,MAAA;AAAA,MACR,KAAA,EAAO,UAAA;AAAA,MACP,OAAA,EAAS;AAAA,QACP,aAAA,EAAe,CAAA,OAAA,EAAU,OAAA,CAAQ,KAAK,CAAA,CAAA;AAAA,QACtC,cAAA,EAAgB;AAAA,OAClB;AAAA,MACA,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,QACnB,GAAI,YAAA,GAAe,EAAE,YAAA,KAAiB,EAAC;AAAA,QACvC,SAAA,EAAW,QAAQ,SAAA,IAAa,IAAA;AAAA,QAChC,aAAA,EAAe,QAAQ,aAAA,IAAiB,IAAA;AAAA,QACxC,SAAA,EAAW,QAAQ,SAAA,IAAa,IAAA;AAAA,QAChC,SAAA,EAAW,QAAQ,SAAA,IAAa,IAAA;AAAA,QAChC,mBAAA,EAAqB,QAAQ,mBAAA,IAAuB,IAAA;AAAA,QACpD,MAAA,EAAQ,QAAA;AAAA,QACR,YAAA,EAAc,YAAA;AAAA,QACd,QAAA,EAAU,QAAQ,KAAA,CAAM,QAAA;AAAA,QACxB,KAAA,EAAO,QAAQ,KAAA,CAAM,KAAA;AAAA,QACrB,SAAA,EAAW,OAAA,CAAQ,KAAA,CAAM,SAAA,IAAa,kBAAA;AAAA,QACtC,SAAA,EAAW,OAAA,CAAQ,KAAA,CAAM,SAAA,IAAa,IAAA;AAAA,QACtC,WAAA,EAAa,QAAQ,KAAA,CAAM,WAAA;AAAA,QAC3B,YAAA,EAAc,QAAQ,KAAA,CAAM,YAAA;AAAA,QAC5B,iBAAA,EAAmB,QAAQ,KAAA,CAAM,iBAAA;AAAA,QACjC,eAAA,EAAiB,QAAQ,KAAA,CAAM,eAAA;AAAA,QAC/B,QAAA,EAAU,QAAQ,KAAA,CAAM,WAAA;AAAA,QACxB,YAAA,EAAc,QAAA;AAAA,QACd,0BAAA,EAA4B,QAAQ,KAAA,CAAM,0BAAA;AAAA,QAC1C,gCAAA,EACE,QAAQ,KAAA,CAAM,gCAAA;AAAA,QAChB,2BAAA,EAA6B,QAAQ,KAAA,CAAM,2BAAA;AAAA,QAC3C,YAAA,EAAc,QAAQ,KAAA,CAAM,YAAA;AAAA,QAC5B,QAAA,EAAU,QAAQ,KAAA,CAAM,QAAA;AAAA,QACxB,aAAA,EAAe,QAAQ,KAAA,CAAM,aAAA;AAAA,QAC7B,kBAAA,EAAoB,yBAAA;AAAA,UAClB,QAAQ,KAAA,CAAM;AAAA,SAChB;AAAA,QACA,SAAA,EAAW,KAAA;AAAA,QACX;AAAA,OACD;AAAA;AACH,GACF;AAEA,EAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,gCAAgC,QAAA,CAAS,MAAM,MAAM,MAAM,QAAA,CAAS,MAAM,CAAA;AAAA,KAC5E;AAAA,EACF;AAEA,EAAA,OAAO,SAAS,IAAA,EAAK;AACvB","file":"spend.mjs","sourcesContent":["import type { OpenAITokenSpend } from \"./openai-usage\";\nimport type { GranularQuotaProgress } from \"./types\";\n\nexport interface GranularSpendContext {\n  tenantId?: string | null;\n  sandboxId?: string | null;\n  environmentId?: string | null;\n  sessionId?: string | null;\n  subjectId?: string | null;\n  permissionProfileId?: string | null;\n  [key: string]: string | null | undefined;\n}\n\nexport interface OpenAIUsageSpendEvent extends OpenAITokenSpend {\n  spendEventId?: string;\n  source?: \"openai\";\n  lineItemType?: \"llm_tokens\";\n  operation?: string;\n  requestId?: string | null;\n  usageContext?: GranularSpendContext;\n  rawUsage?: unknown;\n}\n\nexport interface RecordOpenAIUsageSpendOptions {\n  apiUrl: string;\n  token: string;\n  usage: OpenAIUsageSpendEvent;\n  context?: GranularSpendContext | null;\n  metadata?: Record<string, unknown> | null;\n}\n\nexport interface RecordOpenAIUsageSpendResult {\n  spendEventId: string;\n  inserted: boolean;\n  quota: GranularQuotaProgress | null;\n}\n\nexport function toGranularHttpBase(apiUrl: string): string {\n  const url = new URL(apiUrl);\n  if (url.protocol === \"ws:\") {\n    url.protocol = \"http:\";\n  } else if (url.protocol === \"wss:\") {\n    url.protocol = \"https:\";\n  }\n\n  url.pathname = url.pathname\n    .replace(/\\/ws\\/connect$/, \"\")\n    .replace(/\\/ws$/, \"\");\n  if (!url.pathname || url.pathname === \"/\") {\n    url.pathname = \"/granular\";\n  }\n  url.search = \"\";\n  url.hash = \"\";\n  return url.toString().replace(/\\/$/, \"\");\n}\n\nfunction cleanIdPart(value: string): string {\n  return value.replace(/[^a-zA-Z0-9_-]+/g, \"_\").replace(/^_+|_+$/g, \"\");\n}\n\nexport function buildOpenAISpendEventId(\n  usage: Pick<OpenAIUsageSpendEvent, \"requestId\">,\n  context: GranularSpendContext = {},\n): string | undefined {\n  const requestId = usage.requestId?.trim();\n  if (!requestId) return undefined;\n\n  const scope =\n    context.sessionId ||\n    context.environmentId ||\n    context.subjectId ||\n    context.sandboxId ||\n    \"global\";\n  return [\"spend\", \"openai\", scope, requestId].map(cleanIdPart).join(\"_\");\n}\n\nfunction pricingEffectiveAtSeconds(value: string | null | undefined) {\n  if (!value) return null;\n  const parsed = Date.parse(value);\n  return Number.isFinite(parsed) ? Math.floor(parsed / 1000) : null;\n}\n\nfunction compactContext(context: GranularSpendContext): GranularSpendContext {\n  return Object.fromEntries(\n    Object.entries(context).filter(\n      ([, value]) => value != null && value !== \"\",\n    ),\n  ) as GranularSpendContext;\n}\n\nfunction omitTenantId(context: GranularSpendContext): GranularSpendContext {\n  const scopedContext = { ...context };\n  delete scopedContext.tenantId;\n  return scopedContext;\n}\n\nexport async function recordOpenAIUsageSpend(\n  options: RecordOpenAIUsageSpendOptions,\n): Promise<RecordOpenAIUsageSpendResult> {\n  const usageContext = compactContext({\n    ...(options.usage.usageContext || {}),\n    ...(options.context || {}),\n  });\n  const context = omitTenantId(usageContext);\n  const spendEventId =\n    options.usage.spendEventId ||\n    buildOpenAISpendEventId(options.usage, context);\n  const metadata = {\n    ...(options.metadata || {}),\n    ...(options.usage.rawUsage !== undefined\n      ? { openaiUsage: options.usage.rawUsage }\n      : {}),\n    usageContext: context,\n  };\n\n  const response = await fetch(\n    `${toGranularHttpBase(options.apiUrl)}/control/spend/events`,\n    {\n      method: \"POST\",\n      cache: \"no-store\",\n      headers: {\n        Authorization: `Bearer ${options.token}`,\n        \"Content-Type\": \"application/json\",\n      },\n      body: JSON.stringify({\n        ...(spendEventId ? { spendEventId } : {}),\n        sandboxId: context.sandboxId || null,\n        environmentId: context.environmentId || null,\n        sessionId: context.sessionId || null,\n        subjectId: context.subjectId || null,\n        permissionProfileId: context.permissionProfileId || null,\n        source: \"openai\",\n        lineItemType: \"llm_tokens\",\n        provider: options.usage.provider,\n        model: options.usage.model,\n        operation: options.usage.operation || \"chat.completions\",\n        requestId: options.usage.requestId || null,\n        inputTokens: options.usage.inputTokens,\n        outputTokens: options.usage.outputTokens,\n        cachedInputTokens: options.usage.cachedInputTokens,\n        reasoningTokens: options.usage.reasoningTokens,\n        quantity: options.usage.totalTokens,\n        quantityUnit: \"tokens\",\n        inputPricePerMillionMicros: options.usage.inputPricePerMillionMicros,\n        cachedInputPricePerMillionMicros:\n          options.usage.cachedInputPricePerMillionMicros,\n        outputPricePerMillionMicros: options.usage.outputPricePerMillionMicros,\n        amountMicros: options.usage.amountMicros,\n        currency: options.usage.currency,\n        pricingSource: options.usage.pricingSource,\n        pricingEffectiveAt: pricingEffectiveAtSeconds(\n          options.usage.pricingEffectiveAt,\n        ),\n        estimated: false,\n        metadata,\n      }),\n    },\n  );\n\n  if (!response.ok) {\n    throw new Error(\n      `Granular spend event failed (${response.status}): ${await response.text()}`,\n    );\n  }\n\n  return response.json() as Promise<RecordOpenAIUsageSpendResult>;\n}\n"]}
+{"version":3,"sources":["../src/spend.ts"],"names":[],"mappings":";AAqCO,SAAS,mBAAmB,MAAA,EAAwB;AACzD,EAAA,MAAM,GAAA,GAAM,IAAI,GAAA,CAAI,MAAM,CAAA;AAC1B,EAAA,IAAI,GAAA,CAAI,aAAa,KAAA,EAAO;AAC1B,IAAA,GAAA,CAAI,QAAA,GAAW,OAAA;AAAA,EACjB,CAAA,MAAA,IAAW,GAAA,CAAI,QAAA,KAAa,MAAA,EAAQ;AAClC,IAAA,GAAA,CAAI,QAAA,GAAW,QAAA;AAAA,EACjB;AAEA,EAAA,GAAA,CAAI,QAAA,GAAW,IAAI,QAAA,CAChB,OAAA,CAAQ,kBAAkB,EAAE,CAAA,CAC5B,OAAA,CAAQ,OAAA,EAAS,EAAE,CAAA;AACtB,EAAA,IAAI,CAAC,GAAA,CAAI,QAAA,IAAY,GAAA,CAAI,aAAa,GAAA,EAAK;AACzC,IAAA,GAAA,CAAI,QAAA,GAAW,WAAA;AAAA,EACjB;AACA,EAAA,GAAA,CAAI,MAAA,GAAS,EAAA;AACb,EAAA,GAAA,CAAI,IAAA,GAAO,EAAA;AACX,EAAA,OAAO,GAAA,CAAI,QAAA,EAAS,CAAE,OAAA,CAAQ,OAAO,EAAE,CAAA;AACzC;AAEA,SAAS,YAAY,KAAA,EAAuB;AAC1C,EAAA,OAAO,MAAM,OAAA,CAAQ,kBAAA,EAAoB,GAAG,CAAA,CAAE,OAAA,CAAQ,YAAY,EAAE,CAAA;AACtE;AAEO,SAAS,uBAAA,CACd,KAAA,EACA,OAAA,GAAgC,EAAC,EACb;AACpB,EAAA,MAAM,SAAA,GAAY,KAAA,CAAM,SAAA,EAAW,IAAA,EAAK;AACxC,EAAA,IAAI,CAAC,WAAW,OAAO,MAAA;AAEvB,EAAA,MAAM,KAAA,GACJ,QAAQ,SAAA,IACR,OAAA,CAAQ,iBACR,OAAA,CAAQ,SAAA,IACR,QAAQ,SAAA,IACR,QAAA;AACF,EAAA,OAAO,CAAC,OAAA,EAAS,QAAA,EAAU,KAAA,EAAO,SAAS,EAAE,GAAA,CAAI,WAAW,CAAA,CAAE,IAAA,CAAK,GAAG,CAAA;AACxE;AAEA,SAAS,0BAA0B,KAAA,EAAkC;AACnE,EAAA,IAAI,CAAC,OAAO,OAAO,IAAA;AACnB,EAAA,MAAM,MAAA,GAAS,IAAA,CAAK,KAAA,CAAM,KAAK,CAAA;AAC/B,EAAA,OAAO,MAAA,CAAO,SAAS,MAAM,CAAA,GAAI,KAAK,KAAA,CAAM,MAAA,GAAS,GAAI,CAAA,GAAI,IAAA;AAC/D;AAEA,SAAS,eAAe,OAAA,EAAqD;AAC3E,EAAA,OAAO,MAAA,CAAO,WAAA;AAAA,IACZ,MAAA,CAAO,OAAA,CAAQ,OAAO,CAAA,CAAE,MAAA;AAAA,MACtB,CAAC,GAAG,KAAK,CAAA,KAAM,KAAA,IAAS,QAAQ,KAAA,KAAU;AAAA;AAC5C,GACF;AACF;AAEA,SAAS,aAAa,OAAA,EAAqD;AACzE,EAAA,MAAM,aAAA,GAAgB,EAAE,GAAG,OAAA,EAAQ;AACnC,EAAA,OAAO,aAAA,CAAc,QAAA;AACrB,EAAA,OAAO,aAAA;AACT;AAEA,eAAsB,uBACpB,OAAA,EACuC;AACvC,EAAA,MAAM,eAAe,cAAA,CAAe;AAAA,IAClC,GAAI,OAAA,CAAQ,KAAA,CAAM,YAAA,IAAgB,EAAC;AAAA,IACnC,GAAI,OAAA,CAAQ,OAAA,IAAW;AAAC,GACzB,CAAA;AACD,EAAA,MAAM,OAAA,GAAU,aAAa,YAAY,CAAA;AACzC,EAAA,MAAM,eACJ,OAAA,CAAQ,KAAA,CAAM,gBACd,uBAAA,CAAwB,OAAA,CAAQ,OAAO,OAAO,CAAA;AAChD,EAAA,MAAM,QAAA,GAAW;AAAA,IACf,GAAI,OAAA,CAAQ,QAAA,IAAY,EAAC;AAAA,IACzB,GAAI,OAAA,CAAQ,KAAA,CAAM,QAAA,KAAa,MAAA,GAC3B,EAAE,WAAA,EAAa,OAAA,CAAQ,KAAA,CAAM,QAAA,EAAS,GACtC,EAAC;AAAA,IACL,YAAA,EAAc,OAAA;AAAA,IACd,kBAAA,EAAoB,QAAQ,KAAA,CAAM,kBAAA;AAAA,IAClC,+BAAA,EACE,QAAQ,KAAA,CAAM,+BAAA;AAAA,IAChB,gBAAA,EAAkB,QAAQ,KAAA,CAAM,gBAAA;AAAA,IAChC,oBAAA,EAAsB,QAAQ,KAAA,CAAM,oBAAA;AAAA,IACpC,0BAAA,EAA4B,QAAQ,KAAA,CAAM;AAAA,GAC5C;AAEA,EAAA,MAAM,WAAW,MAAM,KAAA;AAAA,IACrB,CAAA,EAAG,kBAAA,CAAmB,OAAA,CAAQ,MAAM,CAAC,CAAA,qBAAA,CAAA;AAAA,IACrC;AAAA,MACE,MAAA,EAAQ,MAAA;AAAA,MACR,KAAA,EAAO,UAAA;AAAA,MACP,OAAA,EAAS;AAAA,QACP,aAAA,EAAe,CAAA,OAAA,EAAU,OAAA,CAAQ,KAAK,CAAA,CAAA;AAAA,QACtC,cAAA,EAAgB;AAAA,OAClB;AAAA,MACA,IAAA,EAAM,KAAK,SAAA,CAAU;AAAA,QACnB,GAAI,YAAA,GAAe,EAAE,YAAA,KAAiB,EAAC;AAAA,QACvC,SAAA,EAAW,QAAQ,SAAA,IAAa,IAAA;AAAA,QAChC,aAAA,EAAe,QAAQ,aAAA,IAAiB,IAAA;AAAA,QACxC,SAAA,EAAW,QAAQ,SAAA,IAAa,IAAA;AAAA,QAChC,SAAA,EAAW,QAAQ,SAAA,IAAa,IAAA;AAAA,QAChC,mBAAA,EAAqB,QAAQ,mBAAA,IAAuB,IAAA;AAAA,QACpD,MAAA,EAAQ,QAAA;AAAA,QACR,YAAA,EAAc,YAAA;AAAA,QACd,QAAA,EAAU,QAAQ,KAAA,CAAM,QAAA;AAAA,QACxB,KAAA,EAAO,QAAQ,KAAA,CAAM,KAAA;AAAA,QACrB,SAAA,EAAW,OAAA,CAAQ,KAAA,CAAM,SAAA,IAAa,kBAAA;AAAA,QACtC,SAAA,EAAW,OAAA,CAAQ,KAAA,CAAM,SAAA,IAAa,IAAA;AAAA,QACtC,WAAA,EAAa,QAAQ,KAAA,CAAM,WAAA;AAAA,QAC3B,YAAA,EAAc,QAAQ,KAAA,CAAM,YAAA;AAAA,QAC5B,iBAAA,EAAmB,QAAQ,KAAA,CAAM,iBAAA;AAAA,QACjC,eAAA,EAAiB,QAAQ,KAAA,CAAM,eAAA;AAAA,QAC/B,QAAA,EAAU,QAAQ,KAAA,CAAM,WAAA;AAAA,QACxB,YAAA,EAAc,QAAA;AAAA,QACd,0BAAA,EAA4B,QAAQ,KAAA,CAAM,0BAAA;AAAA,QAC1C,gCAAA,EACE,QAAQ,KAAA,CAAM,gCAAA;AAAA,QAChB,2BAAA,EAA6B,QAAQ,KAAA,CAAM,2BAAA;AAAA,QAC3C,YAAA,EAAc,QAAQ,KAAA,CAAM,YAAA;AAAA,QAC5B,QAAA,EAAU,QAAQ,KAAA,CAAM,QAAA;AAAA,QACxB,aAAA,EAAe,QAAQ,KAAA,CAAM,aAAA;AAAA,QAC7B,kBAAA,EAAoB,yBAAA;AAAA,UAClB,QAAQ,KAAA,CAAM;AAAA,SAChB;AAAA,QACA,SAAA,EAAW,KAAA;AAAA,QACX;AAAA,OACD;AAAA;AACH,GACF;AAEA,EAAA,IAAI,CAAC,SAAS,EAAA,EAAI;AAChB,IAAA,MAAM,IAAI,KAAA;AAAA,MACR,gCAAgC,QAAA,CAAS,MAAM,MAAM,MAAM,QAAA,CAAS,MAAM,CAAA;AAAA,KAC5E;AAAA,EACF;AAEA,EAAA,OAAO,SAAS,IAAA,EAAK;AACvB","file":"spend.mjs","sourcesContent":["import type { OpenAITokenSpend } from \"./openai-usage\";\nimport type { GranularQuotaProgress } from \"./types\";\n\nexport interface GranularSpendContext {\n  tenantId?: string | null;\n  sandboxId?: string | null;\n  environmentId?: string | null;\n  sessionId?: string | null;\n  subjectId?: string | null;\n  permissionProfileId?: string | null;\n  [key: string]: string | null | undefined;\n}\n\nexport interface OpenAIUsageSpendEvent extends OpenAITokenSpend {\n  spendEventId?: string;\n  source?: \"openai\";\n  lineItemType?: \"llm_tokens\";\n  operation?: string;\n  requestId?: string | null;\n  usageContext?: GranularSpendContext;\n  rawUsage?: unknown;\n}\n\nexport interface RecordOpenAIUsageSpendOptions {\n  apiUrl: string;\n  token: string;\n  usage: OpenAIUsageSpendEvent;\n  context?: GranularSpendContext | null;\n  metadata?: Record<string, unknown> | null;\n}\n\nexport interface RecordOpenAIUsageSpendResult {\n  spendEventId: string;\n  inserted: boolean;\n  quota: GranularQuotaProgress | null;\n}\n\nexport function toGranularHttpBase(apiUrl: string): string {\n  const url = new URL(apiUrl);\n  if (url.protocol === \"ws:\") {\n    url.protocol = \"http:\";\n  } else if (url.protocol === \"wss:\") {\n    url.protocol = \"https:\";\n  }\n\n  url.pathname = url.pathname\n    .replace(/\\/ws\\/connect$/, \"\")\n    .replace(/\\/ws$/, \"\");\n  if (!url.pathname || url.pathname === \"/\") {\n    url.pathname = \"/granular\";\n  }\n  url.search = \"\";\n  url.hash = \"\";\n  return url.toString().replace(/\\/$/, \"\");\n}\n\nfunction cleanIdPart(value: string): string {\n  return value.replace(/[^a-zA-Z0-9_-]+/g, \"_\").replace(/^_+|_+$/g, \"\");\n}\n\nexport function buildOpenAISpendEventId(\n  usage: Pick<OpenAIUsageSpendEvent, \"requestId\">,\n  context: GranularSpendContext = {},\n): string | undefined {\n  const requestId = usage.requestId?.trim();\n  if (!requestId) return undefined;\n\n  const scope =\n    context.sessionId ||\n    context.environmentId ||\n    context.subjectId ||\n    context.sandboxId ||\n    \"global\";\n  return [\"spend\", \"openai\", scope, requestId].map(cleanIdPart).join(\"_\");\n}\n\nfunction pricingEffectiveAtSeconds(value: string | null | undefined) {\n  if (!value) return null;\n  const parsed = Date.parse(value);\n  return Number.isFinite(parsed) ? Math.floor(parsed / 1000) : null;\n}\n\nfunction compactContext(context: GranularSpendContext): GranularSpendContext {\n  return Object.fromEntries(\n    Object.entries(context).filter(\n      ([, value]) => value != null && value !== \"\",\n    ),\n  ) as GranularSpendContext;\n}\n\nfunction omitTenantId(context: GranularSpendContext): GranularSpendContext {\n  const scopedContext = { ...context };\n  delete scopedContext.tenantId;\n  return scopedContext;\n}\n\nexport async function recordOpenAIUsageSpend(\n  options: RecordOpenAIUsageSpendOptions,\n): Promise<RecordOpenAIUsageSpendResult> {\n  const usageContext = compactContext({\n    ...(options.usage.usageContext || {}),\n    ...(options.context || {}),\n  });\n  const context = omitTenantId(usageContext);\n  const spendEventId =\n    options.usage.spendEventId ||\n    buildOpenAISpendEventId(options.usage, context);\n  const metadata = {\n    ...(options.metadata || {}),\n    ...(options.usage.rawUsage !== undefined\n      ? { openaiUsage: options.usage.rawUsage }\n      : {}),\n    usageContext: context,\n    pricingContextTier: options.usage.pricingContextTier,\n    cacheWritePricePerMillionMicros:\n      options.usage.cacheWritePricePerMillionMicros,\n    cacheWriteTokens: options.usage.cacheWriteTokens,\n    cacheWriteCostMicros: options.usage.cacheWriteCostMicros,\n    longContextThresholdTokens: options.usage.longContextThresholdTokens,\n  };\n\n  const response = await fetch(\n    `${toGranularHttpBase(options.apiUrl)}/control/spend/events`,\n    {\n      method: \"POST\",\n      cache: \"no-store\",\n      headers: {\n        Authorization: `Bearer ${options.token}`,\n        \"Content-Type\": \"application/json\",\n      },\n      body: JSON.stringify({\n        ...(spendEventId ? { spendEventId } : {}),\n        sandboxId: context.sandboxId || null,\n        environmentId: context.environmentId || null,\n        sessionId: context.sessionId || null,\n        subjectId: context.subjectId || null,\n        permissionProfileId: context.permissionProfileId || null,\n        source: \"openai\",\n        lineItemType: \"llm_tokens\",\n        provider: options.usage.provider,\n        model: options.usage.model,\n        operation: options.usage.operation || \"chat.completions\",\n        requestId: options.usage.requestId || null,\n        inputTokens: options.usage.inputTokens,\n        outputTokens: options.usage.outputTokens,\n        cachedInputTokens: options.usage.cachedInputTokens,\n        reasoningTokens: options.usage.reasoningTokens,\n        quantity: options.usage.totalTokens,\n        quantityUnit: \"tokens\",\n        inputPricePerMillionMicros: options.usage.inputPricePerMillionMicros,\n        cachedInputPricePerMillionMicros:\n          options.usage.cachedInputPricePerMillionMicros,\n        outputPricePerMillionMicros: options.usage.outputPricePerMillionMicros,\n        amountMicros: options.usage.amountMicros,\n        currency: options.usage.currency,\n        pricingSource: options.usage.pricingSource,\n        pricingEffectiveAt: pricingEffectiveAtSeconds(\n          options.usage.pricingEffectiveAt,\n        ),\n        estimated: false,\n        metadata,\n      }),\n    },\n  );\n\n  if (!response.ok) {\n    throw new Error(\n      `Granular spend event failed (${response.status}): ${await response.text()}`,\n    );\n  }\n\n  return response.json() as Promise<RecordOpenAIUsageSpendResult>;\n}\n"]}

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@granular-software/sdk",
-  "version": "0.4.50",
+  "version": "0.4.51",
   "description": "TypeScript SDK and CLI for Granular - define, build, and deploy AI sandboxes",
   "main": "./dist/index.js",
   "module": "./dist/index.mjs",