braintrust 0.0.145 → 0.0.147

package/dist/browser.d.mts

@@ -1,4 +1,4 @@
-import { GitMetadataSettings, LogFeedbackFullArgs, BackgroundLogEvent, ExperimentEvent, ExperimentLogFullArgs, ExperimentLogPartialArgs, IdField, SpanType, RepoInfo, DEFAULT_IS_LEGACY_DATASET, TRANSACTION_ID_FIELD, TransactionId, SpanObjectTypeV2, DatasetRecord } from '@braintrust/core';
+import { GitMetadataSettings, LogFeedbackFullArgs, ExperimentEvent, BackgroundLogEvent, ExperimentLogFullArgs, ExperimentLogPartialArgs, IdField, SpanType, RepoInfo, DEFAULT_IS_LEGACY_DATASET, TRANSACTION_ID_FIELD, TransactionId, SpanObjectTypeV2, DatasetRecord } from '@braintrust/core';
 import { PromptData, OpenAIMessage, Tools, AnyModelParam, Prompt as Prompt$1, PromptSessionEvent } from '@braintrust/core/typespecs';
 import { z } from 'zod';
 
@@ -201,7 +201,7 @@ declare class BraintrustState {
     resetLoginInfo(): void;
     copyLoginInfo(other: BraintrustState): void;
     serialize(): SerializedBraintrustState;
-    static deserialize(serialized: unknown, opts?: BackgroundLoggerOpts): BraintrustState;
+    static deserialize(serialized: unknown, opts?: LoginOptions): BraintrustState;
     setFetch(fetch: typeof globalThis.fetch): void;
     login(loginParams: LoginOptions & {
         forceLogin?: boolean;
@@ -253,6 +253,18 @@ interface LogOptions<IsAsyncFlush> {
     computeMetadataArgs?: Record<string, any>;
 }
 type PromiseUnless<B, R> = B extends true ? R : Promise<Awaited<R>>;
+/**
+ * Update a span using the output of `span.export()`. It is important that you only resume updating
+ * to a span once the original span has been fully written and flushed, since otherwise updates to
+ * the span may conflict with the original span.
+ *
+ * @param exported The output of `span.export()`.
+ * @param event The event data to update the span with. See `Experiment.log` for a full list of valid fields.
+ * @param state (optional) Login state to use. If not provided, the global state will be used.
+ */
+declare function updateSpan({ exported, state, ...event }: {
+    exported: string;
+} & Omit<Partial<ExperimentEvent>, "id"> & OptionalStateArg): void;
 interface ParentSpanIds {
     spanId: string;
     rootSpanId: string;
@@ -316,6 +328,13 @@ declare class Logger<IsAsyncFlush extends boolean> implements Exportable {
      * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
      */
     logFeedback(event: LogFeedbackFullArgs): void;
+    /**
+     * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+     * since otherwise updates to the span may conflict with the original span.
+     *
+     * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+     */
+    updateSpan(event: Omit<Partial<ExperimentEvent>, "id"> & Required<Pick<ExperimentEvent, "id">>): void;
     /**
      * Return a serialized representation of the logger that can be used to start subspans in other places. See `Span.start_span` for more details.
      */
@@ -780,6 +799,13 @@ declare class Experiment extends ObjectFetcher<ExperimentEvent> implements Expor
      * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
      */
     logFeedback(event: LogFeedbackFullArgs): void;
+    /**
+     * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+     * since otherwise updates to the span may conflict with the original span.
+     *
+     * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+     */
+    updateSpan(event: Omit<Partial<ExperimentEvent>, "id"> & Required<Pick<ExperimentEvent, "id">>): void;
     /**
      * Return a serialized representation of the experiment that can be used to start subspans in other places. See `Span.start_span` for more details.
      */
@@ -1079,6 +1105,13 @@ declare class BraintrustStream {
      * @returns The underlying ReadableStream<BraintrustStreamChunk>.
      */
     toReadableStream(): ReadableStream<BraintrustStreamChunk>;
+    /**
+     * Returns an async iterator for the BraintrustStream.
+     * This allows for easy consumption of the stream using a for-await...of loop.
+     *
+     * @returns An async iterator that yields BraintrustStreamChunk objects.
+     */
+    [Symbol.asyncIterator](): AsyncIterator<BraintrustStreamChunk>;
     /**
      * Get the final value of the stream. The final value is the concatenation of all
      * the chunks in the stream, deserialized into a string or JSON object, depending on
@@ -1116,7 +1149,7 @@ interface InvokeFunctionArgs<Input, Output, Stream extends boolean = false> {
     /**
      * The name of the project containing the function to invoke.
      */
-    project_name?: string;
+    projectName?: string;
     /**
      * The slug of the function to invoke.
      */
@@ -1124,15 +1157,15 @@ interface InvokeFunctionArgs<Input, Output, Stream extends boolean = false> {
     /**
      * The name of the global function to invoke.
      */
-    global_function?: string;
+    globalFunction?: string;
     /**
      * The ID of the prompt session to invoke the function from.
      */
-    prompt_session_id?: string;
+    promptSessionId?: string;
     /**
      * The ID of the function in the prompt session to invoke.
      */
-    prompt_session_function_id?: string;
+    promptSessionFunctionId?: string;
     /**
      * The version of the function to invoke.
      */
@@ -1215,4 +1248,4 @@ declare const LEGACY_CACHED_HEADER = "x-cached";
 declare const X_CACHED_HEADER = "x-bt-cached";
 declare function parseCachedHeader(value: string | null | undefined): number | undefined;
 
-export { type AnyDataset, type BackgroundLoggerOpts, type BaseMetadata, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, type EndSpanArgs, type EvalCase, Experiment, type ExperimentSummary, type Exportable, type FullInitOptions, type FullLoginOptions, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NoopSpan, type ObjectMetadata, type PromiseUnless, Prompt, ReadonlyExperiment, type ScoreSummary, type SerializedBraintrustState, type SetCurrentArg, type Span, SpanImpl, type StartSpanArgs, type WithTransactionId, X_CACHED_HEADER, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, devNullWritableStream, flush, getSpanParentObject, init, initDataset, initExperiment, initLogger, invoke, loadPrompt, log, login, loginToState, newId, parseCachedHeader, setFetch, startSpan, summarize, traceable, traced, withDataset, withExperiment, withLogger, wrapOpenAI, wrapOpenAIv4, wrapTraced };
+export { type AnyDataset, type BackgroundLoggerOpts, type BaseMetadata, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, type EndSpanArgs, type EvalCase, Experiment, type ExperimentSummary, type Exportable, type FullInitOptions, type FullLoginOptions, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NoopSpan, type ObjectMetadata, type PromiseUnless, Prompt, ReadonlyExperiment, type ScoreSummary, type SerializedBraintrustState, type SetCurrentArg, type Span, SpanImpl, type StartSpanArgs, type WithTransactionId, X_CACHED_HEADER, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, devNullWritableStream, flush, getSpanParentObject, init, initDataset, initExperiment, initLogger, invoke, loadPrompt, log, login, loginToState, newId, parseCachedHeader, setFetch, startSpan, summarize, traceable, traced, updateSpan, withDataset, withExperiment, withLogger, wrapOpenAI, wrapOpenAIv4, wrapTraced };

package/dist/browser.d.ts

@@ -1,4 +1,4 @@
-import { GitMetadataSettings, LogFeedbackFullArgs, BackgroundLogEvent, ExperimentEvent, ExperimentLogFullArgs, ExperimentLogPartialArgs, IdField, SpanType, RepoInfo, DEFAULT_IS_LEGACY_DATASET, TRANSACTION_ID_FIELD, TransactionId, SpanObjectTypeV2, DatasetRecord } from '@braintrust/core';
+import { GitMetadataSettings, LogFeedbackFullArgs, ExperimentEvent, BackgroundLogEvent, ExperimentLogFullArgs, ExperimentLogPartialArgs, IdField, SpanType, RepoInfo, DEFAULT_IS_LEGACY_DATASET, TRANSACTION_ID_FIELD, TransactionId, SpanObjectTypeV2, DatasetRecord } from '@braintrust/core';
 import { PromptData, OpenAIMessage, Tools, AnyModelParam, Prompt as Prompt$1, PromptSessionEvent } from '@braintrust/core/typespecs';
 import { z } from 'zod';
 
@@ -201,7 +201,7 @@ declare class BraintrustState {
     resetLoginInfo(): void;
     copyLoginInfo(other: BraintrustState): void;
     serialize(): SerializedBraintrustState;
-    static deserialize(serialized: unknown, opts?: BackgroundLoggerOpts): BraintrustState;
+    static deserialize(serialized: unknown, opts?: LoginOptions): BraintrustState;
     setFetch(fetch: typeof globalThis.fetch): void;
     login(loginParams: LoginOptions & {
         forceLogin?: boolean;
@@ -253,6 +253,18 @@ interface LogOptions<IsAsyncFlush> {
     computeMetadataArgs?: Record<string, any>;
 }
 type PromiseUnless<B, R> = B extends true ? R : Promise<Awaited<R>>;
+/**
+ * Update a span using the output of `span.export()`. It is important that you only resume updating
+ * to a span once the original span has been fully written and flushed, since otherwise updates to
+ * the span may conflict with the original span.
+ *
+ * @param exported The output of `span.export()`.
+ * @param event The event data to update the span with. See `Experiment.log` for a full list of valid fields.
+ * @param state (optional) Login state to use. If not provided, the global state will be used.
+ */
+declare function updateSpan({ exported, state, ...event }: {
+    exported: string;
+} & Omit<Partial<ExperimentEvent>, "id"> & OptionalStateArg): void;
 interface ParentSpanIds {
     spanId: string;
     rootSpanId: string;
@@ -316,6 +328,13 @@ declare class Logger<IsAsyncFlush extends boolean> implements Exportable {
      * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
      */
     logFeedback(event: LogFeedbackFullArgs): void;
+    /**
+     * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+     * since otherwise updates to the span may conflict with the original span.
+     *
+     * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+     */
+    updateSpan(event: Omit<Partial<ExperimentEvent>, "id"> & Required<Pick<ExperimentEvent, "id">>): void;
     /**
      * Return a serialized representation of the logger that can be used to start subspans in other places. See `Span.start_span` for more details.
      */
@@ -780,6 +799,13 @@ declare class Experiment extends ObjectFetcher<ExperimentEvent> implements Expor
      * @param event.source (Optional) the source of the feedback. Must be one of "external" (default), "app", or "api".
      */
     logFeedback(event: LogFeedbackFullArgs): void;
+    /**
+     * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+     * since otherwise updates to the span may conflict with the original span.
+     *
+     * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+     */
+    updateSpan(event: Omit<Partial<ExperimentEvent>, "id"> & Required<Pick<ExperimentEvent, "id">>): void;
     /**
      * Return a serialized representation of the experiment that can be used to start subspans in other places. See `Span.start_span` for more details.
      */
@@ -1079,6 +1105,13 @@ declare class BraintrustStream {
      * @returns The underlying ReadableStream<BraintrustStreamChunk>.
      */
     toReadableStream(): ReadableStream<BraintrustStreamChunk>;
+    /**
+     * Returns an async iterator for the BraintrustStream.
+     * This allows for easy consumption of the stream using a for-await...of loop.
+     *
+     * @returns An async iterator that yields BraintrustStreamChunk objects.
+     */
+    [Symbol.asyncIterator](): AsyncIterator<BraintrustStreamChunk>;
     /**
      * Get the final value of the stream. The final value is the concatenation of all
      * the chunks in the stream, deserialized into a string or JSON object, depending on
@@ -1116,7 +1149,7 @@ interface InvokeFunctionArgs<Input, Output, Stream extends boolean = false> {
     /**
      * The name of the project containing the function to invoke.
      */
-    project_name?: string;
+    projectName?: string;
     /**
      * The slug of the function to invoke.
      */
@@ -1124,15 +1157,15 @@ interface InvokeFunctionArgs<Input, Output, Stream extends boolean = false> {
     /**
      * The name of the global function to invoke.
      */
-    global_function?: string;
+    globalFunction?: string;
     /**
      * The ID of the prompt session to invoke the function from.
      */
-    prompt_session_id?: string;
+    promptSessionId?: string;
     /**
      * The ID of the function in the prompt session to invoke.
      */
-    prompt_session_function_id?: string;
+    promptSessionFunctionId?: string;
     /**
      * The version of the function to invoke.
      */
@@ -1215,4 +1248,4 @@ declare const LEGACY_CACHED_HEADER = "x-cached";
 declare const X_CACHED_HEADER = "x-bt-cached";
 declare function parseCachedHeader(value: string | null | undefined): number | undefined;
 
-export { type AnyDataset, type BackgroundLoggerOpts, type BaseMetadata, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, type EndSpanArgs, type EvalCase, Experiment, type ExperimentSummary, type Exportable, type FullInitOptions, type FullLoginOptions, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NoopSpan, type ObjectMetadata, type PromiseUnless, Prompt, ReadonlyExperiment, type ScoreSummary, type SerializedBraintrustState, type SetCurrentArg, type Span, SpanImpl, type StartSpanArgs, type WithTransactionId, X_CACHED_HEADER, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, devNullWritableStream, flush, getSpanParentObject, init, initDataset, initExperiment, initLogger, invoke, loadPrompt, log, login, loginToState, newId, parseCachedHeader, setFetch, startSpan, summarize, traceable, traced, withDataset, withExperiment, withLogger, wrapOpenAI, wrapOpenAIv4, wrapTraced };
+export { type AnyDataset, type BackgroundLoggerOpts, type BaseMetadata, BraintrustState, BraintrustStream, type BraintrustStreamChunk, type ChatPrompt, type CompiledPrompt, type CompiledPromptParams, type CompletionPrompt, type DataSummary, Dataset, type DatasetSummary, type DefaultMetadataType, type DefaultPromptArgs, type EndSpanArgs, type EvalCase, Experiment, type ExperimentSummary, type Exportable, type FullInitOptions, type FullLoginOptions, type InitOptions, type InvokeFunctionArgs, type InvokeReturn, LEGACY_CACHED_HEADER, type LogOptions, Logger, type LoginOptions, type MetricSummary, NOOP_SPAN, NoopSpan, type ObjectMetadata, type PromiseUnless, Prompt, ReadonlyExperiment, type ScoreSummary, type SerializedBraintrustState, type SetCurrentArg, type Span, SpanImpl, type StartSpanArgs, type WithTransactionId, X_CACHED_HEADER, _internalGetGlobalState, _internalSetInitialState, braintrustStreamChunkSchema, createFinalValuePassThroughStream, currentExperiment, currentLogger, currentSpan, devNullWritableStream, flush, getSpanParentObject, init, initDataset, initExperiment, initLogger, invoke, loadPrompt, log, login, loginToState, newId, parseCachedHeader, setFetch, startSpan, summarize, traceable, traced, updateSpan, withDataset, withExperiment, withLogger, wrapOpenAI, wrapOpenAIv4, wrapTraced };

package/dist/browser.js

@@ -68,6 +68,7 @@ __export(browser_exports, {
   summarize: () => summarize,
   traceable: () => traceable,
   traced: () => traced,
+  updateSpan: () => updateSpan,
   withDataset: () => withDataset,
   withExperiment: () => withExperiment,
   withLogger: () => withLogger,
@@ -194,6 +195,33 @@ var BraintrustStream = class _BraintrustStream {
   toReadableStream() {
     return this.stream;
   }
+  /**
+   * Returns an async iterator for the BraintrustStream.
+   * This allows for easy consumption of the stream using a for-await...of loop.
+   *
+   * @returns An async iterator that yields BraintrustStreamChunk objects.
+   */
+  [Symbol.asyncIterator]() {
+    const reader = this.stream.getReader();
+    return {
+      async next() {
+        const { done, value } = await reader.read();
+        if (done) {
+          reader.releaseLock();
+          return { done: true, value: void 0 };
+        }
+        return { done: false, value };
+      },
+      async return() {
+        reader.releaseLock();
+        return { done: true, value: void 0 };
+      },
+      async throw(error) {
+        reader.releaseLock();
+        throw error;
+      }
+    };
+  }
   /**
    * Get the final value of the stream. The final value is the concatenation of all
    * the chunks in the stream, deserialized into a string or JSON object, depending on
@@ -482,8 +510,10 @@ var BraintrustState = class _BraintrustState {
     state.apiConn().set_token(state.loginToken);
     state.apiConn().make_long_lived();
     state.appConn().set_token(state.loginToken);
-    state.proxyConn().make_long_lived();
-    state.proxyConn().set_token(state.loginToken);
+    if (state.proxyUrl) {
+      state.proxyConn().make_long_lived();
+      state.proxyConn().set_token(state.loginToken);
+    }
     state.loggedIn = true;
     state.loginReplaceApiConn(state.apiConn());
     return state;
@@ -523,6 +553,9 @@ var BraintrustState = class _BraintrustState {
     return this._apiConn;
   }
   proxyConn() {
+    if (!this.proxyUrl) {
+      return this.apiConn();
+    }
     if (!this._proxyConn) {
       if (!this.proxyUrl) {
         throw new Error("Must initialize proxyUrl before requesting proxyConn");
@@ -740,6 +773,41 @@ function logFeedbackImpl(state, parentObjectType, parentObjectId, {
     state.bgLogger().log([record]);
   }
 }
+function updateSpanImpl(state, parentObjectType, parentObjectId, id, event) {
+  const updateEvent = validateAndSanitizeExperimentLogPartialArgs({
+    id,
+    ...event
+  });
+  const parentIds = async () => new import_core.SpanComponentsV2({
+    objectType: parentObjectType,
+    objectId: await parentObjectId.get()
+  }).objectIdFields();
+  const record = new LazyValue(async () => ({
+    id,
+    ...updateEvent,
+    ...await parentIds(),
+    [import_core.IS_MERGE_FIELD]: true
+  }));
+  state.bgLogger().log([record]);
+}
+function updateSpan({
+  exported,
+  state,
+  ...event
+}) {
+  const resolvedState = state ?? _globalState;
+  const components = import_core.SpanComponentsV2.fromStr(exported);
+  if (!components.rowIds?.rowId) {
+    throw new Error("Exported span must have a row id");
+  }
+  updateSpanImpl(
+    resolvedState,
+    components.objectType,
+    new LazyValue(spanComponentsToObjectIdLambda(resolvedState, components)),
+    components.rowIds?.rowId,
+    event
+  );
+}
 function spanComponentsToObjectIdLambda(state, components) {
   if (components.objectId) {
     const ret = components.objectId;
@@ -944,6 +1012,25 @@ var Logger = class {
   logFeedback(event) {
     logFeedbackImpl(this.state, this.parentObjectType(), this.lazyId, event);
   }
+  /**
+   * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+   * since otherwise updates to the span may conflict with the original span.
+   *
+   * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+   */
+  updateSpan(event) {
+    const { id, ...eventRest } = event;
+    if (!id) {
+      throw new Error("Span id is required to update a span");
+    }
+    updateSpanImpl(
+      this.state,
+      this.parentObjectType(),
+      this.lazyId,
+      id,
+      eventRest
+    );
+  }
   /**
    * Return a serialized representation of the logger that can be used to start subspans in other places. See `Span.start_span` for more details.
    */
@@ -1663,7 +1750,7 @@ async function loadPrompt({
     throw new Error("Must specify slug");
   }
   const state = stateArg ?? _globalState;
-  state.login({
+  await state.login({
     orgName,
     apiKey,
     appUrl,
@@ -1750,7 +1837,9 @@ async function loginToState(options = {}) {
   }
   conn.make_long_lived();
   state.appConn().set_token(apiKey);
-  state.proxyConn().set_token(apiKey);
+  if (state.proxyUrl) {
+    state.proxyConn().set_token(apiKey);
+  }
   state.loginToken = conn.token;
   state.loggedIn = true;
   state.loginReplaceApiConn(conn);
@@ -2281,6 +2370,25 @@ var Experiment = class extends ObjectFetcher {
   logFeedback(event) {
     logFeedbackImpl(this.state, this.parentObjectType(), this.lazyId, event);
   }
+  /**
+   * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+   * since otherwise updates to the span may conflict with the original span.
+   *
+   * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+   */
+  updateSpan(event) {
+    const { id, ...eventRest } = event;
+    if (!id) {
+      throw new Error("Span id is required to update a span");
+    }
+    updateSpanImpl(
+      this.state,
+      this.parentObjectType(),
+      this.lazyId,
+      id,
+      eventRest
+    );
+  }
   /**
    * Return a serialized representation of the experiment that can be used to start subspans in other places. See `Span.start_span` for more details.
    */
@@ -2896,7 +3004,14 @@ async function invoke(args) {
     forceLogin
   });
   const parent = parentArg ? typeof parentArg === "string" ? parentArg : await parentArg.export() : await getSpanParentObject().export();
-  const functionId = import_typespecs3.functionIdSchema.safeParse(functionIdArgs);
+  const functionId = import_typespecs3.functionIdSchema.safeParse({
+    project_name: functionIdArgs.projectName,
+    slug: functionIdArgs.slug,
+    global_function: functionIdArgs.globalFunction,
+    prompt_session_id: functionIdArgs.promptSessionId,
+    prompt_session_function_id: functionIdArgs.promptSessionFunctionId,
+    version: functionIdArgs.version
+  });
   if (!functionId.success) {
     throw new Error(
       `Invalid function ID arguments: ${functionId.error.message}`
@@ -2906,8 +3021,7 @@ async function invoke(args) {
     ...functionId.data,
     input,
     parent,
-    stream,
-    api_version: import_typespecs3.INVOKE_API_VERSION
+    stream
   };
   const resp = await state.proxyConn().post(`function/invoke`, request, {
     headers: {
@@ -3319,6 +3433,7 @@ configureBrowser();
   summarize,
   traceable,
   traced,
+  updateSpan,
   withDataset,
   withExperiment,
   withLogger,

package/dist/browser.mjs

@@ -141,6 +141,33 @@ var BraintrustStream = class _BraintrustStream {
   toReadableStream() {
     return this.stream;
   }
+  /**
+   * Returns an async iterator for the BraintrustStream.
+   * This allows for easy consumption of the stream using a for-await...of loop.
+   *
+   * @returns An async iterator that yields BraintrustStreamChunk objects.
+   */
+  [Symbol.asyncIterator]() {
+    const reader = this.stream.getReader();
+    return {
+      async next() {
+        const { done, value } = await reader.read();
+        if (done) {
+          reader.releaseLock();
+          return { done: true, value: void 0 };
+        }
+        return { done: false, value };
+      },
+      async return() {
+        reader.releaseLock();
+        return { done: true, value: void 0 };
+      },
+      async throw(error) {
+        reader.releaseLock();
+        throw error;
+      }
+    };
+  }
   /**
    * Get the final value of the stream. The final value is the concatenation of all
    * the chunks in the stream, deserialized into a string or JSON object, depending on
@@ -429,8 +456,10 @@ var BraintrustState = class _BraintrustState {
     state.apiConn().set_token(state.loginToken);
     state.apiConn().make_long_lived();
     state.appConn().set_token(state.loginToken);
-    state.proxyConn().make_long_lived();
-    state.proxyConn().set_token(state.loginToken);
+    if (state.proxyUrl) {
+      state.proxyConn().make_long_lived();
+      state.proxyConn().set_token(state.loginToken);
+    }
     state.loggedIn = true;
     state.loginReplaceApiConn(state.apiConn());
     return state;
@@ -470,6 +499,9 @@ var BraintrustState = class _BraintrustState {
     return this._apiConn;
   }
   proxyConn() {
+    if (!this.proxyUrl) {
+      return this.apiConn();
+    }
     if (!this._proxyConn) {
       if (!this.proxyUrl) {
         throw new Error("Must initialize proxyUrl before requesting proxyConn");
@@ -687,6 +719,41 @@ function logFeedbackImpl(state, parentObjectType, parentObjectId, {
     state.bgLogger().log([record]);
   }
 }
+function updateSpanImpl(state, parentObjectType, parentObjectId, id, event) {
+  const updateEvent = validateAndSanitizeExperimentLogPartialArgs({
+    id,
+    ...event
+  });
+  const parentIds = async () => new SpanComponentsV2({
+    objectType: parentObjectType,
+    objectId: await parentObjectId.get()
+  }).objectIdFields();
+  const record = new LazyValue(async () => ({
+    id,
+    ...updateEvent,
+    ...await parentIds(),
+    [IS_MERGE_FIELD]: true
+  }));
+  state.bgLogger().log([record]);
+}
+function updateSpan({
+  exported,
+  state,
+  ...event
+}) {
+  const resolvedState = state ?? _globalState;
+  const components = SpanComponentsV2.fromStr(exported);
+  if (!components.rowIds?.rowId) {
+    throw new Error("Exported span must have a row id");
+  }
+  updateSpanImpl(
+    resolvedState,
+    components.objectType,
+    new LazyValue(spanComponentsToObjectIdLambda(resolvedState, components)),
+    components.rowIds?.rowId,
+    event
+  );
+}
 function spanComponentsToObjectIdLambda(state, components) {
   if (components.objectId) {
     const ret = components.objectId;
@@ -891,6 +958,25 @@ var Logger = class {
   logFeedback(event) {
     logFeedbackImpl(this.state, this.parentObjectType(), this.lazyId, event);
   }
+  /**
+   * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+   * since otherwise updates to the span may conflict with the original span.
+   *
+   * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+   */
+  updateSpan(event) {
+    const { id, ...eventRest } = event;
+    if (!id) {
+      throw new Error("Span id is required to update a span");
+    }
+    updateSpanImpl(
+      this.state,
+      this.parentObjectType(),
+      this.lazyId,
+      id,
+      eventRest
+    );
+  }
   /**
    * Return a serialized representation of the logger that can be used to start subspans in other places. See `Span.start_span` for more details.
    */
@@ -1610,7 +1696,7 @@ async function loadPrompt({
     throw new Error("Must specify slug");
   }
   const state = stateArg ?? _globalState;
-  state.login({
+  await state.login({
     orgName,
     apiKey,
     appUrl,
@@ -1697,7 +1783,9 @@ async function loginToState(options = {}) {
   }
   conn.make_long_lived();
   state.appConn().set_token(apiKey);
-  state.proxyConn().set_token(apiKey);
+  if (state.proxyUrl) {
+    state.proxyConn().set_token(apiKey);
+  }
   state.loginToken = conn.token;
   state.loggedIn = true;
   state.loginReplaceApiConn(conn);
@@ -2228,6 +2316,25 @@ var Experiment = class extends ObjectFetcher {
   logFeedback(event) {
     logFeedbackImpl(this.state, this.parentObjectType(), this.lazyId, event);
   }
+  /**
+   * Update a span in the experiment using its id. It is important that you only update a span once the original span has been fully written and flushed,
+   * since otherwise updates to the span may conflict with the original span.
+   *
+   * @param event The event data to update the span with. Must include `id`. See `Experiment.log` for a full list of valid fields.
+   */
+  updateSpan(event) {
+    const { id, ...eventRest } = event;
+    if (!id) {
+      throw new Error("Span id is required to update a span");
+    }
+    updateSpanImpl(
+      this.state,
+      this.parentObjectType(),
+      this.lazyId,
+      id,
+      eventRest
+    );
+  }
   /**
    * Return a serialized representation of the experiment that can be used to start subspans in other places. See `Span.start_span` for more details.
    */
@@ -2821,7 +2928,6 @@ function configureBrowser() {
 
 // src/functions/invoke.ts
 import {
-  INVOKE_API_VERSION,
   functionIdSchema
 } from "@braintrust/core/typespecs";
 async function invoke(args) {
@@ -2846,7 +2952,14 @@ async function invoke(args) {
     forceLogin
   });
   const parent = parentArg ? typeof parentArg === "string" ? parentArg : await parentArg.export() : await getSpanParentObject().export();
-  const functionId = functionIdSchema.safeParse(functionIdArgs);
+  const functionId = functionIdSchema.safeParse({
+    project_name: functionIdArgs.projectName,
+    slug: functionIdArgs.slug,
+    global_function: functionIdArgs.globalFunction,
+    prompt_session_id: functionIdArgs.promptSessionId,
+    prompt_session_function_id: functionIdArgs.promptSessionFunctionId,
+    version: functionIdArgs.version
+  });
   if (!functionId.success) {
     throw new Error(
       `Invalid function ID arguments: ${functionId.error.message}`
@@ -2856,8 +2969,7 @@ async function invoke(args) {
     ...functionId.data,
     input,
     parent,
-    stream,
-    api_version: INVOKE_API_VERSION
+    stream
   };
   const resp = await state.proxyConn().post(`function/invoke`, request, {
     headers: {
@@ -3268,6 +3380,7 @@ export {
   summarize,
   traceable,
   traced,
+  updateSpan,
   withDataset,
   withExperiment,
   withLogger,