braintrust 0.0.181 → 0.0.182

package/dist/browser.d.mts

@@ -1845,6 +1845,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/browser.d.ts

@@ -1845,6 +1845,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/browser.js

@@ -3990,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;
@@ -4007,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") {
@@ -4029,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;
@@ -4046,7 +4034,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -4054,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({
@@ -4209,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 = {
@@ -4218,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(
@@ -4231,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

@@ -3940,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;
@@ -3957,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") {
@@ -3979,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;
@@ -3996,7 +3984,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -4004,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({
@@ -4159,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 = {
@@ -4168,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(
@@ -4181,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.181",
+      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.75",
+        "@braintrust/core": "0.0.76",
         "@next/env": "^14.2.3",
         "@vercel/functions": "^1.0.2",
         ai: "^3.2.16",

package/dist/index.d.mts

@@ -2200,6 +2200,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/index.d.ts

@@ -2200,6 +2200,7 @@ interface ChatLike {
 interface OpenAILike {
     chat: ChatLike;
     embeddings: any;
+    moderations: any;
     beta?: BetaLike;
 }
 declare global {

package/dist/index.js

@@ -6190,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;
@@ -6207,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") {
@@ -6229,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;
@@ -6246,7 +6234,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -6254,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({
@@ -6409,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 = {
@@ -6418,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(
@@ -6431,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

@@ -6126,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;
@@ -6143,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") {
@@ -6165,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;
@@ -6182,7 +6170,7 @@ function wrapOpenAIv4(openai) {
       }
     });
   }
-  let proxy = new Proxy(openai, {
+  return new Proxy(openai, {
     get(target, name, receiver) {
       if (name === "chat") {
         return chatProxy;
@@ -6190,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({
@@ -6345,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 = {
@@ -6354,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(
@@ -6367,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.181",
+  "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.75",
+    "@braintrust/core": "0.0.76",
     "@next/env": "^14.2.3",
     "@vercel/functions": "^1.0.2",
     "ai": "^3.2.16",