langchain 1.0.0-alpha.8 → 1.0.0-alpha.9

package/dist/node_modules/.pnpm/{langsmith@0.3.64_@opentelemetry_api@1.9.0_openai@5.11.0_ws@8.18.3_bufferutil@4.0.9_utf-8-validate@6.0.5__zod@3.25.76_ → langsmith@0.3.74_@opentelemetry_api@1.9.0_openai@5.12.2_ws@8.18.3_bufferutil@4.0.9_utf-8-validate@6.0.5__zod@3.25.76_}/node_modules/langsmith/dist/client.js

@@ -1,4 +1,4 @@
-import { getEnv, getEnvironmentVariable, getLangChainEnvVarsMetadata, getLangSmithEnvironmentVariable, getOtelEnabled, getRuntimeEnvironment } from "./utils/env.js";
+import { getEnv, getEnvironmentVariable, getLangSmithEnvVarsMetadata, getLangSmithEnvironmentVariable, getOtelEnabled, getRuntimeEnvironment } from "./utils/env.js";
 import { getDefaultOTLPTracerComponents, getOTELContext, getOTELTrace } from "./singletons/otel.js";
 import { LangSmithToOTELTranslator } from "./experimental/otel/translator.js";
 import { AsyncCaller } from "./utils/async_caller.js";
@@ -12,10 +12,10 @@ import { serialize } from "./utils/fast-safe-stringify/index.js";
 import { __version__ } from "./index.js";
 import * as uuid from "uuid";
 
-//#region ../../node_modules/.pnpm/langsmith@0.3.64_@opentelemetry+api@1.9.0_openai@5.11.0_ws@8.18.3_bufferutil@4.0.9_utf-8-validate@6.0.5__zod@3.25.76_/node_modules/langsmith/dist/client.js
-function mergeRuntimeEnvIntoRun(run) {
+//#region ../../node_modules/.pnpm/langsmith@0.3.74_@opentelemetry+api@1.9.0_openai@5.12.2_ws@8.18.3_bufferutil@4.0.9_utf-8-validate@6.0.5__zod@3.25.76_/node_modules/langsmith/dist/client.js
+function mergeRuntimeEnvIntoRun(run, cachedEnvVars) {
 	const runtimeEnv = getRuntimeEnvironment();
-	const envVars = getLangChainEnvVarsMetadata();
+	const envVars = cachedEnvVars ?? getLangSmithEnvVarsMetadata();
 	const extra = run.extra ?? {};
 	const metadata = extra.metadata;
 	run.extra = {
@@ -104,11 +104,11 @@ var AutoBatchQueue = class {
 		this.sizeBytes += size;
 		return itemPromise;
 	}
-	pop(upToSizeBytes) {
+	pop({ upToSizeBytes, upToSize }) {
 		if (upToSizeBytes < 1) throw new Error("Number of bytes to pop off may not be less than 1.");
 		const popped = [];
 		let poppedSizeBytes = 0;
-		while (poppedSizeBytes + (this.peek()?.size ?? 0) < upToSizeBytes && this.items.length > 0) {
+		while (poppedSizeBytes + (this.peek()?.size ?? 0) < upToSizeBytes && this.items.length > 0 && popped.length < upToSize) {
 			const item = this.items.shift();
 			if (item) {
 				popped.push(item);
@@ -133,6 +133,8 @@ var AutoBatchQueue = class {
 };
 const DEFAULT_UNCOMPRESSED_BATCH_SIZE_LIMIT_BYTES = 24 * 1024 * 1024;
 const SERVER_INFO_REQUEST_TIMEOUT_MS = 1e4;
+/** Maximum number of operations to batch in a single request. */
+const DEFAULT_BATCH_SIZE_LIMIT = 100;
 const DEFAULT_API_URL = "https://api.smith.langchain.com";
 var Client = class Client {
 	get _fetch() {
@@ -157,6 +159,12 @@ var Client = class Client {
 			writable: true,
 			value: void 0
 		});
+		Object.defineProperty(this, "workspaceId", {
+			enumerable: true,
+			configurable: true,
+			writable: true,
+			value: void 0
+		});
 		Object.defineProperty(this, "caller", {
 			enumerable: true,
 			configurable: true,
@@ -235,6 +243,12 @@ var Client = class Client {
 			writable: true,
 			value: void 0
 		});
+		Object.defineProperty(this, "batchSizeLimit", {
+			enumerable: true,
+			configurable: true,
+			writable: true,
+			value: void 0
+		});
 		Object.defineProperty(this, "fetchOptions", {
 			enumerable: true,
 			configurable: true,
@@ -289,6 +303,12 @@ var Client = class Client {
 			writable: true,
 			value: void 0
 		});
+		Object.defineProperty(this, "cachedLSEnvVarsForMetadata", {
+			enumerable: true,
+			configurable: true,
+			writable: true,
+			value: void 0
+		});
 		Object.defineProperty(this, "multipartStreamingDisabled", {
 			enumerable: true,
 			configurable: true,
@@ -308,6 +328,7 @@ var Client = class Client {
 		this.apiKey = trimQuotes(config.apiKey ?? defaultConfig.apiKey);
 		this.webUrl = trimQuotes(config.webUrl ?? defaultConfig.webUrl);
 		if (this.webUrl?.endsWith("/")) this.webUrl = this.webUrl.slice(0, -1);
+		this.workspaceId = trimQuotes(config.workspaceId ?? getLangSmithEnvironmentVariable("WORKSPACE_ID"));
 		this.timeout_ms = config.timeout_ms ?? 9e4;
 		this.caller = new AsyncCaller({
 			...config.callerOptions ?? {},
@@ -330,9 +351,11 @@ var Client = class Client {
 		this.autoBatchTracing = config.autoBatchTracing ?? this.autoBatchTracing;
 		this.blockOnRootRunFinalization = config.blockOnRootRunFinalization ?? this.blockOnRootRunFinalization;
 		this.batchSizeBytesLimit = config.batchSizeBytesLimit;
+		this.batchSizeLimit = config.batchSizeLimit;
 		this.fetchOptions = config.fetchOptions || {};
 		this.manualFlushMode = config.manualFlushMode ?? this.manualFlushMode;
 		if (getOtelEnabled()) this.langSmithToOTELTranslator = new LangSmithToOTELTranslator();
+		this.cachedLSEnvVarsForMetadata = getLangSmithEnvVarsMetadata();
 	}
 	static getDefaultClientConfig() {
 		const apiKey = getLangSmithEnvironmentVariable("API_KEY");
@@ -375,6 +398,7 @@ var Client = class Client {
 	get headers() {
 		const headers = { "User-Agent": `langsmith-js/${__version__}` };
 		if (this.apiKey) headers["x-api-key"] = `${this.apiKey}`;
+		if (this.workspaceId) headers["x-tenant-id"] = this.workspaceId;
 		return headers;
 	}
 	_getPlatformEndpointPath(path) {
@@ -409,7 +433,7 @@ var Client = class Client {
 				signal: AbortSignal.timeout(this.timeout_ms),
 				...this.fetchOptions
 			});
-			await raiseForStatus(res, `Failed to fetch ${path}`);
+			await raiseForStatus(res, `fetch ${path}`);
 			return res;
 		});
 		return response;
@@ -432,7 +456,7 @@ var Client = class Client {
 					signal: AbortSignal.timeout(this.timeout_ms),
 					...this.fetchOptions
 				});
-				await raiseForStatus(res, `Failed to fetch ${path}`);
+				await raiseForStatus(res, `fetch ${path}`);
 				return res;
 			});
 			const items = transform ? transform(await response.json()) : await response.json();
@@ -457,7 +481,7 @@ var Client = class Client {
 					...this.fetchOptions,
 					body: body$1
 				});
-				await raiseForStatus(res, `Failed to fetch ${path}`);
+				await raiseForStatus(res, `fetch ${path}`);
 				return res;
 			});
 			const responseBody = await response.json();
@@ -497,14 +521,24 @@ var Client = class Client {
 		const serverInfo = await this._ensureServerInfo();
 		return this.batchSizeBytesLimit ?? serverInfo.batch_ingest_config?.size_limit_bytes ?? DEFAULT_UNCOMPRESSED_BATCH_SIZE_LIMIT_BYTES;
 	}
+	/**
+	* Get the maximum number of operations to batch in a single request.
+	*/
+	async _getBatchSizeLimit() {
+		const serverInfo = await this._ensureServerInfo();
+		return this.batchSizeLimit ?? serverInfo.batch_ingest_config?.size_limit ?? DEFAULT_BATCH_SIZE_LIMIT;
+	}
 	async _getDatasetExamplesMultiPartSupport() {
 		const serverInfo = await this._ensureServerInfo();
 		return serverInfo.instance_flags?.dataset_examples_multipart_enabled ?? false;
 	}
-	drainAutoBatchQueue(batchSizeLimit) {
+	drainAutoBatchQueue({ batchSizeLimitBytes, batchSizeLimit }) {
 		const promises = [];
 		while (this.autoBatchQueue.items.length > 0) {
-			const [batch, done] = this.autoBatchQueue.pop(batchSizeLimit);
+			const [batch, done] = this.autoBatchQueue.pop({
+				upToSizeBytes: batchSizeLimitBytes,
+				upToSize: batchSizeLimit
+			});
 			if (!batch.length) {
 				done();
 				break;
@@ -578,14 +612,21 @@ var Client = class Client {
 	async processRunOperation(item) {
 		clearTimeout(this.autoBatchTimeout);
 		this.autoBatchTimeout = void 0;
-		item.item = mergeRuntimeEnvIntoRun(item.item);
+		item.item = mergeRuntimeEnvIntoRun(item.item, this.cachedLSEnvVarsForMetadata);
 		const itemPromise = this.autoBatchQueue.push(item);
 		if (this.manualFlushMode) return itemPromise;
 		const sizeLimitBytes = await this._getBatchSizeLimitBytes();
-		if (this.autoBatchQueue.sizeBytes > sizeLimitBytes) this.drainAutoBatchQueue(sizeLimitBytes);
+		const sizeLimit = await this._getBatchSizeLimit();
+		if (this.autoBatchQueue.sizeBytes > sizeLimitBytes || this.autoBatchQueue.items.length > sizeLimit) this.drainAutoBatchQueue({
+			batchSizeLimitBytes: sizeLimitBytes,
+			batchSizeLimit: sizeLimit
+		});
 		if (this.autoBatchQueue.items.length > 0) this.autoBatchTimeout = setTimeout(() => {
 			this.autoBatchTimeout = void 0;
-			this.drainAutoBatchQueue(sizeLimitBytes);
+			this.drainAutoBatchQueue({
+				batchSizeLimitBytes: sizeLimitBytes,
+				batchSizeLimit: sizeLimit
+			});
 		}, this.autoBatchAggregationDelayMs);
 		return itemPromise;
 	}
@@ -627,7 +668,11 @@ var Client = class Client {
 	*/
 	async flush() {
 		const sizeLimitBytes = await this._getBatchSizeLimitBytes();
-		await this.drainAutoBatchQueue(sizeLimitBytes);
+		const sizeLimit = await this._getBatchSizeLimit();
+		await this.drainAutoBatchQueue({
+			batchSizeLimitBytes: sizeLimitBytes,
+			batchSizeLimit: sizeLimit
+		});
 	}
 	_cloneCurrentOTELContext() {
 		const otel_trace = getOTELTrace();
@@ -662,8 +707,9 @@ var Client = class Client {
 			}).catch(console.error);
 			return;
 		}
-		const mergedRunCreateParam = mergeRuntimeEnvIntoRun(runCreate);
+		const mergedRunCreateParam = mergeRuntimeEnvIntoRun(runCreate, this.cachedLSEnvVarsForMetadata);
 		if (options?.apiKey !== void 0) headers["x-api-key"] = options.apiKey;
+		if (options?.workspaceId !== void 0) headers["x-tenant-id"] = options.workspaceId;
 		const body = serialize(mergedRunCreateParam, `Creating run with id: ${mergedRunCreateParam.id}`);
 		await this.caller.call(async () => {
 			const res = await this._fetch(`${options?.apiUrl ?? this.apiUrl}/runs`, {
@@ -953,6 +999,7 @@ var Client = class Client {
 			"Content-Type": "application/json"
 		};
 		if (options?.apiKey !== void 0) headers["x-api-key"] = options.apiKey;
+		if (options?.workspaceId !== void 0) headers["x-tenant-id"] = options.workspaceId;
 		const body = serialize(run, `Serializing payload to update run with id: ${runId}`);
 		await this.caller.call(async () => {
 			const res = await this._fetch(`${options?.apiUrl ?? this.apiUrl}/runs/${runId}`, {
@@ -1151,6 +1198,7 @@ var Client = class Client {
 			is_root: isRoot,
 			order
 		};
+		if (body.select.includes("child_run_ids")) warnOnce("Deprecated: 'child_run_ids' in the listRuns select parameter is deprecated and will be removed in a future version.");
 		let runsYielded = 0;
 		for await (const runs of this._getCursorPaginatedList("/runs/query", body)) if (limit) {
 			if (runsYielded >= limit) break;
@@ -1231,7 +1279,10 @@ var Client = class Client {
 		const response = await this.caller.call(async () => {
 			const res = await this._fetch(`${this.apiUrl}/runs/stats`, {
 				method: "POST",
-				headers: this.headers,
+				headers: {
+					...this.headers,
+					"Content-Type": "application/json"
+				},
 				signal: AbortSignal.timeout(this.timeout_ms),
 				...this.fetchOptions,
 				body
@@ -1551,7 +1602,7 @@ var Client = class Client {
 		}
 		throw new Error("No projects found to resolve tenant.");
 	}
-	async *listProjects({ projectIds, name, nameContains, referenceDatasetId, referenceDatasetName, referenceFree, metadata } = {}) {
+	async *listProjects({ projectIds, name, nameContains, referenceDatasetId, referenceDatasetName, includeStats, datasetVersion, referenceFree, metadata } = {}) {
 		const params = new URLSearchParams();
 		if (projectIds !== void 0) for (const projectId of projectIds) params.append("id", projectId);
 		if (name !== void 0) params.append("name", name);
@@ -1561,6 +1612,8 @@ var Client = class Client {
 			const dataset = await this.readDataset({ datasetName: referenceDatasetName });
 			params.append("reference_dataset", dataset.id);
 		}
+		if (includeStats !== void 0) params.append("include_stats", includeStats.toString());
+		if (datasetVersion !== void 0) params.append("dataset_version", datasetVersion);
 		if (referenceFree !== void 0) params.append("reference_free", referenceFree.toString());
 		if (metadata !== void 0) params.append("metadata", JSON.stringify(metadata));
 		for await (const projects of this._getPaginated("/sessions", params)) yield* projects;
@@ -2242,7 +2295,10 @@ var Client = class Client {
 	}
 	async *listFeedback({ runIds, feedbackKeys, feedbackSourceTypes } = {}) {
 		const queryParams = new URLSearchParams();
-		if (runIds) queryParams.append("run", runIds.join(","));
+		if (runIds) for (const runId of runIds) {
+			assertUuid(runId);
+			queryParams.append("run", runId);
+		}
 		if (feedbackKeys) for (const key of feedbackKeys) queryParams.append("key", key);
 		if (feedbackSourceTypes) for (const type of feedbackSourceTypes) queryParams.append("source", type);
 		for await (const feedbacks of this._getPaginated("/feedback", queryParams)) yield* feedbacks;