langchain 0.0.137 → 0.0.138
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/chat_models/minimax.cjs +1 -0
- package/chat_models/minimax.d.ts +1 -0
- package/chat_models/minimax.js +1 -0
- package/dist/callbacks/handlers/tracer.cjs +17 -17
- package/dist/callbacks/handlers/tracer.d.ts +1 -1
- package/dist/callbacks/handlers/tracer.js +17 -17
- package/dist/callbacks/manager.cjs +25 -10
- package/dist/callbacks/manager.d.ts +3 -2
- package/dist/callbacks/manager.js +25 -10
- package/dist/chat_models/minimax.cjs +547 -0
- package/dist/chat_models/minimax.d.ts +364 -0
- package/dist/chat_models/minimax.js +543 -0
- package/dist/chat_models/ollama.cjs +136 -0
- package/dist/chat_models/ollama.d.ts +34 -0
- package/dist/chat_models/ollama.js +136 -0
- package/dist/embeddings/minimax.cjs +152 -0
- package/dist/embeddings/minimax.d.ts +104 -0
- package/dist/embeddings/minimax.js +148 -0
- package/dist/llms/ollama.cjs +136 -0
- package/dist/llms/ollama.d.ts +34 -0
- package/dist/llms/ollama.js +136 -0
- package/dist/load/import_map.cjs +4 -2
- package/dist/load/import_map.d.ts +2 -0
- package/dist/load/import_map.js +2 -0
- package/dist/util/ollama.d.ts +34 -0
- package/dist/vectorstores/redis.cjs +17 -2
- package/dist/vectorstores/redis.d.ts +10 -1
- package/dist/vectorstores/redis.js +17 -2
- package/dist/vectorstores/zep.cjs +2 -1
- package/dist/vectorstores/zep.js +3 -2
- package/embeddings/minimax.cjs +1 -0
- package/embeddings/minimax.d.ts +1 -0
- package/embeddings/minimax.js +1 -0
- package/package.json +19 -3
|
@@ -0,0 +1 @@
|
|
|
1
|
+
module.exports = require('../dist/chat_models/minimax.cjs');
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from '../dist/chat_models/minimax.js'
|
|
@@ -0,0 +1 @@
|
|
|
1
|
+
export * from '../dist/chat_models/minimax.js'
|
|
@@ -60,7 +60,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
60
60
|
events: [
|
|
61
61
|
{
|
|
62
62
|
name: "start",
|
|
63
|
-
time: start_time,
|
|
63
|
+
time: new Date(start_time).toISOString(),
|
|
64
64
|
},
|
|
65
65
|
],
|
|
66
66
|
inputs: { prompts },
|
|
@@ -89,7 +89,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
89
89
|
events: [
|
|
90
90
|
{
|
|
91
91
|
name: "start",
|
|
92
|
-
time: start_time,
|
|
92
|
+
time: new Date(start_time).toISOString(),
|
|
93
93
|
},
|
|
94
94
|
],
|
|
95
95
|
inputs: { messages },
|
|
@@ -112,7 +112,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
112
112
|
run.outputs = output;
|
|
113
113
|
run.events.push({
|
|
114
114
|
name: "end",
|
|
115
|
-
time: run.end_time,
|
|
115
|
+
time: new Date(run.end_time).toISOString(),
|
|
116
116
|
});
|
|
117
117
|
await this.onLLMEnd?.(run);
|
|
118
118
|
await this._endTrace(run);
|
|
@@ -126,7 +126,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
126
126
|
run.error = error.message;
|
|
127
127
|
run.events.push({
|
|
128
128
|
name: "error",
|
|
129
|
-
time: run.end_time,
|
|
129
|
+
time: new Date(run.end_time).toISOString(),
|
|
130
130
|
});
|
|
131
131
|
await this.onLLMError?.(run);
|
|
132
132
|
await this._endTrace(run);
|
|
@@ -143,7 +143,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
143
143
|
events: [
|
|
144
144
|
{
|
|
145
145
|
name: "start",
|
|
146
|
-
time: start_time,
|
|
146
|
+
time: new Date(start_time).toISOString(),
|
|
147
147
|
},
|
|
148
148
|
],
|
|
149
149
|
inputs,
|
|
@@ -166,7 +166,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
166
166
|
run.outputs = outputs;
|
|
167
167
|
run.events.push({
|
|
168
168
|
name: "end",
|
|
169
|
-
time: run.end_time,
|
|
169
|
+
time: new Date(run.end_time).toISOString(),
|
|
170
170
|
});
|
|
171
171
|
await this.onChainEnd?.(run);
|
|
172
172
|
await this._endTrace(run);
|
|
@@ -180,7 +180,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
180
180
|
run.error = error.message;
|
|
181
181
|
run.events.push({
|
|
182
182
|
name: "error",
|
|
183
|
-
time: run.end_time,
|
|
183
|
+
time: new Date(run.end_time).toISOString(),
|
|
184
184
|
});
|
|
185
185
|
await this.onChainError?.(run);
|
|
186
186
|
await this._endTrace(run);
|
|
@@ -197,7 +197,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
197
197
|
events: [
|
|
198
198
|
{
|
|
199
199
|
name: "start",
|
|
200
|
-
time: start_time,
|
|
200
|
+
time: new Date(start_time).toISOString(),
|
|
201
201
|
},
|
|
202
202
|
],
|
|
203
203
|
inputs: { input },
|
|
@@ -220,7 +220,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
220
220
|
run.outputs = { output };
|
|
221
221
|
run.events.push({
|
|
222
222
|
name: "end",
|
|
223
|
-
time: run.end_time,
|
|
223
|
+
time: new Date(run.end_time).toISOString(),
|
|
224
224
|
});
|
|
225
225
|
await this.onToolEnd?.(run);
|
|
226
226
|
await this._endTrace(run);
|
|
@@ -234,7 +234,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
234
234
|
run.error = error.message;
|
|
235
235
|
run.events.push({
|
|
236
236
|
name: "error",
|
|
237
|
-
time: run.end_time,
|
|
237
|
+
time: new Date(run.end_time).toISOString(),
|
|
238
238
|
});
|
|
239
239
|
await this.onToolError?.(run);
|
|
240
240
|
await this._endTrace(run);
|
|
@@ -249,7 +249,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
249
249
|
agentRun.actions.push(action);
|
|
250
250
|
agentRun.events.push({
|
|
251
251
|
name: "agent_action",
|
|
252
|
-
time: Date.
|
|
252
|
+
time: new Date().toISOString(),
|
|
253
253
|
kwargs: { action },
|
|
254
254
|
});
|
|
255
255
|
await this.onAgentAction?.(run);
|
|
@@ -261,7 +261,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
261
261
|
}
|
|
262
262
|
run.events.push({
|
|
263
263
|
name: "agent_end",
|
|
264
|
-
time: Date.
|
|
264
|
+
time: new Date().toISOString(),
|
|
265
265
|
kwargs: { action },
|
|
266
266
|
});
|
|
267
267
|
await this.onAgentEnd?.(run);
|
|
@@ -278,7 +278,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
278
278
|
events: [
|
|
279
279
|
{
|
|
280
280
|
name: "start",
|
|
281
|
-
time: start_time,
|
|
281
|
+
time: new Date(start_time).toISOString(),
|
|
282
282
|
},
|
|
283
283
|
],
|
|
284
284
|
inputs: { query },
|
|
@@ -301,7 +301,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
301
301
|
run.outputs = { documents };
|
|
302
302
|
run.events.push({
|
|
303
303
|
name: "end",
|
|
304
|
-
time: run.end_time,
|
|
304
|
+
time: new Date(run.end_time).toISOString(),
|
|
305
305
|
});
|
|
306
306
|
await this.onRetrieverEnd?.(run);
|
|
307
307
|
await this._endTrace(run);
|
|
@@ -315,7 +315,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
315
315
|
run.error = error.message;
|
|
316
316
|
run.events.push({
|
|
317
317
|
name: "error",
|
|
318
|
-
time: run.end_time,
|
|
318
|
+
time: new Date(run.end_time).toISOString(),
|
|
319
319
|
});
|
|
320
320
|
await this.onRetrieverError?.(run);
|
|
321
321
|
await this._endTrace(run);
|
|
@@ -327,7 +327,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
327
327
|
}
|
|
328
328
|
run.events.push({
|
|
329
329
|
name: "text",
|
|
330
|
-
time: Date.
|
|
330
|
+
time: new Date().toISOString(),
|
|
331
331
|
kwargs: { text },
|
|
332
332
|
});
|
|
333
333
|
await this.onText?.(run);
|
|
@@ -339,7 +339,7 @@ class BaseTracer extends base_js_1.BaseCallbackHandler {
|
|
|
339
339
|
}
|
|
340
340
|
run.events.push({
|
|
341
341
|
name: "new_token",
|
|
342
|
-
time: Date.
|
|
342
|
+
time: new Date().toISOString(),
|
|
343
343
|
kwargs: { token, idx, chunk: fields?.chunk },
|
|
344
344
|
});
|
|
345
345
|
await this.onLLMNewToken?.(run);
|
|
@@ -57,7 +57,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
57
57
|
events: [
|
|
58
58
|
{
|
|
59
59
|
name: "start",
|
|
60
|
-
time: start_time,
|
|
60
|
+
time: new Date(start_time).toISOString(),
|
|
61
61
|
},
|
|
62
62
|
],
|
|
63
63
|
inputs: { prompts },
|
|
@@ -86,7 +86,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
86
86
|
events: [
|
|
87
87
|
{
|
|
88
88
|
name: "start",
|
|
89
|
-
time: start_time,
|
|
89
|
+
time: new Date(start_time).toISOString(),
|
|
90
90
|
},
|
|
91
91
|
],
|
|
92
92
|
inputs: { messages },
|
|
@@ -109,7 +109,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
109
109
|
run.outputs = output;
|
|
110
110
|
run.events.push({
|
|
111
111
|
name: "end",
|
|
112
|
-
time: run.end_time,
|
|
112
|
+
time: new Date(run.end_time).toISOString(),
|
|
113
113
|
});
|
|
114
114
|
await this.onLLMEnd?.(run);
|
|
115
115
|
await this._endTrace(run);
|
|
@@ -123,7 +123,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
123
123
|
run.error = error.message;
|
|
124
124
|
run.events.push({
|
|
125
125
|
name: "error",
|
|
126
|
-
time: run.end_time,
|
|
126
|
+
time: new Date(run.end_time).toISOString(),
|
|
127
127
|
});
|
|
128
128
|
await this.onLLMError?.(run);
|
|
129
129
|
await this._endTrace(run);
|
|
@@ -140,7 +140,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
140
140
|
events: [
|
|
141
141
|
{
|
|
142
142
|
name: "start",
|
|
143
|
-
time: start_time,
|
|
143
|
+
time: new Date(start_time).toISOString(),
|
|
144
144
|
},
|
|
145
145
|
],
|
|
146
146
|
inputs,
|
|
@@ -163,7 +163,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
163
163
|
run.outputs = outputs;
|
|
164
164
|
run.events.push({
|
|
165
165
|
name: "end",
|
|
166
|
-
time: run.end_time,
|
|
166
|
+
time: new Date(run.end_time).toISOString(),
|
|
167
167
|
});
|
|
168
168
|
await this.onChainEnd?.(run);
|
|
169
169
|
await this._endTrace(run);
|
|
@@ -177,7 +177,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
177
177
|
run.error = error.message;
|
|
178
178
|
run.events.push({
|
|
179
179
|
name: "error",
|
|
180
|
-
time: run.end_time,
|
|
180
|
+
time: new Date(run.end_time).toISOString(),
|
|
181
181
|
});
|
|
182
182
|
await this.onChainError?.(run);
|
|
183
183
|
await this._endTrace(run);
|
|
@@ -194,7 +194,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
194
194
|
events: [
|
|
195
195
|
{
|
|
196
196
|
name: "start",
|
|
197
|
-
time: start_time,
|
|
197
|
+
time: new Date(start_time).toISOString(),
|
|
198
198
|
},
|
|
199
199
|
],
|
|
200
200
|
inputs: { input },
|
|
@@ -217,7 +217,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
217
217
|
run.outputs = { output };
|
|
218
218
|
run.events.push({
|
|
219
219
|
name: "end",
|
|
220
|
-
time: run.end_time,
|
|
220
|
+
time: new Date(run.end_time).toISOString(),
|
|
221
221
|
});
|
|
222
222
|
await this.onToolEnd?.(run);
|
|
223
223
|
await this._endTrace(run);
|
|
@@ -231,7 +231,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
231
231
|
run.error = error.message;
|
|
232
232
|
run.events.push({
|
|
233
233
|
name: "error",
|
|
234
|
-
time: run.end_time,
|
|
234
|
+
time: new Date(run.end_time).toISOString(),
|
|
235
235
|
});
|
|
236
236
|
await this.onToolError?.(run);
|
|
237
237
|
await this._endTrace(run);
|
|
@@ -246,7 +246,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
246
246
|
agentRun.actions.push(action);
|
|
247
247
|
agentRun.events.push({
|
|
248
248
|
name: "agent_action",
|
|
249
|
-
time: Date.
|
|
249
|
+
time: new Date().toISOString(),
|
|
250
250
|
kwargs: { action },
|
|
251
251
|
});
|
|
252
252
|
await this.onAgentAction?.(run);
|
|
@@ -258,7 +258,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
258
258
|
}
|
|
259
259
|
run.events.push({
|
|
260
260
|
name: "agent_end",
|
|
261
|
-
time: Date.
|
|
261
|
+
time: new Date().toISOString(),
|
|
262
262
|
kwargs: { action },
|
|
263
263
|
});
|
|
264
264
|
await this.onAgentEnd?.(run);
|
|
@@ -275,7 +275,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
275
275
|
events: [
|
|
276
276
|
{
|
|
277
277
|
name: "start",
|
|
278
|
-
time: start_time,
|
|
278
|
+
time: new Date(start_time).toISOString(),
|
|
279
279
|
},
|
|
280
280
|
],
|
|
281
281
|
inputs: { query },
|
|
@@ -298,7 +298,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
298
298
|
run.outputs = { documents };
|
|
299
299
|
run.events.push({
|
|
300
300
|
name: "end",
|
|
301
|
-
time: run.end_time,
|
|
301
|
+
time: new Date(run.end_time).toISOString(),
|
|
302
302
|
});
|
|
303
303
|
await this.onRetrieverEnd?.(run);
|
|
304
304
|
await this._endTrace(run);
|
|
@@ -312,7 +312,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
312
312
|
run.error = error.message;
|
|
313
313
|
run.events.push({
|
|
314
314
|
name: "error",
|
|
315
|
-
time: run.end_time,
|
|
315
|
+
time: new Date(run.end_time).toISOString(),
|
|
316
316
|
});
|
|
317
317
|
await this.onRetrieverError?.(run);
|
|
318
318
|
await this._endTrace(run);
|
|
@@ -324,7 +324,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
324
324
|
}
|
|
325
325
|
run.events.push({
|
|
326
326
|
name: "text",
|
|
327
|
-
time: Date.
|
|
327
|
+
time: new Date().toISOString(),
|
|
328
328
|
kwargs: { text },
|
|
329
329
|
});
|
|
330
330
|
await this.onText?.(run);
|
|
@@ -336,7 +336,7 @@ export class BaseTracer extends BaseCallbackHandler {
|
|
|
336
336
|
}
|
|
337
337
|
run.events.push({
|
|
338
338
|
name: "new_token",
|
|
339
|
-
time: Date.
|
|
339
|
+
time: new Date().toISOString(),
|
|
340
340
|
kwargs: { token, idx, chunk: fields?.chunk },
|
|
341
341
|
});
|
|
342
342
|
await this.onLLMNewToken?.(run);
|
|
@@ -571,42 +571,57 @@ class TraceGroup {
|
|
|
571
571
|
value: void 0
|
|
572
572
|
});
|
|
573
573
|
}
|
|
574
|
-
async getTraceGroupCallbackManager(group_name, options) {
|
|
574
|
+
async getTraceGroupCallbackManager(group_name, inputs, options) {
|
|
575
575
|
const cb = new tracer_langchain_js_1.LangChainTracer(options);
|
|
576
576
|
const cm = await CallbackManager.configure([cb]);
|
|
577
577
|
const runManager = await cm?.handleChainStart({
|
|
578
578
|
lc: 1,
|
|
579
579
|
type: "not_implemented",
|
|
580
580
|
id: ["langchain", "callbacks", "groups", group_name],
|
|
581
|
-
}, {});
|
|
581
|
+
}, inputs ?? {});
|
|
582
582
|
if (!runManager) {
|
|
583
583
|
throw new Error("Failed to create run group callback manager.");
|
|
584
584
|
}
|
|
585
585
|
return runManager;
|
|
586
586
|
}
|
|
587
|
-
async start() {
|
|
587
|
+
async start(inputs) {
|
|
588
588
|
if (!this.runManager) {
|
|
589
|
-
this.runManager = await this.getTraceGroupCallbackManager(this.groupName, this.options);
|
|
589
|
+
this.runManager = await this.getTraceGroupCallbackManager(this.groupName, inputs, this.options);
|
|
590
590
|
}
|
|
591
591
|
return this.runManager.getChild();
|
|
592
592
|
}
|
|
593
|
-
async
|
|
593
|
+
async error(err) {
|
|
594
594
|
if (this.runManager) {
|
|
595
|
-
await this.runManager.
|
|
595
|
+
await this.runManager.handleChainError(err);
|
|
596
|
+
this.runManager = undefined;
|
|
597
|
+
}
|
|
598
|
+
}
|
|
599
|
+
async end(output) {
|
|
600
|
+
if (this.runManager) {
|
|
601
|
+
await this.runManager.handleChainEnd(output ?? {});
|
|
596
602
|
this.runManager = undefined;
|
|
597
603
|
}
|
|
598
604
|
}
|
|
599
605
|
}
|
|
600
606
|
exports.TraceGroup = TraceGroup;
|
|
601
607
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
608
|
+
function _coerceToDict(value, defaultKey) {
|
|
609
|
+
return value && !Array.isArray(value) && typeof value === "object"
|
|
610
|
+
? value
|
|
611
|
+
: { [defaultKey]: value };
|
|
612
|
+
}
|
|
613
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
602
614
|
async function traceAsGroup(groupOptions, enclosedCode, ...args) {
|
|
603
615
|
const traceGroup = new TraceGroup(groupOptions.name, groupOptions);
|
|
604
|
-
const callbackManager = await traceGroup.start();
|
|
616
|
+
const callbackManager = await traceGroup.start({ ...args });
|
|
605
617
|
try {
|
|
606
|
-
|
|
618
|
+
const result = await enclosedCode(callbackManager, ...args);
|
|
619
|
+
await traceGroup.end(_coerceToDict(result, "output"));
|
|
620
|
+
return result;
|
|
607
621
|
}
|
|
608
|
-
|
|
609
|
-
await traceGroup.
|
|
622
|
+
catch (err) {
|
|
623
|
+
await traceGroup.error(err);
|
|
624
|
+
throw err;
|
|
610
625
|
}
|
|
611
626
|
}
|
|
612
627
|
exports.traceAsGroup = traceAsGroup;
|
|
@@ -113,8 +113,9 @@ export declare class TraceGroup {
|
|
|
113
113
|
exampleId?: string | undefined;
|
|
114
114
|
} | undefined);
|
|
115
115
|
private getTraceGroupCallbackManager;
|
|
116
|
-
start(): Promise<CallbackManager>;
|
|
117
|
-
|
|
116
|
+
start(inputs?: ChainValues): Promise<CallbackManager>;
|
|
117
|
+
error(err: Error | unknown): Promise<void>;
|
|
118
|
+
end(output?: ChainValues): Promise<void>;
|
|
118
119
|
}
|
|
119
120
|
export declare function traceAsGroup<T, A extends any[]>(groupOptions: {
|
|
120
121
|
name: string;
|
|
@@ -561,40 +561,55 @@ export class TraceGroup {
|
|
|
561
561
|
value: void 0
|
|
562
562
|
});
|
|
563
563
|
}
|
|
564
|
-
async getTraceGroupCallbackManager(group_name, options) {
|
|
564
|
+
async getTraceGroupCallbackManager(group_name, inputs, options) {
|
|
565
565
|
const cb = new LangChainTracer(options);
|
|
566
566
|
const cm = await CallbackManager.configure([cb]);
|
|
567
567
|
const runManager = await cm?.handleChainStart({
|
|
568
568
|
lc: 1,
|
|
569
569
|
type: "not_implemented",
|
|
570
570
|
id: ["langchain", "callbacks", "groups", group_name],
|
|
571
|
-
}, {});
|
|
571
|
+
}, inputs ?? {});
|
|
572
572
|
if (!runManager) {
|
|
573
573
|
throw new Error("Failed to create run group callback manager.");
|
|
574
574
|
}
|
|
575
575
|
return runManager;
|
|
576
576
|
}
|
|
577
|
-
async start() {
|
|
577
|
+
async start(inputs) {
|
|
578
578
|
if (!this.runManager) {
|
|
579
|
-
this.runManager = await this.getTraceGroupCallbackManager(this.groupName, this.options);
|
|
579
|
+
this.runManager = await this.getTraceGroupCallbackManager(this.groupName, inputs, this.options);
|
|
580
580
|
}
|
|
581
581
|
return this.runManager.getChild();
|
|
582
582
|
}
|
|
583
|
-
async
|
|
583
|
+
async error(err) {
|
|
584
584
|
if (this.runManager) {
|
|
585
|
-
await this.runManager.
|
|
585
|
+
await this.runManager.handleChainError(err);
|
|
586
586
|
this.runManager = undefined;
|
|
587
587
|
}
|
|
588
588
|
}
|
|
589
|
+
async end(output) {
|
|
590
|
+
if (this.runManager) {
|
|
591
|
+
await this.runManager.handleChainEnd(output ?? {});
|
|
592
|
+
this.runManager = undefined;
|
|
593
|
+
}
|
|
594
|
+
}
|
|
595
|
+
}
|
|
596
|
+
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
597
|
+
function _coerceToDict(value, defaultKey) {
|
|
598
|
+
return value && !Array.isArray(value) && typeof value === "object"
|
|
599
|
+
? value
|
|
600
|
+
: { [defaultKey]: value };
|
|
589
601
|
}
|
|
590
602
|
// eslint-disable-next-line @typescript-eslint/no-explicit-any
|
|
591
603
|
export async function traceAsGroup(groupOptions, enclosedCode, ...args) {
|
|
592
604
|
const traceGroup = new TraceGroup(groupOptions.name, groupOptions);
|
|
593
|
-
const callbackManager = await traceGroup.start();
|
|
605
|
+
const callbackManager = await traceGroup.start({ ...args });
|
|
594
606
|
try {
|
|
595
|
-
|
|
607
|
+
const result = await enclosedCode(callbackManager, ...args);
|
|
608
|
+
await traceGroup.end(_coerceToDict(result, "output"));
|
|
609
|
+
return result;
|
|
596
610
|
}
|
|
597
|
-
|
|
598
|
-
await traceGroup.
|
|
611
|
+
catch (err) {
|
|
612
|
+
await traceGroup.error(err);
|
|
613
|
+
throw err;
|
|
599
614
|
}
|
|
600
615
|
}
|