cloudbroker-sdk 0.4.0 → 0.5.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cloudbroker-sdk",
-  "version": "0.4.0",
+  "version": "0.5.0",
   "description": "CloudBroker SDK — Token Intelligence & Operational Financial Intelligence",
   "main": "src/index.js",
   "type": "module",

package/src/client.js

@@ -37,7 +37,7 @@ export class CloudBrokerClient {
     try {
       const payload = {
         ...event,
-        sdkVersion: "0.4.0",
+        sdkVersion: "0.5.0",
         sentAt: new Date().toISOString(),
       };
       if (this.#cloudQueue.length < MAX_QUEUE_SIZE) {
@@ -93,7 +93,7 @@ export class CloudBrokerClient {
     try {
       const payload = {
         ...event,
-        sdkVersion: "0.4.0",
+        sdkVersion: "0.5.0",
         sentAt: new Date().toISOString(),
       };
 

package/src/test.js

@@ -173,5 +173,57 @@ assert.strictEqual(ctxEvent.product, "Billing",     `product do contexto esperad
 assert.strictEqual(ctxEvent.team,    "Plataforma",  `team do contexto esperado, veio ${ctxEvent.team}`);
 console.log("✅ expressMiddleware: label/product/team do contexto aplicados sem cb* explícito");
 
+// 4) CloudBrokerAI.wrap — label de código nas chamadas de IA (mesma
+// convenção do wrapAWS: explícito na chamada > contexto cbStore > nenhum),
+// pra agregar custo de IA junto com custo de nuvem na visão "Custo por Código".
+const capturedAi = [];
+globalThis.fetch = async (url, opts) => {
+  if (String(url).includes("/api/ai/events")) {
+    capturedAi.push(JSON.parse(opts.body));
+  }
+  return { ok: true, status: 200, json: async () => ({}) };
+};
+
+let sawOpenAiOptions;
+const mockOpenAI2 = {
+  chat: {
+    completions: {
+      create: async (params, options) => {
+        sawOpenAiOptions = options;
+        return {
+          model: params.model,
+          choices: [{ message: { role: "assistant", content: "ok" } }],
+          usage: { prompt_tokens: 10, completion_tokens: 5 },
+        };
+      },
+    },
+  },
+};
+const wrappedAi = CloudBrokerAI.wrap(mockOpenAI2, {
+  apiKey: "cb-test-key", product: "FinOps Advisor", team: "platform", debug: true,
+});
+
+await wrappedAi.chat.completions.create(
+  { model: "gpt-4o", messages: [{ role: "user", content: "x" }] },
+  { cbLabel: "finops-advisor:intent-classify" }
+);
+assert.strictEqual(sawOpenAiOptions, undefined, "cbLabel não pode vazar pro SDK do provider");
+await new Promise((r) => setTimeout(r, 50));
+assert.strictEqual(capturedAi.length, 1, `esperado 1 evento de IA, veio ${capturedAi.length}`);
+assert.strictEqual(capturedAi[0].events[0].label, "finops-advisor:intent-classify");
+console.log("✅ CloudBrokerAI.wrap: label explícito (cbLabel) capturado, sem vazar pro provider");
+
+// label via contexto (cbStore, mesmo ALS de CloudBrokerCloud.expressMiddleware)
+const mw2 = CloudBrokerCloud.expressMiddleware({ label: "finops-advisor:root-cause-agent" });
+await new Promise((resolve) => {
+  mw2({ method: "POST", path: "/api/chat/response" }, {}, async () => {
+    await wrappedAi.chat.completions.create({ model: "gpt-4o", messages: [{ role: "user", content: "y" }] });
+    resolve();
+  });
+});
+await new Promise((r) => setTimeout(r, 50));
+assert.strictEqual(capturedAi[1].events[0].label, "finops-advisor:root-cause-agent");
+console.log("✅ CloudBrokerAI.wrap: label via contexto cbStore (sem cbLabel explícito)");
+
 globalThis.fetch = realFetch;
 console.log("\n🎉 Todos os testes passaram.");

package/src/wrapper.js

@@ -5,6 +5,7 @@
 
 import { CloudBrokerClient } from "./client.js";
 import { PRICING } from "./pricing.js";
+import { cbStore } from "./context.js";
 
 export class CloudBrokerAI {
   /**
@@ -17,6 +18,13 @@ export class CloudBrokerAI {
    * @param {string} options.team      — nome do squad (ex: "mobile")
    * @param {string} options.endpoint  — endpoint CloudBroker (opcional)
    * @param {boolean} options.debug    — logs de debug (opcional)
+   *
+   * Label opcional do trecho de código, pra agregar custo de IA junto com
+   * custo de nuvem na mesma visão "Custo por Código" (mesma convenção de
+   * CloudBrokerCloud.wrapAWS):
+   *   await client.chat.completions.create(params, { cbLabel: "checkout:resumo-pedido" });
+   *   await client.messages.create(params, { cbLabel: "checkout:resumo-pedido" });
+   * Ou via CloudBrokerCloud.expressMiddleware() (contexto automático por requisição).
    */
   static wrap(client, options = {}) {
     const { apiKey, product, team, endpoint, debug = false } = options;
@@ -77,6 +85,11 @@ export class CloudBrokerAI {
 
     client.chat.completions.create = async (params, options) => {
       const start = Date.now();
+      // Mesma precedência de cloud-wrapper.js: explícito na chamada > contexto
+      // cbStore (expressMiddleware) > nenhum. Nunca vaza cbLabel pro SDK do provider.
+      const cbLabel = options?.cbLabel || cbStore.getStore()?.label || null;
+      const { cbLabel: _l, ...providerOptions } = options || {};
+      const fwdOptions = Object.keys(providerOptions).length ? providerOptions : undefined;
 
       // ── Streaming: pede usage no chunk final e instrumenta o iterador ────────
       if (params?.stream) {
@@ -84,7 +97,7 @@ export class CloudBrokerAI {
           ...params,
           stream_options: { include_usage: true, ...(params.stream_options || {}) },
         };
-        const stream = await originalCreate(streamParams, options);
+        const stream = await originalCreate(streamParams, fwdOptions);
 
         let model = params.model || "unknown";
         let inputTokens = 0, outputTokens = 0;
@@ -104,6 +117,7 @@ export class CloudBrokerAI {
               model,
               product:     product || null,
               team:        team    || null,
+              label:       cbLabel,
               inputTokens,
               outputTokens,
               totalTokens: inputTokens + outputTokens,
@@ -121,7 +135,7 @@ export class CloudBrokerAI {
       }
 
       // ── Não-streaming ───────────────────────────────────────────────────────
-      const res = await originalCreate(params, options);
+      const res = await originalCreate(params, fwdOptions);
       const ms  = Date.now() - start;
 
       // Fire-and-forget
@@ -137,6 +151,7 @@ export class CloudBrokerAI {
           model,
           product:       product || null,
           team:          team    || null,
+          label:         cbLabel,
           inputTokens,
           outputTokens,
           totalTokens:   inputTokens + outputTokens,
@@ -159,10 +174,13 @@ export class CloudBrokerAI {
 
     client.messages.create = async (params, options) => {
       const start = Date.now();
+      const cbLabel = options?.cbLabel || cbStore.getStore()?.label || null;
+      const { cbLabel: _l, ...providerOptions } = options || {};
+      const fwdOptions = Object.keys(providerOptions).length ? providerOptions : undefined;
 
       // ── Streaming: acumula usage dos eventos message_start / message_delta ───
       if (params?.stream) {
-        const stream = await originalCreate(params, options);
+        const stream = await originalCreate(params, fwdOptions);
 
         let model = params.model || "unknown";
         let inputTokens = 0, outputTokens = 0;
@@ -185,6 +203,7 @@ export class CloudBrokerAI {
               model,
               product:     product || null,
               team:        team    || null,
+              label:       cbLabel,
               inputTokens,
               outputTokens,
               totalTokens: inputTokens + outputTokens,
@@ -202,7 +221,7 @@ export class CloudBrokerAI {
       }
 
       // ── Não-streaming ───────────────────────────────────────────────────────
-      const res = await originalCreate(params, options);
+      const res = await originalCreate(params, fwdOptions);
       const ms  = Date.now() - start;
 
       try {
@@ -217,6 +236,7 @@ export class CloudBrokerAI {
           model,
           product:       product || null,
           team:          team    || null,
+          label:         cbLabel,
           inputTokens,
           outputTokens,
           totalTokens:   inputTokens + outputTokens,