@aigne/agent-library 1.13.0 → 1.13.2

package/CHANGELOG.md

@@ -5,6 +5,31 @@
 
 * add schema transform ([#35](https://github.com/AIGNE-io/aigne-framework/issues/35)) ([c7d9a2c](https://github.com/AIGNE-io/aigne-framework/commit/c7d9a2c9fcab8d384d4198db5ff6ba4603846cdf))
 
+## [1.13.2](https://github.com/AIGNE-io/aigne-framework/compare/agent-library-v1.13.1...agent-library-v1.13.2) (2025-06-19)
+
+
+### Bug Fixes
+
+* use `inputKey` instead of implicit $message for AIAgent ([#165](https://github.com/AIGNE-io/aigne-framework/issues/165)) ([8b6e589](https://github.com/AIGNE-io/aigne-framework/commit/8b6e5896bba8209fd2eecb0f5b9263618bffdaf8))
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/core bumped to 1.20.1
+    * @aigne/openai bumped to 0.3.2
+
+## [1.13.1](https://github.com/AIGNE-io/aigne-framework/compare/agent-library-v1.13.0...agent-library-v1.13.1) (2025-06-17)
+
+
+### Dependencies
+
+* The following workspace dependencies were updated
+  * dependencies
+    * @aigne/core bumped to 1.20.0
+    * @aigne/openai bumped to 0.3.1
+
 ## [1.13.0](https://github.com/AIGNE-io/aigne-framework/compare/agent-library-v1.12.6...agent-library-v1.13.0) (2025-06-16)
 
 

package/lib/cjs/data-mapper/agents/mapper.d.ts

@@ -1,11 +1,6 @@
 import { AIAgent } from "@aigne/core";
-declare const mapper: AIAgent<{
-    sourceData: string;
-    responseSchema: string;
-    sourceSchema?: string | undefined;
-    instruction?: string | undefined;
-    responseData?: string | undefined;
-    feedback?: string | undefined;
+declare const mapper: AIAgent<string, import("@aigne/core").Message & {
+    [x: string]: string;
 }, {
     jsonata: string;
     confidence: number;

package/lib/cjs/fs-memory/index.d.ts

@@ -36,7 +36,7 @@ export declare class FSMemory extends MemoryAgent {
      */
     constructor(options: FSMemoryOptions);
 }
-interface FSMemoryRetrieverOptions extends AIAgentOptions<FSMemoryRetrieverAgentInput, FSMemoryRetrieverAgentOutput> {
+interface FSMemoryRetrieverOptions extends AIAgentOptions<any, FSMemoryRetrieverAgentInput, FSMemoryRetrieverAgentOutput> {
     memoryFileName: string;
 }
 interface FSMemoryRetrieverAgentInput extends MemoryRetrieverInput {
@@ -47,7 +47,7 @@ interface FSMemoryRetrieverAgentOutput extends Message {
         content: string;
     }[];
 }
-interface FSMemoryRecorderOptions extends AIAgentOptions<FSMemoryRecorderAgentInput, FSMemoryRecorderAgentOutput> {
+interface FSMemoryRecorderOptions extends AIAgentOptions<any, FSMemoryRecorderAgentInput, FSMemoryRecorderAgentOutput> {
     memoryFileName: string;
 }
 type FSMemoryRecorderAgentInput = MemoryRecorderInput;

package/lib/cjs/orchestrator/index.d.ts

@@ -44,6 +44,7 @@ export interface OrchestratorAgentOptions<I extends Message = Message, O extends
      * Default: 5
      */
     tasksConcurrency?: number;
+    inputKey: string;
 }
 /**
  * Orchestrator Agent Class
@@ -74,6 +75,7 @@ export declare class OrchestratorAgent<I extends Message = Message, O extends Me
     constructor(options: OrchestratorAgentOptions<I, O>);
     private planner;
     private completer;
+    inputKey: string;
     /**
      * Maximum number of iterations
      * Prevents infinite execution loops

package/lib/cjs/orchestrator/index.js

@@ -68,6 +68,7 @@ class OrchestratorAgent extends core_1.Agent {
         super({ ...options });
         this.maxIterations = options.maxIterations;
         this.tasksConcurrency = options.tasksConcurrency;
+        this.inputKey = options.inputKey;
         this.planner = new core_1.AIAgent({
             name: "llm_orchestration_planner",
             instructions: orchestrator_prompts_js_1.FULL_PLAN_PROMPT_TEMPLATE,
@@ -81,6 +82,7 @@ class OrchestratorAgent extends core_1.Agent {
     }
     planner;
     completer;
+    inputKey;
     /**
      * Maximum number of iterations
      * Prevents infinite execution loops
@@ -109,8 +111,8 @@ class OrchestratorAgent extends core_1.Agent {
         const { model } = options.context;
         if (!model)
             throw new Error("model is required to run OrchestratorAgent");
-        const objective = (0, core_1.getMessage)(input);
-        if (!objective)
+        const objective = input[this.inputKey];
+        if (typeof objective !== "string")
             throw new Error("Objective is required to run OrchestratorAgent");
         const result = {
             objective,
@@ -152,7 +154,7 @@ class OrchestratorAgent extends core_1.Agent {
     async synthesizePlanResult(planResult, context) {
         return context.invoke(this.completer, {
             ...this.getFullPlanInput(planResult),
-            ...(0, core_1.createMessage)(orchestrator_prompts_js_1.SYNTHESIZE_PLAN_USER_PROMPT_TEMPLATE),
+            message: orchestrator_prompts_js_1.SYNTHESIZE_PLAN_USER_PROMPT_TEMPLATE,
         });
     }
     async executeStep(planResult, step, context) {
@@ -172,7 +174,7 @@ class OrchestratorAgent extends core_1.Agent {
             });
             let result;
             if (agent.isInvokable) {
-                result = getMessageOrJsonString(await context.invoke(agent, prompt));
+                result = getMessageOrJsonString(await context.invoke(agent, { message: prompt }));
             }
             else {
                 const executor = core_1.AIAgent.from({

package/lib/dts/data-mapper/agents/mapper.d.ts

@@ -1,11 +1,6 @@
 import { AIAgent } from "@aigne/core";
-declare const mapper: AIAgent<{
-    sourceData: string;
-    responseSchema: string;
-    sourceSchema?: string | undefined;
-    instruction?: string | undefined;
-    responseData?: string | undefined;
-    feedback?: string | undefined;
+declare const mapper: AIAgent<string, import("@aigne/core").Message & {
+    [x: string]: string;
 }, {
     jsonata: string;
     confidence: number;

package/lib/dts/fs-memory/index.d.ts

@@ -36,7 +36,7 @@ export declare class FSMemory extends MemoryAgent {
      */
     constructor(options: FSMemoryOptions);
 }
-interface FSMemoryRetrieverOptions extends AIAgentOptions<FSMemoryRetrieverAgentInput, FSMemoryRetrieverAgentOutput> {
+interface FSMemoryRetrieverOptions extends AIAgentOptions<any, FSMemoryRetrieverAgentInput, FSMemoryRetrieverAgentOutput> {
     memoryFileName: string;
 }
 interface FSMemoryRetrieverAgentInput extends MemoryRetrieverInput {
@@ -47,7 +47,7 @@ interface FSMemoryRetrieverAgentOutput extends Message {
         content: string;
     }[];
 }
-interface FSMemoryRecorderOptions extends AIAgentOptions<FSMemoryRecorderAgentInput, FSMemoryRecorderAgentOutput> {
+interface FSMemoryRecorderOptions extends AIAgentOptions<any, FSMemoryRecorderAgentInput, FSMemoryRecorderAgentOutput> {
     memoryFileName: string;
 }
 type FSMemoryRecorderAgentInput = MemoryRecorderInput;

package/lib/dts/orchestrator/index.d.ts

@@ -44,6 +44,7 @@ export interface OrchestratorAgentOptions<I extends Message = Message, O extends
      * Default: 5
      */
     tasksConcurrency?: number;
+    inputKey: string;
 }
 /**
  * Orchestrator Agent Class
@@ -74,6 +75,7 @@ export declare class OrchestratorAgent<I extends Message = Message, O extends Me
     constructor(options: OrchestratorAgentOptions<I, O>);
     private planner;
     private completer;
+    inputKey: string;
     /**
      * Maximum number of iterations
      * Prevents infinite execution loops

package/lib/esm/data-mapper/agents/mapper.d.ts

@@ -1,11 +1,6 @@
 import { AIAgent } from "@aigne/core";
-declare const mapper: AIAgent<{
-    sourceData: string;
-    responseSchema: string;
-    sourceSchema?: string | undefined;
-    instruction?: string | undefined;
-    responseData?: string | undefined;
-    feedback?: string | undefined;
+declare const mapper: AIAgent<string, import("@aigne/core").Message & {
+    [x: string]: string;
 }, {
     jsonata: string;
     confidence: number;

package/lib/esm/fs-memory/index.d.ts

@@ -36,7 +36,7 @@ export declare class FSMemory extends MemoryAgent {
      */
     constructor(options: FSMemoryOptions);
 }
-interface FSMemoryRetrieverOptions extends AIAgentOptions<FSMemoryRetrieverAgentInput, FSMemoryRetrieverAgentOutput> {
+interface FSMemoryRetrieverOptions extends AIAgentOptions<any, FSMemoryRetrieverAgentInput, FSMemoryRetrieverAgentOutput> {
     memoryFileName: string;
 }
 interface FSMemoryRetrieverAgentInput extends MemoryRetrieverInput {
@@ -47,7 +47,7 @@ interface FSMemoryRetrieverAgentOutput extends Message {
         content: string;
     }[];
 }
-interface FSMemoryRecorderOptions extends AIAgentOptions<FSMemoryRecorderAgentInput, FSMemoryRecorderAgentOutput> {
+interface FSMemoryRecorderOptions extends AIAgentOptions<any, FSMemoryRecorderAgentInput, FSMemoryRecorderAgentOutput> {
     memoryFileName: string;
 }
 type FSMemoryRecorderAgentInput = MemoryRecorderInput;

package/lib/esm/orchestrator/index.d.ts

@@ -44,6 +44,7 @@ export interface OrchestratorAgentOptions<I extends Message = Message, O extends
      * Default: 5
      */
     tasksConcurrency?: number;
+    inputKey: string;
 }
 /**
  * Orchestrator Agent Class
@@ -74,6 +75,7 @@ export declare class OrchestratorAgent<I extends Message = Message, O extends Me
     constructor(options: OrchestratorAgentOptions<I, O>);
     private planner;
     private completer;
+    inputKey: string;
     /**
      * Maximum number of iterations
      * Prevents infinite execution loops

package/lib/esm/orchestrator/index.js

@@ -1,4 +1,4 @@
-import { AIAgent, Agent, PromptTemplate, createMessage, getMessage, } from "@aigne/core";
+import { AIAgent, Agent, PromptTemplate, } from "@aigne/core";
 import { checkArguments } from "@aigne/core/utils/type-utils.js";
 import fastq from "fastq";
 import { z } from "zod";
@@ -48,6 +48,7 @@ export class OrchestratorAgent extends Agent {
         super({ ...options });
         this.maxIterations = options.maxIterations;
         this.tasksConcurrency = options.tasksConcurrency;
+        this.inputKey = options.inputKey;
         this.planner = new AIAgent({
             name: "llm_orchestration_planner",
             instructions: FULL_PLAN_PROMPT_TEMPLATE,
@@ -61,6 +62,7 @@ export class OrchestratorAgent extends Agent {
     }
     planner;
     completer;
+    inputKey;
     /**
      * Maximum number of iterations
      * Prevents infinite execution loops
@@ -89,8 +91,8 @@ export class OrchestratorAgent extends Agent {
         const { model } = options.context;
         if (!model)
             throw new Error("model is required to run OrchestratorAgent");
-        const objective = getMessage(input);
-        if (!objective)
+        const objective = input[this.inputKey];
+        if (typeof objective !== "string")
             throw new Error("Objective is required to run OrchestratorAgent");
         const result = {
             objective,
@@ -132,7 +134,7 @@ export class OrchestratorAgent extends Agent {
     async synthesizePlanResult(planResult, context) {
         return context.invoke(this.completer, {
             ...this.getFullPlanInput(planResult),
-            ...createMessage(SYNTHESIZE_PLAN_USER_PROMPT_TEMPLATE),
+            message: SYNTHESIZE_PLAN_USER_PROMPT_TEMPLATE,
         });
     }
     async executeStep(planResult, step, context) {
@@ -152,7 +154,7 @@ export class OrchestratorAgent extends Agent {
             });
             let result;
             if (agent.isInvokable) {
-                result = getMessageOrJsonString(await context.invoke(agent, prompt));
+                result = getMessageOrJsonString(await context.invoke(agent, { message: prompt }));
             }
             else {
                 const executor = AIAgent.from({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/agent-library",
-  "version": "1.13.0",
+  "version": "1.13.2",
   "description": "Collection of agent libraries for AIGNE framework",
   "publishConfig": {
     "access": "public"
@@ -47,9 +47,9 @@
     "uuid": "^11.1.0",
     "yaml": "^2.7.1",
     "zod": "^3.24.4",
-    "@aigne/openai": "^0.3.0",
+    "@aigne/core": "^1.20.1",
     "@aigne/sqlite": "^0.1.1",
-    "@aigne/core": "^1.19.0"
+    "@aigne/openai": "^0.3.2"
   },
   "devDependencies": {
     "@types/bun": "^1.2.12",