@aigne/core 0.4.201-4 → 0.4.201-6

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

@@ -15,11 +15,9 @@ var LLMDecisionAgent_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.LLMDecisionAgent = void 0;
 exports.createLLMDecisionAgentDefinition = createLLMDecisionAgentDefinition;
-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");
@@ -83,27 +81,8 @@ let LLMDecisionAgent = LLMDecisionAgent_1 = class LLMDecisionAgent extends runna
         const caseToCall = cases.find((i) => i.name === functionNameToCall);
         if (!caseToCall)
             throw new Error('Case not found');
-        // NOTE: 将 input 转换为 variables，其中 key 为 inputId，value 为 input 的值
-        const variables = Object.fromEntries(utils_1.OrderedRecord.map(this.definition.inputs, (i) => {
-            const value = input[i.name || i.id];
-            if ((0, lodash_1.isNil)(value))
-                return null;
-            return [i.id, value];
-        }).filter(utils_1.isNonNullable));
-        const inputForCase = Object.fromEntries(Object.entries(caseToCall.input ?? {})
-            .map(([inputId, { from, fromVariableId, fromVariablePropPath }]) => {
-            const targetInput = utils_1.OrderedRecord.find(caseToCall.runnable.definition.inputs, (i) => i.id === inputId);
-            if (!targetInput?.name)
-                return null;
-            if (from !== 'variable' || !fromVariableId)
-                return null;
-            const v = variables[fromVariableId];
-            const value = fromVariablePropPath?.length ? (0, lodash_1.get)(v, fromVariablePropPath) : v;
-            return [targetInput.name, value];
-        })
-            .filter(utils_1.isNonNullable));
         // TODO: check result structure and omit undefined values
-        return (await caseToCall.runnable.run(inputForCase, options));
+        return (await caseToCall.runnable.run(input, options));
     }
 };
 exports.LLMDecisionAgent = LLMDecisionAgent;
@@ -115,7 +94,6 @@ exports.LLMDecisionAgent = LLMDecisionAgent = LLMDecisionAgent_1 = __decorate([
     __metadata("design:paramtypes", [Object, llm_model_1.LLMModel, Object])
 ], LLMDecisionAgent);
 function createLLMDecisionAgentDefinition(options) {
-    const inputs = (0, data_type_schema_1.SchemaToDataType)(options.inputs);
     const messages = utils_1.OrderedRecord.fromArray([
         {
             id: (0, nanoid_1.nanoid)(),
@@ -128,34 +106,13 @@ function createLLMDecisionAgentDefinition(options) {
         name: c.name || c.runnable.name,
         description: c.description,
         runnable: { id: c.runnable.id },
-        // TODO: pass input from decision to case runnable
-        input: Object.fromEntries(utils_1.OrderedRecord.map(c.runnable.definition.inputs, (inputOfCase) => {
-            const i = c.input?.[inputOfCase.name || inputOfCase.id];
-            if (!i) {
-                if (inputOfCase.required) {
-                    throw new Error(`Input ${inputOfCase.name || inputOfCase.id} for case ${c.runnable.name || c.runnable.id} is required`);
-                }
-                // ignore optional input
-                return null;
-            }
-            const inputFromDecision = utils_1.OrderedRecord.find(inputs, (input) => input.name === i.fromVariable);
-            if (!inputFromDecision)
-                throw new Error(`Input ${i.fromVariable} not found`);
-            return [
-                inputOfCase.id,
-                {
-                    from: 'variable',
-                    fromVariableId: inputFromDecision.id,
-                    fromVariablePropPath: i.fromVariablePropPath,
-                },
-            ];
-        }).filter(utils_1.isNonNullable)),
     })));
     return {
         id: options.id || options.name || (0, nanoid_1.nanoid)(),
         name: options.name,
         type: 'llm_decision_agent',
-        inputs,
+        // TODO: decision agent inputs should be the intersection of all case inputs
+        inputs: utils_1.OrderedRecord.fromArray([]),
         // TODO: decision agent outputs should be the union of all case outputs
         outputs: utils_1.OrderedRecord.fromArray([]),
         messages,