@aigne/core 1.63.0-beta.1 → 1.63.0-beta.2

package/CHANGELOG.md

@@ -1,5 +1,12 @@
 # Changelog
 
+## [1.63.0-beta.2](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta.1...core-v1.63.0-beta.2) (2025-10-09)
+
+
+### Bug Fixes
+
+* input schema of AI agent should includes input key and input file key ([#600](https://github.com/AIGNE-io/aigne-framework/issues/600)) ([b4ca076](https://github.com/AIGNE-io/aigne-framework/commit/b4ca076d6b4a1a1ecb8d4ebb008abd0d7561aadd))
+
 ## [1.63.0-beta.1](https://github.com/AIGNE-io/aigne-framework/compare/core-v1.63.0-beta...core-v1.63.0-beta.1) (2025-10-09)
 
 

package/lib/cjs/agents/ai-agent.d.ts

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -311,6 +311,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

package/lib/cjs/agents/ai-agent.js

@@ -12,6 +12,7 @@ const template_js_1 = require("../prompt/template.js");
 const structured_stream_extractor_js_1 = require("../utils/structured-stream-extractor.js");
 const type_utils_js_1 = require("../utils/type-utils.js");
 const agent_js_1 = require("./agent.js");
+const model_js_1 = require("./model.js");
 const types_js_1 = require("./types.js");
 exports.DEFAULT_OUTPUT_KEY = "message";
 exports.DEFAULT_OUTPUT_FILE_KEY = "files";
@@ -242,6 +243,20 @@ class AIAgent extends agent_js_1.Agent {
      * which outputs structured data in YAML format within <metadata> tags.
      */
     customStructuredStreamInstructions;
+    get inputSchema() {
+        let schema = super.inputSchema;
+        if (this.inputKey) {
+            schema = schema.extend({
+                [this.inputKey]: zod_1.z.string().nullish(),
+            });
+        }
+        if (this.inputFileKey) {
+            schema = schema.extend({
+                [this.inputFileKey]: model_js_1.fileUnionContentsSchema.nullish(),
+            });
+        }
+        return schema;
+    }
     /**
      * Process an input message and generate a response
      *

package/lib/dts/agents/ai-agent.d.ts

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -311,6 +311,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

package/lib/esm/agents/ai-agent.d.ts

@@ -3,7 +3,7 @@ import { PromptBuilder } from "../prompt/prompt-builder.js";
 import { Agent, type AgentInvokeOptions, type AgentOptions, type AgentProcessAsyncGenerator, type AgentProcessResult, type Message } from "./agent.js";
 import type { ChatModel, ChatModelInput } from "./chat-model.js";
 import type { GuideRailAgentOutput } from "./guide-rail-agent.js";
-import type { FileType } from "./model.js";
+import { type FileType } from "./model.js";
 export declare const DEFAULT_OUTPUT_KEY = "message";
 export declare const DEFAULT_OUTPUT_FILE_KEY = "files";
 /**
@@ -311,6 +311,7 @@ export declare class AIAgent<I extends Message = any, O extends Message = any> e
         metadataEnd: string;
         parse: (raw: string) => object;
     };
+    get inputSchema(): ZodType<I>;
     /**
      * Process an input message and generate a response
      *

package/lib/esm/agents/ai-agent.js

@@ -6,6 +6,7 @@ import { AgentMessageTemplate, ToolMessageTemplate } from "../prompt/template.js
 import { ExtractMetadataTransform } from "../utils/structured-stream-extractor.js";
 import { checkArguments, isEmpty } from "../utils/type-utils.js";
 import { Agent, agentOptionsSchema, isAgentResponseDelta, } from "./agent.js";
+import { fileUnionContentsSchema } from "./model.js";
 import { isTransferAgentOutput } from "./types.js";
 export const DEFAULT_OUTPUT_KEY = "message";
 export const DEFAULT_OUTPUT_FILE_KEY = "files";
@@ -236,6 +237,20 @@ export class AIAgent extends Agent {
      * which outputs structured data in YAML format within <metadata> tags.
      */
     customStructuredStreamInstructions;
+    get inputSchema() {
+        let schema = super.inputSchema;
+        if (this.inputKey) {
+            schema = schema.extend({
+                [this.inputKey]: z.string().nullish(),
+            });
+        }
+        if (this.inputFileKey) {
+            schema = schema.extend({
+                [this.inputFileKey]: fileUnionContentsSchema.nullish(),
+            });
+        }
+        return schema;
+    }
     /**
      * Process an input message and generate a response
      *

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aigne/core",
-  "version": "1.63.0-beta.1",
+  "version": "1.63.0-beta.2",
   "description": "The functional core of agentic AI",
   "publishConfig": {
     "access": "public"
@@ -93,8 +93,8 @@
     "zod-from-json-schema": "^0.0.5",
     "zod-to-json-schema": "^3.24.6",
     "@aigne/afs": "^1.0.0",
-    "@aigne/observability-api": "^0.11.2-beta",
-    "@aigne/platform-helpers": "^0.6.3"
+    "@aigne/platform-helpers": "^0.6.3",
+    "@aigne/observability-api": "^0.11.2-beta"
   },
   "devDependencies": {
     "@types/bun": "^1.2.22",