@aigne/core 0.4.201-6 → 0.4.201-7

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

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

package/lib/cjs/function-agent.js

@@ -62,8 +62,8 @@ function createFunctionAgentDefinition(options) {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),
         name: options.name,
         type: 'function_agent',
-        inputs: (0, data_type_schema_1.SchemaToDataType)(options.inputs),
-        outputs: (0, data_type_schema_1.SchemaToDataType)(options.outputs),
+        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/llm-agent.js

@@ -117,8 +117,8 @@ function createLLMAgentDefinition(options) {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),
         name: options.name,
         type: 'llm_agent',
-        inputs: (0, data_type_schema_1.SchemaToDataType)(options.inputs),
-        outputs: (0, data_type_schema_1.SchemaToDataType)(options.outputs),
+        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/local-function-agent.js

@@ -58,8 +58,8 @@ exports.LocalFunctionAgent = LocalFunctionAgent = LocalFunctionAgent_1 = __decor
     __metadata("design:paramtypes", [Object, Object])
 ], LocalFunctionAgent);
 function createLocalFunctionAgentDefinition(options) {
-    const inputs = (0, data_type_schema_1.SchemaToDataType)(options.inputs);
-    const outputs = (0, data_type_schema_1.SchemaToDataType)(options.outputs);
+    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

@@ -146,7 +146,7 @@ exports.PipelineAgent = PipelineAgent = PipelineAgent_1 = __decorate([
     __metadata("design:paramtypes", [Object, Object])
 ], PipelineAgent);
 function createPipelineAgentDefinition(options) {
-    const inputs = (0, data_type_schema_1.SchemaToDataType)(options.inputs);
+    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, {
@@ -177,7 +177,7 @@ function createPipelineAgentDefinition(options) {
             }).filter(is_non_nullable_1.isNonNullable)),
         });
     }
-    const outputs = ordered_map_1.OrderedRecord.fromArray(ordered_map_1.OrderedRecord.map((0, data_type_schema_1.SchemaToDataType)(options.outputs), (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);