@nocobase/plugin-ai 1.9.32 → 1.9.34

package/dist/node_modules/@langchain/core/dist/tracers/base.cjs

@@ -1,37 +1,86 @@
 "use strict";
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.BaseTracer = exports.isBaseTracer = void 0;
+exports.BaseTracer = void 0;
+exports.isBaseTracer = isBaseTracer;
+const run_trees_1 = require("langsmith/run_trees");
 const base_js_1 = require("../callbacks/base.cjs");
+const env_js_1 = require("../utils/env.cjs");
+// TODO: Remove and just use base LangSmith Run type
+const convertRunTreeToRun = (runTree) => {
+    if (!runTree) {
+        return undefined;
+    }
+    // Important that we return the raw run tree object since the reference
+    // is mutated in other places.
+    // TODO: Remove places where this is being done.
+    // eslint-disable-next-line no-param-reassign
+    runTree.events = runTree.events ?? [];
+    // eslint-disable-next-line no-param-reassign
+    runTree.child_runs = runTree.child_runs ?? [];
+    // TODO: Remove this cast and just use the LangSmith RunTree type.
+    return runTree;
+};
+function convertRunToRunTree(run, parentRun) {
+    if (!run) {
+        return undefined;
+    }
+    return new run_trees_1.RunTree({
+        ...run,
+        start_time: run._serialized_start_time ?? run.start_time,
+        parent_run: convertRunToRunTree(parentRun),
+        child_runs: run.child_runs
+            .map((r) => convertRunToRunTree(r))
+            .filter((r) => r !== undefined),
+        extra: {
+            ...run.extra,
+            runtime: (0, env_js_1.getRuntimeEnvironmentSync)(),
+        },
+        tracingEnabled: false,
+    });
+}
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 function _coerceToDict(value, defaultKey) {
     return value && !Array.isArray(value) && typeof value === "object"
         ? value
         : { [defaultKey]: value };
 }
-function stripNonAlphanumeric(input) {
-    return input.replace(/[-:.]/g, "");
-}
-function convertToDottedOrderFormat(epoch, runId, executionOrder) {
-    const paddedOrder = executionOrder.toFixed(0).slice(0, 3).padStart(3, "0");
-    return (stripNonAlphanumeric(`${new Date(epoch).toISOString().slice(0, -1)}${paddedOrder}Z`) + runId);
-}
 function isBaseTracer(x) {
     return typeof x._addRunToRunMap === "function";
 }
-exports.isBaseTracer = isBaseTracer;
 class BaseTracer extends base_js_1.BaseCallbackHandler {
     constructor(_fields) {
         super(...arguments);
+        /** @deprecated Use `runTreeMap` instead. */
         Object.defineProperty(this, "runMap", {
             enumerable: true,
             configurable: true,
             writable: true,
             value: new Map()
         });
+        Object.defineProperty(this, "runTreeMap", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: new Map()
+        });
+        Object.defineProperty(this, "usesRunTreeMap", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: false
+        });
     }
     copy() {
         return this;
     }
+    getRunById(runId) {
+        if (runId === undefined) {
+            return undefined;
+        }
+        return this.usesRunTreeMap
+            ? convertRunTreeToRun(this.runTreeMap.get(runId))
+            : this.runMap.get(runId);
+    }
     stringifyError(error) {
         // eslint-disable-next-line no-instanceof/no-instanceof
         if (error instanceof Error) {
@@ -46,10 +95,10 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         parentRun.child_runs.push(childRun);
     }
     _addRunToRunMap(run) {
-        const currentDottedOrder = convertToDottedOrderFormat(run.start_time, run.id, run.execution_order);
+        const { dottedOrder: currentDottedOrder, microsecondPrecisionDatestring } = (0, run_trees_1.convertToDottedOrderFormat)(new Date(run.start_time).getTime(), run.id, run.execution_order);
         const storedRun = { ...run };
+        const parentRun = this.getRunById(storedRun.parent_run_id);
         if (storedRun.parent_run_id !== undefined) {
-            const parentRun = this.runMap.get(storedRun.parent_run_id);
             if (parentRun) {
                 this._addChildRun(parentRun, storedRun);
                 parentRun.child_execution_order = Math.max(parentRun.child_execution_order, storedRun.child_execution_order);
@@ -59,6 +108,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
                         parentRun.dotted_order,
                         currentDottedOrder,
                     ].join(".");
+                    storedRun._serialized_start_time = microsecondPrecisionDatestring;
                 }
                 else {
                     // This can happen naturally for callbacks added within a run
@@ -75,23 +125,37 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         else {
             storedRun.trace_id = storedRun.id;
             storedRun.dotted_order = currentDottedOrder;
+            storedRun._serialized_start_time = microsecondPrecisionDatestring;
+        }
+        if (this.usesRunTreeMap) {
+            const runTree = convertRunToRunTree(storedRun, parentRun);
+            if (runTree !== undefined) {
+                this.runTreeMap.set(storedRun.id, runTree);
+            }
+        }
+        else {
+            this.runMap.set(storedRun.id, storedRun);
         }
-        this.runMap.set(storedRun.id, storedRun);
         return storedRun;
     }
     async _endTrace(run) {
-        const parentRun = run.parent_run_id !== undefined && this.runMap.get(run.parent_run_id);
+        const parentRun = run.parent_run_id !== undefined && this.getRunById(run.parent_run_id);
         if (parentRun) {
             parentRun.child_execution_order = Math.max(parentRun.child_execution_order, run.child_execution_order);
         }
         else {
             await this.persistRun(run);
         }
-        this.runMap.delete(run.id);
         await this.onRunUpdate?.(run);
+        if (this.usesRunTreeMap) {
+            this.runTreeMap.delete(run.id);
+        }
+        else {
+            this.runMap.delete(run.id);
+        }
     }
     _getExecutionOrder(parentRunId) {
-        const parentRun = parentRunId !== undefined && this.runMap.get(parentRunId);
+        const parentRun = parentRunId !== undefined && this.getRunById(parentRunId);
         // If a run has no parent then execution order is 1
         if (!parentRun) {
             return 1;
@@ -132,7 +196,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return this._addRunToRunMap(run);
     }
     async handleLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name) {
-        const run = this.runMap.get(runId) ??
+        const run = this.getRunById(runId) ??
             this._createRunForLLMStart(llm, prompts, runId, parentRunId, extraParams, tags, metadata, name);
         await this.onRunCreate?.(run);
         await this.onLLMStart?.(run);
@@ -172,14 +236,14 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return this._addRunToRunMap(run);
     }
     async handleChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name) {
-        const run = this.runMap.get(runId) ??
+        const run = this.getRunById(runId) ??
             this._createRunForChatModelStart(llm, messages, runId, parentRunId, extraParams, tags, metadata, name);
         await this.onRunCreate?.(run);
         await this.onLLMStart?.(run);
         return run;
     }
     async handleLLMEnd(output, runId, _parentRunId, _tags, extraParams) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "llm") {
             throw new Error("No LLM run to end.");
         }
@@ -195,7 +259,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return run;
     }
     async handleLLMError(error, runId, _parentRunId, _tags, extraParams) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "llm") {
             throw new Error("No LLM run to end.");
         }
@@ -241,14 +305,14 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return this._addRunToRunMap(run);
     }
     async handleChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name) {
-        const run = this.runMap.get(runId) ??
+        const run = this.getRunById(runId) ??
             this._createRunForChainStart(chain, inputs, runId, parentRunId, tags, metadata, runType, name);
         await this.onRunCreate?.(run);
         await this.onChainStart?.(run);
         return run;
     }
     async handleChainEnd(outputs, runId, _parentRunId, _tags, kwargs) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run) {
             throw new Error("No chain run to end.");
         }
@@ -266,7 +330,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return run;
     }
     async handleChainError(error, runId, _parentRunId, _tags, kwargs) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run) {
             throw new Error("No chain run to end.");
         }
@@ -314,7 +378,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return this._addRunToRunMap(run);
     }
     async handleToolStart(tool, input, runId, parentRunId, tags, metadata, name) {
-        const run = this.runMap.get(runId) ??
+        const run = this.getRunById(runId) ??
             this._createRunForToolStart(tool, input, runId, parentRunId, tags, metadata, name);
         await this.onRunCreate?.(run);
         await this.onToolStart?.(run);
@@ -322,7 +386,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
     }
     // eslint-disable-next-line @typescript-eslint/no-explicit-any
     async handleToolEnd(output, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "tool") {
             throw new Error("No tool run to end");
         }
@@ -337,7 +401,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return run;
     }
     async handleToolError(error, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "tool") {
             throw new Error("No tool run to end");
         }
@@ -352,7 +416,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return run;
     }
     async handleAgentAction(action, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "chain") {
             return;
         }
@@ -367,7 +431,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         await this.onAgentAction?.(run);
     }
     async handleAgentEnd(action, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "chain") {
             return;
         }
@@ -409,14 +473,14 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return this._addRunToRunMap(run);
     }
     async handleRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name) {
-        const run = this.runMap.get(runId) ??
+        const run = this.getRunById(runId) ??
             this._createRunForRetrieverStart(retriever, query, runId, parentRunId, tags, metadata, name);
         await this.onRunCreate?.(run);
         await this.onRetrieverStart?.(run);
         return run;
     }
     async handleRetrieverEnd(documents, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "retriever") {
             throw new Error("No retriever run to end");
         }
@@ -431,7 +495,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return run;
     }
     async handleRetrieverError(error, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "retriever") {
             throw new Error("No retriever run to end");
         }
@@ -446,7 +510,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         return run;
     }
     async handleText(text, runId) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "chain") {
             return;
         }
@@ -458,7 +522,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
         await this.onText?.(run);
     }
     async handleLLMNewToken(token, idx, runId, _parentRunId, _tags, fields) {
-        const run = this.runMap.get(runId);
+        const run = this.getRunById(runId);
         if (!run || run?.run_type !== "llm") {
             throw new Error(`Invalid "runId" provided to "handleLLMNewToken" callback.`);
         }

package/dist/node_modules/@langchain/core/dist/tracers/base.d.ts

@@ -1,4 +1,5 @@
 import { KVMap, BaseRun } from "langsmith/schemas";
+import { RunTree } from "langsmith/run_trees";
 import type { ChainValues } from "../utils/types/index.js";
 import type { AgentAction, AgentFinish } from "../agents.js";
 import type { LLMResult } from "../outputs.js";
@@ -10,6 +11,7 @@ export type RunType = string;
 export interface Run extends BaseRun {
     id: string;
     start_time: number;
+    end_time?: number;
     execution_order: number;
     child_runs: this[];
     child_execution_order: number;
@@ -20,43 +22,51 @@ export interface Run extends BaseRun {
     }>;
     trace_id?: string;
     dotted_order?: string;
+    /** @internal */
+    _serialized_start_time?: string;
 }
 export interface AgentRun extends Run {
     actions: AgentAction[];
 }
 export declare function isBaseTracer(x: BaseCallbackHandler): x is BaseTracer;
 export declare abstract class BaseTracer extends BaseCallbackHandler {
+    /** @deprecated Use `runTreeMap` instead. */
     protected runMap: Map<string, Run>;
+    protected runTreeMap: Map<string, RunTree>;
+    protected usesRunTreeMap: boolean;
     constructor(_fields?: BaseCallbackHandlerInput);
     copy(): this;
+    protected getRunById(runId?: string): Run | undefined;
     protected stringifyError(error: unknown): string;
     protected abstract persistRun(run: Run): Promise<void>;
     protected _addChildRun(parentRun: Run, childRun: Run): void;
     _addRunToRunMap(run: Run): {
         id: string;
         start_time: number;
+        end_time?: number;
         execution_order: number;
         child_runs: Run[];
         child_execution_order: number;
-        events: {
+        events: Array<{
             name: string;
             time: string;
-            kwargs?: Record<string, unknown> | undefined;
-        }[];
-        trace_id?: string | undefined;
-        dotted_order?: string | undefined;
+            kwargs?: Record<string, unknown>;
+        }>;
+        trace_id?: string;
+        dotted_order?: string;
+        /** @internal */
+        _serialized_start_time?: string;
         name: string;
         run_type: string;
-        end_time?: number | undefined;
-        extra?: KVMap | undefined;
-        error?: string | undefined;
-        serialized?: object | undefined;
+        extra?: KVMap;
+        error?: string;
+        serialized?: object;
         inputs: KVMap;
-        outputs?: KVMap | undefined;
-        reference_example_id?: string | undefined;
-        parent_run_id?: string | undefined;
-        tags?: string[] | undefined;
-        attachments?: import("langsmith/schemas").Attachments | undefined;
+        outputs?: KVMap;
+        reference_example_id?: string;
+        parent_run_id?: string;
+        tags?: string[];
+        attachments?: import("langsmith/schemas").Attachments;
     };
     protected _endTrace(run: Run): Promise<void>;
     protected _getExecutionOrder(parentRunId: string | undefined): number;
@@ -68,28 +78,30 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     _createRunForLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[], metadata?: KVMap, name?: string): {
         id: string;
         start_time: number;
+        end_time?: number;
         execution_order: number;
         child_runs: Run[];
         child_execution_order: number;
-        events: {
+        events: Array<{
             name: string;
             time: string;
-            kwargs?: Record<string, unknown> | undefined;
-        }[];
-        trace_id?: string | undefined;
-        dotted_order?: string | undefined;
+            kwargs?: Record<string, unknown>;
+        }>;
+        trace_id?: string;
+        dotted_order?: string;
+        /** @internal */
+        _serialized_start_time?: string;
         name: string;
         run_type: string;
-        end_time?: number | undefined;
-        extra?: KVMap | undefined;
-        error?: string | undefined;
-        serialized?: object | undefined;
+        extra?: KVMap;
+        error?: string;
+        serialized?: object;
         inputs: KVMap;
-        outputs?: KVMap | undefined;
-        reference_example_id?: string | undefined;
-        parent_run_id?: string | undefined;
-        tags?: string[] | undefined;
-        attachments?: import("langsmith/schemas").Attachments | undefined;
+        outputs?: KVMap;
+        reference_example_id?: string;
+        parent_run_id?: string;
+        tags?: string[];
+        attachments?: import("langsmith/schemas").Attachments;
     };
     handleLLMStart(llm: Serialized, prompts: string[], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[], metadata?: KVMap, name?: string): Promise<Run>;
     /**
@@ -100,28 +112,30 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     _createRunForChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[], metadata?: KVMap, name?: string): {
         id: string;
         start_time: number;
+        end_time?: number;
         execution_order: number;
         child_runs: Run[];
         child_execution_order: number;
-        events: {
+        events: Array<{
             name: string;
             time: string;
-            kwargs?: Record<string, unknown> | undefined;
-        }[];
-        trace_id?: string | undefined;
-        dotted_order?: string | undefined;
+            kwargs?: Record<string, unknown>;
+        }>;
+        trace_id?: string;
+        dotted_order?: string;
+        /** @internal */
+        _serialized_start_time?: string;
         name: string;
         run_type: string;
-        end_time?: number | undefined;
-        extra?: KVMap | undefined;
-        error?: string | undefined;
-        serialized?: object | undefined;
+        extra?: KVMap;
+        error?: string;
+        serialized?: object;
         inputs: KVMap;
-        outputs?: KVMap | undefined;
-        reference_example_id?: string | undefined;
-        parent_run_id?: string | undefined;
-        tags?: string[] | undefined;
-        attachments?: import("langsmith/schemas").Attachments | undefined;
+        outputs?: KVMap;
+        reference_example_id?: string;
+        parent_run_id?: string;
+        tags?: string[];
+        attachments?: import("langsmith/schemas").Attachments;
     };
     handleChatModelStart(llm: Serialized, messages: BaseMessage[][], runId: string, parentRunId?: string, extraParams?: KVMap, tags?: string[], metadata?: KVMap, name?: string): Promise<Run>;
     handleLLMEnd(output: LLMResult, runId: string, _parentRunId?: string, _tags?: string[], extraParams?: Record<string, unknown>): Promise<Run>;
@@ -134,28 +148,30 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     _createRunForChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, runType?: string, name?: string): {
         id: string;
         start_time: number;
+        end_time?: number;
         execution_order: number;
         child_runs: Run[];
         child_execution_order: number;
-        events: {
+        events: Array<{
             name: string;
             time: string;
-            kwargs?: Record<string, unknown> | undefined;
-        }[];
-        trace_id?: string | undefined;
-        dotted_order?: string | undefined;
+            kwargs?: Record<string, unknown>;
+        }>;
+        trace_id?: string;
+        dotted_order?: string;
+        /** @internal */
+        _serialized_start_time?: string;
         name: string;
         run_type: string;
-        end_time?: number | undefined;
-        extra?: KVMap | undefined;
-        error?: string | undefined;
-        serialized?: object | undefined;
+        extra?: KVMap;
+        error?: string;
+        serialized?: object;
         inputs: KVMap;
-        outputs?: KVMap | undefined;
-        reference_example_id?: string | undefined;
-        parent_run_id?: string | undefined;
-        tags?: string[] | undefined;
-        attachments?: import("langsmith/schemas").Attachments | undefined;
+        outputs?: KVMap;
+        reference_example_id?: string;
+        parent_run_id?: string;
+        tags?: string[];
+        attachments?: import("langsmith/schemas").Attachments;
     };
     handleChainStart(chain: Serialized, inputs: ChainValues, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, runType?: string, name?: string): Promise<Run>;
     handleChainEnd(outputs: ChainValues, runId: string, _parentRunId?: string, _tags?: string[], kwargs?: {
@@ -172,28 +188,30 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     _createRunForToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, name?: string): {
         id: string;
         start_time: number;
+        end_time?: number;
         execution_order: number;
         child_runs: Run[];
         child_execution_order: number;
-        events: {
+        events: Array<{
             name: string;
             time: string;
-            kwargs?: Record<string, unknown> | undefined;
-        }[];
-        trace_id?: string | undefined;
-        dotted_order?: string | undefined;
+            kwargs?: Record<string, unknown>;
+        }>;
+        trace_id?: string;
+        dotted_order?: string;
+        /** @internal */
+        _serialized_start_time?: string;
         name: string;
         run_type: string;
-        end_time?: number | undefined;
-        extra?: KVMap | undefined;
-        error?: string | undefined;
-        serialized?: object | undefined;
+        extra?: KVMap;
+        error?: string;
+        serialized?: object;
         inputs: KVMap;
-        outputs?: KVMap | undefined;
-        reference_example_id?: string | undefined;
-        parent_run_id?: string | undefined;
-        tags?: string[] | undefined;
-        attachments?: import("langsmith/schemas").Attachments | undefined;
+        outputs?: KVMap;
+        reference_example_id?: string;
+        parent_run_id?: string;
+        tags?: string[];
+        attachments?: import("langsmith/schemas").Attachments;
     };
     handleToolStart(tool: Serialized, input: string, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, name?: string): Promise<Run>;
     handleToolEnd(output: any, runId: string): Promise<Run>;
@@ -208,28 +226,30 @@ export declare abstract class BaseTracer extends BaseCallbackHandler {
     _createRunForRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, name?: string): {
         id: string;
         start_time: number;
+        end_time?: number;
         execution_order: number;
         child_runs: Run[];
         child_execution_order: number;
-        events: {
+        events: Array<{
             name: string;
             time: string;
-            kwargs?: Record<string, unknown> | undefined;
-        }[];
-        trace_id?: string | undefined;
-        dotted_order?: string | undefined;
+            kwargs?: Record<string, unknown>;
+        }>;
+        trace_id?: string;
+        dotted_order?: string;
+        /** @internal */
+        _serialized_start_time?: string;
         name: string;
         run_type: string;
-        end_time?: number | undefined;
-        extra?: KVMap | undefined;
-        error?: string | undefined;
-        serialized?: object | undefined;
+        extra?: KVMap;
+        error?: string;
+        serialized?: object;
         inputs: KVMap;
-        outputs?: KVMap | undefined;
-        reference_example_id?: string | undefined;
-        parent_run_id?: string | undefined;
-        tags?: string[] | undefined;
-        attachments?: import("langsmith/schemas").Attachments | undefined;
+        outputs?: KVMap;
+        reference_example_id?: string;
+        parent_run_id?: string;
+        tags?: string[];
+        attachments?: import("langsmith/schemas").Attachments;
     };
     handleRetrieverStart(retriever: Serialized, query: string, runId: string, parentRunId?: string, tags?: string[], metadata?: KVMap, name?: string): Promise<Run>;
     handleRetrieverEnd(documents: Document<Record<string, unknown>>[], runId: string): Promise<Run>;