pulse-coder-engine 0.0.1-alpha.10 → 0.0.1-alpha.11

package/dist/{index-C7fdydbL.d.cts → index-BeWyLfso.d.cts}

@@ -1,5 +1,5 @@
-import { FlexibleSchema, ModelMessage, LanguageModel, Tool as Tool$1 } from 'ai';
 import { EventEmitter } from 'events';
+import { ModelMessage, FlexibleSchema, LanguageModel } from 'ai';
 
 /**
  * Custom LLM provider factory - receives a model name and returns a LanguageModel.
@@ -77,6 +77,58 @@ interface ILogger {
     error(message: string, error?: Error, meta?: Record<string, unknown>): void;
 }
 
+/** Convenience: a value that may be wrapped in a Promise. */
+type Promisable<T> = T | Promise<T>;
+interface BeforeRunInput {
+    context: Context;
+    systemPrompt?: SystemPromptOption;
+    tools: Record<string, any>;
+}
+interface BeforeRunResult {
+    systemPrompt?: SystemPromptOption;
+    tools?: Record<string, any>;
+}
+type BeforeLLMCallInput = BeforeRunInput;
+type BeforeLLMCallResult = BeforeRunResult;
+interface AfterRunInput {
+    context: Context;
+    result: string;
+}
+interface AfterLLMCallInput {
+    context: Context;
+    finishReason: string;
+    text: string;
+}
+interface BeforeToolCallInput {
+    name: string;
+    input: any;
+}
+interface BeforeToolCallResult {
+    input?: any;
+}
+interface AfterToolCallInput {
+    name: string;
+    input: any;
+    output: any;
+}
+interface AfterToolCallResult {
+    output?: any;
+}
+interface EngineHookMap {
+    /** Fires once at the start of Engine.run(), before entering the loop. */
+    beforeRun: (input: BeforeRunInput) => Promisable<BeforeRunResult | void>;
+    /** Fires once when Engine.run() finishes. Read-only. */
+    afterRun: (input: AfterRunInput) => Promisable<void>;
+    /** Fires before each LLM call inside the loop (including retries after tool-calls). */
+    beforeLLMCall: (input: BeforeLLMCallInput) => Promisable<BeforeLLMCallResult | void>;
+    /** Fires after each LLM call completes. Read-only. */
+    afterLLMCall: (input: AfterLLMCallInput) => Promisable<void>;
+    /** Fires before each individual tool execution. Can modify input or throw to abort. */
+    beforeToolCall: (input: BeforeToolCallInput) => Promisable<BeforeToolCallResult | void>;
+    /** Fires after each individual tool execution. Can modify output. */
+    afterToolCall: (input: AfterToolCallInput) => Promisable<AfterToolCallResult | void>;
+}
+type EngineHookName = keyof EngineHookMap;
 interface EnginePlugin {
     name: string;
     version: string;
@@ -86,28 +138,13 @@ interface EnginePlugin {
     afterInitialize?(context: EnginePluginContext): Promise<void>;
     destroy?(context: EnginePluginContext): Promise<void>;
 }
-interface EngineRunHookInput {
-    context: Context;
-    messages: ModelMessage[];
-    tools: Record<string, Tool$1>;
-    systemPrompt?: SystemPromptOption;
-    hooks?: ToolHooks;
-}
-interface EngineRunHookResult {
-    systemPrompt?: SystemPromptOption;
-    hooks?: ToolHooks;
-}
-type EngineRunHook = (input: EngineRunHookInput) => Promise<EngineRunHookResult | void> | EngineRunHookResult | void;
 interface EnginePluginContext {
-    registerTool(name: string, tool: Tool$1): void;
-    registerTools(tools: Record<string, Tool$1>): void;
-    getTool(name: string): Tool$1 | undefined;
+    registerTool(name: string, tool: any): void;
+    registerTools(tools: Record<string, any>): void;
+    getTool(name: string): any;
     /** Returns a snapshot of all tools registered by plugins so far. */
-    getTools(): Record<string, Tool$1>;
-    registerRunHook(name: string, hook: EngineRunHook): void;
-    getRunHook(name: string): EngineRunHook | undefined;
-    registerProtocol(name: string, handler: ProtocolHandler): void;
-    getProtocol(name: string): ProtocolHandler | undefined;
+    getTools(): Record<string, any>;
+    registerHook<K extends EngineHookName>(hookName: K, handler: EngineHookMap[K]): void;
     registerService<T>(name: string, service: T): void;
     getService<T>(name: string): T | undefined;
     getConfig<T>(key: string): T | undefined;
@@ -120,10 +157,6 @@ interface EnginePluginContext {
         error(message: string, error?: Error, meta?: any): void;
     };
 }
-interface ProtocolHandler {
-    name: string;
-    handle(message: any): Promise<any>;
-}
 interface EnginePluginLoadOptions {
     plugins?: EnginePlugin[];
     dirs?: string[];
@@ -241,7 +274,6 @@ interface PlanModeService {
     getModePolicy(mode?: PlanMode): ModePolicy;
     getToolMetadata(toolNames: string[]): ToolMeta[];
     buildPromptAppend(toolNames: string[], transition?: PlanModeTransitionResult): string;
-    applyHooks(baseHooks?: ToolHooks): ToolHooks;
     getEvents(limit?: number): PlanModeEvent[];
 }
 type LoggerLike = EnginePluginContext['logger'];
@@ -258,9 +290,8 @@ declare class BuiltInPlanModeService implements PlanModeService {
     getModePolicy(mode?: PlanMode): ModePolicy;
     getToolMetadata(toolNames: string[]): ToolMeta[];
     buildPromptAppend(toolNames: string[], transition?: PlanModeTransitionResult): string;
-    applyHooks(baseHooks?: ToolHooks): ToolHooks;
     getEvents(limit?: number): PlanModeEvent[];
-    private observePotentialPolicyViolation;
+    observePotentialPolicyViolation(toolName: string, input: unknown): void;
     private inferToolMeta;
     private getLatestUserText;
     private messageContentToText;
@@ -274,4 +305,4 @@ declare const builtInPlanModePlugin: EnginePlugin;
  */
 declare const builtInPlugins: (EnginePlugin | SubAgentPlugin)[];
 
-export { BuiltInPlanModeService as B, type ClarificationRequest as C, type EnginePluginLoadOptions as E, type ILogger as I, type LLMProviderFactory as L, type ModePolicy as M, type PlanMode as P, type SystemPromptOption as S, type Tool as T, type ToolHooks as a, type Context as b, type ToolExecutionContext as c, type EngineRunHook as d, BuiltInSkillRegistry as e, type EnginePlugin as f, type EnginePluginContext as g, type PlanIntentLabel as h, type PlanModeEvent as i, type PlanModeEventName as j, type PlanModeService as k, type PlanModeTransitionResult as l, type ToolCategory as m, type ToolMeta as n, type ToolRisk as o, builtInMCPPlugin as p, builtInPlanModePlugin as q, builtInPlugins as r, builtInSkillsPlugin as s, SubAgentPlugin as t };
+export { type AfterLLMCallInput as A, type BeforeLLMCallInput as B, type ClarificationRequest as C, builtInPlanModePlugin as D, type EngineHookMap as E, builtInPlugins as F, builtInSkillsPlugin as G, SubAgentPlugin as H, type ILogger as I, type LLMProviderFactory as L, type ModePolicy as M, type PlanMode as P, type SystemPromptOption as S, type Tool as T, type Context as a, type ToolExecutionContext as b, type EnginePluginLoadOptions as c, type ToolHooks as d, type EngineHookName as e, type AfterRunInput as f, type AfterToolCallInput as g, type AfterToolCallResult as h, type BeforeLLMCallResult as i, type BeforeRunInput as j, type BeforeRunResult as k, type BeforeToolCallInput as l, type BeforeToolCallResult as m, BuiltInPlanModeService as n, BuiltInSkillRegistry as o, type EnginePlugin as p, type EnginePluginContext as q, type PlanIntentLabel as r, type PlanModeEvent as s, type PlanModeEventName as t, type PlanModeService as u, type PlanModeTransitionResult as v, type ToolCategory as w, type ToolMeta as x, type ToolRisk as y, builtInMCPPlugin as z };

package/dist/{index-C7fdydbL.d.ts → index-BeWyLfso.d.ts}

@@ -1,5 +1,5 @@
-import { FlexibleSchema, ModelMessage, LanguageModel, Tool as Tool$1 } from 'ai';
 import { EventEmitter } from 'events';
+import { ModelMessage, FlexibleSchema, LanguageModel } from 'ai';
 
 /**
  * Custom LLM provider factory - receives a model name and returns a LanguageModel.
@@ -77,6 +77,58 @@ interface ILogger {
     error(message: string, error?: Error, meta?: Record<string, unknown>): void;
 }
 
+/** Convenience: a value that may be wrapped in a Promise. */
+type Promisable<T> = T | Promise<T>;
+interface BeforeRunInput {
+    context: Context;
+    systemPrompt?: SystemPromptOption;
+    tools: Record<string, any>;
+}
+interface BeforeRunResult {
+    systemPrompt?: SystemPromptOption;
+    tools?: Record<string, any>;
+}
+type BeforeLLMCallInput = BeforeRunInput;
+type BeforeLLMCallResult = BeforeRunResult;
+interface AfterRunInput {
+    context: Context;
+    result: string;
+}
+interface AfterLLMCallInput {
+    context: Context;
+    finishReason: string;
+    text: string;
+}
+interface BeforeToolCallInput {
+    name: string;
+    input: any;
+}
+interface BeforeToolCallResult {
+    input?: any;
+}
+interface AfterToolCallInput {
+    name: string;
+    input: any;
+    output: any;
+}
+interface AfterToolCallResult {
+    output?: any;
+}
+interface EngineHookMap {
+    /** Fires once at the start of Engine.run(), before entering the loop. */
+    beforeRun: (input: BeforeRunInput) => Promisable<BeforeRunResult | void>;
+    /** Fires once when Engine.run() finishes. Read-only. */
+    afterRun: (input: AfterRunInput) => Promisable<void>;
+    /** Fires before each LLM call inside the loop (including retries after tool-calls). */
+    beforeLLMCall: (input: BeforeLLMCallInput) => Promisable<BeforeLLMCallResult | void>;
+    /** Fires after each LLM call completes. Read-only. */
+    afterLLMCall: (input: AfterLLMCallInput) => Promisable<void>;
+    /** Fires before each individual tool execution. Can modify input or throw to abort. */
+    beforeToolCall: (input: BeforeToolCallInput) => Promisable<BeforeToolCallResult | void>;
+    /** Fires after each individual tool execution. Can modify output. */
+    afterToolCall: (input: AfterToolCallInput) => Promisable<AfterToolCallResult | void>;
+}
+type EngineHookName = keyof EngineHookMap;
 interface EnginePlugin {
     name: string;
     version: string;
@@ -86,28 +138,13 @@ interface EnginePlugin {
     afterInitialize?(context: EnginePluginContext): Promise<void>;
     destroy?(context: EnginePluginContext): Promise<void>;
 }
-interface EngineRunHookInput {
-    context: Context;
-    messages: ModelMessage[];
-    tools: Record<string, Tool$1>;
-    systemPrompt?: SystemPromptOption;
-    hooks?: ToolHooks;
-}
-interface EngineRunHookResult {
-    systemPrompt?: SystemPromptOption;
-    hooks?: ToolHooks;
-}
-type EngineRunHook = (input: EngineRunHookInput) => Promise<EngineRunHookResult | void> | EngineRunHookResult | void;
 interface EnginePluginContext {
-    registerTool(name: string, tool: Tool$1): void;
-    registerTools(tools: Record<string, Tool$1>): void;
-    getTool(name: string): Tool$1 | undefined;
+    registerTool(name: string, tool: any): void;
+    registerTools(tools: Record<string, any>): void;
+    getTool(name: string): any;
     /** Returns a snapshot of all tools registered by plugins so far. */
-    getTools(): Record<string, Tool$1>;
-    registerRunHook(name: string, hook: EngineRunHook): void;
-    getRunHook(name: string): EngineRunHook | undefined;
-    registerProtocol(name: string, handler: ProtocolHandler): void;
-    getProtocol(name: string): ProtocolHandler | undefined;
+    getTools(): Record<string, any>;
+    registerHook<K extends EngineHookName>(hookName: K, handler: EngineHookMap[K]): void;
     registerService<T>(name: string, service: T): void;
     getService<T>(name: string): T | undefined;
     getConfig<T>(key: string): T | undefined;
@@ -120,10 +157,6 @@ interface EnginePluginContext {
         error(message: string, error?: Error, meta?: any): void;
     };
 }
-interface ProtocolHandler {
-    name: string;
-    handle(message: any): Promise<any>;
-}
 interface EnginePluginLoadOptions {
     plugins?: EnginePlugin[];
     dirs?: string[];
@@ -241,7 +274,6 @@ interface PlanModeService {
     getModePolicy(mode?: PlanMode): ModePolicy;
     getToolMetadata(toolNames: string[]): ToolMeta[];
     buildPromptAppend(toolNames: string[], transition?: PlanModeTransitionResult): string;
-    applyHooks(baseHooks?: ToolHooks): ToolHooks;
     getEvents(limit?: number): PlanModeEvent[];
 }
 type LoggerLike = EnginePluginContext['logger'];
@@ -258,9 +290,8 @@ declare class BuiltInPlanModeService implements PlanModeService {
     getModePolicy(mode?: PlanMode): ModePolicy;
     getToolMetadata(toolNames: string[]): ToolMeta[];
     buildPromptAppend(toolNames: string[], transition?: PlanModeTransitionResult): string;
-    applyHooks(baseHooks?: ToolHooks): ToolHooks;
     getEvents(limit?: number): PlanModeEvent[];
-    private observePotentialPolicyViolation;
+    observePotentialPolicyViolation(toolName: string, input: unknown): void;
     private inferToolMeta;
     private getLatestUserText;
     private messageContentToText;
@@ -274,4 +305,4 @@ declare const builtInPlanModePlugin: EnginePlugin;
  */
 declare const builtInPlugins: (EnginePlugin | SubAgentPlugin)[];
 
-export { BuiltInPlanModeService as B, type ClarificationRequest as C, type EnginePluginLoadOptions as E, type ILogger as I, type LLMProviderFactory as L, type ModePolicy as M, type PlanMode as P, type SystemPromptOption as S, type Tool as T, type ToolHooks as a, type Context as b, type ToolExecutionContext as c, type EngineRunHook as d, BuiltInSkillRegistry as e, type EnginePlugin as f, type EnginePluginContext as g, type PlanIntentLabel as h, type PlanModeEvent as i, type PlanModeEventName as j, type PlanModeService as k, type PlanModeTransitionResult as l, type ToolCategory as m, type ToolMeta as n, type ToolRisk as o, builtInMCPPlugin as p, builtInPlanModePlugin as q, builtInPlugins as r, builtInSkillsPlugin as s, SubAgentPlugin as t };
+export { type AfterLLMCallInput as A, type BeforeLLMCallInput as B, type ClarificationRequest as C, builtInPlanModePlugin as D, type EngineHookMap as E, builtInPlugins as F, builtInSkillsPlugin as G, SubAgentPlugin as H, type ILogger as I, type LLMProviderFactory as L, type ModePolicy as M, type PlanMode as P, type SystemPromptOption as S, type Tool as T, type Context as a, type ToolExecutionContext as b, type EnginePluginLoadOptions as c, type ToolHooks as d, type EngineHookName as e, type AfterRunInput as f, type AfterToolCallInput as g, type AfterToolCallResult as h, type BeforeLLMCallResult as i, type BeforeRunInput as j, type BeforeRunResult as k, type BeforeToolCallInput as l, type BeforeToolCallResult as m, BuiltInPlanModeService as n, BuiltInSkillRegistry as o, type EnginePlugin as p, type EnginePluginContext as q, type PlanIntentLabel as r, type PlanModeEvent as s, type PlanModeEventName as t, type PlanModeService as u, type PlanModeTransitionResult as v, type ToolCategory as w, type ToolMeta as x, type ToolRisk as y, builtInMCPPlugin as z };

package/dist/index.cjs

@@ -415,15 +415,28 @@ var maybeCompactContext = async (context, options) => {
 };
 
 // src/core/loop.ts
-function applyToolHooks(tools, hooks) {
+function wrapToolsWithHooks(tools, beforeHooks, afterHooks) {
   const wrapped = {};
   for (const [name, t] of Object.entries(tools)) {
     wrapped[name] = {
       ...t,
       execute: async (input, ctx) => {
-        const finalInput = hooks.onBeforeToolCall ? await hooks.onBeforeToolCall(name, input) ?? input : input;
+        let finalInput = input;
+        for (const hook of beforeHooks) {
+          const result = await hook({ name, input: finalInput });
+          if (result && "input" in result) {
+            finalInput = result.input;
+          }
+        }
         const output = await t.execute(finalInput, ctx);
-        return hooks.onAfterToolCall ? await hooks.onAfterToolCall(name, finalInput, output) ?? output : output;
+        let finalOutput = output;
+        for (const hook of afterHooks) {
+          const result = await hook({ name, input: finalInput, output: finalOutput });
+          if (result && "output" in result) {
+            finalOutput = result.output;
+          }
+        }
+        return finalOutput;
       }
     };
   }
@@ -433,6 +446,7 @@ async function loop(context, options) {
   let errorCount = 0;
   let totalSteps = 0;
   let compactionAttempts = 0;
+  const loopHooks = options?.hooks ?? {};
   while (true) {
     try {
       if (compactionAttempts < MAX_COMPACTION_ATTEMPTS) {
@@ -449,8 +463,24 @@ async function loop(context, options) {
         }
       }
       let tools = options?.tools || {};
-      if (options?.hooks) {
-        tools = applyToolHooks(tools, options.hooks);
+      let systemPrompt = options?.systemPrompt;
+      if (loopHooks.beforeLLMCall?.length) {
+        for (const hook of loopHooks.beforeLLMCall) {
+          const result2 = await hook({ context, systemPrompt, tools });
+          if (result2) {
+            if ("systemPrompt" in result2 && result2.systemPrompt !== void 0) {
+              systemPrompt = result2.systemPrompt;
+            }
+            if ("tools" in result2 && result2.tools !== void 0) {
+              tools = result2.tools;
+            }
+          }
+        }
+      }
+      const beforeToolHooks = loopHooks.beforeToolCall ?? [];
+      const afterToolHooks = loopHooks.afterToolCall ?? [];
+      if (beforeToolHooks.length || afterToolHooks.length) {
+        tools = wrapToolsWithHooks(tools, beforeToolHooks, afterToolHooks);
       }
       const toolExecutionContext = {
         onClarificationRequest: options?.onClarificationRequest,
@@ -461,7 +491,7 @@ async function loop(context, options) {
         toolExecutionContext,
         provider: options?.provider,
         model: options?.model,
-        systemPrompt: options?.systemPrompt,
+        systemPrompt,
         onStepFinish: (step) => {
           options?.onStepFinish?.(step);
         },
@@ -489,6 +519,11 @@ async function loop(context, options) {
           options?.onResponse?.(messages);
         }
       }
+      if (loopHooks.afterLLMCall?.length) {
+        for (const hook of loopHooks.afterLLMCall) {
+          await hook({ context, finishReason, text });
+        }
+      }
       if (finishReason === "stop") {
         return text || "Task completed.";
       }
@@ -1066,9 +1101,15 @@ var PluginManager = class {
   enginePlugins = /* @__PURE__ */ new Map();
   userConfigPlugins = [];
   tools = /* @__PURE__ */ new Map();
-  runHooks = /* @__PURE__ */ new Map();
+  hooks = {
+    beforeRun: [],
+    afterRun: [],
+    beforeLLMCall: [],
+    afterLLMCall: [],
+    beforeToolCall: [],
+    afterToolCall: []
+  };
   services = /* @__PURE__ */ new Map();
-  protocols = /* @__PURE__ */ new Map();
   config = /* @__PURE__ */ new Map();
   events = new import_events.EventEmitter();
   logger;
@@ -1177,14 +1218,10 @@ var PluginManager = class {
         },
         getTool: (name) => this.tools.get(name),
         getTools: () => Object.fromEntries(this.tools),
-        registerRunHook: (name, hook) => {
-          this.runHooks.set(name, hook);
-        },
-        getRunHook: (name) => this.runHooks.get(name),
-        registerProtocol: (name, handler) => {
-          this.protocols.set(name, handler);
+        registerHook: (hookName, handler) => {
+          this.hooks[hookName].push(handler);
+          this.logger.debug(`Plugin "${plugin.name}" registered hook: ${hookName}`);
         },
-        getProtocol: (name) => this.protocols.get(name),
         registerService: (name, service) => {
           this.services.set(name, service);
         },
@@ -1380,10 +1417,10 @@ var PluginManager = class {
     return Object.fromEntries(this.tools);
   }
   /**
-   * 运行时钩子获取
+   * 获取指定类型的 hook 处理函数数组
    */
-  getRunHooks() {
-    return Array.from(this.runHooks.values());
+  getHooks(hookName) {
+    return [...this.hooks[hookName]];
   }
   /**
    * 服务获取
@@ -1391,12 +1428,6 @@ var PluginManager = class {
   getService(name) {
     return this.services.get(name);
   }
-  /**
-   * 协议获取
-   */
-  getProtocol(name) {
-    return this.protocols.get(name);
-  }
   /**
    * 获取插件状态
    */
@@ -1405,9 +1436,10 @@ var PluginManager = class {
       enginePlugins: Array.from(this.enginePlugins.keys()),
       userConfigPlugins: this.userConfigPlugins.map((c) => c.name || "unnamed"),
       tools: Array.from(this.tools.keys()),
-      runHooks: Array.from(this.runHooks.keys()),
-      services: Array.from(this.services.keys()),
-      protocols: Array.from(this.protocols.keys())
+      hooks: Object.fromEntries(
+        Object.entries(this.hooks).map(([k, v]) => [k, v.length])
+      ),
+      services: Array.from(this.services.keys())
     };
   }
   /**
@@ -1884,25 +1916,6 @@ var BuiltInPlanModeService = class {
     }
     return lines.join("\n");
   }
-  applyHooks(baseHooks) {
-    return {
-      onBeforeToolCall: async (name, input) => {
-        this.observePotentialPolicyViolation(name, input);
-        if (baseHooks?.onBeforeToolCall) {
-          const nextInput = await baseHooks.onBeforeToolCall(name, input);
-          return nextInput ?? input;
-        }
-        return input;
-      },
-      onAfterToolCall: async (name, input, output) => {
-        if (baseHooks?.onAfterToolCall) {
-          const nextOutput = await baseHooks.onAfterToolCall(name, input, output);
-          return nextOutput ?? output;
-        }
-        return output;
-      }
-    };
-  }
   getEvents(limit = 50) {
     return this.events.slice(-Math.max(0, limit));
   }
@@ -2039,21 +2052,18 @@ var builtInPlanModePlugin = {
   version: "1.0.0",
   async initialize(context) {
     const service = new BuiltInPlanModeService(context.logger, context.events, "executing");
-    context.registerRunHook("plan-mode", ({ context: runContext, tools, systemPrompt, hooks }) => {
+    context.registerHook("beforeLLMCall", ({ context: runContext, tools, systemPrompt }) => {
       const mode = service.getMode();
       if (mode === "executing") {
-        return {
-          systemPrompt,
-          hooks
-        };
+        return;
       }
       const transition = service.processContextMessages(runContext.messages);
       const append = service.buildPromptAppend(Object.keys(tools), transition);
       const finalSystemPrompt = appendSystemPrompt(systemPrompt, append);
-      return {
-        systemPrompt: finalSystemPrompt,
-        hooks: service.applyHooks(hooks)
-      };
+      return { systemPrompt: finalSystemPrompt };
+    });
+    context.registerHook("beforeToolCall", ({ name, input }) => {
+      service.observePotentialPolicyViolation(name, input);
     });
     context.registerService("planMode", service);
     context.registerService("planModeService", service);
@@ -2264,52 +2274,70 @@ var Engine = class {
       // 默认启用扫描
     };
   }
-  async applyRunHooks(context, systemPrompt, hooks) {
-    let nextSystemPrompt = systemPrompt;
-    let nextHooks = hooks;
-    for (const runHook of this.pluginManager.getRunHooks()) {
-      const result = await runHook({
-        context,
-        messages: context.messages,
-        tools: this.tools,
-        systemPrompt: nextSystemPrompt,
-        hooks: nextHooks
+  /**
+   * Collect all hooks for a given loop invocation.
+   * Merges plugin hooks with the legacy EngineOptions.hooks (ToolHooks).
+   */
+  collectLoopHooks() {
+    const loopHooks = {
+      beforeLLMCall: this.pluginManager.getHooks("beforeLLMCall"),
+      afterLLMCall: this.pluginManager.getHooks("afterLLMCall"),
+      beforeToolCall: [...this.pluginManager.getHooks("beforeToolCall")],
+      afterToolCall: [...this.pluginManager.getHooks("afterToolCall")]
+    };
+    const legacyHooks = this.options.hooks;
+    if (legacyHooks?.onBeforeToolCall) {
+      const legacyBefore = legacyHooks.onBeforeToolCall;
+      loopHooks.beforeToolCall.push(async ({ name, input }) => {
+        const modified = await legacyBefore(name, input);
+        return modified !== void 0 ? { input: modified } : void 0;
       });
-      if (!result) {
-        continue;
-      }
-      if ("systemPrompt" in result) {
-        nextSystemPrompt = result.systemPrompt;
-      }
-      if ("hooks" in result) {
-        nextHooks = result.hooks;
-      }
     }
-    return {
-      systemPrompt: nextSystemPrompt,
-      hooks: nextHooks
-    };
+    if (legacyHooks?.onAfterToolCall) {
+      const legacyAfter = legacyHooks.onAfterToolCall;
+      loopHooks.afterToolCall.push(async ({ name, input, output }) => {
+        const modified = await legacyAfter(name, input, output);
+        return modified !== void 0 ? { output: modified } : void 0;
+      });
+    }
+    return loopHooks;
   }
   /**
    * 运行AI循环
    */
   async run(context, options) {
-    const baseSystemPrompt = options?.systemPrompt ?? this.options.systemPrompt;
-    const baseHooks = options?.hooks ?? this.options.hooks;
-    const { systemPrompt, hooks } = await this.applyRunHooks(context, baseSystemPrompt, baseHooks);
-    return loop(context, {
+    let systemPrompt = options?.systemPrompt ?? this.options.systemPrompt;
+    let tools = { ...this.tools };
+    const beforeRunHooks = this.pluginManager.getHooks("beforeRun");
+    for (const hook of beforeRunHooks) {
+      const result = await hook({ context, systemPrompt, tools });
+      if (result) {
+        if ("systemPrompt" in result && result.systemPrompt !== void 0) {
+          systemPrompt = result.systemPrompt;
+        }
+        if ("tools" in result && result.tools !== void 0) {
+          tools = result.tools;
+        }
+      }
+    }
+    const loopHooks = this.collectLoopHooks();
+    const resultText = await loop(context, {
       ...options,
-      tools: this.tools,
-      // Engine 级别选项作为默认值；调用方通过 options 传入可在单次调用中覆盖
+      tools,
       provider: options?.provider ?? this.options.llmProvider,
       model: options?.model ?? this.options.model,
       systemPrompt,
-      hooks,
+      hooks: loopHooks,
       onToolCall: (toolCall) => {
         options?.onToolCall?.(toolCall);
       },
       onClarificationRequest: options?.onClarificationRequest
     });
+    const afterRunHooks = this.pluginManager.getHooks("afterRun");
+    for (const hook of afterRunHooks) {
+      await hook({ context, result: resultText });
+    }
+    return resultText;
   }
   /**
    * 获取插件状态