@aigne/core 0.0.1

package/lib/cjs/executor/agent.js

@@ -0,0 +1,10 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AgentExecutor = void 0;
+const base_1 = require("./base");
+class AgentExecutor extends base_1.AgentExecutorBase {
+    async process() {
+        // ignore
+    }
+}
+exports.AgentExecutor = AgentExecutor;

package/lib/cjs/executor/aigc.js

@@ -0,0 +1,28 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AIGCAgentExecutor = void 0;
+const base_1 = require("./base");
+class AIGCAgentExecutor extends base_1.AgentExecutorBase {
+    async process() {
+        const { agent } = this;
+        if (!agent.prompt?.length)
+            throw new Error('Prompt cannot be empty');
+        const prompt = await this.renderMessage(agent.prompt
+            .split('\n')
+            .filter((i) => !i.startsWith('//'))
+            .join('\n'));
+        const { data } = await this.context.callAIImage({
+            assistant: agent,
+            input: {
+                prompt,
+                n: agent.n,
+                model: agent.model,
+                quality: agent.quality,
+                size: agent.size,
+                style: agent.style,
+            },
+        });
+        return { $images: data };
+    }
+}
+exports.AIGCAgentExecutor = AIGCAgentExecutor;

package/lib/cjs/executor/api.js

@@ -0,0 +1,64 @@
+"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 (mod) {
+    if (mod && mod.__esModule) return mod;
+    var result = {};
+    if (mod != null) for (var k in mod) if (k !== "default" && Object.prototype.hasOwnProperty.call(mod, k)) __createBinding(result, mod, k);
+    __setModuleDefault(result, mod);
+    return result;
+};
+var __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.APIAgentExecutor = void 0;
+const axios_1 = __importStar(require("axios"));
+const js_cookie_1 = __importDefault(require("js-cookie"));
+const lodash_1 = require("lodash");
+const base_1 = require("./base");
+class APIAgentExecutor extends base_1.AgentExecutorBase {
+    async process({ inputs }) {
+        const { agent } = this;
+        if (!agent.requestUrl)
+            throw new Error(`Assistant ${agent.id}'s url is empty`);
+        const args = Object.fromEntries(await Promise.all((agent.parameters ?? [])
+            .filter((i) => !!i.key && !i.hidden)
+            .map(async (i) => [i.key, inputs?.[i.key] || i.defaultValue])));
+        const method = agent.requestMethod || 'GET';
+        const isGet = method === 'get';
+        try {
+            const response = await (0, axios_1.default)({
+                url: agent.requestUrl,
+                method,
+                params: isGet ? args : undefined,
+                data: isGet ? undefined : args,
+                headers: {
+                    'x-csrf-token': js_cookie_1.default.get('x-csrf-token'),
+                },
+            });
+            return response.data;
+        }
+        catch (e) {
+            if ((0, axios_1.isAxiosError)(e)) {
+                Object.assign(e, (0, lodash_1.pick)(e.response, 'status', 'statusText', 'data'));
+            }
+            throw e;
+        }
+    }
+}
+exports.APIAgentExecutor = APIAgentExecutor;