langwatch 1.0.0 → 1.1.0

package/dist/index.mjs

@@ -4,10 +4,10 @@ import {
 import {
   createLangWatchSpan,
   getLangWatchTracer
-} from "./chunk-C323F5VC.mjs";
+} from "./chunk-PYODMDPV.mjs";
 import {
   ensureSetup
-} from "./chunk-ALPIFEXL.mjs";
+} from "./chunk-EW233MAD.mjs";
 import {
   DEFAULT_ENDPOINT,
   LANGWATCH_SDK_LANGUAGE,
@@ -19,7 +19,7 @@ import {
   buildAuthHeaders,
   getLangWatchLogger,
   version
-} from "./chunk-5Q2DUSMQ.mjs";
+} from "./chunk-5AGWTUI2.mjs";
 import {
   ConsoleLogger,
   NoOpLogger,
@@ -35,6 +35,18 @@ import {
   shouldCaptureOutput
 } from "./chunk-L5KBRKHS.mjs";
 
+// src/internal/credentialContext.ts
+import { AsyncLocalStorage } from "async_hooks";
+var storage = new AsyncLocalStorage();
+var fallbackHolder = {};
+function currentHolder() {
+  var _a;
+  return (_a = storage.getStore()) != null ? _a : fallbackHolder;
+}
+function scopedApiKey() {
+  return currentHolder().apiKey;
+}
+
 // src/client-sdk/services/prompts/prompts-api.service.ts
 import { z } from "zod";
 
@@ -643,7 +655,7 @@ var handledErrorMiddleware = {
     if (handledError) throw handledError;
   }
 };
-var createLangWatchApiClient = (apiKey = ((_a) => (_a = process.env.LANGWATCH_API_KEY) != null ? _a : "")(), endpoint = ((_b) => (_b = process.env.LANGWATCH_ENDPOINT) != null ? _b : DEFAULT_ENDPOINT)(), projectId = process.env.LANGWATCH_PROJECT_ID) => {
+var createLangWatchApiClient = (apiKey = ((_b) => (_b = ((_a) => (_a = scopedApiKey()) != null ? _a : process.env.LANGWATCH_API_KEY)()) != null ? _b : "")(), endpoint = ((_c) => (_c = process.env.LANGWATCH_ENDPOINT) != null ? _c : DEFAULT_ENDPOINT)(), projectId = process.env.LANGWATCH_PROJECT_ID) => {
   const client = openApiCreateClient({
     baseUrl: endpoint,
     headers: __spreadProps(__spreadValues({}, buildAuthHeaders({ apiKey, projectId })), {
@@ -2474,7 +2486,7 @@ var DatasetsFacade = class {
 _datasetService = new WeakMap();
 
 // src/client-sdk/services/experiments/experiment.ts
-import { AsyncLocalStorage } from "async_hooks";
+import { AsyncLocalStorage as AsyncLocalStorage2 } from "async_hooks";
 import { trace, SpanStatusCode, ROOT_CONTEXT } from "@opentelemetry/api";
 
 // src/client-sdk/services/experiments/humanReadableId.ts
@@ -2683,8 +2695,8 @@ function printSummary(result) {
 // src/client-sdk/services/experiments/experiment.ts
 var DEFAULT_CONCURRENCY = 4;
 var DEBOUNCE_INTERVAL_MS = 1e3;
-var iterationContextStorage = new AsyncLocalStorage();
-var targetContextStorage = new AsyncLocalStorage();
+var iterationContextStorage = new AsyncLocalStorage2();
+var targetContextStorage = new AsyncLocalStorage2();
 var Experiment = class _Experiment {
   constructor(name, options) {
     this.initialized = false;
@@ -5149,9 +5161,9 @@ var MonitorsApiError = class extends Error {
 };
 var MonitorsApiService = class {
   constructor(config) {
-    var _a, _b, _c, _d;
-    this.apiKey = (_b = (_a = config == null ? void 0 : config.apiKey) != null ? _a : process.env.LANGWATCH_API_KEY) != null ? _b : "";
-    this.endpoint = (_d = (_c = config == null ? void 0 : config.endpoint) != null ? _c : process.env.LANGWATCH_ENDPOINT) != null ? _d : DEFAULT_ENDPOINT;
+    var _a, _b, _c, _d, _e;
+    this.apiKey = (_c = (_b = (_a = config == null ? void 0 : config.apiKey) != null ? _a : scopedApiKey()) != null ? _b : process.env.LANGWATCH_API_KEY) != null ? _c : "";
+    this.endpoint = (_e = (_d = config == null ? void 0 : config.endpoint) != null ? _d : process.env.LANGWATCH_ENDPOINT) != null ? _e : DEFAULT_ENDPOINT;
   }
   async request(path2, options) {
     var _a, _b;
@@ -5224,9 +5236,9 @@ var SecretsApiError = class extends Error {
 };
 var SecretsApiService = class {
   constructor(config) {
-    var _a, _b, _c, _d;
-    this.apiKey = (_b = (_a = config == null ? void 0 : config.apiKey) != null ? _a : process.env.LANGWATCH_API_KEY) != null ? _b : "";
-    this.endpoint = (_d = (_c = config == null ? void 0 : config.endpoint) != null ? _c : process.env.LANGWATCH_ENDPOINT) != null ? _d : DEFAULT_ENDPOINT;
+    var _a, _b, _c, _d, _e;
+    this.apiKey = (_c = (_b = (_a = config == null ? void 0 : config.apiKey) != null ? _a : scopedApiKey()) != null ? _b : process.env.LANGWATCH_API_KEY) != null ? _c : "";
+    this.endpoint = (_e = (_d = config == null ? void 0 : config.endpoint) != null ? _d : process.env.LANGWATCH_ENDPOINT) != null ? _e : DEFAULT_ENDPOINT;
   }
   async request(path2, options) {
     var _a, _b;
@@ -5361,9 +5373,9 @@ var _LangWatch_instances, createInternalConfig_fn;
 var LangWatch = class {
   constructor(options = {}) {
     __privateAdd(this, _LangWatch_instances);
-    var _a, _b, _c, _d;
-    const apiKey = (_b = (_a = options.apiKey) != null ? _a : process.env.LANGWATCH_API_KEY) != null ? _b : "";
-    const endpoint = (_d = (_c = options.endpoint) != null ? _c : process.env.LANGWATCH_ENDPOINT) != null ? _d : DEFAULT_ENDPOINT;
+    var _a, _b, _c, _d, _e;
+    const apiKey = (_c = (_b = (_a = options.apiKey) != null ? _a : scopedApiKey()) != null ? _b : process.env.LANGWATCH_API_KEY) != null ? _c : "";
+    const endpoint = (_e = (_d = options.endpoint) != null ? _d : process.env.LANGWATCH_ENDPOINT) != null ? _e : DEFAULT_ENDPOINT;
     this.config = __privateMethod(this, _LangWatch_instances, createInternalConfig_fn).call(this, {
       apiKey,
       endpoint,
@@ -5425,6 +5437,199 @@ createInternalConfig_fn = function({
   };
 };
 
+// src/client-sdk/services/virtual-keys/virtual-keys-api.service.ts
+var VirtualKeysApiError = class extends Error {
+  constructor(message, operation, originalError) {
+    super(message);
+    this.operation = operation;
+    this.originalError = originalError;
+    this.name = "VirtualKeysApiError";
+  }
+};
+var VirtualKeysApiService = class {
+  constructor(config) {
+    var _a, _b, _c, _d, _e;
+    this.endpoint = ((_b = (_a = config == null ? void 0 : config.endpoint) != null ? _a : process.env.LANGWATCH_ENDPOINT) != null ? _b : DEFAULT_ENDPOINT).replace(/\/+$/, "");
+    this.apiKey = (_e = (_d = (_c = config == null ? void 0 : config.apiKey) != null ? _c : scopedApiKey()) != null ? _d : process.env.LANGWATCH_API_KEY) != null ? _e : "";
+  }
+  headers() {
+    return {
+      Authorization: `Bearer ${this.apiKey}`,
+      "Content-Type": "application/json"
+    };
+  }
+  async request(operation, path2, init) {
+    var _a;
+    const response = await fetch(`${this.endpoint}${path2}`, __spreadProps(__spreadValues({}, init), {
+      headers: __spreadValues(__spreadValues({}, this.headers()), (_a = init == null ? void 0 : init.headers) != null ? _a : {})
+    }));
+    if (!response.ok) {
+      let parsedBody;
+      try {
+        parsedBody = await response.json();
+      } catch (e) {
+        parsedBody = await response.text();
+      }
+      const message = formatApiErrorForOperation({
+        operation,
+        error: parsedBody,
+        options: { status: response.status }
+      });
+      throwIfHandledError({
+        operation,
+        error: parsedBody,
+        status: response.status,
+        message
+      });
+      throw new VirtualKeysApiError(message, operation, parsedBody);
+    }
+    return await response.json();
+  }
+  async list() {
+    const { data } = await this.request(
+      "list virtual keys",
+      "/api/gateway/v1/virtual-keys"
+    );
+    return data;
+  }
+  async get(id) {
+    const { virtual_key } = await this.request(
+      `get virtual key "${id}"`,
+      `/api/gateway/v1/virtual-keys/${encodeURIComponent(id)}`
+    );
+    return virtual_key;
+  }
+  async create(input) {
+    return this.request(
+      "create virtual key",
+      "/api/gateway/v1/virtual-keys",
+      { method: "POST", body: JSON.stringify(input) }
+    );
+  }
+  async update(id, input) {
+    const { virtual_key } = await this.request(
+      `update virtual key "${id}"`,
+      `/api/gateway/v1/virtual-keys/${encodeURIComponent(id)}`,
+      { method: "PATCH", body: JSON.stringify(input) }
+    );
+    return virtual_key;
+  }
+  async rotate(id) {
+    return this.request(
+      `rotate virtual key "${id}"`,
+      `/api/gateway/v1/virtual-keys/${encodeURIComponent(id)}/rotate`,
+      { method: "POST" }
+    );
+  }
+  async revoke(id) {
+    const { virtual_key } = await this.request(
+      `revoke virtual key "${id}"`,
+      `/api/gateway/v1/virtual-keys/${encodeURIComponent(id)}/revoke`,
+      { method: "POST" }
+    );
+    return virtual_key;
+  }
+  /**
+   * Aggregate spend for one key. Defaults to the current UTC calendar
+   * month server-side. Reads the same cost path the dashboard reads, so
+   * this number and the UI agree by construction.
+   */
+  async spend(id, window) {
+    const params = new URLSearchParams();
+    if (window == null ? void 0 : window.from) params.set("from", window.from);
+    if (window == null ? void 0 : window.to) params.set("to", window.to);
+    const query = params.size > 0 ? `?${params.toString()}` : "";
+    return this.request(
+      `read virtual key spend "${id}"`,
+      `/api/gateway/v1/virtual-keys/${encodeURIComponent(id)}/spend${query}`
+    );
+  }
+};
+
+// src/client-sdk/services/gateway-budgets/gateway-budgets-api.service.ts
+var GatewayBudgetsApiError = class extends Error {
+  constructor(message, operation, originalError) {
+    super(message);
+    this.operation = operation;
+    this.originalError = originalError;
+    this.name = "GatewayBudgetsApiError";
+  }
+};
+var GatewayBudgetsApiService = class {
+  constructor(config) {
+    var _a, _b, _c, _d, _e;
+    this.endpoint = ((_b = (_a = config == null ? void 0 : config.endpoint) != null ? _a : process.env.LANGWATCH_ENDPOINT) != null ? _b : DEFAULT_ENDPOINT).replace(/\/+$/, "");
+    this.apiKey = (_e = (_d = (_c = config == null ? void 0 : config.apiKey) != null ? _c : scopedApiKey()) != null ? _d : process.env.LANGWATCH_API_KEY) != null ? _e : "";
+  }
+  headers() {
+    return {
+      Authorization: `Bearer ${this.apiKey}`,
+      "Content-Type": "application/json"
+    };
+  }
+  async request(operation, path2, init) {
+    var _a;
+    const response = await fetch(`${this.endpoint}${path2}`, __spreadProps(__spreadValues({}, init), {
+      headers: __spreadValues(__spreadValues({}, this.headers()), (_a = init == null ? void 0 : init.headers) != null ? _a : {})
+    }));
+    if (!response.ok) {
+      let parsedBody;
+      try {
+        parsedBody = await response.json();
+      } catch (e) {
+        parsedBody = await response.text();
+      }
+      const message = formatApiErrorForOperation({
+        operation,
+        error: parsedBody,
+        options: { status: response.status }
+      });
+      throwIfHandledError({
+        operation,
+        error: parsedBody,
+        status: response.status,
+        message
+      });
+      throw new GatewayBudgetsApiError(message, operation, parsedBody);
+    }
+    return await response.json();
+  }
+  /**
+   * Every non-archived budget in the organization across all six scope
+   * types, optionally filtered by `scopeTypes`.
+   */
+  async list(options) {
+    var _a;
+    const filter = ((_a = options == null ? void 0 : options.scopeTypes) == null ? void 0 : _a.length) ? `?scope_type=${options.scopeTypes.join(",")}` : "";
+    const { data, spend_available } = await this.request("list gateway budgets", `/api/gateway/v1/budgets${filter}`);
+    return { budgets: data, spend_available };
+  }
+  async create(input) {
+    const { budget } = await this.request(
+      "create gateway budget",
+      "/api/gateway/v1/budgets",
+      { method: "POST", body: JSON.stringify(input) }
+    );
+    return budget;
+  }
+  async update(id, input) {
+    const { budget } = await this.request(
+      `update gateway budget "${id}"`,
+      `/api/gateway/v1/budgets/${encodeURIComponent(id)}`,
+      { method: "PATCH", body: JSON.stringify(input) }
+    );
+    return budget;
+  }
+  async archive(id) {
+    const { budget } = await this.request(
+      `archive gateway budget "${id}"`,
+      `/api/gateway/v1/budgets/${encodeURIComponent(id)}`,
+      { method: "DELETE" }
+    );
+    return budget;
+  }
+};
+
 // src/index.ts
 var logger = {
   ConsoleLogger,
@@ -5452,11 +5657,15 @@ export {
   ExperimentsFacade,
   FetchPolicy,
   FilterableBatchSpanProcessor,
+  GatewayBudgetsApiError,
+  GatewayBudgetsApiService,
   LangWatch,
   LangWatchApiError,
   LangWatchTraceExporter as LangWatchExporter,
   LangWatchHandledError,
   TargetMetadataConflictError,
+  VirtualKeysApiError,
+  VirtualKeysApiService,
   attributes_exports as attributes,
   getLangWatchLogger,
   getLangWatchTracer,