@aigne/core 1.71.0 → 1.72.0-beta.1

package/lib/cjs/prompt/skills/afs/read.js

@@ -21,7 +21,7 @@ class AFSReadAgent extends agent_js_1.Agent {
                 tool: zod_1.z.string(),
                 path: zod_1.z.string(),
                 withLineNumbers: zod_1.z.boolean().optional(),
-                result: zod_1.z.custom().optional(),
+                data: zod_1.z.custom().optional(),
                 message: zod_1.z.string().optional(),
             }),
         });
@@ -30,7 +30,7 @@ class AFSReadAgent extends agent_js_1.Agent {
         if (!this.afs)
             throw new Error("AFS is not configured for this agent.");
         const result = await this.afs.read(input.path);
-        let content = result.result?.content;
+        let content = result.data?.content;
         if (input.withLineNumbers && typeof content === "string") {
             content = content
                 .split("\n")
@@ -43,8 +43,8 @@ class AFSReadAgent extends agent_js_1.Agent {
             path: input.path,
             withLineNumbers: input.withLineNumbers,
             ...result,
-            result: result.result && {
-                ...result.result,
+            data: result.data && {
+                ...result.data,
                 content,
             },
         };

package/lib/cjs/prompt/skills/afs/search.d.ts

@@ -11,7 +11,7 @@ export interface AFSSearchOutput extends Message {
     path: string;
     query: string;
     options?: AFSSearchOptions;
-    list: AFSEntry[];
+    data: AFSEntry[];
     message?: string;
 }
 export interface AFSSearchAgentOptions extends AgentOptions<AFSSearchInput, AFSSearchOutput> {

package/lib/cjs/prompt/skills/afs/search.js

@@ -33,7 +33,7 @@ class AFSSearchAgent extends agent_js_1.Agent {
                     caseSensitive: zod_1.z.boolean().optional(),
                 })
                     .optional(),
-                list: zod_1.z.array(zod_1.z.custom()),
+                data: zod_1.z.array(zod_1.z.custom()),
                 message: zod_1.z.string().optional(),
             }),
         });

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

@@ -1,4 +1,4 @@
-import { AFS, type AFSEntry, type AFSListOptions, type AFSModule, type AFSOptions, type AFSSearchOptions } from "@aigne/afs";
+import { AFS, type AFSExecOptions, type AFSExecResult, type AFSListOptions, type AFSListResult, type AFSModule, type AFSOptions, type AFSReadResult, type AFSSearchOptions } from "@aigne/afs";
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
 import type * as prompts from "@inquirer/prompts";
 import { type ZodObject, type ZodType } from "zod";
@@ -573,23 +573,10 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
     /** For AFSModule interface **/
     private agentToAFSEntry;
     private findAgentByAFSPath;
-    list(_path: string, _options?: AFSListOptions): Promise<{
-        list: AFSEntry[];
-        message?: string;
-    }>;
-    read(path: string): Promise<{
-        result?: AFSEntry;
-        message?: string;
-    }>;
-    search(path: string, _query: string, options?: AFSSearchOptions): Promise<{
-        list: AFSEntry[];
-        message?: string;
-    }>;
-    exec(path: string, args: Record<string, any>, options: {
-        context: Context;
-    }): Promise<{
-        result: Record<string, any>;
-    }>;
+    list(_path: string, _options?: AFSListOptions): Promise<AFSListResult>;
+    read(path: string): Promise<AFSReadResult>;
+    search(path: string, _query: string, options?: AFSSearchOptions): Promise<AFSListResult>;
+    exec(path: string, args: Record<string, any>, options: AFSExecOptions): Promise<AFSExecResult>;
 }
 export type AgentInput<T extends Agent> = T extends Agent<infer I, any> ? I : never;
 export type AgentOutput<T extends Agent> = T extends Agent<any, infer O> ? O : never;

package/lib/dts/loader/agent-yaml.d.ts

@@ -25,6 +25,26 @@ export type AFSModuleSchema = string | {
     module: string;
     options?: Record<string, any>;
 };
+export interface AFSContextPresetSchema {
+    view?: string;
+    select?: {
+        agent: NestAgentSchema;
+    };
+    per?: {
+        agent: NestAgentSchema;
+    };
+    dedupe?: {
+        agent: NestAgentSchema;
+    };
+}
+export interface AFSContextSchema {
+    search?: {
+        presets?: Record<string, AFSContextPresetSchema>;
+    };
+    list?: {
+        presets?: Record<string, AFSContextPresetSchema>;
+    };
+}
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
@@ -42,8 +62,9 @@ export interface BaseAgentSchema {
         provider: string;
         subscribeTopic?: string[];
     };
-    afs?: boolean | (Omit<AFSOptions, "modules"> & {
+    afs?: boolean | (Omit<AFSOptions, "modules" | "context"> & {
         modules?: AFSModuleSchema[];
+        context?: AFSContextSchema;
     });
     shareAFS?: boolean;
 }

package/lib/dts/prompt/context/afs/history.d.ts

@@ -0,0 +1,5 @@
+import type { Agent } from "../../../agents/agent.js";
+export declare function getHistories(agent: Agent): Promise<{
+    role: "user" | "agent";
+    content: unknown;
+}[]>;

package/lib/dts/prompt/context/afs/index.d.ts

@@ -0,0 +1,20 @@
+import type { AFSRootListOptions, AFSRootSearchOptions } from "@aigne/afs";
+import type { Agent, AgentInvokeOptions } from "../../../agents/agent.js";
+export declare function createAFSContext(agent?: Agent<any, any>, context?: AgentInvokeOptions["context"]): {
+    readonly enabled: boolean;
+    description: string;
+    readonly modules: Promise<Pick<{
+        name: string;
+        path: string;
+        description?: string;
+        module: import("@aigne/afs").AFSModule;
+    }, "path" | "name" | "description">[]> | undefined;
+    readonly histories: Promise<{
+        role: "user" | "agent";
+        content: unknown;
+    }[]>;
+    readonly skills: never[] | Promise<Pick<Agent<any, any>, "name" | "description">[]>;
+    list(path: string, options?: AFSRootListOptions): Promise<any>;
+    read(path: string): Promise<import("@aigne/afs").AFSEntry<any> | undefined>;
+    search(path: string, query: string, options?: AFSRootSearchOptions): Promise<any>;
+};

package/lib/dts/prompt/context/index.d.ts

@@ -0,0 +1,31 @@
+import type { PromptBuildOptions } from "../prompt-builder.js";
+export declare function createPromptBuilderContext(options: PromptBuildOptions): {
+    $afs: {
+        readonly enabled: boolean;
+        description: string;
+        readonly modules: Promise<Pick<{
+            name: string;
+            path: string;
+            description?: string;
+            module: import("@aigne/afs").AFSModule;
+        }, "path" | "name" | "description">[]> | undefined;
+        readonly histories: Promise<{
+            role: "user" | "agent";
+            content: unknown;
+        }[]>;
+        readonly skills: never[] | Promise<Pick<import("../../index.js").Agent<any, any>, "name" | "description">[]>;
+        list(path: string, options?: import("@aigne/afs").AFSRootListOptions): Promise<any>;
+        read(path: string): Promise<import("@aigne/afs").AFSEntry<any> | undefined>;
+        search(path: string, query: string, options?: import("@aigne/afs").AFSRootSearchOptions): Promise<any>;
+    };
+    $agent: {
+        readonly skills: Pick<import("../../index.js").Agent<any, any>, "name" | "description">[] | undefined;
+    };
+    $meta?: {
+        usage?: import("../../index.js").ContextUsage;
+        [key: string]: unknown;
+    };
+    userId?: string;
+    sessionId?: string;
+    userContext: import("../../index.js").UserContext | undefined;
+};

package/lib/dts/prompt/skills/afs/edit.d.ts

@@ -14,7 +14,7 @@ export interface AFSEditOutput extends Message {
     tool: string;
     path: string;
     message: string;
-    content: string;
+    data: string;
 }
 export interface AFSEditAgentOptions extends AgentOptions<AFSEditInput, AFSEditOutput> {
     afs: NonNullable<AgentOptions<AFSEditInput, AFSEditOutput>["afs"]>;

package/lib/dts/prompt/skills/afs/exec.d.ts

@@ -4,7 +4,7 @@ export interface AFSExecInput extends Message {
     args: string;
 }
 export interface AFSExecOutput extends Message {
-    result: Record<string, any>;
+    data: Record<string, any>;
 }
 export interface AFSExecAgentOptions extends AgentOptions<AFSExecInput, AFSExecOutput> {
     afs: NonNullable<AgentOptions<AFSExecInput, AFSExecOutput>["afs"]>;

package/lib/dts/prompt/skills/afs/list.d.ts

@@ -10,7 +10,7 @@ export interface AFSListOutput extends Message {
     path: string;
     options?: AFSListOptions;
     message?: string;
-    result: string;
+    data?: unknown;
 }
 export interface AFSListAgentOptions extends AgentOptions<AFSListInput, AFSListOutput> {
     afs: NonNullable<AgentOptions<AFSListInput, AFSListOutput>["afs"]>;
@@ -18,5 +18,4 @@ export interface AFSListAgentOptions extends AgentOptions<AFSListInput, AFSListO
 export declare class AFSListAgent extends Agent<AFSListInput, AFSListOutput> {
     constructor(options: AFSListAgentOptions);
     process(input: AFSListInput, _options: AgentInvokeOptions): Promise<AFSListOutput>;
-    private buildTreeView;
 }

package/lib/dts/prompt/skills/afs/read.d.ts

@@ -9,7 +9,7 @@ export interface AFSReadOutput extends Message {
     tool: string;
     path: string;
     withLineNumbers?: boolean;
-    result?: AFSEntry;
+    data?: AFSEntry;
     message?: string;
 }
 export interface AFSReadAgentOptions extends AgentOptions<AFSReadInput, AFSReadOutput> {

package/lib/dts/prompt/skills/afs/search.d.ts

@@ -11,7 +11,7 @@ export interface AFSSearchOutput extends Message {
     path: string;
     query: string;
     options?: AFSSearchOptions;
-    list: AFSEntry[];
+    data: AFSEntry[];
     message?: string;
 }
 export interface AFSSearchAgentOptions extends AgentOptions<AFSSearchInput, AFSSearchOutput> {

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

@@ -1,4 +1,4 @@
-import { AFS, type AFSEntry, type AFSListOptions, type AFSModule, type AFSOptions, type AFSSearchOptions } from "@aigne/afs";
+import { AFS, type AFSExecOptions, type AFSExecResult, type AFSListOptions, type AFSListResult, type AFSModule, type AFSOptions, type AFSReadResult, type AFSSearchOptions } from "@aigne/afs";
 import { nodejs } from "@aigne/platform-helpers/nodejs/index.js";
 import type * as prompts from "@inquirer/prompts";
 import { type ZodObject, type ZodType } from "zod";
@@ -573,23 +573,10 @@ export declare abstract class Agent<I extends Message = any, O extends Message =
     /** For AFSModule interface **/
     private agentToAFSEntry;
     private findAgentByAFSPath;
-    list(_path: string, _options?: AFSListOptions): Promise<{
-        list: AFSEntry[];
-        message?: string;
-    }>;
-    read(path: string): Promise<{
-        result?: AFSEntry;
-        message?: string;
-    }>;
-    search(path: string, _query: string, options?: AFSSearchOptions): Promise<{
-        list: AFSEntry[];
-        message?: string;
-    }>;
-    exec(path: string, args: Record<string, any>, options: {
-        context: Context;
-    }): Promise<{
-        result: Record<string, any>;
-    }>;
+    list(_path: string, _options?: AFSListOptions): Promise<AFSListResult>;
+    read(path: string): Promise<AFSReadResult>;
+    search(path: string, _query: string, options?: AFSSearchOptions): Promise<AFSListResult>;
+    exec(path: string, args: Record<string, any>, options: AFSExecOptions): Promise<AFSExecResult>;
 }
 export type AgentInput<T extends Agent> = T extends Agent<infer I, any> ? I : never;
 export type AgentOutput<T extends Agent> = T extends Agent<any, infer O> ? O : never;

package/lib/esm/agents/agent.js

@@ -733,7 +733,7 @@ export class Agent {
     // TODO: support list skills inside agent path, and use options to filter skills
     async list(_path, _options) {
         const agents = [this, ...this.skills];
-        return { list: agents.map((agent) => this.agentToAFSEntry(agent)) };
+        return { data: agents.map((agent) => this.agentToAFSEntry(agent)) };
     }
     async read(path) {
         const agent = this.findAgentByAFSPath(path);
@@ -741,7 +741,7 @@ export class Agent {
             return { message: `Agent not found at path: ${path}` };
         }
         return {
-            result: this.agentToAFSEntry(agent),
+            data: this.agentToAFSEntry(agent),
         };
     }
     // TODO: implement search inside agent skills
@@ -752,7 +752,7 @@ export class Agent {
         const agent = this.findAgentByAFSPath(path);
         if (!agent)
             throw new Error(`Agent not found at path: ${path}`);
-        return { result: await options.context.invoke(agent, args) };
+        return { data: await options.context.invoke(agent, args) };
     }
 }
 /**

package/lib/esm/loader/agent-yaml.d.ts

@@ -25,6 +25,26 @@ export type AFSModuleSchema = string | {
     module: string;
     options?: Record<string, any>;
 };
+export interface AFSContextPresetSchema {
+    view?: string;
+    select?: {
+        agent: NestAgentSchema;
+    };
+    per?: {
+        agent: NestAgentSchema;
+    };
+    dedupe?: {
+        agent: NestAgentSchema;
+    };
+}
+export interface AFSContextSchema {
+    search?: {
+        presets?: Record<string, AFSContextPresetSchema>;
+    };
+    list?: {
+        presets?: Record<string, AFSContextPresetSchema>;
+    };
+}
 export interface BaseAgentSchema {
     name?: string;
     description?: string;
@@ -42,8 +62,9 @@ export interface BaseAgentSchema {
         provider: string;
         subscribeTopic?: string[];
     };
-    afs?: boolean | (Omit<AFSOptions, "modules"> & {
+    afs?: boolean | (Omit<AFSOptions, "modules" | "context"> & {
         modules?: AFSModuleSchema[];
+        context?: AFSContextSchema;
     });
     shareAFS?: boolean;
 }

package/lib/esm/loader/agent-yaml.js

@@ -80,6 +80,20 @@ export const getAgentSchema = ({ filepath, options, }) => {
             onSkillEnd: optionalize(nestAgentSchema),
             onHandoff: optionalize(nestAgentSchema),
         }));
+        const afsContextPresetsSchema = z.object({
+            presets: optionalize(z.record(z.string(), z.object({
+                view: optionalize(z.string()),
+                select: optionalize(z.object({
+                    agent: nestAgentSchema,
+                })),
+                per: optionalize(z.object({
+                    agent: nestAgentSchema,
+                })),
+                dedupe: optionalize(z.object({
+                    agent: nestAgentSchema,
+                })),
+            }))),
+        });
         const baseAgentSchema = z.object({
             name: optionalize(z.string()),
             alias: optionalize(z.array(z.string())),
@@ -111,6 +125,10 @@ export const getAgentSchema = ({ filepath, options, }) => {
                             options: optionalize(z.record(z.any())),
                         })),
                     ]))),
+                    context: optionalize(z.object({
+                        search: optionalize(afsContextPresetsSchema),
+                        list: optionalize(afsContextPresetsSchema),
+                    })),
                 })),
             ])),
             shareAFS: optionalize(z.boolean()),

package/lib/esm/loader/index.js

@@ -123,7 +123,32 @@ export async function parseAgent(path, agent, options, agentOptions) {
         afs = new AFS();
     }
     else if (agent.afs) {
-        afs = new AFS();
+        afs = new AFS({});
+        const loadAFSContextPresets = async (presets) => {
+            return Object.fromEntries(await Promise.all(Object.entries(presets).map(async ([key, value]) => {
+                const [select, per, dedupe] = await Promise.all([value.select, value.per, value.dedupe].map(async (item) => {
+                    if (!item?.agent)
+                        return undefined;
+                    const agent = await loadNestAgent(path, item.agent, options, { afs });
+                    return {
+                        invoke: (input, options) => options.context.invoke(agent, input, {
+                            ...options,
+                            streaming: false,
+                        }),
+                    };
+                }));
+                return [key, { ...value, select, per, dedupe }];
+            })));
+        };
+        const context = {
+            search: {
+                presets: await loadAFSContextPresets(agent.afs.context?.search?.presets || {}),
+            },
+            list: {
+                presets: await loadAFSContextPresets(agent.afs.context?.list?.presets || {}),
+            },
+        };
+        afs.options.context = context;
         for (const m of agent.afs.modules || []) {
             const moduleName = typeof m === "string" ? m : m.module;
             const mod = options?.afs?.availableModules?.find((mod) => mod.module === moduleName || mod.alias?.includes(moduleName));

package/lib/esm/prompt/context/afs/history.d.ts

@@ -0,0 +1,5 @@
+import type { Agent } from "../../../agents/agent.js";
+export declare function getHistories(agent: Agent): Promise<{
+    role: "user" | "agent";
+    content: unknown;
+}[]>;

package/lib/esm/prompt/context/afs/history.js

@@ -0,0 +1,29 @@
+import { AFSHistory } from "@aigne/afs-history";
+import { isNonNullable } from "../../../utils/type-utils.js";
+export async function getHistories(agent) {
+    const afs = agent?.afs;
+    if (!afs)
+        return [];
+    const historyModule = (await afs.listModules()).find((m) => m.module instanceof AFSHistory);
+    if (!historyModule)
+        return [];
+    const history = (await afs.list(historyModule.path, {
+        limit: agent.historyConfig?.maxItems || 10,
+        orderBy: [["createdAt", "desc"]],
+    })).data;
+    return history
+        .reverse()
+        .map((i) => {
+        if (!i.content)
+            return;
+        const { input, output } = i.content;
+        if (!input || !output)
+            return;
+        return [
+            { role: "user", content: input },
+            { role: "agent", content: output },
+        ];
+    })
+        .filter(isNonNullable)
+        .flat();
+}

package/lib/esm/prompt/context/afs/index.d.ts

@@ -0,0 +1,20 @@
+import type { AFSRootListOptions, AFSRootSearchOptions } from "@aigne/afs";
+import type { Agent, AgentInvokeOptions } from "../../../agents/agent.js";
+export declare function createAFSContext(agent?: Agent<any, any>, context?: AgentInvokeOptions["context"]): {
+    readonly enabled: boolean;
+    description: string;
+    readonly modules: Promise<Pick<{
+        name: string;
+        path: string;
+        description?: string;
+        module: import("@aigne/afs").AFSModule;
+    }, "path" | "name" | "description">[]> | undefined;
+    readonly histories: Promise<{
+        role: "user" | "agent";
+        content: unknown;
+    }[]>;
+    readonly skills: never[] | Promise<Pick<Agent<any, any>, "name" | "description">[]>;
+    list(path: string, options?: AFSRootListOptions): Promise<any>;
+    read(path: string): Promise<import("@aigne/afs").AFSEntry<any> | undefined>;
+    search(path: string, query: string, options?: AFSRootSearchOptions): Promise<any>;
+};

package/lib/esm/prompt/context/afs/index.js

@@ -0,0 +1,45 @@
+import { pick } from "../../../utils/type-utils.js";
+import { AFS_DESCRIPTION_PROMPT_TEMPLATE } from "../../prompts/afs-builtin-prompt.js";
+import { getAFSSkills } from "../../skills/afs/index.js";
+import { getHistories } from "./history.js";
+export function createAFSContext(agent, context) {
+    const afs = agent?.afs;
+    return {
+        get enabled() {
+            return !!afs;
+        },
+        description: AFS_DESCRIPTION_PROMPT_TEMPLATE,
+        get modules() {
+            return afs
+                ?.listModules()
+                .then((list) => list.map((i) => pick(i, ["name", "path", "description"])));
+        },
+        get histories() {
+            if (!agent)
+                return Promise.resolve([]);
+            return getHistories(agent);
+        },
+        get skills() {
+            const afs = agent?.afs;
+            if (!afs)
+                return [];
+            return getAFSSkills(afs).then((skills) => skills.map((s) => pick(s, ["name", "description"])));
+        },
+        async list(path, options) {
+            if (!afs)
+                throw new Error("AFS is not configured for this agent.");
+            return (await afs.list(path, { ...options, context, format: options?.format || "tree" }))
+                .data;
+        },
+        async read(path) {
+            if (!afs)
+                throw new Error("AFS is not configured for this agent.");
+            return (await afs.read(path)).data;
+        },
+        async search(path, query, options = {}) {
+            if (!afs)
+                throw new Error("AFS is not configured for this agent.");
+            return (await afs.search(path, query, { ...options, context })).data;
+        },
+    };
+}

package/lib/esm/prompt/context/index.d.ts

@@ -0,0 +1,31 @@
+import type { PromptBuildOptions } from "../prompt-builder.js";
+export declare function createPromptBuilderContext(options: PromptBuildOptions): {
+    $afs: {
+        readonly enabled: boolean;
+        description: string;
+        readonly modules: Promise<Pick<{
+            name: string;
+            path: string;
+            description?: string;
+            module: import("@aigne/afs").AFSModule;
+        }, "path" | "name" | "description">[]> | undefined;
+        readonly histories: Promise<{
+            role: "user" | "agent";
+            content: unknown;
+        }[]>;
+        readonly skills: never[] | Promise<Pick<import("../../index.js").Agent<any, any>, "name" | "description">[]>;
+        list(path: string, options?: import("@aigne/afs").AFSRootListOptions): Promise<any>;
+        read(path: string): Promise<import("@aigne/afs").AFSEntry<any> | undefined>;
+        search(path: string, query: string, options?: import("@aigne/afs").AFSRootSearchOptions): Promise<any>;
+    };
+    $agent: {
+        readonly skills: Pick<import("../../index.js").Agent<any, any>, "name" | "description">[] | undefined;
+    };
+    $meta?: {
+        usage?: import("../../index.js").ContextUsage;
+        [key: string]: unknown;
+    };
+    userId?: string;
+    sessionId?: string;
+    userContext: import("../../index.js").UserContext | undefined;
+};

package/lib/esm/prompt/context/index.js

@@ -0,0 +1,15 @@
+import { pick } from "../../utils/type-utils.js";
+import { createAFSContext } from "./afs/index.js";
+export function createPromptBuilderContext(options) {
+    return {
+        userContext: options.context?.userContext,
+        ...options.context?.userContext,
+        ...options.input,
+        $afs: createAFSContext(options.agent, options.context),
+        $agent: {
+            get skills() {
+                return options.agent?.skills.map((s) => pick(s, ["name", "description"]));
+            },
+        },
+    };
+}

package/lib/esm/prompt/prompt-builder.js

@@ -8,8 +8,9 @@ import { DEFAULT_OUTPUT_FILE_KEY, DEFAULT_OUTPUT_KEY } from "../agents/ai-agent.
 import { fileUnionContentsSchema } from "../agents/model.js";
 import { optionalize } from "../loader/schema.js";
 import { outputSchemaToResponseFormatSchema } from "../utils/json-schema.js";
-import { checkArguments, flat, isNonNullable, isRecord, partition, pick, unique, } from "../utils/type-utils.js";
-import { AFS_DESCRIPTION_PROMPT_TEMPLATE, AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE, getAFSSystemPrompt, } from "./prompts/afs-builtin-prompt.js";
+import { checkArguments, flat, isNonNullable, isRecord, partition, unique, } from "../utils/type-utils.js";
+import { createPromptBuilderContext } from "./context/index.js";
+import { AFS_EXECUTABLE_TOOLS_PROMPT_TEMPLATE, getAFSSystemPrompt, } from "./prompts/afs-builtin-prompt.js";
 import { MEMORY_MESSAGE_TEMPLATE } from "./prompts/memory-message-template.js";
 import { STRUCTURED_STREAM_INSTRUCTIONS } from "./prompts/structured-stream-instructions.js";
 import { getAFSSkills } from "./skills/afs/index.js";
@@ -92,37 +93,7 @@ export class PromptBuilder {
         };
     }
     getTemplateVariables(options) {
-        const self = this;
-        return {
-            userContext: options.context?.userContext,
-            ...options.context?.userContext,
-            ...options.input,
-            $afs: {
-                get enabled() {
-                    return !!options.agent?.afs;
-                },
-                description: AFS_DESCRIPTION_PROMPT_TEMPLATE,
-                get modules() {
-                    return options.agent?.afs
-                        ?.listModules()
-                        .then((list) => list.map((i) => pick(i, ["name", "path", "description"])));
-                },
-                get histories() {
-                    return self.getHistories(options);
-                },
-                get skills() {
-                    const afs = options.agent?.afs;
-                    if (!afs)
-                        return [];
-                    return getAFSSkills(afs).then((skills) => skills.map((s) => pick(s, ["name", "description"])));
-                },
-            },
-            $agent: {
-                get skills() {
-                    return options.agent?.skills.map((s) => pick(s, ["name", "description"]));
-                },
-            },
-        };
+        return createPromptBuilderContext(options);
     }
     async buildMessages(options) {
         const { input } = options;
@@ -151,12 +122,12 @@ export class PromptBuilder {
                     limit: options.agent?.maxRetrieveMemoryCount || 10,
                     orderBy: [["createdAt", "desc"]],
                 });
-                memories.push(...history.list
+                memories.push(...history.data
                     .reverse()
                     .filter((i) => isNonNullable(i.content)));
                 if (message) {
-                    const result = await afs.search("/", message);
-                    const ms = result.list
+                    const result = (await afs.search("/", message)).data;
+                    const ms = result
                         .map((entry) => {
                         if (entry.metadata?.execute)
                             return null;
@@ -170,7 +141,7 @@ export class PromptBuilder {
                     })
                         .filter(isNonNullable);
                     memories.push(...ms);
-                    const executable = result.list.filter((i) => !!i.metadata?.execute);
+                    const executable = result.filter((i) => !!i.metadata?.execute);
                     if (executable.length) {
                         messages.push({
                             role: "system",
@@ -229,10 +200,10 @@ export class PromptBuilder {
         const historyModule = (await afs.listModules()).find((m) => m.module instanceof AFSHistory);
         if (!historyModule)
             return [];
-        const { list: history } = await afs.list(historyModule.path, {
+        const history = (await afs.list(historyModule.path, {
             limit: agent.historyConfig?.maxItems || 10,
             orderBy: [["createdAt", "desc"]],
-        });
+        })).data;
         return history
             .reverse()
             .map((i) => {

package/lib/esm/prompt/skills/afs/edit.d.ts

@@ -14,7 +14,7 @@ export interface AFSEditOutput extends Message {
     tool: string;
     path: string;
     message: string;
-    content: string;
+    data: string;
 }
 export interface AFSEditAgentOptions extends AgentOptions<AFSEditInput, AFSEditOutput> {
     afs: NonNullable<AgentOptions<AFSEditInput, AFSEditOutput>["afs"]>;