braintrust 0.4.4 → 0.4.6

package/dev/dist/index.js

@@ -848,7 +848,9 @@ function mergeDictsWithPathsHelper({
         mergePaths
       });
     } else {
-      mergeInto[k] = mergeFromV;
+      if (mergeFromV !== void 0) {
+        mergeInto[k] = mergeFromV;
+      }
     }
   });
   return mergeInto;
@@ -5447,7 +5449,7 @@ async function permalink(slug, opts) {
     return state.appUrl;
   };
   try {
-    const components = SpanComponentsV3.fromStr(slug);
+    const components = getSpanComponentsClass().fromStr(slug);
     const object_type = spanObjectTypeV3ToString(components.data.object_type);
     const [orgName, appUrl, object_id] = await Promise.all([
       getOrgName(),
@@ -5475,7 +5477,7 @@ function startSpanParentArgs(args) {
     if (args.parentSpanIds) {
       throw new Error("Cannot specify both parent and parentSpanIds");
     }
-    const parentComponents = SpanComponentsV3.fromStr(args.parent);
+    const parentComponents = getSpanComponentsClass().fromStr(args.parent);
     if (args.parentObjectType !== parentComponents.data.object_type) {
       throw new Error(
         `Mismatch between expected span parent object type ${args.parentObjectType} and provided type ${parentComponents.data.object_type}`
@@ -5919,7 +5921,7 @@ function getSpanParentObject(options) {
     return parentSpan;
   }
   const parentStr = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _40 => _40.parent]), () => ( state.currentParent.getStore()));
-  if (parentStr) return SpanComponentsV3.fromStr(parentStr);
+  if (parentStr) return getSpanComponentsClass().fromStr(parentStr);
   const experiment = currentExperiment();
   if (experiment) {
     return experiment;
@@ -5985,7 +5987,7 @@ function startSpanAndIsLogger(args) {
     parent: _optionalChain([args, 'optionalAccess', _47 => _47.parent]),
     state
   });
-  if (parentObject instanceof SpanComponentsV3) {
+  if (parentObject instanceof SpanComponentsV3 || parentObject instanceof SpanComponentsV4) {
     const parentSpanIds = parentObject.data.row_id ? {
       spanId: parentObject.data.span_id,
       rootSpanId: parentObject.data.root_span_id
@@ -7828,7 +7830,10 @@ Error: ${errorText}`;
                   limit: INTERNAL_BTQL_LIMIT
                 },
                 use_columnstore: false,
-                brainstore_realtime: true
+                brainstore_realtime: true,
+                ...this.pinnedVersion !== void 0 ? {
+                  version: this.pinnedVersion
+                } : {}
               },
               { headers: { "Accept-Encoding": "gzip" } }
             );
@@ -11392,9 +11397,17 @@ function makeCheckAuthorized(allowedOrgName) {
       return next(_httperrors2.default.call(void 0, 401, "Unauthorized"));
     }
     try {
+      const orgName = parseHeader(req.headers, "x-bt-org-name");
+      if (!orgName) {
+        return next(_httperrors2.default.call(void 0, 400, "Missing x-bt-org-name header"));
+      }
+      if (allowedOrgName && allowedOrgName !== orgName) {
+        const errorMessage = `Org '${orgName}' is not allowed. Only org '${allowedOrgName}' is allowed.`;
+        return next(_httperrors2.default.call(void 0, 403, errorMessage));
+      }
       const state = await cachedLogin({
         apiKey: _optionalChain([req, 'access', _124 => _124.ctx, 'optionalAccess', _125 => _125.token]),
-        orgName: allowedOrgName
+        orgName
       });
       req.ctx.state = state;
       next();

package/dev/dist/index.mjs

@@ -848,7 +848,9 @@ function mergeDictsWithPathsHelper({
         mergePaths
       });
     } else {
-      mergeInto[k] = mergeFromV;
+      if (mergeFromV !== void 0) {
+        mergeInto[k] = mergeFromV;
+      }
     }
   });
   return mergeInto;
@@ -5447,7 +5449,7 @@ async function permalink(slug, opts) {
     return state.appUrl;
   };
   try {
-    const components = SpanComponentsV3.fromStr(slug);
+    const components = getSpanComponentsClass().fromStr(slug);
     const object_type = spanObjectTypeV3ToString(components.data.object_type);
     const [orgName, appUrl, object_id] = await Promise.all([
       getOrgName(),
@@ -5475,7 +5477,7 @@ function startSpanParentArgs(args) {
     if (args.parentSpanIds) {
       throw new Error("Cannot specify both parent and parentSpanIds");
     }
-    const parentComponents = SpanComponentsV3.fromStr(args.parent);
+    const parentComponents = getSpanComponentsClass().fromStr(args.parent);
     if (args.parentObjectType !== parentComponents.data.object_type) {
       throw new Error(
         `Mismatch between expected span parent object type ${args.parentObjectType} and provided type ${parentComponents.data.object_type}`
@@ -5919,7 +5921,7 @@ function getSpanParentObject(options) {
     return parentSpan;
   }
   const parentStr = options?.parent ?? state.currentParent.getStore();
-  if (parentStr) return SpanComponentsV3.fromStr(parentStr);
+  if (parentStr) return getSpanComponentsClass().fromStr(parentStr);
   const experiment = currentExperiment();
   if (experiment) {
     return experiment;
@@ -5985,7 +5987,7 @@ function startSpanAndIsLogger(args) {
     parent: args?.parent,
     state
   });
-  if (parentObject instanceof SpanComponentsV3) {
+  if (parentObject instanceof SpanComponentsV3 || parentObject instanceof SpanComponentsV4) {
     const parentSpanIds = parentObject.data.row_id ? {
       spanId: parentObject.data.span_id,
       rootSpanId: parentObject.data.root_span_id
@@ -7828,7 +7830,10 @@ Error: ${errorText}`;
                   limit: INTERNAL_BTQL_LIMIT
                 },
                 use_columnstore: false,
-                brainstore_realtime: true
+                brainstore_realtime: true,
+                ...this.pinnedVersion !== void 0 ? {
+                  version: this.pinnedVersion
+                } : {}
               },
               { headers: { "Accept-Encoding": "gzip" } }
             );
@@ -11392,9 +11397,17 @@ function makeCheckAuthorized(allowedOrgName) {
       return next(createError(401, "Unauthorized"));
     }
     try {
+      const orgName = parseHeader(req.headers, "x-bt-org-name");
+      if (!orgName) {
+        return next(createError(400, "Missing x-bt-org-name header"));
+      }
+      if (allowedOrgName && allowedOrgName !== orgName) {
+        const errorMessage = `Org '${orgName}' is not allowed. Only org '${allowedOrgName}' is allowed.`;
+        return next(createError(403, errorMessage));
+      }
       const state = await cachedLogin({
         apiKey: req.ctx?.token,
-        orgName: allowedOrgName
+        orgName
       });
       req.ctx.state = state;
       next();

package/dist/browser.d.mts

@@ -318,6 +318,69 @@ interface Score {
     error?: unknown;
 }
 
+declare const spanComponentsV4Schema: z.ZodIntersection<z.ZodIntersection<z.ZodObject<{
+    object_type: z.ZodNativeEnum<typeof SpanObjectTypeV3>;
+    propagated_event: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
+}, "strip", z.ZodTypeAny, {
+    object_type: SpanObjectTypeV3;
+    propagated_event?: Record<string, unknown> | null | undefined;
+}, {
+    object_type: SpanObjectTypeV3;
+    propagated_event?: Record<string, unknown> | null | undefined;
+}>, z.ZodUnion<[z.ZodObject<{
+    object_id: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    compute_object_metadata_args: z.ZodOptional<z.ZodNull>;
+}, "strip", z.ZodTypeAny, {
+    object_id?: string | null | undefined;
+    compute_object_metadata_args?: null | undefined;
+}, {
+    object_id?: string | null | undefined;
+    compute_object_metadata_args?: null | undefined;
+}>, z.ZodObject<{
+    object_id: z.ZodOptional<z.ZodNull>;
+    compute_object_metadata_args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+}, "strip", z.ZodTypeAny, {
+    compute_object_metadata_args: Record<string, unknown>;
+    object_id?: null | undefined;
+}, {
+    compute_object_metadata_args: Record<string, unknown>;
+    object_id?: null | undefined;
+}>]>>, z.ZodUnion<[z.ZodObject<{
+    row_id: z.ZodString;
+    span_id: z.ZodString;
+    root_span_id: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    span_id: string;
+    root_span_id: string;
+    row_id: string;
+}, {
+    span_id: string;
+    root_span_id: string;
+    row_id: string;
+}>, z.ZodObject<{
+    row_id: z.ZodOptional<z.ZodNull>;
+    span_id: z.ZodOptional<z.ZodNull>;
+    root_span_id: z.ZodOptional<z.ZodNull>;
+}, "strip", z.ZodTypeAny, {
+    span_id?: null | undefined;
+    root_span_id?: null | undefined;
+    row_id?: null | undefined;
+}, {
+    span_id?: null | undefined;
+    root_span_id?: null | undefined;
+    row_id?: null | undefined;
+}>]>>;
+type SpanComponentsV4Data = z.infer<typeof spanComponentsV4Schema>;
+declare class SpanComponentsV4 {
+    data: SpanComponentsV4Data;
+    constructor(data: SpanComponentsV4Data);
+    toStr(): string;
+    static fromStr(s: string): SpanComponentsV4;
+    objectIdFields(): ParentExperimentIds | ParentProjectLogIds | ParentPlaygroundLogIds;
+    export(): Promise<string>;
+    private static fromJsonObj;
+}
+
 declare const spanTypeAttributeValues: readonly ["llm", "score", "function", "eval", "task", "tool"];
 type SpanType = (typeof spanTypeAttributeValues)[number];
 
@@ -8919,7 +8982,7 @@ declare function currentSpan(options?: OptionalStateArg): Span;
  */
 declare function getSpanParentObject<IsAsyncFlush extends boolean>(options?: AsyncFlushArg<IsAsyncFlush> & OptionalStateArg & {
     parent?: string;
-}): SpanComponentsV3 | Span | Experiment | Logger<IsAsyncFlush>;
+}): SpanComponentsV3 | SpanComponentsV4 | Span | Experiment | Logger<IsAsyncFlush>;
 declare function logError(span: Span, error: unknown): void;
 /**
  * Toplevel function for starting a span. It checks the following (in precedence order):

package/dist/browser.d.ts

@@ -318,6 +318,69 @@ interface Score {
     error?: unknown;
 }
 
+declare const spanComponentsV4Schema: z.ZodIntersection<z.ZodIntersection<z.ZodObject<{
+    object_type: z.ZodNativeEnum<typeof SpanObjectTypeV3>;
+    propagated_event: z.ZodOptional<z.ZodNullable<z.ZodRecord<z.ZodString, z.ZodUnknown>>>;
+}, "strip", z.ZodTypeAny, {
+    object_type: SpanObjectTypeV3;
+    propagated_event?: Record<string, unknown> | null | undefined;
+}, {
+    object_type: SpanObjectTypeV3;
+    propagated_event?: Record<string, unknown> | null | undefined;
+}>, z.ZodUnion<[z.ZodObject<{
+    object_id: z.ZodOptional<z.ZodNullable<z.ZodString>>;
+    compute_object_metadata_args: z.ZodOptional<z.ZodNull>;
+}, "strip", z.ZodTypeAny, {
+    object_id?: string | null | undefined;
+    compute_object_metadata_args?: null | undefined;
+}, {
+    object_id?: string | null | undefined;
+    compute_object_metadata_args?: null | undefined;
+}>, z.ZodObject<{
+    object_id: z.ZodOptional<z.ZodNull>;
+    compute_object_metadata_args: z.ZodRecord<z.ZodString, z.ZodUnknown>;
+}, "strip", z.ZodTypeAny, {
+    compute_object_metadata_args: Record<string, unknown>;
+    object_id?: null | undefined;
+}, {
+    compute_object_metadata_args: Record<string, unknown>;
+    object_id?: null | undefined;
+}>]>>, z.ZodUnion<[z.ZodObject<{
+    row_id: z.ZodString;
+    span_id: z.ZodString;
+    root_span_id: z.ZodString;
+}, "strip", z.ZodTypeAny, {
+    span_id: string;
+    root_span_id: string;
+    row_id: string;
+}, {
+    span_id: string;
+    root_span_id: string;
+    row_id: string;
+}>, z.ZodObject<{
+    row_id: z.ZodOptional<z.ZodNull>;
+    span_id: z.ZodOptional<z.ZodNull>;
+    root_span_id: z.ZodOptional<z.ZodNull>;
+}, "strip", z.ZodTypeAny, {
+    span_id?: null | undefined;
+    root_span_id?: null | undefined;
+    row_id?: null | undefined;
+}, {
+    span_id?: null | undefined;
+    root_span_id?: null | undefined;
+    row_id?: null | undefined;
+}>]>>;
+type SpanComponentsV4Data = z.infer<typeof spanComponentsV4Schema>;
+declare class SpanComponentsV4 {
+    data: SpanComponentsV4Data;
+    constructor(data: SpanComponentsV4Data);
+    toStr(): string;
+    static fromStr(s: string): SpanComponentsV4;
+    objectIdFields(): ParentExperimentIds | ParentProjectLogIds | ParentPlaygroundLogIds;
+    export(): Promise<string>;
+    private static fromJsonObj;
+}
+
 declare const spanTypeAttributeValues: readonly ["llm", "score", "function", "eval", "task", "tool"];
 type SpanType = (typeof spanTypeAttributeValues)[number];
 
@@ -8919,7 +8982,7 @@ declare function currentSpan(options?: OptionalStateArg): Span;
  */
 declare function getSpanParentObject<IsAsyncFlush extends boolean>(options?: AsyncFlushArg<IsAsyncFlush> & OptionalStateArg & {
     parent?: string;
-}): SpanComponentsV3 | Span | Experiment | Logger<IsAsyncFlush>;
+}): SpanComponentsV3 | SpanComponentsV4 | Span | Experiment | Logger<IsAsyncFlush>;
 declare function logError(span: Span, error: unknown): void;
 /**
  * Toplevel function for starting a span. It checks the following (in precedence order):

package/dist/browser.js

@@ -845,7 +845,9 @@ function mergeDictsWithPathsHelper({
         mergePaths
       });
     } else {
-      mergeInto[k] = mergeFromV;
+      if (mergeFromV !== void 0) {
+        mergeInto[k] = mergeFromV;
+      }
     }
   });
   return mergeInto;
@@ -5391,7 +5393,7 @@ function updateSpan({
   ...event
 }) {
   const resolvedState = _nullishCoalesce(state, () => ( _globalState));
-  const components = SpanComponentsV3.fromStr(exported);
+  const components = getSpanComponentsClass().fromStr(exported);
   if (!components.data.row_id) {
     throw new Error("Exported span must have a row id");
   }
@@ -5474,7 +5476,7 @@ async function permalink(slug, opts) {
     return state.appUrl;
   };
   try {
-    const components = SpanComponentsV3.fromStr(slug);
+    const components = getSpanComponentsClass().fromStr(slug);
     const object_type = spanObjectTypeV3ToString(components.data.object_type);
     const [orgName, appUrl, object_id] = await Promise.all([
       getOrgName(),
@@ -5502,7 +5504,7 @@ function startSpanParentArgs(args) {
     if (args.parentSpanIds) {
       throw new Error("Cannot specify both parent and parentSpanIds");
     }
-    const parentComponents = SpanComponentsV3.fromStr(args.parent);
+    const parentComponents = getSpanComponentsClass().fromStr(args.parent);
     if (args.parentObjectType !== parentComponents.data.object_type) {
       throw new Error(
         `Mismatch between expected span parent object type ${args.parentObjectType} and provided type ${parentComponents.data.object_type}`
@@ -6164,7 +6166,7 @@ function getSpanParentObject(options) {
     return parentSpan;
   }
   const parentStr = _nullishCoalesce(_optionalChain([options, 'optionalAccess', _43 => _43.parent]), () => ( state.currentParent.getStore()));
-  if (parentStr) return SpanComponentsV3.fromStr(parentStr);
+  if (parentStr) return getSpanComponentsClass().fromStr(parentStr);
   const experiment = currentExperiment();
   if (experiment) {
     return experiment;
@@ -6385,7 +6387,7 @@ function startSpanAndIsLogger(args) {
     parent: _optionalChain([args, 'optionalAccess', _52 => _52.parent]),
     state
   });
-  if (parentObject instanceof SpanComponentsV3) {
+  if (parentObject instanceof SpanComponentsV3 || parentObject instanceof SpanComponentsV4) {
     const parentSpanIds = parentObject.data.row_id ? {
       spanId: parentObject.data.span_id,
       rootSpanId: parentObject.data.root_span_id
@@ -8505,7 +8507,10 @@ Error: ${errorText}`;
                   limit: INTERNAL_BTQL_LIMIT
                 },
                 use_columnstore: false,
-                brainstore_realtime: true
+                brainstore_realtime: true,
+                ...this.pinnedVersion !== void 0 ? {
+                  version: this.pinnedVersion
+                } : {}
               },
               { headers: { "Accept-Encoding": "gzip" } }
             );

package/dist/browser.mjs

@@ -845,7 +845,9 @@ function mergeDictsWithPathsHelper({
         mergePaths
       });
     } else {
-      mergeInto[k] = mergeFromV;
+      if (mergeFromV !== void 0) {
+        mergeInto[k] = mergeFromV;
+      }
     }
   });
   return mergeInto;
@@ -5391,7 +5393,7 @@ function updateSpan({
   ...event
 }) {
   const resolvedState = state ?? _globalState;
-  const components = SpanComponentsV3.fromStr(exported);
+  const components = getSpanComponentsClass().fromStr(exported);
   if (!components.data.row_id) {
     throw new Error("Exported span must have a row id");
   }
@@ -5474,7 +5476,7 @@ async function permalink(slug, opts) {
     return state.appUrl;
   };
   try {
-    const components = SpanComponentsV3.fromStr(slug);
+    const components = getSpanComponentsClass().fromStr(slug);
     const object_type = spanObjectTypeV3ToString(components.data.object_type);
     const [orgName, appUrl, object_id] = await Promise.all([
       getOrgName(),
@@ -5502,7 +5504,7 @@ function startSpanParentArgs(args) {
     if (args.parentSpanIds) {
       throw new Error("Cannot specify both parent and parentSpanIds");
     }
-    const parentComponents = SpanComponentsV3.fromStr(args.parent);
+    const parentComponents = getSpanComponentsClass().fromStr(args.parent);
     if (args.parentObjectType !== parentComponents.data.object_type) {
       throw new Error(
         `Mismatch between expected span parent object type ${args.parentObjectType} and provided type ${parentComponents.data.object_type}`
@@ -6164,7 +6166,7 @@ function getSpanParentObject(options) {
     return parentSpan;
   }
   const parentStr = options?.parent ?? state.currentParent.getStore();
-  if (parentStr) return SpanComponentsV3.fromStr(parentStr);
+  if (parentStr) return getSpanComponentsClass().fromStr(parentStr);
   const experiment = currentExperiment();
   if (experiment) {
     return experiment;
@@ -6385,7 +6387,7 @@ function startSpanAndIsLogger(args) {
     parent: args?.parent,
     state
   });
-  if (parentObject instanceof SpanComponentsV3) {
+  if (parentObject instanceof SpanComponentsV3 || parentObject instanceof SpanComponentsV4) {
     const parentSpanIds = parentObject.data.row_id ? {
       spanId: parentObject.data.span_id,
       rootSpanId: parentObject.data.root_span_id
@@ -8505,7 +8507,10 @@ Error: ${errorText}`;
                   limit: INTERNAL_BTQL_LIMIT
                 },
                 use_columnstore: false,
-                brainstore_realtime: true
+                brainstore_realtime: true,
+                ...this.pinnedVersion !== void 0 ? {
+                  version: this.pinnedVersion
+                } : {}
               },
               { headers: { "Accept-Encoding": "gzip" } }
             );

package/dist/cli.js

@@ -2052,7 +2052,9 @@ function mergeDictsWithPathsHelper({
         mergePaths
       });
     } else {
-      mergeInto[k] = mergeFromV;
+      if (mergeFromV !== void 0) {
+        mergeInto[k] = mergeFromV;
+      }
     }
   });
   return mergeInto;
@@ -6686,7 +6688,7 @@ async function permalink(slug, opts) {
     return state.appUrl;
   };
   try {
-    const components = SpanComponentsV3.fromStr(slug);
+    const components = getSpanComponentsClass().fromStr(slug);
     const object_type = spanObjectTypeV3ToString(components.data.object_type);
     const [orgName, appUrl, object_id] = await Promise.all([
       getOrgName(),
@@ -6714,7 +6716,7 @@ function startSpanParentArgs(args) {
     if (args.parentSpanIds) {
       throw new Error("Cannot specify both parent and parentSpanIds");
     }
-    const parentComponents = SpanComponentsV3.fromStr(args.parent);
+    const parentComponents = getSpanComponentsClass().fromStr(args.parent);
     if (args.parentObjectType !== parentComponents.data.object_type) {
       throw new Error(
         `Mismatch between expected span parent object type ${args.parentObjectType} and provided type ${parentComponents.data.object_type}`
@@ -7158,7 +7160,7 @@ function getSpanParentObject(options) {
     return parentSpan;
   }
   const parentStr = options?.parent ?? state.currentParent.getStore();
-  if (parentStr) return SpanComponentsV3.fromStr(parentStr);
+  if (parentStr) return getSpanComponentsClass().fromStr(parentStr);
   const experiment = currentExperiment();
   if (experiment) {
     return experiment;
@@ -7224,7 +7226,7 @@ function startSpanAndIsLogger(args) {
     parent: args?.parent,
     state
   });
-  if (parentObject instanceof SpanComponentsV3) {
+  if (parentObject instanceof SpanComponentsV3 || parentObject instanceof SpanComponentsV4) {
     const parentSpanIds = parentObject.data.row_id ? {
       spanId: parentObject.data.span_id,
       rootSpanId: parentObject.data.root_span_id
@@ -9071,7 +9073,10 @@ Error: ${errorText}`;
                   limit: INTERNAL_BTQL_LIMIT
                 },
                 use_columnstore: false,
-                brainstore_realtime: true
+                brainstore_realtime: true,
+                ...this.pinnedVersion !== void 0 ? {
+                  version: this.pinnedVersion
+                } : {}
               },
               { headers: { "Accept-Encoding": "gzip" } }
             );
@@ -10228,7 +10233,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "braintrust",
-      version: "0.4.4",
+      version: "0.4.6",
       description: "SDK for integrating Braintrust",
       repository: {
         type: "git",
@@ -10284,10 +10289,11 @@ var require_package = __commonJS({
         watch: "tsup --watch",
         clean: "rm -r dist/* && rm -r dev/dist/*",
         docs: "npx typedoc --options typedoc.json src/index.ts",
-        test: "vitest run --exclude src/wrappers/anthropic.test.ts --exclude src/wrappers/oai.test.ts --exclude src/otel.test.ts --exclude src/otel-no-deps.test.ts  --exclude src/wrappers/ai-sdk-v1.test.ts --exclude src/wrappers/ai-sdk-v2.test.ts --exclude src/wrappers/ai-sdk-v3.test.ts --exclude src/wrappers/mastra.test.ts --exclude src/wrappers/claude-agent-sdk.test.ts",
+        test: "vitest run --exclude src/wrappers/anthropic.test.ts --exclude src/wrappers/oai.test.ts --exclude src/otel.test.ts --exclude src/otel-no-deps.test.ts  --exclude src/wrappers/ai-sdk-v1.test.ts --exclude src/wrappers/ai-sdk-v2.test.ts --exclude src/wrappers/ai-sdk-v3.test.ts --exclude src/wrappers/mastra.test.ts --exclude src/wrappers/claude-agent-sdk.test.ts --exclude src/wrappers/google-genai.test.ts",
         "test:anthropic": "vitest run src/wrappers/anthropic.test.ts",
         "test:openai": "vitest run src/wrappers/oai.test.ts",
         "test:otel": "vitest run src/otel.test.ts",
+        "test:google-genai": "vitest run src/wrappers/google-genai.test.ts",
         "test:otel-no-deps": "vitest run src/otel-no-deps.test.ts --reporter=verbose",
         "test:ai-sdk-v1": "vitest run src/wrappers/ai-sdk-v1.test.ts",
         "test:ai-sdk-v2": "vitest run src/wrappers/ai-sdk-v2.test.ts src/wrappers/ai-sdk-v1.test.ts",
@@ -10300,6 +10306,7 @@ var require_package = __commonJS({
       devDependencies: {
         "@ai-sdk/anthropic": "^1.0.5",
         "@anthropic-ai/sdk": "^0.60.0",
+        "@google/genai": "^1.25.0",
         "@jest/globals": "^29.7.0",
         "@nodelib/fs.walk": "^1.2.8",
         "@openai/agents": "^0.0.14",
@@ -13731,9 +13738,17 @@ function makeCheckAuthorized(allowedOrgName) {
       return next((0, import_http_errors.default)(401, "Unauthorized"));
     }
     try {
+      const orgName = parseHeader(req.headers, "x-bt-org-name");
+      if (!orgName) {
+        return next((0, import_http_errors.default)(400, "Missing x-bt-org-name header"));
+      }
+      if (allowedOrgName && allowedOrgName !== orgName) {
+        const errorMessage = `Org '${orgName}' is not allowed. Only org '${allowedOrgName}' is allowed.`;
+        return next((0, import_http_errors.default)(403, errorMessage));
+      }
       const state = await cachedLogin({
         apiKey: req.ctx?.token,
-        orgName: allowedOrgName
+        orgName
       });
       req.ctx.state = state;
       next();