@flink-app/flink 1.0.0 → 2.0.0-alpha.48

package/spec/mock-project/dist/src/ai/AgentExecutor.js

@@ -0,0 +1,279 @@
+"use strict";
+var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) {
+    function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); }
+    return new (P || (P = Promise))(function (resolve, reject) {
+        function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } }
+        function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } }
+        function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); }
+        step((generator = generator.apply(thisArg, _arguments || [])).next());
+    });
+};
+var __generator = (this && this.__generator) || function (thisArg, body) {
+    var _ = { label: 0, sent: function() { if (t[0] & 1) throw t[1]; return t[1]; }, trys: [], ops: [] }, f, y, t, g = Object.create((typeof Iterator === "function" ? Iterator : Object).prototype);
+    return g.next = verb(0), g["throw"] = verb(1), g["return"] = verb(2), typeof Symbol === "function" && (g[Symbol.iterator] = function() { return this; }), g;
+    function verb(n) { return function (v) { return step([n, v]); }; }
+    function step(op) {
+        if (f) throw new TypeError("Generator is already executing.");
+        while (g && (g = 0, op[0] && (_ = 0)), _) try {
+            if (f = 1, y && (t = op[0] & 2 ? y["return"] : op[0] ? y["throw"] || ((t = y["return"]) && t.call(y), 0) : y.next) && !(t = t.call(y, op[1])).done) return t;
+            if (y = 0, t) op = [op[0] & 2, t.value];
+            switch (op[0]) {
+                case 0: case 1: t = op; break;
+                case 4: _.label++; return { value: op[1], done: false };
+                case 5: _.label++; y = op[1]; op = [0]; continue;
+                case 7: op = _.ops.pop(); _.trys.pop(); continue;
+                default:
+                    if (!(t = _.trys, t = t.length > 0 && t[t.length - 1]) && (op[0] === 6 || op[0] === 2)) { _ = 0; continue; }
+                    if (op[0] === 3 && (!t || (op[1] > t[0] && op[1] < t[3]))) { _.label = op[1]; break; }
+                    if (op[0] === 6 && _.label < t[1]) { _.label = t[1]; t = op; break; }
+                    if (t && _.label < t[2]) { _.label = t[2]; _.ops.push(op); break; }
+                    if (t[2]) _.ops.pop();
+                    _.trys.pop(); continue;
+            }
+            op = body.call(thisArg, _);
+        } catch (e) { op = [6, e]; y = 0; } finally { f = t = 0; }
+        if (op[0] & 5) throw op[1]; return { value: op[0] ? op[1] : void 0, done: true };
+    }
+};
+var __await = (this && this.__await) || function (v) { return this instanceof __await ? (this.v = v, this) : new __await(v); }
+var __asyncGenerator = (this && this.__asyncGenerator) || function (thisArg, _arguments, generator) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var g = generator.apply(thisArg, _arguments || []), i, q = [];
+    return i = Object.create((typeof AsyncIterator === "function" ? AsyncIterator : Object).prototype), verb("next"), verb("throw"), verb("return", awaitReturn), i[Symbol.asyncIterator] = function () { return this; }, i;
+    function awaitReturn(f) { return function (v) { return Promise.resolve(v).then(f, reject); }; }
+    function verb(n, f) { if (g[n]) { i[n] = function (v) { return new Promise(function (a, b) { q.push([n, v, a, b]) > 1 || resume(n, v); }); }; if (f) i[n] = f(i[n]); } }
+    function resume(n, v) { try { step(g[n](v)); } catch (e) { settle(q[0][3], e); } }
+    function step(r) { r.value instanceof __await ? Promise.resolve(r.value.v).then(fulfill, reject) : settle(q[0][2], r); }
+    function fulfill(value) { resume("next", value); }
+    function reject(value) { resume("throw", value); }
+    function settle(f, v) { if (f(v), q.shift(), q.length) resume(q[0][0], q[0][1]); }
+};
+var __asyncValues = (this && this.__asyncValues) || function (o) {
+    if (!Symbol.asyncIterator) throw new TypeError("Symbol.asyncIterator is not defined.");
+    var m = o[Symbol.asyncIterator], i;
+    return m ? m.call(o) : (o = typeof __values === "function" ? __values(o) : o[Symbol.iterator](), i = {}, verb("next"), verb("throw"), verb("return"), i[Symbol.asyncIterator] = function () { return this; }, i);
+    function verb(n) { i[n] = o[n] && function (v) { return new Promise(function (resolve, reject) { v = o[n](v), settle(resolve, reject, v.done, v.value); }); }; }
+    function settle(resolve, reject, d, v) { Promise.resolve(v).then(function(v) { resolve({ value: v, done: d }); }, reject); }
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentExecutor = void 0;
+var AgentRunner_1 = require("./AgentRunner");
+var FlinkLog_1 = require("../FlinkLog");
+var AgentExecutor = /** @class */ (function () {
+    function AgentExecutor(agentProps, tools, ctx, llmAdapters, agentName) {
+        this.agentProps = agentProps;
+        this.tools = tools;
+        this.ctx = ctx;
+        this.llmAdapters = llmAdapters;
+        this.agentName = agentName;
+    }
+    /**
+     * Unified execution method - supports both awaiting and streaming
+     *
+     * Uses lazy generator pattern (similar to Vercel AI SDK) to allow
+     * multiple consumption without re-execution.
+     *
+     * Examples:
+     *   const response = agent.run({ message: "Hello" });
+     *   const result = await response.result;  // Await final result
+     *   for await (const text of response.textStream) { ... }  // Stream text
+     *   for await (const chunk of response.fullStream) { ... }  // Stream all events
+     */
+    AgentExecutor.prototype.run = function (input) {
+        // Permission check
+        if (this.agentProps.permissions) {
+            this.checkPermissionsSync(input.user);
+        }
+        var runner = this.getRunner();
+        FlinkLog_1.log.debug("Running agent ".concat(this.agentName || "unknown"));
+        // Lazy evaluation - generator only starts when first consumed
+        var baseGenerator = null;
+        var buffer = [];
+        var done = false;
+        var getBaseGenerator = function () {
+            if (!baseGenerator) {
+                baseGenerator = runner.streamGenerator(input);
+            }
+            return baseGenerator;
+        };
+        // Create independent iterators that share buffered chunks
+        var createIterator = function () {
+            var index = 0;
+            return (function () {
+                return __asyncGenerator(this, arguments, function () {
+                    var gen, _a, value, isDone;
+                    return __generator(this, function (_b) {
+                        switch (_b.label) {
+                            case 0:
+                                if (!true) return [3 /*break*/, 7];
+                                if (!(index < buffer.length)) return [3 /*break*/, 3];
+                                return [4 /*yield*/, __await(buffer[index++])];
+                            case 1: return [4 /*yield*/, _b.sent()];
+                            case 2:
+                                _b.sent();
+                                return [3 /*break*/, 0];
+                            case 3:
+                                // If already done, exit
+                                if (done) {
+                                    return [3 /*break*/, 7];
+                                }
+                                gen = getBaseGenerator();
+                                return [4 /*yield*/, __await(gen.next())];
+                            case 4:
+                                _a = _b.sent(), value = _a.value, isDone = _a.done;
+                                if (isDone) {
+                                    done = true;
+                                    return [3 /*break*/, 7];
+                                }
+                                // Buffer and yield
+                                buffer.push(value);
+                                return [4 /*yield*/, __await(value)];
+                            case 5: return [4 /*yield*/, _b.sent()];
+                            case 6:
+                                _b.sent();
+                                index++;
+                                return [3 /*break*/, 0];
+                            case 7: return [2 /*return*/];
+                        }
+                    });
+                });
+            })();
+        };
+        return {
+            result: this.consumeAsResult(createIterator()),
+            textStream: this.consumeAsTextStream(createIterator()),
+            fullStream: createIterator(),
+        };
+    };
+    /**
+     * Consume stream as final result (for await pattern)
+     */
+    AgentExecutor.prototype.consumeAsResult = function (stream) {
+        return __awaiter(this, void 0, void 0, function () {
+            var result, chunk, e_1_1;
+            var _a, stream_1, stream_1_1;
+            var _b, e_1, _c, _d;
+            return __generator(this, function (_e) {
+                switch (_e.label) {
+                    case 0:
+                        _e.trys.push([0, 5, 6, 11]);
+                        _a = true, stream_1 = __asyncValues(stream);
+                        _e.label = 1;
+                    case 1: return [4 /*yield*/, stream_1.next()];
+                    case 2:
+                        if (!(stream_1_1 = _e.sent(), _b = stream_1_1.done, !_b)) return [3 /*break*/, 4];
+                        _d = stream_1_1.value;
+                        _a = false;
+                        chunk = _d;
+                        if (chunk.type === "complete") {
+                            result = chunk.result;
+                        }
+                        _e.label = 3;
+                    case 3:
+                        _a = true;
+                        return [3 /*break*/, 1];
+                    case 4: return [3 /*break*/, 11];
+                    case 5:
+                        e_1_1 = _e.sent();
+                        e_1 = { error: e_1_1 };
+                        return [3 /*break*/, 11];
+                    case 6:
+                        _e.trys.push([6, , 9, 10]);
+                        if (!(!_a && !_b && (_c = stream_1.return))) return [3 /*break*/, 8];
+                        return [4 /*yield*/, _c.call(stream_1)];
+                    case 7:
+                        _e.sent();
+                        _e.label = 8;
+                    case 8: return [3 /*break*/, 10];
+                    case 9:
+                        if (e_1) throw e_1.error;
+                        return [7 /*endfinally*/];
+                    case 10: return [7 /*endfinally*/];
+                    case 11:
+                        if (!result) {
+                            throw new Error("Agent execution did not complete");
+                        }
+                        return [2 /*return*/, result];
+                }
+            });
+        });
+    };
+    /**
+     * Consume stream as text-only stream (for simple streaming UX)
+     */
+    AgentExecutor.prototype.consumeAsTextStream = function (stream) {
+        return __asyncGenerator(this, arguments, function consumeAsTextStream_1() {
+            var _a, stream_2, stream_2_1, chunk, e_2_1;
+            var _b, e_2, _c, _d;
+            return __generator(this, function (_e) {
+                switch (_e.label) {
+                    case 0:
+                        _e.trys.push([0, 7, 8, 13]);
+                        _a = true, stream_2 = __asyncValues(stream);
+                        _e.label = 1;
+                    case 1: return [4 /*yield*/, __await(stream_2.next())];
+                    case 2:
+                        if (!(stream_2_1 = _e.sent(), _b = stream_2_1.done, !_b)) return [3 /*break*/, 6];
+                        _d = stream_2_1.value;
+                        _a = false;
+                        chunk = _d;
+                        if (!(chunk.type === "text_delta")) return [3 /*break*/, 5];
+                        return [4 /*yield*/, __await(chunk.delta)];
+                    case 3: return [4 /*yield*/, _e.sent()];
+                    case 4:
+                        _e.sent();
+                        _e.label = 5;
+                    case 5:
+                        _a = true;
+                        return [3 /*break*/, 1];
+                    case 6: return [3 /*break*/, 13];
+                    case 7:
+                        e_2_1 = _e.sent();
+                        e_2 = { error: e_2_1 };
+                        return [3 /*break*/, 13];
+                    case 8:
+                        _e.trys.push([8, , 11, 12]);
+                        if (!(!_a && !_b && (_c = stream_2.return))) return [3 /*break*/, 10];
+                        return [4 /*yield*/, __await(_c.call(stream_2))];
+                    case 9:
+                        _e.sent();
+                        _e.label = 10;
+                    case 10: return [3 /*break*/, 12];
+                    case 11:
+                        if (e_2) throw e_2.error;
+                        return [7 /*endfinally*/];
+                    case 12: return [7 /*endfinally*/];
+                    case 13: return [2 /*return*/];
+                }
+            });
+        });
+    };
+    AgentExecutor.prototype.getRunner = function () {
+        if (!this.runner) {
+            this.runner = new AgentRunner_1.AgentRunner(this.agentProps, this.tools, this.llmAdapters, this.agentName);
+        }
+        return this.runner;
+    };
+    AgentExecutor.prototype.checkPermissionsSync = function (user) {
+        var perms = this.agentProps.permissions;
+        if (!perms)
+            return;
+        if (typeof perms === "function") {
+            var hasPermission = perms(user);
+            if (!hasPermission) {
+                throw new Error("Permission denied for agent ".concat(this.agentName || "unknown"));
+            }
+            return;
+        }
+        if (!user) {
+            throw new Error("Permission denied for agent ".concat(this.agentName || "unknown"));
+        }
+        var requiredPerms = Array.isArray(perms) ? perms : [perms];
+        var userPerms = user.permissions || [];
+        if (!requiredPerms.every(function (p) { return userPerms.includes(p); })) {
+            throw new Error("Permission denied for agent ".concat(this.agentName || "unknown"));
+        }
+    };
+    return AgentExecutor;
+}());
+exports.AgentExecutor = AgentExecutor;