@aigne/core 1.72.0 → 1.74.0-beta

package/dist/loader/schema.d.mts

@@ -0,0 +1,177 @@
+import { Role } from "../agents/chat-model.mjs";
+import { PromptBuilder } from "../prompt/prompt-builder.mjs";
+import { ZodType, z as z$1 } from "zod";
+
+//#region src/loader/schema.d.ts
+declare const inputOutputSchema: ({
+  path
+}: {
+  path: string;
+}) => z$1.ZodUnion<[ZodType<any, z$1.ZodTypeDef, any>, z$1.ZodEffects<z$1.ZodObject<{
+  type: z$1.ZodLiteral<"object">;
+  properties: z$1.ZodRecord<z$1.ZodString, z$1.ZodAny>;
+  required: z$1.ZodOptional<z$1.ZodArray<z$1.ZodString, "many">>;
+  additionalProperties: z$1.ZodOptional<z$1.ZodBoolean>;
+}, "strip", z$1.ZodTypeAny, {
+  type: "object";
+  properties: Record<string, any>;
+  required?: string[] | undefined;
+  additionalProperties?: boolean | undefined;
+}, {
+  type: "object";
+  properties: Record<string, any>;
+  required?: string[] | undefined;
+  additionalProperties?: boolean | undefined;
+}>, any, {
+  type: "object";
+  properties: Record<string, any>;
+  required?: string[] | undefined;
+  additionalProperties?: boolean | undefined;
+}>]>;
+declare const defaultInputSchema: z$1.ZodRecord<z$1.ZodString, z$1.ZodUnion<[z$1.ZodUnknown, z$1.ZodObject<{
+  $get: z$1.ZodString;
+}, "strip", z$1.ZodTypeAny, {
+  $get: string;
+}, {
+  $get: string;
+}>]>>;
+declare const chatModelObjectSchema: z$1.ZodObject<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+  temperature: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  topP: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  frequencyPenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  presencePenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined>;
+}, "passthrough", z$1.ZodTypeAny, z$1.objectOutputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+  temperature: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  topP: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  frequencyPenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  presencePenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined>;
+}, z$1.ZodTypeAny, "passthrough">, z$1.objectInputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+  temperature: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  topP: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  frequencyPenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  presencePenalty: ZodType<number | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | {
+    $get: string;
+  } | undefined>;
+  thinkingEffort: ZodType<number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, number | "high" | "medium" | "low" | "minimal" | {
+    $get: string;
+  } | undefined>;
+}, z$1.ZodTypeAny, "passthrough">>;
+declare const chatModelSchema: typeof chatModelObjectSchema;
+declare const imageModelObjectSchema: z$1.ZodObject<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+}, "passthrough", z$1.ZodTypeAny, z$1.objectOutputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+}, z$1.ZodTypeAny, "passthrough">, z$1.objectInputType<{
+  model: ZodType<string | {
+    $get: string;
+  } | undefined, z$1.ZodTypeDef, string | {
+    $get: string;
+  } | undefined>;
+}, z$1.ZodTypeAny, "passthrough">>;
+declare const imageModelSchema: typeof imageModelObjectSchema;
+declare function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined>;
+declare function camelizeSchema<T extends ZodType>(schema: T, {
+  shallow
+}?: {
+  shallow?: boolean;
+}): T;
+declare function preprocessSchema<T extends ZodType>(fn: (data: unknown) => unknown, schema: T): T;
+type Instructions = {
+  role: Exclude<Role, "tool">;
+  content: string;
+  path: string;
+  cacheControl?: {
+    type: "ephemeral";
+    ttl?: "5m" | "1h";
+  };
+}[];
+declare const getInstructionsSchema: ({
+  filepath
+}: {
+  filepath: string;
+}) => ZodType<Instructions>;
+declare function instructionsToPromptBuilder(instructions: Instructions | string): PromptBuilder;
+//#endregion
+export { Instructions, camelizeSchema, chatModelSchema, defaultInputSchema, getInstructionsSchema, imageModelSchema, inputOutputSchema, instructionsToPromptBuilder, optionalize, preprocessSchema };
+//# sourceMappingURL=schema.d.mts.map

package/dist/loader/schema.d.mts.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.d.mts","names":[],"sources":["../../src/loader/schema.ts"],"mappings":";;;;;cAUa,iBAAA;EAAA;AAAA;EAAA,IAAA;AAAA,MAA+C,GAAA,CAAA,QAAA,EAAA,OAAA,MAAA,GAAA,CAAA,UAAA,QAAA,GAAA,CAAA,UAAA,CAAA,GAAA,CAAA,SAAA;EAAA,IAAA;;;;;;;;;;;;;;;;;;;;cA2D/C,kBAAA,EAAkB,GAAA,CAAA,SAAA,CAAA,GAAA,CAAA,SAAA,EAAA,GAAA,CAAA,QAAA,EAAA,GAAA,CAAA,UAAA,EAAA,GAAA,CAAA,SAAA;EAAA,IAAA;;;;;;cAEzB,qBAAA,EAAqB,GAAA,CAAA,SAAA;EAAA,KAAA;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;cAuBd,eAAA,SAgBY,qBAAA;AAAA,cAEnB,sBAAA,EAAsB,GAAA,CAAA,SAAA;EAAA,KAAA;;;;;;;;;;;;;;;;;;cAQf,gBAAA,SAIY,sBAAA;AAAA,iBAET,WAAA,GAAA,CAAA,MAAA,EAAuB,OAAA,CAAQ,CAAA,IAAK,OAAA,CAAQ,CAAA;AAAA,iBAI5C,cAAA,WAAyB,OAAA,CAAA,CAAA,MAAA,EAC/B,CAAA;EAAA;AAAA;EAAA,OAAA;AAAA,IAEP,CAAA;AAAA,iBAIa,gBAAA,WAA2B,OAAA,CAAA,CAAA,EAAA,GAAA,IAAA,uBAAA,MAAA,EAAiD,CAAA,GAAI,CAAA;AAAA,KA6BpF,YAAA;EAAA,IAAA,EACJ,OAAA,CAAQ,IAAA;EAAA,OAAA;EAAA,IAAA;EAAA,YAAA;IAAA,IAAA;IAAA,GAAA;EAAA;AAAA;AAAA,cAmCH,qBAAA;EAAA;AAAA;EAAA,QAAA;AAAA,MAWQ,OAAA,CAAQ,YAAA;AAAA,iBAEb,2BAAA,CAAA,YAAA,EAA0C,YAAA,YAAqB,aAAA"}

package/dist/loader/schema.mjs

@@ -0,0 +1,127 @@
+import { isRecord } from "../utils/type-utils.mjs";
+import { getterSchema } from "../agents/types.mjs";
+import "../agents/agent.mjs";
+import { roleSchema } from "../agents/chat-model.mjs";
+import { ChatMessagesTemplate, parseChatMessages } from "../prompt/template.mjs";
+import { PromptBuilder } from "../prompt/prompt-builder.mjs";
+import { camelize } from "../utils/camelize.mjs";
+import { nodejs } from "@aigne/utils/nodejs";
+import { parse } from "yaml";
+import { z as z$1 } from "zod";
+
+//#region src/loader/schema.ts
+const inputOutputSchema = ({ path }) => {
+	const includeExternalSchema = async (schema) => {
+		if (schema?.type === "object" && schema.properties) return {
+			...schema,
+			properties: Object.fromEntries(await Promise.all(Object.entries(schema.properties).map(async ([key, value]) => [key, await includeExternalSchema(value)])))
+		};
+		if (schema?.type === "array" && schema.items) return {
+			...schema,
+			items: await includeExternalSchema(schema.items)
+		};
+		if (typeof schema === "string") {
+			const raw = await nodejs.fs.readFile(nodejs.path.join(nodejs.path.dirname(path), schema), "utf8");
+			return nestedJsonSchema.parseAsync(parse(raw));
+		}
+		return schema;
+	};
+	const nestedJsonSchema = z$1.object({ type: z$1.string() }).passthrough().transform((v) => includeExternalSchema(v));
+	const jsonSchemaSchema = z$1.object({
+		type: z$1.literal("object"),
+		properties: z$1.record(z$1.any()),
+		required: z$1.array(z$1.string()).optional(),
+		additionalProperties: z$1.boolean().optional()
+	}).transform((v) => includeExternalSchema(v));
+	return z$1.union([z$1.string().transform((v) => nodejs.fs.readFile(nodejs.path.join(nodejs.path.dirname(path), v), "utf8").then((raw) => jsonSchemaSchema.parseAsync(parse(raw)))), jsonSchemaSchema]);
+};
+const defaultInputSchema = z$1.record(z$1.string(), getterSchema(z$1.unknown()));
+const chatModelObjectSchema = camelizeSchema(z$1.object({
+	model: optionalize(getterSchema(z$1.string())),
+	temperature: optionalize(getterSchema(z$1.number().min(0).max(2))),
+	topP: optionalize(getterSchema(z$1.number().min(0))),
+	frequencyPenalty: optionalize(getterSchema(z$1.number().min(-2).max(2))),
+	presencePenalty: optionalize(getterSchema(z$1.number().min(-2).max(2))),
+	thinkingEffort: optionalize(getterSchema(z$1.union([
+		z$1.number().int(),
+		z$1.literal("high"),
+		z$1.literal("medium"),
+		z$1.literal("low"),
+		z$1.literal("minimal")
+	])))
+}).passthrough());
+const chatModelSchema = z$1.preprocess((v) => {
+	if (!isRecord(v)) return v;
+	return {
+		...v,
+		model: v.model || (v.provider && v.name ? `${v.provider || ""}:${v.name || ""}` : void 0) || void 0
+	};
+}, z$1.union([z$1.string(), chatModelObjectSchema])).transform((v) => typeof v === "string" ? { model: v } : v);
+const imageModelObjectSchema = camelizeSchema(z$1.object({ model: optionalize(getterSchema(z$1.string())) }).passthrough());
+const imageModelSchema = z$1.union([z$1.string(), imageModelObjectSchema]).transform((v) => typeof v === "string" ? { model: v } : v);
+function optionalize(schema) {
+	return schema.nullish().transform((v) => v ?? void 0);
+}
+function camelizeSchema(schema, { shallow = true } = {}) {
+	return z$1.preprocess((v) => isRecord(v) ? camelize(v, shallow) : v, schema);
+}
+function preprocessSchema(fn, schema) {
+	return z$1.preprocess(fn, schema);
+}
+const instructionItemSchema = camelizeSchema(z$1.union([z$1.object({
+	role: roleSchema.default("system"),
+	url: z$1.string(),
+	cacheControl: optionalize(z$1.object({
+		type: z$1.literal("ephemeral"),
+		ttl: optionalize(z$1.union([z$1.literal("5m"), z$1.literal("1h")]))
+	}))
+}), z$1.object({
+	role: roleSchema.default("system"),
+	content: z$1.string(),
+	cacheControl: optionalize(z$1.object({
+		type: z$1.literal("ephemeral"),
+		ttl: optionalize(z$1.union([z$1.literal("5m"), z$1.literal("1h")]))
+	}))
+})]));
+const parseInstructionItem = ({ filepath }) => async ({ role, cacheControl, ...v }) => {
+	if (role === "tool") throw new Error(`'tool' role is not allowed in instruction item in agent file ${filepath}`);
+	if ("content" in v && typeof v.content === "string") return {
+		role,
+		content: v.content,
+		path: filepath,
+		cacheControl
+	};
+	if ("url" in v && typeof v.url === "string") {
+		const url = nodejs.path.isAbsolute(v.url) ? v.url : nodejs.path.join(nodejs.path.dirname(filepath), v.url);
+		return nodejs.fs.readFile(url, "utf8").then((content) => ({
+			role,
+			content,
+			path: url,
+			cacheControl
+		}));
+	}
+	throw new Error(`Invalid instruction item in agent file ${filepath}. Expected 'content' or 'url' property`);
+};
+const getInstructionsSchema = ({ filepath }) => z$1.union([
+	z$1.string(),
+	instructionItemSchema,
+	z$1.array(instructionItemSchema)
+]).transform(async (v) => {
+	if (typeof v === "string") return [{
+		role: "system",
+		content: v,
+		path: filepath
+	}];
+	if (Array.isArray(v)) return Promise.all(v.map((item) => parseInstructionItem({ filepath })(item)));
+	return [await parseInstructionItem({ filepath })(v)];
+});
+function instructionsToPromptBuilder(instructions) {
+	return new PromptBuilder({ instructions: typeof instructions === "string" ? instructions : ChatMessagesTemplate.from(parseChatMessages(instructions.map((i) => ({
+		...i,
+		options: { workingDir: nodejs.path.dirname(i.path) }
+	})))) });
+}
+
+//#endregion
+export { camelizeSchema, chatModelSchema, defaultInputSchema, getInstructionsSchema, imageModelSchema, inputOutputSchema, instructionsToPromptBuilder, optionalize, preprocessSchema };
+//# sourceMappingURL=schema.mjs.map

package/dist/loader/schema.mjs.map

@@ -0,0 +1 @@
+{"version":3,"file":"schema.mjs","names":["z"],"sources":["../../src/loader/schema.ts"],"sourcesContent":["import { nodejs } from \"@aigne/utils/nodejs\";\nimport { parse } from \"yaml\";\nimport { type ZodType, z } from \"zod\";\nimport { getterSchema } from \"../agents/agent.js\";\nimport { type Role, roleSchema } from \"../agents/chat-model.js\";\nimport { PromptBuilder } from \"../prompt/prompt-builder.js\";\nimport { ChatMessagesTemplate, parseChatMessages } from \"../prompt/template.js\";\nimport { camelize } from \"../utils/camelize.js\";\nimport { isRecord } from \"../utils/type-utils.js\";\n\nexport const inputOutputSchema = ({ path }: { path: string }) => {\n  const includeExternalSchema = async (schema: any): Promise<typeof schema> => {\n    if (schema?.type === \"object\" && schema.properties) {\n      return {\n        ...schema,\n        properties: Object.fromEntries(\n          await Promise.all(\n            Object.entries(schema.properties).map(async ([key, value]) => [\n              key,\n              await includeExternalSchema(value),\n            ]),\n          ),\n        ),\n      };\n    }\n\n    if (schema?.type === \"array\" && schema.items) {\n      return { ...schema, items: await includeExternalSchema(schema.items) };\n    }\n\n    // Load nested schemas from file\n    if (typeof schema === \"string\") {\n      const raw = await nodejs.fs.readFile(\n        nodejs.path.join(nodejs.path.dirname(path), schema),\n        \"utf8\",\n      );\n      return nestedJsonSchema.parseAsync(parse(raw));\n    }\n    return schema;\n  };\n\n  const nestedJsonSchema = z\n    .object({\n      type: z.string(),\n    })\n    .passthrough()\n    .transform((v) => includeExternalSchema(v));\n\n  const jsonSchemaSchema = z\n    .object({\n      type: z.literal(\"object\"),\n      properties: z.record(z.any()),\n      required: z.array(z.string()).optional(),\n      additionalProperties: z.boolean().optional(),\n    })\n    .transform((v) => includeExternalSchema(v));\n\n  return z.union([\n    z\n      .string()\n      .transform((v) =>\n        nodejs.fs\n          .readFile(nodejs.path.join(nodejs.path.dirname(path), v), \"utf8\")\n          .then((raw) => jsonSchemaSchema.parseAsync(parse(raw))),\n      ) as unknown as ZodType<z.infer<typeof jsonSchemaSchema>>,\n    jsonSchemaSchema,\n  ]);\n};\n\nexport const defaultInputSchema = z.record(z.string(), getterSchema(z.unknown()));\n\nconst chatModelObjectSchema = camelizeSchema(\n  z\n    .object({\n      model: optionalize(getterSchema(z.string())),\n      temperature: optionalize(getterSchema(z.number().min(0).max(2))),\n      topP: optionalize(getterSchema(z.number().min(0))),\n      frequencyPenalty: optionalize(getterSchema(z.number().min(-2).max(2))),\n      presencePenalty: optionalize(getterSchema(z.number().min(-2).max(2))),\n      thinkingEffort: optionalize(\n        getterSchema(\n          z.union([\n            z.number().int(),\n            z.literal(\"high\"),\n            z.literal(\"medium\"),\n            z.literal(\"low\"),\n            z.literal(\"minimal\"),\n          ]),\n        ),\n      ),\n    })\n    .passthrough(),\n);\n\nexport const chatModelSchema = z\n  .preprocess(\n    (v) => {\n      if (!isRecord(v)) return v;\n      return {\n        ...v,\n        model:\n          v.model ||\n          (v.provider && v.name ? `${v.provider || \"\"}:${v.name || \"\"}` : undefined) ||\n          undefined,\n      };\n    },\n    z.union([z.string(), chatModelObjectSchema]),\n  )\n  .transform((v) =>\n    typeof v === \"string\" ? { model: v } : v,\n  ) as unknown as typeof chatModelObjectSchema;\n\nconst imageModelObjectSchema = camelizeSchema(\n  z\n    .object({\n      model: optionalize(getterSchema(z.string())),\n    })\n    .passthrough(),\n);\n\nexport const imageModelSchema = z\n  .union([z.string(), imageModelObjectSchema])\n  .transform((v) =>\n    typeof v === \"string\" ? { model: v } : v,\n  ) as unknown as typeof imageModelObjectSchema;\n\nexport function optionalize<T>(schema: ZodType<T>): ZodType<T | undefined> {\n  return schema.nullish().transform((v) => v ?? undefined) as ZodType<T | undefined>;\n}\n\nexport function camelizeSchema<T extends ZodType>(\n  schema: T,\n  { shallow = true }: { shallow?: boolean } = {},\n): T {\n  return z.preprocess((v) => (isRecord(v) ? camelize(v, shallow) : v), schema) as unknown as T;\n}\n\nexport function preprocessSchema<T extends ZodType>(fn: (data: unknown) => unknown, schema: T): T {\n  return z.preprocess(fn, schema) as unknown as T;\n}\n\nconst instructionItemSchema = camelizeSchema(\n  z.union([\n    z.object({\n      role: roleSchema.default(\"system\"),\n      url: z.string(),\n      cacheControl: optionalize(\n        z.object({\n          type: z.literal(\"ephemeral\"),\n          ttl: optionalize(z.union([z.literal(\"5m\"), z.literal(\"1h\")])),\n        }),\n      ),\n    }),\n    z.object({\n      role: roleSchema.default(\"system\"),\n      content: z.string(),\n      cacheControl: optionalize(\n        z.object({\n          type: z.literal(\"ephemeral\"),\n          ttl: optionalize(z.union([z.literal(\"5m\"), z.literal(\"1h\")])),\n        }),\n      ),\n    }),\n  ]),\n);\n\nexport type Instructions = {\n  role: Exclude<Role, \"tool\">;\n  content: string;\n  path: string;\n  cacheControl?: {\n    type: \"ephemeral\";\n    ttl?: \"5m\" | \"1h\";\n  };\n}[];\n\nconst parseInstructionItem =\n  ({ filepath }: { filepath: string }) =>\n  async ({\n    role,\n    cacheControl,\n    ...v\n  }: z.infer<typeof instructionItemSchema>): Promise<Instructions[number]> => {\n    if (role === \"tool\")\n      throw new Error(`'tool' role is not allowed in instruction item in agent file ${filepath}`);\n\n    if (\"content\" in v && typeof v.content === \"string\") {\n      return { role, content: v.content, path: filepath, cacheControl };\n    }\n    if (\"url\" in v && typeof v.url === \"string\") {\n      const url = nodejs.path.isAbsolute(v.url)\n        ? v.url\n        : nodejs.path.join(nodejs.path.dirname(filepath), v.url);\n      return nodejs.fs\n        .readFile(url, \"utf8\")\n        .then((content) => ({ role, content, path: url, cacheControl }));\n    }\n    throw new Error(\n      `Invalid instruction item in agent file ${filepath}. Expected 'content' or 'url' property`,\n    );\n  };\n\nexport const getInstructionsSchema = ({ filepath }: { filepath: string }) =>\n  z\n    .union([z.string(), instructionItemSchema, z.array(instructionItemSchema)])\n    .transform(async (v): Promise<Instructions> => {\n      if (typeof v === \"string\") return [{ role: \"system\", content: v, path: filepath }];\n\n      if (Array.isArray(v)) {\n        return Promise.all(v.map((item) => parseInstructionItem({ filepath })(item)));\n      }\n\n      return [await parseInstructionItem({ filepath })(v)];\n    }) as unknown as ZodType<Instructions>;\n\nexport function instructionsToPromptBuilder(instructions: Instructions | string) {\n  return new PromptBuilder({\n    instructions:\n      typeof instructions === \"string\"\n        ? instructions\n        : ChatMessagesTemplate.from(\n            parseChatMessages(\n              instructions.map((i) => ({\n                ...i,\n                options: { workingDir: nodejs.path.dirname(i.path) },\n              })),\n            ),\n          ),\n  });\n}\n"],"mappings":";;;;;;;;;;;;AAUA,MAAa,qBAAqB,EAAE,WAA6B;CAC/D,MAAM,wBAAwB,OAAO,WAAwC;AAC3E,MAAI,QAAQ,SAAS,YAAY,OAAO,WACtC,QAAO;GACL,GAAG;GACH,YAAY,OAAO,YACjB,MAAM,QAAQ,IACZ,OAAO,QAAQ,OAAO,WAAW,CAAC,IAAI,OAAO,CAAC,KAAK,WAAW,CAC5D,KACA,MAAM,sBAAsB,MAAM,CACnC,CAAC,CACH,CACF;GACF;AAGH,MAAI,QAAQ,SAAS,WAAW,OAAO,MACrC,QAAO;GAAE,GAAG;GAAQ,OAAO,MAAM,sBAAsB,OAAO,MAAM;GAAE;AAIxE,MAAI,OAAO,WAAW,UAAU;GAC9B,MAAM,MAAM,MAAM,OAAO,GAAG,SAC1B,OAAO,KAAK,KAAK,OAAO,KAAK,QAAQ,KAAK,EAAE,OAAO,EACnD,OACD;AACD,UAAO,iBAAiB,WAAW,MAAM,IAAI,CAAC;;AAEhD,SAAO;;CAGT,MAAM,mBAAmBA,IACtB,OAAO,EACN,MAAMA,IAAE,QAAQ,EACjB,CAAC,CACD,aAAa,CACb,WAAW,MAAM,sBAAsB,EAAE,CAAC;CAE7C,MAAM,mBAAmBA,IACtB,OAAO;EACN,MAAMA,IAAE,QAAQ,SAAS;EACzB,YAAYA,IAAE,OAAOA,IAAE,KAAK,CAAC;EAC7B,UAAUA,IAAE,MAAMA,IAAE,QAAQ,CAAC,CAAC,UAAU;EACxC,sBAAsBA,IAAE,SAAS,CAAC,UAAU;EAC7C,CAAC,CACD,WAAW,MAAM,sBAAsB,EAAE,CAAC;AAE7C,QAAOA,IAAE,MAAM,CACbA,IACG,QAAQ,CACR,WAAW,MACV,OAAO,GACJ,SAAS,OAAO,KAAK,KAAK,OAAO,KAAK,QAAQ,KAAK,EAAE,EAAE,EAAE,OAAO,CAChE,MAAM,QAAQ,iBAAiB,WAAW,MAAM,IAAI,CAAC,CAAC,CAC1D,EACH,iBACD,CAAC;;AAGJ,MAAa,qBAAqBA,IAAE,OAAOA,IAAE,QAAQ,EAAE,aAAaA,IAAE,SAAS,CAAC,CAAC;AAEjF,MAAM,wBAAwB,eAC5BA,IACG,OAAO;CACN,OAAO,YAAY,aAAaA,IAAE,QAAQ,CAAC,CAAC;CAC5C,aAAa,YAAY,aAAaA,IAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,IAAI,EAAE,CAAC,CAAC;CAChE,MAAM,YAAY,aAAaA,IAAE,QAAQ,CAAC,IAAI,EAAE,CAAC,CAAC;CAClD,kBAAkB,YAAY,aAAaA,IAAE,QAAQ,CAAC,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;CACtE,iBAAiB,YAAY,aAAaA,IAAE,QAAQ,CAAC,IAAI,GAAG,CAAC,IAAI,EAAE,CAAC,CAAC;CACrE,gBAAgB,YACd,aACEA,IAAE,MAAM;EACNA,IAAE,QAAQ,CAAC,KAAK;EAChBA,IAAE,QAAQ,OAAO;EACjBA,IAAE,QAAQ,SAAS;EACnBA,IAAE,QAAQ,MAAM;EAChBA,IAAE,QAAQ,UAAU;EACrB,CAAC,CACH,CACF;CACF,CAAC,CACD,aAAa,CACjB;AAED,MAAa,kBAAkBA,IAC5B,YACE,MAAM;AACL,KAAI,CAAC,SAAS,EAAE,CAAE,QAAO;AACzB,QAAO;EACL,GAAG;EACH,OACE,EAAE,UACD,EAAE,YAAY,EAAE,OAAO,GAAG,EAAE,YAAY,GAAG,GAAG,EAAE,QAAQ,OAAO,WAChE;EACH;GAEHA,IAAE,MAAM,CAACA,IAAE,QAAQ,EAAE,sBAAsB,CAAC,CAC7C,CACA,WAAW,MACV,OAAO,MAAM,WAAW,EAAE,OAAO,GAAG,GAAG,EACxC;AAEH,MAAM,yBAAyB,eAC7BA,IACG,OAAO,EACN,OAAO,YAAY,aAAaA,IAAE,QAAQ,CAAC,CAAC,EAC7C,CAAC,CACD,aAAa,CACjB;AAED,MAAa,mBAAmBA,IAC7B,MAAM,CAACA,IAAE,QAAQ,EAAE,uBAAuB,CAAC,CAC3C,WAAW,MACV,OAAO,MAAM,WAAW,EAAE,OAAO,GAAG,GAAG,EACxC;AAEH,SAAgB,YAAe,QAA4C;AACzE,QAAO,OAAO,SAAS,CAAC,WAAW,MAAM,KAAK,OAAU;;AAG1D,SAAgB,eACd,QACA,EAAE,UAAU,SAAgC,EAAE,EAC3C;AACH,QAAOA,IAAE,YAAY,MAAO,SAAS,EAAE,GAAG,SAAS,GAAG,QAAQ,GAAG,GAAI,OAAO;;AAG9E,SAAgB,iBAAoC,IAAgC,QAAc;AAChG,QAAOA,IAAE,WAAW,IAAI,OAAO;;AAGjC,MAAM,wBAAwB,eAC5BA,IAAE,MAAM,CACNA,IAAE,OAAO;CACP,MAAM,WAAW,QAAQ,SAAS;CAClC,KAAKA,IAAE,QAAQ;CACf,cAAc,YACZA,IAAE,OAAO;EACP,MAAMA,IAAE,QAAQ,YAAY;EAC5B,KAAK,YAAYA,IAAE,MAAM,CAACA,IAAE,QAAQ,KAAK,EAAEA,IAAE,QAAQ,KAAK,CAAC,CAAC,CAAC;EAC9D,CAAC,CACH;CACF,CAAC,EACFA,IAAE,OAAO;CACP,MAAM,WAAW,QAAQ,SAAS;CAClC,SAASA,IAAE,QAAQ;CACnB,cAAc,YACZA,IAAE,OAAO;EACP,MAAMA,IAAE,QAAQ,YAAY;EAC5B,KAAK,YAAYA,IAAE,MAAM,CAACA,IAAE,QAAQ,KAAK,EAAEA,IAAE,QAAQ,KAAK,CAAC,CAAC,CAAC;EAC9D,CAAC,CACH;CACF,CAAC,CACH,CAAC,CACH;AAYD,MAAM,wBACH,EAAE,eACH,OAAO,EACL,MACA,cACA,GAAG,QACuE;AAC1E,KAAI,SAAS,OACX,OAAM,IAAI,MAAM,gEAAgE,WAAW;AAE7F,KAAI,aAAa,KAAK,OAAO,EAAE,YAAY,SACzC,QAAO;EAAE;EAAM,SAAS,EAAE;EAAS,MAAM;EAAU;EAAc;AAEnE,KAAI,SAAS,KAAK,OAAO,EAAE,QAAQ,UAAU;EAC3C,MAAM,MAAM,OAAO,KAAK,WAAW,EAAE,IAAI,GACrC,EAAE,MACF,OAAO,KAAK,KAAK,OAAO,KAAK,QAAQ,SAAS,EAAE,EAAE,IAAI;AAC1D,SAAO,OAAO,GACX,SAAS,KAAK,OAAO,CACrB,MAAM,aAAa;GAAE;GAAM;GAAS,MAAM;GAAK;GAAc,EAAE;;AAEpE,OAAM,IAAI,MACR,0CAA0C,SAAS,wCACpD;;AAGL,MAAa,yBAAyB,EAAE,eACtCA,IACG,MAAM;CAACA,IAAE,QAAQ;CAAE;CAAuBA,IAAE,MAAM,sBAAsB;CAAC,CAAC,CAC1E,UAAU,OAAO,MAA6B;AAC7C,KAAI,OAAO,MAAM,SAAU,QAAO,CAAC;EAAE,MAAM;EAAU,SAAS;EAAG,MAAM;EAAU,CAAC;AAElF,KAAI,MAAM,QAAQ,EAAE,CAClB,QAAO,QAAQ,IAAI,EAAE,KAAK,SAAS,qBAAqB,EAAE,UAAU,CAAC,CAAC,KAAK,CAAC,CAAC;AAG/E,QAAO,CAAC,MAAM,qBAAqB,EAAE,UAAU,CAAC,CAAC,EAAE,CAAC;EACpD;AAEN,SAAgB,4BAA4B,cAAqC;AAC/E,QAAO,IAAI,cAAc,EACvB,cACE,OAAO,iBAAiB,WACpB,eACA,qBAAqB,KACnB,kBACE,aAAa,KAAK,OAAO;EACvB,GAAG;EACH,SAAS,EAAE,YAAY,OAAO,KAAK,QAAQ,EAAE,KAAK,EAAE;EACrD,EAAE,CACJ,CACF,EACR,CAAC"}

package/dist/memory/index.cjs

@@ -0,0 +1,3 @@
+const require_recorder = require('./recorder.cjs');
+const require_retriever = require('./retriever.cjs');
+const require_memory = require('./memory.cjs');

package/dist/memory/index.d.mts

@@ -0,0 +1,3 @@
+import { MemoryRecorder, MemoryRecorderInput, MemoryRecorderOptions, MemoryRecorderOutput, memoryRecorderInputSchema, memoryRecorderOutputSchema } from "./recorder.mjs";
+import { MemoryRetriever, MemoryRetrieverInput, MemoryRetrieverOptions, MemoryRetrieverOutput, memoryRetrieverInputSchema, memoryRetrieverOutputSchema } from "./retriever.mjs";
+import { Memory, MemoryAgent, MemoryAgentOptions, newMemoryId } from "./memory.mjs";

package/dist/memory/index.mjs

@@ -0,0 +1,5 @@
+import { MemoryRecorder, memoryRecorderInputSchema, memoryRecorderOutputSchema } from "./recorder.mjs";
+import { MemoryRetriever, memoryRetrieverInputSchema, memoryRetrieverOutputSchema } from "./retriever.mjs";
+import { MemoryAgent, newMemoryId } from "./memory.mjs";
+
+export {  };

package/dist/memory/memory.cjs

@@ -0,0 +1,134 @@
+const require_rolldown_runtime = require('../_virtual/rolldown_runtime.cjs');
+const require_utils_type_utils = require('../utils/type-utils.cjs');
+const require_agent = require('../agents/agent.cjs');
+const require_recorder = require('./recorder.cjs');
+const require_retriever = require('./retriever.cjs');
+let zod = require("zod");
+let _aigne_uuid = require("@aigne/uuid");
+
+//#region src/memory/memory.ts
+const newMemoryId = () => (0, _aigne_uuid.v7)();
+/**
+* A specialized agent responsible for managing, storing, and retrieving memories within the agent system.
+*
+* MemoryAgent serves as a bridge between application logic and memory storage/retrieval mechanisms.
+* It delegates the actual memory operations to specialized recorder and retriever agents that
+* are attached as skills. This agent doesn't directly process messages like other agents but
+* instead provides memory management capabilities to the system.
+*/
+var MemoryAgent = class extends require_agent.Agent {
+	tag = "MemoryAgent";
+	/**
+	* Creates a new MemoryAgent instance.
+	*/
+	constructor(options) {
+		require_utils_type_utils.checkArguments("MemoryAgent", memoryAgentOptionsSchema, options);
+		super({
+			subscribeTopic: options.subscribeTopic,
+			skills: options.skills
+		});
+		this.recorder = options.recorder instanceof require_recorder.MemoryRecorder ? options.recorder : options.recorder && new require_recorder.MemoryRecorder(typeof options.recorder === "function" ? { process: options.recorder } : options.recorder);
+		this.retriever = options.retriever instanceof require_retriever.MemoryRetriever ? options.retriever : options.retriever && new require_retriever.MemoryRetriever(typeof options.retriever === "function" ? { process: options.retriever } : options.retriever);
+		this.autoUpdate = options.autoUpdate;
+	}
+	_retriever;
+	/**
+	* Agent used for retrieving memories from storage.
+	*
+	* This retriever is automatically added to the agent's skills when set.
+	* Setting a new retriever will remove any previously set retriever from skills.
+	*/
+	get retriever() {
+		return this._retriever;
+	}
+	set retriever(value) {
+		if (this._retriever) require_utils_type_utils.remove(this.skills, [this._retriever]);
+		this._retriever = value;
+		if (value) this.skills.push(value);
+	}
+	_recorder;
+	/**
+	* Agent used for recording and storing new memories.
+	*
+	* This recorder is automatically added to the agent's skills when set.
+	* Setting a new recorder will remove any previously set recorder from skills.
+	*/
+	get recorder() {
+		return this._recorder;
+	}
+	set recorder(value) {
+		if (this._recorder) require_utils_type_utils.remove(this.skills, [this._recorder]);
+		this._recorder = value;
+		if (value) this.skills.push(value);
+	}
+	/**
+	* Controls whether to automatically update the memory when agent call completes.
+	*
+	* When true, the agent will automatically record any relevant information
+	* after completing operations, creating a history of interactions.
+	*/
+	autoUpdate;
+	/**
+	* Indicates whether this agent can be directly called.
+	*
+	* MemoryAgent is designed to be used as a supporting component rather than
+	* being directly invoked for processing, so this returns false.
+	*/
+	get isCallable() {
+		return false;
+	}
+	/**
+	* The standard message processing method required by the Agent interface.
+	*
+	* MemoryAgent doesn't directly process messages like other agents, so this method
+	* throws an error when called. Use the specialized retrieve() and record() methods instead.
+	*/
+	async process(_input, _options) {
+		throw new Error("Method not implemented.");
+	}
+	/**
+	* Retrieves memories based on the provided input criteria.
+	*
+	* Delegates the actual retrieval operation to the configured retriever agent.
+	*
+	* @param input - The retrieval parameters (can include search terms, limits, etc.)
+	* @param context - The execution context
+	* @returns A promise resolving to the retrieved memories
+	* @throws Error - If no retriever has been initialized
+	*/
+	async retrieve(input, context) {
+		if (!this.retriever) throw new Error("MemoryAgent retriever not initialized");
+		return context.invoke(this.retriever, input);
+	}
+	/**
+	* Records new memories based on the provided input content.
+	*
+	* Delegates the actual recording operation to the configured recorder agent.
+	*
+	* @param input - The content to be recorded as memories
+	* @param context - The execution context
+	* @returns A promise resolving to the recorded memories
+	* @throws Error - If no recorder has been initialized
+	*/
+	async record(input, context) {
+		if (!this.recorder) throw new Error("MemoryAgent recorder not initialized");
+		return context.invoke(this.recorder, input);
+	}
+	async onMessage({ role, source, message, context }) {
+		this.record({ content: [role === "user" ? { input: message } : {
+			output: message,
+			source
+		}] }, context);
+	}
+};
+const memoryAgentOptionsSchema = zod.z.object({
+	retriever: zod.z.custom().optional(),
+	recorder: zod.z.custom().optional(),
+	autoUpdate: zod.z.boolean().optional(),
+	subscribeTopic: zod.z.union([zod.z.string(), zod.z.array(zod.z.string())]).optional(),
+	skills: zod.z.array(zod.z.custom()).optional()
+});
+
+//#endregion
+exports.MemoryAgent = MemoryAgent;
+exports.newMemoryId = newMemoryId;

package/dist/memory/memory.d.cts

@@ -0,0 +1,103 @@
+import { MessagePayload } from "../aigne/message-queue.cjs";
+import { MemoryRecorder, MemoryRecorderInput, MemoryRecorderOptions, MemoryRecorderOutput } from "./recorder.cjs";
+import { MemoryRetriever, MemoryRetrieverInput, MemoryRetrieverOptions, MemoryRetrieverOutput } from "./retriever.cjs";
+import { Context } from "../aigne/context.cjs";
+import { Agent, AgentInvokeOptions, AgentOptions, Message } from "../agents/agent.cjs";
+
+//#region src/memory/memory.d.ts
+interface Memory {
+  id: string;
+  sessionId?: string | null;
+  content: unknown;
+  createdAt: string;
+}
+declare const newMemoryId: () => string;
+interface MemoryAgentOptions extends Partial<Pick<MemoryAgent, "autoUpdate">>, Pick<AgentOptions, "subscribeTopic" | "skills"> {
+  recorder?: MemoryRecorder | MemoryRecorderOptions["process"] | MemoryRecorderOptions;
+  retriever?: MemoryRetriever | MemoryRetrieverOptions["process"] | MemoryRetrieverOptions;
+}
+/**
+ * A specialized agent responsible for managing, storing, and retrieving memories within the agent system.
+ *
+ * MemoryAgent serves as a bridge between application logic and memory storage/retrieval mechanisms.
+ * It delegates the actual memory operations to specialized recorder and retriever agents that
+ * are attached as skills. This agent doesn't directly process messages like other agents but
+ * instead provides memory management capabilities to the system.
+ */
+declare class MemoryAgent extends Agent {
+  tag: string;
+  /**
+       * Creates a new MemoryAgent instance.
+       */
+  constructor(options: MemoryAgentOptions);
+  private _retriever?;
+  /**
+       * Agent used for retrieving memories from storage.
+       *
+       * This retriever is automatically added to the agent's skills when set.
+       * Setting a new retriever will remove any previously set retriever from skills.
+       */
+  get retriever(): MemoryRetriever | undefined;
+  set retriever(value: MemoryRetriever | undefined);
+  private _recorder?;
+  /**
+       * Agent used for recording and storing new memories.
+       *
+       * This recorder is automatically added to the agent's skills when set.
+       * Setting a new recorder will remove any previously set recorder from skills.
+       */
+  get recorder(): MemoryRecorder | undefined;
+  set recorder(value: MemoryRecorder | undefined);
+  /**
+       * Controls whether to automatically update the memory when agent call completes.
+       *
+       * When true, the agent will automatically record any relevant information
+       * after completing operations, creating a history of interactions.
+       */
+  autoUpdate?: boolean;
+  /**
+       * Indicates whether this agent can be directly called.
+       *
+       * MemoryAgent is designed to be used as a supporting component rather than
+       * being directly invoked for processing, so this returns false.
+       */
+  get isCallable(): boolean;
+  /**
+       * The standard message processing method required by the Agent interface.
+       *
+       * MemoryAgent doesn't directly process messages like other agents, so this method
+       * throws an error when called. Use the specialized retrieve() and record() methods instead.
+       */
+  process(_input: Message, _options: AgentInvokeOptions): Promise<Message>;
+  /**
+       * Retrieves memories based on the provided input criteria.
+       *
+       * Delegates the actual retrieval operation to the configured retriever agent.
+       *
+       * @param input - The retrieval parameters (can include search terms, limits, etc.)
+       * @param context - The execution context
+       * @returns A promise resolving to the retrieved memories
+       * @throws Error - If no retriever has been initialized
+       */
+  retrieve(input: MemoryRetrieverInput, context: Context): Promise<MemoryRetrieverOutput>;
+  /**
+       * Records new memories based on the provided input content.
+       *
+       * Delegates the actual recording operation to the configured recorder agent.
+       *
+       * @param input - The content to be recorded as memories
+       * @param context - The execution context
+       * @returns A promise resolving to the recorded memories
+       * @throws Error - If no recorder has been initialized
+       */
+  record(input: MemoryRecorderInput, context: Context): Promise<MemoryRecorderOutput>;
+  onMessage({
+    role,
+    source,
+    message,
+    context
+  }: MessagePayload): Promise<void>;
+}
+//#endregion
+export { Memory, MemoryAgent, MemoryAgentOptions, newMemoryId };
+//# sourceMappingURL=memory.d.cts.map

package/dist/memory/memory.d.cts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory.d.cts","names":[],"sources":["../../src/memory/memory.ts"],"mappings":";;;;;;;UAwBiB,MAAA;EAAA,EAAA;EAAA,SAAA;EAAA,OAAA;EAAA,SAAA;AAAA;AAAA,cAOJ,WAAA;AAAA,UAEI,kBAAA,SACP,OAAA,CAAQ,IAAA,CAAK,WAAA,kBACnB,IAAA,CAAK,YAAA;EAAA,QAAA,GACI,cAAA,GAAiB,qBAAA,cAAmC,qBAAA;EAAA,SAAA,GACnD,eAAA,GAAkB,sBAAA,cAAoC,sBAAA;AAAA;AAAA;;;;AAWpE;;;;AAXoE,cAWvD,WAAA,SAAoB,KAAA;EAAA,GAAA;EAAA;;;EAAA,YAAA,OAAA,EAMV,kBAAA;EAAA,QAAA,UAAA;EAAA;;;;;;EAAA,IAAA,UAAA,GAoCJ,eAAA;EAAA,IAAA,UAAA,KAAA,EAII,eAAA;EAAA,QAAA,SAAA;EAAA;;;;;;EAAA,IAAA,SAAA,GAeL,cAAA;EAAA,IAAA,SAAA,KAAA,EAII,cAAA;EAAA;;;;;;EAAA,UAAA;EAAA;;;;;;EAAA,IAAA,WAAA;EAAA;;;;;;EAAA,QAAA,MAAA,EA+BE,OAAA,EAAA,QAAA,EAAmB,kBAAA,GAAqB,OAAA,CAAQ,OAAA;EAAA;;;;;;;;;;EAAA,SAAA,KAAA,EAchD,oBAAA,EAAA,OAAA,EAA+B,OAAA,GAAU,OAAA,CAAQ,qBAAA;EAAA;;;;;;;;;;EAAA,OAAA,KAAA,EAenD,mBAAA,EAAA,OAAA,EAA8B,OAAA,GAAU,OAAA,CAAQ,oBAAA;EAAA,UAAA;IAAA,IAAA;IAAA,MAAA;IAAA,OAAA;IAAA;EAAA,GAKP,cAAA,GAAiB,OAAA;AAAA"}