braintrust 0.0.157 → 0.0.158

package/dist/browser.d.mts

@@ -1,5 +1,5 @@
-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 { GitMetadataSettings, LogFeedbackFullArgs, ExperimentEvent, BackgroundLogEvent, ExperimentLogFullArgs, ExperimentLogPartialArgs, IdField, SpanType, RepoInfo, DEFAULT_IS_LEGACY_DATASET, TRANSACTION_ID_FIELD, TransactionId, SpanObjectTypeV3, DatasetRecord } from '@braintrust/core';
+import { PromptData, OpenAIMessage, Tools, AnyModelParam, Message, Prompt as Prompt$1, PromptSessionEvent } from '@braintrust/core/typespecs';
 import { z } from 'zod';
 
 interface IsoAsyncLocalStorage<T> {
@@ -16,8 +16,6 @@ declare class LazyValue<T> {
     get hasComputed(): boolean;
 }
 
-/// <reference lib="dom" />
-
 type SetCurrentArg = {
     setCurrent?: boolean;
 };
@@ -29,6 +27,7 @@ type StartSpanArgs = {
     startTime?: number;
     parent?: string;
     event?: StartSpanEventArgs;
+    propagatedEvent?: StartSpanEventArgs;
 };
 type EndSpanArgs = {
     endTime?: number;
@@ -685,6 +684,10 @@ declare function flush(options?: OptionalStateArg): Promise<void>;
  * @param fetch The fetch implementation to use.
  */
 declare function setFetch(fetch: typeof globalThis.fetch): void;
+/**
+ * Runs the provided callback with the span as the current span.
+ */
+declare function withCurrent<R>(span: Span, callback: (span: Span) => R, state?: BraintrustState): R;
 type WithTransactionId<R> = R & {
     [TRANSACTION_ID_FIELD]: TransactionId;
 };
@@ -844,6 +847,7 @@ declare class SpanImpl implements Span {
     private state;
     private isMerge;
     private loggedEndTime;
+    private propagatedEvent;
     private parentObjectType;
     private parentObjectId;
     private parentComputeObjectMetadataArgs;
@@ -854,7 +858,7 @@ declare class SpanImpl implements Span {
     kind: "span";
     constructor(args: {
         state: BraintrustState;
-        parentObjectType: SpanObjectTypeV2;
+        parentObjectType: SpanObjectTypeV3;
         parentObjectId: LazyValue<string>;
         parentComputeObjectMetadataArgs: Record<string, any> | undefined;
         parentSpanIds: ParentSpanIds | undefined;
@@ -887,6 +891,8 @@ declare class Dataset<IsLegacyDataset extends boolean = typeof DEFAULT_IS_LEGACY
     get name(): Promise<string>;
     get project(): Promise<ObjectMetadata>;
     protected getState(): Promise<BraintrustState>;
+    private validateEvent;
+    private createArgs;
     /**
      * Insert a single record to the dataset. The record will be batched and uploaded behind the scenes. If you pass in an `id`,
      * and a record with that `id` already exists, it will be overwritten (upsert).
@@ -911,6 +917,26 @@ declare class Dataset<IsLegacyDataset extends boolean = typeof DEFAULT_IS_LEGACY
         readonly id?: string;
         readonly output?: unknown;
     }): string;
+    /**
+     * Update fields of a single record in the dataset. The updated fields will be batched and uploaded behind the scenes.
+     * You must pass in an `id` of the record to update. Only the fields provided will be updated; other fields will remain unchanged.
+     *
+     * @param event The fields to update in the record.
+     * @param event.id The unique identifier of the record to update.
+     * @param event.input (Optional) The new input value for the record (an arbitrary, JSON serializable object).
+     * @param event.expected (Optional) The new expected output value for the record (an arbitrary, JSON serializable object).
+     * @param event.tags (Optional) A list of strings to update the tags of the record.
+     * @param event.metadata (Optional) A dictionary to update the metadata of the record. The values in `metadata` can be any
+     * JSON-serializable type, but its keys must be strings.
+     * @returns The `id` of the updated record.
+     */
+    update({ input, expected, metadata, tags, id, }: {
+        readonly id: string;
+        readonly input?: unknown;
+        readonly expected?: unknown;
+        readonly tags?: string[];
+        readonly metadata?: Record<string, unknown>;
+    }): string;
     delete(id: string): string;
     /**
      * Summarize the dataset, including high level metrics about its size and other metadata.
@@ -955,10 +981,13 @@ type CompiledPrompt<Flavor extends "chat" | "completion"> = CompiledPromptParams
     };
 } & (Flavor extends "chat" ? ChatPrompt : Flavor extends "completion" ? CompletionPrompt : {});
 type DefaultPromptArgs = Partial<CompiledPromptParams & AnyModelParam & ChatPrompt & CompletionPrompt>;
+declare function renderMessage<T extends Message>(render: (template: string) => string, message: T): T;
 declare class Prompt {
     private metadata;
     private defaults;
     private noTrace;
+    private parsedPromptData;
+    private hasParsedPromptData;
     constructor(metadata: Omit<Prompt$1, "log_id"> | PromptSessionEvent, defaults: DefaultPromptArgs, noTrace: boolean);
     get id(): string;
     get projectId(): string;
@@ -978,6 +1007,7 @@ declare class Prompt {
         flavor?: Flavor;
     }): CompiledPrompt<Flavor>;
     private runBuild;
+    private getParsedPromptData;
 }
 type AnyDataset = Dataset<boolean>;
 /**
@@ -1251,4 +1281,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, logError, login, loginToState, newId, parseCachedHeader, setFetch, startSpan, summarize, traceable, traced, updateSpan, 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, logError, login, loginToState, newId, parseCachedHeader, renderMessage, setFetch, startSpan, summarize, traceable, traced, updateSpan, withCurrent, withDataset, withExperiment, withLogger, wrapOpenAI, wrapOpenAIv4, wrapTraced };

package/dist/browser.d.ts

@@ -1,5 +1,5 @@
-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 { GitMetadataSettings, LogFeedbackFullArgs, ExperimentEvent, BackgroundLogEvent, ExperimentLogFullArgs, ExperimentLogPartialArgs, IdField, SpanType, RepoInfo, DEFAULT_IS_LEGACY_DATASET, TRANSACTION_ID_FIELD, TransactionId, SpanObjectTypeV3, DatasetRecord } from '@braintrust/core';
+import { PromptData, OpenAIMessage, Tools, AnyModelParam, Message, Prompt as Prompt$1, PromptSessionEvent } from '@braintrust/core/typespecs';
 import { z } from 'zod';
 
 interface IsoAsyncLocalStorage<T> {
@@ -16,8 +16,6 @@ declare class LazyValue<T> {
     get hasComputed(): boolean;
 }
 
-/// <reference lib="dom" />
-
 type SetCurrentArg = {
     setCurrent?: boolean;
 };
@@ -29,6 +27,7 @@ type StartSpanArgs = {
     startTime?: number;
     parent?: string;
     event?: StartSpanEventArgs;
+    propagatedEvent?: StartSpanEventArgs;
 };
 type EndSpanArgs = {
     endTime?: number;
@@ -685,6 +684,10 @@ declare function flush(options?: OptionalStateArg): Promise<void>;
  * @param fetch The fetch implementation to use.
  */
 declare function setFetch(fetch: typeof globalThis.fetch): void;
+/**
+ * Runs the provided callback with the span as the current span.
+ */
+declare function withCurrent<R>(span: Span, callback: (span: Span) => R, state?: BraintrustState): R;
 type WithTransactionId<R> = R & {
     [TRANSACTION_ID_FIELD]: TransactionId;
 };
@@ -844,6 +847,7 @@ declare class SpanImpl implements Span {
     private state;
     private isMerge;
     private loggedEndTime;
+    private propagatedEvent;
     private parentObjectType;
     private parentObjectId;
     private parentComputeObjectMetadataArgs;
@@ -854,7 +858,7 @@ declare class SpanImpl implements Span {
     kind: "span";
     constructor(args: {
         state: BraintrustState;
-        parentObjectType: SpanObjectTypeV2;
+        parentObjectType: SpanObjectTypeV3;
         parentObjectId: LazyValue<string>;
         parentComputeObjectMetadataArgs: Record<string, any> | undefined;
         parentSpanIds: ParentSpanIds | undefined;
@@ -887,6 +891,8 @@ declare class Dataset<IsLegacyDataset extends boolean = typeof DEFAULT_IS_LEGACY
     get name(): Promise<string>;
     get project(): Promise<ObjectMetadata>;
     protected getState(): Promise<BraintrustState>;
+    private validateEvent;
+    private createArgs;
     /**
      * Insert a single record to the dataset. The record will be batched and uploaded behind the scenes. If you pass in an `id`,
      * and a record with that `id` already exists, it will be overwritten (upsert).
@@ -911,6 +917,26 @@ declare class Dataset<IsLegacyDataset extends boolean = typeof DEFAULT_IS_LEGACY
         readonly id?: string;
         readonly output?: unknown;
     }): string;
+    /**
+     * Update fields of a single record in the dataset. The updated fields will be batched and uploaded behind the scenes.
+     * You must pass in an `id` of the record to update. Only the fields provided will be updated; other fields will remain unchanged.
+     *
+     * @param event The fields to update in the record.
+     * @param event.id The unique identifier of the record to update.
+     * @param event.input (Optional) The new input value for the record (an arbitrary, JSON serializable object).
+     * @param event.expected (Optional) The new expected output value for the record (an arbitrary, JSON serializable object).
+     * @param event.tags (Optional) A list of strings to update the tags of the record.
+     * @param event.metadata (Optional) A dictionary to update the metadata of the record. The values in `metadata` can be any
+     * JSON-serializable type, but its keys must be strings.
+     * @returns The `id` of the updated record.
+     */
+    update({ input, expected, metadata, tags, id, }: {
+        readonly id: string;
+        readonly input?: unknown;
+        readonly expected?: unknown;
+        readonly tags?: string[];
+        readonly metadata?: Record<string, unknown>;
+    }): string;
     delete(id: string): string;
     /**
      * Summarize the dataset, including high level metrics about its size and other metadata.
@@ -955,10 +981,13 @@ type CompiledPrompt<Flavor extends "chat" | "completion"> = CompiledPromptParams
     };
 } & (Flavor extends "chat" ? ChatPrompt : Flavor extends "completion" ? CompletionPrompt : {});
 type DefaultPromptArgs = Partial<CompiledPromptParams & AnyModelParam & ChatPrompt & CompletionPrompt>;
+declare function renderMessage<T extends Message>(render: (template: string) => string, message: T): T;
 declare class Prompt {
     private metadata;
     private defaults;
     private noTrace;
+    private parsedPromptData;
+    private hasParsedPromptData;
     constructor(metadata: Omit<Prompt$1, "log_id"> | PromptSessionEvent, defaults: DefaultPromptArgs, noTrace: boolean);
     get id(): string;
     get projectId(): string;
@@ -978,6 +1007,7 @@ declare class Prompt {
         flavor?: Flavor;
     }): CompiledPrompt<Flavor>;
     private runBuild;
+    private getParsedPromptData;
 }
 type AnyDataset = Dataset<boolean>;
 /**
@@ -1251,4 +1281,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, logError, login, loginToState, newId, parseCachedHeader, setFetch, startSpan, summarize, traceable, traced, updateSpan, 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, logError, login, loginToState, newId, parseCachedHeader, renderMessage, setFetch, startSpan, summarize, traceable, traced, updateSpan, withCurrent, withDataset, withExperiment, withLogger, wrapOpenAI, wrapOpenAIv4, wrapTraced };