UNPKG

@langchain/langgraph

Version:
85 lines (84 loc) 3.17 kB
const require_config = require("./pregel/utils/config.cjs"); let _langchain_core_singletons = require("@langchain/core/singletons"); let _langchain_core_runnables = require("@langchain/core/runnables"); //#region src/utils.ts var RunnableCallable = class extends _langchain_core_runnables.Runnable { lc_namespace = ["langgraph"]; func; tags; config; trace = true; recurse = true; constructor(fields) { super(); this.name = fields.name ?? fields.func.name; this.func = fields.func; this.config = fields.tags ? { tags: fields.tags } : void 0; this.trace = fields.trace ?? this.trace; this.recurse = fields.recurse ?? this.recurse; } async _tracedInvoke(input, config, runManager) { return new Promise((resolve, reject) => { const childConfig = (0, _langchain_core_runnables.patchConfig)(config, { callbacks: runManager?.getChild() }); _langchain_core_singletons.AsyncLocalStorageProviderSingleton.runWithConfig(childConfig, async () => { try { resolve(await this.func(input, childConfig)); } catch (e) { reject(e); } }); }); } async invoke(input, options) { let returnValue; const config = require_config.ensureLangGraphConfig(options); const mergedConfig = (0, _langchain_core_runnables.mergeConfigs)(this.config, config); if (this.trace) returnValue = await this._callWithConfig(this._tracedInvoke, input, mergedConfig); else returnValue = await _langchain_core_singletons.AsyncLocalStorageProviderSingleton.runWithConfig(mergedConfig, async () => this.func(input, mergedConfig)); if (_langchain_core_runnables.Runnable.isRunnable(returnValue) && this.recurse) return await _langchain_core_singletons.AsyncLocalStorageProviderSingleton.runWithConfig(mergedConfig, async () => returnValue.invoke(input, mergedConfig)); return returnValue; } }; function* prefixGenerator(generator, prefix) { if (prefix === void 0) yield* generator; else for (const value of generator) yield [prefix, value]; } async function gatherIterator(i) { const out = []; for await (const item of await i) out.push(item); return out; } function gatherIteratorSync(i) { const out = []; for (const item of i) out.push(item); return out; } function patchConfigurable(config, patch) { if (!config) return { configurable: patch }; else if (!("configurable" in config)) return { ...config, configurable: patch }; else return { ...config, configurable: { ...config.configurable, ...patch } }; } function isAsyncGeneratorFunction(val) { return val != null && typeof val === "function" && val instanceof Object.getPrototypeOf(async function* () {}).constructor; } function isGeneratorFunction(val) { return val != null && typeof val === "function" && val instanceof Object.getPrototypeOf(function* () {}).constructor; } //#endregion exports.RunnableCallable = RunnableCallable; exports.gatherIterator = gatherIterator; exports.gatherIteratorSync = gatherIteratorSync; exports.isAsyncGeneratorFunction = isAsyncGeneratorFunction; exports.isGeneratorFunction = isGeneratorFunction; exports.patchConfigurable = patchConfigurable; exports.prefixGenerator = prefixGenerator; //# sourceMappingURL=utils.cjs.map