@difizen/magent-au
Version:
418 lines (417 loc) • 33 kB
JavaScript
var _dec, _dec2, _dec3, _dec4, _dec5, _dec6, _class, _class2, _descriptor, _descriptor2, _descriptor3, _descriptor4;
function _typeof(o) { "@babel/helpers - typeof"; return _typeof = "function" == typeof Symbol && "symbol" == typeof Symbol.iterator ? function (o) { return typeof o; } : function (o) { return o && "function" == typeof Symbol && o.constructor === Symbol && o !== Symbol.prototype ? "symbol" : typeof o; }, _typeof(o); }
function _regeneratorRuntime() { "use strict"; /*! regenerator-runtime -- Copyright (c) 2014-present, Facebook, Inc. -- license (MIT): https://github.com/facebook/regenerator/blob/main/LICENSE */ _regeneratorRuntime = function _regeneratorRuntime() { return e; }; var t, e = {}, r = Object.prototype, n = r.hasOwnProperty, o = Object.defineProperty || function (t, e, r) { t[e] = r.value; }, i = "function" == typeof Symbol ? Symbol : {}, a = i.iterator || "@@iterator", c = i.asyncIterator || "@@asyncIterator", u = i.toStringTag || "@@toStringTag"; function define(t, e, r) { return Object.defineProperty(t, e, { value: r, enumerable: !0, configurable: !0, writable: !0 }), t[e]; } try { define({}, ""); } catch (t) { define = function define(t, e, r) { return t[e] = r; }; } function wrap(t, e, r, n) { var i = e && e.prototype instanceof Generator ? e : Generator, a = Object.create(i.prototype), c = new Context(n || []); return o(a, "_invoke", { value: makeInvokeMethod(t, r, c) }), a; } function tryCatch(t, e, r) { try { return { type: "normal", arg: t.call(e, r) }; } catch (t) { return { type: "throw", arg: t }; } } e.wrap = wrap; var h = "suspendedStart", l = "suspendedYield", f = "executing", s = "completed", y = {}; function Generator() {} function GeneratorFunction() {} function GeneratorFunctionPrototype() {} var p = {}; define(p, a, function () { return this; }); var d = Object.getPrototypeOf, v = d && d(d(values([]))); v && v !== r && n.call(v, a) && (p = v); var g = GeneratorFunctionPrototype.prototype = Generator.prototype = Object.create(p); function defineIteratorMethods(t) { ["next", "throw", "return"].forEach(function (e) { define(t, e, function (t) { return this._invoke(e, t); }); }); } function AsyncIterator(t, e) { function invoke(r, o, i, a) { var c = tryCatch(t[r], t, o); if ("throw" !== c.type) { var u = c.arg, h = u.value; return h && "object" == _typeof(h) && n.call(h, "__await") ? e.resolve(h.__await).then(function (t) { invoke("next", t, i, a); }, function (t) { invoke("throw", t, i, a); }) : e.resolve(h).then(function (t) { u.value = t, i(u); }, function (t) { return invoke("throw", t, i, a); }); } a(c.arg); } var r; o(this, "_invoke", { value: function value(t, n) { function callInvokeWithMethodAndArg() { return new e(function (e, r) { invoke(t, n, e, r); }); } return r = r ? r.then(callInvokeWithMethodAndArg, callInvokeWithMethodAndArg) : callInvokeWithMethodAndArg(); } }); } function makeInvokeMethod(e, r, n) { var o = h; return function (i, a) { if (o === f) throw new Error("Generator is already running"); if (o === s) { if ("throw" === i) throw a; return { value: t, done: !0 }; } for (n.method = i, n.arg = a;;) { var c = n.delegate; if (c) { var u = maybeInvokeDelegate(c, n); if (u) { if (u === y) continue; return u; } } if ("next" === n.method) n.sent = n._sent = n.arg;else if ("throw" === n.method) { if (o === h) throw o = s, n.arg; n.dispatchException(n.arg); } else "return" === n.method && n.abrupt("return", n.arg); o = f; var p = tryCatch(e, r, n); if ("normal" === p.type) { if (o = n.done ? s : l, p.arg === y) continue; return { value: p.arg, done: n.done }; } "throw" === p.type && (o = s, n.method = "throw", n.arg = p.arg); } }; } function maybeInvokeDelegate(e, r) { var n = r.method, o = e.iterator[n]; if (o === t) return r.delegate = null, "throw" === n && e.iterator.return && (r.method = "return", r.arg = t, maybeInvokeDelegate(e, r), "throw" === r.method) || "return" !== n && (r.method = "throw", r.arg = new TypeError("The iterator does not provide a '" + n + "' method")), y; var i = tryCatch(o, e.iterator, r.arg); if ("throw" === i.type) return r.method = "throw", r.arg = i.arg, r.delegate = null, y; var a = i.arg; return a ? a.done ? (r[e.resultName] = a.value, r.next = e.nextLoc, "return" !== r.method && (r.method = "next", r.arg = t), r.delegate = null, y) : a : (r.method = "throw", r.arg = new TypeError("iterator result is not an object"), r.delegate = null, y); } function pushTryEntry(t) { var e = { tryLoc: t[0] }; 1 in t && (e.catchLoc = t[1]), 2 in t && (e.finallyLoc = t[2], e.afterLoc = t[3]), this.tryEntries.push(e); } function resetTryEntry(t) { var e = t.completion || {}; e.type = "normal", delete e.arg, t.completion = e; } function Context(t) { this.tryEntries = [{ tryLoc: "root" }], t.forEach(pushTryEntry, this), this.reset(!0); } function values(e) { if (e || "" === e) { var r = e[a]; if (r) return r.call(e); if ("function" == typeof e.next) return e; if (!isNaN(e.length)) { var o = -1, i = function next() { for (; ++o < e.length;) if (n.call(e, o)) return next.value = e[o], next.done = !1, next; return next.value = t, next.done = !0, next; }; return i.next = i; } } throw new TypeError(_typeof(e) + " is not iterable"); } return GeneratorFunction.prototype = GeneratorFunctionPrototype, o(g, "constructor", { value: GeneratorFunctionPrototype, configurable: !0 }), o(GeneratorFunctionPrototype, "constructor", { value: GeneratorFunction, configurable: !0 }), GeneratorFunction.displayName = define(GeneratorFunctionPrototype, u, "GeneratorFunction"), e.isGeneratorFunction = function (t) { var e = "function" == typeof t && t.constructor; return !!e && (e === GeneratorFunction || "GeneratorFunction" === (e.displayName || e.name)); }, e.mark = function (t) { return Object.setPrototypeOf ? Object.setPrototypeOf(t, GeneratorFunctionPrototype) : (t.__proto__ = GeneratorFunctionPrototype, define(t, u, "GeneratorFunction")), t.prototype = Object.create(g), t; }, e.awrap = function (t) { return { __await: t }; }, defineIteratorMethods(AsyncIterator.prototype), define(AsyncIterator.prototype, c, function () { return this; }), e.AsyncIterator = AsyncIterator, e.async = function (t, r, n, o, i) { void 0 === i && (i = Promise); var a = new AsyncIterator(wrap(t, r, n, o), i); return e.isGeneratorFunction(r) ? a : a.next().then(function (t) { return t.done ? t.value : a.next(); }); }, defineIteratorMethods(g), define(g, u, "Generator"), define(g, a, function () { return this; }), define(g, "toString", function () { return "[object Generator]"; }), e.keys = function (t) { var e = Object(t), r = []; for (var n in e) r.push(n); return r.reverse(), function next() { for (; r.length;) { var t = r.pop(); if (t in e) return next.value = t, next.done = !1, next; } return next.done = !0, next; }; }, e.values = values, Context.prototype = { constructor: Context, reset: function reset(e) { if (this.prev = 0, this.next = 0, this.sent = this._sent = t, this.done = !1, this.delegate = null, this.method = "next", this.arg = t, this.tryEntries.forEach(resetTryEntry), !e) for (var r in this) "t" === r.charAt(0) && n.call(this, r) && !isNaN(+r.slice(1)) && (this[r] = t); }, stop: function stop() { this.done = !0; var t = this.tryEntries[0].completion; if ("throw" === t.type) throw t.arg; return this.rval; }, dispatchException: function dispatchException(e) { if (this.done) throw e; var r = this; function handle(n, o) { return a.type = "throw", a.arg = e, r.next = n, o && (r.method = "next", r.arg = t), !!o; } for (var o = this.tryEntries.length - 1; o >= 0; --o) { var i = this.tryEntries[o], a = i.completion; if ("root" === i.tryLoc) return handle("end"); if (i.tryLoc <= this.prev) { var c = n.call(i, "catchLoc"), u = n.call(i, "finallyLoc"); if (c && u) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } else if (c) { if (this.prev < i.catchLoc) return handle(i.catchLoc, !0); } else { if (!u) throw new Error("try statement without catch or finally"); if (this.prev < i.finallyLoc) return handle(i.finallyLoc); } } } }, abrupt: function abrupt(t, e) { for (var r = this.tryEntries.length - 1; r >= 0; --r) { var o = this.tryEntries[r]; if (o.tryLoc <= this.prev && n.call(o, "finallyLoc") && this.prev < o.finallyLoc) { var i = o; break; } } i && ("break" === t || "continue" === t) && i.tryLoc <= e && e <= i.finallyLoc && (i = null); var a = i ? i.completion : {}; return a.type = t, a.arg = e, i ? (this.method = "next", this.next = i.finallyLoc, y) : this.complete(a); }, complete: function complete(t, e) { if ("throw" === t.type) throw t.arg; return "break" === t.type || "continue" === t.type ? this.next = t.arg : "return" === t.type ? (this.rval = this.arg = t.arg, this.method = "return", this.next = "end") : "normal" === t.type && e && (this.next = e), y; }, finish: function finish(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.finallyLoc === t) return this.complete(r.completion, r.afterLoc), resetTryEntry(r), y; } }, catch: function _catch(t) { for (var e = this.tryEntries.length - 1; e >= 0; --e) { var r = this.tryEntries[e]; if (r.tryLoc === t) { var n = r.completion; if ("throw" === n.type) { var o = n.arg; resetTryEntry(r); } return o; } } throw new Error("illegal catch attempt"); }, delegateYield: function delegateYield(e, r, n) { return this.delegate = { iterator: values(e), resultName: r, nextLoc: n }, "next" === this.method && (this.arg = t), y; } }, e; }
function asyncGeneratorStep(gen, resolve, reject, _next, _throw, key, arg) { try { var info = gen[key](arg); var value = info.value; } catch (error) { reject(error); return; } if (info.done) { resolve(value); } else { Promise.resolve(value).then(_next, _throw); } }
function _asyncToGenerator(fn) { return function () { var self = this, args = arguments; return new Promise(function (resolve, reject) { var gen = fn.apply(self, args); function _next(value) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "next", value); } function _throw(err) { asyncGeneratorStep(gen, resolve, reject, _next, _throw, "throw", err); } _next(undefined); }); }; }
function _initializerDefineProperty(target, property, descriptor, context) { if (!descriptor) return; Object.defineProperty(target, property, { enumerable: descriptor.enumerable, configurable: descriptor.configurable, writable: descriptor.writable, value: descriptor.initializer ? descriptor.initializer.call(context) : void 0 }); }
function _classCallCheck(instance, Constructor) { if (!(instance instanceof Constructor)) { throw new TypeError("Cannot call a class as a function"); } }
function _defineProperties(target, props) { for (var i = 0; i < props.length; i++) { var descriptor = props[i]; descriptor.enumerable = descriptor.enumerable || false; descriptor.configurable = true; if ("value" in descriptor) descriptor.writable = true; Object.defineProperty(target, _toPropertyKey(descriptor.key), descriptor); } }
function _createClass(Constructor, protoProps, staticProps) { if (protoProps) _defineProperties(Constructor.prototype, protoProps); if (staticProps) _defineProperties(Constructor, staticProps); Object.defineProperty(Constructor, "prototype", { writable: false }); return Constructor; }
function _inherits(subClass, superClass) { if (typeof superClass !== "function" && superClass !== null) { throw new TypeError("Super expression must either be null or a function"); } subClass.prototype = Object.create(superClass && superClass.prototype, { constructor: { value: subClass, writable: true, configurable: true } }); Object.defineProperty(subClass, "prototype", { writable: false }); if (superClass) _setPrototypeOf(subClass, superClass); }
function _setPrototypeOf(o, p) { _setPrototypeOf = Object.setPrototypeOf ? Object.setPrototypeOf.bind() : function _setPrototypeOf(o, p) { o.__proto__ = p; return o; }; return _setPrototypeOf(o, p); }
function _createSuper(Derived) { var hasNativeReflectConstruct = _isNativeReflectConstruct(); return function _createSuperInternal() { var Super = _getPrototypeOf(Derived), result; if (hasNativeReflectConstruct) { var NewTarget = _getPrototypeOf(this).constructor; result = Reflect.construct(Super, arguments, NewTarget); } else { result = Super.apply(this, arguments); } return _possibleConstructorReturn(this, result); }; }
function _possibleConstructorReturn(self, call) { if (call && (_typeof(call) === "object" || typeof call === "function")) { return call; } else if (call !== void 0) { throw new TypeError("Derived constructors may only return object or undefined"); } return _assertThisInitialized(self); }
function _assertThisInitialized(self) { if (self === void 0) { throw new ReferenceError("this hasn't been initialised - super() hasn't been called"); } return self; }
function _isNativeReflectConstruct() { if (typeof Reflect === "undefined" || !Reflect.construct) return false; if (Reflect.construct.sham) return false; if (typeof Proxy === "function") return true; try { Boolean.prototype.valueOf.call(Reflect.construct(Boolean, [], function () {})); return true; } catch (e) { return false; } }
function _getPrototypeOf(o) { _getPrototypeOf = Object.setPrototypeOf ? Object.getPrototypeOf.bind() : function _getPrototypeOf(o) { return o.__proto__ || Object.getPrototypeOf(o); }; return _getPrototypeOf(o); }
function _applyDecoratedDescriptor(target, property, decorators, descriptor, context) { var desc = {}; Object.keys(descriptor).forEach(function (key) { desc[key] = descriptor[key]; }); desc.enumerable = !!desc.enumerable; desc.configurable = !!desc.configurable; if ('value' in desc || desc.initializer) { desc.writable = true; } desc = decorators.slice().reverse().reduce(function (desc, decorator) { return decorator(target, property, desc) || desc; }, desc); if (context && desc.initializer !== void 0) { desc.value = desc.initializer ? desc.initializer.call(context) : void 0; desc.initializer = undefined; } if (desc.initializer === void 0) { Object.defineProperty(target, property, desc); desc = null; } return desc; }
function _initializerWarningHelper(descriptor, context) { throw new Error('Decorating class property failed. Please ensure that ' + 'transform-class-properties is enabled and runs after the decorators transform.'); }
function _toConsumableArray(arr) { return _arrayWithoutHoles(arr) || _iterableToArray(arr) || _unsupportedIterableToArray(arr) || _nonIterableSpread(); }
function _nonIterableSpread() { throw new TypeError("Invalid attempt to spread non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _iterableToArray(iter) { if (typeof Symbol !== "undefined" && iter[Symbol.iterator] != null || iter["@@iterator"] != null) return Array.from(iter); }
function _arrayWithoutHoles(arr) { if (Array.isArray(arr)) return _arrayLikeToArray(arr); }
function _slicedToArray(arr, i) { return _arrayWithHoles(arr) || _iterableToArrayLimit(arr, i) || _unsupportedIterableToArray(arr, i) || _nonIterableRest(); }
function _nonIterableRest() { throw new TypeError("Invalid attempt to destructure non-iterable instance.\nIn order to be iterable, non-array objects must have a [Symbol.iterator]() method."); }
function _unsupportedIterableToArray(o, minLen) { if (!o) return; if (typeof o === "string") return _arrayLikeToArray(o, minLen); var n = Object.prototype.toString.call(o).slice(8, -1); if (n === "Object" && o.constructor) n = o.constructor.name; if (n === "Map" || n === "Set") return Array.from(o); if (n === "Arguments" || /^(?:Ui|I)nt(?:8|16|32)(?:Clamped)?Array$/.test(n)) return _arrayLikeToArray(o, minLen); }
function _arrayLikeToArray(arr, len) { if (len == null || len > arr.length) len = arr.length; for (var i = 0, arr2 = new Array(len); i < len; i++) arr2[i] = arr[i]; return arr2; }
function _iterableToArrayLimit(r, l) { var t = null == r ? null : "undefined" != typeof Symbol && r[Symbol.iterator] || r["@@iterator"]; if (null != t) { var e, n, i, u, a = [], f = !0, o = !1; try { if (i = (t = t.call(r)).next, 0 === l) { if (Object(t) !== t) return; f = !1; } else for (; !(f = (e = i.call(t)).done) && (a.push(e.value), a.length !== l); f = !0); } catch (r) { o = !0, n = r; } finally { try { if (!f && null != t.return && (u = t.return(), Object(u) !== u)) return; } finally { if (o) throw n; } } return a; } }
function _arrayWithHoles(arr) { if (Array.isArray(arr)) return arr; }
function ownKeys(e, r) { var t = Object.keys(e); if (Object.getOwnPropertySymbols) { var o = Object.getOwnPropertySymbols(e); r && (o = o.filter(function (r) { return Object.getOwnPropertyDescriptor(e, r).enumerable; })), t.push.apply(t, o); } return t; }
function _objectSpread(e) { for (var r = 1; r < arguments.length; r++) { var t = null != arguments[r] ? arguments[r] : {}; r % 2 ? ownKeys(Object(t), !0).forEach(function (r) { _defineProperty(e, r, t[r]); }) : Object.getOwnPropertyDescriptors ? Object.defineProperties(e, Object.getOwnPropertyDescriptors(t)) : ownKeys(Object(t)).forEach(function (r) { Object.defineProperty(e, r, Object.getOwnPropertyDescriptor(t, r)); }); } return e; }
function _defineProperty(obj, key, value) { key = _toPropertyKey(key); if (key in obj) { Object.defineProperty(obj, key, { value: value, enumerable: true, configurable: true, writable: true }); } else { obj[key] = value; } return obj; }
function _toPropertyKey(t) { var i = _toPrimitive(t, "string"); return "symbol" == _typeof(i) ? i : String(i); }
function _toPrimitive(t, r) { if ("object" != _typeof(t) || !t) return t; var e = t[Symbol.toPrimitive]; if (void 0 !== e) { var i = e.call(t, r || "default"); if ("object" != _typeof(i)) return i; throw new TypeError("@@toPrimitive must return a primitive value."); } return ("string" === r ? String : Number)(t); }
import { DeleteOutlined } from '@ant-design/icons';
import { Fetcher } from '@difizen/magent-core';
import { useFlowStore, useKnowledgeStore, useModelStore } from '@difizen/magent-flow';
import { BaseView, inject, prop, view, ViewOption, transient, useInject, ViewInstance } from '@difizen/mana-app';
import { Button } from 'antd';
// import yaml from 'js-yaml';
import { forwardRef, useEffect, useState } from 'react';
import { AgentManager } from "../../agent/agent-manager.js";
import { ModelSelector } from "../../model/model-selector/index.js";
import { KnowledgeModal, KnowledgeModalComponent } from "../agent-config/knowledge-modal/modal.js";
import { InitEdgeParser, InitNodeParser } from "./flow-utils.js";
import { FlowWithTabs } from "./flow-with-tabs/index.js";
import "./index.less";
import { jsx as _jsx } from "react/jsx-runtime";
import { Fragment as _Fragment } from "react/jsx-runtime";
import { jsxs as _jsxs } from "react/jsx-runtime";
var viewId = 'magent-agent-flow';
var AgentFlowComponent = /*#__PURE__*/forwardRef(function AgentConfigViewComponent(props, ref) {
var _useModelStore = useModelStore(),
setModelSelector = _useModelStore.setModelSelector;
var _useKnowledgeStore = useKnowledgeStore(),
setKnowledgeSelector = _useKnowledgeStore.setKnowledgeSelector;
var _useFlowStore = useFlowStore(),
setNode = _useFlowStore.setNode,
initFlow = _useFlowStore.initFlow;
var instance = useInject(ViewInstance);
useEffect(function () {
// 注册 flow 中模型选择
var Ele = function Ele(_ref) {
var _llmParam$find, _llmParam$find2, _llmParam$find3;
var nodeId = _ref.nodeId,
llmParam = _ref.llmParam;
var onChange = function onChange(val) {
if (!val) {
return;
}
setNode(nodeId, function (old) {
return _objectSpread(_objectSpread({}, old), {}, {
data: _objectSpread(_objectSpread({}, old.data), {}, {
config: _objectSpread(_objectSpread({}, old.data['config']), {}, {
inputs: _objectSpread(_objectSpread({}, old.data['config']['inputs']), {}, {
llm_param: [llmParam.find(function (p) {
return p.name === 'prompt';
}), {
name: 'id',
type: 'string',
value: {
type: 'value',
content: val.id
}
}, {
name: 'temperature',
type: 'string',
value: {
type: 'value',
content: val.temperature
}
}, {
name: 'model_name',
type: 'string',
value: {
type: 'value',
content: val.model_name[0]
}
}]
})
})
})
});
});
};
var value = {
id: (_llmParam$find = llmParam.find(function (p) {
return p.name === 'id';
})) === null || _llmParam$find === void 0 || (_llmParam$find = _llmParam$find.value) === null || _llmParam$find === void 0 ? void 0 : _llmParam$find.content,
temperature: Number((_llmParam$find2 = llmParam.find(function (p) {
return p.name === 'temperature';
})) === null || _llmParam$find2 === void 0 || (_llmParam$find2 = _llmParam$find2.value) === null || _llmParam$find2 === void 0 ? void 0 : _llmParam$find2.content),
nickname: '',
model_name: [(_llmParam$find3 = llmParam.find(function (p) {
return p.name === 'model_name';
})) === null || _llmParam$find3 === void 0 || (_llmParam$find3 = _llmParam$find3.value) === null || _llmParam$find3 === void 0 ? void 0 : _llmParam$find3.content]
};
return /*#__PURE__*/_jsx(_Fragment, {
children: /*#__PURE__*/_jsx(ModelSelector, {
value: value,
onChange: onChange
})
});
};
setModelSelector(Ele);
var Ele2 = function Ele2(_ref2) {
var _knowledgeParam$find, _knowledgeParam$find2;
var nodeId = _ref2.nodeId,
knowledgeParam = _ref2.knowledgeParam;
var _useState = useState(false),
_useState2 = _slicedToArray(_useState, 2),
knowledgeModal = _useState2[0],
setKnowledgeModal = _useState2[1];
var _onChange = function onChange(val) {
setNode(nodeId, function (old) {
return _objectSpread(_objectSpread({}, old), {}, {
data: _objectSpread(_objectSpread({}, old.data), {}, {
config: _objectSpread(_objectSpread({}, old.data['config']), {}, {
inputs: _objectSpread(_objectSpread({}, old.data['config']['inputs']), {}, {
knowledge_param: [{
name: 'top_k',
type: 'string',
value: {
type: 'value',
content: val.top_k
}
}, {
name: 'id',
type: 'string',
value: {
type: 'value',
content: val.id
}
}]
})
})
})
});
});
};
var value = {
knowledge: (((_knowledgeParam$find = knowledgeParam.find(function (p) {
return p.name === 'id';
})) === null || _knowledgeParam$find === void 0 || (_knowledgeParam$find = _knowledgeParam$find.value) === null || _knowledgeParam$find === void 0 ? void 0 : _knowledgeParam$find.content) || []).map(function (k) {
return {
id: k
};
}),
top_k: (_knowledgeParam$find2 = knowledgeParam.find(function (p) {
return p.name === 'top_k';
})) === null || _knowledgeParam$find2 === void 0 || (_knowledgeParam$find2 = _knowledgeParam$find2.value) === null || _knowledgeParam$find2 === void 0 ? void 0 : _knowledgeParam$find2.content
};
return /*#__PURE__*/_jsxs(_Fragment, {
children: [/*#__PURE__*/_jsx(KnowledgeModalComponent, {
visible: knowledgeModal,
close: function close() {
return setKnowledgeModal(false);
},
data: {
dataProvider: {
knowledge: _toConsumableArray(value.knowledge)
},
onChange: function onChange(knowledges) {
_onChange({
id: _toConsumableArray(knowledges.map(function (k) {
return k.id;
})),
top_k: value.top_k
});
}
},
modalItem: KnowledgeModal
}), /*#__PURE__*/_jsxs("div", {
children: [/*#__PURE__*/_jsx("div", {
children: value.knowledge.map(function (k) {
return /*#__PURE__*/_jsxs("div", {
style: {
width: '100%',
border: '1px solid #d9d9d9',
borderRadius: '4px',
marginBottom: '6px',
paddingLeft: '12px',
paddingRight: '12px',
display: 'flex',
justifyContent: 'space-between',
alignItems: 'center',
fontWeight: '500'
},
children: [/*#__PURE__*/_jsx("div", {
children: k.id
}), /*#__PURE__*/_jsx(Button, {
type: "link",
icon: /*#__PURE__*/_jsx(DeleteOutlined, {}),
onClick: function onClick() {
_onChange({
id: value.knowledge.filter(function (kn) {
return kn.id !== k.id;
}).map(function (kn) {
return kn.id;
})
});
}
})]
}, k.id);
})
}), /*#__PURE__*/_jsx(Button, {
style: {
width: '100%',
border: '1px dashed #d9d9d9',
padding: '6px'
// zIndex,
},
type: "link",
size: "small",
onClick: function onClick(e) {
e.stopPropagation();
setKnowledgeModal(true);
},
children: "\u6DFB\u52A0\u77E5\u8BC6\u5E93"
})]
})]
});
};
setKnowledgeSelector(Ele2);
}, [setKnowledgeSelector, setModelSelector, setNode]);
useEffect(function () {
instance.getGraphInfo().then(function (info) {
// const graph = yaml.load(mockGraph) as Graph;
var graph = info.graph || {
nodes: [],
edges: []
};
var nodes = graph.nodes.map(function (n) {
return InitNodeParser(n);
});
var edges = graph.edges.map(function (e) {
return InitEdgeParser(e);
});
// 获取 yaml 初始化 flow
initFlow({
nodes: _toConsumableArray(nodes),
edges: _toConsumableArray(edges)
});
return;
}).catch(console.error);
}, [initFlow, instance]);
return /*#__PURE__*/_jsx("div", {
ref: ref,
className: viewId,
children: /*#__PURE__*/_jsx(FlowWithTabs, {
toolbar: /*#__PURE__*/_jsx(_Fragment, {})
})
});
});
export var AgentFlowView = (_dec = transient(), _dec2 = view(viewId), _dec3 = inject(Fetcher), _dec4 = prop(), _dec5 = prop(), _dec6 = prop(), _dec(_class = _dec2(_class = (_class2 = /*#__PURE__*/function (_BaseView) {
_inherits(AgentFlowView, _BaseView);
var _super = _createSuper(AgentFlowView);
function AgentFlowView(option, agentManager) {
var _this;
_classCallCheck(this, AgentFlowView);
_this = _super.call(this);
_initializerDefineProperty(_this, "request", _descriptor, _assertThisInitialized(_this));
_this.agentId = void 0;
_this.view = AgentFlowComponent;
_initializerDefineProperty(_this, "agent", _descriptor2, _assertThisInitialized(_this));
_initializerDefineProperty(_this, "workflowId", _descriptor3, _assertThisInitialized(_this));
_initializerDefineProperty(_this, "workflow", _descriptor4, _assertThisInitialized(_this));
_this.agentManager = void 0;
_this.initAgent = function () {
var agentId = arguments.length > 0 && arguments[0] !== undefined ? arguments[0] : _this.agentId;
if (agentId) {
var agent = _this.agentManager.getOrCreate({
id: agentId
});
agent.fetchInfo();
_this.agent = agent;
return agent;
}
return undefined;
};
_this.getWorkflowInfo = /*#__PURE__*/function () {
var _ref3 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee(workflowId) {
var res;
return _regeneratorRuntime().wrap(function _callee$(_context) {
while (1) switch (_context.prev = _context.next) {
case 0:
_context.next = 2;
return _this.request.get("/api/v1/workflows/".concat(workflowId), {});
case 2:
res = _context.sent;
if (res.status === 200) {
_this.workflow = res.data;
}
return _context.abrupt("return", _this.workflow);
case 5:
case "end":
return _context.stop();
}
}, _callee);
}));
return function (_x) {
return _ref3.apply(this, arguments);
};
}();
_this.getGraphInfo = /*#__PURE__*/_asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee2() {
var planner, workflowId;
return _regeneratorRuntime().wrap(function _callee2$(_context2) {
while (1) switch (_context2.prev = _context2.next) {
case 0:
_context2.next = 2;
return _this.agent.ready;
case 2:
planner = _this.agent.planner;
workflowId = planner.workflow_id;
_this.workflowId = workflowId;
_context2.next = 7;
return _this.getWorkflowInfo(workflowId);
case 7:
return _context2.abrupt("return", _context2.sent);
case 8:
case "end":
return _context2.stop();
}
}, _callee2);
}));
_this.saveGraph = /*#__PURE__*/function () {
var _ref5 = _asyncToGenerator( /*#__PURE__*/_regeneratorRuntime().mark(function _callee3(graph) {
var res;
return _regeneratorRuntime().wrap(function _callee3$(_context3) {
while (1) switch (_context3.prev = _context3.next) {
case 0:
_context3.next = 2;
return _this.agent.ready;
case 2:
if (!(!_this.workflowId || !_this.workflow)) {
_context3.next = 4;
break;
}
return _context3.abrupt("return");
case 4:
_this.workflow.graph = graph;
_context3.next = 7;
return _this.request.put("/api/v1/workflows/".concat(_this.workflowId), _this.workflow);
case 7:
res = _context3.sent;
return _context3.abrupt("return", res);
case 9:
case "end":
return _context3.stop();
}
}, _callee3);
}));
return function (_x2) {
return _ref5.apply(this, arguments);
};
}();
_this.agentId = option.agentId;
_this.agentManager = agentManager;
_this.initAgent(option.agentId);
return _this;
}
AgentFlowView = inject(AgentManager)(AgentFlowView, undefined, 1) || AgentFlowView;
AgentFlowView = inject(ViewOption)(AgentFlowView, undefined, 0) || AgentFlowView;
_createClass(AgentFlowView, [{
key: "modelOptions",
get: function get() {
var _this$agent;
// TODO 大模型optios列表和对应存取值要怎么取？
return ((_this$agent = this.agent) === null || _this$agent === void 0 || (_this$agent = _this$agent.llm) === null || _this$agent === void 0 || (_this$agent = _this$agent.models) === null || _this$agent === void 0 ? void 0 : _this$agent.map(function (item) {
return {
label: item,
value: item
};
})) || [];
}
}]);
return AgentFlowView;
}(BaseView), (_descriptor = _applyDecoratedDescriptor(_class2.prototype, "request", [_dec3], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor2 = _applyDecoratedDescriptor(_class2.prototype, "agent", [_dec4], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor3 = _applyDecoratedDescriptor(_class2.prototype, "workflowId", [_dec5], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
}), _descriptor4 = _applyDecoratedDescriptor(_class2.prototype, "workflow", [_dec6], {
configurable: true,
enumerable: true,
writable: true,
initializer: null
})), _class2)) || _class) || _class);