langsmith 0.3.11 → 0.3.12

package/dist/client.cjs

@@ -1053,6 +1053,7 @@ class Client {
      * @param query - The query string to filter by.
      * @param filter - The filter string to apply to the run spans.
      * @param traceFilter - The filter string to apply on the root run of the trace.
+     * @param treeFilter - The filter string to apply on other runs in the trace.
      * @param limit - The maximum number of runs to retrieve.
      * @returns {AsyncIterable<Run>} - The runs.
      *
@@ -1736,7 +1737,7 @@ class Client {
             datasetId = (await this.readDataset({ datasetName })).id;
         }
         else {
-            throw new Error("Must provide datasetName or datasetId");
+            throw new Error("Must provide either datasetName or datasetId");
         }
         const response = await this._getResponse(`${path}/${datasetId}/openai_ft`);
         const datasetText = await response.text();
@@ -1928,74 +1929,104 @@ class Client {
         const result = await response.json();
         return result["examples"];
     }
-    async createExample(inputs, outputs, { datasetId, datasetName, createdAt, exampleId, metadata, split, sourceRunId, }) {
-        let datasetId_ = datasetId;
-        if (datasetId_ === undefined && datasetName === undefined) {
+    async createExample(inputsOrUpdate, outputs, options) {
+        if (isExampleCreate(inputsOrUpdate)) {
+            if (outputs !== undefined || options !== undefined) {
+                throw new Error("Cannot provide outputs or options when using ExampleCreate object");
+            }
+        }
+        let datasetId_ = outputs ? options?.datasetId : inputsOrUpdate.dataset_id;
+        const datasetName_ = outputs
+            ? options?.datasetName
+            : inputsOrUpdate.dataset_name;
+        if (datasetId_ === undefined && datasetName_ === undefined) {
             throw new Error("Must provide either datasetName or datasetId");
         }
-        else if (datasetId_ !== undefined && datasetName !== undefined) {
+        else if (datasetId_ !== undefined && datasetName_ !== undefined) {
             throw new Error("Must provide either datasetName or datasetId, not both");
         }
         else if (datasetId_ === undefined) {
-            const dataset = await this.readDataset({ datasetName });
+            const dataset = await this.readDataset({ datasetName: datasetName_ });
             datasetId_ = dataset.id;
         }
-        const createdAt_ = createdAt || new Date();
-        const data = {
-            dataset_id: datasetId_,
-            inputs,
-            outputs,
-            created_at: createdAt_?.toISOString(),
-            id: exampleId,
-            metadata,
-            split,
-            source_run_id: sourceRunId,
-        };
-        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/examples`, {
-            method: "POST",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(data),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await (0, error_js_1.raiseForStatus)(response, "create example");
-        const result = await response.json();
-        return result;
+        const createdAt_ = (outputs ? options?.createdAt : inputsOrUpdate.created_at) || new Date();
+        let data;
+        if (!isExampleCreate(inputsOrUpdate)) {
+            data = {
+                inputs: inputsOrUpdate,
+                outputs,
+                created_at: createdAt_?.toISOString(),
+                id: options?.exampleId,
+                metadata: options?.metadata,
+                split: options?.split,
+                source_run_id: options?.sourceRunId,
+                use_source_run_io: options?.useSourceRunIO,
+                use_source_run_attachments: options?.useSourceRunAttachments,
+                attachments: options?.attachments,
+            };
+        }
+        else {
+            data = inputsOrUpdate;
+        }
+        const response = await this._uploadExamplesMultipart(datasetId_, [data]);
+        const example = await this.readExample(response.example_ids?.[0] ?? uuid.v4());
+        return example;
     }
-    async createExamples(props) {
-        const { inputs, outputs, metadata, sourceRunIds, exampleIds, datasetId, datasetName, } = props;
+    async createExamples(propsOrUploads) {
+        if (Array.isArray(propsOrUploads)) {
+            if (propsOrUploads.length === 0) {
+                return [];
+            }
+            const uploads = propsOrUploads;
+            let datasetId_ = uploads[0].dataset_id;
+            const datasetName_ = uploads[0].dataset_name;
+            if (datasetId_ === undefined && datasetName_ === undefined) {
+                throw new Error("Must provide either datasetName or datasetId");
+            }
+            else if (datasetId_ !== undefined && datasetName_ !== undefined) {
+                throw new Error("Must provide either datasetName or datasetId, not both");
+            }
+            else if (datasetId_ === undefined) {
+                const dataset = await this.readDataset({ datasetName: datasetName_ });
+                datasetId_ = dataset.id;
+            }
+            const response = await this._uploadExamplesMultipart(datasetId_, uploads);
+            const examples = await Promise.all(response.example_ids.map((id) => this.readExample(id)));
+            return examples;
+        }
+        const { inputs, outputs, metadata, splits, sourceRunIds, useSourceRunIOs, useSourceRunAttachments, attachments, exampleIds, datasetId, datasetName, } = propsOrUploads;
+        if (inputs === undefined) {
+            throw new Error("Must provide inputs when using legacy parameters");
+        }
         let datasetId_ = datasetId;
-        if (datasetId_ === undefined && datasetName === undefined) {
+        const datasetName_ = datasetName;
+        if (datasetId_ === undefined && datasetName_ === undefined) {
             throw new Error("Must provide either datasetName or datasetId");
         }
-        else if (datasetId_ !== undefined && datasetName !== undefined) {
+        else if (datasetId_ !== undefined && datasetName_ !== undefined) {
             throw new Error("Must provide either datasetName or datasetId, not both");
         }
         else if (datasetId_ === undefined) {
-            const dataset = await this.readDataset({ datasetName });
+            const dataset = await this.readDataset({ datasetName: datasetName_ });
             datasetId_ = dataset.id;
         }
         const formattedExamples = inputs.map((input, idx) => {
             return {
                 dataset_id: datasetId_,
                 inputs: input,
-                outputs: outputs ? outputs[idx] : undefined,
-                metadata: metadata ? metadata[idx] : undefined,
-                split: props.splits ? props.splits[idx] : undefined,
-                id: exampleIds ? exampleIds[idx] : undefined,
-                source_run_id: sourceRunIds ? sourceRunIds[idx] : undefined,
+                outputs: outputs?.[idx],
+                metadata: metadata?.[idx],
+                split: splits?.[idx],
+                id: exampleIds?.[idx],
+                attachments: attachments?.[idx],
+                source_run_id: sourceRunIds?.[idx],
+                use_source_run_io: useSourceRunIOs?.[idx],
+                use_source_run_attachments: useSourceRunAttachments?.[idx],
             };
         });
-        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/examples/bulk`, {
-            method: "POST",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(formattedExamples),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await (0, error_js_1.raiseForStatus)(response, "create examples");
-        const result = await response.json();
-        return result;
+        const response = await this._uploadExamplesMultipart(datasetId_, formattedExamples);
+        const examples = await Promise.all(response.example_ids.map((id) => this.readExample(id)));
+        return examples;
     }
     async createLLMExample(input, generation, options) {
         return this.createExample({ input }, { output: generation }, options);
@@ -2019,7 +2050,6 @@ class Client {
         const { attachment_urls, ...rest } = rawExample;
         const example = rest;
         if (attachment_urls) {
-            // add attachments back to the example
             example.attachments = Object.entries(attachment_urls).reduce((acc, [key, value]) => {
                 acc[key.slice("attachment.".length)] = {
                     presigned_url: value.presigned_url,
@@ -2116,30 +2146,43 @@ class Client {
         await (0, error_js_1.raiseForStatus)(response, `delete ${path}`);
         await response.json();
     }
-    async updateExample(exampleId, update) {
+    async updateExample(exampleIdOrUpdate, update) {
+        let exampleId;
+        if (update) {
+            exampleId = exampleIdOrUpdate;
+        }
+        else {
+            exampleId = exampleIdOrUpdate.id;
+        }
         (0, _uuid_js_1.assertUuid)(exampleId);
-        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/examples/${exampleId}`, {
-            method: "PATCH",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(update),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await (0, error_js_1.raiseForStatus)(response, "update example");
-        const result = await response.json();
-        return result;
+        let updateToUse;
+        if (update) {
+            updateToUse = { id: exampleId, ...update };
+        }
+        else {
+            updateToUse = exampleIdOrUpdate;
+        }
+        let datasetId;
+        if (updateToUse.dataset_id !== undefined) {
+            datasetId = updateToUse.dataset_id;
+        }
+        else {
+            const example = await this.readExample(exampleId);
+            datasetId = example.dataset_id;
+        }
+        return this._updateExamplesMultipart(datasetId, [updateToUse]);
     }
     async updateExamples(update) {
-        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/examples/bulk`, {
-            method: "PATCH",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(update),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await (0, error_js_1.raiseForStatus)(response, "update examples");
-        const result = await response.json();
-        return result;
+        // We will naively get dataset id from first example and assume it works for all
+        let datasetId;
+        if (update[0].dataset_id === undefined) {
+            const example = await this.readExample(update[0].id);
+            datasetId = example.dataset_id;
+        }
+        else {
+            datasetId = update[0].dataset_id;
+        }
+        return this._updateExamplesMultipart(datasetId, update);
     }
     /**
      * Get dataset version by closest date or exact tag.
@@ -2379,10 +2422,10 @@ class Client {
      * applications the ability to submit feedback without needing
      * to expose an API key.
      *
-     * @param runId - The ID of the run.
-     * @param feedbackKey - The feedback key.
-     * @param options - Additional options for the token.
-     * @param options.expiration - The expiration time for the token.
+     * @param runId The ID of the run.
+     * @param feedbackKey The feedback key.
+     * @param options Additional options for the token.
+     * @param options.expiration The expiration time for the token.
      *
      * @returns A promise that resolves to a FeedbackIngestToken.
      */
@@ -2844,6 +2887,9 @@ class Client {
      * @returns Promise with the update response
      */
     async updateExamplesMultipart(datasetId, updates = []) {
+        return this._updateExamplesMultipart(datasetId, updates);
+    }
+    async _updateExamplesMultipart(datasetId, updates = []) {
         if (!(await this._getMultiPartSupport())) {
             throw new Error("Your LangSmith version does not allow using the multipart examples endpoint, please update to the latest version.");
         }
@@ -2861,7 +2907,7 @@ class Client {
                 type: "application/json",
             });
             formData.append(exampleId, exampleBlob);
-            // Add inputs
+            // Add inputs if present
             if (example.inputs) {
                 const stringifiedInputs = (0, index_js_2.serialize)(example.inputs);
                 const inputsBlob = new Blob([stringifiedInputs], {
@@ -2903,7 +2949,8 @@ class Client {
                 formData.append(`${exampleId}.attachments_operations`, attachmentsOperationsBlob);
             }
         }
-        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/v1/platform/datasets/${datasetId}/examples`, {
+        const datasetIdToUse = datasetId ?? updates[0]?.dataset_id;
+        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/v1/platform/datasets/${datasetIdToUse}/examples`, {
             method: "PATCH",
             headers: this.headers,
             body: formData,
@@ -2915,8 +2962,12 @@ class Client {
      * Upload examples with attachments using multipart form data.
      * @param uploads List of ExampleUploadWithAttachments objects to upload
      * @returns Promise with the upload response
+     * @deprecated This method is deprecated and will be removed in future LangSmith versions, please use `createExamples` instead
      */
     async uploadExamplesMultipart(datasetId, uploads = []) {
+        return this._uploadExamplesMultipart(datasetId, uploads);
+    }
+    async _uploadExamplesMultipart(datasetId, uploads = []) {
         if (!(await this._getMultiPartSupport())) {
             throw new Error("Your LangSmith version does not allow using the multipart examples endpoint, please update to the latest version.");
         }
@@ -2928,6 +2979,13 @@ class Client {
                 created_at: example.created_at,
                 ...(example.metadata && { metadata: example.metadata }),
                 ...(example.split && { split: example.split }),
+                ...(example.source_run_id && { source_run_id: example.source_run_id }),
+                ...(example.use_source_run_io && {
+                    use_source_run_io: example.use_source_run_io,
+                }),
+                ...(example.use_source_run_attachments && {
+                    use_source_run_attachments: example.use_source_run_attachments,
+                }),
             };
             // Add main example data
             const stringifiedExample = (0, index_js_2.serialize)(exampleBody);
@@ -2935,12 +2993,14 @@ class Client {
                 type: "application/json",
             });
             formData.append(exampleId, exampleBlob);
-            // Add inputs
-            const stringifiedInputs = (0, index_js_2.serialize)(example.inputs);
-            const inputsBlob = new Blob([stringifiedInputs], {
-                type: "application/json",
-            });
-            formData.append(`${exampleId}.inputs`, inputsBlob);
+            // Add inputs if present
+            if (example.inputs) {
+                const stringifiedInputs = (0, index_js_2.serialize)(example.inputs);
+                const inputsBlob = new Blob([stringifiedInputs], {
+                    type: "application/json",
+                });
+                formData.append(`${exampleId}.inputs`, inputsBlob);
+            }
             // Add outputs if present
             if (example.outputs) {
                 const stringifiedOutputs = (0, index_js_2.serialize)(example.outputs);
@@ -3201,3 +3261,6 @@ class Client {
     }
 }
 exports.Client = Client;
+function isExampleCreate(input) {
+    return "dataset_id" in input || "dataset_name" in input;
+}

package/dist/client.d.ts

@@ -1,5 +1,5 @@
 import { AsyncCallerParams } from "./utils/async_caller.js";
-import { ComparativeExperiment, DataType, Dataset, DatasetDiffInfo, DatasetShareSchema, Example, ExampleUpdate, ExampleUpdateWithId, Feedback, FeedbackConfig, FeedbackIngestToken, KVMap, LangChainBaseMessage, LangSmithSettings, LikePromptResponse, Prompt, PromptCommit, PromptSortField, Run, RunCreate, RunUpdate, ScoreType, ExampleSearch, TimeDelta, TracerSession, TracerSessionResult, ValueType, AnnotationQueue, RunWithAnnotationQueueInfo, Attachments, ExampleUploadWithAttachments, UploadExamplesResponse, ExampleUpdateWithAttachments, UpdateExamplesResponse, DatasetVersion } from "./schemas.js";
+import { ComparativeExperiment, DataType, Dataset, DatasetDiffInfo, DatasetShareSchema, Example, ExampleCreate, ExampleUpdate, ExampleUpdateWithoutId, Feedback, FeedbackConfig, FeedbackIngestToken, KVMap, LangChainBaseMessage, LangSmithSettings, LikePromptResponse, Prompt, PromptCommit, PromptSortField, Run, RunCreate, RunUpdate, ScoreType, ExampleSearch, TimeDelta, TracerSession, TracerSessionResult, ValueType, AnnotationQueue, RunWithAnnotationQueueInfo, Attachments, UploadExamplesResponse, UpdateExamplesResponse, DatasetVersion } from "./schemas.js";
 import { EvaluationResult, EvaluationResults, RunEvaluator } from "./evaluation/evaluator.js";
 export interface ClientConfig {
     apiUrl?: string;
@@ -163,6 +163,12 @@ export type CreateExampleOptions = {
     split?: string | string[];
     /** The ID of the source run associated with this example. */
     sourceRunId?: string;
+    /** Whether to use the inputs and outputs from the source run. */
+    useSourceRunIO?: boolean;
+    /** Which attachments from the source run to use. */
+    useSourceRunAttachments?: string[];
+    /** Attachments for the example */
+    attachments?: Attachments;
 };
 export type CreateProjectParams = {
     projectName: string;
@@ -295,6 +301,7 @@ export declare class Client implements LangSmithTracingClientInterface {
      * @param query - The query string to filter by.
      * @param filter - The filter string to apply to the run spans.
      * @param traceFilter - The filter string to apply on the root run of the trace.
+     * @param treeFilter - The filter string to apply on other runs in the trace.
      * @param limit - The maximum number of runs to retrieve.
      * @returns {AsyncIterable<Run>} - The runs.
      *
@@ -550,13 +557,22 @@ export declare class Client implements LangSmithTracingClientInterface {
     similarExamples(inputs: KVMap, datasetId: string, limit: number, { filter, }?: {
         filter?: string;
     }): Promise<ExampleSearch[]>;
-    createExample(inputs: KVMap, outputs: KVMap, { datasetId, datasetName, createdAt, exampleId, metadata, split, sourceRunId, }: CreateExampleOptions): Promise<Example>;
+    createExample(update: ExampleCreate): Promise<Example>;
+    /**
+     * @deprecated This signature is deprecated, use createExample(update: ExampleCreate) instead
+     */
+    createExample(inputs: KVMap, outputs: KVMap, options: CreateExampleOptions): Promise<Example>;
+    createExamples(uploads: ExampleCreate[]): Promise<Example[]>;
+    /** @deprecated Use the uploads-only overload instead */
     createExamples(props: {
-        inputs: Array<KVMap>;
+        inputs?: Array<KVMap>;
         outputs?: Array<KVMap>;
         metadata?: Array<KVMap>;
         splits?: Array<string | Array<string>>;
         sourceRunIds?: Array<string>;
+        useSourceRunIOs?: Array<boolean>;
+        useSourceRunAttachments?: Array<string[]>;
+        attachments?: Array<Attachments>;
         exampleIds?: Array<string>;
         datasetId?: string;
         datasetName?: string;
@@ -578,8 +594,12 @@ export declare class Client implements LangSmithTracingClientInterface {
         includeAttachments?: boolean;
     }): AsyncIterable<Example>;
     deleteExample(exampleId: string): Promise<void>;
-    updateExample(exampleId: string, update: ExampleUpdate): Promise<object>;
-    updateExamples(update: ExampleUpdateWithId[]): Promise<object>;
+    /**
+     * @deprecated This signature is deprecated, use updateExample(update: ExampleUpdate) instead
+     */
+    updateExample(exampleId: string, update: ExampleUpdateWithoutId): Promise<object>;
+    updateExample(update: ExampleUpdate): Promise<object>;
+    updateExamples(update: ExampleUpdate[]): Promise<object>;
     /**
      * Get dataset version by closest date or exact tag.
      *
@@ -653,10 +673,10 @@ export declare class Client implements LangSmithTracingClientInterface {
      * applications the ability to submit feedback without needing
      * to expose an API key.
      *
-     * @param runId - The ID of the run.
-     * @param feedbackKey - The feedback key.
-     * @param options - Additional options for the token.
-     * @param options.expiration - The expiration time for the token.
+     * @param runId The ID of the run.
+     * @param feedbackKey The feedback key.
+     * @param options Additional options for the token.
+     * @param options.expiration The expiration time for the token.
      *
      * @returns A promise that resolves to a FeedbackIngestToken.
      */
@@ -796,13 +816,16 @@ export declare class Client implements LangSmithTracingClientInterface {
      * @param updates List of ExampleUpdateWithAttachments objects to upsert
      * @returns Promise with the update response
      */
-    updateExamplesMultipart(datasetId: string, updates?: ExampleUpdateWithAttachments[]): Promise<UpdateExamplesResponse>;
+    updateExamplesMultipart(datasetId: string, updates?: ExampleUpdate[]): Promise<UpdateExamplesResponse>;
+    private _updateExamplesMultipart;
     /**
      * Upload examples with attachments using multipart form data.
      * @param uploads List of ExampleUploadWithAttachments objects to upload
      * @returns Promise with the upload response
+     * @deprecated This method is deprecated and will be removed in future LangSmith versions, please use `createExamples` instead
      */
-    uploadExamplesMultipart(datasetId: string, uploads?: ExampleUploadWithAttachments[]): Promise<UploadExamplesResponse>;
+    uploadExamplesMultipart(datasetId: string, uploads?: ExampleCreate[]): Promise<UploadExamplesResponse>;
+    private _uploadExamplesMultipart;
     updatePrompt(promptIdentifier: string, options?: {
         description?: string;
         readme?: string;

package/dist/client.js

@@ -1025,6 +1025,7 @@ export class Client {
      * @param query - The query string to filter by.
      * @param filter - The filter string to apply to the run spans.
      * @param traceFilter - The filter string to apply on the root run of the trace.
+     * @param treeFilter - The filter string to apply on other runs in the trace.
      * @param limit - The maximum number of runs to retrieve.
      * @returns {AsyncIterable<Run>} - The runs.
      *
@@ -1708,7 +1709,7 @@ export class Client {
             datasetId = (await this.readDataset({ datasetName })).id;
         }
         else {
-            throw new Error("Must provide datasetName or datasetId");
+            throw new Error("Must provide either datasetName or datasetId");
         }
         const response = await this._getResponse(`${path}/${datasetId}/openai_ft`);
         const datasetText = await response.text();
@@ -1900,74 +1901,104 @@ export class Client {
         const result = await response.json();
         return result["examples"];
     }
-    async createExample(inputs, outputs, { datasetId, datasetName, createdAt, exampleId, metadata, split, sourceRunId, }) {
-        let datasetId_ = datasetId;
-        if (datasetId_ === undefined && datasetName === undefined) {
+    async createExample(inputsOrUpdate, outputs, options) {
+        if (isExampleCreate(inputsOrUpdate)) {
+            if (outputs !== undefined || options !== undefined) {
+                throw new Error("Cannot provide outputs or options when using ExampleCreate object");
+            }
+        }
+        let datasetId_ = outputs ? options?.datasetId : inputsOrUpdate.dataset_id;
+        const datasetName_ = outputs
+            ? options?.datasetName
+            : inputsOrUpdate.dataset_name;
+        if (datasetId_ === undefined && datasetName_ === undefined) {
             throw new Error("Must provide either datasetName or datasetId");
         }
-        else if (datasetId_ !== undefined && datasetName !== undefined) {
+        else if (datasetId_ !== undefined && datasetName_ !== undefined) {
             throw new Error("Must provide either datasetName or datasetId, not both");
         }
         else if (datasetId_ === undefined) {
-            const dataset = await this.readDataset({ datasetName });
+            const dataset = await this.readDataset({ datasetName: datasetName_ });
             datasetId_ = dataset.id;
         }
-        const createdAt_ = createdAt || new Date();
-        const data = {
-            dataset_id: datasetId_,
-            inputs,
-            outputs,
-            created_at: createdAt_?.toISOString(),
-            id: exampleId,
-            metadata,
-            split,
-            source_run_id: sourceRunId,
-        };
-        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/examples`, {
-            method: "POST",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(data),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await raiseForStatus(response, "create example");
-        const result = await response.json();
-        return result;
+        const createdAt_ = (outputs ? options?.createdAt : inputsOrUpdate.created_at) || new Date();
+        let data;
+        if (!isExampleCreate(inputsOrUpdate)) {
+            data = {
+                inputs: inputsOrUpdate,
+                outputs,
+                created_at: createdAt_?.toISOString(),
+                id: options?.exampleId,
+                metadata: options?.metadata,
+                split: options?.split,
+                source_run_id: options?.sourceRunId,
+                use_source_run_io: options?.useSourceRunIO,
+                use_source_run_attachments: options?.useSourceRunAttachments,
+                attachments: options?.attachments,
+            };
+        }
+        else {
+            data = inputsOrUpdate;
+        }
+        const response = await this._uploadExamplesMultipart(datasetId_, [data]);
+        const example = await this.readExample(response.example_ids?.[0] ?? uuid.v4());
+        return example;
     }
-    async createExamples(props) {
-        const { inputs, outputs, metadata, sourceRunIds, exampleIds, datasetId, datasetName, } = props;
+    async createExamples(propsOrUploads) {
+        if (Array.isArray(propsOrUploads)) {
+            if (propsOrUploads.length === 0) {
+                return [];
+            }
+            const uploads = propsOrUploads;
+            let datasetId_ = uploads[0].dataset_id;
+            const datasetName_ = uploads[0].dataset_name;
+            if (datasetId_ === undefined && datasetName_ === undefined) {
+                throw new Error("Must provide either datasetName or datasetId");
+            }
+            else if (datasetId_ !== undefined && datasetName_ !== undefined) {
+                throw new Error("Must provide either datasetName or datasetId, not both");
+            }
+            else if (datasetId_ === undefined) {
+                const dataset = await this.readDataset({ datasetName: datasetName_ });
+                datasetId_ = dataset.id;
+            }
+            const response = await this._uploadExamplesMultipart(datasetId_, uploads);
+            const examples = await Promise.all(response.example_ids.map((id) => this.readExample(id)));
+            return examples;
+        }
+        const { inputs, outputs, metadata, splits, sourceRunIds, useSourceRunIOs, useSourceRunAttachments, attachments, exampleIds, datasetId, datasetName, } = propsOrUploads;
+        if (inputs === undefined) {
+            throw new Error("Must provide inputs when using legacy parameters");
+        }
         let datasetId_ = datasetId;
-        if (datasetId_ === undefined && datasetName === undefined) {
+        const datasetName_ = datasetName;
+        if (datasetId_ === undefined && datasetName_ === undefined) {
             throw new Error("Must provide either datasetName or datasetId");
         }
-        else if (datasetId_ !== undefined && datasetName !== undefined) {
+        else if (datasetId_ !== undefined && datasetName_ !== undefined) {
             throw new Error("Must provide either datasetName or datasetId, not both");
         }
         else if (datasetId_ === undefined) {
-            const dataset = await this.readDataset({ datasetName });
+            const dataset = await this.readDataset({ datasetName: datasetName_ });
             datasetId_ = dataset.id;
         }
         const formattedExamples = inputs.map((input, idx) => {
             return {
                 dataset_id: datasetId_,
                 inputs: input,
-                outputs: outputs ? outputs[idx] : undefined,
-                metadata: metadata ? metadata[idx] : undefined,
-                split: props.splits ? props.splits[idx] : undefined,
-                id: exampleIds ? exampleIds[idx] : undefined,
-                source_run_id: sourceRunIds ? sourceRunIds[idx] : undefined,
+                outputs: outputs?.[idx],
+                metadata: metadata?.[idx],
+                split: splits?.[idx],
+                id: exampleIds?.[idx],
+                attachments: attachments?.[idx],
+                source_run_id: sourceRunIds?.[idx],
+                use_source_run_io: useSourceRunIOs?.[idx],
+                use_source_run_attachments: useSourceRunAttachments?.[idx],
             };
         });
-        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/examples/bulk`, {
-            method: "POST",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(formattedExamples),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await raiseForStatus(response, "create examples");
-        const result = await response.json();
-        return result;
+        const response = await this._uploadExamplesMultipart(datasetId_, formattedExamples);
+        const examples = await Promise.all(response.example_ids.map((id) => this.readExample(id)));
+        return examples;
     }
     async createLLMExample(input, generation, options) {
         return this.createExample({ input }, { output: generation }, options);
@@ -1991,7 +2022,6 @@ export class Client {
         const { attachment_urls, ...rest } = rawExample;
         const example = rest;
         if (attachment_urls) {
-            // add attachments back to the example
             example.attachments = Object.entries(attachment_urls).reduce((acc, [key, value]) => {
                 acc[key.slice("attachment.".length)] = {
                     presigned_url: value.presigned_url,
@@ -2088,30 +2118,43 @@ export class Client {
         await raiseForStatus(response, `delete ${path}`);
         await response.json();
     }
-    async updateExample(exampleId, update) {
+    async updateExample(exampleIdOrUpdate, update) {
+        let exampleId;
+        if (update) {
+            exampleId = exampleIdOrUpdate;
+        }
+        else {
+            exampleId = exampleIdOrUpdate.id;
+        }
         assertUuid(exampleId);
-        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/examples/${exampleId}`, {
-            method: "PATCH",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(update),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await raiseForStatus(response, "update example");
-        const result = await response.json();
-        return result;
+        let updateToUse;
+        if (update) {
+            updateToUse = { id: exampleId, ...update };
+        }
+        else {
+            updateToUse = exampleIdOrUpdate;
+        }
+        let datasetId;
+        if (updateToUse.dataset_id !== undefined) {
+            datasetId = updateToUse.dataset_id;
+        }
+        else {
+            const example = await this.readExample(exampleId);
+            datasetId = example.dataset_id;
+        }
+        return this._updateExamplesMultipart(datasetId, [updateToUse]);
     }
     async updateExamples(update) {
-        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/examples/bulk`, {
-            method: "PATCH",
-            headers: { ...this.headers, "Content-Type": "application/json" },
-            body: JSON.stringify(update),
-            signal: AbortSignal.timeout(this.timeout_ms),
-            ...this.fetchOptions,
-        });
-        await raiseForStatus(response, "update examples");
-        const result = await response.json();
-        return result;
+        // We will naively get dataset id from first example and assume it works for all
+        let datasetId;
+        if (update[0].dataset_id === undefined) {
+            const example = await this.readExample(update[0].id);
+            datasetId = example.dataset_id;
+        }
+        else {
+            datasetId = update[0].dataset_id;
+        }
+        return this._updateExamplesMultipart(datasetId, update);
     }
     /**
      * Get dataset version by closest date or exact tag.
@@ -2351,10 +2394,10 @@ export class Client {
      * applications the ability to submit feedback without needing
      * to expose an API key.
      *
-     * @param runId - The ID of the run.
-     * @param feedbackKey - The feedback key.
-     * @param options - Additional options for the token.
-     * @param options.expiration - The expiration time for the token.
+     * @param runId The ID of the run.
+     * @param feedbackKey The feedback key.
+     * @param options Additional options for the token.
+     * @param options.expiration The expiration time for the token.
      *
      * @returns A promise that resolves to a FeedbackIngestToken.
      */
@@ -2816,6 +2859,9 @@ export class Client {
      * @returns Promise with the update response
      */
     async updateExamplesMultipart(datasetId, updates = []) {
+        return this._updateExamplesMultipart(datasetId, updates);
+    }
+    async _updateExamplesMultipart(datasetId, updates = []) {
         if (!(await this._getMultiPartSupport())) {
             throw new Error("Your LangSmith version does not allow using the multipart examples endpoint, please update to the latest version.");
         }
@@ -2833,7 +2879,7 @@ export class Client {
                 type: "application/json",
             });
             formData.append(exampleId, exampleBlob);
-            // Add inputs
+            // Add inputs if present
             if (example.inputs) {
                 const stringifiedInputs = serializePayloadForTracing(example.inputs);
                 const inputsBlob = new Blob([stringifiedInputs], {
@@ -2875,7 +2921,8 @@ export class Client {
                 formData.append(`${exampleId}.attachments_operations`, attachmentsOperationsBlob);
             }
         }
-        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/v1/platform/datasets/${datasetId}/examples`, {
+        const datasetIdToUse = datasetId ?? updates[0]?.dataset_id;
+        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/v1/platform/datasets/${datasetIdToUse}/examples`, {
             method: "PATCH",
             headers: this.headers,
             body: formData,
@@ -2887,8 +2934,12 @@ export class Client {
      * Upload examples with attachments using multipart form data.
      * @param uploads List of ExampleUploadWithAttachments objects to upload
      * @returns Promise with the upload response
+     * @deprecated This method is deprecated and will be removed in future LangSmith versions, please use `createExamples` instead
      */
     async uploadExamplesMultipart(datasetId, uploads = []) {
+        return this._uploadExamplesMultipart(datasetId, uploads);
+    }
+    async _uploadExamplesMultipart(datasetId, uploads = []) {
         if (!(await this._getMultiPartSupport())) {
             throw new Error("Your LangSmith version does not allow using the multipart examples endpoint, please update to the latest version.");
         }
@@ -2900,6 +2951,13 @@ export class Client {
                 created_at: example.created_at,
                 ...(example.metadata && { metadata: example.metadata }),
                 ...(example.split && { split: example.split }),
+                ...(example.source_run_id && { source_run_id: example.source_run_id }),
+                ...(example.use_source_run_io && {
+                    use_source_run_io: example.use_source_run_io,
+                }),
+                ...(example.use_source_run_attachments && {
+                    use_source_run_attachments: example.use_source_run_attachments,
+                }),
             };
             // Add main example data
             const stringifiedExample = serializePayloadForTracing(exampleBody);
@@ -2907,12 +2965,14 @@ export class Client {
                 type: "application/json",
             });
             formData.append(exampleId, exampleBlob);
-            // Add inputs
-            const stringifiedInputs = serializePayloadForTracing(example.inputs);
-            const inputsBlob = new Blob([stringifiedInputs], {
-                type: "application/json",
-            });
-            formData.append(`${exampleId}.inputs`, inputsBlob);
+            // Add inputs if present
+            if (example.inputs) {
+                const stringifiedInputs = serializePayloadForTracing(example.inputs);
+                const inputsBlob = new Blob([stringifiedInputs], {
+                    type: "application/json",
+                });
+                formData.append(`${exampleId}.inputs`, inputsBlob);
+            }
             // Add outputs if present
             if (example.outputs) {
                 const stringifiedOutputs = serializePayloadForTracing(example.outputs);
@@ -3172,3 +3232,6 @@ export class Client {
         ]);
     }
 }
+function isExampleCreate(input) {
+    return "dataset_id" in input || "dataset_name" in input;
+}

package/dist/index.cjs

@@ -8,4 +8,4 @@ Object.defineProperty(exports, "RunTree", { enumerable: true, get: function () {
 var fetch_js_1 = require("./singletons/fetch.cjs");
 Object.defineProperty(exports, "overrideFetchImplementation", { enumerable: true, get: function () { return fetch_js_1.overrideFetchImplementation; } });
 // Update using yarn bump-version
-exports.__version__ = "0.3.11";
+exports.__version__ = "0.3.12";

package/dist/index.d.ts

@@ -2,4 +2,4 @@ export { Client, type ClientConfig, type LangSmithTracingClientInterface, } from
 export type { Dataset, Example, TracerSession, Run, Feedback, RetrieverOutput, } from "./schemas.js";
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
-export declare const __version__ = "0.3.11";
+export declare const __version__ = "0.3.12";

package/dist/index.js

@@ -2,4 +2,4 @@ export { Client, } from "./client.js";
 export { RunTree } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 // Update using yarn bump-version
-export const __version__ = "0.3.11";
+export const __version__ = "0.3.12";

package/dist/schemas.d.ts

@@ -183,12 +183,7 @@ export interface RunUpdate {
      */
     attachments?: Attachments;
 }
-export interface ExampleCreate extends BaseExample {
-    id?: string;
-    created_at?: string;
-    split?: string | string[];
-}
-export interface ExampleUploadWithAttachments {
+export interface ExampleCreate {
     id?: string;
     inputs: KVMap;
     outputs?: KVMap;
@@ -196,8 +191,15 @@ export interface ExampleUploadWithAttachments {
     split?: string | string[];
     attachments?: Attachments;
     created_at?: string;
+    dataset_id?: string;
+    dataset_name?: string;
+    source_run_id?: string;
+    use_source_run_io?: boolean;
+    use_source_run_attachments?: string[];
+}
+export interface ExampleUploadWithAttachments extends ExampleCreate {
 }
-export interface ExampleUpdateWithAttachments {
+export interface ExampleUpdate {
     id: string;
     inputs?: KVMap;
     outputs?: KVMap;
@@ -205,6 +207,11 @@ export interface ExampleUpdateWithAttachments {
     split?: string | string[];
     attachments?: Attachments;
     attachments_operations?: KVMap;
+    dataset_id?: string;
+}
+export interface ExampleUpdateWithoutId extends Omit<ExampleUpdate, "id"> {
+}
+export interface ExampleUpdateWithAttachments extends ExampleUpdate {
 }
 export interface UploadExamplesResponse {
     count: number;
@@ -234,15 +241,7 @@ export interface RawExample extends BaseExample {
     runs: Run[];
     attachment_urls?: Record<string, RawAttachmentInfo>;
 }
-export interface ExampleUpdate {
-    dataset_id?: string;
-    inputs?: KVMap;
-    outputs?: KVMap;
-    metadata?: KVMap;
-    split?: string | string[];
-}
 export interface ExampleUpdateWithId extends ExampleUpdate {
-    id: string;
 }
 export interface ExampleSearch extends BaseExample {
     id: string;

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "langsmith",
-  "version": "0.3.11",
+  "version": "0.3.12",
   "description": "Client library to connect to the LangSmith LLM Tracing and Evaluation Platform.",
   "packageManager": "yarn@1.22.19",
   "files": [