@langchain/langgraph 1.4.13 → 1.4.15-rc.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/func/index.cjs +1 -1
- package/dist/func/index.js +1 -1
- package/dist/graph/message.cjs +7 -4
- package/dist/graph/message.cjs.map +1 -1
- package/dist/graph/message.d.cts.map +1 -1
- package/dist/graph/message.d.ts.map +1 -1
- package/dist/graph/message.js +7 -4
- package/dist/graph/message.js.map +1 -1
- package/dist/graph/state.cjs +3 -1
- package/dist/graph/state.cjs.map +1 -1
- package/dist/graph/state.d.cts +4 -2
- package/dist/graph/state.d.cts.map +1 -1
- package/dist/graph/state.d.ts +4 -2
- package/dist/graph/state.d.ts.map +1 -1
- package/dist/graph/state.js +3 -1
- package/dist/graph/state.js.map +1 -1
- package/dist/index.cjs +6 -4
- package/dist/index.cjs.map +1 -1
- package/dist/index.d.cts +2 -2
- package/dist/index.d.ts +2 -2
- package/dist/index.js +2 -1
- package/dist/index.js.map +1 -1
- package/dist/pregel/algo.cjs +1 -1
- package/dist/pregel/algo.js +1 -1
- package/dist/pregel/index.cjs +1 -1
- package/dist/pregel/index.js +1 -1
- package/dist/pregel/loop.cjs +1 -1
- package/dist/pregel/loop.js +1 -1
- package/dist/pregel/messages-v2.cjs +6 -0
- package/dist/pregel/messages-v2.cjs.map +1 -1
- package/dist/pregel/messages-v2.js +6 -0
- package/dist/pregel/messages-v2.js.map +1 -1
- package/dist/pregel/read.cjs +18 -7
- package/dist/pregel/read.cjs.map +1 -1
- package/dist/pregel/read.d.cts +3 -1
- package/dist/pregel/read.d.cts.map +1 -1
- package/dist/pregel/read.d.ts +3 -1
- package/dist/pregel/read.d.ts.map +1 -1
- package/dist/pregel/read.js +18 -7
- package/dist/pregel/read.js.map +1 -1
- package/dist/pregel/runnable.cjs +103 -0
- package/dist/pregel/runnable.cjs.map +1 -0
- package/dist/pregel/runnable.js +103 -0
- package/dist/pregel/runnable.js.map +1 -0
- package/dist/pregel/runner.cjs +1 -1
- package/dist/pregel/runner.js +1 -1
- package/dist/pregel/utils/config.cjs +3 -2
- package/dist/pregel/utils/config.cjs.map +1 -1
- package/dist/pregel/utils/config.d.cts.map +1 -1
- package/dist/pregel/utils/config.d.ts.map +1 -1
- package/dist/pregel/utils/config.js +3 -2
- package/dist/pregel/utils/config.js.map +1 -1
- package/dist/pregel/utils/index.cjs +8 -0
- package/dist/pregel/utils/index.cjs.map +1 -1
- package/dist/pregel/utils/index.d.cts +37 -1
- package/dist/pregel/utils/index.d.cts.map +1 -1
- package/dist/pregel/utils/index.d.ts +37 -1
- package/dist/pregel/utils/index.d.ts.map +1 -1
- package/dist/pregel/utils/index.js +8 -1
- package/dist/pregel/utils/index.js.map +1 -1
- package/dist/web.cjs +6 -4
- package/dist/web.d.cts +2 -2
- package/dist/web.d.ts +2 -2
- package/dist/web.js +2 -1
- package/package.json +7 -7
package/dist/pregel/read.d.cts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { TimeoutPolicy } from "./utils/timeout.cjs";
|
|
2
|
-
import { CachePolicy, RetryPolicy } from "./utils/index.cjs";
|
|
2
|
+
import { CachePolicy, RetryPolicy, TracePolicy } from "./utils/index.cjs";
|
|
3
3
|
import { Runnable, RunnableBinding, RunnableBindingArgs, RunnableConfig, RunnableLike } from "@langchain/core/runnables";
|
|
4
4
|
|
|
5
5
|
//#region src/pregel/read.d.ts
|
|
@@ -16,6 +16,7 @@ interface PregelNodeArgs<RunInput, RunOutput> extends Partial<RunnableBindingArg
|
|
|
16
16
|
retryPolicy?: RetryPolicy;
|
|
17
17
|
cachePolicy?: CachePolicy;
|
|
18
18
|
timeout?: TimeoutPolicy;
|
|
19
|
+
tracePolicy?: TracePolicy;
|
|
19
20
|
subgraphs?: Runnable[];
|
|
20
21
|
ends?: string[];
|
|
21
22
|
/** Whether this node is an auto-generated node-level error handler. */
|
|
@@ -38,6 +39,7 @@ declare class PregelNode<RunInput = PregelNodeInputType, RunOutput = PregelNodeO
|
|
|
38
39
|
retryPolicy?: RetryPolicy;
|
|
39
40
|
cachePolicy?: CachePolicy;
|
|
40
41
|
timeout?: TimeoutPolicy;
|
|
42
|
+
tracePolicy?: TracePolicy;
|
|
41
43
|
subgraphs?: Runnable[];
|
|
42
44
|
ends?: string[];
|
|
43
45
|
isErrorHandler?: boolean;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"read.d.cts","names":[],"sources":["../../src/pregel/read.ts"],"mappings":";;;;;
|
|
1
|
+
{"version":3,"file":"read.d.cts","names":[],"sources":["../../src/pregel/read.ts"],"mappings":";;;;;UA8EU,cAAA,8BAA4C,OAAA,CACpD,mBAAA,CAAoB,QAAA,EAAU,SAAA;EAE9B,QAAA,EAAU,MAAA;EACV,QAAA,EAAU,KAAA;EAEV,MAAA,IAAU,IAAA;EACV,OAAA,GAAU,QAAA,CAAS,SAAA;EACnB,IAAA;EACA,KAAA,GAAQ,QAAA,CAAS,QAAA,EAAU,SAAA;EAE3B,MAAA,GAAS,MAAA;EACT,MAAA,GAAS,cAAA;EACT,QAAA,GAAW,MAAA;EACX,WAAA,GAAc,WAAA;EACd,WAAA,GAAc,WAAA;EACd,OAAA,GAAU,aAAA;EACV,WAAA,GAAc,WAAA;EACd,SAAA,GAAY,QAAA;EACZ,IAAA;EAnB2D;EAqB3D,cAAA;EArBuB;EAuBvB,gBAAA;AAAA;AAAA,KAIU,mBAAA;AAAA,KAGA,oBAAA;AAAA,cAEC,UAAA,YACA,mBAAA,cACC,oBAAA,UACJ,eAAA,CAAgB,QAAA,EAAU,SAAA,EAAW,cAAA;EAC7C,aAAA;EAEA,QAAA,EAAU,MAAA;EAEV,QAAA;EAGA,MAAA,IAAU,IAAA;EAEV,OAAA,EAAS,QAAA;EAET,KAAA,EAAO,QAAA,CAAS,QAAA,EAAU,SAAA;EAG1B,MAAA,EAAQ,MAAA;EAER,QAAA,EAAU,MAAA;EAEV,IAAA;EAEA,WAAA,GAAc,WAAA;EAEd,WAAA,GAAc,WAAA;EAEd,OAAA,GAAU,aAAA;EAEV,WAAA,GAAc,WAAA;EAEd,SAAA,GAAY,QAAA;EAEZ,IAAA;EAEA,cAAA;EAEA,gBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,cAAA,CAAe,QAAA,EAAU,SAAA;EAqD7C,UAAA,CAAA,GAAc,KAAA,CAAM,QAAA;EAsBpB,OAAA,CAAA,GAAW,QAAA,CAAS,QAAA,EAAU,SAAA;EAiC9B,IAAA,CAAK,QAAA,EAAU,KAAA,WAAgB,UAAA,CAAW,QAAA,EAAU,SAAA;EA0BpD,IAAA,cAAA,CACE,UAAA,EAAY,YAAA,GACX,UAAA,CAAW,QAAA,EAAU,OAAA,CAAQ,YAAA,EAAc,KAAA;AAAA"}
|
package/dist/pregel/read.d.ts
CHANGED
|
@@ -1,5 +1,5 @@
|
|
|
1
1
|
import { TimeoutPolicy } from "./utils/timeout.js";
|
|
2
|
-
import { CachePolicy, RetryPolicy } from "./utils/index.js";
|
|
2
|
+
import { CachePolicy, RetryPolicy, TracePolicy } from "./utils/index.js";
|
|
3
3
|
import { Runnable, RunnableBinding, RunnableBindingArgs, RunnableConfig, RunnableLike } from "@langchain/core/runnables";
|
|
4
4
|
|
|
5
5
|
//#region src/pregel/read.d.ts
|
|
@@ -16,6 +16,7 @@ interface PregelNodeArgs<RunInput, RunOutput> extends Partial<RunnableBindingArg
|
|
|
16
16
|
retryPolicy?: RetryPolicy;
|
|
17
17
|
cachePolicy?: CachePolicy;
|
|
18
18
|
timeout?: TimeoutPolicy;
|
|
19
|
+
tracePolicy?: TracePolicy;
|
|
19
20
|
subgraphs?: Runnable[];
|
|
20
21
|
ends?: string[];
|
|
21
22
|
/** Whether this node is an auto-generated node-level error handler. */
|
|
@@ -38,6 +39,7 @@ declare class PregelNode<RunInput = PregelNodeInputType, RunOutput = PregelNodeO
|
|
|
38
39
|
retryPolicy?: RetryPolicy;
|
|
39
40
|
cachePolicy?: CachePolicy;
|
|
40
41
|
timeout?: TimeoutPolicy;
|
|
42
|
+
tracePolicy?: TracePolicy;
|
|
41
43
|
subgraphs?: Runnable[];
|
|
42
44
|
ends?: string[];
|
|
43
45
|
isErrorHandler?: boolean;
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"read.d.ts","names":[],"sources":["../../src/pregel/read.ts"],"mappings":";;;;;
|
|
1
|
+
{"version":3,"file":"read.d.ts","names":[],"sources":["../../src/pregel/read.ts"],"mappings":";;;;;UA8EU,cAAA,8BAA4C,OAAA,CACpD,mBAAA,CAAoB,QAAA,EAAU,SAAA;EAE9B,QAAA,EAAU,MAAA;EACV,QAAA,EAAU,KAAA;EAEV,MAAA,IAAU,IAAA;EACV,OAAA,GAAU,QAAA,CAAS,SAAA;EACnB,IAAA;EACA,KAAA,GAAQ,QAAA,CAAS,QAAA,EAAU,SAAA;EAE3B,MAAA,GAAS,MAAA;EACT,MAAA,GAAS,cAAA;EACT,QAAA,GAAW,MAAA;EACX,WAAA,GAAc,WAAA;EACd,WAAA,GAAc,WAAA;EACd,OAAA,GAAU,aAAA;EACV,WAAA,GAAc,WAAA;EACd,SAAA,GAAY,QAAA;EACZ,IAAA;EAnBoD;EAqBpD,cAAA;EArB2D;EAuB3D,gBAAA;AAAA;AAAA,KAIU,mBAAA;AAAA,KAGA,oBAAA;AAAA,cAEC,UAAA,YACA,mBAAA,cACC,oBAAA,UACJ,eAAA,CAAgB,QAAA,EAAU,SAAA,EAAW,cAAA;EAC7C,aAAA;EAEA,QAAA,EAAU,MAAA;EAEV,QAAA;EAGA,MAAA,IAAU,IAAA;EAEV,OAAA,EAAS,QAAA;EAET,KAAA,EAAO,QAAA,CAAS,QAAA,EAAU,SAAA;EAG1B,MAAA,EAAQ,MAAA;EAER,QAAA,EAAU,MAAA;EAEV,IAAA;EAEA,WAAA,GAAc,WAAA;EAEd,WAAA,GAAc,WAAA;EAEd,OAAA,GAAU,aAAA;EAEV,WAAA,GAAc,WAAA;EAEd,SAAA,GAAY,QAAA;EAEZ,IAAA;EAEA,cAAA;EAEA,gBAAA;EAEA,WAAA,CAAY,MAAA,EAAQ,cAAA,CAAe,QAAA,EAAU,SAAA;EAqD7C,UAAA,CAAA,GAAc,KAAA,CAAM,QAAA;EAsBpB,OAAA,CAAA,GAAW,QAAA,CAAS,QAAA,EAAU,SAAA;EAiC9B,IAAA,CAAK,QAAA,EAAU,KAAA,WAAgB,UAAA,CAAW,QAAA,EAAU,SAAA;EA0BpD,IAAA,cAAA,CACE,UAAA,EAAY,YAAA,GACX,UAAA,CAAW,QAAA,EAAU,OAAA,CAAQ,YAAA,EAAc,KAAA;AAAA"}
|
package/dist/pregel/read.js
CHANGED
|
@@ -1,6 +1,7 @@
|
|
|
1
1
|
import { CONFIG_KEY_READ } from "../constants.js";
|
|
2
2
|
import { RunnableCallable } from "../utils.js";
|
|
3
3
|
import { ChannelWrite } from "./write.js";
|
|
4
|
+
import { RunnableSeq } from "./runnable.js";
|
|
4
5
|
import { RunnableBinding, RunnablePassthrough, RunnableSequence, _coerceToRunnable } from "@langchain/core/runnables";
|
|
5
6
|
//#region src/pregel/read.ts
|
|
6
7
|
var ChannelRead = class ChannelRead extends RunnableCallable {
|
|
@@ -39,12 +40,13 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
39
40
|
retryPolicy;
|
|
40
41
|
cachePolicy;
|
|
41
42
|
timeout;
|
|
43
|
+
tracePolicy;
|
|
42
44
|
subgraphs;
|
|
43
45
|
ends;
|
|
44
46
|
isErrorHandler;
|
|
45
47
|
errorHandlerNode;
|
|
46
48
|
constructor(fields) {
|
|
47
|
-
const { channels, triggers, mapper, writers, bound, kwargs, metadata, retryPolicy, cachePolicy, timeout, tags, subgraphs, ends, isErrorHandler, errorHandlerNode } = fields;
|
|
49
|
+
const { channels, triggers, mapper, writers, bound, kwargs, metadata, retryPolicy, cachePolicy, timeout, tracePolicy, tags, subgraphs, ends, isErrorHandler, errorHandlerNode } = fields;
|
|
48
50
|
const mergedTags = [...fields.config?.tags ? fields.config.tags : [], ...tags ?? []];
|
|
49
51
|
super({
|
|
50
52
|
...fields,
|
|
@@ -65,6 +67,7 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
65
67
|
this.retryPolicy = retryPolicy;
|
|
66
68
|
this.cachePolicy = cachePolicy;
|
|
67
69
|
this.timeout = timeout;
|
|
70
|
+
this.tracePolicy = tracePolicy;
|
|
68
71
|
this.subgraphs = subgraphs;
|
|
69
72
|
this.ends = ends;
|
|
70
73
|
this.isErrorHandler = isErrorHandler;
|
|
@@ -82,15 +85,19 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
82
85
|
}
|
|
83
86
|
getNode() {
|
|
84
87
|
const writers = this.getWriters();
|
|
88
|
+
const sequence = (fields) => this.tracePolicy ? new RunnableSeq({
|
|
89
|
+
...fields,
|
|
90
|
+
tracePolicy: this.tracePolicy
|
|
91
|
+
}) : new RunnableSequence(fields);
|
|
85
92
|
if (this.bound === defaultRunnableBound && writers.length === 0) return;
|
|
86
93
|
else if (this.bound === defaultRunnableBound && writers.length === 1) return writers[0];
|
|
87
|
-
else if (this.bound === defaultRunnableBound) return
|
|
94
|
+
else if (this.bound === defaultRunnableBound) return sequence({
|
|
88
95
|
first: writers[0],
|
|
89
96
|
middle: writers.slice(1, writers.length - 1),
|
|
90
97
|
last: writers[writers.length - 1],
|
|
91
98
|
omitSequenceTags: true
|
|
92
99
|
});
|
|
93
|
-
else if (writers.length > 0) return
|
|
100
|
+
else if (writers.length > 0) return sequence({
|
|
94
101
|
first: this.bound,
|
|
95
102
|
middle: writers.slice(0, writers.length - 1),
|
|
96
103
|
last: writers[writers.length - 1],
|
|
@@ -114,7 +121,8 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
114
121
|
config: this.config,
|
|
115
122
|
retryPolicy: this.retryPolicy,
|
|
116
123
|
cachePolicy: this.cachePolicy,
|
|
117
|
-
timeout: this.timeout
|
|
124
|
+
timeout: this.timeout,
|
|
125
|
+
tracePolicy: this.tracePolicy
|
|
118
126
|
});
|
|
119
127
|
}
|
|
120
128
|
pipe(coerceable) {
|
|
@@ -128,7 +136,8 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
128
136
|
kwargs: this.kwargs,
|
|
129
137
|
retryPolicy: this.retryPolicy,
|
|
130
138
|
cachePolicy: this.cachePolicy,
|
|
131
|
-
timeout: this.timeout
|
|
139
|
+
timeout: this.timeout,
|
|
140
|
+
tracePolicy: this.tracePolicy
|
|
132
141
|
});
|
|
133
142
|
else if (this.bound === defaultRunnableBound) return new PregelNode({
|
|
134
143
|
channels: this.channels,
|
|
@@ -140,7 +149,8 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
140
149
|
kwargs: this.kwargs,
|
|
141
150
|
retryPolicy: this.retryPolicy,
|
|
142
151
|
cachePolicy: this.cachePolicy,
|
|
143
|
-
timeout: this.timeout
|
|
152
|
+
timeout: this.timeout,
|
|
153
|
+
tracePolicy: this.tracePolicy
|
|
144
154
|
});
|
|
145
155
|
else return new PregelNode({
|
|
146
156
|
channels: this.channels,
|
|
@@ -152,7 +162,8 @@ var PregelNode = class PregelNode extends RunnableBinding {
|
|
|
152
162
|
kwargs: this.kwargs,
|
|
153
163
|
retryPolicy: this.retryPolicy,
|
|
154
164
|
cachePolicy: this.cachePolicy,
|
|
155
|
-
timeout: this.timeout
|
|
165
|
+
timeout: this.timeout,
|
|
166
|
+
tracePolicy: this.tracePolicy
|
|
156
167
|
});
|
|
157
168
|
}
|
|
158
169
|
};
|
package/dist/pregel/read.js.map
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"read.js","names":[],"sources":["../../src/pregel/read.ts"],"sourcesContent":["import {\n Runnable,\n RunnableBinding,\n RunnableBindingArgs,\n RunnableConfig,\n RunnablePassthrough,\n RunnableSequence,\n _coerceToRunnable,\n type RunnableLike,\n} from \"@langchain/core/runnables\";\nimport { CONFIG_KEY_READ } from \"../constants.js\";\nimport { ChannelWrite } from \"./write.js\";\nimport { RunnableCallable } from \"../utils.js\";\nimport type { CachePolicy, RetryPolicy, TimeoutPolicy } from \"./utils/index.js\";\n\nexport class ChannelRead<\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n RunInput = any,\n> extends RunnableCallable {\n lc_graph_name = \"ChannelRead\";\n\n channel: string | Array<string>;\n\n fresh: boolean = false;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n constructor(\n channel: string | Array<string>,\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any,\n fresh: boolean = false\n ) {\n super({\n trace: false,\n func: (_: RunInput, config: RunnableConfig) =>\n ChannelRead.doRead(config, this.channel, this.fresh, this.mapper),\n });\n this.fresh = fresh;\n this.mapper = mapper;\n this.channel = channel;\n this.name = Array.isArray(channel)\n ? `ChannelRead<${channel.join(\",\")}>`\n : `ChannelRead<${channel}>`;\n }\n\n static doRead<T = unknown>(\n config: RunnableConfig,\n channel: string | Array<string>,\n fresh: boolean,\n mapper?: (args: unknown) => unknown\n ): T {\n const read: (arg: string | string[], fresh: boolean) => unknown =\n config.configurable?.[CONFIG_KEY_READ];\n if (!read) {\n throw new Error(\n \"Runnable is not configured with a read function. Make sure to call in the context of a Pregel process\"\n );\n }\n if (mapper) {\n return mapper(read(channel, fresh)) as T;\n } else {\n return read(channel, fresh) as T;\n }\n }\n}\n\nconst defaultRunnableBound =\n /* #__PURE__ */ new RunnablePassthrough<PregelNodeInputType>();\n\ninterface PregelNodeArgs<RunInput, RunOutput> extends Partial<\n RunnableBindingArgs<RunInput, RunOutput>\n> {\n channels: Record<string, string> | string[];\n triggers: Array<string>;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any;\n writers?: Runnable<RunOutput, unknown>[];\n tags?: string[];\n bound?: Runnable<RunInput, RunOutput>;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n kwargs?: Record<string, any>;\n config?: RunnableConfig;\n metadata?: Record<string, unknown>;\n retryPolicy?: RetryPolicy;\n cachePolicy?: CachePolicy;\n timeout?: TimeoutPolicy;\n subgraphs?: Runnable[];\n ends?: string[];\n /** Whether this node is an auto-generated node-level error handler. */\n isErrorHandler?: boolean;\n /** Name of the error handler node to run if this node's execution fails. */\n errorHandlerNode?: string;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type PregelNodeInputType = any;\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type PregelNodeOutputType = any;\n\nexport class PregelNode<\n RunInput = PregelNodeInputType,\n RunOutput = PregelNodeOutputType,\n> extends RunnableBinding<RunInput, RunOutput, RunnableConfig> {\n lc_graph_name = \"PregelNode\";\n\n channels: Record<string, string> | string[];\n\n triggers: string[] = [];\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any;\n\n writers: Runnable[] = [];\n\n bound: Runnable<RunInput, RunOutput> = defaultRunnableBound;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n kwargs: Record<string, any> = {};\n\n metadata: Record<string, unknown> = {};\n\n tags: string[] = [];\n\n retryPolicy?: RetryPolicy;\n\n cachePolicy?: CachePolicy;\n\n timeout?: TimeoutPolicy;\n\n subgraphs?: Runnable[];\n\n ends?: string[];\n\n isErrorHandler?: boolean;\n\n errorHandlerNode?: string;\n\n constructor(fields: PregelNodeArgs<RunInput, RunOutput>) {\n const {\n channels,\n triggers,\n mapper,\n writers,\n bound,\n kwargs,\n metadata,\n retryPolicy,\n cachePolicy,\n timeout,\n tags,\n subgraphs,\n ends,\n isErrorHandler,\n errorHandlerNode,\n } = fields;\n const mergedTags = [\n ...(fields.config?.tags ? fields.config.tags : []),\n ...(tags ?? []),\n ];\n\n super({\n ...fields,\n bound:\n fields.bound ??\n (defaultRunnableBound as unknown as Runnable<RunInput, RunOutput>),\n config: {\n ...(fields.config ? fields.config : {}),\n tags: mergedTags,\n },\n });\n\n this.channels = channels;\n this.triggers = triggers;\n this.mapper = mapper;\n this.writers = writers ?? this.writers;\n this.bound = bound ?? this.bound;\n this.kwargs = kwargs ?? this.kwargs;\n this.metadata = metadata ?? this.metadata;\n this.tags = mergedTags;\n this.retryPolicy = retryPolicy;\n this.cachePolicy = cachePolicy;\n this.timeout = timeout;\n this.subgraphs = subgraphs;\n this.ends = ends;\n this.isErrorHandler = isErrorHandler;\n this.errorHandlerNode = errorHandlerNode;\n }\n\n getWriters(): Array<Runnable> {\n const newWriters = [...this.writers];\n while (\n newWriters.length > 1 &&\n // eslint-disable-next-line no-instanceof/no-instanceof\n newWriters[newWriters.length - 1] instanceof ChannelWrite &&\n // eslint-disable-next-line no-instanceof/no-instanceof\n newWriters[newWriters.length - 2] instanceof ChannelWrite\n ) {\n // we can combine writes if they are consecutive\n // careful to not modify the original writers list or ChannelWrite\n const endWriters = newWriters.slice(-2) as ChannelWrite[];\n const combinedWrites = endWriters[0].writes.concat(endWriters[1].writes);\n newWriters[newWriters.length - 2] = new ChannelWrite(\n combinedWrites,\n endWriters[0].config?.tags\n );\n newWriters.pop();\n }\n return newWriters;\n }\n\n getNode(): Runnable<RunInput, RunOutput> | undefined {\n const writers = this.getWriters();\n if (this.bound === defaultRunnableBound && writers.length === 0) {\n return undefined;\n } else if (this.bound === defaultRunnableBound && writers.length === 1) {\n return writers[0];\n } else if (this.bound === defaultRunnableBound) {\n return new RunnableSequence({\n first: writers[0],\n middle: writers.slice(1, writers.length - 1),\n last: writers[writers.length - 1],\n omitSequenceTags: true,\n });\n } else if (writers.length > 0) {\n return new RunnableSequence({\n first: this.bound,\n middle: writers.slice(0, writers.length - 1),\n last: writers[writers.length - 1],\n omitSequenceTags: true,\n });\n } else {\n return this.bound;\n }\n }\n\n join(channels: Array<string>): PregelNode<RunInput, RunOutput> {\n if (!Array.isArray(channels)) {\n throw new Error(\"channels must be a list\");\n }\n if (typeof this.channels !== \"object\") {\n throw new Error(\"all channels must be named when using .join()\");\n }\n\n return new PregelNode<RunInput, RunOutput>({\n channels: {\n ...this.channels,\n ...Object.fromEntries(channels.map((chan) => [chan, chan])),\n },\n triggers: this.triggers,\n mapper: this.mapper,\n writers: this.writers,\n bound: this.bound,\n kwargs: this.kwargs,\n config: this.config,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n });\n }\n\n pipe<NewRunOutput>(\n coerceable: RunnableLike\n ): PregelNode<RunInput, Exclude<NewRunOutput, Error>> {\n if (ChannelWrite.isWriter(coerceable)) {\n return new PregelNode<RunInput, Exclude<NewRunOutput, Error>>({\n channels: this.channels,\n triggers: this.triggers,\n mapper: this.mapper,\n writers: [...this.writers, coerceable],\n bound: this.bound as unknown as PregelNode<\n RunInput,\n Exclude<NewRunOutput, Error>\n >,\n config: this.config,\n kwargs: this.kwargs,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n });\n } else if (this.bound === defaultRunnableBound) {\n return new PregelNode<RunInput, Exclude<NewRunOutput, Error>>({\n channels: this.channels,\n triggers: this.triggers,\n mapper: this.mapper,\n writers: this.writers,\n bound: _coerceToRunnable<RunInput, NewRunOutput>(coerceable),\n config: this.config,\n kwargs: this.kwargs,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n });\n } else {\n return new PregelNode<RunInput, Exclude<NewRunOutput, Error>>({\n channels: this.channels,\n triggers: this.triggers,\n mapper: this.mapper,\n writers: this.writers,\n bound: this.bound.pipe(coerceable),\n config: this.config,\n kwargs: this.kwargs,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n });\n }\n }\n}\n"],"mappings":";;;;;AAeA,IAAa,cAAb,MAAa,oBAGH,iBAAiB;CACzB,gBAAgB;CAEhB;CAEA,QAAiB;CAGjB;CAGA,YACE,SAEA,QACA,QAAiB,OACjB;AACA,QAAM;GACJ,OAAO;GACP,OAAO,GAAa,WAClB,YAAY,OAAO,QAAQ,KAAK,SAAS,KAAK,OAAO,KAAK,OAAO;GACpE,CAAC;AACF,OAAK,QAAQ;AACb,OAAK,SAAS;AACd,OAAK,UAAU;AACf,OAAK,OAAO,MAAM,QAAQ,QAAQ,GAC9B,eAAe,QAAQ,KAAK,IAAI,CAAC,KACjC,eAAe,QAAQ;;CAG7B,OAAO,OACL,QACA,SACA,OACA,QACG;EACH,MAAM,OACJ,OAAO,eAAe;AACxB,MAAI,CAAC,KACH,OAAM,IAAI,MACR,wGACD;AAEH,MAAI,OACF,QAAO,OAAO,KAAK,SAAS,MAAM,CAAC;MAEnC,QAAO,KAAK,SAAS,MAAM;;;AAKjC,MAAM,uCACY,IAAI,qBAA0C;AAiChE,IAAa,aAAb,MAAa,mBAGH,gBAAqD;CAC7D,gBAAgB;CAEhB;CAEA,WAAqB,EAAE;CAGvB;CAEA,UAAsB,EAAE;CAExB,QAAuC;CAGvC,SAA8B,EAAE;CAEhC,WAAoC,EAAE;CAEtC,OAAiB,EAAE;CAEnB;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,QAA6C;EACvD,MAAM,EACJ,UACA,UACA,QACA,SACA,OACA,QACA,UACA,aACA,aACA,SACA,MACA,WACA,MACA,gBACA,qBACE;EACJ,MAAM,aAAa,CACjB,GAAI,OAAO,QAAQ,OAAO,OAAO,OAAO,OAAO,EAAE,EACjD,GAAI,QAAQ,EAAE,CACf;AAED,QAAM;GACJ,GAAG;GACH,OACE,OAAO,SACN;GACH,QAAQ;IACN,GAAI,OAAO,SAAS,OAAO,SAAS,EAAE;IACtC,MAAM;IACP;GACF,CAAC;AAEF,OAAK,WAAW;AAChB,OAAK,WAAW;AAChB,OAAK,SAAS;AACd,OAAK,UAAU,WAAW,KAAK;AAC/B,OAAK,QAAQ,SAAS,KAAK;AAC3B,OAAK,SAAS,UAAU,KAAK;AAC7B,OAAK,WAAW,YAAY,KAAK;AACjC,OAAK,OAAO;AACZ,OAAK,cAAc;AACnB,OAAK,cAAc;AACnB,OAAK,UAAU;AACf,OAAK,YAAY;AACjB,OAAK,OAAO;AACZ,OAAK,iBAAiB;AACtB,OAAK,mBAAmB;;CAG1B,aAA8B;EAC5B,MAAM,aAAa,CAAC,GAAG,KAAK,QAAQ;AACpC,SACE,WAAW,SAAS,KAEpB,WAAW,WAAW,SAAS,cAAc,gBAE7C,WAAW,WAAW,SAAS,cAAc,cAC7C;GAGA,MAAM,aAAa,WAAW,MAAM,GAAG;GACvC,MAAM,iBAAiB,WAAW,GAAG,OAAO,OAAO,WAAW,GAAG,OAAO;AACxE,cAAW,WAAW,SAAS,KAAK,IAAI,aACtC,gBACA,WAAW,GAAG,QAAQ,KACvB;AACD,cAAW,KAAK;;AAElB,SAAO;;CAGT,UAAqD;EACnD,MAAM,UAAU,KAAK,YAAY;AACjC,MAAI,KAAK,UAAU,wBAAwB,QAAQ,WAAW,EAC5D;WACS,KAAK,UAAU,wBAAwB,QAAQ,WAAW,EACnE,QAAO,QAAQ;WACN,KAAK,UAAU,qBACxB,QAAO,IAAI,iBAAiB;GAC1B,OAAO,QAAQ;GACf,QAAQ,QAAQ,MAAM,GAAG,QAAQ,SAAS,EAAE;GAC5C,MAAM,QAAQ,QAAQ,SAAS;GAC/B,kBAAkB;GACnB,CAAC;WACO,QAAQ,SAAS,EAC1B,QAAO,IAAI,iBAAiB;GAC1B,OAAO,KAAK;GACZ,QAAQ,QAAQ,MAAM,GAAG,QAAQ,SAAS,EAAE;GAC5C,MAAM,QAAQ,QAAQ,SAAS;GAC/B,kBAAkB;GACnB,CAAC;MAEF,QAAO,KAAK;;CAIhB,KAAK,UAA0D;AAC7D,MAAI,CAAC,MAAM,QAAQ,SAAS,CAC1B,OAAM,IAAI,MAAM,0BAA0B;AAE5C,MAAI,OAAO,KAAK,aAAa,SAC3B,OAAM,IAAI,MAAM,gDAAgD;AAGlE,SAAO,IAAI,WAAgC;GACzC,UAAU;IACR,GAAG,KAAK;IACR,GAAG,OAAO,YAAY,SAAS,KAAK,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC;IAC5D;GACD,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,OAAO,KAAK;GACZ,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACf,CAAC;;CAGJ,KACE,YACoD;AACpD,MAAI,aAAa,SAAS,WAAW,CACnC,QAAO,IAAI,WAAmD;GAC5D,UAAU,KAAK;GACf,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,CAAC,GAAG,KAAK,SAAS,WAAW;GACtC,OAAO,KAAK;GAIZ,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACf,CAAC;WACO,KAAK,UAAU,qBACxB,QAAO,IAAI,WAAmD;GAC5D,UAAU,KAAK;GACf,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,OAAO,kBAA0C,WAAW;GAC5D,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACf,CAAC;MAEF,QAAO,IAAI,WAAmD;GAC5D,UAAU,KAAK;GACf,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,OAAO,KAAK,MAAM,KAAK,WAAW;GAClC,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACf,CAAC"}
|
|
1
|
+
{"version":3,"file":"read.js","names":[],"sources":["../../src/pregel/read.ts"],"sourcesContent":["import {\n Runnable,\n RunnableBinding,\n RunnableBindingArgs,\n RunnableConfig,\n RunnablePassthrough,\n RunnableSequence,\n _coerceToRunnable,\n type RunnableLike,\n} from \"@langchain/core/runnables\";\nimport { CONFIG_KEY_READ } from \"../constants.js\";\nimport { ChannelWrite } from \"./write.js\";\nimport { RunnableCallable } from \"../utils.js\";\nimport type {\n CachePolicy,\n RetryPolicy,\n TimeoutPolicy,\n TracePolicy,\n} from \"./utils/index.js\";\nimport { RunnableSeq } from \"./runnable.js\";\n\nexport class ChannelRead<\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n RunInput = any,\n> extends RunnableCallable {\n lc_graph_name = \"ChannelRead\";\n\n channel: string | Array<string>;\n\n fresh: boolean = false;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n constructor(\n channel: string | Array<string>,\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any,\n fresh: boolean = false\n ) {\n super({\n trace: false,\n func: (_: RunInput, config: RunnableConfig) =>\n ChannelRead.doRead(config, this.channel, this.fresh, this.mapper),\n });\n this.fresh = fresh;\n this.mapper = mapper;\n this.channel = channel;\n this.name = Array.isArray(channel)\n ? `ChannelRead<${channel.join(\",\")}>`\n : `ChannelRead<${channel}>`;\n }\n\n static doRead<T = unknown>(\n config: RunnableConfig,\n channel: string | Array<string>,\n fresh: boolean,\n mapper?: (args: unknown) => unknown\n ): T {\n const read: (arg: string | string[], fresh: boolean) => unknown =\n config.configurable?.[CONFIG_KEY_READ];\n if (!read) {\n throw new Error(\n \"Runnable is not configured with a read function. Make sure to call in the context of a Pregel process\"\n );\n }\n if (mapper) {\n return mapper(read(channel, fresh)) as T;\n } else {\n return read(channel, fresh) as T;\n }\n }\n}\n\nconst defaultRunnableBound =\n /* #__PURE__ */ new RunnablePassthrough<PregelNodeInputType>();\n\ninterface PregelNodeArgs<RunInput, RunOutput> extends Partial<\n RunnableBindingArgs<RunInput, RunOutput>\n> {\n channels: Record<string, string> | string[];\n triggers: Array<string>;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any;\n writers?: Runnable<RunOutput, unknown>[];\n tags?: string[];\n bound?: Runnable<RunInput, RunOutput>;\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n kwargs?: Record<string, any>;\n config?: RunnableConfig;\n metadata?: Record<string, unknown>;\n retryPolicy?: RetryPolicy;\n cachePolicy?: CachePolicy;\n timeout?: TimeoutPolicy;\n tracePolicy?: TracePolicy;\n subgraphs?: Runnable[];\n ends?: string[];\n /** Whether this node is an auto-generated node-level error handler. */\n isErrorHandler?: boolean;\n /** Name of the error handler node to run if this node's execution fails. */\n errorHandlerNode?: string;\n}\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type PregelNodeInputType = any;\n\n// eslint-disable-next-line @typescript-eslint/no-explicit-any\nexport type PregelNodeOutputType = any;\n\nexport class PregelNode<\n RunInput = PregelNodeInputType,\n RunOutput = PregelNodeOutputType,\n> extends RunnableBinding<RunInput, RunOutput, RunnableConfig> {\n lc_graph_name = \"PregelNode\";\n\n channels: Record<string, string> | string[];\n\n triggers: string[] = [];\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n mapper?: (args: any) => any;\n\n writers: Runnable[] = [];\n\n bound: Runnable<RunInput, RunOutput> = defaultRunnableBound;\n\n // eslint-disable-next-line @typescript-eslint/no-explicit-any\n kwargs: Record<string, any> = {};\n\n metadata: Record<string, unknown> = {};\n\n tags: string[] = [];\n\n retryPolicy?: RetryPolicy;\n\n cachePolicy?: CachePolicy;\n\n timeout?: TimeoutPolicy;\n\n tracePolicy?: TracePolicy;\n\n subgraphs?: Runnable[];\n\n ends?: string[];\n\n isErrorHandler?: boolean;\n\n errorHandlerNode?: string;\n\n constructor(fields: PregelNodeArgs<RunInput, RunOutput>) {\n const {\n channels,\n triggers,\n mapper,\n writers,\n bound,\n kwargs,\n metadata,\n retryPolicy,\n cachePolicy,\n timeout,\n tracePolicy,\n tags,\n subgraphs,\n ends,\n isErrorHandler,\n errorHandlerNode,\n } = fields;\n const mergedTags = [\n ...(fields.config?.tags ? fields.config.tags : []),\n ...(tags ?? []),\n ];\n\n super({\n ...fields,\n bound:\n fields.bound ??\n (defaultRunnableBound as unknown as Runnable<RunInput, RunOutput>),\n config: {\n ...(fields.config ? fields.config : {}),\n tags: mergedTags,\n },\n });\n\n this.channels = channels;\n this.triggers = triggers;\n this.mapper = mapper;\n this.writers = writers ?? this.writers;\n this.bound = bound ?? this.bound;\n this.kwargs = kwargs ?? this.kwargs;\n this.metadata = metadata ?? this.metadata;\n this.tags = mergedTags;\n this.retryPolicy = retryPolicy;\n this.cachePolicy = cachePolicy;\n this.timeout = timeout;\n this.tracePolicy = tracePolicy;\n this.subgraphs = subgraphs;\n this.ends = ends;\n this.isErrorHandler = isErrorHandler;\n this.errorHandlerNode = errorHandlerNode;\n }\n\n getWriters(): Array<Runnable> {\n const newWriters = [...this.writers];\n while (\n newWriters.length > 1 &&\n // eslint-disable-next-line no-instanceof/no-instanceof\n newWriters[newWriters.length - 1] instanceof ChannelWrite &&\n // eslint-disable-next-line no-instanceof/no-instanceof\n newWriters[newWriters.length - 2] instanceof ChannelWrite\n ) {\n // we can combine writes if they are consecutive\n // careful to not modify the original writers list or ChannelWrite\n const endWriters = newWriters.slice(-2) as ChannelWrite[];\n const combinedWrites = endWriters[0].writes.concat(endWriters[1].writes);\n newWriters[newWriters.length - 2] = new ChannelWrite(\n combinedWrites,\n endWriters[0].config?.tags\n );\n newWriters.pop();\n }\n return newWriters;\n }\n\n getNode(): Runnable<RunInput, RunOutput> | undefined {\n const writers = this.getWriters();\n const sequence = (\n fields: ConstructorParameters<\n typeof RunnableSequence<RunInput, RunOutput>\n >[0]\n ) =>\n this.tracePolicy\n ? new RunnableSeq({ ...fields, tracePolicy: this.tracePolicy })\n : new RunnableSequence(fields);\n if (this.bound === defaultRunnableBound && writers.length === 0) {\n return undefined;\n } else if (this.bound === defaultRunnableBound && writers.length === 1) {\n return writers[0];\n } else if (this.bound === defaultRunnableBound) {\n return sequence({\n first: writers[0],\n middle: writers.slice(1, writers.length - 1),\n last: writers[writers.length - 1],\n omitSequenceTags: true,\n });\n } else if (writers.length > 0) {\n return sequence({\n first: this.bound,\n middle: writers.slice(0, writers.length - 1),\n last: writers[writers.length - 1],\n omitSequenceTags: true,\n });\n } else {\n return this.bound;\n }\n }\n\n join(channels: Array<string>): PregelNode<RunInput, RunOutput> {\n if (!Array.isArray(channels)) {\n throw new Error(\"channels must be a list\");\n }\n if (typeof this.channels !== \"object\") {\n throw new Error(\"all channels must be named when using .join()\");\n }\n\n return new PregelNode<RunInput, RunOutput>({\n channels: {\n ...this.channels,\n ...Object.fromEntries(channels.map((chan) => [chan, chan])),\n },\n triggers: this.triggers,\n mapper: this.mapper,\n writers: this.writers,\n bound: this.bound,\n kwargs: this.kwargs,\n config: this.config,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n tracePolicy: this.tracePolicy,\n });\n }\n\n pipe<NewRunOutput>(\n coerceable: RunnableLike\n ): PregelNode<RunInput, Exclude<NewRunOutput, Error>> {\n if (ChannelWrite.isWriter(coerceable)) {\n return new PregelNode<RunInput, Exclude<NewRunOutput, Error>>({\n channels: this.channels,\n triggers: this.triggers,\n mapper: this.mapper,\n writers: [...this.writers, coerceable],\n bound: this.bound as unknown as PregelNode<\n RunInput,\n Exclude<NewRunOutput, Error>\n >,\n config: this.config,\n kwargs: this.kwargs,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n tracePolicy: this.tracePolicy,\n });\n } else if (this.bound === defaultRunnableBound) {\n return new PregelNode<RunInput, Exclude<NewRunOutput, Error>>({\n channels: this.channels,\n triggers: this.triggers,\n mapper: this.mapper,\n writers: this.writers,\n bound: _coerceToRunnable<RunInput, NewRunOutput>(coerceable),\n config: this.config,\n kwargs: this.kwargs,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n tracePolicy: this.tracePolicy,\n });\n } else {\n return new PregelNode<RunInput, Exclude<NewRunOutput, Error>>({\n channels: this.channels,\n triggers: this.triggers,\n mapper: this.mapper,\n writers: this.writers,\n bound: this.bound.pipe(coerceable),\n config: this.config,\n kwargs: this.kwargs,\n retryPolicy: this.retryPolicy,\n cachePolicy: this.cachePolicy,\n timeout: this.timeout,\n tracePolicy: this.tracePolicy,\n });\n }\n }\n}\n"],"mappings":";;;;;;AAqBA,IAAa,cAAb,MAAa,oBAGH,iBAAiB;CACzB,gBAAgB;CAEhB;CAEA,QAAiB;CAGjB;CAGA,YACE,SAEA,QACA,QAAiB,OACjB;AACA,QAAM;GACJ,OAAO;GACP,OAAO,GAAa,WAClB,YAAY,OAAO,QAAQ,KAAK,SAAS,KAAK,OAAO,KAAK,OAAO;GACpE,CAAC;AACF,OAAK,QAAQ;AACb,OAAK,SAAS;AACd,OAAK,UAAU;AACf,OAAK,OAAO,MAAM,QAAQ,QAAQ,GAC9B,eAAe,QAAQ,KAAK,IAAI,CAAC,KACjC,eAAe,QAAQ;;CAG7B,OAAO,OACL,QACA,SACA,OACA,QACG;EACH,MAAM,OACJ,OAAO,eAAe;AACxB,MAAI,CAAC,KACH,OAAM,IAAI,MACR,wGACD;AAEH,MAAI,OACF,QAAO,OAAO,KAAK,SAAS,MAAM,CAAC;MAEnC,QAAO,KAAK,SAAS,MAAM;;;AAKjC,MAAM,uCACY,IAAI,qBAA0C;AAkChE,IAAa,aAAb,MAAa,mBAGH,gBAAqD;CAC7D,gBAAgB;CAEhB;CAEA,WAAqB,EAAE;CAGvB;CAEA,UAAsB,EAAE;CAExB,QAAuC;CAGvC,SAA8B,EAAE;CAEhC,WAAoC,EAAE;CAEtC,OAAiB,EAAE;CAEnB;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA;CAEA,YAAY,QAA6C;EACvD,MAAM,EACJ,UACA,UACA,QACA,SACA,OACA,QACA,UACA,aACA,aACA,SACA,aACA,MACA,WACA,MACA,gBACA,qBACE;EACJ,MAAM,aAAa,CACjB,GAAI,OAAO,QAAQ,OAAO,OAAO,OAAO,OAAO,EAAE,EACjD,GAAI,QAAQ,EAAE,CACf;AAED,QAAM;GACJ,GAAG;GACH,OACE,OAAO,SACN;GACH,QAAQ;IACN,GAAI,OAAO,SAAS,OAAO,SAAS,EAAE;IACtC,MAAM;IACP;GACF,CAAC;AAEF,OAAK,WAAW;AAChB,OAAK,WAAW;AAChB,OAAK,SAAS;AACd,OAAK,UAAU,WAAW,KAAK;AAC/B,OAAK,QAAQ,SAAS,KAAK;AAC3B,OAAK,SAAS,UAAU,KAAK;AAC7B,OAAK,WAAW,YAAY,KAAK;AACjC,OAAK,OAAO;AACZ,OAAK,cAAc;AACnB,OAAK,cAAc;AACnB,OAAK,UAAU;AACf,OAAK,cAAc;AACnB,OAAK,YAAY;AACjB,OAAK,OAAO;AACZ,OAAK,iBAAiB;AACtB,OAAK,mBAAmB;;CAG1B,aAA8B;EAC5B,MAAM,aAAa,CAAC,GAAG,KAAK,QAAQ;AACpC,SACE,WAAW,SAAS,KAEpB,WAAW,WAAW,SAAS,cAAc,gBAE7C,WAAW,WAAW,SAAS,cAAc,cAC7C;GAGA,MAAM,aAAa,WAAW,MAAM,GAAG;GACvC,MAAM,iBAAiB,WAAW,GAAG,OAAO,OAAO,WAAW,GAAG,OAAO;AACxE,cAAW,WAAW,SAAS,KAAK,IAAI,aACtC,gBACA,WAAW,GAAG,QAAQ,KACvB;AACD,cAAW,KAAK;;AAElB,SAAO;;CAGT,UAAqD;EACnD,MAAM,UAAU,KAAK,YAAY;EACjC,MAAM,YACJ,WAIA,KAAK,cACD,IAAI,YAAY;GAAE,GAAG;GAAQ,aAAa,KAAK;GAAa,CAAC,GAC7D,IAAI,iBAAiB,OAAO;AAClC,MAAI,KAAK,UAAU,wBAAwB,QAAQ,WAAW,EAC5D;WACS,KAAK,UAAU,wBAAwB,QAAQ,WAAW,EACnE,QAAO,QAAQ;WACN,KAAK,UAAU,qBACxB,QAAO,SAAS;GACd,OAAO,QAAQ;GACf,QAAQ,QAAQ,MAAM,GAAG,QAAQ,SAAS,EAAE;GAC5C,MAAM,QAAQ,QAAQ,SAAS;GAC/B,kBAAkB;GACnB,CAAC;WACO,QAAQ,SAAS,EAC1B,QAAO,SAAS;GACd,OAAO,KAAK;GACZ,QAAQ,QAAQ,MAAM,GAAG,QAAQ,SAAS,EAAE;GAC5C,MAAM,QAAQ,QAAQ,SAAS;GAC/B,kBAAkB;GACnB,CAAC;MAEF,QAAO,KAAK;;CAIhB,KAAK,UAA0D;AAC7D,MAAI,CAAC,MAAM,QAAQ,SAAS,CAC1B,OAAM,IAAI,MAAM,0BAA0B;AAE5C,MAAI,OAAO,KAAK,aAAa,SAC3B,OAAM,IAAI,MAAM,gDAAgD;AAGlE,SAAO,IAAI,WAAgC;GACzC,UAAU;IACR,GAAG,KAAK;IACR,GAAG,OAAO,YAAY,SAAS,KAAK,SAAS,CAAC,MAAM,KAAK,CAAC,CAAC;IAC5D;GACD,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,OAAO,KAAK;GACZ,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACd,aAAa,KAAK;GACnB,CAAC;;CAGJ,KACE,YACoD;AACpD,MAAI,aAAa,SAAS,WAAW,CACnC,QAAO,IAAI,WAAmD;GAC5D,UAAU,KAAK;GACf,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,CAAC,GAAG,KAAK,SAAS,WAAW;GACtC,OAAO,KAAK;GAIZ,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACd,aAAa,KAAK;GACnB,CAAC;WACO,KAAK,UAAU,qBACxB,QAAO,IAAI,WAAmD;GAC5D,UAAU,KAAK;GACf,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,OAAO,kBAA0C,WAAW;GAC5D,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACd,aAAa,KAAK;GACnB,CAAC;MAEF,QAAO,IAAI,WAAmD;GAC5D,UAAU,KAAK;GACf,UAAU,KAAK;GACf,QAAQ,KAAK;GACb,SAAS,KAAK;GACd,OAAO,KAAK,MAAM,KAAK,WAAW;GAClC,QAAQ,KAAK;GACb,QAAQ,KAAK;GACb,aAAa,KAAK;GAClB,aAAa,KAAK;GAClB,SAAS,KAAK;GACd,aAAa,KAAK;GACnB,CAAC"}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
const require_index = require("./utils/index.cjs");
|
|
2
|
+
let _langchain_core_runnables = require("@langchain/core/runnables");
|
|
3
|
+
//#region src/pregel/runnable.ts
|
|
4
|
+
function tracePayload(value, processor) {
|
|
5
|
+
if (processor === void 0) return value;
|
|
6
|
+
try {
|
|
7
|
+
return processor(value);
|
|
8
|
+
} catch (error) {
|
|
9
|
+
console.warn("Trace input/output processor raised; recording untransformed payload", error);
|
|
10
|
+
return value;
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* A node sequence with transforms at its own tracing boundaries. Keep execution
|
|
15
|
+
* and child callback propagation aligned with Core's RunnableSequence; only the
|
|
16
|
+
* values passed to handleChainStart/handleChainEnd are transformed.
|
|
17
|
+
* @internal
|
|
18
|
+
*/
|
|
19
|
+
var RunnableSeq = class extends _langchain_core_runnables.RunnableSequence {
|
|
20
|
+
tracePolicy;
|
|
21
|
+
constructor(fields) {
|
|
22
|
+
const { tracePolicy, ...sequenceFields } = fields;
|
|
23
|
+
super(sequenceFields);
|
|
24
|
+
this.tracePolicy = tracePolicy;
|
|
25
|
+
}
|
|
26
|
+
async startTrace(input, config) {
|
|
27
|
+
const payload = tracePayload(input, this.tracePolicy?.processInputs);
|
|
28
|
+
return (await (0, _langchain_core_runnables.getCallbackManagerForConfig)(config))?.handleChainStart(this.toJSON(), require_index._coerceToDict(payload, "input"), config.runId, void 0, void 0, void 0, config.runName);
|
|
29
|
+
}
|
|
30
|
+
async endTrace(output, manager) {
|
|
31
|
+
const payload = tracePayload(output, this.tracePolicy?.processOutputs);
|
|
32
|
+
await manager?.handleChainEnd(require_index._coerceToDict(payload, "output"));
|
|
33
|
+
}
|
|
34
|
+
async invoke(input, options) {
|
|
35
|
+
const config = (0, _langchain_core_runnables.ensureConfig)(options);
|
|
36
|
+
const manager = await this.startTrace(input, config);
|
|
37
|
+
delete config.runId;
|
|
38
|
+
let nextInput = input;
|
|
39
|
+
let output;
|
|
40
|
+
try {
|
|
41
|
+
const steps = [this.first, ...this.middle];
|
|
42
|
+
for (let i = 0; i < steps.length; i += 1) nextInput = await (0, _langchain_core_runnables.raceWithSignal)(steps[i].invoke(nextInput, (0, _langchain_core_runnables.patchConfig)(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) })), config.signal);
|
|
43
|
+
if (config.signal?.aborted) await (0, _langchain_core_runnables.raceWithSignal)(Promise.resolve(), config.signal);
|
|
44
|
+
output = await this.last.invoke(nextInput, (0, _langchain_core_runnables.patchConfig)(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${this.steps.length}`) }));
|
|
45
|
+
} catch (error) {
|
|
46
|
+
await manager?.handleChainError(error);
|
|
47
|
+
throw error;
|
|
48
|
+
}
|
|
49
|
+
await this.endTrace(output, manager);
|
|
50
|
+
return output;
|
|
51
|
+
}
|
|
52
|
+
async batch(inputs, options, batchOptions) {
|
|
53
|
+
const configList = this._getOptionsList(options ?? {}, inputs.length);
|
|
54
|
+
const managers = await Promise.all(configList.map(async (config, i) => {
|
|
55
|
+
const manager = await this.startTrace(inputs[i], config);
|
|
56
|
+
delete config.runId;
|
|
57
|
+
return manager;
|
|
58
|
+
}));
|
|
59
|
+
let outputs = inputs;
|
|
60
|
+
try {
|
|
61
|
+
for (let i = 0; i < this.steps.length; i += 1) outputs = await (0, _langchain_core_runnables.raceWithSignal)(this.steps[i].batch(outputs, managers.map((manager, j) => (0, _langchain_core_runnables.patchConfig)(configList[j], { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) })), batchOptions), configList[0]?.signal);
|
|
62
|
+
} catch (error) {
|
|
63
|
+
await Promise.all(managers.map((manager) => manager?.handleChainError(error)));
|
|
64
|
+
throw error;
|
|
65
|
+
}
|
|
66
|
+
await Promise.all(managers.map((manager, i) => this.endTrace(outputs[i], manager)));
|
|
67
|
+
return outputs;
|
|
68
|
+
}
|
|
69
|
+
async *_streamIterator(input, options) {
|
|
70
|
+
const config = (0, _langchain_core_runnables.ensureConfig)(options);
|
|
71
|
+
const manager = await this.startTrace(input, config);
|
|
72
|
+
delete config.runId;
|
|
73
|
+
const steps = this.steps;
|
|
74
|
+
let output;
|
|
75
|
+
let concatSupported = true;
|
|
76
|
+
async function* inputGenerator() {
|
|
77
|
+
yield input;
|
|
78
|
+
}
|
|
79
|
+
try {
|
|
80
|
+
let generator = steps[0].transform(inputGenerator(), (0, _langchain_core_runnables.patchConfig)(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : "seq:step:1") }));
|
|
81
|
+
for (let i = 1; i < steps.length; i += 1) generator = steps[i].transform(generator, (0, _langchain_core_runnables.patchConfig)(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) }));
|
|
82
|
+
for await (const chunk of generator) {
|
|
83
|
+
config.signal?.throwIfAborted();
|
|
84
|
+
yield chunk;
|
|
85
|
+
if (concatSupported) if (output === void 0) output = chunk;
|
|
86
|
+
else try {
|
|
87
|
+
output = this._concatOutputChunks(output, chunk);
|
|
88
|
+
} catch {
|
|
89
|
+
output = void 0;
|
|
90
|
+
concatSupported = false;
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
} catch (error) {
|
|
94
|
+
await manager?.handleChainError(error);
|
|
95
|
+
throw error;
|
|
96
|
+
}
|
|
97
|
+
await this.endTrace(output, manager);
|
|
98
|
+
}
|
|
99
|
+
};
|
|
100
|
+
//#endregion
|
|
101
|
+
exports.RunnableSeq = RunnableSeq;
|
|
102
|
+
|
|
103
|
+
//# sourceMappingURL=runnable.cjs.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runnable.cjs","names":["RunnableSequence","_coerceToDict"],"sources":["../../src/pregel/runnable.ts"],"sourcesContent":["import type { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport {\n ensureConfig,\n getCallbackManagerForConfig,\n patchConfig,\n raceWithSignal,\n RunnableSequence,\n type RunnableBatchOptions,\n type RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { _coerceToDict, type TracePolicy } from \"./utils/index.js\";\n\nfunction tracePayload(\n value: unknown,\n processor: ((value: unknown) => unknown) | undefined\n): unknown {\n if (processor === undefined) return value;\n try {\n return processor(value);\n } catch (error) {\n console.warn(\n \"Trace input/output processor raised; recording untransformed payload\",\n error\n );\n return value;\n }\n}\n\n/**\n * A node sequence with transforms at its own tracing boundaries. Keep execution\n * and child callback propagation aligned with Core's RunnableSequence; only the\n * values passed to handleChainStart/handleChainEnd are transformed.\n * @internal\n */\nexport class RunnableSeq<\n RunInput = unknown,\n RunOutput = unknown,\n> extends RunnableSequence<RunInput, RunOutput> {\n private tracePolicy?: TracePolicy;\n\n constructor(\n fields: ConstructorParameters<\n typeof RunnableSequence<RunInput, RunOutput>\n >[0] & {\n tracePolicy?: TracePolicy;\n }\n ) {\n const { tracePolicy, ...sequenceFields } = fields;\n super(sequenceFields);\n this.tracePolicy = tracePolicy;\n }\n\n private async startTrace(input: RunInput, config: RunnableConfig) {\n // Evaluate outside optional chaining: processors run without callbacks too.\n const payload = tracePayload(input, this.tracePolicy?.processInputs);\n const manager = await getCallbackManagerForConfig(config);\n return manager?.handleChainStart(\n this.toJSON(),\n _coerceToDict(payload, \"input\"),\n config.runId,\n undefined,\n undefined,\n undefined,\n config.runName\n );\n }\n\n private async endTrace(\n output: unknown,\n manager: CallbackManagerForChainRun | undefined\n ) {\n const payload = tracePayload(output, this.tracePolicy?.processOutputs);\n await manager?.handleChainEnd(_coerceToDict(payload, \"output\"));\n }\n\n override async invoke(\n input: RunInput,\n options?: RunnableConfig\n ): Promise<RunOutput> {\n const config = ensureConfig(options);\n const manager = await this.startTrace(input, config);\n delete config.runId;\n let nextInput = input;\n let output: RunOutput;\n try {\n const steps = [this.first, ...this.middle];\n for (let i = 0; i < steps.length; i += 1) {\n nextInput = await raceWithSignal(\n steps[i].invoke(\n nextInput,\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${i + 1}`\n ),\n })\n ),\n config.signal\n );\n }\n if (config.signal?.aborted) {\n // Use Core's abort-error normalization before invoking the last step.\n await raceWithSignal(Promise.resolve(), config.signal);\n }\n output = await this.last.invoke(\n nextInput,\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${this.steps.length}`\n ),\n })\n );\n } catch (error) {\n await manager?.handleChainError(error);\n throw error;\n }\n await this.endTrace(output, manager);\n return output;\n }\n\n override batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions & { returnExceptions?: false }\n ): Promise<RunOutput[]>;\n override batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions & { returnExceptions: true }\n ): Promise<(RunOutput | Error)[]>;\n override batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions\n ): Promise<(RunOutput | Error)[]>;\n override async batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions\n ): Promise<(RunOutput | Error)[]> {\n const configList = this._getOptionsList(options ?? {}, inputs.length);\n const managers = await Promise.all(\n configList.map(async (config, i) => {\n const manager = await this.startTrace(inputs[i], config);\n delete config.runId;\n return manager;\n })\n );\n let outputs: unknown[] = inputs;\n try {\n // Preserve RunnableSequence's step-wise batching so child runnables can\n // use their native batch implementations and handle returnExceptions.\n for (let i = 0; i < this.steps.length; i += 1) {\n outputs = await raceWithSignal(\n this.steps[i].batch(\n outputs,\n managers.map((manager, j) =>\n patchConfig(configList[j], {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${i + 1}`\n ),\n })\n ),\n batchOptions\n ),\n configList[0]?.signal\n );\n }\n } catch (error) {\n await Promise.all(\n managers.map((manager) => manager?.handleChainError(error))\n );\n throw error;\n }\n await Promise.all(\n managers.map((manager, i) => this.endTrace(outputs[i], manager))\n );\n return outputs as (RunOutput | Error)[];\n }\n\n override async *_streamIterator(input: RunInput, options?: RunnableConfig) {\n const config = ensureConfig(options);\n const manager = await this.startTrace(input, config);\n delete config.runId;\n const steps = this.steps;\n let output: RunOutput | undefined;\n let concatSupported = true;\n async function* inputGenerator() {\n yield input;\n }\n try {\n let generator = steps[0].transform(\n inputGenerator(),\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : \"seq:step:1\"\n ),\n })\n );\n for (let i = 1; i < steps.length; i += 1) {\n generator = steps[i].transform(\n generator,\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${i + 1}`\n ),\n })\n );\n }\n for await (const chunk of generator) {\n config.signal?.throwIfAborted();\n yield chunk as RunOutput;\n if (concatSupported) {\n if (output === undefined) output = chunk;\n else {\n try {\n output = this._concatOutputChunks(output, chunk);\n } catch {\n output = undefined;\n concatSupported = false;\n }\n }\n }\n }\n } catch (error) {\n await manager?.handleChainError(error);\n throw error;\n }\n await this.endTrace(output, manager);\n }\n}\n"],"mappings":";;;AAYA,SAAS,aACP,OACA,WACS;AACT,KAAI,cAAc,KAAA,EAAW,QAAO;AACpC,KAAI;AACF,SAAO,UAAU,MAAM;UAChB,OAAO;AACd,UAAQ,KACN,wEACA,MACD;AACD,SAAO;;;;;;;;;AAUX,IAAa,cAAb,cAGUA,0BAAAA,iBAAsC;CAC9C;CAEA,YACE,QAKA;EACA,MAAM,EAAE,aAAa,GAAG,mBAAmB;AAC3C,QAAM,eAAe;AACrB,OAAK,cAAc;;CAGrB,MAAc,WAAW,OAAiB,QAAwB;EAEhE,MAAM,UAAU,aAAa,OAAO,KAAK,aAAa,cAAc;AAEpE,UADgB,OAAA,GAAA,0BAAA,6BAAkC,OAAO,GACzC,iBACd,KAAK,QAAQ,EACbC,cAAAA,cAAc,SAAS,QAAQ,EAC/B,OAAO,OACP,KAAA,GACA,KAAA,GACA,KAAA,GACA,OAAO,QACR;;CAGH,MAAc,SACZ,QACA,SACA;EACA,MAAM,UAAU,aAAa,QAAQ,KAAK,aAAa,eAAe;AACtE,QAAM,SAAS,eAAeA,cAAAA,cAAc,SAAS,SAAS,CAAC;;CAGjE,MAAe,OACb,OACA,SACoB;EACpB,MAAM,UAAA,GAAA,0BAAA,cAAsB,QAAQ;EACpC,MAAM,UAAU,MAAM,KAAK,WAAW,OAAO,OAAO;AACpD,SAAO,OAAO;EACd,IAAI,YAAY;EAChB,IAAI;AACJ,MAAI;GACF,MAAM,QAAQ,CAAC,KAAK,OAAO,GAAG,KAAK,OAAO;AAC1C,QAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,EACrC,aAAY,OAAA,GAAA,0BAAA,gBACV,MAAM,GAAG,OACP,YAAA,GAAA,0BAAA,aACY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,IAAI,IACrD,EACF,CAAC,CACH,EACD,OAAO,OACR;AAEH,OAAI,OAAO,QAAQ,QAEjB,QAAA,GAAA,0BAAA,gBAAqB,QAAQ,SAAS,EAAE,OAAO,OAAO;AAExD,YAAS,MAAM,KAAK,KAAK,OACvB,YAAA,GAAA,0BAAA,aACY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,KAAK,MAAM,SAC5D,EACF,CAAC,CACH;WACM,OAAO;AACd,SAAM,SAAS,iBAAiB,MAAM;AACtC,SAAM;;AAER,QAAM,KAAK,SAAS,QAAQ,QAAQ;AACpC,SAAO;;CAkBT,MAAe,MACb,QACA,SACA,cACgC;EAChC,MAAM,aAAa,KAAK,gBAAgB,WAAW,EAAE,EAAE,OAAO,OAAO;EACrE,MAAM,WAAW,MAAM,QAAQ,IAC7B,WAAW,IAAI,OAAO,QAAQ,MAAM;GAClC,MAAM,UAAU,MAAM,KAAK,WAAW,OAAO,IAAI,OAAO;AACxD,UAAO,OAAO;AACd,UAAO;IACP,CACH;EACD,IAAI,UAAqB;AACzB,MAAI;AAGF,QAAK,IAAI,IAAI,GAAG,IAAI,KAAK,MAAM,QAAQ,KAAK,EAC1C,WAAU,OAAA,GAAA,0BAAA,gBACR,KAAK,MAAM,GAAG,MACZ,SACA,SAAS,KAAK,SAAS,OAAA,GAAA,0BAAA,aACT,WAAW,IAAI,EACzB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,IAAI,IACrD,EACF,CAAC,CACH,EACD,aACD,EACD,WAAW,IAAI,OAChB;WAEI,OAAO;AACd,SAAM,QAAQ,IACZ,SAAS,KAAK,YAAY,SAAS,iBAAiB,MAAM,CAAC,CAC5D;AACD,SAAM;;AAER,QAAM,QAAQ,IACZ,SAAS,KAAK,SAAS,MAAM,KAAK,SAAS,QAAQ,IAAI,QAAQ,CAAC,CACjE;AACD,SAAO;;CAGT,OAAgB,gBAAgB,OAAiB,SAA0B;EACzE,MAAM,UAAA,GAAA,0BAAA,cAAsB,QAAQ;EACpC,MAAM,UAAU,MAAM,KAAK,WAAW,OAAO,OAAO;AACpD,SAAO,OAAO;EACd,MAAM,QAAQ,KAAK;EACnB,IAAI;EACJ,IAAI,kBAAkB;EACtB,gBAAgB,iBAAiB;AAC/B,SAAM;;AAER,MAAI;GACF,IAAI,YAAY,MAAM,GAAG,UACvB,gBAAgB,GAAA,GAAA,0BAAA,aACJ,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,aACrC,EACF,CAAC,CACH;AACD,QAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,EACrC,aAAY,MAAM,GAAG,UACnB,YAAA,GAAA,0BAAA,aACY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,IAAI,IACrD,EACF,CAAC,CACH;AAEH,cAAW,MAAM,SAAS,WAAW;AACnC,WAAO,QAAQ,gBAAgB;AAC/B,UAAM;AACN,QAAI,gBACF,KAAI,WAAW,KAAA,EAAW,UAAS;QAEjC,KAAI;AACF,cAAS,KAAK,oBAAoB,QAAQ,MAAM;YAC1C;AACN,cAAS,KAAA;AACT,uBAAkB;;;WAKnB,OAAO;AACd,SAAM,SAAS,iBAAiB,MAAM;AACtC,SAAM;;AAER,QAAM,KAAK,SAAS,QAAQ,QAAQ"}
|
|
@@ -0,0 +1,103 @@
|
|
|
1
|
+
import { _coerceToDict } from "./utils/index.js";
|
|
2
|
+
import { RunnableSequence, ensureConfig, getCallbackManagerForConfig, patchConfig, raceWithSignal } from "@langchain/core/runnables";
|
|
3
|
+
//#region src/pregel/runnable.ts
|
|
4
|
+
function tracePayload(value, processor) {
|
|
5
|
+
if (processor === void 0) return value;
|
|
6
|
+
try {
|
|
7
|
+
return processor(value);
|
|
8
|
+
} catch (error) {
|
|
9
|
+
console.warn("Trace input/output processor raised; recording untransformed payload", error);
|
|
10
|
+
return value;
|
|
11
|
+
}
|
|
12
|
+
}
|
|
13
|
+
/**
|
|
14
|
+
* A node sequence with transforms at its own tracing boundaries. Keep execution
|
|
15
|
+
* and child callback propagation aligned with Core's RunnableSequence; only the
|
|
16
|
+
* values passed to handleChainStart/handleChainEnd are transformed.
|
|
17
|
+
* @internal
|
|
18
|
+
*/
|
|
19
|
+
var RunnableSeq = class extends RunnableSequence {
|
|
20
|
+
tracePolicy;
|
|
21
|
+
constructor(fields) {
|
|
22
|
+
const { tracePolicy, ...sequenceFields } = fields;
|
|
23
|
+
super(sequenceFields);
|
|
24
|
+
this.tracePolicy = tracePolicy;
|
|
25
|
+
}
|
|
26
|
+
async startTrace(input, config) {
|
|
27
|
+
const payload = tracePayload(input, this.tracePolicy?.processInputs);
|
|
28
|
+
return (await getCallbackManagerForConfig(config))?.handleChainStart(this.toJSON(), _coerceToDict(payload, "input"), config.runId, void 0, void 0, void 0, config.runName);
|
|
29
|
+
}
|
|
30
|
+
async endTrace(output, manager) {
|
|
31
|
+
const payload = tracePayload(output, this.tracePolicy?.processOutputs);
|
|
32
|
+
await manager?.handleChainEnd(_coerceToDict(payload, "output"));
|
|
33
|
+
}
|
|
34
|
+
async invoke(input, options) {
|
|
35
|
+
const config = ensureConfig(options);
|
|
36
|
+
const manager = await this.startTrace(input, config);
|
|
37
|
+
delete config.runId;
|
|
38
|
+
let nextInput = input;
|
|
39
|
+
let output;
|
|
40
|
+
try {
|
|
41
|
+
const steps = [this.first, ...this.middle];
|
|
42
|
+
for (let i = 0; i < steps.length; i += 1) nextInput = await raceWithSignal(steps[i].invoke(nextInput, patchConfig(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) })), config.signal);
|
|
43
|
+
if (config.signal?.aborted) await raceWithSignal(Promise.resolve(), config.signal);
|
|
44
|
+
output = await this.last.invoke(nextInput, patchConfig(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${this.steps.length}`) }));
|
|
45
|
+
} catch (error) {
|
|
46
|
+
await manager?.handleChainError(error);
|
|
47
|
+
throw error;
|
|
48
|
+
}
|
|
49
|
+
await this.endTrace(output, manager);
|
|
50
|
+
return output;
|
|
51
|
+
}
|
|
52
|
+
async batch(inputs, options, batchOptions) {
|
|
53
|
+
const configList = this._getOptionsList(options ?? {}, inputs.length);
|
|
54
|
+
const managers = await Promise.all(configList.map(async (config, i) => {
|
|
55
|
+
const manager = await this.startTrace(inputs[i], config);
|
|
56
|
+
delete config.runId;
|
|
57
|
+
return manager;
|
|
58
|
+
}));
|
|
59
|
+
let outputs = inputs;
|
|
60
|
+
try {
|
|
61
|
+
for (let i = 0; i < this.steps.length; i += 1) outputs = await raceWithSignal(this.steps[i].batch(outputs, managers.map((manager, j) => patchConfig(configList[j], { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) })), batchOptions), configList[0]?.signal);
|
|
62
|
+
} catch (error) {
|
|
63
|
+
await Promise.all(managers.map((manager) => manager?.handleChainError(error)));
|
|
64
|
+
throw error;
|
|
65
|
+
}
|
|
66
|
+
await Promise.all(managers.map((manager, i) => this.endTrace(outputs[i], manager)));
|
|
67
|
+
return outputs;
|
|
68
|
+
}
|
|
69
|
+
async *_streamIterator(input, options) {
|
|
70
|
+
const config = ensureConfig(options);
|
|
71
|
+
const manager = await this.startTrace(input, config);
|
|
72
|
+
delete config.runId;
|
|
73
|
+
const steps = this.steps;
|
|
74
|
+
let output;
|
|
75
|
+
let concatSupported = true;
|
|
76
|
+
async function* inputGenerator() {
|
|
77
|
+
yield input;
|
|
78
|
+
}
|
|
79
|
+
try {
|
|
80
|
+
let generator = steps[0].transform(inputGenerator(), patchConfig(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : "seq:step:1") }));
|
|
81
|
+
for (let i = 1; i < steps.length; i += 1) generator = steps[i].transform(generator, patchConfig(config, { callbacks: manager?.getChild(this.omitSequenceTags ? void 0 : `seq:step:${i + 1}`) }));
|
|
82
|
+
for await (const chunk of generator) {
|
|
83
|
+
config.signal?.throwIfAborted();
|
|
84
|
+
yield chunk;
|
|
85
|
+
if (concatSupported) if (output === void 0) output = chunk;
|
|
86
|
+
else try {
|
|
87
|
+
output = this._concatOutputChunks(output, chunk);
|
|
88
|
+
} catch {
|
|
89
|
+
output = void 0;
|
|
90
|
+
concatSupported = false;
|
|
91
|
+
}
|
|
92
|
+
}
|
|
93
|
+
} catch (error) {
|
|
94
|
+
await manager?.handleChainError(error);
|
|
95
|
+
throw error;
|
|
96
|
+
}
|
|
97
|
+
await this.endTrace(output, manager);
|
|
98
|
+
}
|
|
99
|
+
};
|
|
100
|
+
//#endregion
|
|
101
|
+
export { RunnableSeq };
|
|
102
|
+
|
|
103
|
+
//# sourceMappingURL=runnable.js.map
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
{"version":3,"file":"runnable.js","names":[],"sources":["../../src/pregel/runnable.ts"],"sourcesContent":["import type { CallbackManagerForChainRun } from \"@langchain/core/callbacks/manager\";\nimport {\n ensureConfig,\n getCallbackManagerForConfig,\n patchConfig,\n raceWithSignal,\n RunnableSequence,\n type RunnableBatchOptions,\n type RunnableConfig,\n} from \"@langchain/core/runnables\";\nimport { _coerceToDict, type TracePolicy } from \"./utils/index.js\";\n\nfunction tracePayload(\n value: unknown,\n processor: ((value: unknown) => unknown) | undefined\n): unknown {\n if (processor === undefined) return value;\n try {\n return processor(value);\n } catch (error) {\n console.warn(\n \"Trace input/output processor raised; recording untransformed payload\",\n error\n );\n return value;\n }\n}\n\n/**\n * A node sequence with transforms at its own tracing boundaries. Keep execution\n * and child callback propagation aligned with Core's RunnableSequence; only the\n * values passed to handleChainStart/handleChainEnd are transformed.\n * @internal\n */\nexport class RunnableSeq<\n RunInput = unknown,\n RunOutput = unknown,\n> extends RunnableSequence<RunInput, RunOutput> {\n private tracePolicy?: TracePolicy;\n\n constructor(\n fields: ConstructorParameters<\n typeof RunnableSequence<RunInput, RunOutput>\n >[0] & {\n tracePolicy?: TracePolicy;\n }\n ) {\n const { tracePolicy, ...sequenceFields } = fields;\n super(sequenceFields);\n this.tracePolicy = tracePolicy;\n }\n\n private async startTrace(input: RunInput, config: RunnableConfig) {\n // Evaluate outside optional chaining: processors run without callbacks too.\n const payload = tracePayload(input, this.tracePolicy?.processInputs);\n const manager = await getCallbackManagerForConfig(config);\n return manager?.handleChainStart(\n this.toJSON(),\n _coerceToDict(payload, \"input\"),\n config.runId,\n undefined,\n undefined,\n undefined,\n config.runName\n );\n }\n\n private async endTrace(\n output: unknown,\n manager: CallbackManagerForChainRun | undefined\n ) {\n const payload = tracePayload(output, this.tracePolicy?.processOutputs);\n await manager?.handleChainEnd(_coerceToDict(payload, \"output\"));\n }\n\n override async invoke(\n input: RunInput,\n options?: RunnableConfig\n ): Promise<RunOutput> {\n const config = ensureConfig(options);\n const manager = await this.startTrace(input, config);\n delete config.runId;\n let nextInput = input;\n let output: RunOutput;\n try {\n const steps = [this.first, ...this.middle];\n for (let i = 0; i < steps.length; i += 1) {\n nextInput = await raceWithSignal(\n steps[i].invoke(\n nextInput,\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${i + 1}`\n ),\n })\n ),\n config.signal\n );\n }\n if (config.signal?.aborted) {\n // Use Core's abort-error normalization before invoking the last step.\n await raceWithSignal(Promise.resolve(), config.signal);\n }\n output = await this.last.invoke(\n nextInput,\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${this.steps.length}`\n ),\n })\n );\n } catch (error) {\n await manager?.handleChainError(error);\n throw error;\n }\n await this.endTrace(output, manager);\n return output;\n }\n\n override batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions & { returnExceptions?: false }\n ): Promise<RunOutput[]>;\n override batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions & { returnExceptions: true }\n ): Promise<(RunOutput | Error)[]>;\n override batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions\n ): Promise<(RunOutput | Error)[]>;\n override async batch(\n inputs: RunInput[],\n options?: Partial<RunnableConfig> | Partial<RunnableConfig>[],\n batchOptions?: RunnableBatchOptions\n ): Promise<(RunOutput | Error)[]> {\n const configList = this._getOptionsList(options ?? {}, inputs.length);\n const managers = await Promise.all(\n configList.map(async (config, i) => {\n const manager = await this.startTrace(inputs[i], config);\n delete config.runId;\n return manager;\n })\n );\n let outputs: unknown[] = inputs;\n try {\n // Preserve RunnableSequence's step-wise batching so child runnables can\n // use their native batch implementations and handle returnExceptions.\n for (let i = 0; i < this.steps.length; i += 1) {\n outputs = await raceWithSignal(\n this.steps[i].batch(\n outputs,\n managers.map((manager, j) =>\n patchConfig(configList[j], {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${i + 1}`\n ),\n })\n ),\n batchOptions\n ),\n configList[0]?.signal\n );\n }\n } catch (error) {\n await Promise.all(\n managers.map((manager) => manager?.handleChainError(error))\n );\n throw error;\n }\n await Promise.all(\n managers.map((manager, i) => this.endTrace(outputs[i], manager))\n );\n return outputs as (RunOutput | Error)[];\n }\n\n override async *_streamIterator(input: RunInput, options?: RunnableConfig) {\n const config = ensureConfig(options);\n const manager = await this.startTrace(input, config);\n delete config.runId;\n const steps = this.steps;\n let output: RunOutput | undefined;\n let concatSupported = true;\n async function* inputGenerator() {\n yield input;\n }\n try {\n let generator = steps[0].transform(\n inputGenerator(),\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : \"seq:step:1\"\n ),\n })\n );\n for (let i = 1; i < steps.length; i += 1) {\n generator = steps[i].transform(\n generator,\n patchConfig(config, {\n callbacks: manager?.getChild(\n this.omitSequenceTags ? undefined : `seq:step:${i + 1}`\n ),\n })\n );\n }\n for await (const chunk of generator) {\n config.signal?.throwIfAborted();\n yield chunk as RunOutput;\n if (concatSupported) {\n if (output === undefined) output = chunk;\n else {\n try {\n output = this._concatOutputChunks(output, chunk);\n } catch {\n output = undefined;\n concatSupported = false;\n }\n }\n }\n }\n } catch (error) {\n await manager?.handleChainError(error);\n throw error;\n }\n await this.endTrace(output, manager);\n }\n}\n"],"mappings":";;;AAYA,SAAS,aACP,OACA,WACS;AACT,KAAI,cAAc,KAAA,EAAW,QAAO;AACpC,KAAI;AACF,SAAO,UAAU,MAAM;UAChB,OAAO;AACd,UAAQ,KACN,wEACA,MACD;AACD,SAAO;;;;;;;;;AAUX,IAAa,cAAb,cAGU,iBAAsC;CAC9C;CAEA,YACE,QAKA;EACA,MAAM,EAAE,aAAa,GAAG,mBAAmB;AAC3C,QAAM,eAAe;AACrB,OAAK,cAAc;;CAGrB,MAAc,WAAW,OAAiB,QAAwB;EAEhE,MAAM,UAAU,aAAa,OAAO,KAAK,aAAa,cAAc;AAEpE,UADgB,MAAM,4BAA4B,OAAO,GACzC,iBACd,KAAK,QAAQ,EACb,cAAc,SAAS,QAAQ,EAC/B,OAAO,OACP,KAAA,GACA,KAAA,GACA,KAAA,GACA,OAAO,QACR;;CAGH,MAAc,SACZ,QACA,SACA;EACA,MAAM,UAAU,aAAa,QAAQ,KAAK,aAAa,eAAe;AACtE,QAAM,SAAS,eAAe,cAAc,SAAS,SAAS,CAAC;;CAGjE,MAAe,OACb,OACA,SACoB;EACpB,MAAM,SAAS,aAAa,QAAQ;EACpC,MAAM,UAAU,MAAM,KAAK,WAAW,OAAO,OAAO;AACpD,SAAO,OAAO;EACd,IAAI,YAAY;EAChB,IAAI;AACJ,MAAI;GACF,MAAM,QAAQ,CAAC,KAAK,OAAO,GAAG,KAAK,OAAO;AAC1C,QAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,EACrC,aAAY,MAAM,eAChB,MAAM,GAAG,OACP,WACA,YAAY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,IAAI,IACrD,EACF,CAAC,CACH,EACD,OAAO,OACR;AAEH,OAAI,OAAO,QAAQ,QAEjB,OAAM,eAAe,QAAQ,SAAS,EAAE,OAAO,OAAO;AAExD,YAAS,MAAM,KAAK,KAAK,OACvB,WACA,YAAY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,KAAK,MAAM,SAC5D,EACF,CAAC,CACH;WACM,OAAO;AACd,SAAM,SAAS,iBAAiB,MAAM;AACtC,SAAM;;AAER,QAAM,KAAK,SAAS,QAAQ,QAAQ;AACpC,SAAO;;CAkBT,MAAe,MACb,QACA,SACA,cACgC;EAChC,MAAM,aAAa,KAAK,gBAAgB,WAAW,EAAE,EAAE,OAAO,OAAO;EACrE,MAAM,WAAW,MAAM,QAAQ,IAC7B,WAAW,IAAI,OAAO,QAAQ,MAAM;GAClC,MAAM,UAAU,MAAM,KAAK,WAAW,OAAO,IAAI,OAAO;AACxD,UAAO,OAAO;AACd,UAAO;IACP,CACH;EACD,IAAI,UAAqB;AACzB,MAAI;AAGF,QAAK,IAAI,IAAI,GAAG,IAAI,KAAK,MAAM,QAAQ,KAAK,EAC1C,WAAU,MAAM,eACd,KAAK,MAAM,GAAG,MACZ,SACA,SAAS,KAAK,SAAS,MACrB,YAAY,WAAW,IAAI,EACzB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,IAAI,IACrD,EACF,CAAC,CACH,EACD,aACD,EACD,WAAW,IAAI,OAChB;WAEI,OAAO;AACd,SAAM,QAAQ,IACZ,SAAS,KAAK,YAAY,SAAS,iBAAiB,MAAM,CAAC,CAC5D;AACD,SAAM;;AAER,QAAM,QAAQ,IACZ,SAAS,KAAK,SAAS,MAAM,KAAK,SAAS,QAAQ,IAAI,QAAQ,CAAC,CACjE;AACD,SAAO;;CAGT,OAAgB,gBAAgB,OAAiB,SAA0B;EACzE,MAAM,SAAS,aAAa,QAAQ;EACpC,MAAM,UAAU,MAAM,KAAK,WAAW,OAAO,OAAO;AACpD,SAAO,OAAO;EACd,MAAM,QAAQ,KAAK;EACnB,IAAI;EACJ,IAAI,kBAAkB;EACtB,gBAAgB,iBAAiB;AAC/B,SAAM;;AAER,MAAI;GACF,IAAI,YAAY,MAAM,GAAG,UACvB,gBAAgB,EAChB,YAAY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,aACrC,EACF,CAAC,CACH;AACD,QAAK,IAAI,IAAI,GAAG,IAAI,MAAM,QAAQ,KAAK,EACrC,aAAY,MAAM,GAAG,UACnB,WACA,YAAY,QAAQ,EAClB,WAAW,SAAS,SAClB,KAAK,mBAAmB,KAAA,IAAY,YAAY,IAAI,IACrD,EACF,CAAC,CACH;AAEH,cAAW,MAAM,SAAS,WAAW;AACnC,WAAO,QAAQ,gBAAgB;AAC/B,UAAM;AACN,QAAI,gBACF,KAAI,WAAW,KAAA,EAAW,UAAS;QAEjC,KAAI;AACF,cAAS,KAAK,oBAAoB,QAAQ,MAAM;YAC1C;AACN,cAAS,KAAA;AACT,uBAAkB;;;WAKnB,OAAO;AACd,SAAM,SAAS,iBAAiB,MAAM;AACtC,SAAM;;AAER,QAAM,KAAK,SAAS,QAAQ,QAAQ"}
|
package/dist/pregel/runner.cjs
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
const require_constants = require("../constants.cjs");
|
|
2
2
|
const require_errors = require("../errors.cjs");
|
|
3
|
-
const require_types = require("./types.cjs");
|
|
4
3
|
const require_index = require("./utils/index.cjs");
|
|
4
|
+
const require_types = require("./types.cjs");
|
|
5
5
|
const require_retry = require("./retry.cjs");
|
|
6
6
|
//#region src/pregel/runner.ts
|
|
7
7
|
const PROMISE_ADDED_SYMBOL = Symbol.for("promiseAdded");
|
package/dist/pregel/runner.js
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
import { CONFIG_KEY_ABORT_SIGNALS, CONFIG_KEY_CALL, CONFIG_KEY_SCRATCHPAD, ERROR, ERROR_SOURCE_NODE, INTERRUPT, NO_WRITES, RESUME, RETURN } from "../constants.js";
|
|
2
2
|
import { isGraphBubbleUp, isGraphDrained, isGraphInterrupt } from "../errors.js";
|
|
3
|
-
import { Call } from "./types.js";
|
|
4
3
|
import { combineAbortSignals, patchConfigurable } from "./utils/index.js";
|
|
4
|
+
import { Call } from "./types.js";
|
|
5
5
|
import { _runWithRetry } from "./retry.js";
|
|
6
6
|
//#region src/pregel/runner.ts
|
|
7
7
|
const PROMISE_ADDED_SYMBOL = Symbol.for("promiseAdded");
|
|
@@ -91,9 +91,10 @@ function mergeCallbacks(base, provided) {
|
|
|
91
91
|
for (const callback of base) manager.addHandler((0, _langchain_core_callbacks_manager.ensureHandler)(callback), true);
|
|
92
92
|
return manager;
|
|
93
93
|
}
|
|
94
|
+
const dedupeHandlers = (handlers) => handlers.filter((handler, index) => handlers.findIndex((candidate) => candidate === handler || candidate.name === "StreamMessagesHandler" && handler.name === "StreamMessagesHandler") === index);
|
|
94
95
|
return new _langchain_core_callbacks_manager.CallbackManager(provided._parentRunId, {
|
|
95
|
-
handlers:
|
|
96
|
-
inheritableHandlers:
|
|
96
|
+
handlers: dedupeHandlers(base.handlers.concat(provided.handlers)),
|
|
97
|
+
inheritableHandlers: dedupeHandlers(base.inheritableHandlers.concat(provided.inheritableHandlers)),
|
|
97
98
|
tags: Array.from(new Set(base.tags.concat(provided.tags))),
|
|
98
99
|
inheritableTags: Array.from(new Set(base.inheritableTags.concat(provided.inheritableTags))),
|
|
99
100
|
metadata: {
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.cjs","names":["CallbackManager","CONFIG_KEY_READ","AsyncLocalStorageProviderSingleton","CONFIG_KEY_SCRATCHPAD"],"sources":["../../../src/pregel/utils/config.ts"],"sourcesContent":["import { RunnableConfig } from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\nimport {\n CallbackManager,\n ensureHandler,\n type Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { BaseStore } from \"@langchain/langgraph-checkpoint\";\nimport { LangGraphRunnableConfig } from \"../runnable_types.js\";\nimport {\n CHECKPOINT_NAMESPACE_END,\n CHECKPOINT_NAMESPACE_SEPARATOR,\n CONFIG_KEY_READ,\n CONFIG_KEY_SCRATCHPAD,\n} from \"../../constants.js\";\n\nconst COPIABLE_KEYS = [\"tags\", \"metadata\", \"callbacks\", \"configurable\"];\n\nconst CONFIG_KEYS = [\n \"tags\",\n \"metadata\",\n \"callbacks\",\n \"runName\",\n \"maxConcurrency\",\n \"recursionLimit\",\n \"configurable\",\n \"runId\",\n \"outputKeys\",\n \"streamMode\",\n \"store\",\n \"writer\",\n \"interrupt\",\n \"context\",\n \"interruptBefore\",\n \"interruptAfter\",\n \"checkpointDuring\",\n \"durability\",\n \"signal\",\n \"heartbeat\",\n \"executionInfo\",\n \"serverInfo\",\n \"control\",\n];\n\nconst DEFAULT_RECURSION_LIMIT = 25;\nexport const PROPAGATE_TO_METADATA = new Set([\n \"thread_id\",\n \"checkpoint_id\",\n \"checkpoint_ns\",\n \"task_id\",\n \"run_id\",\n \"assistant_id\",\n \"graph_id\",\n]);\n\nexport function propagateConfigurableToMetadata(\n configurable?: Record<string, unknown>,\n metadata?: Record<string, unknown>\n): Record<string, unknown> | undefined {\n if (!configurable) {\n return metadata;\n }\n const result = metadata ?? {};\n for (const key of PROPAGATE_TO_METADATA) {\n if (key in result) {\n continue;\n }\n const value = configurable[key];\n if (value !== undefined) {\n result[key] = value;\n }\n }\n return result;\n}\n\n/**\n * Drop langgraph's internal `seq:step*` bookkeeping tags.\n *\n * `seq:step:N` tags are added internally to mark sequence steps; everything\n * else (user-supplied tags and any other framework tags) is kept. Returns the\n * surviving tags, or `undefined` if none remain. Shared by the stream handlers\n * (e.g. {@link mapDebugTasks}) so the same tag set is surfaced consistently.\n */\nexport function filterToUserTags(\n tags: readonly string[] | undefined\n): string[] | undefined {\n if (tags == null || tags.length === 0) return undefined;\n const filtered = tags.filter((tag) => !tag.startsWith(\"seq:step\"));\n return filtered.length > 0 ? filtered : undefined;\n}\n\n/**\n * Merge two `callbacks` values across configs.\n *\n * A `callbacks` value may be `undefined`, an array of handlers, or a\n * {@link CallbackManager}, so merging two of them has six cases. This\n * mirrors the callbacks branch of langchain-core's `mergeConfigs` and\n * langgraph's `_merge_callbacks`, so a handler bound via\n * `.withConfig({ callbacks: [...] })` is preserved when a later config\n * (e.g. `streamEvents` injecting its own internal handler) is merged on\n * top instead of overwriting it.\n */\nfunction mergeCallbacks(\n base: Callbacks | undefined,\n provided: Callbacks | undefined\n): Callbacks | undefined {\n if (provided === undefined) return base;\n if (base === undefined) {\n return Array.isArray(provided) ? [...provided] : provided.copy();\n }\n if (Array.isArray(provided)) {\n if (Array.isArray(base)) return base.concat(provided);\n // base is a manager\n const manager = base.copy();\n for (const callback of provided) {\n manager.addHandler(ensureHandler(callback), true);\n }\n return manager;\n }\n // provided is a manager\n if (Array.isArray(base)) {\n const manager = provided.copy();\n for (const callback of base) {\n manager.addHandler(ensureHandler(callback), true);\n }\n return manager;\n }\n // both are managers\n return new CallbackManager(provided._parentRunId, {\n handlers: Array.from(new Set(base.handlers.concat(provided.handlers))),\n inheritableHandlers: Array.from(\n new Set(base.inheritableHandlers.concat(provided.inheritableHandlers))\n ),\n tags: Array.from(new Set(base.tags.concat(provided.tags))),\n inheritableTags: Array.from(\n new Set(base.inheritableTags.concat(provided.inheritableTags))\n ),\n metadata: { ...base.metadata, ...provided.metadata },\n inheritableMetadata: {\n ...base.inheritableMetadata,\n ...provided.inheritableMetadata,\n },\n });\n}\n\n/**\n * True when the caller is starting a fresh top-level run (invoke-time\n * `thread_id`, no active nesting keys). In that case the ambient `configurable`\n * from `AsyncLocalStorage` cannot be trusted per-key — it may belong to another\n * concurrent invocation on a shared singleton agent (scratchpad/task-input as\n * well as arbitrary user keys like `tenant_id`/`user_id`). The whole ambient\n * `configurable` is therefore ignored; any value the caller actually wants for\n * this run arrives through the explicit (bound + invoke-time) configs instead.\n *\n * Only the last caller-supplied config is treated as invoke-time options.\n * Earlier entries are graph-bound defaults from `.withConfig()` / compile and\n * must not count — a child graph bound with `thread_id` and invoked from a\n * parent task without a fresh config still needs ambient nesting keys from ALS.\n */\nfunction isRootLevelExplicitInvoke(\n configs: (LangGraphRunnableConfig | undefined)[]\n): boolean {\n let invokeConfig: LangGraphRunnableConfig | undefined;\n for (let i = configs.length - 1; i >= 0; i -= 1) {\n if (configs[i] !== undefined) {\n invokeConfig = configs[i];\n break;\n }\n }\n const hasInvokeTimeThreadId =\n invokeConfig?.configurable?.thread_id !== undefined;\n\n const hasExplicitNesting = configs.some(\n (c) => c?.configurable?.[CONFIG_KEY_READ] !== undefined\n );\n\n const ambientConfigurable =\n AsyncLocalStorageProviderSingleton.getRunnableConfig()?.configurable;\n const hasAmbientNesting =\n ambientConfigurable?.[CONFIG_KEY_READ] !== undefined;\n\n return hasInvokeTimeThreadId && !hasExplicitNesting && !hasAmbientNesting;\n}\n\nexport function ensureLangGraphConfig(\n ...configs: (LangGraphRunnableConfig | undefined)[]\n): RunnableConfig {\n const empty: LangGraphRunnableConfig = {\n tags: [],\n metadata: {},\n callbacks: undefined,\n recursionLimit: DEFAULT_RECURSION_LIMIT,\n configurable: {},\n };\n\n const skipImplicitConfigurable = isRootLevelExplicitInvoke(configs);\n\n const implicitConfig: RunnableConfig =\n AsyncLocalStorageProviderSingleton.getRunnableConfig();\n if (implicitConfig !== undefined) {\n for (const [k, v] of Object.entries(implicitConfig)) {\n if (v !== undefined) {\n // Stale-ALS root invoke: ignore the ambient `configurable` entirely,\n // since its keys may belong to another concurrent invocation.\n if (k === \"configurable\" && skipImplicitConfigurable) {\n continue;\n }\n if (COPIABLE_KEYS.includes(k)) {\n let copiedValue;\n if (Array.isArray(v)) {\n copiedValue = [...v];\n } else if (typeof v === \"object\") {\n if (\n k === \"callbacks\" &&\n \"copy\" in v &&\n typeof v.copy === \"function\"\n ) {\n copiedValue = v.copy();\n } else {\n copiedValue = { ...v };\n }\n } else {\n copiedValue = v;\n }\n empty[k as keyof RunnableConfig] = copiedValue;\n } else {\n empty[k as keyof RunnableConfig] = v;\n }\n }\n }\n }\n\n for (const config of configs) {\n if (config === undefined) {\n continue;\n }\n\n for (const [k, v] of Object.entries(config)) {\n if (v === undefined || !CONFIG_KEYS.includes(k)) {\n continue;\n }\n\n // Merge (rather than overwrite) the fields that compose across\n // multiple configs, matching langgraph's `merge_configs` /\n // langchain-core's `mergeConfigs`. Without this, a value bound via\n // `.withConfig({...})` is silently dropped whenever a later config\n // (e.g. invoke-time, or `streamEvents`' internal handler) supplies\n // any other key in the same field. Later configs still win per key\n // on collision.\n if (k === \"configurable\") {\n empty.configurable = {\n ...(empty.configurable as Record<string, unknown> | undefined),\n ...(v as Record<string, unknown>),\n };\n } else if (k === \"metadata\") {\n empty.metadata = {\n ...(empty.metadata as Record<string, unknown> | undefined),\n ...(v as Record<string, unknown>),\n };\n } else if (k === \"tags\") {\n // Plain concat (matches langgraph's `merge_configs`): no dedup,\n // no sort.\n empty.tags = [\n ...((empty.tags as string[] | undefined) ?? []),\n ...(v as string[]),\n ];\n } else if (k === \"callbacks\") {\n empty.callbacks = mergeCallbacks(empty.callbacks, v as Callbacks);\n } else {\n empty[k as keyof LangGraphRunnableConfig] = v;\n }\n }\n }\n\n empty.metadata =\n propagateConfigurableToMetadata(\n empty.configurable as Record<string, unknown> | undefined,\n empty.metadata as Record<string, unknown> | undefined\n ) ?? {};\n return empty;\n}\n\n/**\n * A helper utility function that returns the {@link BaseStore} that was set when the graph was initialized\n *\n * @returns a reference to the {@link BaseStore} that was set when the graph was initialized\n */\nexport function getStore(\n config?: LangGraphRunnableConfig\n): BaseStore | undefined {\n const runConfig: LangGraphRunnableConfig =\n config ?? AsyncLocalStorageProviderSingleton.getRunnableConfig();\n\n if (runConfig === undefined) {\n throw new Error(\n [\n \"Config not retrievable. This is likely because you are running in an environment without support for AsyncLocalStorage.\",\n \"If you're running `getStore` in such environment, pass the `config` from the node function directly.\",\n ].join(\"\\n\")\n );\n }\n\n return runConfig?.store;\n}\n\n/**\n * A helper utility function that returns the {@link LangGraphRunnableConfig#writer} if \"custom\" stream mode is enabled, otherwise undefined.\n *\n * @returns a reference to the {@link LangGraphRunnableConfig#writer} if \"custom\" stream mode is enabled, otherwise undefined\n */\nexport function getWriter(\n config?: LangGraphRunnableConfig\n): ((chunk: unknown) => void) | undefined {\n const runConfig: LangGraphRunnableConfig =\n config ?? AsyncLocalStorageProviderSingleton.getRunnableConfig();\n\n if (runConfig === undefined) {\n throw new Error(\n [\n \"Config not retrievable. This is likely because you are running in an environment without support for AsyncLocalStorage.\",\n \"If you're running `getWriter` in such environment, pass the `config` from the node function directly.\",\n ].join(\"\\n\")\n );\n }\n\n return runConfig?.writer || runConfig?.configurable?.writer;\n}\n\n/**\n * A helper utility function that returns the {@link LangGraphRunnableConfig} that was set when the graph was initialized.\n *\n * Note: This only works when running in an environment that supports node:async_hooks and AsyncLocalStorage. If you're running this in a\n * web environment, access the LangGraphRunnableConfig from the node function directly.\n *\n * @returns the {@link LangGraphRunnableConfig} that was set when the graph was initialized\n */\nexport function getConfig(): LangGraphRunnableConfig {\n return AsyncLocalStorageProviderSingleton.getRunnableConfig();\n}\n\n/**\n * A helper utility function that returns the input for the currently executing\n * task.\n *\n * Note: When called without arguments, this relies on `node:async_hooks` /\n * `AsyncLocalStorage`, which is available in many JavaScript environments\n * (Node.js, Deno, Cloudflare Workers) but not in web browsers. In environments\n * without `AsyncLocalStorage` support, pass the `config` that your node/tool\n * function receives directly, e.g. `getCurrentTaskInput(config)`.\n *\n * Tip: Inside a tool run by a `ToolNode`, prefer reading graph state from\n * `runtime.state` on the second tool argument (typed as `ToolRuntime` from\n * `@langchain/core/tools`). It works in every runtime, including web browsers.\n *\n * @param config - Optional {@link LangGraphRunnableConfig} to read the task\n * input from. Provide this when running in an environment without\n * `AsyncLocalStorage` support (e.g. web browsers).\n * @returns the input for the currently executing task\n */\nexport function getCurrentTaskInput<T = unknown>(\n config?: LangGraphRunnableConfig\n): T {\n const runConfig: LangGraphRunnableConfig =\n config ?? AsyncLocalStorageProviderSingleton.getRunnableConfig();\n\n if (runConfig === undefined) {\n throw new Error(\n [\n \"Config not retrievable. This is likely because you are running in an environment without support for AsyncLocalStorage.\",\n \"If you're running `getCurrentTaskInput` in such environment, pass the `config` from the node function directly.\",\n ].join(\"\\n\")\n );\n }\n\n if (\n runConfig.configurable?.[CONFIG_KEY_SCRATCHPAD]?.currentTaskInput ===\n undefined\n ) {\n throw new Error(\"BUG: internal scratchpad not initialized.\");\n }\n\n return runConfig!.configurable![CONFIG_KEY_SCRATCHPAD]!.currentTaskInput as T;\n}\n\nexport function recastCheckpointNamespace(namespace: string): string {\n return namespace\n .split(CHECKPOINT_NAMESPACE_SEPARATOR)\n .filter((part) => !part.match(/^\\d+$/))\n .map((part) => part.split(CHECKPOINT_NAMESPACE_END)[0])\n .join(CHECKPOINT_NAMESPACE_SEPARATOR);\n}\n\nexport function getParentCheckpointNamespace(namespace: string): string {\n const parts = namespace.split(CHECKPOINT_NAMESPACE_SEPARATOR);\n while (parts.length > 1 && parts[parts.length - 1].match(/^\\d+$/)) {\n parts.pop();\n }\n return parts.slice(0, -1).join(CHECKPOINT_NAMESPACE_SEPARATOR);\n}\n"],"mappings":";;;;AAgBA,MAAM,gBAAgB;CAAC;CAAQ;CAAY;CAAa;CAAe;AAEvE,MAAM,cAAc;CAClB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD;AAED,MAAM,0BAA0B;AAChC,MAAa,wBAAwB,IAAI,IAAI;CAC3C;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAEF,SAAgB,gCACd,cACA,UACqC;AACrC,KAAI,CAAC,aACH,QAAO;CAET,MAAM,SAAS,YAAY,EAAE;AAC7B,MAAK,MAAM,OAAO,uBAAuB;AACvC,MAAI,OAAO,OACT;EAEF,MAAM,QAAQ,aAAa;AAC3B,MAAI,UAAU,KAAA,EACZ,QAAO,OAAO;;AAGlB,QAAO;;;;;;;;;;AAWT,SAAgB,iBACd,MACsB;AACtB,KAAI,QAAQ,QAAQ,KAAK,WAAW,EAAG,QAAO,KAAA;CAC9C,MAAM,WAAW,KAAK,QAAQ,QAAQ,CAAC,IAAI,WAAW,WAAW,CAAC;AAClE,QAAO,SAAS,SAAS,IAAI,WAAW,KAAA;;;;;;;;;;;;;AAc1C,SAAS,eACP,MACA,UACuB;AACvB,KAAI,aAAa,KAAA,EAAW,QAAO;AACnC,KAAI,SAAS,KAAA,EACX,QAAO,MAAM,QAAQ,SAAS,GAAG,CAAC,GAAG,SAAS,GAAG,SAAS,MAAM;AAElE,KAAI,MAAM,QAAQ,SAAS,EAAE;AAC3B,MAAI,MAAM,QAAQ,KAAK,CAAE,QAAO,KAAK,OAAO,SAAS;EAErD,MAAM,UAAU,KAAK,MAAM;AAC3B,OAAK,MAAM,YAAY,SACrB,SAAQ,YAAA,GAAA,kCAAA,eAAyB,SAAS,EAAE,KAAK;AAEnD,SAAO;;AAGT,KAAI,MAAM,QAAQ,KAAK,EAAE;EACvB,MAAM,UAAU,SAAS,MAAM;AAC/B,OAAK,MAAM,YAAY,KACrB,SAAQ,YAAA,GAAA,kCAAA,eAAyB,SAAS,EAAE,KAAK;AAEnD,SAAO;;AAGT,QAAO,IAAIA,kCAAAA,gBAAgB,SAAS,cAAc;EAChD,UAAU,MAAM,KAAK,IAAI,IAAI,KAAK,SAAS,OAAO,SAAS,SAAS,CAAC,CAAC;EACtE,qBAAqB,MAAM,KACzB,IAAI,IAAI,KAAK,oBAAoB,OAAO,SAAS,oBAAoB,CAAC,CACvE;EACD,MAAM,MAAM,KAAK,IAAI,IAAI,KAAK,KAAK,OAAO,SAAS,KAAK,CAAC,CAAC;EAC1D,iBAAiB,MAAM,KACrB,IAAI,IAAI,KAAK,gBAAgB,OAAO,SAAS,gBAAgB,CAAC,CAC/D;EACD,UAAU;GAAE,GAAG,KAAK;GAAU,GAAG,SAAS;GAAU;EACpD,qBAAqB;GACnB,GAAG,KAAK;GACR,GAAG,SAAS;GACb;EACF,CAAC;;;;;;;;;;;;;;;;AAiBJ,SAAS,0BACP,SACS;CACT,IAAI;AACJ,MAAK,IAAI,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK,EAC5C,KAAI,QAAQ,OAAO,KAAA,GAAW;AAC5B,iBAAe,QAAQ;AACvB;;CAGJ,MAAM,wBACJ,cAAc,cAAc,cAAc,KAAA;CAE5C,MAAM,qBAAqB,QAAQ,MAChC,MAAM,GAAG,eAAeC,kBAAAA,qBAAqB,KAAA,EAC/C;CAID,MAAM,qBADJC,2BAAAA,mCAAmC,mBAAmB,EAAE,gBAElCD,kBAAAA,qBAAqB,KAAA;AAE7C,QAAO,yBAAyB,CAAC,sBAAsB,CAAC;;AAG1D,SAAgB,sBACd,GAAG,SACa;CAChB,MAAM,QAAiC;EACrC,MAAM,EAAE;EACR,UAAU,EAAE;EACZ,WAAW,KAAA;EACX,gBAAgB;EAChB,cAAc,EAAE;EACjB;CAED,MAAM,2BAA2B,0BAA0B,QAAQ;CAEnE,MAAM,iBACJC,2BAAAA,mCAAmC,mBAAmB;AACxD,KAAI,mBAAmB,KAAA;OAChB,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ,eAAe,CACjD,KAAI,MAAM,KAAA,GAAW;AAGnB,OAAI,MAAM,kBAAkB,yBAC1B;AAEF,OAAI,cAAc,SAAS,EAAE,EAAE;IAC7B,IAAI;AACJ,QAAI,MAAM,QAAQ,EAAE,CAClB,eAAc,CAAC,GAAG,EAAE;aACX,OAAO,MAAM,SACtB,KACE,MAAM,eACN,UAAU,KACV,OAAO,EAAE,SAAS,WAElB,eAAc,EAAE,MAAM;QAEtB,eAAc,EAAE,GAAG,GAAG;QAGxB,eAAc;AAEhB,UAAM,KAA6B;SAEnC,OAAM,KAA6B;;;AAM3C,MAAK,MAAM,UAAU,SAAS;AAC5B,MAAI,WAAW,KAAA,EACb;AAGF,OAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ,OAAO,EAAE;AAC3C,OAAI,MAAM,KAAA,KAAa,CAAC,YAAY,SAAS,EAAE,CAC7C;AAUF,OAAI,MAAM,eACR,OAAM,eAAe;IACnB,GAAI,MAAM;IACV,GAAI;IACL;YACQ,MAAM,WACf,OAAM,WAAW;IACf,GAAI,MAAM;IACV,GAAI;IACL;YACQ,MAAM,OAGf,OAAM,OAAO,CACX,GAAK,MAAM,QAAiC,EAAE,EAC9C,GAAI,EACL;YACQ,MAAM,YACf,OAAM,YAAY,eAAe,MAAM,WAAW,EAAe;OAEjE,OAAM,KAAsC;;;AAKlD,OAAM,WACJ,gCACE,MAAM,cACN,MAAM,SACP,IAAI,EAAE;AACT,QAAO;;;;;;;AAQT,SAAgB,SACd,QACuB;CACvB,MAAM,YACJ,UAAUA,2BAAAA,mCAAmC,mBAAmB;AAElE,KAAI,cAAc,KAAA,EAChB,OAAM,IAAI,MACR,CACE,2HACA,uGACD,CAAC,KAAK,KAAK,CACb;AAGH,QAAO,WAAW;;;;;;;AAQpB,SAAgB,UACd,QACwC;CACxC,MAAM,YACJ,UAAUA,2BAAAA,mCAAmC,mBAAmB;AAElE,KAAI,cAAc,KAAA,EAChB,OAAM,IAAI,MACR,CACE,2HACA,wGACD,CAAC,KAAK,KAAK,CACb;AAGH,QAAO,WAAW,UAAU,WAAW,cAAc;;;;;;;;;;AAWvD,SAAgB,YAAqC;AACnD,QAAOA,2BAAAA,mCAAmC,mBAAmB;;;;;;;;;;;;;;;;;;;;;AAsB/D,SAAgB,oBACd,QACG;CACH,MAAM,YACJ,UAAUA,2BAAAA,mCAAmC,mBAAmB;AAElE,KAAI,cAAc,KAAA,EAChB,OAAM,IAAI,MACR,CACE,2HACA,kHACD,CAAC,KAAK,KAAK,CACb;AAGH,KACE,UAAU,eAAA,wBAAuC,qBACjD,KAAA,EAEA,OAAM,IAAI,MAAM,4CAA4C;AAG9D,QAAO,UAAW,aAAcC,kBAAAA,uBAAwB;;AAG1D,SAAgB,0BAA0B,WAA2B;AACnE,QAAO,UACJ,MAAA,IAAqC,CACrC,QAAQ,SAAS,CAAC,KAAK,MAAM,QAAQ,CAAC,CACtC,KAAK,SAAS,KAAK,MAAA,IAA+B,CAAC,GAAG,CACtD,KAAA,IAAoC;;AAGzC,SAAgB,6BAA6B,WAA2B;CACtE,MAAM,QAAQ,UAAU,MAAA,IAAqC;AAC7D,QAAO,MAAM,SAAS,KAAK,MAAM,MAAM,SAAS,GAAG,MAAM,QAAQ,CAC/D,OAAM,KAAK;AAEb,QAAO,MAAM,MAAM,GAAG,GAAG,CAAC,KAAA,IAAoC"}
|
|
1
|
+
{"version":3,"file":"config.cjs","names":["CallbackManager","CONFIG_KEY_READ","AsyncLocalStorageProviderSingleton","CONFIG_KEY_SCRATCHPAD"],"sources":["../../../src/pregel/utils/config.ts"],"sourcesContent":["import { RunnableConfig } from \"@langchain/core/runnables\";\nimport { AsyncLocalStorageProviderSingleton } from \"@langchain/core/singletons\";\nimport {\n CallbackManager,\n ensureHandler,\n type Callbacks,\n} from \"@langchain/core/callbacks/manager\";\nimport { BaseStore } from \"@langchain/langgraph-checkpoint\";\nimport { LangGraphRunnableConfig } from \"../runnable_types.js\";\nimport {\n CHECKPOINT_NAMESPACE_END,\n CHECKPOINT_NAMESPACE_SEPARATOR,\n CONFIG_KEY_READ,\n CONFIG_KEY_SCRATCHPAD,\n} from \"../../constants.js\";\n\nconst COPIABLE_KEYS = [\"tags\", \"metadata\", \"callbacks\", \"configurable\"];\n\nconst CONFIG_KEYS = [\n \"tags\",\n \"metadata\",\n \"callbacks\",\n \"runName\",\n \"maxConcurrency\",\n \"recursionLimit\",\n \"configurable\",\n \"runId\",\n \"outputKeys\",\n \"streamMode\",\n \"store\",\n \"writer\",\n \"interrupt\",\n \"context\",\n \"interruptBefore\",\n \"interruptAfter\",\n \"checkpointDuring\",\n \"durability\",\n \"signal\",\n \"heartbeat\",\n \"executionInfo\",\n \"serverInfo\",\n \"control\",\n];\n\nconst DEFAULT_RECURSION_LIMIT = 25;\nexport const PROPAGATE_TO_METADATA = new Set([\n \"thread_id\",\n \"checkpoint_id\",\n \"checkpoint_ns\",\n \"task_id\",\n \"run_id\",\n \"assistant_id\",\n \"graph_id\",\n]);\n\nexport function propagateConfigurableToMetadata(\n configurable?: Record<string, unknown>,\n metadata?: Record<string, unknown>\n): Record<string, unknown> | undefined {\n if (!configurable) {\n return metadata;\n }\n const result = metadata ?? {};\n for (const key of PROPAGATE_TO_METADATA) {\n if (key in result) {\n continue;\n }\n const value = configurable[key];\n if (value !== undefined) {\n result[key] = value;\n }\n }\n return result;\n}\n\n/**\n * Drop langgraph's internal `seq:step*` bookkeeping tags.\n *\n * `seq:step:N` tags are added internally to mark sequence steps; everything\n * else (user-supplied tags and any other framework tags) is kept. Returns the\n * surviving tags, or `undefined` if none remain. Shared by the stream handlers\n * (e.g. {@link mapDebugTasks}) so the same tag set is surfaced consistently.\n */\nexport function filterToUserTags(\n tags: readonly string[] | undefined\n): string[] | undefined {\n if (tags == null || tags.length === 0) return undefined;\n const filtered = tags.filter((tag) => !tag.startsWith(\"seq:step\"));\n return filtered.length > 0 ? filtered : undefined;\n}\n\n/**\n * Merge two `callbacks` values across configs.\n *\n * A `callbacks` value may be `undefined`, an array of handlers, or a\n * {@link CallbackManager}, so merging two of them has six cases. This\n * mirrors the callbacks branch of langchain-core's `mergeConfigs` and\n * langgraph's `_merge_callbacks`, so a handler bound via\n * `.withConfig({ callbacks: [...] })` is preserved when a later config\n * (e.g. `streamEvents` injecting its own internal handler) is merged on\n * top instead of overwriting it.\n */\nfunction mergeCallbacks(\n base: Callbacks | undefined,\n provided: Callbacks | undefined\n): Callbacks | undefined {\n if (provided === undefined) return base;\n if (base === undefined) {\n return Array.isArray(provided) ? [...provided] : provided.copy();\n }\n if (Array.isArray(provided)) {\n if (Array.isArray(base)) return base.concat(provided);\n // base is a manager\n const manager = base.copy();\n for (const callback of provided) {\n manager.addHandler(ensureHandler(callback), true);\n }\n return manager;\n }\n // provided is a manager\n if (Array.isArray(base)) {\n const manager = provided.copy();\n for (const callback of base) {\n manager.addHandler(ensureHandler(callback), true);\n }\n return manager;\n }\n // both are managers\n const dedupeHandlers = (\n handlers: CallbackManager[\"handlers\"]\n ): CallbackManager[\"handlers\"] =>\n handlers.filter(\n (handler, index) =>\n handlers.findIndex(\n (candidate) =>\n candidate === handler ||\n (candidate.name === \"StreamMessagesHandler\" &&\n handler.name === \"StreamMessagesHandler\")\n ) === index\n );\n return new CallbackManager(provided._parentRunId, {\n handlers: dedupeHandlers(base.handlers.concat(provided.handlers)),\n inheritableHandlers: dedupeHandlers(\n base.inheritableHandlers.concat(provided.inheritableHandlers)\n ),\n tags: Array.from(new Set(base.tags.concat(provided.tags))),\n inheritableTags: Array.from(\n new Set(base.inheritableTags.concat(provided.inheritableTags))\n ),\n metadata: { ...base.metadata, ...provided.metadata },\n inheritableMetadata: {\n ...base.inheritableMetadata,\n ...provided.inheritableMetadata,\n },\n });\n}\n\n/**\n * True when the caller is starting a fresh top-level run (invoke-time\n * `thread_id`, no active nesting keys). In that case the ambient `configurable`\n * from `AsyncLocalStorage` cannot be trusted per-key — it may belong to another\n * concurrent invocation on a shared singleton agent (scratchpad/task-input as\n * well as arbitrary user keys like `tenant_id`/`user_id`). The whole ambient\n * `configurable` is therefore ignored; any value the caller actually wants for\n * this run arrives through the explicit (bound + invoke-time) configs instead.\n *\n * Only the last caller-supplied config is treated as invoke-time options.\n * Earlier entries are graph-bound defaults from `.withConfig()` / compile and\n * must not count — a child graph bound with `thread_id` and invoked from a\n * parent task without a fresh config still needs ambient nesting keys from ALS.\n */\nfunction isRootLevelExplicitInvoke(\n configs: (LangGraphRunnableConfig | undefined)[]\n): boolean {\n let invokeConfig: LangGraphRunnableConfig | undefined;\n for (let i = configs.length - 1; i >= 0; i -= 1) {\n if (configs[i] !== undefined) {\n invokeConfig = configs[i];\n break;\n }\n }\n const hasInvokeTimeThreadId =\n invokeConfig?.configurable?.thread_id !== undefined;\n\n const hasExplicitNesting = configs.some(\n (c) => c?.configurable?.[CONFIG_KEY_READ] !== undefined\n );\n\n const ambientConfigurable =\n AsyncLocalStorageProviderSingleton.getRunnableConfig()?.configurable;\n const hasAmbientNesting =\n ambientConfigurable?.[CONFIG_KEY_READ] !== undefined;\n\n return hasInvokeTimeThreadId && !hasExplicitNesting && !hasAmbientNesting;\n}\n\nexport function ensureLangGraphConfig(\n ...configs: (LangGraphRunnableConfig | undefined)[]\n): RunnableConfig {\n const empty: LangGraphRunnableConfig = {\n tags: [],\n metadata: {},\n callbacks: undefined,\n recursionLimit: DEFAULT_RECURSION_LIMIT,\n configurable: {},\n };\n\n const skipImplicitConfigurable = isRootLevelExplicitInvoke(configs);\n\n const implicitConfig: RunnableConfig =\n AsyncLocalStorageProviderSingleton.getRunnableConfig();\n if (implicitConfig !== undefined) {\n for (const [k, v] of Object.entries(implicitConfig)) {\n if (v !== undefined) {\n // Stale-ALS root invoke: ignore the ambient `configurable` entirely,\n // since its keys may belong to another concurrent invocation.\n if (k === \"configurable\" && skipImplicitConfigurable) {\n continue;\n }\n if (COPIABLE_KEYS.includes(k)) {\n let copiedValue;\n if (Array.isArray(v)) {\n copiedValue = [...v];\n } else if (typeof v === \"object\") {\n if (\n k === \"callbacks\" &&\n \"copy\" in v &&\n typeof v.copy === \"function\"\n ) {\n copiedValue = v.copy();\n } else {\n copiedValue = { ...v };\n }\n } else {\n copiedValue = v;\n }\n empty[k as keyof RunnableConfig] = copiedValue;\n } else {\n empty[k as keyof RunnableConfig] = v;\n }\n }\n }\n }\n\n for (const config of configs) {\n if (config === undefined) {\n continue;\n }\n\n for (const [k, v] of Object.entries(config)) {\n if (v === undefined || !CONFIG_KEYS.includes(k)) {\n continue;\n }\n\n // Merge (rather than overwrite) the fields that compose across\n // multiple configs, matching langgraph's `merge_configs` /\n // langchain-core's `mergeConfigs`. Without this, a value bound via\n // `.withConfig({...})` is silently dropped whenever a later config\n // (e.g. invoke-time, or `streamEvents`' internal handler) supplies\n // any other key in the same field. Later configs still win per key\n // on collision.\n if (k === \"configurable\") {\n empty.configurable = {\n ...(empty.configurable as Record<string, unknown> | undefined),\n ...(v as Record<string, unknown>),\n };\n } else if (k === \"metadata\") {\n empty.metadata = {\n ...(empty.metadata as Record<string, unknown> | undefined),\n ...(v as Record<string, unknown>),\n };\n } else if (k === \"tags\") {\n // Plain concat (matches langgraph's `merge_configs`): no dedup,\n // no sort.\n empty.tags = [\n ...((empty.tags as string[] | undefined) ?? []),\n ...(v as string[]),\n ];\n } else if (k === \"callbacks\") {\n empty.callbacks = mergeCallbacks(empty.callbacks, v as Callbacks);\n } else {\n empty[k as keyof LangGraphRunnableConfig] = v;\n }\n }\n }\n\n empty.metadata =\n propagateConfigurableToMetadata(\n empty.configurable as Record<string, unknown> | undefined,\n empty.metadata as Record<string, unknown> | undefined\n ) ?? {};\n return empty;\n}\n\n/**\n * A helper utility function that returns the {@link BaseStore} that was set when the graph was initialized\n *\n * @returns a reference to the {@link BaseStore} that was set when the graph was initialized\n */\nexport function getStore(\n config?: LangGraphRunnableConfig\n): BaseStore | undefined {\n const runConfig: LangGraphRunnableConfig =\n config ?? AsyncLocalStorageProviderSingleton.getRunnableConfig();\n\n if (runConfig === undefined) {\n throw new Error(\n [\n \"Config not retrievable. This is likely because you are running in an environment without support for AsyncLocalStorage.\",\n \"If you're running `getStore` in such environment, pass the `config` from the node function directly.\",\n ].join(\"\\n\")\n );\n }\n\n return runConfig?.store;\n}\n\n/**\n * A helper utility function that returns the {@link LangGraphRunnableConfig#writer} if \"custom\" stream mode is enabled, otherwise undefined.\n *\n * @returns a reference to the {@link LangGraphRunnableConfig#writer} if \"custom\" stream mode is enabled, otherwise undefined\n */\nexport function getWriter(\n config?: LangGraphRunnableConfig\n): ((chunk: unknown) => void) | undefined {\n const runConfig: LangGraphRunnableConfig =\n config ?? AsyncLocalStorageProviderSingleton.getRunnableConfig();\n\n if (runConfig === undefined) {\n throw new Error(\n [\n \"Config not retrievable. This is likely because you are running in an environment without support for AsyncLocalStorage.\",\n \"If you're running `getWriter` in such environment, pass the `config` from the node function directly.\",\n ].join(\"\\n\")\n );\n }\n\n return runConfig?.writer || runConfig?.configurable?.writer;\n}\n\n/**\n * A helper utility function that returns the {@link LangGraphRunnableConfig} that was set when the graph was initialized.\n *\n * Note: This only works when running in an environment that supports node:async_hooks and AsyncLocalStorage. If you're running this in a\n * web environment, access the LangGraphRunnableConfig from the node function directly.\n *\n * @returns the {@link LangGraphRunnableConfig} that was set when the graph was initialized\n */\nexport function getConfig(): LangGraphRunnableConfig {\n return AsyncLocalStorageProviderSingleton.getRunnableConfig();\n}\n\n/**\n * A helper utility function that returns the input for the currently executing\n * task.\n *\n * Note: When called without arguments, this relies on `node:async_hooks` /\n * `AsyncLocalStorage`, which is available in many JavaScript environments\n * (Node.js, Deno, Cloudflare Workers) but not in web browsers. In environments\n * without `AsyncLocalStorage` support, pass the `config` that your node/tool\n * function receives directly, e.g. `getCurrentTaskInput(config)`.\n *\n * Tip: Inside a tool run by a `ToolNode`, prefer reading graph state from\n * `runtime.state` on the second tool argument (typed as `ToolRuntime` from\n * `@langchain/core/tools`). It works in every runtime, including web browsers.\n *\n * @param config - Optional {@link LangGraphRunnableConfig} to read the task\n * input from. Provide this when running in an environment without\n * `AsyncLocalStorage` support (e.g. web browsers).\n * @returns the input for the currently executing task\n */\nexport function getCurrentTaskInput<T = unknown>(\n config?: LangGraphRunnableConfig\n): T {\n const runConfig: LangGraphRunnableConfig =\n config ?? AsyncLocalStorageProviderSingleton.getRunnableConfig();\n\n if (runConfig === undefined) {\n throw new Error(\n [\n \"Config not retrievable. This is likely because you are running in an environment without support for AsyncLocalStorage.\",\n \"If you're running `getCurrentTaskInput` in such environment, pass the `config` from the node function directly.\",\n ].join(\"\\n\")\n );\n }\n\n if (\n runConfig.configurable?.[CONFIG_KEY_SCRATCHPAD]?.currentTaskInput ===\n undefined\n ) {\n throw new Error(\"BUG: internal scratchpad not initialized.\");\n }\n\n return runConfig!.configurable![CONFIG_KEY_SCRATCHPAD]!.currentTaskInput as T;\n}\n\nexport function recastCheckpointNamespace(namespace: string): string {\n return namespace\n .split(CHECKPOINT_NAMESPACE_SEPARATOR)\n .filter((part) => !part.match(/^\\d+$/))\n .map((part) => part.split(CHECKPOINT_NAMESPACE_END)[0])\n .join(CHECKPOINT_NAMESPACE_SEPARATOR);\n}\n\nexport function getParentCheckpointNamespace(namespace: string): string {\n const parts = namespace.split(CHECKPOINT_NAMESPACE_SEPARATOR);\n while (parts.length > 1 && parts[parts.length - 1].match(/^\\d+$/)) {\n parts.pop();\n }\n return parts.slice(0, -1).join(CHECKPOINT_NAMESPACE_SEPARATOR);\n}\n"],"mappings":";;;;AAgBA,MAAM,gBAAgB;CAAC;CAAQ;CAAY;CAAa;CAAe;AAEvE,MAAM,cAAc;CAClB;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACA;CACD;AAED,MAAM,0BAA0B;AAChC,MAAa,wBAAwB,IAAI,IAAI;CAC3C;CACA;CACA;CACA;CACA;CACA;CACA;CACD,CAAC;AAEF,SAAgB,gCACd,cACA,UACqC;AACrC,KAAI,CAAC,aACH,QAAO;CAET,MAAM,SAAS,YAAY,EAAE;AAC7B,MAAK,MAAM,OAAO,uBAAuB;AACvC,MAAI,OAAO,OACT;EAEF,MAAM,QAAQ,aAAa;AAC3B,MAAI,UAAU,KAAA,EACZ,QAAO,OAAO;;AAGlB,QAAO;;;;;;;;;;AAWT,SAAgB,iBACd,MACsB;AACtB,KAAI,QAAQ,QAAQ,KAAK,WAAW,EAAG,QAAO,KAAA;CAC9C,MAAM,WAAW,KAAK,QAAQ,QAAQ,CAAC,IAAI,WAAW,WAAW,CAAC;AAClE,QAAO,SAAS,SAAS,IAAI,WAAW,KAAA;;;;;;;;;;;;;AAc1C,SAAS,eACP,MACA,UACuB;AACvB,KAAI,aAAa,KAAA,EAAW,QAAO;AACnC,KAAI,SAAS,KAAA,EACX,QAAO,MAAM,QAAQ,SAAS,GAAG,CAAC,GAAG,SAAS,GAAG,SAAS,MAAM;AAElE,KAAI,MAAM,QAAQ,SAAS,EAAE;AAC3B,MAAI,MAAM,QAAQ,KAAK,CAAE,QAAO,KAAK,OAAO,SAAS;EAErD,MAAM,UAAU,KAAK,MAAM;AAC3B,OAAK,MAAM,YAAY,SACrB,SAAQ,YAAA,GAAA,kCAAA,eAAyB,SAAS,EAAE,KAAK;AAEnD,SAAO;;AAGT,KAAI,MAAM,QAAQ,KAAK,EAAE;EACvB,MAAM,UAAU,SAAS,MAAM;AAC/B,OAAK,MAAM,YAAY,KACrB,SAAQ,YAAA,GAAA,kCAAA,eAAyB,SAAS,EAAE,KAAK;AAEnD,SAAO;;CAGT,MAAM,kBACJ,aAEA,SAAS,QACN,SAAS,UACR,SAAS,WACN,cACC,cAAc,WACb,UAAU,SAAS,2BAClB,QAAQ,SAAS,wBACtB,KAAK,MACT;AACH,QAAO,IAAIA,kCAAAA,gBAAgB,SAAS,cAAc;EAChD,UAAU,eAAe,KAAK,SAAS,OAAO,SAAS,SAAS,CAAC;EACjE,qBAAqB,eACnB,KAAK,oBAAoB,OAAO,SAAS,oBAAoB,CAC9D;EACD,MAAM,MAAM,KAAK,IAAI,IAAI,KAAK,KAAK,OAAO,SAAS,KAAK,CAAC,CAAC;EAC1D,iBAAiB,MAAM,KACrB,IAAI,IAAI,KAAK,gBAAgB,OAAO,SAAS,gBAAgB,CAAC,CAC/D;EACD,UAAU;GAAE,GAAG,KAAK;GAAU,GAAG,SAAS;GAAU;EACpD,qBAAqB;GACnB,GAAG,KAAK;GACR,GAAG,SAAS;GACb;EACF,CAAC;;;;;;;;;;;;;;;;AAiBJ,SAAS,0BACP,SACS;CACT,IAAI;AACJ,MAAK,IAAI,IAAI,QAAQ,SAAS,GAAG,KAAK,GAAG,KAAK,EAC5C,KAAI,QAAQ,OAAO,KAAA,GAAW;AAC5B,iBAAe,QAAQ;AACvB;;CAGJ,MAAM,wBACJ,cAAc,cAAc,cAAc,KAAA;CAE5C,MAAM,qBAAqB,QAAQ,MAChC,MAAM,GAAG,eAAeC,kBAAAA,qBAAqB,KAAA,EAC/C;CAID,MAAM,qBADJC,2BAAAA,mCAAmC,mBAAmB,EAAE,gBAElCD,kBAAAA,qBAAqB,KAAA;AAE7C,QAAO,yBAAyB,CAAC,sBAAsB,CAAC;;AAG1D,SAAgB,sBACd,GAAG,SACa;CAChB,MAAM,QAAiC;EACrC,MAAM,EAAE;EACR,UAAU,EAAE;EACZ,WAAW,KAAA;EACX,gBAAgB;EAChB,cAAc,EAAE;EACjB;CAED,MAAM,2BAA2B,0BAA0B,QAAQ;CAEnE,MAAM,iBACJC,2BAAAA,mCAAmC,mBAAmB;AACxD,KAAI,mBAAmB,KAAA;OAChB,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ,eAAe,CACjD,KAAI,MAAM,KAAA,GAAW;AAGnB,OAAI,MAAM,kBAAkB,yBAC1B;AAEF,OAAI,cAAc,SAAS,EAAE,EAAE;IAC7B,IAAI;AACJ,QAAI,MAAM,QAAQ,EAAE,CAClB,eAAc,CAAC,GAAG,EAAE;aACX,OAAO,MAAM,SACtB,KACE,MAAM,eACN,UAAU,KACV,OAAO,EAAE,SAAS,WAElB,eAAc,EAAE,MAAM;QAEtB,eAAc,EAAE,GAAG,GAAG;QAGxB,eAAc;AAEhB,UAAM,KAA6B;SAEnC,OAAM,KAA6B;;;AAM3C,MAAK,MAAM,UAAU,SAAS;AAC5B,MAAI,WAAW,KAAA,EACb;AAGF,OAAK,MAAM,CAAC,GAAG,MAAM,OAAO,QAAQ,OAAO,EAAE;AAC3C,OAAI,MAAM,KAAA,KAAa,CAAC,YAAY,SAAS,EAAE,CAC7C;AAUF,OAAI,MAAM,eACR,OAAM,eAAe;IACnB,GAAI,MAAM;IACV,GAAI;IACL;YACQ,MAAM,WACf,OAAM,WAAW;IACf,GAAI,MAAM;IACV,GAAI;IACL;YACQ,MAAM,OAGf,OAAM,OAAO,CACX,GAAK,MAAM,QAAiC,EAAE,EAC9C,GAAI,EACL;YACQ,MAAM,YACf,OAAM,YAAY,eAAe,MAAM,WAAW,EAAe;OAEjE,OAAM,KAAsC;;;AAKlD,OAAM,WACJ,gCACE,MAAM,cACN,MAAM,SACP,IAAI,EAAE;AACT,QAAO;;;;;;;AAQT,SAAgB,SACd,QACuB;CACvB,MAAM,YACJ,UAAUA,2BAAAA,mCAAmC,mBAAmB;AAElE,KAAI,cAAc,KAAA,EAChB,OAAM,IAAI,MACR,CACE,2HACA,uGACD,CAAC,KAAK,KAAK,CACb;AAGH,QAAO,WAAW;;;;;;;AAQpB,SAAgB,UACd,QACwC;CACxC,MAAM,YACJ,UAAUA,2BAAAA,mCAAmC,mBAAmB;AAElE,KAAI,cAAc,KAAA,EAChB,OAAM,IAAI,MACR,CACE,2HACA,wGACD,CAAC,KAAK,KAAK,CACb;AAGH,QAAO,WAAW,UAAU,WAAW,cAAc;;;;;;;;;;AAWvD,SAAgB,YAAqC;AACnD,QAAOA,2BAAAA,mCAAmC,mBAAmB;;;;;;;;;;;;;;;;;;;;;AAsB/D,SAAgB,oBACd,QACG;CACH,MAAM,YACJ,UAAUA,2BAAAA,mCAAmC,mBAAmB;AAElE,KAAI,cAAc,KAAA,EAChB,OAAM,IAAI,MACR,CACE,2HACA,kHACD,CAAC,KAAK,KAAK,CACb;AAGH,KACE,UAAU,eAAA,wBAAuC,qBACjD,KAAA,EAEA,OAAM,IAAI,MAAM,4CAA4C;AAG9D,QAAO,UAAW,aAAcC,kBAAAA,uBAAwB;;AAG1D,SAAgB,0BAA0B,WAA2B;AACnE,QAAO,UACJ,MAAA,IAAqC,CACrC,QAAQ,SAAS,CAAC,KAAK,MAAM,QAAQ,CAAC,CACtC,KAAK,SAAS,KAAK,MAAA,IAA+B,CAAC,GAAG,CACtD,KAAA,IAAoC;;AAGzC,SAAgB,6BAA6B,WAA2B;CACtE,MAAM,QAAQ,UAAU,MAAA,IAAqC;AAC7D,QAAO,MAAM,SAAS,KAAK,MAAM,MAAM,SAAS,GAAG,MAAM,QAAQ,CAC/D,OAAM,KAAK;AAEb,QAAO,MAAM,MAAM,GAAG,GAAG,CAAC,KAAA,IAAoC"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.cts","names":[],"sources":["../../../src/pregel/utils/config.ts"],"mappings":";;;;;;;;;;
|
|
1
|
+
{"version":3,"file":"config.d.cts","names":[],"sources":["../../../src/pregel/utils/config.ts"],"mappings":";;;;;;;;;;iBA2SgB,QAAA,CACd,MAAA,GAAS,uBAAA,GACR,SAAA;;;AA+CH;;;iBA1BgB,SAAA,CACd,MAAA,GAAS,uBAAA,KACN,KAAA;;AA+CL;;;;;;;iBAvBgB,SAAA,CAAA,GAAa,uBAAA;;;;;;;;;;;;;;;;;;;;iBAuBb,mBAAA,aAAA,CACd,MAAA,GAAS,uBAAA,GACR,CAAA"}
|
|
@@ -1 +1 @@
|
|
|
1
|
-
{"version":3,"file":"config.d.ts","names":[],"sources":["../../../src/pregel/utils/config.ts"],"mappings":";;;;;;;;;;
|
|
1
|
+
{"version":3,"file":"config.d.ts","names":[],"sources":["../../../src/pregel/utils/config.ts"],"mappings":";;;;;;;;;;iBA2SgB,QAAA,CACd,MAAA,GAAS,uBAAA,GACR,SAAA;;;AA+CH;;;iBA1BgB,SAAA,CACd,MAAA,GAAS,uBAAA,KACN,KAAA;;AA+CL;;;;;;;iBAvBgB,SAAA,CAAA,GAAa,uBAAA;;;;;;;;;;;;;;;;;;;;iBAuBb,mBAAA,aAAA,CACd,MAAA,GAAS,uBAAA,GACR,CAAA"}
|