braintrust 3.26.0 → 3.27.0

package/dist/index.mjs

@@ -63,6 +63,7 @@ import {
   braintrustFlueObserver,
   braintrustStreamChunkSchema,
   buildBedrockRuntimeSpanInfo,
+  collectAnthropicSession,
   configureInstrumentation,
   configureNode,
   constructLogs3OverflowRequest,
@@ -140,7 +141,7 @@ import {
   wrapMastraAgent,
   wrapTraced,
   zodToJsonSchema
-} from "./chunk-XIZOM2HO.mjs";
+} from "./chunk-YKD22IMR.mjs";
 import {
   INSTRUMENTATION_NAMES,
   __export,
@@ -149,6 +150,7 @@ import {
   bedrockRuntimeChannels,
   claudeAgentSDKChannels,
   cloudflareAgentsChannels,
+  cloudflareThinkChannels,
   cohereChannels,
   cursorSDKChannels,
   genkitChannels,
@@ -163,6 +165,7 @@ import {
   isomorph_default,
   langSmithChannels,
   mistralChannels,
+  ollamaChannels,
   openAIChannels,
   openAICodexChannels,
   openRouterAgentChannels,
@@ -170,7 +173,7 @@ import {
   piCodingAgentChannels,
   strandsAgentSDKChannels,
   withSpanInstrumentationName
-} from "./chunk-PN7EWK66.mjs";
+} from "./chunk-CZM5JIQL.mjs";
 
 // src/exports.ts
 var exports_exports = {};
@@ -244,6 +247,7 @@ __export(exports_exports, {
   braintrustFlueObserver: () => braintrustFlueObserver,
   braintrustStreamChunkSchema: () => braintrustStreamChunkSchema,
   buildLocalSummary: () => buildLocalSummary,
+  collectAnthropicSession: () => collectAnthropicSession,
   configureInstrumentation: () => configureInstrumentation,
   constructLogs3OverflowRequest: () => constructLogs3OverflowRequest,
   createFinalValuePassThroughStream: () => createFinalValuePassThroughStream,
@@ -323,6 +327,7 @@ __export(exports_exports, {
   wrapClaudeAgentSDK: () => wrapClaudeAgentSDK,
   wrapCloudflareAIChat: () => wrapCloudflareAIChat,
   wrapCloudflareAgent: () => wrapCloudflareAgent,
+  wrapCloudflareThink: () => wrapCloudflareThink,
   wrapCohere: () => wrapCohere,
   wrapCopilotClient: () => wrapCopilotClient,
   wrapCursorSDK: () => wrapCursorSDK,
@@ -337,6 +342,7 @@ __export(exports_exports, {
   wrapLangSmithTraceable: () => wrapLangSmithTraceable,
   wrapMastraAgent: () => wrapMastraAgent,
   wrapMistral: () => wrapMistral,
+  wrapOllama: () => wrapOllama,
   wrapOpenAI: () => wrapOpenAI,
   wrapOpenAICodexSDK: () => wrapOpenAICodexSDK,
   wrapOpenAIv4: () => wrapOpenAIv4,
@@ -3541,6 +3547,66 @@ function betaProxy(beta, anthropic) {
       if (prop === "messages") {
         return betaMessagesProxy(target.messages, anthropic);
       }
+      if (prop === "sessions") {
+        return target.sessions ? betaSessionsProxy(target.sessions) : target.sessions;
+      }
+      return Reflect.get(target, prop, receiver);
+    }
+  });
+}
+function betaSessionsProxy(sessions) {
+  return new Proxy(sessions, {
+    get(target, prop, receiver) {
+      if (prop === "events") {
+        return betaSessionEventsProxy(target.events);
+      }
+      if (prop === "threads") {
+        return target.threads ? betaSessionThreadsProxy(target.threads) : target.threads;
+      }
+      return Reflect.get(target, prop, receiver);
+    }
+  });
+}
+function betaSessionEventsProxy(events) {
+  return new Proxy(events, {
+    get(target, prop, receiver) {
+      if (prop === "stream") {
+        return new TypedApplyProxy(target.stream, {
+          apply(stream, thisArg, argArray) {
+            return anthropicChannels.betaSessionsEventsStream.tracePromise(
+              () => Reflect.apply(stream, thisArg, argArray),
+              { arguments: argArray }
+            );
+          }
+        });
+      }
+      return Reflect.get(target, prop, receiver);
+    }
+  });
+}
+function betaSessionThreadsProxy(threads) {
+  return new Proxy(threads, {
+    get(target, prop, receiver) {
+      if (prop === "events") {
+        return betaSessionThreadEventsProxy(target.events);
+      }
+      return Reflect.get(target, prop, receiver);
+    }
+  });
+}
+function betaSessionThreadEventsProxy(events) {
+  return new Proxy(events, {
+    get(target, prop, receiver) {
+      if (prop === "stream") {
+        return new TypedApplyProxy(target.stream, {
+          apply(stream, thisArg, argArray) {
+            return anthropicChannels.betaSessionsThreadsEventsStream.tracePromise(
+              () => Reflect.apply(stream, thisArg, argArray),
+              { arguments: argArray }
+            );
+          }
+        });
+      }
       return Reflect.get(target, prop, receiver);
     }
   });
@@ -3734,6 +3800,52 @@ function claudeAgentSDKProxy(sdk) {
   });
 }
 
+// src/wrappers/cloudflare-think.ts
+var WRAPPED_THINK = /* @__PURE__ */ Symbol.for("braintrust.cloudflare-think.wrapped");
+function wrapCloudflareThink(sdk) {
+  if (!sdk || typeof sdk !== "object") {
+    return sdk;
+  }
+  const thinkModule = sdk;
+  if (typeof thinkModule.Think !== "function") {
+    return sdk;
+  }
+  patchThinkClass(thinkModule.Think);
+  return sdk;
+}
+function patchThinkClass(Think) {
+  const prototype = Think.prototype;
+  if (!prototype || prototype[WRAPPED_THINK]) {
+    return;
+  }
+  const descriptor = Object.getOwnPropertyDescriptor(
+    prototype,
+    "_runInferenceLoop"
+  );
+  if (!descriptor || typeof descriptor.value !== "function") {
+    return;
+  }
+  const original = descriptor.value;
+  Object.defineProperty(prototype, "_runInferenceLoop", {
+    ...descriptor,
+    value: function wrappedCloudflareThinkRunInferenceLoop(input) {
+      const args = [input];
+      return cloudflareThinkChannels.runInferenceLoop.tracePromise(
+        () => Reflect.apply(original, this, args),
+        {
+          arguments: args,
+          self: this
+        }
+      );
+    }
+  });
+  Object.defineProperty(prototype, WRAPPED_THINK, {
+    configurable: false,
+    enumerable: false,
+    value: true
+  });
+}
+
 // src/wrappers/openai-codex.ts
 var WRAPPED_CLIENT = /* @__PURE__ */ Symbol.for("braintrust.openai-codex.wrapped-client");
 var WRAPPED_THREAD = /* @__PURE__ */ Symbol.for("braintrust.openai-codex.wrapped-thread");
@@ -4026,14 +4138,9 @@ function patchAgentSessionClass(AgentSession) {
     ...descriptor,
     value: function wrappedPiCodingAgentPrompt(text, options) {
       const args = [text, options];
-      return piCodingAgentChannels.prompt.tracePromise(
-        () => Reflect.apply(originalPrompt, this, args),
-        {
-          arguments: args,
-          self: this,
-          session: this
-        }
-      );
+      return piCodingAgentChannels.prompt.invoke(originalPrompt, this, args, {
+        session: this
+      });
     }
   });
   Object.defineProperty(prototype, WRAPPED_PROMPT, {
@@ -5473,6 +5580,83 @@ function wrapAgentsStream(stream) {
   });
 }
 
+// src/wrappers/ollama.ts
+function wrapOllama(ollama) {
+  if (isSupportedOllamaClient(ollama)) {
+    return ollamaProxy(ollama);
+  }
+  debugLogger.warn("Unsupported Ollama library. Not wrapping.");
+  return ollama;
+}
+var ollamaProxyCache = /* @__PURE__ */ new WeakMap();
+function isSupportedOllamaClient(value) {
+  return isObject(value) && ["chat", "generate", "embed"].some(
+    (name) => typeof value[name] === "function"
+  );
+}
+function ollamaProxy(ollama) {
+  const cached = ollamaProxyCache.get(ollama);
+  if (cached) {
+    return cached;
+  }
+  let chatSource;
+  let wrappedChat;
+  let generateSource;
+  let wrappedGenerate;
+  let embedSource;
+  let wrappedEmbed;
+  const proxy = new Proxy(ollama, {
+    get(target, prop, receiver) {
+      switch (prop) {
+        case "chat": {
+          const source = target.chat;
+          if (source !== chatSource) {
+            chatSource = source;
+            wrappedChat = typeof source === "function" ? wrapChat(source.bind(target)) : source;
+          }
+          return wrappedChat;
+        }
+        case "generate": {
+          const source = target.generate;
+          if (source !== generateSource) {
+            generateSource = source;
+            wrappedGenerate = typeof source === "function" ? wrapGenerate2(source.bind(target)) : source;
+          }
+          return wrappedGenerate;
+        }
+        case "embed": {
+          const source = target.embed;
+          if (source !== embedSource) {
+            embedSource = source;
+            wrappedEmbed = typeof source === "function" ? wrapEmbed3(source.bind(target)) : source;
+          }
+          return wrappedEmbed;
+        }
+        default:
+          return Reflect.get(target, prop, receiver);
+      }
+    }
+  });
+  ollamaProxyCache.set(ollama, proxy);
+  ollamaProxyCache.set(proxy, proxy);
+  return proxy;
+}
+function wrapChat(chat) {
+  return (request) => ollamaChannels.chat.tracePromise(() => chat(request), {
+    arguments: [request]
+  });
+}
+function wrapGenerate2(generate) {
+  return (request) => ollamaChannels.generate.tracePromise(() => generate(request), {
+    arguments: [request]
+  });
+}
+function wrapEmbed3(embed) {
+  return (request) => ollamaChannels.embed.tracePromise(() => embed(request), {
+    arguments: [request]
+  });
+}
+
 // src/wrappers/cohere.ts
 function wrapCohere(cohere) {
   if (isSupportedCohereClient(cohere)) {
@@ -5503,11 +5687,11 @@ function cohereProxy(cohere) {
     get(target, prop, receiver) {
       switch (prop) {
         case "chat":
-          return typeof target.chat === "function" ? wrapChat(target.chat.bind(target)) : target.chat;
+          return typeof target.chat === "function" ? wrapChat2(target.chat.bind(target)) : target.chat;
         case "chatStream":
           return typeof target.chatStream === "function" ? wrapChatStream2(target.chatStream.bind(target)) : target.chatStream;
         case "embed":
-          return typeof target.embed === "function" ? wrapEmbed3(target.embed.bind(target)) : target.embed;
+          return typeof target.embed === "function" ? wrapEmbed4(target.embed.bind(target)) : target.embed;
         case "rerank":
           return typeof target.rerank === "function" ? wrapRerank3(target.rerank.bind(target)) : target.rerank;
         default: {
@@ -5520,7 +5704,7 @@ function cohereProxy(cohere) {
   cohereProxyCache.set(cohere, proxy);
   return proxy;
 }
-function wrapChat(chat) {
+function wrapChat2(chat) {
   return (request, options) => cohereChannels.chat.tracePromise(() => chat(request, options), {
     arguments: [request]
   });
@@ -5530,7 +5714,7 @@ function wrapChatStream2(chatStream) {
     arguments: [request]
   });
 }
-function wrapEmbed3(embed) {
+function wrapEmbed4(embed) {
   return (request, options) => cohereChannels.embed.tracePromise(() => embed(request, options), {
     arguments: [request]
   });
@@ -10382,6 +10566,7 @@ export {
   braintrustFlueObserver,
   braintrustStreamChunkSchema,
   buildLocalSummary,
+  collectAnthropicSession,
   configureInstrumentation,
   constructLogs3OverflowRequest,
   createFinalValuePassThroughStream,
@@ -10462,6 +10647,7 @@ export {
   wrapClaudeAgentSDK,
   wrapCloudflareAIChat,
   wrapCloudflareAgent,
+  wrapCloudflareThink,
   wrapCohere,
   wrapCopilotClient,
   wrapCursorSDK,
@@ -10476,6 +10662,7 @@ export {
   wrapLangSmithTraceable,
   wrapMastraAgent,
   wrapMistral,
+  wrapOllama,
   wrapOpenAI,
   wrapOpenAICodexSDK,
   wrapOpenAIv4,

package/dist/instrumentation/index.d.mts

@@ -6857,6 +6857,7 @@ interface InstrumentationIntegrationsConfig {
     huggingface?: boolean;
     claudeAgentSDK?: boolean;
     cloudflareAIChat?: boolean;
+    cloudflareThink?: boolean;
     cursor?: boolean;
     cursorSDK?: boolean;
     flue?: boolean;
@@ -6865,6 +6866,7 @@ interface InstrumentationIntegrationsConfig {
     openrouter?: boolean;
     openrouterAgent?: boolean;
     mistral?: boolean;
+    ollama?: boolean;
     cohere?: boolean;
     groq?: boolean;
     bedrock?: boolean;
@@ -6903,6 +6905,7 @@ interface BraintrustPluginConfig {
  * - HuggingFace Inference SDK
  * - LangChain.js and LangGraph
  * - Mistral SDK
+ * - Ollama SDK
  * - Cohere SDK
  *
  * The plugin is automatically enabled when the Braintrust library is loaded.
@@ -6915,6 +6918,7 @@ declare class BraintrustPlugin extends BasePlugin {
     private anthropicPlugin;
     private aiSDKPlugin;
     private claudeAgentSDKPlugin;
+    private cloudflareThinkPlugin;
     private cursorSDKPlugin;
     private openAIAgentsPlugin;
     private googleGenAIPlugin;
@@ -6923,6 +6927,7 @@ declare class BraintrustPlugin extends BasePlugin {
     private openRouterPlugin;
     private openRouterAgentPlugin;
     private mistralPlugin;
+    private ollamaPlugin;
     private googleADKPlugin;
     private coherePlugin;
     private groqPlugin;
@@ -9042,6 +9047,9 @@ type FlueExecutionOperation = {
 } | {
     type: "task";
     taskId: string;
+} | {
+    type: "coordinator";
+    phase: "reconcile";
 };
 interface FlueTraceCarrier {
     traceparent: string;

package/dist/instrumentation/index.d.ts

@@ -6857,6 +6857,7 @@ interface InstrumentationIntegrationsConfig {
     huggingface?: boolean;
     claudeAgentSDK?: boolean;
     cloudflareAIChat?: boolean;
+    cloudflareThink?: boolean;
     cursor?: boolean;
     cursorSDK?: boolean;
     flue?: boolean;
@@ -6865,6 +6866,7 @@ interface InstrumentationIntegrationsConfig {
     openrouter?: boolean;
     openrouterAgent?: boolean;
     mistral?: boolean;
+    ollama?: boolean;
     cohere?: boolean;
     groq?: boolean;
     bedrock?: boolean;
@@ -6903,6 +6905,7 @@ interface BraintrustPluginConfig {
  * - HuggingFace Inference SDK
  * - LangChain.js and LangGraph
  * - Mistral SDK
+ * - Ollama SDK
  * - Cohere SDK
  *
  * The plugin is automatically enabled when the Braintrust library is loaded.
@@ -6915,6 +6918,7 @@ declare class BraintrustPlugin extends BasePlugin {
     private anthropicPlugin;
     private aiSDKPlugin;
     private claudeAgentSDKPlugin;
+    private cloudflareThinkPlugin;
     private cursorSDKPlugin;
     private openAIAgentsPlugin;
     private googleGenAIPlugin;
@@ -6923,6 +6927,7 @@ declare class BraintrustPlugin extends BasePlugin {
     private openRouterPlugin;
     private openRouterAgentPlugin;
     private mistralPlugin;
+    private ollamaPlugin;
     private googleADKPlugin;
     private coherePlugin;
     private groqPlugin;
@@ -9042,6 +9047,9 @@ type FlueExecutionOperation = {
 } | {
     type: "task";
     taskId: string;
+} | {
+    type: "coordinator";
+    phase: "reconcile";
 };
 interface FlueTraceCarrier {
     traceparent: string;