@aigne/core 0.4.201-2 → 0.4.201-4

package/lib/cjs/data-type-schema.js

@@ -0,0 +1,51 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.SchemaToDataType = SchemaToDataType;
+const nanoid_1 = require("nanoid");
+const utils_1 = require("./utils");
+function SchemaToDataType(dataType) {
+    return utils_1.OrderedRecord.fromArray(Object.entries(dataType).map(([name, schema]) => {
+        switch (schema.type) {
+            case 'string':
+                return {
+                    id: (0, nanoid_1.nanoid)(),
+                    name,
+                    type: 'string',
+                    required: schema.required,
+                };
+            case 'number':
+                return {
+                    id: (0, nanoid_1.nanoid)(),
+                    name,
+                    type: 'number',
+                    required: schema.required,
+                };
+            case 'boolean':
+                return {
+                    id: (0, nanoid_1.nanoid)(),
+                    name,
+                    type: 'boolean',
+                    required: schema.required,
+                };
+            case 'object':
+                return {
+                    id: (0, nanoid_1.nanoid)(),
+                    name,
+                    type: 'object',
+                    required: schema.required,
+                    properties: SchemaToDataType(schema.properties),
+                };
+            case 'array':
+                return {
+                    id: (0, nanoid_1.nanoid)(),
+                    name,
+                    type: 'array',
+                    required: schema.required,
+                    items: SchemaToDataType({ items: schema.items })['items'],
+                };
+            default: {
+                throw new Error(`Unknown data type: ${schema.type}`);
+            }
+        }
+    }));
+}

package/lib/cjs/function-agent.js

@@ -18,6 +18,7 @@ exports.createFunctionAgentDefinition = createFunctionAgentDefinition;
 const nanoid_1 = require("nanoid");
 const tsyringe_1 = require("tsyringe");
 const constants_1 = require("./constants");
+const data_type_schema_1 = require("./data-type-schema");
 const function_runner_1 = require("./function-runner");
 const runnable_1 = require("./runnable");
 const utils_1 = require("./utils");
@@ -61,16 +62,8 @@ function createFunctionAgentDefinition(options) {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),
         name: options.name,
         type: 'function_agent',
-        inputs: utils_1.OrderedRecord.fromArray(Object.entries(options.inputs).map(([name, { ...dataType }]) => ({
-            ...dataType,
-            id: (0, nanoid_1.nanoid)(),
-            name,
-        }))),
-        outputs: utils_1.OrderedRecord.fromArray(Object.entries(options.outputs).map(([name, { ...dataType }]) => ({
-            ...dataType,
-            id: (0, nanoid_1.nanoid)(),
-            name,
-        }))),
+        inputs: (0, data_type_schema_1.SchemaToDataType)(options.inputs),
+        outputs: (0, data_type_schema_1.SchemaToDataType)(options.outputs),
         language: options.language,
         code: options.code,
     };

package/lib/cjs/index.js

@@ -17,6 +17,7 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./utils"), exports);
 __exportStar(require("./constants"), exports);
 __exportStar(require("./data-type"), exports);
+__exportStar(require("./data-type-schema"), exports);
 __exportStar(require("./context"), exports);
 __exportStar(require("./runnable"), exports);
 __exportStar(require("./pipeline-agent"), exports);

package/lib/cjs/llm-agent.js

@@ -19,6 +19,7 @@ const lodash_1 = require("lodash");
 const nanoid_1 = require("nanoid");
 const tsyringe_1 = require("tsyringe");
 const constants_1 = require("./constants");
+const data_type_schema_1 = require("./data-type-schema");
 const llm_model_1 = require("./llm-model");
 const runnable_1 = require("./runnable");
 const utils_1 = require("./utils");
@@ -116,16 +117,8 @@ function createLLMAgentDefinition(options) {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),
         name: options.name,
         type: 'llm_agent',
-        inputs: ordered_map_1.OrderedRecord.fromArray(Object.entries(options.inputs).map(([name, { ...dataType }]) => ({
-            ...dataType,
-            id: (0, nanoid_1.nanoid)(),
-            name,
-        }))),
-        outputs: ordered_map_1.OrderedRecord.fromArray(Object.entries(options.outputs).map(([name, { ...dataType }]) => ({
-            ...dataType,
-            id: (0, nanoid_1.nanoid)(),
-            name,
-        }))),
+        inputs: (0, data_type_schema_1.SchemaToDataType)(options.inputs),
+        outputs: (0, data_type_schema_1.SchemaToDataType)(options.outputs),
         modelOptions: options.modelOptions,
         messages: ordered_map_1.OrderedRecord.fromArray(options.messages?.map((i) => ({
             id: (0, nanoid_1.nanoid)(),

package/lib/cjs/llm-decision-agent.js

@@ -19,6 +19,7 @@ const lodash_1 = require("lodash");
 const nanoid_1 = require("nanoid");
 const tsyringe_1 = require("tsyringe");
 const constants_1 = require("./constants");
+const data_type_schema_1 = require("./data-type-schema");
 const llm_model_1 = require("./llm-model");
 const runnable_1 = require("./runnable");
 const utils_1 = require("./utils");
@@ -114,11 +115,7 @@ exports.LLMDecisionAgent = LLMDecisionAgent = LLMDecisionAgent_1 = __decorate([
     __metadata("design:paramtypes", [Object, llm_model_1.LLMModel, Object])
 ], LLMDecisionAgent);
 function createLLMDecisionAgentDefinition(options) {
-    const inputs = utils_1.OrderedRecord.fromArray(Object.entries(options.inputs).map(([name, { ...dataType }]) => ({
-        ...dataType,
-        id: (0, nanoid_1.nanoid)(),
-        name: name,
-    })));
+    const inputs = (0, data_type_schema_1.SchemaToDataType)(options.inputs);
     const messages = utils_1.OrderedRecord.fromArray([
         {
             id: (0, nanoid_1.nanoid)(),

package/lib/cjs/local-function-agent.js

@@ -18,6 +18,7 @@ exports.createLocalFunctionAgentDefinition = createLocalFunctionAgentDefinition;
 const nanoid_1 = require("nanoid");
 const tsyringe_1 = require("tsyringe");
 const constants_1 = require("./constants");
+const data_type_schema_1 = require("./data-type-schema");
 const runnable_1 = require("./runnable");
 const utils_1 = require("./utils");
 let LocalFunctionAgent = LocalFunctionAgent_1 = class LocalFunctionAgent extends runnable_1.Runnable {
@@ -57,16 +58,8 @@ exports.LocalFunctionAgent = LocalFunctionAgent = LocalFunctionAgent_1 = __decor
     __metadata("design:paramtypes", [Object, Object])
 ], LocalFunctionAgent);
 function createLocalFunctionAgentDefinition(options) {
-    const inputs = utils_1.OrderedRecord.fromArray(Object.entries(options.inputs).map(([name, i]) => ({
-        ...i,
-        id: (0, nanoid_1.nanoid)(),
-        name,
-    })));
-    const outputs = utils_1.OrderedRecord.fromArray(Object.entries(options.outputs).map(([name, o]) => ({
-        ...o,
-        id: (0, nanoid_1.nanoid)(),
-        name,
-    })));
+    const inputs = (0, data_type_schema_1.SchemaToDataType)(options.inputs);
+    const outputs = (0, data_type_schema_1.SchemaToDataType)(options.outputs);
     return {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),
         name: options.name,

package/lib/cjs/pipeline-agent.js

@@ -22,6 +22,7 @@ const lodash_1 = require("lodash");
 const nanoid_1 = require("nanoid");
 const tsyringe_1 = require("tsyringe");
 const constants_1 = require("./constants");
+const data_type_schema_1 = require("./data-type-schema");
 const logger_1 = __importDefault(require("./logger"));
 const runnable_1 = require("./runnable");
 const utils_1 = require("./utils");
@@ -145,11 +146,7 @@ exports.PipelineAgent = PipelineAgent = PipelineAgent_1 = __decorate([
     __metadata("design:paramtypes", [Object, Object])
 ], PipelineAgent);
 function createPipelineAgentDefinition(options) {
-    const inputs = ordered_map_1.OrderedRecord.fromArray(Object.entries(options.inputs).map(([name, { ...dataType }]) => ({
-        ...dataType,
-        id: (0, nanoid_1.nanoid)(),
-        name,
-    })));
+    const inputs = (0, data_type_schema_1.SchemaToDataType)(options.inputs);
     const processes = ordered_map_1.OrderedRecord.fromArray([]);
     for (const p of options.processes || []) {
         ordered_map_1.OrderedRecord.push(processes, {
@@ -180,18 +177,13 @@ function createPipelineAgentDefinition(options) {
             }).filter(is_non_nullable_1.isNonNullable)),
         });
     }
-    const outputs = ordered_map_1.OrderedRecord.fromArray(Object.entries(options.outputs).map(([name, { fromVariable, ...output }]) => {
+    const outputs = ordered_map_1.OrderedRecord.fromArray(ordered_map_1.OrderedRecord.map((0, data_type_schema_1.SchemaToDataType)(options.outputs), (output) => {
+        const { fromVariable, fromVariablePropPath } = options.outputs[output.name];
         const from = ordered_map_1.OrderedRecord.find(inputs, (i) => i.name === fromVariable) ||
             ordered_map_1.OrderedRecord.find(processes, (p) => p.name === fromVariable);
         if (!from)
             throw new Error(`Output ${name} not found in inputs or processes`);
-        return {
-            ...output,
-            id: (0, nanoid_1.nanoid)(),
-            name,
-            from: 'variable',
-            fromVariableId: from.id,
-        };
+        return { ...output, from: 'variable', fromVariableId: from.id, fromVariablePropPath };
     }));
     return {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),