langsmith 0.3.49-rc.2 → 0.3.49-rc.4

package/dist/experimental/otel/processor.cjs

@@ -5,6 +5,7 @@ exports.isTraceableSpan = isTraceableSpan;
 const sdk_trace_base_1 = require("@opentelemetry/sdk-trace-base");
 const constants_js_1 = require("./constants.cjs");
 const utils_js_1 = require("./utils.cjs");
+const run_trees_js_1 = require("../../run_trees.cjs");
 function isTraceableSpan(span) {
     return (span.attributes[constants_js_1.LANGSMITH_TRACEABLE] === "true" ||
         typeof span.attributes["ai.operationId"] === "string");
@@ -27,45 +28,15 @@ class LangSmithOTLPSpanProcessor extends sdk_trace_base_1.BatchSpanProcessor {
             writable: true,
             value: {}
         });
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        Object.defineProperty(this, "cleanupInterval", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "TRACE_TTL_MS", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 10 * 60 * 1000
-        }); // 10 minutes
-        // We must use a cleanup interval because LangSmith can start child spans
-        // after arbitrary OTEL parent spans have ended since it uses batching.
-        this.cleanupInterval = setInterval(() => this.cleanupStaleTraces(), 60000);
-    }
-    cleanupStaleTraces() {
-        const now = Date.now();
-        for (const [traceId, traceInfo] of Object.entries(this.traceMap)) {
-            if (now - traceInfo.lastAccessed > this.TRACE_TTL_MS) {
-                delete this.traceMap[traceId];
-            }
-        }
-    }
-    shutdown() {
-        if (this.cleanupInterval) {
-            clearInterval(this.cleanupInterval);
-        }
-        return super.shutdown();
     }
     onStart(span, parentContext) {
         if (!this.traceMap[span.spanContext().traceId]) {
             this.traceMap[span.spanContext().traceId] = {
                 spanInfo: {},
-                lastAccessed: Date.now(),
+                spanCount: 0,
             };
         }
-        this.traceMap[span.spanContext().traceId].lastAccessed = Date.now();
+        this.traceMap[span.spanContext().traceId].spanCount++;
         const isTraceable = isTraceableSpan(span);
         const parentSpanId = getParentSpanId(span);
         this.traceMap[span.spanContext().traceId].spanInfo[span.spanContext().spanId] = {
@@ -97,13 +68,21 @@ class LangSmithOTLPSpanProcessor extends sdk_trace_base_1.BatchSpanProcessor {
         const traceInfo = this.traceMap[span.spanContext().traceId];
         if (!traceInfo)
             return;
-        traceInfo.lastAccessed = Date.now();
         const spanInfo = traceInfo.spanInfo[span.spanContext().spanId];
         if (!spanInfo)
             return;
+        // Decrement span count and cleanup trace if all spans are done
+        traceInfo.spanCount--;
+        if (traceInfo.spanCount <= 0) {
+            delete this.traceMap[span.spanContext().traceId];
+        }
         if (spanInfo.isTraceable) {
             super.onEnd(span);
         }
     }
+    async shutdown() {
+        await run_trees_js_1.RunTree.getSharedClient().awaitPendingTraceBatches();
+        await super.shutdown();
+    }
 }
 exports.LangSmithOTLPSpanProcessor = LangSmithOTLPSpanProcessor;

package/dist/experimental/otel/processor.d.ts

@@ -7,11 +7,8 @@ export declare function isTraceableSpan(span: ReadableSpan): boolean;
  */
 export declare class LangSmithOTLPSpanProcessor extends BatchSpanProcessor {
     private traceMap;
-    private cleanupInterval;
-    private TRACE_TTL_MS;
     constructor(...args: ConstructorParameters<typeof BatchSpanProcessor>);
-    private cleanupStaleTraces;
-    shutdown(): Promise<void>;
     onStart(span: Span, parentContext: Context): void;
     onEnd(span: ReadableSpan): void;
+    shutdown(): Promise<void>;
 }

package/dist/experimental/otel/processor.js

@@ -1,6 +1,7 @@
 import { BatchSpanProcessor, } from "@opentelemetry/sdk-trace-base";
 import { LANGSMITH_IS_ROOT, LANGSMITH_PARENT_RUN_ID, LANGSMITH_TRACEABLE, } from "./constants.js";
 import { getUuidFromOtelSpanId } from "./utils.js";
+import { RunTree } from "../../run_trees.js";
 export function isTraceableSpan(span) {
     return (span.attributes[LANGSMITH_TRACEABLE] === "true" ||
         typeof span.attributes["ai.operationId"] === "string");
@@ -23,45 +24,15 @@ export class LangSmithOTLPSpanProcessor extends BatchSpanProcessor {
             writable: true,
             value: {}
         });
-        // eslint-disable-next-line @typescript-eslint/no-explicit-any
-        Object.defineProperty(this, "cleanupInterval", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: void 0
-        });
-        Object.defineProperty(this, "TRACE_TTL_MS", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: 10 * 60 * 1000
-        }); // 10 minutes
-        // We must use a cleanup interval because LangSmith can start child spans
-        // after arbitrary OTEL parent spans have ended since it uses batching.
-        this.cleanupInterval = setInterval(() => this.cleanupStaleTraces(), 60000);
-    }
-    cleanupStaleTraces() {
-        const now = Date.now();
-        for (const [traceId, traceInfo] of Object.entries(this.traceMap)) {
-            if (now - traceInfo.lastAccessed > this.TRACE_TTL_MS) {
-                delete this.traceMap[traceId];
-            }
-        }
-    }
-    shutdown() {
-        if (this.cleanupInterval) {
-            clearInterval(this.cleanupInterval);
-        }
-        return super.shutdown();
     }
     onStart(span, parentContext) {
         if (!this.traceMap[span.spanContext().traceId]) {
             this.traceMap[span.spanContext().traceId] = {
                 spanInfo: {},
-                lastAccessed: Date.now(),
+                spanCount: 0,
             };
         }
-        this.traceMap[span.spanContext().traceId].lastAccessed = Date.now();
+        this.traceMap[span.spanContext().traceId].spanCount++;
         const isTraceable = isTraceableSpan(span);
         const parentSpanId = getParentSpanId(span);
         this.traceMap[span.spanContext().traceId].spanInfo[span.spanContext().spanId] = {
@@ -93,12 +64,20 @@ export class LangSmithOTLPSpanProcessor extends BatchSpanProcessor {
         const traceInfo = this.traceMap[span.spanContext().traceId];
         if (!traceInfo)
             return;
-        traceInfo.lastAccessed = Date.now();
         const spanInfo = traceInfo.spanInfo[span.spanContext().spanId];
         if (!spanInfo)
             return;
+        // Decrement span count and cleanup trace if all spans are done
+        traceInfo.spanCount--;
+        if (traceInfo.spanCount <= 0) {
+            delete this.traceMap[span.spanContext().traceId];
+        }
         if (spanInfo.isTraceable) {
             super.onEnd(span);
         }
     }
+    async shutdown() {
+        await RunTree.getSharedClient().awaitPendingTraceBatches();
+        await super.shutdown();
+    }
 }

package/dist/index.cjs

@@ -10,4 +10,4 @@ Object.defineProperty(exports, "overrideFetchImplementation", { enumerable: true
 var project_js_1 = require("./utils/project.cjs");
 Object.defineProperty(exports, "getDefaultProjectName", { enumerable: true, get: function () { return project_js_1.getDefaultProjectName; } });
 // Update using yarn bump-version
-exports.__version__ = "0.3.49-rc.2";
+exports.__version__ = "0.3.49-rc.4";

package/dist/index.d.ts

@@ -3,4 +3,4 @@ export type { Dataset, Example, TracerSession, Run, Feedback, RetrieverOutput, }
 export { RunTree, type RunTreeConfig } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
-export declare const __version__ = "0.3.49-rc.2";
+export declare const __version__ = "0.3.49-rc.4";

package/dist/index.js

@@ -3,4 +3,4 @@ export { RunTree } from "./run_trees.js";
 export { overrideFetchImplementation } from "./singletons/fetch.js";
 export { getDefaultProjectName } from "./utils/project.js";
 // Update using yarn bump-version
-export const __version__ = "0.3.49-rc.2";
+export const __version__ = "0.3.49-rc.4";

package/dist/run_trees.d.ts

@@ -101,7 +101,7 @@ export declare class RunTree implements BaseRun {
     set metadata(metadata: KVMap);
     get metadata(): KVMap;
     private static getDefaultConfig;
-    private static getSharedClient;
+    static getSharedClient(): Client;
     createChild(config: RunTreeConfig): RunTree;
     end(outputs?: KVMap, error?: string, endTime?: number, metadata?: KVMap): Promise<void>;
     private _convertToCreate;

package/package.json

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