cloudbroker-sdk 0.3.0 → 0.5.0

package/README.md

@@ -152,26 +152,28 @@ Em vez de passar `cbLabel` em cada `.send()`, use o middleware: todas as chamada
 de nuvem feitas durante a requisição herdam um label de negócio automaticamente
 (via `AsyncLocalStorage`). Ótimo pra agrupar custo por **módulo** ou **rota**.
 
+Cada campo (`label`, `product`, `team`) pode ser fixo (string) ou uma função
+`(req) => string`. Isso permite a hierarquia **Grupo (team) → App (product) →
+Trecho (label) → Serviço** na visão de custo.
+
 ```js
 import express from "express";
 import { CloudBrokerCloud } from "cloudbroker-sdk";
 
 const app = express();
 
-// label fixo, "MÉTODO /caminho" (default), ou uma função req → label:
 app.use(CloudBrokerCloud.expressMiddleware({
-  label: (req) => {
-    if (req.path.startsWith("/api/auth"))    return "Autenticação";
-    if (req.path.startsWith("/api/billing")) return "Billing";
-    return "Outros";
-  },
+  team:    (req) => (req.path.startsWith("/api/admin") ? "Plataforma" : "Produto"), // Grupo
+  product: (req) => req.path.split("/")[2] || "core",                              // App
+  label:   (req) => `${req.method} ${req.path}`,                                   // Trecho
 }));
 
 // daqui pra frente, qualquer s3.send(...)/ddb.send(...) dentro da request
-// é atribuído ao módulo retornado — sem cbLabel manual.
+// é atribuído ao grupo/app/trecho — sem cbLabel manual.
 ```
 
-Prioridade do label: `cbLabel` explícito no `.send()` > contexto do middleware > `null`.
+Prioridade de cada campo: explícito no `.send()` (`cbLabel`/`cbProduct`/`cbTeam`) >
+contexto do middleware > valor estático do `wrapAWS` > `null`.
 
 ## Garantias
 

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "cloudbroker-sdk",
-  "version": "0.3.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.3.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.3.0",
+        sdkVersion: "0.5.0",
         sentAt: new Date().toISOString(),
       };
 

package/src/cloud-wrapper.js

@@ -24,7 +24,7 @@
 
 import { CloudBrokerClient } from "./client.js";
 import { estimateCloudCost } from "./pricing-cloud.js";
-import { cbStore, currentLabel } from "./context.js";
+import { cbStore } from "./context.js";
 
 export class CloudBrokerCloud {
   /**
@@ -54,13 +54,16 @@ export class CloudBrokerCloud {
     client.send = async (command, sendOptions) => {
       const start = Date.now();
       const operation = (command?.constructor?.name || "").replace(/Command$/, "");
-      // Label: explícito no .send() tem prioridade; senão usa o do contexto da
-      // requisição (expressMiddleware), senão null.
-      const cbLabel = sendOptions?.cbLabel || currentLabel() || null;
+      // Prioridade: explícito no .send() > contexto da requisição (expressMiddleware)
+      // > valor estático do wrapAWS. Permite Grupo/App/Trecho por requisição.
+      const ctx = cbStore.getStore() || {};
+      const cbLabel   = sendOptions?.cbLabel   || ctx.label   || null;
+      const evProduct = sendOptions?.cbProduct || ctx.product || product || null;
+      const evTeam    = sendOptions?.cbTeam    || ctx.team    || team    || null;
       const bytesIn = CloudBrokerCloud.#inputBytes(command?.input);
 
-      // Passa adiante apenas as options nativas — nunca vaza cbLabel pro AWS SDK.
-      const { cbLabel: _omit, ...awsOptions } = sendOptions || {};
+      // Passa adiante apenas as options nativas — nunca vaza campos cb* pro AWS SDK.
+      const { cbLabel: _l, cbProduct: _p, cbTeam: _t, ...awsOptions } = sendOptions || {};
       const result = await originalSend(command, Object.keys(awsOptions).length ? awsOptions : undefined);
 
       try {
@@ -73,8 +76,8 @@ export class CloudBrokerCloud {
           provider:  "aws",
           service,
           operation,
-          product:   product || null,
-          team:      team    || null,
+          product:   evProduct,
+          team:      evTeam,
           label:     cbLabel,
           requests:  1,
           bytesIn,
@@ -95,26 +98,29 @@ export class CloudBrokerCloud {
   }
 
   /**
-   * Middleware Express que atribui um label de negócio a todas as chamadas de
-   * nuvem feitas durante a requisição (via AsyncLocalStorage). Assim o custo por
-   * código agrupa por módulo/rota sem precisar de cbLabel manual em cada send.
+   * Middleware Express que atribui contexto de negócio a todas as chamadas de
+   * nuvem feitas durante a requisição (via AsyncLocalStorage). Cada campo pode ser
+   * fixo (string) ou uma função (req) => string — ex: mapear rota → módulo/app/grupo.
+   * Sem precisar passar cbLabel/cbProduct/cbTeam manual em cada send.
    *
    * @param {object} [options]
-   * @param {string|((req)=>string)} [options.label] — label fixo, ou função que
-   *   recebe o req e devolve o label (ex: mapear rota → módulo de negócio).
-   *   Default: "MÉTODO /caminho" da requisição.
+   * @param {string|((req)=>string)} [options.label]   — trecho (default: "MÉTODO /caminho")
+   * @param {string|((req)=>string)} [options.product] — App (sobrescreve o product do wrapAWS)
+   * @param {string|((req)=>string)} [options.team]    — Grupo (sobrescreve o team do wrapAWS)
    * @returns {(req,res,next)=>void}
    */
   static expressMiddleware(options = {}) {
-    const resolve =
-      typeof options.label === "function" ? options.label :
-      typeof options.label === "string"   ? () => options.label :
-      (req) => `${req.method || ""} ${req.baseUrl || ""}${req.route?.path || req.path || ""}`.trim();
+    const mk = (opt, fallback) =>
+      typeof opt === "function" ? opt :
+      typeof opt === "string"   ? () => opt :
+      fallback;
+    const resolveLabel   = mk(options.label, (req) => `${req.method || ""} ${req.baseUrl || ""}${req.route?.path || req.path || ""}`.trim());
+    const resolveProduct = mk(options.product, () => undefined);
+    const resolveTeam    = mk(options.team, () => undefined);
 
     return (req, res, next) => {
-      let label = null;
-      try { label = resolve(req); } catch { label = null; }
-      cbStore.run({ label }, () => next());
+      const safe = (fn) => { try { return fn(req); } catch { return undefined; } };
+      cbStore.run({ label: safe(resolveLabel), product: safe(resolveProduct), team: safe(resolveTeam) }, () => next());
     };
   }
 

package/src/test.js

@@ -154,18 +154,76 @@ assert.ok(ev.costUSD > 0, "deve ter custo estimado");
 console.log("✅ wrapAWS: evento capturado, cbLabel não vazou, resultado preservado");
 
 // 3) expressMiddleware — label do contexto (ALS) entra sem cbLabel explícito
-const mw = CloudBrokerCloud.expressMiddleware({ label: (req) => `modulo:${req.path}` });
+const mw = CloudBrokerCloud.expressMiddleware({
+  label:   (req) => `modulo:${req.path}`,
+  product: (req) => (req.path.startsWith("/api/billing") ? "Billing" : "Outros"),
+  team:    "Plataforma",
+});
 const fakeReq = { method: "GET", path: "/api/billing" };
 await new Promise((resolve) => {
   mw(fakeReq, {}, async () => {
-    await s3.send(new PutObjectCommand({ Bucket: "b", Key: "k", Body: "x" })); // sem cbLabel — herda do contexto
+    await s3.send(new PutObjectCommand({ Bucket: "b", Key: "k", Body: "x" })); // sem cb* — herda do contexto
     resolve();
   });
 });
 await new Promise((r) => setTimeout(r, 50));
 const ctxEvent = captured[captured.length - 1].events[0];
-assert.strictEqual(ctxEvent.label, "modulo:/api/billing", `label do contexto esperado, veio ${ctxEvent.label}`);
-console.log("✅ expressMiddleware: label do contexto aplicado sem cbLabel explícito");
+assert.strictEqual(ctxEvent.label,   "modulo:/api/billing", `label do contexto esperado, veio ${ctxEvent.label}`);
+assert.strictEqual(ctxEvent.product, "Billing",     `product do contexto esperado, veio ${ctxEvent.product}`);
+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,