@eggjs/tegg-langchain 0.0.0

package/lib/graph/GraphLoadUnitHook.d.ts

@@ -0,0 +1,12 @@
+import { EggProtoImplClass, LifecycleHook } from '@eggjs/tegg';
+import { EggPrototypeFactory, LoadUnit, LoadUnitLifecycleContext } from '@eggjs/tegg-metadata';
+import { CompiledStateGraphProto } from './CompiledStateGraphProto';
+import { IGraphMetadata } from '@eggjs/tegg-langchain-decorator';
+export declare class GraphLoadUnitHook implements LifecycleHook<LoadUnitLifecycleContext, LoadUnit> {
+    private readonly eggPrototypeFactory;
+    clazzMap: Map<EggProtoImplClass, IGraphMetadata>;
+    graphCompiledNameMap: Map<string, CompiledStateGraphProto>;
+    constructor(eggPrototypeFactory: EggPrototypeFactory);
+    preCreate(ctx: LoadUnitLifecycleContext, loadUnit: LoadUnit): Promise<void>;
+    postCreate(_ctx: LoadUnitLifecycleContext, obj: LoadUnit): Promise<void>;
+}

package/lib/graph/GraphLoadUnitHook.js

@@ -0,0 +1,48 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GraphLoadUnitHook = void 0;
+const CompiledStateGraphProto_1 = require("./CompiledStateGraphProto");
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+const node_assert_1 = __importDefault(require("node:assert"));
+class GraphLoadUnitHook {
+    constructor(eggPrototypeFactory) {
+        this.graphCompiledNameMap = new Map();
+        this.eggPrototypeFactory = eggPrototypeFactory;
+        this.clazzMap = tegg_langchain_decorator_1.GraphInfoUtil.getAllGraphMetadata();
+    }
+    async preCreate(ctx, loadUnit) {
+        const clazzList = await ctx.loader.load();
+        for (const clazz of clazzList) {
+            const meta = this.clazzMap.get(clazz);
+            if (meta) {
+                const protoName = clazz.name[0].toLowerCase() + clazz.name.substring(1);
+                const graphMetadata = tegg_langchain_decorator_1.GraphInfoUtil.getGraphMetadata(clazz);
+                (0, node_assert_1.default)(graphMetadata, `${clazz.name} graphMetadata should not be null`);
+                const proto = new CompiledStateGraphProto_1.CompiledStateGraphProto(loadUnit, `compiled${protoName.replace(protoName[0], protoName[0].toUpperCase())}`, protoName, graphMetadata);
+                this.eggPrototypeFactory.registerPrototype(proto, loadUnit);
+                this.graphCompiledNameMap.set(protoName, proto);
+            }
+        }
+    }
+    async postCreate(_ctx, obj) {
+        for (const graphName of this.graphCompiledNameMap.keys()) {
+            const graphProto = obj.getEggPrototype(graphName, [])[0];
+            if (graphProto) {
+                const compiledGraphProto = this.graphCompiledNameMap.get(graphName);
+                if (!compiledGraphProto.injectObjects.find(injectObject => injectObject.objName === graphProto.name)) {
+                    compiledGraphProto.injectObjects.push({
+                        refName: 'stateGraph',
+                        objName: graphProto.name,
+                        qualifiers: [],
+                        proto: graphProto,
+                    });
+                }
+            }
+        }
+    }
+}
+exports.GraphLoadUnitHook = GraphLoadUnitHook;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiR3JhcGhMb2FkVW5pdEhvb2suanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJHcmFwaExvYWRVbml0SG9vay50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7QUFNQSx1RUFBb0U7QUFDcEUsOEVBQWdGO0FBQ2hGLDhEQUFpQztBQUVqQyxNQUFhLGlCQUFpQjtJQUs1QixZQUFZLG1CQUF3QztRQUZwRCx5QkFBb0IsR0FBeUMsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUdyRSxJQUFJLENBQUMsbUJBQW1CLEdBQUcsbUJBQW1CLENBQUM7UUFDL0MsSUFBSSxDQUFDLFFBQVEsR0FBRyx3Q0FBYSxDQUFDLG1CQUFtQixFQUFFLENBQUM7SUFDdEQsQ0FBQztJQUVELEtBQUssQ0FBQyxTQUFTLENBQUMsR0FBNkIsRUFBRSxRQUFrQjtRQUMvRCxNQUFNLFNBQVMsR0FBRyxNQUFNLEdBQUcsQ0FBQyxNQUFNLENBQUMsSUFBSSxFQUFFLENBQUM7UUFDMUMsS0FBSyxNQUFNLEtBQUssSUFBSSxTQUFTLEVBQUUsQ0FBQztZQUM5QixNQUFNLElBQUksR0FBRyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxLQUEwQixDQUFDLENBQUM7WUFDM0QsSUFBSSxJQUFJLEVBQUUsQ0FBQztnQkFDVCxNQUFNLFNBQVMsR0FBRyxLQUFLLENBQUMsSUFBSSxDQUFDLENBQUMsQ0FBQyxDQUFDLFdBQVcsRUFBRSxHQUFHLEtBQUssQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUMsQ0FBQyxDQUFDO2dCQUN4RSxNQUFNLGFBQWEsR0FBRyx3Q0FBYSxDQUFDLGdCQUFnQixDQUFDLEtBQTBCLENBQUMsQ0FBQztnQkFDakYsSUFBQSxxQkFBTSxFQUFDLGFBQWEsRUFBRSxHQUFHLEtBQUssQ0FBQyxJQUFJLG1DQUFtQyxDQUFDLENBQUM7Z0JBQ3hFLE1BQU0sS0FBSyxHQUFHLElBQUksaURBQXVCLENBQUMsUUFBUSxFQUFFLFdBQVcsU0FBUyxDQUFDLE9BQU8sQ0FBQyxTQUFTLENBQUMsQ0FBQyxDQUFDLEVBQUUsU0FBUyxDQUFDLENBQUMsQ0FBQyxDQUFDLFdBQVcsRUFBRSxDQUFDLEVBQUUsRUFBRSxTQUFTLEVBQUUsYUFBYSxDQUFDLENBQUM7Z0JBQ3hKLElBQUksQ0FBQyxtQkFBbUIsQ0FBQyxpQkFBaUIsQ0FBQyxLQUFLLEVBQUUsUUFBUSxDQUFDLENBQUM7Z0JBQzVELElBQUksQ0FBQyxvQkFBb0IsQ0FBQyxHQUFHLENBQUMsU0FBUyxFQUFFLEtBQUssQ0FBQyxDQUFDO1lBQ2xELENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztJQUVELEtBQUssQ0FBQyxVQUFVLENBQUMsSUFBOEIsRUFBRSxHQUFhO1FBQzVELEtBQUssTUFBTSxTQUFTLElBQUksSUFBSSxDQUFDLG9CQUFvQixDQUFDLElBQUksRUFBRSxFQUFFLENBQUM7WUFDekQsTUFBTSxVQUFVLEdBQUcsR0FBRyxDQUFDLGVBQWUsQ0FBQyxTQUFTLEVBQUUsRUFBRSxDQUFDLENBQUMsQ0FBQyxDQUFDLENBQUM7WUFDekQsSUFBSSxVQUFVLEVBQUUsQ0FBQztnQkFDZixNQUFNLGtCQUFrQixHQUFHLElBQUksQ0FBQyxvQkFBb0IsQ0FBQyxHQUFHLENBQUMsU0FBUyxDQUE0QixDQUFDO2dCQUMvRixJQUFJLENBQUMsa0JBQWtCLENBQUMsYUFBYSxDQUFDLElBQUksQ0FBQyxZQUFZLENBQUMsRUFBRSxDQUFDLFlBQVksQ0FBQyxPQUFPLEtBQUssVUFBVSxDQUFDLElBQUksQ0FBQyxFQUFFLENBQUM7b0JBQ3JHLGtCQUFrQixDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUM7d0JBQ3BDLE9BQU8sRUFBRSxZQUFZO3dCQUNyQixPQUFPLEVBQUUsVUFBVSxDQUFDLElBQUk7d0JBQ3hCLFVBQVUsRUFBRSxFQUFFO3dCQUNkLEtBQUssRUFBRSxVQUFVO3FCQUNsQixDQUFDLENBQUM7Z0JBQ0wsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztDQUVGO0FBMUNELDhDQTBDQyJ9

package/lib/graph/GraphObjectHook.d.ts

@@ -0,0 +1,5 @@
+import { type LifecycleHook } from '@eggjs/tegg';
+import { EggObjectLifeCycleContext, EggObject } from '@eggjs/tegg/helper';
+export declare class GraphObjectHook implements LifecycleHook<EggObjectLifeCycleContext, EggObject> {
+    postCreate(_: EggObjectLifeCycleContext, eggObject: EggObject): Promise<void>;
+}

package/lib/graph/GraphObjectHook.js

@@ -0,0 +1,142 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __setModuleDefault = (this && this.__setModuleDefault) || (Object.create ? (function(o, v) {
+    Object.defineProperty(o, "default", { enumerable: true, value: v });
+}) : function(o, v) {
+    o["default"] = v;
+});
+var __importStar = (this && this.__importStar) || (function () {
+    var ownKeys = function(o) {
+        ownKeys = Object.getOwnPropertyNames || function (o) {
+            var ar = [];
+            for (var k in o) if (Object.prototype.hasOwnProperty.call(o, k)) ar[ar.length] = k;
+            return ar;
+        };
+        return ownKeys(o);
+    };
+    return function (mod) {
+        if (mod && mod.__esModule) return mod;
+        var result = {};
+        if (mod != null) for (var k = ownKeys(mod), i = 0; i < k.length; i++) if (k[i] !== "default") __createBinding(result, mod, k[i]);
+        __setModuleDefault(result, mod);
+        return result;
+    };
+})();
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GraphObjectHook = void 0;
+const z = __importStar(require("zod/v4"));
+const tegg_1 = require("@eggjs/tegg");
+const helper_1 = require("@eggjs/tegg/helper");
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+const mcp_client_1 = require("@eggjs/mcp-client");
+const mcp_adapters_1 = require("@langchain/mcp-adapters");
+const langchain_1 = require("langchain");
+const prebuilt_1 = require("@langchain/langgraph/prebuilt");
+class GraphNodeHandler {
+    constructor(nodeMetadata, eggObject) {
+        this.nodeMetadata = nodeMetadata;
+        this.eggObject = eggObject;
+    }
+    async findGraphTools() {
+        var _a;
+        const tools = (_a = this.nodeMetadata.tools) !== null && _a !== void 0 ? _a : [];
+        let dTools = [];
+        for (let i = 0; i < tools.length; i++) {
+            const toolsObj = await helper_1.EggContainerFactory.getOrCreateEggObjectFromClazz(tools[i]);
+            const toolMetadata = tegg_langchain_decorator_1.GraphToolInfoUtil.getGraphToolMetadata(toolsObj.proto.clazz);
+            const ToolDetail = tegg_1.MCPInfoUtil.getMCPToolArgsIndex(toolsObj.proto.clazz, 'execute');
+            if (toolMetadata && ToolDetail) {
+                const tool = new langchain_1.DynamicStructuredTool({
+                    description: toolMetadata.description,
+                    name: toolMetadata.toolName,
+                    func: toolsObj.obj.execute.bind(toolsObj.obj),
+                    schema: z.object(ToolDetail.argsSchema),
+                });
+                dTools = dTools.concat(tool);
+            }
+        }
+        return dTools;
+    }
+    async findMcpServerTools() {
+        var _a;
+        const mcpServers = (_a = this.nodeMetadata.mcpServers) !== null && _a !== void 0 ? _a : [];
+        let sTools = [];
+        for (let i = 0; i < mcpServers.length; i++) {
+            const mcpClientObj = await helper_1.EggContainerFactory.getOrCreateEggObjectFromName(mcp_client_1.MCPClientInjectName, [{
+                    attribute: mcp_client_1.MCPClientQualifierAttribute,
+                    value: mcpServers[i],
+                }]);
+            const tool = await (0, mcp_adapters_1.loadMcpTools)(mcpServers[i], mcpClientObj.obj, {
+                throwOnLoadError: true,
+                prefixToolNameWithServerName: false,
+                additionalToolNamePrefix: '',
+            });
+            sTools = sTools.concat(tool);
+        }
+        return sTools;
+    }
+    async boundTools() {
+        const nodeObj = this.eggObject.obj;
+        const dTools = await this.findGraphTools();
+        const sTools = await this.findMcpServerTools();
+        if (tegg_langchain_decorator_1.TeggToolNode.prototype.isPrototypeOf(nodeObj)) {
+            const toolNode = new prebuilt_1.ToolNode([...dTools, ...sTools]);
+            Object.defineProperty(nodeObj, 'toolNode', {
+                get: () => toolNode,
+            });
+            return;
+        }
+        // 如果用户写了 build 方法，则让他自己绑定
+        if (nodeObj.build) {
+            nodeObj.build([...dTools, ...sTools]);
+        }
+        else {
+            // 否则自动绑定
+            const injectObjects = this.eggObject.proto.injectObjects;
+            for (let i = 0; i < injectObjects.length; i++) {
+                const injectObject = injectObjects[i];
+                const qualifiers = injectObject.qualifiers;
+                for (let j = 0; j < qualifiers.length; j++) {
+                    const qualifier = qualifiers[j];
+                    if (qualifier.attribute === tegg_langchain_decorator_1.ChatModelQualifierAttribute) {
+                        // 找到当前 eggObject 上的 chatModel
+                        const chatModelObj = await helper_1.EggContainerFactory.getOrCreateEggObjectFromName(tegg_langchain_decorator_1.ChatModelInjectName, [{
+                                attribute: tegg_langchain_decorator_1.ChatModelQualifierAttribute,
+                                value: qualifier.value,
+                            }]);
+                        const chatModel = chatModelObj.obj;
+                        // 绑定
+                        const boundChatModel = chatModel.bindTools([...dTools, ...sTools]);
+                        // 劫持
+                        Object.defineProperty(nodeObj, injectObject.objName, {
+                            get: () => boundChatModel,
+                        });
+                    }
+                }
+            }
+        }
+    }
+}
+class GraphObjectHook {
+    async postCreate(_, eggObject) {
+        const nodeMetadata = eggObject.proto.getMetaData(tegg_langchain_decorator_1.GRAPH_NODE_METADATA);
+        // 找到 graph node
+        if (nodeMetadata) {
+            const handler = new GraphNodeHandler(nodeMetadata, eggObject);
+            await handler.boundTools();
+            return;
+        }
+    }
+}
+exports.GraphObjectHook = GraphObjectHook;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiR3JhcGhPYmplY3RIb29rLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiR3JhcGhPYmplY3RIb29rLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7OztBQUFBLDBDQUE0QjtBQUM1QixzQ0FBOEQ7QUFDOUQsK0NBQXNIO0FBQ3RILDhFQVN5QztBQUV6QyxrREFBcUY7QUFDckYsMERBQXVEO0FBRXZELHlDQUFrRDtBQUNsRCw0REFBeUQ7QUFFekQsTUFBTSxnQkFBZ0I7SUFJcEIsWUFBWSxZQUErQixFQUFFLFNBQW9CO1FBQy9ELElBQUksQ0FBQyxZQUFZLEdBQUcsWUFBWSxDQUFDO1FBQ2pDLElBQUksQ0FBQyxTQUFTLEdBQUcsU0FBUyxDQUFDO0lBQzdCLENBQUM7SUFFRCxLQUFLLENBQUMsY0FBYzs7UUFDbEIsTUFBTSxLQUFLLEdBQUcsTUFBQSxJQUFJLENBQUMsWUFBWSxDQUFDLEtBQUssbUNBQUksRUFBRSxDQUFDO1FBQzVDLElBQUksTUFBTSxHQUFvRCxFQUFFLENBQUM7UUFDakUsS0FBSyxJQUFJLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxHQUFHLEtBQUssQ0FBQyxNQUFNLEVBQUUsQ0FBQyxFQUFFLEVBQUUsQ0FBQztZQUN0QyxNQUFNLFFBQVEsR0FBRyxNQUFNLDRCQUFtQixDQUFDLDZCQUE2QixDQUFDLEtBQUssQ0FBQyxDQUFDLENBQUMsQ0FBQyxDQUFDO1lBQ25GLE1BQU0sWUFBWSxHQUFHLDRDQUFpQixDQUFDLG9CQUFvQixDQUFFLFFBQVEsQ0FBQyxLQUEwQyxDQUFDLEtBQU0sQ0FBQyxDQUFDO1lBQ3pILE1BQU0sVUFBVSxHQUFHLGtCQUFXLENBQUMsbUJBQW1CLENBQUUsUUFBUSxDQUFDLEtBQTBDLENBQUMsS0FBTSxFQUFFLFNBQVMsQ0FBQyxDQUFDO1lBQzNILElBQUksWUFBWSxJQUFJLFVBQVUsRUFBRSxDQUFDO2dCQUMvQixNQUFNLElBQUksR0FBRyxJQUFJLGlDQUFxQixDQUFDO29CQUNyQyxXQUFXLEVBQUUsWUFBWSxDQUFDLFdBQVc7b0JBQ3JDLElBQUksRUFBRSxZQUFZLENBQUMsUUFBUTtvQkFDM0IsSUFBSSxFQUFHLFFBQVEsQ0FBQyxHQUFrQyxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQztvQkFDN0UsTUFBTSxFQUFFLENBQUMsQ0FBQyxNQUFNLENBQUMsVUFBVSxDQUFDLFVBQVUsQ0FBUTtpQkFDL0MsQ0FBQyxDQUFDO2dCQUNILE1BQU0sR0FBRyxNQUFNLENBQUMsTUFBTSxDQUFDLElBQUksQ0FBQyxDQUFDO1lBQy9CLENBQUM7UUFDSCxDQUFDO1FBRUQsT0FBTyxNQUFNLENBQUM7SUFDaEIsQ0FBQztJQUVELEtBQUssQ0FBQyxrQkFBa0I7O1FBQ3RCLE1BQU0sVUFBVSxHQUFHLE1BQUEsSUFBSSxDQUFDLFlBQVksQ0FBQyxVQUFVLG1DQUFJLEVBQUUsQ0FBQztRQUN0RCxJQUFJLE1BQU0sR0FBb0QsRUFBRSxDQUFDO1FBQ2pFLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxVQUFVLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7WUFDM0MsTUFBTSxZQUFZLEdBQUcsTUFBTSw0QkFBbUIsQ0FBQyw0QkFBNEIsQ0FBQyxnQ0FBbUIsRUFBRSxDQUFDO29CQUNoRyxTQUFTLEVBQUUsd0NBQTJCO29CQUN0QyxLQUFLLEVBQUUsVUFBVSxDQUFDLENBQUMsQ0FBQztpQkFDckIsQ0FBQyxDQUFDLENBQUM7WUFDSixNQUFNLElBQUksR0FBRyxNQUFNLElBQUEsMkJBQVksRUFBQyxVQUFVLENBQUMsQ0FBQyxDQUFDLEVBQUUsWUFBWSxDQUFDLEdBQVUsRUFBRTtnQkFDdEUsZ0JBQWdCLEVBQUUsSUFBSTtnQkFDdEIsNEJBQTRCLEVBQUUsS0FBSztnQkFDbkMsd0JBQXdCLEVBQUUsRUFBRTthQUM3QixDQUFDLENBQUM7WUFDSCxNQUFNLEdBQUcsTUFBTSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUMvQixDQUFDO1FBQ0QsT0FBTyxNQUFNLENBQUM7SUFDaEIsQ0FBQztJQUVELEtBQUssQ0FBQyxVQUFVO1FBQ2QsTUFBTSxPQUFPLEdBQUcsSUFBSSxDQUFDLFNBQVMsQ0FBQyxHQUFzQixDQUFDO1FBQ3RELE1BQU0sTUFBTSxHQUFHLE1BQU0sSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO1FBQzNDLE1BQU0sTUFBTSxHQUFHLE1BQU0sSUFBSSxDQUFDLGtCQUFrQixFQUFFLENBQUM7UUFFL0MsSUFBSSx1Q0FBWSxDQUFDLFNBQVMsQ0FBQyxhQUFhLENBQUMsT0FBTyxDQUFDLEVBQUUsQ0FBQztZQUNsRCxNQUFNLFFBQVEsR0FBRyxJQUFJLG1CQUFRLENBQUMsQ0FBRSxHQUFJLE1BQWtDLEVBQUUsR0FBSSxNQUFrQyxDQUFFLENBQUMsQ0FBQztZQUNsSCxNQUFNLENBQUMsY0FBYyxDQUFDLE9BQU8sRUFBRSxVQUFVLEVBQUU7Z0JBQ3pDLEdBQUcsRUFBRSxHQUFHLEVBQUUsQ0FBQyxRQUFRO2FBQ3BCLENBQUMsQ0FBQztZQUNILE9BQU87UUFDVCxDQUFDO1FBRUQsMEJBQTBCO1FBQzFCLElBQUksT0FBTyxDQUFDLEtBQUssRUFBRSxDQUFDO1lBQ2xCLE9BQU8sQ0FBQyxLQUFNLENBQUMsQ0FBRSxHQUFHLE1BQU0sRUFBRSxHQUFHLE1BQU0sQ0FBRSxDQUFDLENBQUM7UUFDM0MsQ0FBQzthQUFNLENBQUM7WUFDTixTQUFTO1lBQ1QsTUFBTSxhQUFhLEdBQUcsSUFBSSxDQUFDLFNBQVMsQ0FBQyxLQUFLLENBQUMsYUFBYSxDQUFDO1lBQ3pELEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxhQUFhLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7Z0JBQzlDLE1BQU0sWUFBWSxHQUFHLGFBQWEsQ0FBQyxDQUFDLENBQUMsQ0FBQztnQkFDdEMsTUFBTSxVQUFVLEdBQUcsWUFBWSxDQUFDLFVBQVUsQ0FBQztnQkFDM0MsS0FBSyxJQUFJLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxHQUFHLFVBQVUsQ0FBQyxNQUFNLEVBQUUsQ0FBQyxFQUFFLEVBQUUsQ0FBQztvQkFDM0MsTUFBTSxTQUFTLEdBQUcsVUFBVSxDQUFDLENBQUMsQ0FBQyxDQUFDO29CQUNoQyxJQUFJLFNBQVMsQ0FBQyxTQUFTLEtBQUssc0RBQTJCLEVBQUUsQ0FBQzt3QkFDeEQsOEJBQThCO3dCQUM5QixNQUFNLFlBQVksR0FBRyxNQUFNLDRCQUFtQixDQUFDLDRCQUE0QixDQUFDLDhDQUFtQixFQUFFLENBQUM7Z0NBQ2hHLFNBQVMsRUFBRSxzREFBMkI7Z0NBQ3RDLEtBQUssRUFBRSxTQUFTLENBQUMsS0FBSzs2QkFDdkIsQ0FBQyxDQUFDLENBQUM7d0JBQ0osTUFBTSxTQUFTLEdBQUcsWUFBWSxDQUFDLEdBQXdCLENBQUM7d0JBQ3hELEtBQUs7d0JBQ0wsTUFBTSxjQUFjLEdBQUcsU0FBUyxDQUFDLFNBQVMsQ0FBQyxDQUFFLEdBQUcsTUFBTSxFQUFFLEdBQUcsTUFBTSxDQUFFLENBQUMsQ0FBQzt3QkFDckUsS0FBSzt3QkFDTCxNQUFNLENBQUMsY0FBYyxDQUFDLE9BQU8sRUFBRSxZQUFZLENBQUMsT0FBTyxFQUFFOzRCQUNuRCxHQUFHLEVBQUUsR0FBRyxFQUFFLENBQUMsY0FBYzt5QkFDMUIsQ0FBQyxDQUFDO29CQUNMLENBQUM7Z0JBQ0gsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztDQUNGO0FBRUQsTUFBYSxlQUFlO0lBRTFCLEtBQUssQ0FBQyxVQUFVLENBQUMsQ0FBNEIsRUFBRSxTQUFvQjtRQUNqRSxNQUFNLFlBQVksR0FBRyxTQUFTLENBQUMsS0FBSyxDQUFDLFdBQVcsQ0FBb0IsOENBQW1CLENBQUMsQ0FBQztRQUN6RixnQkFBZ0I7UUFDaEIsSUFBSSxZQUFZLEVBQUUsQ0FBQztZQUNqQixNQUFNLE9BQU8sR0FBRyxJQUFJLGdCQUFnQixDQUFDLFlBQVksRUFBRSxTQUFTLENBQUMsQ0FBQztZQUM5RCxNQUFNLE9BQU8sQ0FBQyxVQUFVLEVBQUUsQ0FBQztZQUMzQixPQUFPO1FBQ1QsQ0FBQztJQUNILENBQUM7Q0FDRjtBQVhELDBDQVdDIn0=

package/lib/graph/GraphPrototypeHook.d.ts

@@ -0,0 +1,20 @@
+import { EggProtoImplClass, LifecycleHook } from '@eggjs/tegg';
+import { EggPrototype, EggPrototypeLifecycleContext } from '@eggjs/tegg-metadata';
+export declare class GraphPrototypeHook implements LifecycleHook<EggPrototypeLifecycleContext, EggPrototype> {
+    toolProtoMap: Map<EggProtoImplClass, EggPrototype>;
+    nodeProtoMap: Map<EggPrototype, EggProtoImplClass[]>;
+    nodeMcpServerProtoMap: Map<EggPrototype, string[]>;
+    mcpServerProto: EggPrototype;
+    checkPointProto: EggPrototype;
+    checkPointList: EggProtoImplClass[];
+    checkPointProtoMap: Map<EggProtoImplClass, EggPrototype>;
+    checkPointGraphMap: Map<EggProtoImplClass, EggPrototype[]>;
+    graphCheckpointProtoMap: Map<EggPrototype, string>;
+    graphNodeProtoMap: Map<EggPrototype, EggProtoImplClass[]>;
+    nodeProto: Map<EggProtoImplClass, EggPrototype>;
+    graphEdgeProtoMap: Map<EggPrototype, EggProtoImplClass[]>;
+    edgeProto: Map<EggProtoImplClass, EggPrototype>;
+    postCreate(ctx: EggPrototypeLifecycleContext, proto: EggPrototype): Promise<void>;
+    makeNodeDependencies(ctx: EggPrototypeLifecycleContext, proto: EggPrototype): Promise<void>;
+    makeGraphDependencies(ctx: EggPrototypeLifecycleContext, proto: EggPrototype): Promise<void>;
+}

package/lib/graph/GraphPrototypeHook.js

@@ -0,0 +1,253 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GraphPrototypeHook = void 0;
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+const mcp_client_1 = require("@eggjs/mcp-client");
+const langgraph_1 = require("@langchain/langgraph");
+class GraphPrototypeHook {
+    constructor() {
+        this.toolProtoMap = new Map();
+        this.nodeProtoMap = new Map();
+        this.nodeMcpServerProtoMap = new Map();
+        this.checkPointList = [];
+        this.checkPointProtoMap = new Map();
+        this.checkPointGraphMap = new Map();
+        this.graphCheckpointProtoMap = new Map();
+        this.graphNodeProtoMap = new Map();
+        this.nodeProto = new Map();
+        this.graphEdgeProtoMap = new Map();
+        this.edgeProto = new Map();
+    }
+    async postCreate(ctx, proto) {
+        await this.makeNodeDependencies(ctx, proto);
+        await this.makeGraphDependencies(ctx, proto);
+    }
+    async makeNodeDependencies(ctx, proto) {
+        var _a, _b;
+        const nodeMetadata = tegg_langchain_decorator_1.GraphNodeInfoUtil.getGraphNodeMetadata(ctx.clazz);
+        if (nodeMetadata) {
+            const tools = (_a = nodeMetadata.tools) !== null && _a !== void 0 ? _a : [];
+            // 找到所有已经创建好的 tool proto
+            for (let i = 0; i < tools.length; i++) {
+                if (this.toolProtoMap.has(tools[i])) {
+                    const toolProto = this.toolProtoMap.get(tools[i]);
+                    proto.injectObjects.push({
+                        refName: `__GRAPH_TOOL_${String(toolProto.name)}__`,
+                        objName: toolProto.name,
+                        qualifiers: [],
+                        proto: toolProto,
+                    });
+                }
+            }
+            const mcpServers = (_b = nodeMetadata.mcpServers) !== null && _b !== void 0 ? _b : [];
+            if (this.mcpServerProto) {
+                for (let i = 0; i < mcpServers.length; i++) {
+                    proto.injectObjects.push({
+                        refName: `__GRAPH_MCPSERVER_${String(mcpServers[i])}__`,
+                        objName: mcp_client_1.MCPClientInjectName,
+                        qualifiers: [{
+                                attribute: mcp_client_1.MCPClientQualifierAttribute,
+                                value: mcpServers[i],
+                            }],
+                        proto: this.mcpServerProto,
+                    });
+                }
+            }
+            // 存入 nodeProtoMap，等 tool 创建好了再注入
+            this.nodeProtoMap.set(proto, tools);
+            // 存入 nodeMcpServerProtoMap，等 mcpServer 创建好了再注入
+            this.nodeMcpServerProtoMap.set(proto, mcpServers);
+        }
+        const toolMetadata = tegg_langchain_decorator_1.GraphToolInfoUtil.getGraphToolMetadata(ctx.clazz);
+        if (toolMetadata) {
+            // 注入到所有用到这个 tool 的 node 上
+            for (const [nodeProto, toolClazzList] of this.nodeProtoMap.entries()) {
+                if (toolClazzList.includes(ctx.clazz)) {
+                    nodeProto.injectObjects.push({
+                        refName: `__GRAPH_TOOL_${String(proto.name)}__`,
+                        objName: proto.name,
+                        qualifiers: [],
+                        proto,
+                    });
+                }
+            }
+            // 记录已经创建好的 tool proto，以便后续 node 注入
+            this.toolProtoMap.set(ctx.clazz, proto);
+        }
+        if (proto.name === mcp_client_1.MCPClientInjectName) {
+            this.mcpServerProto = proto;
+            // 注入到所有用到 mcpServer 的 node 上
+            for (const [nodeProto, mcpServers] of this.nodeMcpServerProtoMap.entries()) {
+                for (let i = 0; i < mcpServers.length; i++) {
+                    nodeProto.injectObjects.push({
+                        refName: `__GRAPH_MCPSERVER_${String(mcpServers[i])}__`,
+                        objName: mcp_client_1.MCPClientInjectName,
+                        qualifiers: [{
+                                attribute: mcp_client_1.MCPClientQualifierAttribute,
+                                value: mcpServers[i],
+                            }],
+                        proto,
+                    });
+                }
+            }
+        }
+    }
+    async makeGraphDependencies(ctx, proto) {
+        var _a, _b, _c;
+        const graphMetadata = tegg_langchain_decorator_1.GraphInfoUtil.getGraphMetadata(ctx.clazz);
+        if (graphMetadata) {
+            if (graphMetadata.checkpoint) {
+                // module.yml ChatCheckpointSaver
+                if (typeof graphMetadata.checkpoint === 'string') {
+                    // 如果 ChatCheckpointSaver 已经创建好了，则直接注入
+                    if (this.checkPointProto) {
+                        proto.injectObjects.push({
+                            refName: 'checkpoint',
+                            objName: tegg_langchain_decorator_1.ChatCheckpointSaverInjectName,
+                            qualifiers: [{
+                                    attribute: tegg_langchain_decorator_1.ChatCheckpointSaverQualifierAttribute,
+                                    value: graphMetadata.checkpoint,
+                                }],
+                            proto: this.checkPointProto,
+                        });
+                    }
+                    // 如果没有创建好，则记录下来，等 ChatCheckpointSaver 创建好了再注入
+                    this.graphCheckpointProtoMap.set(proto, graphMetadata.checkpoint);
+                }
+                else {
+                    // 用户自己编写的 CheckpointSaver class
+                    if (this.checkPointProtoMap.has(graphMetadata.checkpoint)) {
+                        // 如果已经创建好了，则直接注入
+                        const checkPointProto = this.checkPointProtoMap.get(graphMetadata.checkpoint);
+                        proto.injectObjects.push({
+                            refName: 'checkpoint',
+                            objName: checkPointProto.name,
+                            qualifiers: [],
+                            proto: checkPointProto,
+                        });
+                    }
+                    // 如果没有创建好，则记录下来，等 CheckpointSaver 创建好了再注入
+                    this.checkPointList.push(graphMetadata.checkpoint);
+                    if (this.checkPointGraphMap.has(graphMetadata.checkpoint)) {
+                        this.checkPointGraphMap.get(graphMetadata.checkpoint).push(proto);
+                    }
+                    else {
+                        this.checkPointGraphMap.set(graphMetadata.checkpoint, [proto]);
+                    }
+                }
+                this.graphNodeProtoMap.set(proto, []);
+                this.graphEdgeProtoMap.set(proto, []);
+                // 将所有 node 注入到 graph 上
+                for (const nodeProto of (_a = graphMetadata.nodes) !== null && _a !== void 0 ? _a : []) {
+                    // 如果有，则是扫描过了，直接注入
+                    if (this.nodeProto.has(nodeProto)) {
+                        proto.injectObjects.push({
+                            refName: `__GRAPH_NODE_${String(nodeProto)}__`,
+                            objName: nodeProto.name,
+                            qualifiers: [],
+                            proto: this.nodeProto.get(nodeProto),
+                        });
+                    }
+                    else {
+                        // 如果没有，记录下来，等 node 创建好了再注入
+                        this.graphNodeProtoMap.get(proto).push(nodeProto);
+                    }
+                }
+                // 将所有 edge 注入到 graph 上
+                for (const edgeProto of (_b = graphMetadata.edges) !== null && _b !== void 0 ? _b : []) {
+                    // 如果有，则是扫描过了，直接注入
+                    if (this.edgeProto.has(edgeProto)) {
+                        proto.injectObjects.push({
+                            refName: `__GRAPH_EDGE_${String(edgeProto)}__`,
+                            objName: edgeProto.name,
+                            qualifiers: [],
+                            proto: this.edgeProto.get(edgeProto),
+                        });
+                    }
+                    else {
+                        // 如果没有，记录下来，等 edge 创建好了再注入
+                        this.graphEdgeProtoMap.get(proto).push(edgeProto);
+                    }
+                }
+            }
+        }
+        // 此时创建的是 ChatCheckpointSaver
+        if (proto.name === tegg_langchain_decorator_1.ChatCheckpointSaverInjectName) {
+            this.checkPointProto = proto;
+            // 注入到所有用到 checkpoint 的 graph 上
+            for (const [graphProto, checkpoint] of this.graphCheckpointProtoMap.entries()) {
+                if (checkpoint === (graphMetadata === null || graphMetadata === void 0 ? void 0 : graphMetadata.checkpoint)) {
+                    if (!graphProto.injectObjects.find(injectObject => injectObject.refName === 'checkpoint')) {
+                        continue;
+                    }
+                    graphProto.injectObjects.push({
+                        refName: 'checkpoint',
+                        objName: tegg_langchain_decorator_1.ChatCheckpointSaverInjectName,
+                        qualifiers: [{
+                                attribute: tegg_langchain_decorator_1.ChatCheckpointSaverQualifierAttribute,
+                                value: checkpoint,
+                            }],
+                        proto,
+                    });
+                }
+            }
+        }
+        // 此时创建的是用户自定义的 BaseCheckpointSaver 的子类
+        if (langgraph_1.BaseCheckpointSaver.isPrototypeOf(ctx.clazz)) {
+            // 如果有 graph 依赖这个 CheckpointSaver，则注入
+            const graphProto = (_c = this.checkPointGraphMap.get(ctx.clazz)) !== null && _c !== void 0 ? _c : [];
+            for (let i = 0; i < graphProto.length; i++) {
+                if (!graphProto[i].injectObjects.find(injectObject => injectObject.refName === 'checkpoint')) {
+                    continue;
+                }
+                graphProto[i].injectObjects.push({
+                    refName: 'checkpoint',
+                    objName: proto.name,
+                    qualifiers: [],
+                    proto,
+                });
+            }
+            this.checkPointProtoMap.set(ctx.clazz, proto);
+        }
+        // 此时创建的是 graph node
+        const nodeMeta = tegg_langchain_decorator_1.GraphNodeInfoUtil.getGraphNodeMetadata(ctx.clazz);
+        if (nodeMeta) {
+            this.nodeProto.set(ctx.clazz, proto);
+            // 注入到所有依赖这个 graph node 的 graph
+            for (const [graphProto, nodeProtos] of this.graphNodeProtoMap.entries()) {
+                if (nodeProtos.includes(ctx.clazz)) {
+                    if (!graphProto.injectObjects.find(injectObject => injectObject.refName === `__GRAPH_NODE_${String(ctx.clazz)}__`)) {
+                        continue;
+                    }
+                    graphProto.injectObjects.push({
+                        refName: `__GRAPH_NODE_${String(ctx.clazz)}__`,
+                        objName: proto.name,
+                        qualifiers: [],
+                        proto,
+                    });
+                }
+            }
+        }
+        // 此时创建的是 graph edge
+        const edgeMeta = tegg_langchain_decorator_1.GraphEdgeInfoUtil.getGraphEdgeMetadata(ctx.clazz);
+        if (edgeMeta) {
+            this.edgeProto.set(ctx.clazz, proto);
+            // 注入到所有依赖这个 graph edge 的 graph
+            for (const [graphProto, edgeProtos] of this.graphEdgeProtoMap.entries()) {
+                if (edgeProtos.includes(ctx.clazz)) {
+                    if (!graphProto.injectObjects.find(injectObject => injectObject.refName === `__GRAPH_EDGE_${String(ctx.clazz)}__`)) {
+                        continue;
+                    }
+                    graphProto.injectObjects.push({
+                        refName: `__GRAPH_EDGE_${String(ctx.clazz)}__`,
+                        objName: proto.name,
+                        qualifiers: [],
+                        proto,
+                    });
+                }
+            }
+        }
+    }
+}
+exports.GraphPrototypeHook = GraphPrototypeHook;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiR3JhcGhQcm90b3R5cGVIb29rLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiR3JhcGhQcm90b3R5cGVIb29rLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7OztBQUVBLDhFQUErTDtBQUMvTCxrREFBcUY7QUFDckYsb0RBQTJEO0FBRTNELE1BQWEsa0JBQWtCO0lBQS9CO1FBQ0UsaUJBQVksR0FBeUMsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUMvRCxpQkFBWSxHQUEyQyxJQUFJLEdBQUcsRUFBRSxDQUFDO1FBQ2pFLDBCQUFxQixHQUFnQyxJQUFJLEdBQUcsRUFBRSxDQUFDO1FBSS9ELG1CQUFjLEdBQXdCLEVBQUUsQ0FBQztRQUN6Qyx1QkFBa0IsR0FBeUMsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUNyRSx1QkFBa0IsR0FBMkMsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUN2RSw0QkFBdUIsR0FBOEIsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUUvRCxzQkFBaUIsR0FBMkMsSUFBSSxHQUFHLEVBQUUsQ0FBQztRQUN0RSxjQUFTLEdBQXlDLElBQUksR0FBRyxFQUFFLENBQUM7UUFFNUQsc0JBQWlCLEdBQTJDLElBQUksR0FBRyxFQUFFLENBQUM7UUFDdEUsY0FBUyxHQUF5QyxJQUFJLEdBQUcsRUFBRSxDQUFDO0lBbVA5RCxDQUFDO0lBalBDLEtBQUssQ0FBQyxVQUFVLENBQUMsR0FBaUMsRUFBRSxLQUFtQjtRQUNyRSxNQUFNLElBQUksQ0FBQyxvQkFBb0IsQ0FBQyxHQUFHLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFDNUMsTUFBTSxJQUFJLENBQUMscUJBQXFCLENBQUMsR0FBRyxFQUFFLEtBQUssQ0FBQyxDQUFDO0lBQy9DLENBQUM7SUFFRCxLQUFLLENBQUMsb0JBQW9CLENBQUMsR0FBaUMsRUFBRSxLQUFtQjs7UUFDL0UsTUFBTSxZQUFZLEdBQUcsNENBQWlCLENBQUMsb0JBQW9CLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ3ZFLElBQUksWUFBWSxFQUFFLENBQUM7WUFDakIsTUFBTSxLQUFLLEdBQUcsTUFBQSxZQUFZLENBQUMsS0FBSyxtQ0FBSSxFQUFFLENBQUM7WUFDdkMsd0JBQXdCO1lBQ3hCLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxLQUFLLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7Z0JBQ3RDLElBQUksSUFBSSxDQUFDLFlBQVksQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQztvQkFDcEMsTUFBTSxTQUFTLEdBQUcsSUFBSSxDQUFDLFlBQVksQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLENBQUMsQ0FBQyxDQUFFLENBQUM7b0JBQ25ELEtBQUssQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDO3dCQUN2QixPQUFPLEVBQUUsZ0JBQWdCLE1BQU0sQ0FBQyxTQUFTLENBQUMsSUFBSSxDQUFDLElBQUk7d0JBQ25ELE9BQU8sRUFBRSxTQUFTLENBQUMsSUFBSTt3QkFDdkIsVUFBVSxFQUFFLEVBQUU7d0JBQ2QsS0FBSyxFQUFFLFNBQVM7cUJBQ2pCLENBQUMsQ0FBQztnQkFDTCxDQUFDO1lBQ0gsQ0FBQztZQUVELE1BQU0sVUFBVSxHQUFHLE1BQUEsWUFBWSxDQUFDLFVBQVUsbUNBQUksRUFBRSxDQUFDO1lBQ2pELElBQUksSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO2dCQUN4QixLQUFLLElBQUksQ0FBQyxHQUFHLENBQUMsRUFBRSxDQUFDLEdBQUcsVUFBVSxDQUFDLE1BQU0sRUFBRSxDQUFDLEVBQUUsRUFBRSxDQUFDO29CQUMzQyxLQUFLLENBQUMsYUFBYSxDQUFDLElBQUksQ0FBQzt3QkFDdkIsT0FBTyxFQUFFLHFCQUFxQixNQUFNLENBQUMsVUFBVSxDQUFDLENBQUMsQ0FBQyxDQUFDLElBQUk7d0JBQ3ZELE9BQU8sRUFBRSxnQ0FBbUI7d0JBQzVCLFVBQVUsRUFBRSxDQUFDO2dDQUNYLFNBQVMsRUFBRSx3Q0FBMkI7Z0NBQ3RDLEtBQUssRUFBRSxVQUFVLENBQUMsQ0FBQyxDQUFDOzZCQUNyQixDQUFDO3dCQUNGLEtBQUssRUFBRSxJQUFJLENBQUMsY0FBYztxQkFDM0IsQ0FBQyxDQUFDO2dCQUNMLENBQUM7WUFDSCxDQUFDO1lBRUQsaUNBQWlDO1lBQ2pDLElBQUksQ0FBQyxZQUFZLENBQUMsR0FBRyxDQUFDLEtBQUssRUFBRSxLQUFLLENBQUMsQ0FBQztZQUNwQywrQ0FBK0M7WUFDL0MsSUFBSSxDQUFDLHFCQUFxQixDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsVUFBVSxDQUFDLENBQUM7UUFDcEQsQ0FBQztRQUVELE1BQU0sWUFBWSxHQUFHLDRDQUFpQixDQUFDLG9CQUFvQixDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsQ0FBQztRQUN2RSxJQUFJLFlBQVksRUFBRSxDQUFDO1lBQ2pCLDBCQUEwQjtZQUMxQixLQUFLLE1BQU0sQ0FBRSxTQUFTLEVBQUUsYUFBYSxDQUFFLElBQUksSUFBSSxDQUFDLFlBQVksQ0FBQyxPQUFPLEVBQUUsRUFBRSxDQUFDO2dCQUN2RSxJQUFJLGFBQWEsQ0FBQyxRQUFRLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxFQUFFLENBQUM7b0JBQ3RDLFNBQVMsQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDO3dCQUMzQixPQUFPLEVBQUUsZ0JBQWdCLE1BQU0sQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLElBQUk7d0JBQy9DLE9BQU8sRUFBRSxLQUFLLENBQUMsSUFBSTt3QkFDbkIsVUFBVSxFQUFFLEVBQUU7d0JBQ2QsS0FBSztxQkFDTixDQUFDLENBQUM7Z0JBQ0wsQ0FBQztZQUNILENBQUM7WUFDRCxtQ0FBbUM7WUFDbkMsSUFBSSxDQUFDLFlBQVksQ0FBQyxHQUFHLENBQUMsR0FBRyxDQUFDLEtBQUssRUFBRSxLQUFLLENBQUMsQ0FBQztRQUMxQyxDQUFDO1FBRUQsSUFBSSxLQUFLLENBQUMsSUFBSSxLQUFLLGdDQUFtQixFQUFFLENBQUM7WUFDdkMsSUFBSSxDQUFDLGNBQWMsR0FBRyxLQUFLLENBQUM7WUFDNUIsNkJBQTZCO1lBQzdCLEtBQUssTUFBTSxDQUFFLFNBQVMsRUFBRSxVQUFVLENBQUUsSUFBSSxJQUFJLENBQUMscUJBQXFCLENBQUMsT0FBTyxFQUFFLEVBQUUsQ0FBQztnQkFDN0UsS0FBSyxJQUFJLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxHQUFHLFVBQVUsQ0FBQyxNQUFNLEVBQUUsQ0FBQyxFQUFFLEVBQUUsQ0FBQztvQkFDM0MsU0FBUyxDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUM7d0JBQzNCLE9BQU8sRUFBRSxxQkFBcUIsTUFBTSxDQUFDLFVBQVUsQ0FBQyxDQUFDLENBQUMsQ0FBQyxJQUFJO3dCQUN2RCxPQUFPLEVBQUUsZ0NBQW1CO3dCQUM1QixVQUFVLEVBQUUsQ0FBQztnQ0FDWCxTQUFTLEVBQUUsd0NBQTJCO2dDQUN0QyxLQUFLLEVBQUUsVUFBVSxDQUFDLENBQUMsQ0FBQzs2QkFDckIsQ0FBQzt3QkFDRixLQUFLO3FCQUNOLENBQUMsQ0FBQztnQkFDTCxDQUFDO1lBQ0gsQ0FBQztRQUNILENBQUM7SUFDSCxDQUFDO0lBR0QsS0FBSyxDQUFDLHFCQUFxQixDQUFDLEdBQWlDLEVBQUUsS0FBbUI7O1FBRWhGLE1BQU0sYUFBYSxHQUFHLHdDQUFhLENBQUMsZ0JBQWdCLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ2hFLElBQUksYUFBYSxFQUFFLENBQUM7WUFDbEIsSUFBSSxhQUFhLENBQUMsVUFBVSxFQUFFLENBQUM7Z0JBQzdCLGlDQUFpQztnQkFDakMsSUFBSSxPQUFPLGFBQWEsQ0FBQyxVQUFVLEtBQUssUUFBUSxFQUFFLENBQUM7b0JBQ2pELHNDQUFzQztvQkFDdEMsSUFBSSxJQUFJLENBQUMsZUFBZSxFQUFFLENBQUM7d0JBQ3pCLEtBQUssQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDOzRCQUN2QixPQUFPLEVBQUUsWUFBWTs0QkFDckIsT0FBTyxFQUFFLHdEQUE2Qjs0QkFDdEMsVUFBVSxFQUFFLENBQUM7b0NBQ1gsU0FBUyxFQUFFLGdFQUFxQztvQ0FDaEQsS0FBSyxFQUFFLGFBQWEsQ0FBQyxVQUFVO2lDQUNoQyxDQUFDOzRCQUNGLEtBQUssRUFBRSxJQUFJLENBQUMsZUFBZTt5QkFDNUIsQ0FBQyxDQUFDO29CQUNMLENBQUM7b0JBRUQsOENBQThDO29CQUM5QyxJQUFJLENBQUMsdUJBQXVCLENBQUMsR0FBRyxDQUFDLEtBQUssRUFBRSxhQUFhLENBQUMsVUFBVSxDQUFDLENBQUM7Z0JBQ3BFLENBQUM7cUJBQU0sQ0FBQztvQkFDTixnQ0FBZ0M7b0JBQ2hDLElBQUksSUFBSSxDQUFDLGtCQUFrQixDQUFDLEdBQUcsQ0FBQyxhQUFhLENBQUMsVUFBVSxDQUFDLEVBQUUsQ0FBQzt3QkFDMUQsaUJBQWlCO3dCQUNqQixNQUFNLGVBQWUsR0FBRyxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLGFBQWEsQ0FBQyxVQUFVLENBQUUsQ0FBQzt3QkFDL0UsS0FBSyxDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUM7NEJBQ3ZCLE9BQU8sRUFBRSxZQUFZOzRCQUNyQixPQUFPLEVBQUUsZUFBZSxDQUFDLElBQUk7NEJBQzdCLFVBQVUsRUFBRSxFQUFFOzRCQUNkLEtBQUssRUFBRSxlQUFlO3lCQUN2QixDQUFDLENBQUM7b0JBQ0wsQ0FBQztvQkFFRCwwQ0FBMEM7b0JBQzFDLElBQUksQ0FBQyxjQUFjLENBQUMsSUFBSSxDQUFDLGFBQWEsQ0FBQyxVQUFVLENBQUMsQ0FBQztvQkFDbkQsSUFBSSxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLGFBQWEsQ0FBQyxVQUFVLENBQUMsRUFBRSxDQUFDO3dCQUMxRCxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLGFBQWEsQ0FBQyxVQUFVLENBQUUsQ0FBQyxJQUFJLENBQUMsS0FBSyxDQUFDLENBQUM7b0JBQ3JFLENBQUM7eUJBQU0sQ0FBQzt3QkFDTixJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLGFBQWEsQ0FBQyxVQUFVLEVBQUUsQ0FBRSxLQUFLLENBQUUsQ0FBQyxDQUFDO29CQUNuRSxDQUFDO2dCQUNILENBQUM7Z0JBRUQsSUFBSSxDQUFDLGlCQUFpQixDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsRUFBRSxDQUFDLENBQUM7Z0JBQ3RDLElBQUksQ0FBQyxpQkFBaUIsQ0FBQyxHQUFHLENBQUMsS0FBSyxFQUFFLEVBQUUsQ0FBQyxDQUFDO2dCQUV0Qyx1QkFBdUI7Z0JBQ3ZCLEtBQUssTUFBTSxTQUFTLElBQUksTUFBQSxhQUFhLENBQUMsS0FBSyxtQ0FBSSxFQUFFLEVBQUUsQ0FBQztvQkFDbEQsa0JBQWtCO29CQUNsQixJQUFJLElBQUksQ0FBQyxTQUFTLENBQUMsR0FBRyxDQUFDLFNBQVMsQ0FBQyxFQUFFLENBQUM7d0JBQ2xDLEtBQUssQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDOzRCQUN2QixPQUFPLEVBQUUsZ0JBQWdCLE1BQU0sQ0FBQyxTQUFTLENBQUMsSUFBSTs0QkFDOUMsT0FBTyxFQUFFLFNBQVMsQ0FBQyxJQUFJOzRCQUN2QixVQUFVLEVBQUUsRUFBRTs0QkFDZCxLQUFLLEVBQUUsSUFBSSxDQUFDLFNBQVMsQ0FBQyxHQUFHLENBQUMsU0FBUyxDQUFFO3lCQUN0QyxDQUFDLENBQUM7b0JBQ0wsQ0FBQzt5QkFBTSxDQUFDO3dCQUNOLDJCQUEyQjt3QkFDM0IsSUFBSSxDQUFDLGlCQUFpQixDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUUsQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUM7b0JBQ3JELENBQUM7Z0JBQ0gsQ0FBQztnQkFFRCx1QkFBdUI7Z0JBQ3ZCLEtBQUssTUFBTSxTQUFTLElBQUksTUFBQSxhQUFhLENBQUMsS0FBSyxtQ0FBSSxFQUFFLEVBQUUsQ0FBQztvQkFDbEQsa0JBQWtCO29CQUNsQixJQUFJLElBQUksQ0FBQyxTQUFTLENBQUMsR0FBRyxDQUFDLFNBQVMsQ0FBQyxFQUFFLENBQUM7d0JBQ2xDLEtBQUssQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDOzRCQUN2QixPQUFPLEVBQUUsZ0JBQWdCLE1BQU0sQ0FBQyxTQUFTLENBQUMsSUFBSTs0QkFDOUMsT0FBTyxFQUFFLFNBQVMsQ0FBQyxJQUFJOzRCQUN2QixVQUFVLEVBQUUsRUFBRTs0QkFDZCxLQUFLLEVBQUUsSUFBSSxDQUFDLFNBQVMsQ0FBQyxHQUFHLENBQUMsU0FBUyxDQUFFO3lCQUN0QyxDQUFDLENBQUM7b0JBQ0wsQ0FBQzt5QkFBTSxDQUFDO3dCQUNOLDJCQUEyQjt3QkFDM0IsSUFBSSxDQUFDLGlCQUFpQixDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUUsQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUM7b0JBQ3JELENBQUM7Z0JBQ0gsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO1FBRUQsNkJBQTZCO1FBQzdCLElBQUksS0FBSyxDQUFDLElBQUksS0FBSyx3REFBNkIsRUFBRSxDQUFDO1lBQ2pELElBQUksQ0FBQyxlQUFlLEdBQUcsS0FBSyxDQUFDO1lBQzdCLCtCQUErQjtZQUMvQixLQUFLLE1BQU0sQ0FBRSxVQUFVLEVBQUUsVUFBVSxDQUFFLElBQUksSUFBSSxDQUFDLHVCQUF1QixDQUFDLE9BQU8sRUFBRSxFQUFFLENBQUM7Z0JBQ2hGLElBQUksVUFBVSxNQUFLLGFBQWEsYUFBYixhQUFhLHVCQUFiLGFBQWEsQ0FBRSxVQUFVLENBQUEsRUFBRSxDQUFDO29CQUM3QyxJQUFJLENBQUMsVUFBVSxDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUMsWUFBWSxDQUFDLEVBQUUsQ0FBQyxZQUFZLENBQUMsT0FBTyxLQUFLLFlBQVksQ0FBQyxFQUFFLENBQUM7d0JBQzFGLFNBQVM7b0JBQ1gsQ0FBQztvQkFDRCxVQUFVLENBQUMsYUFBYSxDQUFDLElBQUksQ0FBQzt3QkFDNUIsT0FBTyxFQUFFLFlBQVk7d0JBQ3JCLE9BQU8sRUFBRSx3REFBNkI7d0JBQ3RDLFVBQVUsRUFBRSxDQUFDO2dDQUNYLFNBQVMsRUFBRSxnRUFBcUM7Z0NBQ2hELEtBQUssRUFBRSxVQUFVOzZCQUNsQixDQUFDO3dCQUNGLEtBQUs7cUJBQ04sQ0FBQyxDQUFDO2dCQUNMLENBQUM7WUFDSCxDQUFDO1FBQ0gsQ0FBQztRQUVELHVDQUF1QztRQUN2QyxJQUFJLCtCQUFtQixDQUFDLGFBQWEsQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLEVBQUUsQ0FBQztZQUNqRCxxQ0FBcUM7WUFDckMsTUFBTSxVQUFVLEdBQUcsTUFBQSxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsbUNBQUksRUFBRSxDQUFDO1lBQ2hFLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxVQUFVLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7Z0JBQzNDLElBQUksQ0FBQyxVQUFVLENBQUMsQ0FBQyxDQUFDLENBQUMsYUFBYSxDQUFDLElBQUksQ0FBQyxZQUFZLENBQUMsRUFBRSxDQUFDLFlBQVksQ0FBQyxPQUFPLEtBQUssWUFBWSxDQUFDLEVBQUUsQ0FBQztvQkFDN0YsU0FBUztnQkFDWCxDQUFDO2dCQUNELFVBQVUsQ0FBQyxDQUFDLENBQUMsQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDO29CQUMvQixPQUFPLEVBQUUsWUFBWTtvQkFDckIsT0FBTyxFQUFFLEtBQUssQ0FBQyxJQUFJO29CQUNuQixVQUFVLEVBQUUsRUFBRTtvQkFDZCxLQUFLO2lCQUNOLENBQUMsQ0FBQztZQUNMLENBQUM7WUFDRCxJQUFJLENBQUMsa0JBQWtCLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsS0FBSyxDQUFDLENBQUM7UUFDaEQsQ0FBQztRQUVELG9CQUFvQjtRQUNwQixNQUFNLFFBQVEsR0FBRyw0Q0FBaUIsQ0FBQyxvQkFBb0IsQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLENBQUM7UUFDbkUsSUFBSSxRQUFRLEVBQUUsQ0FBQztZQUNiLElBQUksQ0FBQyxTQUFTLENBQUMsR0FBRyxDQUFDLEdBQUcsQ0FBQyxLQUFLLEVBQUUsS0FBSyxDQUFDLENBQUM7WUFDckMsK0JBQStCO1lBQy9CLEtBQUssTUFBTSxDQUFFLFVBQVUsRUFBRSxVQUFVLENBQUUsSUFBSSxJQUFJLENBQUMsaUJBQWlCLENBQUMsT0FBTyxFQUFFLEVBQUUsQ0FBQztnQkFDMUUsSUFBSSxVQUFVLENBQUMsUUFBUSxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsRUFBRSxDQUFDO29CQUNuQyxJQUFJLENBQUMsVUFBVSxDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUMsWUFBWSxDQUFDLEVBQUUsQ0FBQyxZQUFZLENBQUMsT0FBTyxLQUFLLGdCQUFnQixNQUFNLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQUMsRUFBRSxDQUFDO3dCQUNuSCxTQUFTO29CQUNYLENBQUM7b0JBQ0QsVUFBVSxDQUFDLGFBQWEsQ0FBQyxJQUFJLENBQUM7d0JBQzVCLE9BQU8sRUFBRSxnQkFBZ0IsTUFBTSxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsSUFBSTt3QkFDOUMsT0FBTyxFQUFFLEtBQUssQ0FBQyxJQUFJO3dCQUNuQixVQUFVLEVBQUUsRUFBRTt3QkFDZCxLQUFLO3FCQUNOLENBQUMsQ0FBQztnQkFDTCxDQUFDO1lBQ0gsQ0FBQztRQUNILENBQUM7UUFFRCxvQkFBb0I7UUFDcEIsTUFBTSxRQUFRLEdBQUcsNENBQWlCLENBQUMsb0JBQW9CLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxDQUFDO1FBQ25FLElBQUksUUFBUSxFQUFFLENBQUM7WUFDYixJQUFJLENBQUMsU0FBUyxDQUFDLEdBQUcsQ0FBQyxHQUFHLENBQUMsS0FBSyxFQUFFLEtBQUssQ0FBQyxDQUFDO1lBQ3JDLCtCQUErQjtZQUMvQixLQUFLLE1BQU0sQ0FBRSxVQUFVLEVBQUUsVUFBVSxDQUFFLElBQUksSUFBSSxDQUFDLGlCQUFpQixDQUFDLE9BQU8sRUFBRSxFQUFFLENBQUM7Z0JBQzFFLElBQUksVUFBVSxDQUFDLFFBQVEsQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLEVBQUUsQ0FBQztvQkFDbkMsSUFBSSxDQUFDLFVBQVUsQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDLFlBQVksQ0FBQyxFQUFFLENBQUMsWUFBWSxDQUFDLE9BQU8sS0FBSyxnQkFBZ0IsTUFBTSxDQUFDLEdBQUcsQ0FBQyxLQUFLLENBQUMsSUFBSSxDQUFDLEVBQUUsQ0FBQzt3QkFDbkgsU0FBUztvQkFDWCxDQUFDO29CQUNELFVBQVUsQ0FBQyxhQUFhLENBQUMsSUFBSSxDQUFDO3dCQUM1QixPQUFPLEVBQUUsZ0JBQWdCLE1BQU0sQ0FBQyxHQUFHLENBQUMsS0FBSyxDQUFDLElBQUk7d0JBQzlDLE9BQU8sRUFBRSxLQUFLLENBQUMsSUFBSTt3QkFDbkIsVUFBVSxFQUFFLEVBQUU7d0JBQ2QsS0FBSztxQkFDTixDQUFDLENBQUM7Z0JBQ0wsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztDQUNGO0FBblFELGdEQW1RQyJ9

package/lib/tracing/LangGraphTracer.d.ts

@@ -0,0 +1,8 @@
+import { Logger } from '@eggjs/tegg';
+import { BaseTracer, Run } from '@langchain/core/tracers/base';
+export declare class LangGraphTracer extends BaseTracer {
+    logger: Logger;
+    name: string;
+    setName(name: string): void;
+    protected persistRun(run: Run): Promise<void>;
+}

package/lib/tracing/LangGraphTracer.js

@@ -0,0 +1,40 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.LangGraphTracer = void 0;
+const tegg_1 = require("@eggjs/tegg");
+// eslint-disable-next-line @typescript-eslint/ban-ts-comment
+// @ts-ignore
+const base_1 = require("@langchain/core/tracers/base");
+let LangGraphTracer = class LangGraphTracer extends base_1.BaseTracer {
+    constructor() {
+        super(...arguments);
+        this.name = 'LangGraphTracer';
+    }
+    setName(name) {
+        this.name = name;
+    }
+    persistRun(run) {
+        this.logger.info(`[agent_run][${this.name}]:traceId=${run.trace_id},run=${JSON.stringify(run)}`);
+        return Promise.resolve(undefined);
+    }
+};
+exports.LangGraphTracer = LangGraphTracer;
+__decorate([
+    (0, tegg_1.Inject)(),
+    __metadata("design:type", Object)
+], LangGraphTracer.prototype, "logger", void 0);
+exports.LangGraphTracer = LangGraphTracer = __decorate([
+    (0, tegg_1.SingletonProto)({
+        accessLevel: tegg_1.AccessLevel.PUBLIC,
+    })
+], LangGraphTracer);
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiTGFuZ0dyYXBoVHJhY2VyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiTGFuZ0dyYXBoVHJhY2VyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7OztBQUFBLHNDQUEwRTtBQUMxRSw2REFBNkQ7QUFDN0QsYUFBYTtBQUNiLHVEQUErRDtBQUt4RCxJQUFNLGVBQWUsR0FBckIsTUFBTSxlQUFnQixTQUFRLGlCQUFVO0lBQXhDOztRQUlMLFNBQUksR0FBRyxpQkFBaUIsQ0FBQztJQVUzQixDQUFDO0lBUkMsT0FBTyxDQUFDLElBQVk7UUFDbEIsSUFBSSxDQUFDLElBQUksR0FBRyxJQUFJLENBQUM7SUFDbkIsQ0FBQztJQUVTLFVBQVUsQ0FBQyxHQUFRO1FBQzNCLElBQUksQ0FBQyxNQUFNLENBQUMsSUFBSSxDQUFDLGVBQWUsSUFBSSxDQUFDLElBQUksYUFBYSxHQUFHLENBQUMsUUFBUSxRQUFRLElBQUksQ0FBQyxTQUFTLENBQUMsR0FBRyxDQUFDLEVBQUUsQ0FBQyxDQUFDO1FBQ2pHLE9BQU8sT0FBTyxDQUFDLE9BQU8sQ0FBQyxTQUFTLENBQUMsQ0FBQztJQUNwQyxDQUFDO0NBQ0YsQ0FBQTtBQWRZLDBDQUFlO0FBRTFCO0lBREMsSUFBQSxhQUFNLEdBQUU7OytDQUNNOzBCQUZKLGVBQWU7SUFIM0IsSUFBQSxxQkFBYyxFQUFDO1FBQ2QsV0FBVyxFQUFFLGtCQUFXLENBQUMsTUFBTTtLQUNoQyxDQUFDO0dBQ1csZUFBZSxDQWMzQiJ9

package/lib/util.d.ts

@@ -0,0 +1,4 @@
+import { ObjectInfo, ModuleConfig } from '@eggjs/tegg';
+export type ConfigTypeHelper<T extends 'ChatModel'> = Required<ModuleConfig>[T]['clients'][keyof Required<ModuleConfig>[T]['clients']];
+export declare function getClientNames(config: ModuleConfig | undefined, key: string): string[];
+export declare function getChatModelConfig(config: ModuleConfig, objectInfo: ObjectInfo): ConfigTypeHelper<'ChatModel'>;

package/lib/util.js

@@ -0,0 +1,27 @@
+"use strict";
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.getClientNames = getClientNames;
+exports.getChatModelConfig = getChatModelConfig;
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+const node_assert_1 = __importDefault(require("node:assert"));
+function getClientNames(config, key) {
+    var _a;
+    const clients = (_a = config === null || config === void 0 ? void 0 : config[key]) === null || _a === void 0 ? void 0 : _a.clients;
+    if (!clients)
+        return [];
+    return Object.keys(clients);
+}
+function getChatModelConfig(config, objectInfo) {
+    var _a, _b;
+    const chatModelName = (_a = objectInfo.qualifiers.find(t => t.attribute === tegg_langchain_decorator_1.ChatModelQualifierAttribute)) === null || _a === void 0 ? void 0 : _a.value;
+    (0, node_assert_1.default)(chatModelName, 'not found ChatModel name');
+    const chatModelConfig = (_b = config.ChatModel) === null || _b === void 0 ? void 0 : _b.clients[chatModelName];
+    if (!config) {
+        throw new Error(`not found ChatModel config for ${chatModelName}`);
+    }
+    return chatModelConfig;
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoidXRpbC5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbInV0aWwudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7QUFTQSx3Q0FJQztBQUVELGdEQVFDO0FBdEJELDhFQUV5QztBQUN6Qyw4REFBaUM7QUFLakMsU0FBZ0IsY0FBYyxDQUFDLE1BQWdDLEVBQUUsR0FBVzs7SUFDMUUsTUFBTSxPQUFPLEdBQUcsTUFBQSxNQUFNLGFBQU4sTUFBTSx1QkFBTixNQUFNLENBQUcsR0FBRyxDQUFDLDBDQUFFLE9BQU8sQ0FBQztJQUN2QyxJQUFJLENBQUMsT0FBTztRQUFFLE9BQU8sRUFBRSxDQUFDO0lBQ3hCLE9BQU8sTUFBTSxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsQ0FBQztBQUM5QixDQUFDO0FBRUQsU0FBZ0Isa0JBQWtCLENBQUMsTUFBb0IsRUFBRSxVQUFzQjs7SUFDN0UsTUFBTSxhQUFhLEdBQUcsTUFBQSxVQUFVLENBQUMsVUFBVSxDQUFDLElBQUksQ0FBQyxDQUFDLENBQUMsRUFBRSxDQUFDLENBQUMsQ0FBQyxTQUFTLEtBQUssc0RBQTJCLENBQUMsMENBQUUsS0FBSyxDQUFDO0lBQzFHLElBQUEscUJBQU0sRUFBQyxhQUFhLEVBQUUsMEJBQTBCLENBQUMsQ0FBQztJQUNsRCxNQUFNLGVBQWUsR0FBRyxNQUFBLE1BQU0sQ0FBQyxTQUFTLDBDQUFFLE9BQU8sQ0FBQyxhQUFhLENBQUMsQ0FBQztJQUNqRSxJQUFJLENBQUMsTUFBTSxFQUFFLENBQUM7UUFDWixNQUFNLElBQUksS0FBSyxDQUFDLGtDQUFrQyxhQUFhLEVBQUUsQ0FBQyxDQUFDO0lBQ3JFLENBQUM7SUFDRCxPQUFPLGVBQWdCLENBQUM7QUFDMUIsQ0FBQyJ9