@aigne/core 1.73.0-beta → 1.74.0-beta

package/dist/prompt/prompt-builder.cjs

@@ -0,0 +1,291 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const require_utils_type_utils = require('../utils/type-utils.cjs');
+const require_utils_json_schema = require('../utils/json-schema.cjs');
+const require_agent = require('../agents/agent.cjs');
+const require_model = require('../agents/model.cjs');
+const require_agent_session = require('./agent-session.cjs');
+const require_index = require('./skills/afs/index.cjs');
+const require_index$1 = require('./context/index.cjs');
+const require_memory_message_template = require('./prompts/memory-message-template.cjs');
+const require_structured_stream_instructions = require('./prompts/structured-stream-instructions.cjs');
+const require_template = require('./template.cjs');
+const require_ai_agent = require('../agents/ai-agent.cjs');
+let _aigne_utils_nodejs = require("@aigne/utils/nodejs");
+let yaml = require("yaml");
+let zod = require("zod");
+let zod_to_json_schema = require("zod-to-json-schema");
+let _aigne_afs_utils_zod = require("@aigne/afs/utils/zod");
+let _aigne_uuid = require("@aigne/uuid");
+
+//#region src/prompt/prompt-builder.ts
+var PromptBuilder = class PromptBuilder {
+	static from(instructions, { workingDir } = {}) {
+		if (typeof instructions === "string") return new PromptBuilder({
+			instructions,
+			workingDir
+		});
+		if (isFromPromptResult(instructions)) return PromptBuilder.fromMCPPromptResult(instructions);
+		if (isFromPath(instructions)) return PromptBuilder.fromFile(instructions.path, { workingDir });
+		throw new Error(`Invalid instructions ${instructions}`);
+	}
+	static fromFile(path, { workingDir }) {
+		const text = _aigne_utils_nodejs.nodejs.fsSync.readFileSync(path, "utf-8");
+		return PromptBuilder.from(text, { workingDir: workingDir || _aigne_utils_nodejs.nodejs.path.dirname(path) });
+	}
+	static fromMCPPromptResult(result) {
+		return new PromptBuilder({ instructions: require_template.ChatMessagesTemplate.from(result.messages.map((i) => {
+			let content;
+			if (i.content.type === "text") content = i.content.text;
+			else if (i.content.type === "resource") {
+				const { resource } = i.content;
+				if ("text" in resource && typeof resource.text === "string") content = resource.text;
+				else if ("blob" in resource && typeof resource.blob === "string") content = [{
+					type: "url",
+					url: resource.blob
+				}];
+			} else if (i.content.type === "image") content = [{
+				type: "url",
+				url: i.content.data
+			}];
+			if (!content) throw new Error(`Unsupported content type ${i.content.type}`);
+			if (i.role === "user") return require_template.UserMessageTemplate.from(content);
+			if (i.role === "assistant") return require_template.AgentMessageTemplate.from(content);
+			throw new Error(`Unsupported role ${i.role}`);
+		})) });
+	}
+	constructor(options) {
+		this.instructions = options?.instructions;
+		this.workingDir = options?.workingDir;
+	}
+	instructions;
+	workingDir;
+	copy() {
+		return new PromptBuilder({
+			instructions: typeof this.instructions === "string" ? this.instructions : this.instructions?.copy(),
+			workingDir: this.workingDir
+		});
+	}
+	async build(options) {
+		let { userId, sessionId } = options.context?.userContext || {};
+		const agentId = options.agent?.name;
+		const afs = options.agent?.afs;
+		sessionId ||= (0, _aigne_uuid.v7)();
+		const session = new require_agent_session.AgentSession({
+			agentId,
+			userId,
+			sessionId,
+			afs,
+			...options.agent?.session
+		});
+		const { systemMessage, userMessage } = await this.buildMessages(options);
+		if (systemMessage) await session.setSystemMessages(systemMessage);
+		return {
+			userMessage,
+			responseFormat: options.agent?.structuredStreamMode ? void 0 : this.buildResponseFormat(options),
+			outputFileType: options.agent?.outputFileType,
+			...await this.buildTools(options),
+			session
+		};
+	}
+	async buildPrompt(options) {
+		const messages = await (typeof this.instructions === "string" ? require_template.ChatMessagesTemplate.from([require_template.SystemMessageTemplate.from(this.instructions)]) : this.instructions)?.format(this.getTemplateVariables(options), { workingDir: this.workingDir }) ?? [];
+		const inputFileKey = options.inputFileKey;
+		const files = require_utils_type_utils.flat(inputFileKey ? require_utils_type_utils.checkArguments("Check input files", (0, _aigne_afs_utils_zod.optionalize)(require_model.fileUnionContentsSchema), options.input?.[inputFileKey]) : null);
+		return {
+			prompt: messages.map((i) => i.content).join("\n"),
+			image: files.length ? files : void 0
+		};
+	}
+	getTemplateVariables(options) {
+		return require_index$1.createPromptBuilderContext(options);
+	}
+	async buildMessages(options) {
+		const { input } = options;
+		const inputKey = options.agent?.inputKey;
+		const message = inputKey && typeof input?.[inputKey] === "string" ? input[inputKey] : void 0;
+		const [systemMessages, userMessages] = require_utils_type_utils.partition(await (typeof this.instructions === "string" ? require_template.ChatMessagesTemplate.from([require_template.SystemMessageTemplate.from(this.instructions)]) : this.instructions)?.format(this.getTemplateVariables(options), { workingDir: this.workingDir }) ?? [], (i) => i.role === "system");
+		const inputFileKey = options.agent?.inputFileKey;
+		const files = require_utils_type_utils.flat(inputFileKey ? require_utils_type_utils.checkArguments("Check input files", (0, _aigne_afs_utils_zod.optionalize)(require_model.fileUnionContentsSchema), input?.[inputFileKey]) : null);
+		if (options.agent?.memories.length || options.context?.memories.length) {
+			const deprecatedMemories = await this.deprecatedMemories(message, options);
+			if (deprecatedMemories.length) systemMessages.push(...deprecatedMemories);
+		}
+		const { structuredStreamMode, outputSchema } = options.agent || {};
+		if (structuredStreamMode && outputSchema) {
+			const instructions = options.agent?.customStructuredStreamInstructions?.instructions || PromptBuilder.from(require_structured_stream_instructions.STRUCTURED_STREAM_INSTRUCTIONS.instructions);
+			systemMessages.push({
+				role: "system",
+				content: (await instructions.buildPrompt({ input: {
+					...input,
+					outputJsonSchema: (0, zod_to_json_schema.zodToJsonSchema)(outputSchema)
+				} })).prompt
+			});
+		}
+		if (message || files.length) {
+			const content = [];
+			if (message && !userMessages.some((i) => i.role === "user" && (typeof i.content === "string" ? i.content.includes(message) : i.content?.some((c) => c.type === "text" && c.text.includes(message))))) content.push({
+				type: "text",
+				text: message
+			});
+			if (files.length) content.push(...files);
+			if (content.length) userMessages.push({
+				role: "user",
+				content
+			});
+		}
+		let systemMessage = this.mergeMessages(systemMessages, "system");
+		if (!systemMessage.content?.length) systemMessage = void 0;
+		let userMessage = this.mergeMessages(userMessages, "user");
+		if (!userMessage.content?.length) {
+			userMessage = {
+				role: "user",
+				content: systemMessage?.content
+			};
+			systemMessage = void 0;
+		}
+		if (!userMessage.content?.length) throw new Error("User message cannot be empty.");
+		return {
+			systemMessage,
+			userMessage
+		};
+	}
+	mergeMessages(messages, role) {
+		const content = [];
+		for (const message of messages) if (typeof message.content === "string") content.push({
+			type: "text",
+			text: message.content
+		});
+		else if (Array.isArray(message.content)) content.push(...message.content);
+		else if (message.content) throw new Error(`Unsupported message content type: ${typeof message.content}`);
+		return {
+			role,
+			content
+		};
+	}
+	async deprecatedMemories(message, options) {
+		const messages = [];
+		const memories = [];
+		if (options.agent && options.context) memories.push(...await options.agent.retrieveMemories({ search: message }, { context: options.context }));
+		if (options.agent?.useMemoriesFromContext && options.context?.memories?.length) memories.push(...options.context.memories);
+		if (memories.length) messages.push(...await this.convertMemoriesToMessages(memories, options));
+		return messages;
+	}
+	async convertMemoriesToMessages(memories, options) {
+		const messages = [];
+		const other = [];
+		const inputKey = options.agent?.inputKey;
+		const inputFileKey = options.agent?.inputFileKey;
+		const outputKey = options.agent?.outputKey || require_ai_agent.DEFAULT_OUTPUT_KEY;
+		const outputFileKey = options.agent?.outputFileKey || require_ai_agent.DEFAULT_OUTPUT_FILE_KEY;
+		const stringOrStringify = (value) => typeof value === "string" ? value : (0, yaml.stringify)(value);
+		const convertMemoryToMessage = async (content) => {
+			const { input, output } = content;
+			if (!input || !output) return [];
+			const result = [];
+			const userMessageContent = [];
+			if (typeof input === "object") {
+				const inputMessage = inputKey ? Reflect.get(input, inputKey) : void 0;
+				const inputFiles = inputFileKey ? Reflect.get(input, inputFileKey) : void 0;
+				if (inputMessage) userMessageContent.push({
+					type: "text",
+					text: stringOrStringify(inputMessage)
+				});
+				if (inputFiles) userMessageContent.push(...require_utils_type_utils.flat(require_utils_type_utils.checkArguments("Check memory input files", (0, _aigne_afs_utils_zod.optionalize)(require_model.fileUnionContentsSchema), inputFiles)));
+			}
+			if (!userMessageContent.length) userMessageContent.push({
+				type: "text",
+				text: stringOrStringify(input)
+			});
+			result.push({
+				role: "user",
+				content: userMessageContent
+			});
+			const agentMessageContent = [];
+			if (typeof output === "object") {
+				const outputMessage = Reflect.get(output, outputKey);
+				const outputFiles = Reflect.get(output, outputFileKey);
+				if (outputMessage) agentMessageContent.push({
+					type: "text",
+					text: stringOrStringify(outputMessage)
+				});
+				if (outputFiles) agentMessageContent.push(...require_utils_type_utils.flat(require_utils_type_utils.checkArguments("Check memory output files", (0, _aigne_afs_utils_zod.optionalize)(require_model.fileUnionContentsSchema), outputFiles)));
+			}
+			if (!agentMessageContent.length) agentMessageContent.push({
+				type: "text",
+				text: stringOrStringify(output)
+			});
+			result.push({
+				role: "agent",
+				content: agentMessageContent
+			});
+			return result;
+		};
+		for (const memory of memories) {
+			const { content } = memory;
+			if (require_utils_type_utils.isRecord(content) && "input" in content && content.input && "output" in content && content.output) messages.push(...await convertMemoryToMessage(content));
+			else other.push(memory);
+		}
+		if (other.length) messages.unshift({
+			role: "system",
+			content: await require_template.PromptTemplate.from(options.agent?.memoryPromptTemplate || require_memory_message_template.MEMORY_MESSAGE_TEMPLATE).format({ memories: (0, yaml.stringify)(other) })
+		});
+		return messages;
+	}
+	buildResponseFormat(options) {
+		const outputSchema = options.outputSchema || options.agent?.outputSchema;
+		if (!outputSchema) return void 0;
+		return !isEmptyObjectType(outputSchema) ? {
+			type: "json_schema",
+			jsonSchema: {
+				name: "output",
+				schema: require_utils_json_schema.outputSchemaToResponseFormatSchema(outputSchema),
+				strict: true
+			}
+		} : void 0;
+	}
+	async buildTools(options) {
+		const toolAgents = require_utils_type_utils.unique((options.context?.skills ?? []).concat(options.agent?.skills ?? []).concat(options.agent?.memoryAgentsAsTools ? options.agent.memories : []).flatMap((i) => i.isInvokable ? i : i.skills), (i) => i.name);
+		if (options.agent?.afs) toolAgents.push(...await require_index.getAFSSkills(options.agent.afs));
+		const tools = toolAgents.map((i) => ({
+			type: "function",
+			function: {
+				name: i.name,
+				description: i.description,
+				parameters: !isEmptyObjectType(i.inputSchema) ? require_utils_json_schema.outputSchemaToResponseFormatSchema(i.inputSchema) : {}
+			}
+		}));
+		let toolChoice;
+		const modelOptions = {};
+		const manualChoice = options.agent?.toolChoice;
+		if (manualChoice) if (manualChoice instanceof require_agent.Agent) toolChoice = {
+			type: "function",
+			function: {
+				name: manualChoice.name,
+				description: manualChoice.description
+			}
+		};
+		else if (manualChoice === "router") {
+			toolChoice = "required";
+			modelOptions.parallelToolCalls = false;
+		} else toolChoice = manualChoice;
+		else toolChoice = tools.length ? "auto" : void 0;
+		return {
+			toolAgents: toolAgents.length ? toolAgents : void 0,
+			tools: tools.length ? tools : void 0,
+			toolChoice,
+			modelOptions: Object.keys(modelOptions).length ? modelOptions : void 0
+		};
+	}
+};
+function isFromPromptResult(value) {
+	return typeof value === "object" && "messages" in value && Array.isArray(value.messages);
+}
+function isFromPath(value) {
+	return typeof value === "object" && "path" in value && typeof value.path === "string";
+}
+function isEmptyObjectType(schema) {
+	return schema instanceof zod.ZodObject && Object.keys(schema.shape).length === 0;
+}
+
+//#endregion
+exports.PromptBuilder = PromptBuilder;

package/dist/prompt/prompt-builder.d.cts

@@ -0,0 +1,57 @@
+import { FileUnionContent } from "../agents/model.cjs";
+import { ChatModel, ChatModelInput, ChatModelInputMessage } from "../agents/chat-model.cjs";
+import { Context } from "../aigne/context.cjs";
+import { Agent, Message } from "../agents/agent.cjs";
+import { AgentSession } from "./agent-session.cjs";
+import { ChatMessagesTemplate } from "./template.cjs";
+import { AIAgent } from "../agents/ai-agent.cjs";
+import { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
+
+//#region src/prompt/prompt-builder.d.ts
+interface PromptBuilderOptions {
+  instructions?: string | ChatMessagesTemplate;
+  workingDir?: string;
+}
+interface PromptBuildOptions {
+  context?: Context;
+  agent?: AIAgent;
+  input?: Message;
+  model?: ChatModel;
+  outputSchema?: Agent["outputSchema"];
+}
+declare class PromptBuilder {
+  static from(instructions: string | {
+    path: string;
+  } | GetPromptResult, {
+    workingDir
+  }?: {
+    workingDir?: string;
+  }): PromptBuilder;
+  private static fromFile;
+  private static fromMCPPromptResult;
+  constructor(options?: PromptBuilderOptions);
+  instructions?: string | ChatMessagesTemplate;
+  workingDir?: string;
+  copy(): PromptBuilder;
+  build(options: PromptBuildOptions): Promise<Omit<ChatModelInput, "messages"> & {
+    session: AgentSession;
+    userMessage: ChatModelInputMessage;
+    toolAgents?: Agent[];
+  }>;
+  buildPrompt(options: Pick<PromptBuildOptions, "input" | "context"> & {
+    inputFileKey?: string;
+  }): Promise<{
+    prompt: string;
+    image?: FileUnionContent[];
+  }>;
+  private getTemplateVariables;
+  private buildMessages;
+  private mergeMessages;
+  protected deprecatedMemories(message: string | undefined, options: PromptBuildOptions): Promise<ChatModelInputMessage[]>;
+  private convertMemoriesToMessages;
+  private buildResponseFormat;
+  private buildTools;
+}
+//#endregion
+export { PromptBuildOptions, PromptBuilder, PromptBuilderOptions };
+//# sourceMappingURL=prompt-builder.d.cts.map

package/dist/prompt/prompt-builder.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-builder.d.cts","names":[],"sources":["../../src/prompt/prompt-builder.ts"],"mappings":";;;;;;;;;;UAsCiB,oBAAA;EAAA,YAAA,YACS,oBAAA;EAAA,UAAA;AAAA;AAAA,UAIT,kBAAA;EAAA,OAAA,GACL,OAAA;EAAA,KAAA,GACF,OAAA;EAAA,KAAA,GACA,OAAA;EAAA,KAAA,GACA,SAAA;EAAA,YAAA,GACO,KAAA;AAAA;AAAA,cAGJ,aAAA;EAAA,OAAA,KAAA,YAAA;IAAA,IAAA;EAAA,IAEiC,eAAA;IAAA;EAAA;IAAA,UAAA;EAAA,IAEzC,aAAA;EAAA,eAAA,QAAA;EAAA,eAAA,mBAAA;EAAA,YAAA,OAAA,GA8CmB,oBAAA;EAAA,YAAA,YAKE,oBAAA;EAAA,UAAA;EAAA,KAAA,GAIhB,aAAA;EAAA,MAAA,OAAA,EAQa,kBAAA,GAAqB,OAAA,CACxC,IAAA,CAAK,cAAA;IAAA,OAAA,EACM,YAAA;IAAA,WAAA,EACI,qBAAA;IAAA,UAAA,GACA,KAAA;EAAA;EAAA,YAAA,OAAA,EAiCN,IAAA,CAAK,kBAAA;IAAA,YAAA;EAAA,IACb,OAAA;IAAA,MAAA;IAAA,KAAA,GAAkC,gBAAA;EAAA;EAAA,QAAA,oBAAA;EAAA,QAAA,aAAA;EAAA,QAAA,aAAA;EAAA,UAAA,mBAAA,OAAA,sBAAA,OAAA,EA6IoC,kBAAA,GAAkB,OAAA,CAAA,qBAAA;EAAA,QAAA,yBAAA;EAAA,QAAA,mBAAA;EAAA,QAAA,UAAA;AAAA"}

package/dist/prompt/prompt-builder.d.mts

@@ -0,0 +1,57 @@
+import { FileUnionContent } from "../agents/model.mjs";
+import { ChatModel, ChatModelInput, ChatModelInputMessage } from "../agents/chat-model.mjs";
+import { Context } from "../aigne/context.mjs";
+import { Agent, Message } from "../agents/agent.mjs";
+import { AgentSession } from "./agent-session.mjs";
+import { ChatMessagesTemplate } from "./template.mjs";
+import { AIAgent } from "../agents/ai-agent.mjs";
+import { GetPromptResult } from "@modelcontextprotocol/sdk/types.js";
+
+//#region src/prompt/prompt-builder.d.ts
+interface PromptBuilderOptions {
+  instructions?: string | ChatMessagesTemplate;
+  workingDir?: string;
+}
+interface PromptBuildOptions {
+  context?: Context;
+  agent?: AIAgent;
+  input?: Message;
+  model?: ChatModel;
+  outputSchema?: Agent["outputSchema"];
+}
+declare class PromptBuilder {
+  static from(instructions: string | {
+    path: string;
+  } | GetPromptResult, {
+    workingDir
+  }?: {
+    workingDir?: string;
+  }): PromptBuilder;
+  private static fromFile;
+  private static fromMCPPromptResult;
+  constructor(options?: PromptBuilderOptions);
+  instructions?: string | ChatMessagesTemplate;
+  workingDir?: string;
+  copy(): PromptBuilder;
+  build(options: PromptBuildOptions): Promise<Omit<ChatModelInput, "messages"> & {
+    session: AgentSession;
+    userMessage: ChatModelInputMessage;
+    toolAgents?: Agent[];
+  }>;
+  buildPrompt(options: Pick<PromptBuildOptions, "input" | "context"> & {
+    inputFileKey?: string;
+  }): Promise<{
+    prompt: string;
+    image?: FileUnionContent[];
+  }>;
+  private getTemplateVariables;
+  private buildMessages;
+  private mergeMessages;
+  protected deprecatedMemories(message: string | undefined, options: PromptBuildOptions): Promise<ChatModelInputMessage[]>;
+  private convertMemoriesToMessages;
+  private buildResponseFormat;
+  private buildTools;
+}
+//#endregion
+export { PromptBuildOptions, PromptBuilder, PromptBuilderOptions };
+//# sourceMappingURL=prompt-builder.d.mts.map

package/dist/prompt/prompt-builder.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"prompt-builder.d.mts","names":[],"sources":["../../src/prompt/prompt-builder.ts"],"mappings":";;;;;;;;;;UAsCiB,oBAAA;EAAA,YAAA,YACS,oBAAA;EAAA,UAAA;AAAA;AAAA,UAIT,kBAAA;EAAA,OAAA,GACL,OAAA;EAAA,KAAA,GACF,OAAA;EAAA,KAAA,GACA,OAAA;EAAA,KAAA,GACA,SAAA;EAAA,YAAA,GACO,KAAA;AAAA;AAAA,cAGJ,aAAA;EAAA,OAAA,KAAA,YAAA;IAAA,IAAA;EAAA,IAEiC,eAAA;IAAA;EAAA;IAAA,UAAA;EAAA,IAEzC,aAAA;EAAA,eAAA,QAAA;EAAA,eAAA,mBAAA;EAAA,YAAA,OAAA,GA8CmB,oBAAA;EAAA,YAAA,YAKE,oBAAA;EAAA,UAAA;EAAA,KAAA,GAIhB,aAAA;EAAA,MAAA,OAAA,EAQa,kBAAA,GAAqB,OAAA,CACxC,IAAA,CAAK,cAAA;IAAA,OAAA,EACM,YAAA;IAAA,WAAA,EACI,qBAAA;IAAA,UAAA,GACA,KAAA;EAAA;EAAA,YAAA,OAAA,EAiCN,IAAA,CAAK,kBAAA;IAAA,YAAA;EAAA,IACb,OAAA;IAAA,MAAA;IAAA,KAAA,GAAkC,gBAAA;EAAA;EAAA,QAAA,oBAAA;EAAA,QAAA,aAAA;EAAA,QAAA,aAAA;EAAA,UAAA,mBAAA,OAAA,sBAAA,OAAA,EA6IoC,kBAAA,GAAkB,OAAA,CAAA,qBAAA;EAAA,QAAA,yBAAA;EAAA,QAAA,mBAAA;EAAA,QAAA,UAAA;AAAA"}