braintrust 0.0.180 → 0.0.182

package/dist/browser.d.mts

@@ -521,6 +521,32 @@ declare class Attachment {
     private initUploader;
     private initData;
 }
+declare const attachmentMetadataSchema: z.ZodObject<{
+    downloadUrl: z.ZodString;
+    status: z.ZodObject<{
+        upload_status: z.ZodEnum<["uploading", "done", "error"]>;
+        error_message: z.ZodEffects<z.ZodOptional<z.ZodNullable<z.ZodString>>, string | undefined, string | null | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    }, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    }>;
+}, "strip", z.ZodTypeAny, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    };
+    downloadUrl: string;
+}, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    };
+    downloadUrl: string;
+}>;
+type AttachmentMetadata = z.infer<typeof attachmentMetadataSchema>;
 /**
  * A readonly alternative to `Attachment`, which can be used for fetching
  * already-uploaded Attachments.
@@ -546,13 +572,17 @@ declare class ReadonlyAttachment {
      * contents from the object store on first access.
      */
     data(): Promise<Blob>;
+    /**
+     * Fetch the attachment metadata, which includes a downloadUrl and a status.
+     * This will re-fetch the status each time in case it changes over time.
+     */
+    metadata(): Promise<AttachmentMetadata>;
     /**
      * Fetch the attachment upload status. This will re-fetch the status each time
      * in case it changes over time.
      */
     status(): Promise<AttachmentStatus>;
     private initDownloader;
-    private fetchMetadata;
 }
 /**
  * Update a span using the output of `span.export()`. It is important that you only resume updating
@@ -1815,6 +1845,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/browser.d.ts

@@ -521,6 +521,32 @@ declare class Attachment {
     private initUploader;
     private initData;
 }
+declare const attachmentMetadataSchema: z.ZodObject<{
+    downloadUrl: z.ZodString;
+    status: z.ZodObject<{
+        upload_status: z.ZodEnum<["uploading", "done", "error"]>;
+        error_message: z.ZodEffects<z.ZodOptional<z.ZodNullable<z.ZodString>>, string | undefined, string | null | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    }, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    }>;
+}, "strip", z.ZodTypeAny, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    };
+    downloadUrl: string;
+}, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    };
+    downloadUrl: string;
+}>;
+type AttachmentMetadata = z.infer<typeof attachmentMetadataSchema>;
 /**
  * A readonly alternative to `Attachment`, which can be used for fetching
  * already-uploaded Attachments.
@@ -546,13 +572,17 @@ declare class ReadonlyAttachment {
      * contents from the object store on first access.
      */
     data(): Promise<Blob>;
+    /**
+     * Fetch the attachment metadata, which includes a downloadUrl and a status.
+     * This will re-fetch the status each time in case it changes over time.
+     */
+    metadata(): Promise<AttachmentMetadata>;
     /**
      * Fetch the attachment upload status. This will re-fetch the status each time
      * in case it changes over time.
      */
     status(): Promise<AttachmentStatus>;
     private initDownloader;
-    private fetchMetadata;
 }
 /**
  * Update a span using the output of `span.export()`. It is important that you only resume updating
@@ -1815,6 +1845,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/browser.js

@@ -1189,16 +1189,35 @@ var ReadonlyAttachment = class {
   async data() {
     return this._data.get();
   }
+  /**
+   * Fetch the attachment metadata, which includes a downloadUrl and a status.
+   * This will re-fetch the status each time in case it changes over time.
+   */
+  async metadata() {
+    const state = this.state ?? _globalState;
+    await state.login({});
+    const resp = await state.apiConn().get("/attachment", {
+      key: this.reference.key,
+      filename: this.reference.filename,
+      content_type: this.reference.content_type,
+      org_id: state.orgId || ""
+    });
+    if (!resp.ok) {
+      const errorStr = JSON.stringify(resp);
+      throw new Error(`Invalid response from API server: ${errorStr}`);
+    }
+    return attachmentMetadataSchema.parse(await resp.json());
+  }
   /**
    * Fetch the attachment upload status. This will re-fetch the status each time
    * in case it changes over time.
    */
   async status() {
-    return (await this.fetchMetadata()).status;
+    return (await this.metadata()).status;
   }
   initDownloader() {
     const download = async () => {
-      const { downloadUrl, status } = await this.fetchMetadata();
+      const { downloadUrl, status } = await this.metadata();
       if (status.upload_status !== "done") {
         throw new Error(
           `Expected attachment status "done", got "${status.upload_status}"`
@@ -1213,21 +1232,6 @@ var ReadonlyAttachment = class {
     };
     return new LazyValue(download);
   }
-  async fetchMetadata() {
-    const state = this.state ?? _globalState;
-    await state.login({});
-    const resp = await state.apiConn().get("/attachment", {
-      key: this.reference.key,
-      filename: this.reference.filename,
-      content_type: this.reference.content_type,
-      org_id: state.orgId || ""
-    });
-    if (!resp.ok) {
-      const errorStr = JSON.stringify(resp);
-      throw new Error(`Invalid response from API server: ${errorStr}`);
-    }
-    return attachmentMetadataSchema.parse(await resp.json());
-  }
 };
 function logFeedbackImpl(state, parentObjectType, parentObjectId, {
   id,
@@ -3986,16 +3990,11 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  let completionProxy = new Proxy(openai.chat.completions, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapChatCompletion(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
-  let chatProxy = new Proxy(openai.chat, {
+  const completionProxy = createEndpointProxy(
+    openai.chat.completions,
+    wrapChatCompletion
+  );
+  const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
         return completionProxy;
@@ -4003,18 +4002,11 @@ function wrapOpenAIv4(openai) {
       return Reflect.get(target, name, receiver);
     }
   });
-  let embeddingProxy = new Proxy(openai.embeddings, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapEmbeddings(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
+  const embeddingProxy = createEndpointProxy(openai.embeddings, wrapEmbeddings);
+  const moderationProxy = createEndpointProxy(openai.moderations, wrapModerations);
   let betaProxy;
   if (openai.beta?.chat?.completions?.stream) {
-    let betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
+    const betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
       get(target, name, receiver) {
         const baseVal = Reflect.get(target, name, receiver);
         if (name === "parse") {
@@ -4025,7 +4017,7 @@ function wrapOpenAIv4(openai) {
         return baseVal;
       }
     });
-    let betaChatProxy = new Proxy(openai.beta.chat, {
+    const betaChatProxy = new Proxy(openai.beta.chat, {
       get(target, name, receiver) {
         if (name === "completions") {
           return betaChatCompletionProxy;
@@ -4042,7 +4034,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -4050,13 +4042,15 @@ function wrapOpenAIv4(openai) {
       if (name === "embeddings") {
         return embeddingProxy;
       }
+      if (name === "moderations") {
+        return moderationProxy;
+      }
       if (name === "beta" && betaProxy) {
         return betaProxy;
       }
       return Reflect.get(target, name, receiver);
     }
   });
-  return proxy;
 }
 function logCompletionResponse(startTime, response, span) {
   span.log({
@@ -4205,7 +4199,7 @@ function wrapChatCompletion(completion) {
     }
   };
 }
-function parseChatCompletionParams(allParams) {
+function parseBaseParams(allParams, inputField) {
   const { span_info, ...params } = allParams;
   const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
   let ret = {
@@ -4214,10 +4208,12 @@ function parseChatCompletionParams(allParams) {
       metadata: spanInfoMetadata
     }
   };
-  const { messages, ...paramsRest } = params;
-  return (0, import_core3.mergeDicts)(ret, { event: { input: messages, metadata: paramsRest } });
+  const input = params[inputField];
+  const paramsRest = { ...params };
+  delete paramsRest[inputField];
+  return (0, import_core3.mergeDicts)(ret, { event: { input, metadata: paramsRest } });
 }
-function wrapEmbeddings(create) {
+function createApiWrapper(name, create, processResponse, parseParams) {
   return async (allParams, options) => {
     const { span_info: _, ...params } = allParams;
     return traced(
@@ -4227,42 +4223,61 @@ function wrapEmbeddings(create) {
           options
         ).withResponse();
         logHeaders(response, span);
-        const embedding_length = result.data[0].embedding.length;
-        span.log({
-          // TODO: Add a flag to control whether to log the full embedding vector,
-          // possibly w/ JSON compression.
-          output: { embedding_length },
-          metrics: {
-            tokens: result.usage?.total_tokens,
-            prompt_tokens: result.usage?.prompt_tokens
-          }
-        });
+        processResponse(result, span);
         return result;
       },
       (0, import_core3.mergeDicts)(
         {
-          name: "Embedding",
+          name,
           spanAttributes: {
             type: import_core2.SpanTypeAttribute.LLM
           }
         },
-        parseEmbeddingParams(allParams)
+        parseParams(allParams)
       )
     );
   };
 }
-function parseEmbeddingParams(allParams) {
-  const { span_info, ...params } = allParams;
-  const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
-  let ret = {
-    ...spanInfoRest,
-    event: {
-      metadata: spanInfoMetadata
+function createEndpointProxy(target, wrapperFn) {
+  return new Proxy(target, {
+    get(target2, name, receiver) {
+      const baseVal = Reflect.get(target2, name, receiver);
+      if (name === "create") {
+        return wrapperFn(baseVal.bind(target2));
+      }
+      return baseVal;
     }
-  };
-  const { input, ...paramsRest } = params;
-  return (0, import_core3.mergeDicts)(ret, { event: { input, metadata: paramsRest } });
+  });
+}
+function parseChatCompletionParams(params) {
+  return parseBaseParams(params, "messages");
+}
+function processEmbeddingResponse(result, span) {
+  span.log({
+    output: { embedding_length: result.data[0].embedding.length },
+    metrics: {
+      tokens: result.usage?.total_tokens,
+      prompt_tokens: result.usage?.prompt_tokens
+    }
+  });
+}
+function processModerationResponse(result, span) {
+  span.log({
+    output: result.results
+  });
 }
+var wrapEmbeddings = (create) => createApiWrapper(
+  "Embedding",
+  create,
+  processEmbeddingResponse,
+  (params) => parseBaseParams(params, "input")
+);
+var wrapModerations = (create) => createApiWrapper(
+  "Moderation",
+  create,
+  processModerationResponse,
+  (params) => parseBaseParams(params, "input")
+);
 function postprocessStreamingResults(allResults) {
   let role = void 0;
   let content = void 0;

package/dist/browser.mjs

@@ -1137,16 +1137,35 @@ var ReadonlyAttachment = class {
   async data() {
     return this._data.get();
   }
+  /**
+   * Fetch the attachment metadata, which includes a downloadUrl and a status.
+   * This will re-fetch the status each time in case it changes over time.
+   */
+  async metadata() {
+    const state = this.state ?? _globalState;
+    await state.login({});
+    const resp = await state.apiConn().get("/attachment", {
+      key: this.reference.key,
+      filename: this.reference.filename,
+      content_type: this.reference.content_type,
+      org_id: state.orgId || ""
+    });
+    if (!resp.ok) {
+      const errorStr = JSON.stringify(resp);
+      throw new Error(`Invalid response from API server: ${errorStr}`);
+    }
+    return attachmentMetadataSchema.parse(await resp.json());
+  }
   /**
    * Fetch the attachment upload status. This will re-fetch the status each time
    * in case it changes over time.
    */
   async status() {
-    return (await this.fetchMetadata()).status;
+    return (await this.metadata()).status;
   }
   initDownloader() {
     const download = async () => {
-      const { downloadUrl, status } = await this.fetchMetadata();
+      const { downloadUrl, status } = await this.metadata();
       if (status.upload_status !== "done") {
         throw new Error(
           `Expected attachment status "done", got "${status.upload_status}"`
@@ -1161,21 +1180,6 @@ var ReadonlyAttachment = class {
     };
     return new LazyValue(download);
   }
-  async fetchMetadata() {
-    const state = this.state ?? _globalState;
-    await state.login({});
-    const resp = await state.apiConn().get("/attachment", {
-      key: this.reference.key,
-      filename: this.reference.filename,
-      content_type: this.reference.content_type,
-      org_id: state.orgId || ""
-    });
-    if (!resp.ok) {
-      const errorStr = JSON.stringify(resp);
-      throw new Error(`Invalid response from API server: ${errorStr}`);
-    }
-    return attachmentMetadataSchema.parse(await resp.json());
-  }
 };
 function logFeedbackImpl(state, parentObjectType, parentObjectId, {
   id,
@@ -3936,16 +3940,11 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  let completionProxy = new Proxy(openai.chat.completions, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapChatCompletion(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
-  let chatProxy = new Proxy(openai.chat, {
+  const completionProxy = createEndpointProxy(
+    openai.chat.completions,
+    wrapChatCompletion
+  );
+  const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
         return completionProxy;
@@ -3953,18 +3952,11 @@ function wrapOpenAIv4(openai) {
       return Reflect.get(target, name, receiver);
     }
   });
-  let embeddingProxy = new Proxy(openai.embeddings, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapEmbeddings(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
+  const embeddingProxy = createEndpointProxy(openai.embeddings, wrapEmbeddings);
+  const moderationProxy = createEndpointProxy(openai.moderations, wrapModerations);
   let betaProxy;
   if (openai.beta?.chat?.completions?.stream) {
-    let betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
+    const betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
       get(target, name, receiver) {
         const baseVal = Reflect.get(target, name, receiver);
         if (name === "parse") {
@@ -3975,7 +3967,7 @@ function wrapOpenAIv4(openai) {
         return baseVal;
       }
     });
-    let betaChatProxy = new Proxy(openai.beta.chat, {
+    const betaChatProxy = new Proxy(openai.beta.chat, {
       get(target, name, receiver) {
         if (name === "completions") {
           return betaChatCompletionProxy;
@@ -3992,7 +3984,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -4000,13 +3992,15 @@ function wrapOpenAIv4(openai) {
       if (name === "embeddings") {
         return embeddingProxy;
       }
+      if (name === "moderations") {
+        return moderationProxy;
+      }
       if (name === "beta" && betaProxy) {
         return betaProxy;
       }
       return Reflect.get(target, name, receiver);
     }
   });
-  return proxy;
 }
 function logCompletionResponse(startTime, response, span) {
   span.log({
@@ -4155,7 +4149,7 @@ function wrapChatCompletion(completion) {
     }
   };
 }
-function parseChatCompletionParams(allParams) {
+function parseBaseParams(allParams, inputField) {
   const { span_info, ...params } = allParams;
   const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
   let ret = {
@@ -4164,10 +4158,12 @@ function parseChatCompletionParams(allParams) {
       metadata: spanInfoMetadata
     }
   };
-  const { messages, ...paramsRest } = params;
-  return mergeDicts2(ret, { event: { input: messages, metadata: paramsRest } });
+  const input = params[inputField];
+  const paramsRest = { ...params };
+  delete paramsRest[inputField];
+  return mergeDicts2(ret, { event: { input, metadata: paramsRest } });
 }
-function wrapEmbeddings(create) {
+function createApiWrapper(name, create, processResponse, parseParams) {
   return async (allParams, options) => {
     const { span_info: _, ...params } = allParams;
     return traced(
@@ -4177,42 +4173,61 @@ function wrapEmbeddings(create) {
           options
         ).withResponse();
         logHeaders(response, span);
-        const embedding_length = result.data[0].embedding.length;
-        span.log({
-          // TODO: Add a flag to control whether to log the full embedding vector,
-          // possibly w/ JSON compression.
-          output: { embedding_length },
-          metrics: {
-            tokens: result.usage?.total_tokens,
-            prompt_tokens: result.usage?.prompt_tokens
-          }
-        });
+        processResponse(result, span);
         return result;
       },
       mergeDicts2(
         {
-          name: "Embedding",
+          name,
           spanAttributes: {
             type: SpanTypeAttribute2.LLM
           }
         },
-        parseEmbeddingParams(allParams)
+        parseParams(allParams)
       )
     );
   };
 }
-function parseEmbeddingParams(allParams) {
-  const { span_info, ...params } = allParams;
-  const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
-  let ret = {
-    ...spanInfoRest,
-    event: {
-      metadata: spanInfoMetadata
+function createEndpointProxy(target, wrapperFn) {
+  return new Proxy(target, {
+    get(target2, name, receiver) {
+      const baseVal = Reflect.get(target2, name, receiver);
+      if (name === "create") {
+        return wrapperFn(baseVal.bind(target2));
+      }
+      return baseVal;
     }
-  };
-  const { input, ...paramsRest } = params;
-  return mergeDicts2(ret, { event: { input, metadata: paramsRest } });
+  });
+}
+function parseChatCompletionParams(params) {
+  return parseBaseParams(params, "messages");
+}
+function processEmbeddingResponse(result, span) {
+  span.log({
+    output: { embedding_length: result.data[0].embedding.length },
+    metrics: {
+      tokens: result.usage?.total_tokens,
+      prompt_tokens: result.usage?.prompt_tokens
+    }
+  });
+}
+function processModerationResponse(result, span) {
+  span.log({
+    output: result.results
+  });
 }
+var wrapEmbeddings = (create) => createApiWrapper(
+  "Embedding",
+  create,
+  processEmbeddingResponse,
+  (params) => parseBaseParams(params, "input")
+);
+var wrapModerations = (create) => createApiWrapper(
+  "Moderation",
+  create,
+  processModerationResponse,
+  (params) => parseBaseParams(params, "input")
+);
 function postprocessStreamingResults(allResults) {
   let role = void 0;
   let content = void 0;

package/dist/cli.js

@@ -1236,7 +1236,7 @@ var require_package = __commonJS({
   "package.json"(exports2, module2) {
     module2.exports = {
       name: "braintrust",
-      version: "0.0.180",
+      version: "0.0.182",
       description: "SDK for integrating Braintrust",
       repository: {
         type: "git",
@@ -1311,7 +1311,7 @@ var require_package = __commonJS({
       },
       dependencies: {
         "@ai-sdk/provider": "^1.0.1",
-        "@braintrust/core": "0.0.74",
+        "@braintrust/core": "0.0.76",
         "@next/env": "^14.2.3",
         "@vercel/functions": "^1.0.2",
         ai: "^3.2.16",
@@ -2457,16 +2457,35 @@ var ReadonlyAttachment = class {
   async data() {
     return this._data.get();
   }
+  /**
+   * Fetch the attachment metadata, which includes a downloadUrl and a status.
+   * This will re-fetch the status each time in case it changes over time.
+   */
+  async metadata() {
+    const state = this.state ?? _globalState;
+    await state.login({});
+    const resp = await state.apiConn().get("/attachment", {
+      key: this.reference.key,
+      filename: this.reference.filename,
+      content_type: this.reference.content_type,
+      org_id: state.orgId || ""
+    });
+    if (!resp.ok) {
+      const errorStr = JSON.stringify(resp);
+      throw new Error(`Invalid response from API server: ${errorStr}`);
+    }
+    return attachmentMetadataSchema.parse(await resp.json());
+  }
   /**
    * Fetch the attachment upload status. This will re-fetch the status each time
    * in case it changes over time.
    */
   async status() {
-    return (await this.fetchMetadata()).status;
+    return (await this.metadata()).status;
   }
   initDownloader() {
     const download = async () => {
-      const { downloadUrl, status } = await this.fetchMetadata();
+      const { downloadUrl, status } = await this.metadata();
       if (status.upload_status !== "done") {
         throw new Error(
           `Expected attachment status "done", got "${status.upload_status}"`
@@ -2481,21 +2500,6 @@ var ReadonlyAttachment = class {
     };
     return new LazyValue(download);
   }
-  async fetchMetadata() {
-    const state = this.state ?? _globalState;
-    await state.login({});
-    const resp = await state.apiConn().get("/attachment", {
-      key: this.reference.key,
-      filename: this.reference.filename,
-      content_type: this.reference.content_type,
-      org_id: state.orgId || ""
-    });
-    if (!resp.ok) {
-      const errorStr = JSON.stringify(resp);
-      throw new Error(`Invalid response from API server: ${errorStr}`);
-    }
-    return attachmentMetadataSchema.parse(await resp.json());
-  }
 };
 function logFeedbackImpl(state, parentObjectType, parentObjectId, {
   id,

package/dist/index.d.mts

@@ -523,6 +523,32 @@ declare class Attachment {
     private initUploader;
     private initData;
 }
+declare const attachmentMetadataSchema: z.ZodObject<{
+    downloadUrl: z.ZodString;
+    status: z.ZodObject<{
+        upload_status: z.ZodEnum<["uploading", "done", "error"]>;
+        error_message: z.ZodEffects<z.ZodOptional<z.ZodNullable<z.ZodString>>, string | undefined, string | null | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    }, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    }>;
+}, "strip", z.ZodTypeAny, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    };
+    downloadUrl: string;
+}, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    };
+    downloadUrl: string;
+}>;
+type AttachmentMetadata = z.infer<typeof attachmentMetadataSchema>;
 /**
  * A readonly alternative to `Attachment`, which can be used for fetching
  * already-uploaded Attachments.
@@ -548,13 +574,17 @@ declare class ReadonlyAttachment {
      * contents from the object store on first access.
      */
     data(): Promise<Blob>;
+    /**
+     * Fetch the attachment metadata, which includes a downloadUrl and a status.
+     * This will re-fetch the status each time in case it changes over time.
+     */
+    metadata(): Promise<AttachmentMetadata>;
     /**
      * Fetch the attachment upload status. This will re-fetch the status each time
      * in case it changes over time.
      */
     status(): Promise<AttachmentStatus>;
     private initDownloader;
-    private fetchMetadata;
 }
 /**
  * Update a span using the output of `span.export()`. It is important that you only resume updating
@@ -2170,6 +2200,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/index.d.ts

@@ -523,6 +523,32 @@ declare class Attachment {
     private initUploader;
     private initData;
 }
+declare const attachmentMetadataSchema: z.ZodObject<{
+    downloadUrl: z.ZodString;
+    status: z.ZodObject<{
+        upload_status: z.ZodEnum<["uploading", "done", "error"]>;
+        error_message: z.ZodEffects<z.ZodOptional<z.ZodNullable<z.ZodString>>, string | undefined, string | null | undefined>;
+    }, "strip", z.ZodTypeAny, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    }, {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    }>;
+}, "strip", z.ZodTypeAny, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | undefined;
+    };
+    downloadUrl: string;
+}, {
+    status: {
+        upload_status: "error" | "done" | "uploading";
+        error_message?: string | null | undefined;
+    };
+    downloadUrl: string;
+}>;
+type AttachmentMetadata = z.infer<typeof attachmentMetadataSchema>;
 /**
  * A readonly alternative to `Attachment`, which can be used for fetching
  * already-uploaded Attachments.
@@ -548,13 +574,17 @@ declare class ReadonlyAttachment {
      * contents from the object store on first access.
      */
     data(): Promise<Blob>;
+    /**
+     * Fetch the attachment metadata, which includes a downloadUrl and a status.
+     * This will re-fetch the status each time in case it changes over time.
+     */
+    metadata(): Promise<AttachmentMetadata>;
     /**
      * Fetch the attachment upload status. This will re-fetch the status each time
      * in case it changes over time.
      */
     status(): Promise<AttachmentStatus>;
     private initDownloader;
-    private fetchMetadata;
 }
 /**
  * Update a span using the output of `span.export()`. It is important that you only resume updating
@@ -2170,6 +2200,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/index.js

@@ -1425,16 +1425,35 @@ var ReadonlyAttachment = class {
   async data() {
     return this._data.get();
   }
+  /**
+   * Fetch the attachment metadata, which includes a downloadUrl and a status.
+   * This will re-fetch the status each time in case it changes over time.
+   */
+  async metadata() {
+    const state = this.state ?? _globalState;
+    await state.login({});
+    const resp = await state.apiConn().get("/attachment", {
+      key: this.reference.key,
+      filename: this.reference.filename,
+      content_type: this.reference.content_type,
+      org_id: state.orgId || ""
+    });
+    if (!resp.ok) {
+      const errorStr = JSON.stringify(resp);
+      throw new Error(`Invalid response from API server: ${errorStr}`);
+    }
+    return attachmentMetadataSchema.parse(await resp.json());
+  }
   /**
    * Fetch the attachment upload status. This will re-fetch the status each time
    * in case it changes over time.
    */
   async status() {
-    return (await this.fetchMetadata()).status;
+    return (await this.metadata()).status;
   }
   initDownloader() {
     const download = async () => {
-      const { downloadUrl, status } = await this.fetchMetadata();
+      const { downloadUrl, status } = await this.metadata();
       if (status.upload_status !== "done") {
         throw new Error(
           `Expected attachment status "done", got "${status.upload_status}"`
@@ -1449,21 +1468,6 @@ var ReadonlyAttachment = class {
     };
     return new LazyValue(download);
   }
-  async fetchMetadata() {
-    const state = this.state ?? _globalState;
-    await state.login({});
-    const resp = await state.apiConn().get("/attachment", {
-      key: this.reference.key,
-      filename: this.reference.filename,
-      content_type: this.reference.content_type,
-      org_id: state.orgId || ""
-    });
-    if (!resp.ok) {
-      const errorStr = JSON.stringify(resp);
-      throw new Error(`Invalid response from API server: ${errorStr}`);
-    }
-    return attachmentMetadataSchema.parse(await resp.json());
-  }
 };
 function logFeedbackImpl(state, parentObjectType, parentObjectId, {
   id,
@@ -6186,16 +6190,11 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  let completionProxy = new Proxy(openai.chat.completions, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapChatCompletion(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
-  let chatProxy = new Proxy(openai.chat, {
+  const completionProxy = createEndpointProxy(
+    openai.chat.completions,
+    wrapChatCompletion
+  );
+  const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
         return completionProxy;
@@ -6203,18 +6202,11 @@ function wrapOpenAIv4(openai) {
       return Reflect.get(target, name, receiver);
     }
   });
-  let embeddingProxy = new Proxy(openai.embeddings, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapEmbeddings(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
+  const embeddingProxy = createEndpointProxy(openai.embeddings, wrapEmbeddings);
+  const moderationProxy = createEndpointProxy(openai.moderations, wrapModerations);
   let betaProxy;
   if (openai.beta?.chat?.completions?.stream) {
-    let betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
+    const betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
       get(target, name, receiver) {
         const baseVal = Reflect.get(target, name, receiver);
         if (name === "parse") {
@@ -6225,7 +6217,7 @@ function wrapOpenAIv4(openai) {
         return baseVal;
       }
     });
-    let betaChatProxy = new Proxy(openai.beta.chat, {
+    const betaChatProxy = new Proxy(openai.beta.chat, {
       get(target, name, receiver) {
         if (name === "completions") {
           return betaChatCompletionProxy;
@@ -6242,7 +6234,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -6250,13 +6242,15 @@ function wrapOpenAIv4(openai) {
       if (name === "embeddings") {
         return embeddingProxy;
       }
+      if (name === "moderations") {
+        return moderationProxy;
+      }
       if (name === "beta" && betaProxy) {
         return betaProxy;
       }
       return Reflect.get(target, name, receiver);
     }
   });
-  return proxy;
 }
 function logCompletionResponse(startTime, response, span) {
   span.log({
@@ -6405,7 +6399,7 @@ function wrapChatCompletion(completion) {
     }
   };
 }
-function parseChatCompletionParams(allParams) {
+function parseBaseParams(allParams, inputField) {
   const { span_info, ...params } = allParams;
   const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
   let ret = {
@@ -6414,10 +6408,12 @@ function parseChatCompletionParams(allParams) {
       metadata: spanInfoMetadata
     }
   };
-  const { messages, ...paramsRest } = params;
-  return (0, import_core4.mergeDicts)(ret, { event: { input: messages, metadata: paramsRest } });
+  const input = params[inputField];
+  const paramsRest = { ...params };
+  delete paramsRest[inputField];
+  return (0, import_core4.mergeDicts)(ret, { event: { input, metadata: paramsRest } });
 }
-function wrapEmbeddings(create) {
+function createApiWrapper(name, create, processResponse, parseParams) {
   return async (allParams, options) => {
     const { span_info: _, ...params } = allParams;
     return traced(
@@ -6427,42 +6423,61 @@ function wrapEmbeddings(create) {
           options
         ).withResponse();
         logHeaders(response, span);
-        const embedding_length = result.data[0].embedding.length;
-        span.log({
-          // TODO: Add a flag to control whether to log the full embedding vector,
-          // possibly w/ JSON compression.
-          output: { embedding_length },
-          metrics: {
-            tokens: result.usage?.total_tokens,
-            prompt_tokens: result.usage?.prompt_tokens
-          }
-        });
+        processResponse(result, span);
         return result;
       },
       (0, import_core4.mergeDicts)(
         {
-          name: "Embedding",
+          name,
           spanAttributes: {
             type: import_core3.SpanTypeAttribute.LLM
           }
         },
-        parseEmbeddingParams(allParams)
+        parseParams(allParams)
       )
     );
   };
 }
-function parseEmbeddingParams(allParams) {
-  const { span_info, ...params } = allParams;
-  const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
-  let ret = {
-    ...spanInfoRest,
-    event: {
-      metadata: spanInfoMetadata
+function createEndpointProxy(target, wrapperFn) {
+  return new Proxy(target, {
+    get(target2, name, receiver) {
+      const baseVal = Reflect.get(target2, name, receiver);
+      if (name === "create") {
+        return wrapperFn(baseVal.bind(target2));
+      }
+      return baseVal;
     }
-  };
-  const { input, ...paramsRest } = params;
-  return (0, import_core4.mergeDicts)(ret, { event: { input, metadata: paramsRest } });
+  });
+}
+function parseChatCompletionParams(params) {
+  return parseBaseParams(params, "messages");
+}
+function processEmbeddingResponse(result, span) {
+  span.log({
+    output: { embedding_length: result.data[0].embedding.length },
+    metrics: {
+      tokens: result.usage?.total_tokens,
+      prompt_tokens: result.usage?.prompt_tokens
+    }
+  });
+}
+function processModerationResponse(result, span) {
+  span.log({
+    output: result.results
+  });
 }
+var wrapEmbeddings = (create) => createApiWrapper(
+  "Embedding",
+  create,
+  processEmbeddingResponse,
+  (params) => parseBaseParams(params, "input")
+);
+var wrapModerations = (create) => createApiWrapper(
+  "Moderation",
+  create,
+  processModerationResponse,
+  (params) => parseBaseParams(params, "input")
+);
 function postprocessStreamingResults(allResults) {
   let role = void 0;
   let content = void 0;

package/dist/index.mjs

@@ -1357,16 +1357,35 @@ var ReadonlyAttachment = class {
   async data() {
     return this._data.get();
   }
+  /**
+   * Fetch the attachment metadata, which includes a downloadUrl and a status.
+   * This will re-fetch the status each time in case it changes over time.
+   */
+  async metadata() {
+    const state = this.state ?? _globalState;
+    await state.login({});
+    const resp = await state.apiConn().get("/attachment", {
+      key: this.reference.key,
+      filename: this.reference.filename,
+      content_type: this.reference.content_type,
+      org_id: state.orgId || ""
+    });
+    if (!resp.ok) {
+      const errorStr = JSON.stringify(resp);
+      throw new Error(`Invalid response from API server: ${errorStr}`);
+    }
+    return attachmentMetadataSchema.parse(await resp.json());
+  }
   /**
    * Fetch the attachment upload status. This will re-fetch the status each time
    * in case it changes over time.
    */
   async status() {
-    return (await this.fetchMetadata()).status;
+    return (await this.metadata()).status;
   }
   initDownloader() {
     const download = async () => {
-      const { downloadUrl, status } = await this.fetchMetadata();
+      const { downloadUrl, status } = await this.metadata();
       if (status.upload_status !== "done") {
         throw new Error(
           `Expected attachment status "done", got "${status.upload_status}"`
@@ -1381,21 +1400,6 @@ var ReadonlyAttachment = class {
     };
     return new LazyValue(download);
   }
-  async fetchMetadata() {
-    const state = this.state ?? _globalState;
-    await state.login({});
-    const resp = await state.apiConn().get("/attachment", {
-      key: this.reference.key,
-      filename: this.reference.filename,
-      content_type: this.reference.content_type,
-      org_id: state.orgId || ""
-    });
-    if (!resp.ok) {
-      const errorStr = JSON.stringify(resp);
-      throw new Error(`Invalid response from API server: ${errorStr}`);
-    }
-    return attachmentMetadataSchema.parse(await resp.json());
-  }
 };
 function logFeedbackImpl(state, parentObjectType, parentObjectId, {
   id,
@@ -6122,16 +6126,11 @@ function wrapOpenAI(openai) {
 }
 globalThis.__inherited_braintrust_wrap_openai = wrapOpenAI;
 function wrapOpenAIv4(openai) {
-  let completionProxy = new Proxy(openai.chat.completions, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapChatCompletion(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
-  let chatProxy = new Proxy(openai.chat, {
+  const completionProxy = createEndpointProxy(
+    openai.chat.completions,
+    wrapChatCompletion
+  );
+  const chatProxy = new Proxy(openai.chat, {
     get(target, name, receiver) {
       if (name === "completions") {
         return completionProxy;
@@ -6139,18 +6138,11 @@ function wrapOpenAIv4(openai) {
       return Reflect.get(target, name, receiver);
     }
   });
-  let embeddingProxy = new Proxy(openai.embeddings, {
-    get(target, name, receiver) {
-      const baseVal = Reflect.get(target, name, receiver);
-      if (name === "create") {
-        return wrapEmbeddings(baseVal.bind(target));
-      }
-      return baseVal;
-    }
-  });
+  const embeddingProxy = createEndpointProxy(openai.embeddings, wrapEmbeddings);
+  const moderationProxy = createEndpointProxy(openai.moderations, wrapModerations);
   let betaProxy;
   if (openai.beta?.chat?.completions?.stream) {
-    let betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
+    const betaChatCompletionProxy = new Proxy(openai?.beta?.chat.completions, {
       get(target, name, receiver) {
         const baseVal = Reflect.get(target, name, receiver);
         if (name === "parse") {
@@ -6161,7 +6153,7 @@ function wrapOpenAIv4(openai) {
         return baseVal;
       }
     });
-    let betaChatProxy = new Proxy(openai.beta.chat, {
+    const betaChatProxy = new Proxy(openai.beta.chat, {
       get(target, name, receiver) {
         if (name === "completions") {
           return betaChatCompletionProxy;
@@ -6178,7 +6170,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -6186,13 +6178,15 @@ function wrapOpenAIv4(openai) {
       if (name === "embeddings") {
         return embeddingProxy;
       }
+      if (name === "moderations") {
+        return moderationProxy;
+      }
       if (name === "beta" && betaProxy) {
         return betaProxy;
       }
       return Reflect.get(target, name, receiver);
     }
   });
-  return proxy;
 }
 function logCompletionResponse(startTime, response, span) {
   span.log({
@@ -6341,7 +6335,7 @@ function wrapChatCompletion(completion) {
     }
   };
 }
-function parseChatCompletionParams(allParams) {
+function parseBaseParams(allParams, inputField) {
   const { span_info, ...params } = allParams;
   const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
   let ret = {
@@ -6350,10 +6344,12 @@ function parseChatCompletionParams(allParams) {
       metadata: spanInfoMetadata
     }
   };
-  const { messages, ...paramsRest } = params;
-  return mergeDicts3(ret, { event: { input: messages, metadata: paramsRest } });
+  const input = params[inputField];
+  const paramsRest = { ...params };
+  delete paramsRest[inputField];
+  return mergeDicts3(ret, { event: { input, metadata: paramsRest } });
 }
-function wrapEmbeddings(create) {
+function createApiWrapper(name, create, processResponse, parseParams) {
   return async (allParams, options) => {
     const { span_info: _, ...params } = allParams;
     return traced(
@@ -6363,42 +6359,61 @@ function wrapEmbeddings(create) {
           options
         ).withResponse();
         logHeaders(response, span);
-        const embedding_length = result.data[0].embedding.length;
-        span.log({
-          // TODO: Add a flag to control whether to log the full embedding vector,
-          // possibly w/ JSON compression.
-          output: { embedding_length },
-          metrics: {
-            tokens: result.usage?.total_tokens,
-            prompt_tokens: result.usage?.prompt_tokens
-          }
-        });
+        processResponse(result, span);
         return result;
       },
       mergeDicts3(
         {
-          name: "Embedding",
+          name,
           spanAttributes: {
             type: SpanTypeAttribute3.LLM
           }
         },
-        parseEmbeddingParams(allParams)
+        parseParams(allParams)
       )
     );
   };
 }
-function parseEmbeddingParams(allParams) {
-  const { span_info, ...params } = allParams;
-  const { metadata: spanInfoMetadata, ...spanInfoRest } = span_info ?? {};
-  let ret = {
-    ...spanInfoRest,
-    event: {
-      metadata: spanInfoMetadata
+function createEndpointProxy(target, wrapperFn) {
+  return new Proxy(target, {
+    get(target2, name, receiver) {
+      const baseVal = Reflect.get(target2, name, receiver);
+      if (name === "create") {
+        return wrapperFn(baseVal.bind(target2));
+      }
+      return baseVal;
     }
-  };
-  const { input, ...paramsRest } = params;
-  return mergeDicts3(ret, { event: { input, metadata: paramsRest } });
+  });
+}
+function parseChatCompletionParams(params) {
+  return parseBaseParams(params, "messages");
+}
+function processEmbeddingResponse(result, span) {
+  span.log({
+    output: { embedding_length: result.data[0].embedding.length },
+    metrics: {
+      tokens: result.usage?.total_tokens,
+      prompt_tokens: result.usage?.prompt_tokens
+    }
+  });
+}
+function processModerationResponse(result, span) {
+  span.log({
+    output: result.results
+  });
 }
+var wrapEmbeddings = (create) => createApiWrapper(
+  "Embedding",
+  create,
+  processEmbeddingResponse,
+  (params) => parseBaseParams(params, "input")
+);
+var wrapModerations = (create) => createApiWrapper(
+  "Moderation",
+  create,
+  processModerationResponse,
+  (params) => parseBaseParams(params, "input")
+);
 function postprocessStreamingResults(allResults) {
   let role = void 0;
   let content = void 0;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "braintrust",
-  "version": "0.0.180",
+  "version": "0.0.182",
   "description": "SDK for integrating Braintrust",
   "repository": {
     "type": "git",
@@ -75,7 +75,7 @@
   },
   "dependencies": {
     "@ai-sdk/provider": "^1.0.1",
-    "@braintrust/core": "0.0.74",
+    "@braintrust/core": "0.0.76",
     "@next/env": "^14.2.3",
     "@vercel/functions": "^1.0.2",
     "ai": "^3.2.16",