langsmith 0.2.16-rc.9 → 0.3.0

package/dist/client.cjs

@@ -1784,6 +1784,40 @@ class Client {
         await (0, error_js_1.raiseForStatus)(response, "update dataset");
         return (await response.json());
     }
+    /**
+     * Updates a tag on a dataset.
+     *
+     * If the tag is already assigned to a different version of this dataset,
+     * the tag will be moved to the new version. The as_of parameter is used to
+     * determine which version of the dataset to apply the new tags to.
+     *
+     * It must be an exact version of the dataset to succeed. You can
+     * use the "readDatasetVersion" method to find the exact version
+     * to apply the tags to.
+     * @param params.datasetId The ID of the dataset to update. Must be provided if "datasetName" is not provided.
+     * @param params.datasetName The name of the dataset to update. Must be provided if "datasetId" is not provided.
+     * @param params.asOf The timestamp of the dataset to apply the new tags to.
+     * @param params.tag The new tag to apply to the dataset.
+     */
+    async updateDatasetTag(props) {
+        const { datasetId, datasetName, asOf, tag } = props;
+        if (!datasetId && !datasetName) {
+            throw new Error("Must provide either datasetName or datasetId");
+        }
+        const _datasetId = datasetId ?? (await this.readDataset({ datasetName })).id;
+        (0, _uuid_js_1.assertUuid)(_datasetId);
+        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/datasets/${_datasetId}/tags`, {
+            method: "PUT",
+            headers: { ...this.headers, "Content-Type": "application/json" },
+            body: JSON.stringify({
+                as_of: typeof asOf === "string" ? asOf : asOf.toISOString(),
+                tag,
+            }),
+            signal: AbortSignal.timeout(this.timeout_ms),
+            ...this.fetchOptions,
+        });
+        await (0, error_js_1.raiseForStatus)(response, "update dataset tags");
+    }
     async deleteDataset({ datasetId, datasetName, }) {
         let path = "/datasets";
         let datasetId_ = datasetId;
@@ -2099,6 +2133,47 @@ class Client {
         const result = await response.json();
         return result;
     }
+    /**
+     * Get dataset version by closest date or exact tag.
+     *
+     * Use this to resolve the nearest version to a given timestamp or for a given tag.
+     *
+     * @param options The options for getting the dataset version
+     * @param options.datasetId The ID of the dataset
+     * @param options.datasetName The name of the dataset
+     * @param options.asOf The timestamp of the dataset to retrieve
+     * @param options.tag The tag of the dataset to retrieve
+     * @returns The dataset version
+     */
+    async readDatasetVersion({ datasetId, datasetName, asOf, tag, }) {
+        let resolvedDatasetId;
+        if (!datasetId) {
+            const dataset = await this.readDataset({ datasetName });
+            resolvedDatasetId = dataset.id;
+        }
+        else {
+            resolvedDatasetId = datasetId;
+        }
+        (0, _uuid_js_1.assertUuid)(resolvedDatasetId);
+        if ((asOf && tag) || (!asOf && !tag)) {
+            throw new Error("Exactly one of asOf and tag must be specified.");
+        }
+        const params = new URLSearchParams();
+        if (asOf !== undefined) {
+            params.append("as_of", typeof asOf === "string" ? asOf : asOf.toISOString());
+        }
+        if (tag !== undefined) {
+            params.append("tag", tag);
+        }
+        const response = await this.caller.call((0, fetch_js_1._getFetchImplementation)(), `${this.apiUrl}/datasets/${resolvedDatasetId}/version?${params.toString()}`, {
+            method: "GET",
+            headers: { ...this.headers },
+            signal: AbortSignal.timeout(this.timeout_ms),
+            ...this.fetchOptions,
+        });
+        await (0, error_js_1.raiseForStatus)(response, "read dataset version");
+        return await response.json();
+    }
     async listDatasetSplits({ datasetId, datasetName, asOf, }) {
         let datasetId_;
         if (datasetId === undefined && datasetName === undefined) {

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 } from "./schemas.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 { EvaluationResult, EvaluationResults, RunEvaluator } from "./evaluation/evaluator.js";
 export interface ClientConfig {
     apiUrl?: string;
@@ -487,6 +487,27 @@ export declare class Client implements LangSmithTracingClientInterface {
         name?: string;
         description?: string;
     }): Promise<Dataset>;
+    /**
+     * Updates a tag on a dataset.
+     *
+     * If the tag is already assigned to a different version of this dataset,
+     * the tag will be moved to the new version. The as_of parameter is used to
+     * determine which version of the dataset to apply the new tags to.
+     *
+     * It must be an exact version of the dataset to succeed. You can
+     * use the "readDatasetVersion" method to find the exact version
+     * to apply the tags to.
+     * @param params.datasetId The ID of the dataset to update. Must be provided if "datasetName" is not provided.
+     * @param params.datasetName The name of the dataset to update. Must be provided if "datasetId" is not provided.
+     * @param params.asOf The timestamp of the dataset to apply the new tags to.
+     * @param params.tag The new tag to apply to the dataset.
+     */
+    updateDatasetTag(props: {
+        datasetId?: string;
+        datasetName?: string;
+        asOf: string | Date;
+        tag: string;
+    }): Promise<void>;
     deleteDataset({ datasetId, datasetName, }: {
         datasetId?: string;
         datasetName?: string;
@@ -559,6 +580,24 @@ export declare class Client implements LangSmithTracingClientInterface {
     deleteExample(exampleId: string): Promise<void>;
     updateExample(exampleId: string, update: ExampleUpdate): Promise<object>;
     updateExamples(update: ExampleUpdateWithId[]): Promise<object>;
+    /**
+     * Get dataset version by closest date or exact tag.
+     *
+     * Use this to resolve the nearest version to a given timestamp or for a given tag.
+     *
+     * @param options The options for getting the dataset version
+     * @param options.datasetId The ID of the dataset
+     * @param options.datasetName The name of the dataset
+     * @param options.asOf The timestamp of the dataset to retrieve
+     * @param options.tag The tag of the dataset to retrieve
+     * @returns The dataset version
+     */
+    readDatasetVersion({ datasetId, datasetName, asOf, tag, }: {
+        datasetId?: string;
+        datasetName?: string;
+        asOf?: string | Date;
+        tag?: string;
+    }): Promise<DatasetVersion>;
     listDatasetSplits({ datasetId, datasetName, asOf, }: {
         datasetId?: string;
         datasetName?: string;

package/dist/client.js

@@ -1756,6 +1756,40 @@ export class Client {
         await raiseForStatus(response, "update dataset");
         return (await response.json());
     }
+    /**
+     * Updates a tag on a dataset.
+     *
+     * If the tag is already assigned to a different version of this dataset,
+     * the tag will be moved to the new version. The as_of parameter is used to
+     * determine which version of the dataset to apply the new tags to.
+     *
+     * It must be an exact version of the dataset to succeed. You can
+     * use the "readDatasetVersion" method to find the exact version
+     * to apply the tags to.
+     * @param params.datasetId The ID of the dataset to update. Must be provided if "datasetName" is not provided.
+     * @param params.datasetName The name of the dataset to update. Must be provided if "datasetId" is not provided.
+     * @param params.asOf The timestamp of the dataset to apply the new tags to.
+     * @param params.tag The new tag to apply to the dataset.
+     */
+    async updateDatasetTag(props) {
+        const { datasetId, datasetName, asOf, tag } = props;
+        if (!datasetId && !datasetName) {
+            throw new Error("Must provide either datasetName or datasetId");
+        }
+        const _datasetId = datasetId ?? (await this.readDataset({ datasetName })).id;
+        assertUuid(_datasetId);
+        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/datasets/${_datasetId}/tags`, {
+            method: "PUT",
+            headers: { ...this.headers, "Content-Type": "application/json" },
+            body: JSON.stringify({
+                as_of: typeof asOf === "string" ? asOf : asOf.toISOString(),
+                tag,
+            }),
+            signal: AbortSignal.timeout(this.timeout_ms),
+            ...this.fetchOptions,
+        });
+        await raiseForStatus(response, "update dataset tags");
+    }
     async deleteDataset({ datasetId, datasetName, }) {
         let path = "/datasets";
         let datasetId_ = datasetId;
@@ -2071,6 +2105,47 @@ export class Client {
         const result = await response.json();
         return result;
     }
+    /**
+     * Get dataset version by closest date or exact tag.
+     *
+     * Use this to resolve the nearest version to a given timestamp or for a given tag.
+     *
+     * @param options The options for getting the dataset version
+     * @param options.datasetId The ID of the dataset
+     * @param options.datasetName The name of the dataset
+     * @param options.asOf The timestamp of the dataset to retrieve
+     * @param options.tag The tag of the dataset to retrieve
+     * @returns The dataset version
+     */
+    async readDatasetVersion({ datasetId, datasetName, asOf, tag, }) {
+        let resolvedDatasetId;
+        if (!datasetId) {
+            const dataset = await this.readDataset({ datasetName });
+            resolvedDatasetId = dataset.id;
+        }
+        else {
+            resolvedDatasetId = datasetId;
+        }
+        assertUuid(resolvedDatasetId);
+        if ((asOf && tag) || (!asOf && !tag)) {
+            throw new Error("Exactly one of asOf and tag must be specified.");
+        }
+        const params = new URLSearchParams();
+        if (asOf !== undefined) {
+            params.append("as_of", typeof asOf === "string" ? asOf : asOf.toISOString());
+        }
+        if (tag !== undefined) {
+            params.append("tag", tag);
+        }
+        const response = await this.caller.call(_getFetchImplementation(), `${this.apiUrl}/datasets/${resolvedDatasetId}/version?${params.toString()}`, {
+            method: "GET",
+            headers: { ...this.headers },
+            signal: AbortSignal.timeout(this.timeout_ms),
+            ...this.fetchOptions,
+        });
+        await raiseForStatus(response, "read dataset version");
+        return await response.json();
+    }
     async listDatasetSplits({ datasetId, datasetName, asOf, }) {
         let datasetId_;
         if (datasetId === undefined && datasetName === undefined) {

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.2.16-rc.9";
+exports.__version__ = "0.3.0";

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.2.16-rc.9";
+export declare const __version__ = "0.3.0";

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.2.16-rc.9";
+export const __version__ = "0.3.0";

package/dist/jest/index.cjs

@@ -16,15 +16,17 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.logOutput = exports.logFeedback = exports.toBeSemanticCloseTo = exports.toBeAbsoluteCloseTo = exports.toBeRelativeCloseTo = exports.expect = exports.describe = exports.it = exports.test = void 0;
+exports.wrapEvaluator = exports.logOutputs = exports.logFeedback = exports.toBeSemanticCloseTo = exports.toBeAbsoluteCloseTo = exports.toBeRelativeCloseTo = exports.expect = exports.describe = exports.it = exports.test = void 0;
 const globals_1 = require("@jest/globals");
 const matchers_js_1 = require("../utils/jestlike/matchers.cjs");
 Object.defineProperty(exports, "toBeRelativeCloseTo", { enumerable: true, get: function () { return matchers_js_1.toBeRelativeCloseTo; } });
 Object.defineProperty(exports, "toBeAbsoluteCloseTo", { enumerable: true, get: function () { return matchers_js_1.toBeAbsoluteCloseTo; } });
 Object.defineProperty(exports, "toBeSemanticCloseTo", { enumerable: true, get: function () { return matchers_js_1.toBeSemanticCloseTo; } });
+const evaluatedBy_js_1 = require("../utils/jestlike/vendor/evaluatedBy.cjs");
+Object.defineProperty(exports, "wrapEvaluator", { enumerable: true, get: function () { return evaluatedBy_js_1.wrapEvaluator; } });
 const index_js_1 = require("../utils/jestlike/index.cjs");
 Object.defineProperty(exports, "logFeedback", { enumerable: true, get: function () { return index_js_1.logFeedback; } });
-Object.defineProperty(exports, "logOutput", { enumerable: true, get: function () { return index_js_1.logOutput; } });
+Object.defineProperty(exports, "logOutputs", { enumerable: true, get: function () { return index_js_1.logOutputs; } });
 const index_js_2 = require("../utils/jestlike/index.cjs");
 globals_1.expect.extend({
     toBeRelativeCloseTo: matchers_js_1.toBeRelativeCloseTo,