@eggjs/tegg-langchain 0.0.0

package/README.md

@@ -0,0 +1,20 @@
+# @eggjs/tegg-langchain
+
+使用注解的方式来开发 egg 中的 orm
+
+## Install
+
+```shell
+npm i --save @eggjs/tegg-orm-plugin
+```
+
+
+## Config
+
+```js
+// config/plugin.js
+exports.teggOrm = {
+  package: '@eggjs/tegg-orm-plugin',
+  enable: true,
+};
+```

package/app.d.ts

@@ -0,0 +1,8 @@
+import type { Application, IBoot } from 'egg';
+export default class ModuleLangChainHook implements IBoot {
+    #private;
+    constructor(app: Application);
+    configWillLoad(): void;
+    configDidLoad(): void;
+    beforeClose(): Promise<void>;
+}

package/app.js

@@ -0,0 +1,55 @@
+"use strict";
+var __classPrivateFieldSet = (this && this.__classPrivateFieldSet) || function (receiver, state, value, kind, f) {
+    if (kind === "m") throw new TypeError("Private method is not writable");
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a setter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot write private member to an object whose class did not declare it");
+    return (kind === "a" ? f.call(receiver, value) : f ? f.value = value : state.set(receiver, value)), value;
+};
+var __classPrivateFieldGet = (this && this.__classPrivateFieldGet) || function (receiver, state, kind, f) {
+    if (kind === "a" && !f) throw new TypeError("Private accessor was defined without a getter");
+    if (typeof state === "function" ? receiver !== state || !f : !state.has(receiver)) throw new TypeError("Cannot read private member from an object whose class did not declare it");
+    return kind === "m" ? f : kind === "a" ? f.call(receiver) : f ? f.value : state.get(receiver);
+};
+var _ModuleLangChainHook_app, _ModuleLangChainHook_graphObjectHook, _ModuleLangChainHook_graphLoadUnitHook, _ModuleLangChainHook_boundModelObjectHook, _ModuleLangChainHook_graphPrototypeHook;
+Object.defineProperty(exports, "__esModule", { value: true });
+const tegg_metadata_1 = require("@eggjs/tegg-metadata");
+const GraphObjectHook_1 = require("./lib/graph/GraphObjectHook");
+const GraphLoadUnitHook_1 = require("./lib/graph/GraphLoadUnitHook");
+const CompiledStateGraphProto_1 = require("./lib/graph/CompiledStateGraphProto");
+const CompiledStateGraphObject_1 = require("./lib/graph/CompiledStateGraphObject");
+const BoundModelObjectHook_1 = require("./lib/boundModel/BoundModelObjectHook");
+const GraphPrototypeHook_1 = require("./lib/graph/GraphPrototypeHook");
+const GraphBuildHook_1 = require("./lib/graph/GraphBuildHook");
+class ModuleLangChainHook {
+    constructor(app) {
+        _ModuleLangChainHook_app.set(this, void 0);
+        _ModuleLangChainHook_graphObjectHook.set(this, void 0);
+        _ModuleLangChainHook_graphLoadUnitHook.set(this, void 0);
+        _ModuleLangChainHook_boundModelObjectHook.set(this, void 0);
+        _ModuleLangChainHook_graphPrototypeHook.set(this, void 0);
+        __classPrivateFieldSet(this, _ModuleLangChainHook_app, app, "f");
+        __classPrivateFieldSet(this, _ModuleLangChainHook_graphObjectHook, new GraphObjectHook_1.GraphObjectHook(), "f");
+        __classPrivateFieldSet(this, _ModuleLangChainHook_graphLoadUnitHook, new GraphLoadUnitHook_1.GraphLoadUnitHook(__classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggPrototypeFactory), "f");
+        __classPrivateFieldSet(this, _ModuleLangChainHook_boundModelObjectHook, new BoundModelObjectHook_1.BoundModelObjectHook(), "f");
+        __classPrivateFieldSet(this, _ModuleLangChainHook_graphPrototypeHook, new GraphPrototypeHook_1.GraphPrototypeHook(), "f");
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").loadUnitLifecycleUtil.registerLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_graphLoadUnitHook, "f"));
+    }
+    configWillLoad() {
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggObjectLifecycleUtil.registerLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_graphObjectHook, "f"));
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggObjectLifecycleUtil.registerLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_boundModelObjectHook, "f"));
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggObjectFactory.registerEggObjectCreateMethod(CompiledStateGraphProto_1.CompiledStateGraphProto, CompiledStateGraphObject_1.CompiledStateGraphObject.createObject);
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggPrototypeLifecycleUtil.registerLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_graphPrototypeHook, "f"));
+    }
+    configDidLoad() {
+        tegg_metadata_1.GlobalGraph.instance.registerBuildHook(GraphBuildHook_1.GraphBuildHook);
+    }
+    async beforeClose() {
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggObjectLifecycleUtil.deleteLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_graphObjectHook, "f"));
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggObjectLifecycleUtil.deleteLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_boundModelObjectHook, "f"));
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").loadUnitLifecycleUtil.deleteLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_graphLoadUnitHook, "f"));
+        __classPrivateFieldGet(this, _ModuleLangChainHook_app, "f").eggPrototypeLifecycleUtil.deleteLifecycle(__classPrivateFieldGet(this, _ModuleLangChainHook_graphPrototypeHook, "f"));
+    }
+}
+_ModuleLangChainHook_app = new WeakMap(), _ModuleLangChainHook_graphObjectHook = new WeakMap(), _ModuleLangChainHook_graphLoadUnitHook = new WeakMap(), _ModuleLangChainHook_boundModelObjectHook = new WeakMap(), _ModuleLangChainHook_graphPrototypeHook = new WeakMap();
+exports.default = ModuleLangChainHook;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiYXBwLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiYXBwLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7Ozs7Ozs7Ozs7Ozs7O0FBQ0Esd0RBQW1EO0FBQ25ELGlFQUE4RDtBQUM5RCxxRUFBa0U7QUFDbEUsaUZBQThFO0FBQzlFLG1GQUFnRjtBQUNoRixnRkFBNkU7QUFDN0UsdUVBQW9FO0FBQ3BFLCtEQUE0RDtBQUU1RCxNQUFxQixtQkFBbUI7SUFPdEMsWUFBWSxHQUFnQjtRQU5uQiwyQ0FBa0I7UUFDbEIsdURBQWtDO1FBQ2xDLHlEQUFzQztRQUN0Qyw0REFBNEM7UUFDNUMsMERBQXdDO1FBRy9DLHVCQUFBLElBQUksNEJBQVEsR0FBRyxNQUFBLENBQUM7UUFDaEIsdUJBQUEsSUFBSSx3Q0FBb0IsSUFBSSxpQ0FBZSxFQUFFLE1BQUEsQ0FBQztRQUM5Qyx1QkFBQSxJQUFJLDBDQUFzQixJQUFJLHFDQUFpQixDQUFDLHVCQUFBLElBQUksZ0NBQUssQ0FBQyxtQkFBbUIsQ0FBQyxNQUFBLENBQUM7UUFDL0UsdUJBQUEsSUFBSSw2Q0FBeUIsSUFBSSwyQ0FBb0IsRUFBRSxNQUFBLENBQUM7UUFDeEQsdUJBQUEsSUFBSSwyQ0FBdUIsSUFBSSx1Q0FBa0IsRUFBRSxNQUFBLENBQUM7UUFDcEQsdUJBQUEsSUFBSSxnQ0FBSyxDQUFDLHFCQUFxQixDQUFDLGlCQUFpQixDQUFDLHVCQUFBLElBQUksOENBQW1CLENBQUMsQ0FBQztJQUM3RSxDQUFDO0lBRUQsY0FBYztRQUNaLHVCQUFBLElBQUksZ0NBQUssQ0FBQyxzQkFBc0IsQ0FBQyxpQkFBaUIsQ0FBQyx1QkFBQSxJQUFJLDRDQUFpQixDQUFDLENBQUM7UUFDMUUsdUJBQUEsSUFBSSxnQ0FBSyxDQUFDLHNCQUFzQixDQUFDLGlCQUFpQixDQUFDLHVCQUFBLElBQUksaURBQXNCLENBQUMsQ0FBQztRQUMvRSx1QkFBQSxJQUFJLGdDQUFLLENBQUMsZ0JBQWdCLENBQUMsNkJBQTZCLENBQUMsaURBQXVCLEVBQUUsbURBQXdCLENBQUMsWUFBWSxDQUFDLENBQUM7UUFDekgsdUJBQUEsSUFBSSxnQ0FBSyxDQUFDLHlCQUF5QixDQUFDLGlCQUFpQixDQUFDLHVCQUFBLElBQUksK0NBQW9CLENBQUMsQ0FBQztJQUNsRixDQUFDO0lBRUQsYUFBYTtRQUNYLDJCQUFXLENBQUMsUUFBUyxDQUFDLGlCQUFpQixDQUFDLCtCQUFjLENBQUMsQ0FBQztJQUMxRCxDQUFDO0lBRUQsS0FBSyxDQUFDLFdBQVc7UUFDZix1QkFBQSxJQUFJLGdDQUFLLENBQUMsc0JBQXNCLENBQUMsZUFBZSxDQUFDLHVCQUFBLElBQUksNENBQWlCLENBQUMsQ0FBQztRQUN4RSx1QkFBQSxJQUFJLGdDQUFLLENBQUMsc0JBQXNCLENBQUMsZUFBZSxDQUFDLHVCQUFBLElBQUksaURBQXNCLENBQUMsQ0FBQztRQUM3RSx1QkFBQSxJQUFJLGdDQUFLLENBQUMscUJBQXFCLENBQUMsZUFBZSxDQUFDLHVCQUFBLElBQUksOENBQW1CLENBQUMsQ0FBQztRQUN6RSx1QkFBQSxJQUFJLGdDQUFLLENBQUMseUJBQXlCLENBQUMsZUFBZSxDQUFDLHVCQUFBLElBQUksK0NBQW9CLENBQUMsQ0FBQztJQUNoRixDQUFDO0NBQ0Y7O2tCQWpDb0IsbUJBQW1CIn0=

package/index.d.ts

@@ -0,0 +1,10 @@
+export * from './lib/ChatOpenAI';
+export * from './lib/boundModel/BoundModelObjectHook';
+export * from './lib/config/QualifierUtil';
+export * from './lib/graph/CompiledStateGraphObject';
+export * from './lib/graph/CompiledStateGraphProto';
+export * from './lib/graph/GraphBuildHook';
+export * from './lib/graph/GraphLoadUnitHook';
+export * from './lib/graph/GraphObjectHook';
+export * from './lib/graph/GraphPrototypeHook';
+export * from './lib/tracing/LangGraphTracer';

package/index.js

@@ -0,0 +1,27 @@
+"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 __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./lib/ChatOpenAI"), exports);
+__exportStar(require("./lib/boundModel/BoundModelObjectHook"), exports);
+__exportStar(require("./lib/config/QualifierUtil"), exports);
+__exportStar(require("./lib/graph/CompiledStateGraphObject"), exports);
+__exportStar(require("./lib/graph/CompiledStateGraphProto"), exports);
+__exportStar(require("./lib/graph/GraphBuildHook"), exports);
+__exportStar(require("./lib/graph/GraphLoadUnitHook"), exports);
+__exportStar(require("./lib/graph/GraphObjectHook"), exports);
+__exportStar(require("./lib/graph/GraphPrototypeHook"), exports);
+__exportStar(require("./lib/tracing/LangGraphTracer"), exports);
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiaW5kZXguanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJpbmRleC50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsbURBQWlDO0FBQ2pDLHdFQUFzRDtBQUN0RCw2REFBMkM7QUFDM0MsdUVBQXFEO0FBQ3JELHNFQUFvRDtBQUNwRCw2REFBMkM7QUFDM0MsZ0VBQThDO0FBQzlDLDhEQUE0QztBQUM1QyxpRUFBK0M7QUFDL0MsZ0VBQThDIn0=

package/lib/ChatModelHelper.d.ts

@@ -0,0 +1,8 @@
+export declare class ChatModelHelper {
+    static getChatModelQualifier(clientName: string): {
+        chatModel: {
+            attribute: symbol;
+            value: string;
+        }[];
+    };
+}

package/lib/ChatModelHelper.js

@@ -0,0 +1,16 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatModelHelper = void 0;
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+class ChatModelHelper {
+    static getChatModelQualifier(clientName) {
+        return {
+            [tegg_langchain_decorator_1.ChatModelInjectName]: [{
+                    attribute: tegg_langchain_decorator_1.ChatModelQualifierAttribute,
+                    value: clientName,
+                }],
+        };
+    }
+}
+exports.ChatModelHelper = ChatModelHelper;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ2hhdE1vZGVsSGVscGVyLmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiQ2hhdE1vZGVsSGVscGVyLnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7OztBQUFBLDhFQUd5QztBQUV6QyxNQUFhLGVBQWU7SUFDMUIsTUFBTSxDQUFDLHFCQUFxQixDQUFDLFVBQWtCO1FBQzdDLE9BQU87WUFDTCxDQUFDLDhDQUFtQixDQUFDLEVBQUUsQ0FBQztvQkFDdEIsU0FBUyxFQUFFLHNEQUEyQjtvQkFDdEMsS0FBSyxFQUFFLFVBQVU7aUJBQ2xCLENBQUM7U0FDSCxDQUFDO0lBQ0osQ0FBQztDQUNGO0FBVEQsMENBU0MifQ==

package/lib/ChatOpenAI.d.ts

@@ -0,0 +1,7 @@
+import { ObjectInfo } from '@eggjs/tegg';
+import { ModuleConfig } from '@eggjs/tegg-common-util';
+import { ChatOpenAI } from '@langchain/openai';
+export declare class ChatOpenAIModel extends ChatOpenAI {
+    readonly moduleConfig: ModuleConfig;
+    constructor(moduleConfig: ModuleConfig, objInfo: ObjectInfo);
+}

package/lib/ChatOpenAI.js

@@ -0,0 +1,63 @@
+"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);
+};
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.ChatOpenAIModel = void 0;
+const tegg_1 = require("@eggjs/tegg");
+const tegg_common_util_1 = require("@eggjs/tegg-common-util");
+const openai_1 = require("@langchain/openai");
+const util_1 = require("./util");
+const QualifierUtil_1 = require("./config/QualifierUtil");
+const urllib_1 = require("urllib");
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+const ChatModelHelper_1 = require("./ChatModelHelper");
+let ChatOpenAIModel = class ChatOpenAIModel extends openai_1.ChatOpenAI {
+    constructor(moduleConfig, objInfo) {
+        const chatConfig = (0, util_1.getChatModelConfig)(moduleConfig, objInfo);
+        chatConfig.configuration = chatConfig.configuration || {};
+        // 如果依赖中存在低版本 urllib 会引发问题，因此需要手动设置好
+        urllib_1.FetchFactory.setDispatcher(new urllib_1.Agent({ allowH2: true }));
+        chatConfig.configuration.fetch = urllib_1.fetch;
+        super(chatConfig);
+        this.moduleConfig = moduleConfig;
+    }
+};
+exports.ChatOpenAIModel = ChatOpenAIModel;
+exports.ChatOpenAIModel = ChatOpenAIModel = __decorate([
+    (0, tegg_1.MultiInstanceProto)({
+        accessLevel: tegg_1.AccessLevel.PUBLIC,
+        initType: tegg_1.ObjectInitType.SINGLETON,
+        // 从 module.yml 中动态获取配置来决定需要初始化几个对象
+        getObjects(ctx) {
+            const config = tegg_common_util_1.ModuleConfigUtil.loadModuleConfigSync(ctx.unitPath);
+            const moduleName = tegg_common_util_1.ModuleConfigUtil.readModuleNameSync(ctx.unitPath);
+            return (0, util_1.getClientNames)(config, 'ChatModel')
+                .filter(name => {
+                return config.ChatModel.clients[name].type === 'openai';
+            })
+                .map(name => {
+                return {
+                    name: tegg_langchain_decorator_1.ChatModelInjectName,
+                    qualifiers: ChatModelHelper_1.ChatModelHelper.getChatModelQualifier(name)[tegg_langchain_decorator_1.ChatModelInjectName],
+                    properQualifiers: {
+                        ...QualifierUtil_1.QualifierUtil.getModuleConfigQualifier(moduleName),
+                    },
+                };
+            });
+        },
+    }),
+    __param(0, (0, tegg_1.Inject)()),
+    __param(1, (0, tegg_1.MultiInstanceInfo)([tegg_langchain_decorator_1.ChatModelQualifierAttribute])),
+    __metadata("design:paramtypes", [Object, Object])
+], ChatOpenAIModel);
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ2hhdE9wZW5BSS5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIkNoYXRPcGVuQUkudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7Ozs7Ozs7Ozs7Ozs7O0FBQUEsc0NBUXFCO0FBQ3JCLDhEQUF5RTtBQUN6RSw4Q0FBK0M7QUFDL0MsaUNBQTREO0FBQzVELDBEQUF1RDtBQUN2RCxtQ0FBb0Q7QUFDcEQsOEVBQW1HO0FBQ25HLHVEQUFvRDtBQXdCN0MsSUFBTSxlQUFlLEdBQXJCLE1BQU0sZUFBZ0IsU0FBUSxtQkFBVTtJQUM3QyxZQUNxQixZQUEwQixFQUNPLE9BQW1CO1FBRXZFLE1BQU0sVUFBVSxHQUFHLElBQUEseUJBQWtCLEVBQUMsWUFBWSxFQUFFLE9BQU8sQ0FBQyxDQUFDO1FBQzdELFVBQVUsQ0FBQyxhQUFhLEdBQUcsVUFBVSxDQUFDLGFBQWEsSUFBSSxFQUFFLENBQUM7UUFDMUQsb0NBQW9DO1FBQ3BDLHFCQUFZLENBQUMsYUFBYSxDQUFDLElBQUksY0FBSyxDQUFDLEVBQUUsT0FBTyxFQUFFLElBQUksRUFBRSxDQUFDLENBQUMsQ0FBQztRQUN4RCxVQUFVLENBQUMsYUFBcUIsQ0FBQyxLQUFLLEdBQUcsY0FBWSxDQUFDO1FBQ3ZELEtBQUssQ0FBQyxVQUFpQixDQUFDLENBQUM7UUFSTixpQkFBWSxHQUFaLFlBQVksQ0FBYztJQVMvQyxDQUFDO0NBQ0YsQ0FBQTtBQVpZLDBDQUFlOzBCQUFmLGVBQWU7SUF0QjNCLElBQUEseUJBQWtCLEVBQUM7UUFDbEIsV0FBVyxFQUFFLGtCQUFXLENBQUMsTUFBTTtRQUMvQixRQUFRLEVBQUUscUJBQWMsQ0FBQyxTQUFTO1FBQ2xDLG1DQUFtQztRQUNuQyxVQUFVLENBQUMsR0FBNEM7WUFDckQsTUFBTSxNQUFNLEdBQWlCLG1DQUFnQixDQUFDLG9CQUFvQixDQUFDLEdBQUcsQ0FBQyxRQUFRLENBQUMsQ0FBQztZQUNqRixNQUFNLFVBQVUsR0FBRyxtQ0FBZ0IsQ0FBQyxrQkFBa0IsQ0FBQyxHQUFHLENBQUMsUUFBUSxDQUFDLENBQUM7WUFDckUsT0FBTyxJQUFBLHFCQUFjLEVBQUMsTUFBTSxFQUFFLFdBQVcsQ0FBQztpQkFDdkMsTUFBTSxDQUFDLElBQUksQ0FBQyxFQUFFO2dCQUNiLE9BQVEsTUFBYyxDQUFDLFNBQVMsQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLENBQUMsSUFBSSxLQUFLLFFBQVEsQ0FBQztZQUNuRSxDQUFDLENBQUM7aUJBQ0QsR0FBRyxDQUFDLElBQUksQ0FBQyxFQUFFO2dCQUNWLE9BQU87b0JBQ0wsSUFBSSxFQUFFLDhDQUFtQjtvQkFDekIsVUFBVSxFQUFFLGlDQUFlLENBQUMscUJBQXFCLENBQUMsSUFBSSxDQUFDLENBQUMsOENBQW1CLENBQUM7b0JBQzVFLGdCQUFnQixFQUFFO3dCQUNoQixHQUFHLDZCQUFhLENBQUMsd0JBQXdCLENBQUMsVUFBVSxDQUFDO3FCQUN0RDtpQkFDRixDQUFDO1lBQ0osQ0FBQyxDQUFDLENBQUM7UUFDUCxDQUFDO0tBQ0YsQ0FBQztJQUdHLFdBQUEsSUFBQSxhQUFNLEdBQUUsQ0FBQTtJQUNSLFdBQUEsSUFBQSx3QkFBaUIsRUFBQyxDQUFFLHNEQUEyQixDQUFFLENBQUMsQ0FBQTs7R0FIMUMsZUFBZSxDQVkzQiJ9

package/lib/boundModel/BoundModelObjectHook.d.ts

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

package/lib/boundModel/BoundModelObjectHook.js

@@ -0,0 +1,112 @@
+"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.BoundModelObjectHook = void 0;
+const z = __importStar(require("zod/v4"));
+const tegg_1 = require("@eggjs/tegg");
+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 tegg_runtime_1 = require("@eggjs/tegg-runtime");
+const langchain_1 = require("langchain");
+class BoundModelHandler {
+    constructor(nodeMetadata, eggObject) {
+        this.boundModelMetadata = nodeMetadata;
+        this.eggObject = eggObject;
+    }
+    async findGraphTools() {
+        var _a;
+        const tools = (_a = this.boundModelMetadata.tools) !== null && _a !== void 0 ? _a : [];
+        let dTools = [];
+        for (let i = 0; i < tools.length; i++) {
+            const toolsObj = await tegg_runtime_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.boundModelMetadata.mcpServers) !== null && _a !== void 0 ? _a : [];
+        let sTools = [];
+        for (let i = 0; i < mcpServers.length; i++) {
+            const mcpClientObj = await tegg_runtime_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();
+        const chatModelObj = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromName(tegg_langchain_decorator_1.ChatModelInjectName, [{
+                attribute: tegg_langchain_decorator_1.ChatModelQualifierAttribute,
+                value: this.boundModelMetadata.modelName,
+            }]);
+        const chatModel = chatModelObj.obj;
+        const boundChatModel = chatModel.bindTools([...dTools, ...sTools]);
+        Object.setPrototypeOf(Object.getPrototypeOf(nodeObj), boundChatModel);
+    }
+}
+class BoundModelObjectHook {
+    async postCreate(_, eggObject) {
+        const BoundModelMetadata = eggObject.proto.getMetaData(tegg_langchain_decorator_1.BOUND_MODEL_METADATA);
+        // 找到 graph node
+        if (BoundModelMetadata) {
+            const handler = new BoundModelHandler(BoundModelMetadata, eggObject);
+            await handler.boundTools();
+            return;
+        }
+    }
+}
+exports.BoundModelObjectHook = BoundModelObjectHook;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQm91bmRNb2RlbE9iamVjdEhvb2suanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJCb3VuZE1vZGVsT2JqZWN0SG9vay50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7Ozs7QUFBQSwwQ0FBNEI7QUFDNUIsc0NBQThEO0FBQzlELDhFQVF5QztBQUl6QyxrREFBcUY7QUFDckYsMERBQXVEO0FBQ3ZELHNEQUFnRztBQUVoRyx5Q0FBa0Q7QUFFbEQsTUFBTSxpQkFBaUI7SUFJckIsWUFBWSxZQUFpQyxFQUFFLFNBQW9CO1FBQ2pFLElBQUksQ0FBQyxrQkFBa0IsR0FBRyxZQUFZLENBQUM7UUFDdkMsSUFBSSxDQUFDLFNBQVMsR0FBRyxTQUFTLENBQUM7SUFDN0IsQ0FBQztJQUVELEtBQUssQ0FBQyxjQUFjOztRQUNsQixNQUFNLEtBQUssR0FBRyxNQUFBLElBQUksQ0FBQyxrQkFBa0IsQ0FBQyxLQUFLLG1DQUFJLEVBQUUsQ0FBQztRQUNsRCxJQUFJLE1BQU0sR0FBb0QsRUFBRSxDQUFDO1FBQ2pFLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxLQUFLLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7WUFDdEMsTUFBTSxRQUFRLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw2QkFBNkIsQ0FBQyxLQUFLLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUNuRixNQUFNLFlBQVksR0FBRyw0Q0FBaUIsQ0FBQyxvQkFBb0IsQ0FBRSxRQUFRLENBQUMsS0FBMEMsQ0FBQyxLQUFNLENBQUMsQ0FBQztZQUN6SCxNQUFNLFVBQVUsR0FBRyxrQkFBVyxDQUFDLG1CQUFtQixDQUFFLFFBQVEsQ0FBQyxLQUEwQyxDQUFDLEtBQU0sRUFBRSxTQUFTLENBQUMsQ0FBQztZQUMzSCxJQUFJLFlBQVksSUFBSSxVQUFVLEVBQUUsQ0FBQztnQkFDL0IsTUFBTSxJQUFJLEdBQUcsSUFBSSxpQ0FBcUIsQ0FBQztvQkFDckMsV0FBVyxFQUFFLFlBQVksQ0FBQyxXQUFXO29CQUNyQyxJQUFJLEVBQUUsWUFBWSxDQUFDLFFBQVE7b0JBQzNCLElBQUksRUFBRyxRQUFRLENBQUMsR0FBa0MsQ0FBQyxPQUFPLENBQUMsSUFBSSxDQUFDLFFBQVEsQ0FBQyxHQUFHLENBQUM7b0JBQzdFLE1BQU0sRUFBRSxDQUFDLENBQUMsTUFBTSxDQUFDLFVBQVUsQ0FBQyxVQUFVLENBQVE7aUJBQy9DLENBQUMsQ0FBQztnQkFDSCxNQUFNLEdBQUcsTUFBTSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsQ0FBQztZQUMvQixDQUFDO1FBQ0gsQ0FBQztRQUVELE9BQU8sTUFBTSxDQUFDO0lBQ2hCLENBQUM7SUFFRCxLQUFLLENBQUMsa0JBQWtCOztRQUN0QixNQUFNLFVBQVUsR0FBRyxNQUFBLElBQUksQ0FBQyxrQkFBa0IsQ0FBQyxVQUFVLG1DQUFJLEVBQUUsQ0FBQztRQUM1RCxJQUFJLE1BQU0sR0FBb0QsRUFBRSxDQUFDO1FBQ2pFLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxVQUFVLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7WUFDM0MsTUFBTSxZQUFZLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw0QkFBNEIsQ0FBQyxnQ0FBbUIsRUFBRSxDQUFDO29CQUNoRyxTQUFTLEVBQUUsd0NBQTJCO29CQUN0QyxLQUFLLEVBQUUsVUFBVSxDQUFDLENBQUMsQ0FBQztpQkFDckIsQ0FBQyxDQUFDLENBQUM7WUFDSixNQUFNLElBQUksR0FBRyxNQUFNLElBQUEsMkJBQVksRUFBQyxVQUFVLENBQUMsQ0FBQyxDQUFDLEVBQUUsWUFBWSxDQUFDLEdBQVUsRUFBRTtnQkFDdEUsZ0JBQWdCLEVBQUUsSUFBSTtnQkFDdEIsNEJBQTRCLEVBQUUsS0FBSztnQkFDbkMsd0JBQXdCLEVBQUUsRUFBRTthQUM3QixDQUFDLENBQUM7WUFDSCxNQUFNLEdBQUcsTUFBTSxDQUFDLE1BQU0sQ0FBQyxJQUFJLENBQUMsQ0FBQztRQUMvQixDQUFDO1FBQ0QsT0FBTyxNQUFNLENBQUM7SUFDaEIsQ0FBQztJQUVELEtBQUssQ0FBQyxVQUFVO1FBQ2QsTUFBTSxPQUFPLEdBQUcsSUFBSSxDQUFDLFNBQVMsQ0FBQyxHQUFzQixDQUFDO1FBQ3RELE1BQU0sTUFBTSxHQUFHLE1BQU0sSUFBSSxDQUFDLGNBQWMsRUFBRSxDQUFDO1FBQzNDLE1BQU0sTUFBTSxHQUFHLE1BQU0sSUFBSSxDQUFDLGtCQUFrQixFQUFFLENBQUM7UUFFL0MsTUFBTSxZQUFZLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw0QkFBNEIsQ0FBQyw4Q0FBbUIsRUFBRSxDQUFDO2dCQUNoRyxTQUFTLEVBQUUsc0RBQTJCO2dCQUN0QyxLQUFLLEVBQUUsSUFBSSxDQUFDLGtCQUFrQixDQUFDLFNBQVM7YUFDekMsQ0FBQyxDQUFDLENBQUM7UUFDSixNQUFNLFNBQVMsR0FBRyxZQUFZLENBQUMsR0FBd0IsQ0FBQztRQUV4RCxNQUFNLGNBQWMsR0FBRyxTQUFTLENBQUMsU0FBUyxDQUFDLENBQUUsR0FBRyxNQUFNLEVBQUUsR0FBRyxNQUFNLENBQUUsQ0FBQyxDQUFDO1FBRXJFLE1BQU0sQ0FBQyxjQUFjLENBQUMsTUFBTSxDQUFDLGNBQWMsQ0FBQyxPQUFPLENBQUMsRUFBRSxjQUFjLENBQUMsQ0FBQztJQUN4RSxDQUFDO0NBQ0Y7QUFFRCxNQUFhLG9CQUFvQjtJQUUvQixLQUFLLENBQUMsVUFBVSxDQUFDLENBQTRCLEVBQUUsU0FBb0I7UUFDakUsTUFBTSxrQkFBa0IsR0FBRyxTQUFTLENBQUMsS0FBSyxDQUFDLFdBQVcsQ0FBc0IsK0NBQW9CLENBQUMsQ0FBQztRQUNsRyxnQkFBZ0I7UUFDaEIsSUFBSSxrQkFBa0IsRUFBRSxDQUFDO1lBQ3ZCLE1BQU0sT0FBTyxHQUFHLElBQUksaUJBQWlCLENBQUMsa0JBQWtCLEVBQUUsU0FBUyxDQUFDLENBQUM7WUFDckUsTUFBTSxPQUFPLENBQUMsVUFBVSxFQUFFLENBQUM7WUFDM0IsT0FBTztRQUNULENBQUM7SUFDSCxDQUFDO0NBQ0Y7QUFYRCxvREFXQyJ9

package/lib/config/QualifierUtil.d.ts

@@ -0,0 +1,8 @@
+export declare class QualifierUtil {
+    static getModuleConfigQualifier(moduleName: string): {
+        moduleConfig: {
+            attribute: symbol;
+            value: string;
+        }[];
+    };
+}

package/lib/config/QualifierUtil.js

@@ -0,0 +1,17 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.QualifierUtil = void 0;
+const tegg_1 = require("@eggjs/tegg");
+// TODO refactor to ModuleConfig and mist impl
+class QualifierUtil {
+    static getModuleConfigQualifier(moduleName) {
+        return {
+            moduleConfig: [{
+                    attribute: tegg_1.ConfigSourceQualifierAttribute,
+                    value: moduleName,
+                }],
+        };
+    }
+}
+exports.QualifierUtil = QualifierUtil;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiUXVhbGlmaWVyVXRpbC5qcyIsInNvdXJjZVJvb3QiOiIiLCJzb3VyY2VzIjpbIlF1YWxpZmllclV0aWwudHMiXSwibmFtZXMiOltdLCJtYXBwaW5ncyI6Ijs7O0FBQUEsc0NBQTZEO0FBRTdELDhDQUE4QztBQUM5QyxNQUFhLGFBQWE7SUFDeEIsTUFBTSxDQUFDLHdCQUF3QixDQUFDLFVBQWtCO1FBQ2hELE9BQU87WUFDTCxZQUFZLEVBQUUsQ0FBQztvQkFDYixTQUFTLEVBQUUscUNBQThCO29CQUN6QyxLQUFLLEVBQUUsVUFBVTtpQkFDbEIsQ0FBQztTQUNILENBQUM7SUFDSixDQUFDO0NBQ0Y7QUFURCxzQ0FTQyJ9

package/lib/graph/CompiledStateGraphObject.d.ts

@@ -0,0 +1,25 @@
+import { EggContext, EggObject } from '@eggjs/tegg-runtime';
+import { EggObjectName, EggPrototypeName, Id } from '@eggjs/tegg';
+import { CompiledStateGraphProto } from './CompiledStateGraphProto';
+import { EggPrototype } from '@eggjs/tegg-metadata';
+import { GraphMetadata } from '@eggjs/tegg-langchain-decorator';
+export declare class CompiledStateGraphObject implements EggObject {
+    private status;
+    id: Id;
+    readonly name: EggPrototypeName;
+    readonly proto: CompiledStateGraphProto;
+    readonly ctx: EggContext;
+    readonly daoName: string;
+    private _obj;
+    readonly graphMetadata: GraphMetadata;
+    readonly graphName: string;
+    constructor(name: EggObjectName, proto: CompiledStateGraphProto);
+    init(): Promise<void>;
+    build(): Promise<any>;
+    boundNodes(stateGraph: EggObject): Promise<void>;
+    boundEdges(stateGraph: EggObject): Promise<void>;
+    injectProperty(): void;
+    get isReady(): boolean;
+    get obj(): object;
+    static createObject(name: EggObjectName, proto: EggPrototype): Promise<CompiledStateGraphObject>;
+}

package/lib/graph/CompiledStateGraphObject.js

@@ -0,0 +1,113 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompiledStateGraphObject = void 0;
+const tegg_runtime_1 = require("@eggjs/tegg-runtime");
+const tegg_1 = require("@eggjs/tegg");
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+class CompiledStateGraphObject {
+    constructor(name, proto) {
+        var _a;
+        this.status = tegg_runtime_1.EggObjectStatus.PENDING;
+        this.name = name;
+        this.proto = proto;
+        this.ctx = tegg_runtime_1.ContextHandler.getContext();
+        this.id = tegg_1.IdenticalUtil.createObjectId(this.proto.id, (_a = this.ctx) === null || _a === void 0 ? void 0 : _a.id);
+        this.graphMetadata = proto.graphMetadata;
+        this.graphName = proto.graphName;
+    }
+    async init() {
+        this._obj = await this.build();
+        const graph = this._obj;
+        const originalInvoke = graph.invoke;
+        const langGraphTraceObj = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromName('langGraphTracer');
+        const tracer = langGraphTraceObj.obj;
+        tracer.setName(this.graphName);
+        graph.invoke = async (input, config) => {
+            var _a;
+            if ((_a = config === null || config === void 0 ? void 0 : config.tags) === null || _a === void 0 ? void 0 : _a.includes('trace-log')) {
+                config.callbacks = [tracer, ...((config === null || config === void 0 ? void 0 : config.callbacks) || [])];
+            }
+            return await originalInvoke.call(graph, input, config);
+        };
+        this.status = tegg_runtime_1.EggObjectStatus.READY;
+    }
+    async build() {
+        const stateGraph = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromName(this.graphName);
+        await this.boundNodes(stateGraph);
+        await this.boundEdges(stateGraph);
+        const graphObj = stateGraph.obj;
+        const checkpoint = this.graphMetadata.checkpoint;
+        let compileGraph;
+        if (checkpoint) {
+            let checkpointObj;
+            if (typeof checkpoint !== 'string') {
+                checkpointObj = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromClazz(checkpoint);
+            }
+            else {
+                checkpointObj = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromName(tegg_langchain_decorator_1.ChatCheckpointSaverInjectName, [{
+                        attribute: tegg_langchain_decorator_1.ChatCheckpointSaverQualifierAttribute,
+                        value: checkpoint,
+                    }]);
+            }
+            compileGraph = graphObj.compile({
+                checkpointer: checkpointObj.obj,
+            });
+        }
+        else {
+            compileGraph = graphObj.compile();
+        }
+        return compileGraph;
+    }
+    async boundNodes(stateGraph) {
+        var _a;
+        const graphObj = stateGraph.obj;
+        const nodes = (_a = this.graphMetadata.nodes) !== null && _a !== void 0 ? _a : [];
+        for (let i = 0; i < nodes.length; i++) {
+            const node = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromClazz(nodes[i]);
+            const nodeObj = node.obj;
+            const nodeMetadata = node.proto.getMetaData(tegg_langchain_decorator_1.GRAPH_NODE_METADATA);
+            if (nodeMetadata) {
+                if (tegg_langchain_decorator_1.TeggToolNode.prototype.isPrototypeOf(nodeObj)) {
+                    graphObj.addNode(nodeMetadata.nodeName, nodeObj.toolNode);
+                }
+                else {
+                    graphObj.addNode(nodeMetadata.nodeName, nodeObj.execute.bind(nodeObj));
+                }
+            }
+        }
+    }
+    async boundEdges(stateGraph) {
+        var _a;
+        const graphObj = stateGraph.obj;
+        const edges = (_a = this.graphMetadata.edges) !== null && _a !== void 0 ? _a : [];
+        for (let i = 0; i < edges.length; i++) {
+            const edge = await tegg_runtime_1.EggContainerFactory.getOrCreateEggObjectFromClazz(edges[i]);
+            const edgeObj = edge.obj;
+            const edgeMetadata = edge.proto.getMetaData(tegg_langchain_decorator_1.GRAPH_EDGE_METADATA);
+            if (edgeMetadata) {
+                if (edgeObj.execute) {
+                    graphObj.addConditionalEdges(edgeMetadata.fromNodeName, edgeObj.execute.bind(edgeObj), edgeMetadata.toNodeNames);
+                }
+                else {
+                    graphObj.addEdge(edgeMetadata.fromNodeName, edgeMetadata.toNodeNames[0]);
+                }
+            }
+        }
+    }
+    injectProperty() {
+        throw new Error('never call GraphObject#injectProperty');
+    }
+    get isReady() {
+        return this.status === tegg_runtime_1.EggObjectStatus.READY;
+    }
+    get obj() {
+        return this._obj;
+    }
+    static async createObject(name, proto) {
+        const compiledStateGraphObject = new CompiledStateGraphObject(name, proto);
+        await compiledStateGraphObject.init();
+        return compiledStateGraphObject;
+    }
+}
+exports.CompiledStateGraphObject = CompiledStateGraphObject;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ29tcGlsZWRTdGF0ZUdyYXBoT2JqZWN0LmpzIiwic291cmNlUm9vdCI6IiIsInNvdXJjZXMiOlsiQ29tcGlsZWRTdGF0ZUdyYXBoT2JqZWN0LnRzIl0sIm5hbWVzIjpbXSwibWFwcGluZ3MiOiI7OztBQUFBLHNEQU02QjtBQUM3QixzQ0FBaUY7QUFHakYsOEVBQW9RO0FBSXBRLE1BQWEsd0JBQXdCO0lBV25DLFlBQVksSUFBbUIsRUFBRSxLQUE4Qjs7UUFWdkQsV0FBTSxHQUFvQiw4QkFBZSxDQUFDLE9BQU8sQ0FBQztRQVd4RCxJQUFJLENBQUMsSUFBSSxHQUFHLElBQUksQ0FBQztRQUNqQixJQUFJLENBQUMsS0FBSyxHQUFHLEtBQUssQ0FBQztRQUNuQixJQUFJLENBQUMsR0FBRyxHQUFHLDZCQUFjLENBQUMsVUFBVSxFQUFHLENBQUM7UUFDeEMsSUFBSSxDQUFDLEVBQUUsR0FBRyxvQkFBYSxDQUFDLGNBQWMsQ0FBQyxJQUFJLENBQUMsS0FBSyxDQUFDLEVBQUUsRUFBRSxNQUFBLElBQUksQ0FBQyxHQUFHLDBDQUFFLEVBQUUsQ0FBQyxDQUFDO1FBQ3BFLElBQUksQ0FBQyxhQUFhLEdBQUcsS0FBSyxDQUFDLGFBQWEsQ0FBQztRQUN6QyxJQUFJLENBQUMsU0FBUyxHQUFHLEtBQUssQ0FBQyxTQUFTLENBQUM7SUFDbkMsQ0FBQztJQUVELEtBQUssQ0FBQyxJQUFJO1FBQ1IsSUFBSSxDQUFDLElBQUksR0FBRyxNQUFNLElBQUksQ0FBQyxLQUFLLEVBQUUsQ0FBQztRQUMvQixNQUFNLEtBQUssR0FBRyxJQUFJLENBQUMsSUFBb0MsQ0FBQztRQUV4RCxNQUFNLGNBQWMsR0FBRyxLQUFLLENBQUMsTUFBTSxDQUFDO1FBQ3BDLE1BQU0saUJBQWlCLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw0QkFBNEIsQ0FBQyxpQkFBaUIsQ0FBQyxDQUFDO1FBQ3BHLE1BQU0sTUFBTSxHQUFHLGlCQUFpQixDQUFDLEdBQXNCLENBQUM7UUFDeEQsTUFBTSxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDL0IsS0FBSyxDQUFDLE1BQU0sR0FBRyxLQUFLLEVBQUUsS0FBVSxFQUFFLE1BQVksRUFBRSxFQUFFOztZQUNoRCxJQUFJLE1BQUEsTUFBTSxhQUFOLE1BQU0sdUJBQU4sTUFBTSxDQUFFLElBQUksMENBQUUsUUFBUSxDQUFDLFdBQVcsQ0FBQyxFQUFFLENBQUM7Z0JBQ3hDLE1BQU0sQ0FBQyxTQUFTLEdBQUcsQ0FBRSxNQUFNLEVBQUUsR0FBRyxDQUFDLENBQUEsTUFBTSxhQUFOLE1BQU0sdUJBQU4sTUFBTSxDQUFFLFNBQVMsS0FBSSxFQUFFLENBQUMsQ0FBRSxDQUFDO1lBQzlELENBQUM7WUFDRCxPQUFPLE1BQU0sY0FBYyxDQUFDLElBQUksQ0FBQyxLQUFLLEVBQUUsS0FBSyxFQUFFLE1BQU0sQ0FBQyxDQUFDO1FBQ3pELENBQUMsQ0FBQztRQUVGLElBQUksQ0FBQyxNQUFNLEdBQUcsOEJBQWUsQ0FBQyxLQUFLLENBQUM7SUFDdEMsQ0FBQztJQUVELEtBQUssQ0FBQyxLQUFLO1FBQ1QsTUFBTSxVQUFVLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw0QkFBNEIsQ0FBQyxJQUFJLENBQUMsU0FBUyxDQUFDLENBQUM7UUFDMUYsTUFBTSxJQUFJLENBQUMsVUFBVSxDQUFDLFVBQVUsQ0FBQyxDQUFDO1FBQ2xDLE1BQU0sSUFBSSxDQUFDLFVBQVUsQ0FBQyxVQUFVLENBQUMsQ0FBQztRQUNsQyxNQUFNLFFBQVEsR0FBRyxVQUFVLENBQUMsR0FBdUIsQ0FBQztRQUNwRCxNQUFNLFVBQVUsR0FBRyxJQUFJLENBQUMsYUFBYSxDQUFDLFVBQVUsQ0FBQztRQUNqRCxJQUFJLFlBQVksQ0FBQztRQUNqQixJQUFJLFVBQVUsRUFBRSxDQUFDO1lBQ2YsSUFBSSxhQUF3QixDQUFDO1lBQzdCLElBQUksT0FBTyxVQUFVLEtBQUssUUFBUSxFQUFFLENBQUM7Z0JBQ25DLGFBQWEsR0FBRyxNQUFNLGtDQUFtQixDQUFDLDZCQUE2QixDQUFDLFVBQVUsQ0FBQyxDQUFDO1lBQ3RGLENBQUM7aUJBQU0sQ0FBQztnQkFDTixhQUFhLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw0QkFBNEIsQ0FBQyx3REFBNkIsRUFBRSxDQUFDO3dCQUNyRyxTQUFTLEVBQUUsZ0VBQXFDO3dCQUNoRCxLQUFLLEVBQUUsVUFBVTtxQkFDbEIsQ0FBQyxDQUFDLENBQUM7WUFDTixDQUFDO1lBQ0QsWUFBWSxHQUFHLFFBQVEsQ0FBQyxPQUFPLENBQUM7Z0JBQzlCLFlBQVksRUFBRSxhQUFhLENBQUMsR0FBMEI7YUFDdkQsQ0FBQyxDQUFDO1FBQ0wsQ0FBQzthQUFNLENBQUM7WUFDTixZQUFZLEdBQUcsUUFBUSxDQUFDLE9BQU8sRUFBRSxDQUFDO1FBQ3BDLENBQUM7UUFDRCxPQUFPLFlBQVksQ0FBQztJQUN0QixDQUFDO0lBRUQsS0FBSyxDQUFDLFVBQVUsQ0FBQyxVQUFxQjs7UUFDcEMsTUFBTSxRQUFRLEdBQUcsVUFBVSxDQUFDLEdBQXVCLENBQUM7UUFDcEQsTUFBTSxLQUFLLEdBQUcsTUFBQSxJQUFJLENBQUMsYUFBYSxDQUFDLEtBQUssbUNBQUksRUFBRSxDQUFDO1FBQzdDLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxLQUFLLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7WUFDdEMsTUFBTSxJQUFJLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw2QkFBNkIsQ0FBQyxLQUFLLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUMvRSxNQUFNLE9BQU8sR0FBRyxJQUFJLENBQUMsR0FBaUMsQ0FBQztZQUN2RCxNQUFNLFlBQVksR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLFdBQVcsQ0FBb0IsOENBQW1CLENBQUMsQ0FBQztZQUNwRixJQUFJLFlBQVksRUFBRSxDQUFDO2dCQUNqQixJQUFJLHVDQUFZLENBQUMsU0FBUyxDQUFDLGFBQWEsQ0FBQyxPQUFPLENBQUMsRUFBRSxDQUFDO29CQUNsRCxRQUFRLENBQUMsT0FBTyxDQUFDLFlBQVksQ0FBQyxRQUFRLEVBQUcsT0FBbUMsQ0FBQyxRQUFRLENBQUMsQ0FBQztnQkFDekYsQ0FBQztxQkFBTSxDQUFDO29CQUNOLFFBQVEsQ0FBQyxPQUFPLENBQUMsWUFBWSxDQUFDLFFBQVEsRUFBRSxPQUFPLENBQUMsT0FBTyxDQUFDLElBQUksQ0FBQyxPQUFPLENBQUMsQ0FBQyxDQUFDO2dCQUN6RSxDQUFDO1lBQ0gsQ0FBQztRQUNILENBQUM7SUFDSCxDQUFDO0lBRUQsS0FBSyxDQUFDLFVBQVUsQ0FBQyxVQUFxQjs7UUFDcEMsTUFBTSxRQUFRLEdBQUcsVUFBVSxDQUFDLEdBQXVCLENBQUM7UUFDcEQsTUFBTSxLQUFLLEdBQUcsTUFBQSxJQUFJLENBQUMsYUFBYSxDQUFDLEtBQUssbUNBQUksRUFBRSxDQUFDO1FBQzdDLEtBQUssSUFBSSxDQUFDLEdBQUcsQ0FBQyxFQUFFLENBQUMsR0FBRyxLQUFLLENBQUMsTUFBTSxFQUFFLENBQUMsRUFBRSxFQUFFLENBQUM7WUFDdEMsTUFBTSxJQUFJLEdBQUcsTUFBTSxrQ0FBbUIsQ0FBQyw2QkFBNkIsQ0FBQyxLQUFLLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQztZQUMvRSxNQUFNLE9BQU8sR0FBRyxJQUFJLENBQUMsR0FBaUMsQ0FBQztZQUN2RCxNQUFNLFlBQVksR0FBRyxJQUFJLENBQUMsS0FBSyxDQUFDLFdBQVcsQ0FBb0IsOENBQW1CLENBQUMsQ0FBQztZQUNwRixJQUFJLFlBQVksRUFBRSxDQUFDO2dCQUNqQixJQUFJLE9BQU8sQ0FBQyxPQUFPLEVBQUUsQ0FBQztvQkFDcEIsUUFBUSxDQUFDLG1CQUFtQixDQUFDLFlBQVksQ0FBQyxZQUFZLEVBQUUsT0FBTyxDQUFDLE9BQU8sQ0FBQyxJQUFJLENBQUMsT0FBTyxDQUFDLEVBQUUsWUFBWSxDQUFDLFdBQVcsQ0FBQyxDQUFDO2dCQUNuSCxDQUFDO3FCQUFNLENBQUM7b0JBQ04sUUFBUSxDQUFDLE9BQU8sQ0FBQyxZQUFZLENBQUMsWUFBWSxFQUFFLFlBQVksQ0FBQyxXQUFXLENBQUMsQ0FBQyxDQUFDLENBQUMsQ0FBQztnQkFDM0UsQ0FBQztZQUNILENBQUM7UUFDSCxDQUFDO0lBQ0gsQ0FBQztJQUVELGNBQWM7UUFDWixNQUFNLElBQUksS0FBSyxDQUFDLHVDQUF1QyxDQUFDLENBQUM7SUFDM0QsQ0FBQztJQUVELElBQUksT0FBTztRQUNULE9BQU8sSUFBSSxDQUFDLE1BQU0sS0FBSyw4QkFBZSxDQUFDLEtBQUssQ0FBQztJQUMvQyxDQUFDO0lBRUQsSUFBSSxHQUFHO1FBQ0wsT0FBTyxJQUFJLENBQUMsSUFBSSxDQUFDO0lBQ25CLENBQUM7SUFFRCxNQUFNLENBQUMsS0FBSyxDQUFDLFlBQVksQ0FBQyxJQUFtQixFQUFFLEtBQW1CO1FBQ2hFLE1BQU0sd0JBQXdCLEdBQUcsSUFBSSx3QkFBd0IsQ0FBQyxJQUFJLEVBQUUsS0FBZ0MsQ0FBQyxDQUFDO1FBQ3RHLE1BQU0sd0JBQXdCLENBQUMsSUFBSSxFQUFFLENBQUM7UUFDdEMsT0FBTyx3QkFBd0IsQ0FBQztJQUNsQyxDQUFDO0NBQ0Y7QUFuSEQsNERBbUhDIn0=

package/lib/graph/CompiledStateGraphProto.d.ts

@@ -0,0 +1,21 @@
+import { EggPrototype, LoadUnit } from '@eggjs/tegg-metadata';
+import { InjectObjectProto } from '@eggjs/tegg-types';
+import { AccessLevel, EggPrototypeName, ObjectInitType, QualifierInfo, Id, QualifierAttribute, QualifierValue } from '@eggjs/tegg';
+import { GraphMetadata } from '@eggjs/tegg-langchain-decorator';
+export declare class CompiledStateGraphProto implements EggPrototype {
+    private readonly qualifiers;
+    readonly accessLevel = AccessLevel.PUBLIC;
+    id: Id;
+    readonly initType = ObjectInitType.SINGLETON;
+    readonly injectObjects: InjectObjectProto[];
+    loadUnitId: string;
+    readonly name: EggPrototypeName;
+    readonly graphMetadata: GraphMetadata;
+    readonly graphName: string;
+    constructor(loadUnit: LoadUnit, protoName: string, graphName: string, graphMetadata: GraphMetadata);
+    constructEggObject(): object;
+    getMetaData<T>(): T | undefined;
+    verifyQualifier(qualifier: QualifierInfo): boolean;
+    verifyQualifiers(qualifiers: QualifierInfo[]): boolean;
+    getQualifier(attribute: QualifierAttribute): QualifierValue | undefined;
+}

package/lib/graph/CompiledStateGraphProto.js

@@ -0,0 +1,42 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.CompiledStateGraphProto = void 0;
+const tegg_1 = require("@eggjs/tegg");
+class CompiledStateGraphProto {
+    constructor(loadUnit, protoName, graphName, graphMetadata) {
+        this.accessLevel = tegg_1.AccessLevel.PUBLIC;
+        this.initType = tegg_1.ObjectInitType.SINGLETON;
+        this.injectObjects = [];
+        this.loadUnitId = loadUnit.id;
+        this.qualifiers = [];
+        this.name = protoName;
+        this.graphMetadata = graphMetadata;
+        this.graphName = graphName;
+        this.id = tegg_1.IdenticalUtil.createProtoId(loadUnit.id, protoName);
+    }
+    constructEggObject() {
+        return {};
+    }
+    getMetaData() {
+        // TODO set default metadata
+        return;
+    }
+    verifyQualifier(qualifier) {
+        const selfQualifiers = this.qualifiers.find(t => t.attribute === qualifier.attribute);
+        return (selfQualifiers === null || selfQualifiers === void 0 ? void 0 : selfQualifiers.value) === qualifier.value;
+    }
+    verifyQualifiers(qualifiers) {
+        for (const qualifier of qualifiers) {
+            if (!this.verifyQualifier(qualifier)) {
+                return false;
+            }
+        }
+        return true;
+    }
+    getQualifier(attribute) {
+        const qualifier = this.qualifiers.find(t => t.attribute === attribute);
+        return qualifier === null || qualifier === void 0 ? void 0 : qualifier.value;
+    }
+}
+exports.CompiledStateGraphProto = CompiledStateGraphProto;
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiQ29tcGlsZWRTdGF0ZUdyYXBoUHJvdG8uanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJDb21waWxlZFN0YXRlR3JhcGhQcm90by50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOzs7QUFJQSxzQ0FTcUI7QUFHckIsTUFBYSx1QkFBdUI7SUFXbEMsWUFBWSxRQUFrQixFQUFFLFNBQWlCLEVBQUUsU0FBaUIsRUFBRSxhQUE0QjtRQVR6RixnQkFBVyxHQUFHLGtCQUFXLENBQUMsTUFBTSxDQUFDO1FBRWpDLGFBQVEsR0FBRyxxQkFBYyxDQUFDLFNBQVMsQ0FBQztRQUNwQyxrQkFBYSxHQUF3QixFQUFFLENBQUM7UUFPL0MsSUFBSSxDQUFDLFVBQVUsR0FBRyxRQUFRLENBQUMsRUFBRSxDQUFDO1FBQzlCLElBQUksQ0FBQyxVQUFVLEdBQUcsRUFBRSxDQUFDO1FBQ3JCLElBQUksQ0FBQyxJQUFJLEdBQUcsU0FBUyxDQUFDO1FBQ3RCLElBQUksQ0FBQyxhQUFhLEdBQUcsYUFBYSxDQUFDO1FBQ25DLElBQUksQ0FBQyxTQUFTLEdBQUcsU0FBUyxDQUFDO1FBQzNCLElBQUksQ0FBQyxFQUFFLEdBQUcsb0JBQWEsQ0FBQyxhQUFhLENBQUMsUUFBUSxDQUFDLEVBQUUsRUFBRSxTQUFTLENBQUMsQ0FBQztJQUNoRSxDQUFDO0lBRUQsa0JBQWtCO1FBQ2hCLE9BQU8sRUFBRSxDQUFDO0lBQ1osQ0FBQztJQUVELFdBQVc7UUFDVCw0QkFBNEI7UUFDNUIsT0FBTztJQUNULENBQUM7SUFFRCxlQUFlLENBQUMsU0FBd0I7UUFDdEMsTUFBTSxjQUFjLEdBQUcsSUFBSSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxDQUFDLENBQUMsU0FBUyxLQUFLLFNBQVMsQ0FBQyxTQUFTLENBQUMsQ0FBQztRQUN0RixPQUFPLENBQUEsY0FBYyxhQUFkLGNBQWMsdUJBQWQsY0FBYyxDQUFFLEtBQUssTUFBSyxTQUFTLENBQUMsS0FBSyxDQUFDO0lBQ25ELENBQUM7SUFFRCxnQkFBZ0IsQ0FBQyxVQUEyQjtRQUMxQyxLQUFLLE1BQU0sU0FBUyxJQUFJLFVBQVUsRUFBRSxDQUFDO1lBQ25DLElBQUksQ0FBQyxJQUFJLENBQUMsZUFBZSxDQUFDLFNBQVMsQ0FBQyxFQUFFLENBQUM7Z0JBQ3JDLE9BQU8sS0FBSyxDQUFDO1lBQ2YsQ0FBQztRQUNILENBQUM7UUFDRCxPQUFPLElBQUksQ0FBQztJQUNkLENBQUM7SUFFRCxZQUFZLENBQUMsU0FBNkI7UUFDeEMsTUFBTSxTQUFTLEdBQUcsSUFBSSxDQUFDLFVBQVUsQ0FBQyxJQUFJLENBQUMsQ0FBQyxDQUFDLEVBQUUsQ0FBQyxDQUFDLENBQUMsU0FBUyxLQUFLLFNBQVMsQ0FBQyxDQUFDO1FBQ3ZFLE9BQU8sU0FBUyxhQUFULFNBQVMsdUJBQVQsU0FBUyxDQUFFLEtBQUssQ0FBQztJQUMxQixDQUFDO0NBQ0Y7QUEvQ0QsMERBK0NDIn0=

package/lib/graph/GraphBuildHook.d.ts

@@ -0,0 +1,2 @@
+import { GlobalGraph } from '@eggjs/tegg-metadata';
+export declare function GraphBuildHook(globalGraph: GlobalGraph): void;

package/lib/graph/GraphBuildHook.js

@@ -0,0 +1,24 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.GraphBuildHook = GraphBuildHook;
+const tegg_langchain_decorator_1 = require("@eggjs/tegg-langchain-decorator");
+const LangGraphTracer_1 = require("../tracing/LangGraphTracer");
+function GraphBuildHook(globalGraph) {
+    var _a, _b;
+    let langchainGraphTracerProtoNode;
+    for (const moduleNode of globalGraph.moduleGraph.nodes.values()) {
+        for (const protoNode of moduleNode.val.protos) {
+            if (((_a = protoNode.val.proto) === null || _a === void 0 ? void 0 : _a.clazz) && (LangGraphTracer_1.LangGraphTracer.isPrototypeOf(protoNode.val.proto.clazz) || protoNode.val.proto.clazz === LangGraphTracer_1.LangGraphTracer)) {
+                langchainGraphTracerProtoNode = protoNode;
+            }
+        }
+    }
+    for (const moduleNode of globalGraph.moduleGraph.nodes.values()) {
+        for (const protoNode of moduleNode.val.protos) {
+            if (((_b = protoNode.val.proto) === null || _b === void 0 ? void 0 : _b.clazz) && tegg_langchain_decorator_1.AbstractStateGraph.isPrototypeOf(protoNode.val.proto.clazz)) {
+                globalGraph.addInject(moduleNode, protoNode, langchainGraphTracerProtoNode, langchainGraphTracerProtoNode.val.proto.name);
+            }
+        }
+    }
+}
+//# sourceMappingURL=data:application/json;base64,eyJ2ZXJzaW9uIjozLCJmaWxlIjoiR3JhcGhCdWlsZEhvb2suanMiLCJzb3VyY2VSb290IjoiIiwic291cmNlcyI6WyJHcmFwaEJ1aWxkSG9vay50cyJdLCJuYW1lcyI6W10sIm1hcHBpbmdzIjoiOztBQUlBLHdDQXdCQztBQTVCRCw4RUFBcUU7QUFDckUsZ0VBQTZEO0FBRzdELFNBQWdCLGNBQWMsQ0FBQyxXQUF3Qjs7SUFDckQsSUFBSSw2QkFBNkIsQ0FBQztJQUNsQyxLQUFLLE1BQU0sVUFBVSxJQUFJLFdBQVcsQ0FBQyxXQUFXLENBQUMsS0FBSyxDQUFDLE1BQU0sRUFBRSxFQUFFLENBQUM7UUFDaEUsS0FBSyxNQUFNLFNBQVMsSUFBSSxVQUFVLENBQUMsR0FBRyxDQUFDLE1BQU0sRUFBRSxDQUFDO1lBQzlDLElBQUksQ0FBQSxNQUFDLFNBQVMsQ0FBQyxHQUFHLENBQUMsS0FBOEIsMENBQUUsS0FBSyxLQUFJLENBQUMsaUNBQWUsQ0FBQyxhQUFhLENBQUUsU0FBUyxDQUFDLEdBQUcsQ0FBQyxLQUE4QixDQUFDLEtBQUssQ0FBQyxJQUFLLFNBQVMsQ0FBQyxHQUFHLENBQUMsS0FBOEIsQ0FBQyxLQUFLLEtBQUssaUNBQWUsQ0FBQyxFQUFFLENBQUM7Z0JBQzVOLDZCQUE2QixHQUFHLFNBQVMsQ0FBQztZQUM1QyxDQUFDO1FBQ0gsQ0FBQztJQUNILENBQUM7SUFFRCxLQUFLLE1BQU0sVUFBVSxJQUFJLFdBQVcsQ0FBQyxXQUFXLENBQUMsS0FBSyxDQUFDLE1BQU0sRUFBRSxFQUFFLENBQUM7UUFDaEUsS0FBSyxNQUFNLFNBQVMsSUFBSSxVQUFVLENBQUMsR0FBRyxDQUFDLE1BQU0sRUFBRSxDQUFDO1lBQzlDLElBQ0UsQ0FBQSxNQUFDLFNBQVMsQ0FBQyxHQUFHLENBQUMsS0FBOEIsMENBQUUsS0FBSyxLQUFJLDZDQUFrQixDQUFDLGFBQWEsQ0FBRSxTQUFTLENBQUMsR0FBRyxDQUFDLEtBQThCLENBQUMsS0FBSyxDQUFDLEVBQzdJLENBQUM7Z0JBQ0QsV0FBVyxDQUFDLFNBQVMsQ0FDbkIsVUFBVSxFQUNWLFNBQVMsRUFDVCw2QkFBNkIsRUFDN0IsNkJBQTZCLENBQUMsR0FBRyxDQUFDLEtBQUssQ0FBQyxJQUFJLENBQzdDLENBQUM7WUFDSixDQUFDO1FBQ0gsQ0FBQztJQUNILENBQUM7QUFDSCxDQUFDIn0=