npm - @pencil-agent/nano-pencil - Versions diffs - 2.0.0-beta.8 → 2.0.0 - Mend

@pencil-agent/nano-pencil 2.0.0-beta.8 → 2.0.0

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.

Files changed (241) hide show

package/dist/core/export-html/template.html CHANGED Viewed

@@ -1,54 +1,54 @@
-<!DOCTYPE html>
-<html lang="en">
-<head>
-  <meta charset="UTF-8">
-  <meta name="viewport" content="width=device-width, initial-scale=1.0">
-  <title>Session Export</title>
-  <style>
-{{CSS}}
-  </style>
-</head>
-<body>
-  <button id="hamburger" title="Open sidebar"><svg width="14" height="14" viewBox="0 0 24 24" fill="currentColor" stroke="none"><circle cx="6" cy="6" r="2.5"/><circle cx="6" cy="18" r="2.5"/><circle cx="18" cy="12" r="2.5"/><rect x="5" y="6" width="2" height="12"/><path d="M6 12h10c1 0 2 0 2-2V8"/></svg></button>
-  <div id="sidebar-overlay"></div>
-  <div id="app">
-    <aside id="sidebar">
-      <div class="sidebar-header">
-        <div class="sidebar-controls">
-          <input type="text" class="sidebar-search" id="tree-search" placeholder="Search...">
-        </div>
-        <div class="sidebar-filters">
-          <button class="filter-btn active" data-filter="default" title="Hide settings entries">Default</button>
-          <button class="filter-btn" data-filter="no-tools" title="Default minus tool results">No-tools</button>
-          <button class="filter-btn" data-filter="user-only" title="Only user messages">User</button>
-          <button class="filter-btn" data-filter="labeled-only" title="Only labeled entries">Labeled</button>
-          <button class="filter-btn" data-filter="all" title="Show everything">All</button>
-          <button class="sidebar-close" id="sidebar-close" title="Close">✕</button>
-        </div>
-      </div>
-      <div class="tree-container" id="tree-container"></div>
-      <div class="tree-status" id="tree-status"></div>
-    </aside>
-    <main id="content">
-      <div id="header-container"></div>
-      <div id="messages"></div>
-    </main>
-    <div id="image-modal" class="image-modal">
-      <img id="modal-image" src="" alt="">
-    </div>
-  </div>
-  <script id="session-data" type="application/json">{{SESSION_DATA}}</script>
-  <!-- Vendored libraries -->
-  <script>{{MARKED_JS}}</script>
-  <!-- highlight.js -->
-  <script>{{HIGHLIGHT_JS}}</script>
-  <!-- Main application code -->
-  <script>
-{{JS}}
-  </script>
-</body>
-</html>
+<!DOCTYPE html>
+<html lang="en">
+<head>
+  <meta charset="UTF-8">
+  <meta name="viewport" content="width=device-width, initial-scale=1.0">
+  <title>Session Export</title>
+  <style>
+{{CSS}}
+  </style>
+</head>
+<body>
+  <button id="hamburger" title="Open sidebar"><svg width="14" height="14" viewBox="0 0 24 24" fill="currentColor" stroke="none"><circle cx="6" cy="6" r="2.5"/><circle cx="6" cy="18" r="2.5"/><circle cx="18" cy="12" r="2.5"/><rect x="5" y="6" width="2" height="12"/><path d="M6 12h10c1 0 2 0 2-2V8"/></svg></button>
+  <div id="sidebar-overlay"></div>
+  <div id="app">
+    <aside id="sidebar">
+      <div class="sidebar-header">
+        <div class="sidebar-controls">
+          <input type="text" class="sidebar-search" id="tree-search" placeholder="Search...">
+        </div>
+        <div class="sidebar-filters">
+          <button class="filter-btn active" data-filter="default" title="Hide settings entries">Default</button>
+          <button class="filter-btn" data-filter="no-tools" title="Default minus tool results">No-tools</button>
+          <button class="filter-btn" data-filter="user-only" title="Only user messages">User</button>
+          <button class="filter-btn" data-filter="labeled-only" title="Only labeled entries">Labeled</button>
+          <button class="filter-btn" data-filter="all" title="Show everything">All</button>
+          <button class="sidebar-close" id="sidebar-close" title="Close">✕</button>
+        </div>
+      </div>
+      <div class="tree-container" id="tree-container"></div>
+      <div class="tree-status" id="tree-status"></div>
+    </aside>
+    <main id="content">
+      <div id="header-container"></div>
+      <div id="messages"></div>
+    </main>
+    <div id="image-modal" class="image-modal">
+      <img id="modal-image" src="" alt="">
+    </div>
+  </div>
+  <script id="session-data" type="application/json">{{SESSION_DATA}}</script>
+  <!-- Vendored libraries -->
+  <script>{{MARKED_JS}}</script>
+  <!-- highlight.js -->
+  <script>{{HIGHLIGHT_JS}}</script>
+  <!-- Main application code -->
+  <script>
+{{JS}}
+  </script>
+</body>
+</html>

package/dist/core/extensions-host/index.d.ts CHANGED Viewed

@@ -8,6 +8,6 @@ export type { SlashCommandInfo, SlashCommandLocation, SlashCommandSource } from
 export { createExtensionRuntime, discoverAndLoadExtensions, loadExtensionFromFactory, loadExtensions, } from "./loader.js";
 export type { ExtensionErrorListener, ForkHandler, NavigateTreeHandler, NewSessionHandler, ShutdownHandler, SwitchSessionHandler, } from "./runner.js";
 export { ExtensionRunner } from "./runner.js";
-export type { AgentEndEvent, AgentResultEvent, AgentStartEvent, AgentToolResult, AgentToolUpdateCallback, AppAction, AppendEntryHandler, BashToolCallEvent, BashToolResultEvent, BeforeAgentStartEvent, BeforeAgentStartEventResult, CompactOptions, ContextEvent, ContextEventResult, ContextUsage, CustomToolCallEvent, CustomToolResultEvent, EditToolCallEvent, EditToolResultEvent, ExecOptions, ExecResult, Extension, ExtensionActions, ExtensionAPI, ExtensionCommandContext, ExtensionCommandContextActions, ExtensionContext, ExtensionContextActions, ExtensionError, ExtensionEvent, ExtensionFactory, ExtensionFlag, ExtensionHandler, ExtensionRuntime, ExtensionShortcut, ExtensionUIContext, ExtensionUIDialogOptions, ExtensionWidgetOptions, FindToolCallEvent, FindToolResultEvent, GetActiveToolsHandler, GetAllToolsHandler, GetCommandsHandler, GetThinkingLevelHandler, GrepToolCallEvent, GrepToolResultEvent, InputEvent, InputEventResult, InputSource, KeybindingsManager, LoadExtensionsResult, LsToolCallEvent, LsToolResultEvent, MessageEndEvent, MessageRenderer, MessageRenderOptions, MessageStartEvent, MessageUpdateEvent, ModelSelectEvent, ModelSelectSource, ProviderConfig, ProviderModelConfig, ReadToolCallEvent, ReadToolResultEvent, RegisteredCommand, RegisteredTool, ResourcesDiscoverEvent, ResourcesDiscoverResult, SendMessageHandler, SendUserMessageHandler, SessionBeforeCompactEvent, SessionBeforeCompactResult, SessionBeforeForkEvent, SessionBeforeForkResult, SessionBeforeSwitchEvent, SessionBeforeSwitchResult, SessionBeforeTreeEvent, SessionBeforeTreeResult, SessionCompactEvent, SessionEvent, SessionForkEvent, SessionShutdownEvent, SessionStartEvent, SessionSwitchEvent, SessionTreeEvent, SetActiveToolsHandler, SetLabelHandler, SetModelHandler, SetThinkingLevelHandler, TerminalInputHandler, ToolCallEvent, ToolCallEventResult, ToolDefinition, ToolExecutionEndEvent, ToolExecutionStartEvent, ToolExecutionUpdateEvent, ToolInfo, ToolRenderResultOptions, ToolResultEvent, ToolResultEventResult, TreePreparation, TurnEndEvent, TurnStartEvent, UserBashEvent, UserBashEventResult, WidgetPlacement, WriteToolCallEvent, WriteToolResultEvent, } from "./types.js";
+export type { AgentEndEvent, AgentResultEvent, AgentStartEvent, AgentToolResult, AgentToolUpdateCallback, AppAction, AppendEntryHandler, BashToolCallEvent, BashToolResultEvent, BeforeAgentStartEvent, BeforeAgentStartEventResult, CompactOptions, ContextEvent, ContextEventResult, ContextUsage, CustomToolCallEvent, CustomToolResultEvent, EditToolCallEvent, EditToolResultEvent, ExecOptions, ExecResult, Extension, ExtensionActions, ExtensionAPI, ExtensionCommandContext, ExtensionCommandContextActions, ExtensionContext, ExtensionContextActions, ExtensionError, ExtensionEvent, ExtensionEventName, ExtensionFactory, ExtensionFlag, ExtensionFlagOptions, ExtensionFlagValue, ExtensionHandler, ExtensionRuntime, ExtensionShortcut, ExtensionUIContext, ExtensionUIDialogOptions, ExtensionWidgetOptions, FindToolCallEvent, FindToolResultEvent, GetActiveToolsHandler, GetAllToolsHandler, GetCommandsHandler, GetThinkingLevelHandler, GrepToolCallEvent, GrepToolResultEvent, InputEvent, InputEventResult, InputSource, KeybindingsManager, LoadExtensionsResult, LsToolCallEvent, LsToolResultEvent, MessageEndEvent, MessageRenderer, MessageRenderOptions, MessageStartEvent, MessageUpdateEvent, ModelSelectEvent, ModelSelectSource, ProviderConfig, ProviderModelConfig, ReadToolCallEvent, ReadToolResultEvent, RegisteredCommand, RegisteredTool, ResourcesDiscoverEvent, ResourcesDiscoverResult, SendMessageHandler, SendUserMessageHandler, SessionBeforeCompactEvent, SessionBeforeCompactResult, SessionBeforeForkEvent, SessionBeforeForkResult, SessionBeforeSwitchEvent, SessionBeforeSwitchResult, SessionBeforeTreeEvent, SessionBeforeTreeResult, SessionCompactEvent, SessionEvent, SessionForkEvent, SessionShutdownEvent, SessionStartEvent, SessionSwitchEvent, SessionTreeEvent, SetActiveToolsHandler, SetLabelHandler, SetModelHandler, SetThinkingLevelHandler, TerminalInputHandler, ToolCallEvent, ToolCallEventResult, ToolDefinition, ToolExecutionEndEvent, ToolExecutionStartEvent, ToolExecutionUpdateEvent, ToolInfo, ToolRenderResultOptions, ToolResultEvent, ToolResultEventResult, TreePreparation, TurnEndEvent, TurnStartEvent, UserBashEvent, UserBashEventResult, WidgetPlacement, WriteToolCallEvent, WriteToolResultEvent, } from "./types.js";
 export { isBashToolResult, isEditToolResult, isFindToolResult, isGrepToolResult, isLsToolResult, isReadToolResult, isToolCallEventType, isWriteToolResult, } from "./types.js";
 export { wrapRegisteredTool, wrapRegisteredTools, wrapToolsWithExtensions, wrapToolWithExtensions, } from "./wrapper.js";

package/dist/core/extensions-host/loader.js CHANGED Viewed

	@@ -1 +1 @@
1	- var P=Object.defineProperty;var c=(n,t)=>P(n,"name",{value:t,configurable:!0});importas d from"node:fs";import{createRequire as T}from"node:module";importas y from"node:os";importas l from"node:path";import{fileURLToPath as A}from"node:url";import{createJiti as M}from"@mariozechner/jiti";importas w from"@pencil-agent/agent-core";importas k from"@pencil-agent/ai";importas L from"@pencil-agent/tui";import*as F from"@sinclair/typebox";import{CONFIG_DIR_NAME as C,getAgentDir as N,isBunBinary as _}from"../../config.js";import{createEventBus as I}from"../runtime/event-bus.js";import{execCommand as R}from"../platform/exec/exec.js";async function z(){return{"@sinclair/typebox":F,"@pencil-agent/agent-core":w,"@pencil-agent/tui":L,"@pencil-agent/ai":k,"@pencil-agent/nano-pencil":await import("../../index.js")}}c(z,"getVirtualModules");const m=T(import.meta.url);let h=null;function U(){if(h)return h;const n=l.dirname(A(import.meta.url)),t=l.resolve(n,"../..","index.js"),r=m.resolve("@sinclair/typebox").replace(/[\\/]build[\\/]cjs[\\/]index\.js$/,"");return h={"@pencil-agent/nano-pencil":t,"@pencil-agent/agent-core":m.resolve("@pencil-agent/agent-core"),"@pencil-agent/tui":m.resolve("@pencil-agent/tui"),"@pencil-agent/ai":m.resolve("@pencil-agent/ai"),"@sinclair/typebox":r},h}c(U,"getAliases");const D=/[\u00A0\u2000-\u200A\u202F\u205F\u3000]/g;function J(n){return n.replace(D," ")}c(J,"normalizeUnicodeSpaces");function V(n){const t=J(n);return t.startsWith("~/")?l.join(y.homedir(),t.slice(2)):t.startsWith("~")?l.join(y.homedir(),t.slice(1)):t}c(V,"expandPath");function v(n,t){const o=V(n);return l.isAbsolute(o)?o:l.resolve(t,o)}c(v,"resolvePath");function W(){const n=c(()=>{throw new Error("Extension runtime not initialized. Action methods cannot be called during extension loading.")},"notInitialized");return{sendMessage:n,sendUserMessage:n,executeCommand:c(()=>Promise.reject(new Error("Extension runtime not initialized")),"executeCommand"),isIdle:n,appendEntry:n,setSessionName:n,getSessionName:n,setLabel:n,getActiveTools:n,getAllTools:n,setActiveTools:n,getCommands:n,setModel:c(()=>Promise.reject(new Error("Extension runtime not initialized")),"setModel"),getThinkingLevel:n,setThinkingLevel:n,flagValues:new Map,pendingProviderRegistrations:[]}}c(W,"createExtensionRuntime");function E(n,t,o,r,i){return{cwd:o,agentDir:r,on(e,s){const u=n.handlers.get(e)??[];u.push(s),n.handlers.set(e,u)},registerTool(e){n.tools.set(e.name,{definition:e,extensionPath:n.path})},registerCommand(e,s){n.commands.set(e,{name:e,...s})},registerShortcut(e,s){n.shortcuts.set(e,{shortcut:e,extensionPath:n.path,...s})},registerFlag(e,s){n.flags.set(e,{name:e,extensionPath:n.path,...s}),s.default!==void 0&&!t.flagValues.has(e)&&t.flagValues.set(e,s.default)},registerMessageRenderer(e,s){n.messageRenderers.set(e,s)},getFlag(e){if(n.flags.has(e))return t.flagValues.get(e)},sendMessage(e,s){t.sendMessage(e,s)},sendUserMessage(e,s){t.sendUserMessage(e,s)},executeCommand(e){return t.executeCommand(e)},isIdle(){return t.isIdle()},appendEntry(e,s){t.appendEntry(e,s)},setSessionName(e){t.setSessionName(e)},getSessionName(){return t.getSessionName()},setLabel(e,s){t.setLabel(e,s)},exec(e,s,u){return R(e,s,u?.cwd??o,u)},getActiveTools(){return t.getActiveTools()},getAllTools(){return t.getAllTools()},setActiveTools(e){t.setActiveTools(e)},getCommands(){return t.getCommands()},setModel(e){return t.setModel(e)},getThinkingLevel(){return t.getThinkingLevel()},setThinkingLevel(e){t.setThinkingLevel(e)},registerProvider(e,s){t.pendingProviderRegistrations.push({name:e,config:s})},events:i}}c(E,"createExtensionAPI");async function B(n){const r=await M(import.meta.url,{moduleCache:!1,..._?{virtualModules:await z(),tryNative:!1}:{alias:U()}}).import(n,{default:!0});return typeof r!="function"?void 0:r}c(B,"loadExtensionModule");function S(n,t){return{path:n,resolvedPath:t,handlers:new Map,tools:new Map,messageRenderers:new Map,commands:new Map,flags:new Map,shortcuts:new Map}}c(S,"createExtension");async function O(n,t,o,r,i){const a=v(n,t);try{const e=await B(a);if(!e)return{extension:null,error:`Extension does not export a valid factory function: ${n}`};const s=S(n,a),u=E(s,i,t,o,r);return await e(u),{extension:s,error:null}}catch(e){return{extension:null,error:`Failed to load extension: ${e instanceof Error?e.message:String(e)}`}}}c(O,"loadExtension");async function ne(n,t,o,r,i,a="<inline>"){const e=S(a,a),s=E(e,i,t,o,r);return await n(s),e}c(ne,"loadExtensionFromFactory");async function $(n,t,o,r){const i=[],a=[],e=~~r??I~~(),s=W(),u=await Promise.all(n.map(f=>O(f,t,o,e,s).then(p=>({...p,path:f}))));for(const{extension:f,error:p,path:g}of u){if(p){a.push({path:g,error:p});continue}f&&i.push(f)}return{extensions:i,errors:a,runtime:s}}c($,"loadExtensions");function j(n){try{const t=d.readFileSync(n,"utf-8"),o=JSON.parse(t);return o.nanopencil&&typeof o.nanopencil=="object"?o.nanopencil:null}catch{return null}}c(j,"readPiManifest");function q(n){return n.endsWith(".ts")\|\|n.endsWith(".js")}c(q,"isExtensionFile");function b(n){const t=l.join(n,"package.json");if(d.existsSync(t)){const i=j(t);if(i?.extensions?.length){const a=[];for(const e of i.extensions){const s=l.resolve(n,e);d.existsSync(s)&&a.push(s)}if(a.length>0)return a}}const o=l.join(n,"index.ts"),r=l.join(n,"index.js");return d.existsSync(o)?[o]:d.existsSync(r)?[r]:null}c(b,"resolveExtensionEntries");function x(n){if(!d.existsSync(n))return[];const t=[];try{const o=d.readdirSync(n,{withFileTypes:!0});for(const r of o){const i=l.join(n,r.name);if((r.isFile()\|\|r.isSymbolicLink())&&q(r.name)){t.push(i);continue}if(r.isDirectory()\|\|r.isSymbolicLink()){const a=b(i);a&&t.push(...a)}}}catch{return[]}return t}c(x,"discoverExtensionsInDir");function G(n){const t=[],o=c(i=>{const a=l.join(i,"package.json");if(!d.existsSync(a))return;const e=j(a);if(e?.extensions?.length)for(const s of e.extensions){const u=l.resolve(i,s);d.existsSync(u)&&t.push(u)}},"scanPackage"),r=c(i=>{if(!d.existsSync(i))return;let a;try{a=d.readdirSync(i,{withFileTypes:!0})}catch{return}for(const e of a)if(!(!e.isDirectory()&&!e.isSymbolicLink())&&!(e.name===".bin"\|\|e.name===".cache")&&e.name!=="@pencil-agent"){if(e.name.startsWith("@")){const s=l.join(i,e.name);let u;try{u=d.readdirSync(s,{withFileTypes:!0})}catch{continue}for(const f of u)(f.isDirectory()\|\|f.isSymbolicLink())&&o(l.join(s,f.name));continue}o(l.join(i,e.name))}},"scanNodeModules");for(const i of n)r(l.join(i,"node_modules"));return t}c(G,"discoverNpmExtensions");async function te(n,t,o=N(),r){const i=[],a=new Set,e=c(f=>{for(const p of f){const g=l.resolve(p);a.has(g)\|\|(a.add(g),i.push(p))}},"addPaths"),s=l.join(t,C,"extensions");e(x(s));const u=l.join(o,"extensions");e(x(u));for(const f of n){const p=v(f,t);if(d.existsSync(p)&&d.statSync(p).isDirectory()){const g=b(p);if(g){e(g);continue}e(x(p));continue}e([p])}return e(G([t,o])),$(i,t,o,r)}c(te,"discoverAndLoadExtensions");export{W as createExtensionRuntime,te as discoverAndLoadExtensions,ne as loadExtensionFromFactory,$ as loadExtensions};
1	+ var b=Object.defineProperty;var c=(n,t)=>b(n,"name",{value:t,configurable:!0});importas d from"node:fs";import{createRequire as j}from"node:module";importas x from"node:os";importas l from"node:path";import{fileURLToPath as T}from"node:url";import{createJiti as _}from"@mariozechner/jiti";importas w from"@pencil-agent/agent-core";importas M from"@pencil-agent/ai";importas k from"@pencil-agent/ai/env";importas F from"@pencil-agent/ai/events";importas L from"@pencil-agent/ai/json";importas N from"@pencil-agent/ai/models";importas C from"@pencil-agent/ai/oauth";importas U from"@pencil-agent/ai/overflow";importas I from"@pencil-agent/ai/registry";importas R from"@pencil-agent/ai/schema";importas D from"@pencil-agent/ai/stream";importas B from"@pencil-agent/ai/types";importas O from"@pencil-agent/tui";import*as z from"@sinclair/typebox";import{CONFIG_DIR_NAME as J,getAgentDir as V,isBunBinary as W}from"../../config.js";import{createEventBus as $}from"../runtime/event-bus.js";import{execCommand as Q}from"../platform/exec/exec.js";const q=["types","schema","events","models","registry","stream","env","overflow","json","oauth"],H={"@pencil-agent/ai/types":B,"@pencil-agent/ai/schema":R,"@pencil-agent/ai/events":F,"@pencil-agent/ai/models":N,"@pencil-agent/ai/registry":I,"@pencil-agent/ai/stream":D,"@pencil-agent/ai/env":k,"@pencil-agent/ai/overflow":U,"@pencil-agent/ai/json":L,"@pencil-agent/ai/oauth":C};async function G(){return{"@sinclair/typebox":z,"@pencil-agent/agent-core":w,"@pencil-agent/tui":O,"@pencil-agent/ai":M,...H,"@pencil-agent/nano-pencil":await import("../../index.js")}}c(G,"getVirtualModules");const g=j(import.meta.url);let h=null;function K(){if(h)return h;const n=l.dirname(T(import.meta.url)),t=l.resolve(n,"../..","index.js"),a=g.resolve("@sinclair/typebox").replace(/[\\/]build[\\/]cjs[\\/]index\.js$/,""),i=Object.fromEntries(q.map(r=>[`@pencil-agent/ai/${r}`,g.resolve(`@pencil-agent/ai/${r}`)]));return h={"@pencil-agent/nano-pencil":t,"@pencil-agent/agent-core":g.resolve("@pencil-agent/agent-core"),"@pencil-agent/tui":g.resolve("@pencil-agent/tui"),...i,"@pencil-agent/ai":g.resolve("@pencil-agent/ai"),"@sinclair/typebox":a},h}c(K,"getAliases");const X=/[\u00A0\u2000-\u200A\u202F\u205F\u3000]/g;function Y(n){return n.replace(X," ")}c(Y,"normalizeUnicodeSpaces");function Z(n){const t=Y(n);return t.startsWith("~/")?l.join(x.homedir(),t.slice(2)):t.startsWith("~")?l.join(x.homedir(),t.slice(1)):t}c(Z,"expandPath");function v(n,t){const o=Z(n);return l.isAbsolute(o)?o:l.resolve(t,o)}c(v,"resolvePath");function ee(){const n=c(()=>{throw new Error("Extension runtime not initialized. Action methods cannot be called during extension loading.")},"notInitialized");return{sendMessage:n,sendUserMessage:n,executeCommand:c(()=>Promise.reject(new Error("Extension runtime not initialized")),"executeCommand"),isIdle:n,clearFollowUpQueue:n,appendEntry:n,setSessionName:n,getSessionName:n,setLabel:n,getActiveTools:n,getAllTools:n,setActiveTools:n,getCommands:n,setModel:c(()=>Promise.reject(new Error("Extension runtime not initialized")),"setModel"),getThinkingLevel:n,setThinkingLevel:n,flagValues:new Map,pendingProviderRegistrations:[]}}c(ee,"createExtensionRuntime");function E(n,t,o,a,i){return{cwd:o,agentDir:a,on(e,s){const u=n.handlers.get(e)??[];u.push(s),n.handlers.set(e,u)},registerTool(e){n.tools.set(e.name,{definition:e,extensionPath:n.path})},registerCommand(e,s){n.commands.set(e,{name:e,...s})},registerShortcut(e,s){n.shortcuts.set(e,{shortcut:e,extensionPath:n.path,...s})},registerFlag(e,s){n.flags.set(e,{name:e,extensionPath:n.path,...s}),s.default!==void 0&&!t.flagValues.has(e)&&t.flagValues.set(e,s.default)},registerMessageRenderer(e,s){n.messageRenderers.set(e,s)},getFlag(e){if(n.flags.has(e))return t.flagValues.get(e)},sendMessage(e,s){t.sendMessage(e,s)},sendUserMessage(e,s){t.sendUserMessage(e,s)},executeCommand(e){return t.executeCommand(e)},isIdle(){return t.isIdle()},clearFollowUpQueue(){t.clearFollowUpQueue()},appendEntry(e,s){t.appendEntry(e,s)},setSessionName(e){t.setSessionName(e)},getSessionName(){return t.getSessionName()},setLabel(e,s){t.setLabel(e,s)},exec(e,s,u){return Q(e,s,u?.cwd??o,u)},getActiveTools(){return t.getActiveTools()},getAllTools(){return t.getAllTools()},setActiveTools(e){t.setActiveTools(e)},getCommands(){return t.getCommands()},setModel(e){return t.setModel(e)},getThinkingLevel(){return t.getThinkingLevel()},setThinkingLevel(e){t.setThinkingLevel(e)},registerProvider(e,s){t.pendingProviderRegistrations.push({name:e,config:s})},events:i}}c(E,"createExtensionAPI");async function ne(n){const a=await _(import.meta.url,{moduleCache:!1,...W?{virtualModules:await G(),tryNative:!1}:{alias:K()}}).import(n,{default:!0});return typeof a!="function"?void 0:a}c(ne,"loadExtensionModule");function S(n,t){return{path:n,resolvedPath:t,handlers:new Map,tools:new Map,messageRenderers:new Map,commands:new Map,flags:new Map,shortcuts:new Map}}c(S,"createExtension");async function te(n,t,o,a,i){const r=v(n,t);try{const e=await ne(r);if(!e)return{extension:null,error:`Extension does not export a valid factory function: ${n}`};const s=S(n,r),u=E(s,i,t,o,a);return await e(u),{extension:s,error:null}}catch(e){return{extension:null,error:`Failed to load extension: ${e instanceof Error?e.message:String(e)}`}}}c(te,"loadExtension");async function fe(n,t,o,a,i,r="<inline>"){const e=S(r,r),s=E(e,i,t,o,a);return await n(s),e}c(fe,"loadExtensionFromFactory");async function se(n,t,o,a){const i=[],r=[],e=a??$(),s=ee(),u=await Promise.all(n.map(p=>te(p,t,o,e,s).then(f=>({...f,path:p}))));for(const{extension:p,error:f,path:m}of u){if(f){r.push({path:m,error:f});continue}p&&i.push(p)}return{extensions:i,errors:r,runtime:s}}c(se,"loadExtensions");function A(n){try{const t=d.readFileSync(n,"utf-8"),o=JSON.parse(t);return o.nanopencil&&typeof o.nanopencil=="object"?o.nanopencil:null}catch{return null}}c(A,"readPiManifest");function oe(n){return n.endsWith(".ts")\|\|n.endsWith(".js")}c(oe,"isExtensionFile");function P(n){const t=l.join(n,"package.json");if(d.existsSync(t)){const i=A(t);if(i?.extensions?.length){const r=[];for(const e of i.extensions){const s=l.resolve(n,e);d.existsSync(s)&&r.push(s)}if(r.length>0)return r}}const o=l.join(n,"index.ts"),a=l.join(n,"index.js");return d.existsSync(o)?[o]:d.existsSync(a)?[a]:null}c(P,"resolveExtensionEntries");function y(n){if(!d.existsSync(n))return[];const t=[];try{const o=d.readdirSync(n,{withFileTypes:!0});for(const a of o){const i=l.join(n,a.name);if((a.isFile()\|\|a.isSymbolicLink())&&oe(a.name)){t.push(i);continue}if(a.isDirectory()\|\|a.isSymbolicLink()){const r=P(i);r&&t.push(...r)}}}catch{return[]}return t}c(y,"discoverExtensionsInDir");function ie(n){const t=[],o=c(i=>{const r=l.join(i,"package.json");if(!d.existsSync(r))return;const e=A(r);if(e?.extensions?.length)for(const s of e.extensions){const u=l.resolve(i,s);d.existsSync(u)&&t.push(u)}},"scanPackage"),a=c(i=>{if(!d.existsSync(i))return;let r;try{r=d.readdirSync(i,{withFileTypes:!0})}catch{return}for(const e of r)if(!(!e.isDirectory()&&!e.isSymbolicLink())&&!(e.name===".bin"\|\|e.name===".cache")&&e.name!=="@pencil-agent"){if(e.name.startsWith("@")){const s=l.join(i,e.name);let u;try{u=d.readdirSync(s,{withFileTypes:!0})}catch{continue}for(const p of u)(p.isDirectory()\|\|p.isSymbolicLink())&&o(l.join(s,p.name));continue}o(l.join(i,e.name))}},"scanNodeModules");for(const i of n)a(l.join(i,"node_modules"));return t}c(ie,"discoverNpmExtensions");async function me(n,t,o=V(),a){const i=[],r=new Set,e=c(p=>{for(const f of p){const m=l.resolve(f);r.has(m)\|\|(r.add(m),i.push(f))}},"addPaths"),s=l.join(t,J,"extensions");e(y(s));const u=l.join(o,"extensions");e(y(u));for(const p of n){const f=v(p,t);if(d.existsSync(f)&&d.statSync(f).isDirectory()){const m=P(f);if(m){e(m);continue}e(y(f));continue}e([f])}return e(ie([t,o])),se(i,t,o,a)}c(me,"discoverAndLoadExtensions");export{ee as createExtensionRuntime,me as discoverAndLoadExtensions,fe as loadExtensionFromFactory,se as loadExtensions};

package/dist/core/extensions-host/runner.d.ts CHANGED Viewed

@@ -76,6 +76,7 @@ export declare class ExtensionRunner {
     private isIdleFn;
     private waitForIdleFn;
     private abortFn;
+    private clearFollowUpQueueFn;
     private hasPendingMessagesFn;
     private getContextUsageFn;
     private compactFn;

package/dist/core/extensions-host/runner.js CHANGED Viewed

@@ -1,3 +1,3 @@
-var k=Object.defineProperty;var s=(f,e)=>k(f,"name",{value:e,configurable:!0});import{theme as E}from"../../modes/interactive/theme/theme.js";import{classifyArgsSignature as v,HOOK_SAMPLE_RATES as C,runWithExtCallerContext as x}from"../platform/telemetry/index.js";const T=["interrupt","clear","exit","suspend","cycleThinkingLevel","cycleModelForward","cycleModelBackward","selectModel","selectProviderThenModel","expandTools","toggleThinking","externalEditor","followUp","submit","selectConfirm","selectCancel","copy","deleteToLineEnd"],P=s(f=>{const e={};for(const[n,t]of Object.entries(f)){const o=n,i=Array.isArray(t)?t:[t],a=T.includes(o);for(const r of i){const c=r.toLowerCase();e[c]={action:o,restrictOverride:a}}}return e},"buildBuiltinKeybindings");async function _(f){return f?.hasHandlers("session_shutdown")?(await f.emit({type:"session_shutdown"}),!0):!1}s(_,"emitSessionShutdownEvent");const p={select:s(async()=>{},"select"),confirm:s(async()=>!1,"confirm"),input:s(async()=>{},"input"),notify:s(()=>{},"notify"),onTerminalInput:s(()=>()=>{},"onTerminalInput"),setStatus:s(()=>{},"setStatus"),setWorkingMessage:s(()=>{},"setWorkingMessage"),setWidget:s(()=>{},"setWidget"),setFooter:s(()=>{},"setFooter"),setHeader:s(()=>{},"setHeader"),setTitle:s(()=>{},"setTitle"),custom:s(async()=>{},"custom"),pasteToEditor:s(()=>{},"pasteToEditor"),setEditorText:s(()=>{},"setEditorText"),getEditorText:s(()=>"","getEditorText"),editor:s(async()=>{},"editor"),openExternalEditor:s(async()=>!1,"openExternalEditor"),setEditorComponent:s(()=>{},"setEditorComponent"),get theme(){return E},getAllThemes:s(()=>[],"getAllThemes"),getTheme:s(()=>{},"getTheme"),setTheme:s(f=>({success:!1,error:"UI not available"}),"setTheme"),getToolsExpanded:s(()=>!1,"getToolsExpanded"),setToolsExpanded:s(()=>{},"setToolsExpanded")};class H{static{s(this,"ExtensionRunner")}extensions;runtime;uiContext;cwd;agentDir;sessionManager;modelRegistry;errorListeners=new Set;getModel=s(()=>{},"getModel");completeSimpleFn=s(async()=>{},"completeSimpleFn");completeSimpleWithUsageFn=s(async()=>{},"completeSimpleWithUsageFn");completeJsonFn;isIdleFn=s(()=>!0,"isIdleFn");waitForIdleFn=s(async()=>{},"waitForIdleFn");abortFn=s(()=>{},"abortFn");hasPendingMessagesFn=s(()=>!1,"hasPendingMessagesFn");getContextUsageFn=s(()=>{},"getContextUsageFn");compactFn=s(()=>{},"compactFn");getSystemPromptFn=s(()=>"","getSystemPromptFn");getSoulManagerFn=s(()=>{},"getSoulManagerFn");getSettingsFn=s(()=>({}),"getSettingsFn");getSkillsFn=s(()=>[],"getSkillsFn");newSessionHandler=s(async()=>({cancelled:!1}),"newSessionHandler");forkHandler=s(async()=>({cancelled:!1}),"forkHandler");navigateTreeHandler=s(async()=>({cancelled:!1}),"navigateTreeHandler");switchSessionHandler=s(async()=>({cancelled:!1}),"switchSessionHandler");reloadHandler=s(async()=>{},"reloadHandler");shutdownHandler=s(()=>{},"shutdownHandler");shortcutDiagnostics=[];commandDiagnostics=[];telemetrySink;_beforeAgentStartTimeoutMs=1500;get beforeAgentStartTimeoutMs(){return this._beforeAgentStartTimeoutMs??1500}beforeAgentStartTimeoutSentinel=Symbol("before_agent_start_timeout");beforeAgentStartTimeoutLastReported=new Map;constructor(e,n,t,o,i,a){this.extensions=e,this.runtime=n,this.uiContext=p,this.cwd=t,this.agentDir=o,this.sessionManager=i,this.modelRegistry=a}async withTimeout(e,n){return new Promise(t=>{const o=setTimeout(()=>t(this.beforeAgentStartTimeoutSentinel),n);Promise.resolve(e).then(i=>{clearTimeout(o),t(i)}).catch(()=>{clearTimeout(o),t(this.beforeAgentStartTimeoutSentinel)})})}reportBeforeAgentStartTimeout(e){const n=Date.now(),t=this.beforeAgentStartTimeoutLastReported.get(e)??0;n-t<6e4||(this.beforeAgentStartTimeoutLastReported.set(e,n),console.warn(`Extension before_agent_start timed out (${this.beforeAgentStartTimeoutMs}ms): ${e}`))}bindCore(e,n){this.runtime.sendMessage=e.sendMessage,this.runtime.sendUserMessage=e.sendUserMessage,this.runtime.executeCommand=e.executeCommand,this.runtime.appendEntry=e.appendEntry,this.runtime.setSessionName=e.setSessionName,this.runtime.getSessionName=e.getSessionName,this.runtime.setLabel=e.setLabel,this.runtime.getActiveTools=e.getActiveTools,this.runtime.getAllTools=e.getAllTools,this.runtime.setActiveTools=e.setActiveTools,this.runtime.getCommands=e.getCommands,this.runtime.setModel=e.setModel,this.runtime.getThinkingLevel=e.getThinkingLevel,this.runtime.setThinkingLevel=e.setThinkingLevel,this.getModel=n.getModel,this.completeSimpleFn=n.completeSimple,this.completeSimpleWithUsageFn=n.completeSimpleWithUsage,this.completeJsonFn=n.completeJson,this.isIdleFn=n.isIdle,this.runtime.isIdle=()=>this.isIdleFn(),this.abortFn=n.abort,this.hasPendingMessagesFn=n.hasPendingMessages,this.shutdownHandler=n.shutdown,this.getContextUsageFn=n.getContextUsage,this.compactFn=n.compact,this.getSystemPromptFn=n.getSystemPrompt,this.getSoulManagerFn=n.getSoulManager,this.getSettingsFn=n.getSettings,this.getSkillsFn=n.getSkills;for(const{name:t,config:o}of this.runtime.pendingProviderRegistrations)this.modelRegistry.registerProvider(t,o);this.runtime.pendingProviderRegistrations=[]}bindCommandContext(e){if(e){this.waitForIdleFn=e.waitForIdle,this.newSessionHandler=e.newSession,this.forkHandler=e.fork,this.navigateTreeHandler=e.navigateTree,this.switchSessionHandler=e.switchSession,this.reloadHandler=e.reload;return}this.waitForIdleFn=async()=>{},this.newSessionHandler=async()=>({cancelled:!1}),this.forkHandler=async()=>({cancelled:!1}),this.navigateTreeHandler=async()=>({cancelled:!1}),this.switchSessionHandler=async()=>({cancelled:!1}),this.reloadHandler=async()=>{}}setUIContext(e){this.uiContext=e??p}getUIContext(){return this.uiContext}hasUI(){return this.uiContext===p?!1:this.uiContext.__nonInteractive!==!0}getExtensionPaths(){return this.extensions.map(e=>e.path)}getAllRegisteredTools(){const e=new Map;for(const n of this.extensions)for(const t of n.tools.values())e.has(t.definition.name)||e.set(t.definition.name,t);return Array.from(e.values())}getToolDefinition(e){for(const n of this.extensions){const t=n.tools.get(e);if(t)return t.definition}}getFlags(){const e=new Map;for(const n of this.extensions)for(const[t,o]of n.flags)e.has(t)||e.set(t,o);return e}setFlagValue(e,n){this.runtime.flagValues.set(e,n)}getFlagValues(){return new Map(this.runtime.flagValues)}getShortcuts(e){this.shortcutDiagnostics=[];const n=P(e),t=new Map,o=s((i,a)=>{this.shortcutDiagnostics.push({type:"warning",message:i,path:a}),this.hasUI()||console.warn(i)},"addDiagnostic");for(const i of this.extensions)for(const[a,r]of i.shortcuts){const c=a.toLowerCase(),l=n[c];if(l?.restrictOverride===!0){o(`Extension shortcut '${a}' from ${r.extensionPath} conflicts with built-in shortcut. Skipping.`,r.extensionPath);continue}l?.restrictOverride===!1&&o(`Extension shortcut conflict: '${a}' is built-in shortcut for ${l.action} and ${r.extensionPath}. Using ${r.extensionPath}.`,r.extensionPath);const h=t.get(c);h&&o(`Extension shortcut conflict: '${a}' registered by both ${h.extensionPath} and ${r.extensionPath}. Using ${r.extensionPath}.`,r.extensionPath),t.set(c,r)}return t}getShortcutDiagnostics(){return this.shortcutDiagnostics}onError(e){return this.errorListeners.add(e),()=>this.errorListeners.delete(e)}emitError(e){for(const n of this.errorListeners)n(e)}setBeforeAgentStartTimeout(e){this._beforeAgentStartTimeoutMs=e}dispose(){this.errorListeners.clear(),this.beforeAgentStartTimeoutLastReported.clear(),this.shortcutDiagnostics=[],this.commandDiagnostics=[];for(const e of this.extensions)e.handlers.clear()}hasHandlers(e){for(const n of this.extensions){const t=n.handlers.get(e);if(t&&t.length>0)return!0}return!1}getMessageRenderer(e){for(const n of this.extensions){const t=n.messageRenderers.get(e);if(t)return t}}getRegisteredCommands(e){this.commandDiagnostics=[];const n=[],t=new Map;for(const o of this.extensions)for(const i of o.commands.values()){if(e?.has(i.name)){const r=`Extension command '${i.name}' from ${o.path} conflicts with built-in commands. Skipping.`;this.commandDiagnostics.push({type:"warning",message:r,path:o.path}),this.hasUI()||console.warn(r);continue}const a=t.get(i.name);if(a){const r=`Extension command '${i.name}' from ${o.path} conflicts with ${a}. Skipping.`;this.commandDiagnostics.push({type:"warning",message:r,path:o.path}),this.hasUI()||console.warn(r);continue}t.set(i.name,o.path),n.push(i)}return n}getCommandDiagnostics(){return this.commandDiagnostics}getRegisteredCommandsWithPaths(){const e=[];for(const n of this.extensions)for(const t of n.commands.values())e.push({command:t,extensionPath:n.path});return e}getCommand(e){for(const n of this.extensions){const t=n.commands.get(e);if(t)return t}}findCommandOwner(e){for(const n of this.extensions)if(n.commands.has(e))return n}deriveExtensionName(e){const t=(e.path||e.resolvedPath||"").replace(/\/+$/,"").split(/[\\/]/);return t[t.length-1]||"unknown"}setTelemetrySink(e){this.telemetrySink=e}writeLlmCallEvent(e){try{this.telemetrySink?.writeLlmCallEvent(e)}catch{}}invokeHookHandler(e,n,t){const o=this.deriveExtensionName(e),i={extensionName:o,callerContext:`hook:${n}`,isUserInitiated:!1};return x(i,async()=>{const a=C[n]??1;if(a<1&&Math.random()>=a)return await t();const r=new Date,c=performance.now();let l=!0,h=null;try{return await t()}catch(u){throw l=!1,h=u instanceof Error?u.constructor.name:"unknown",u}finally{try{this.telemetrySink?.writeHookEvent({extensionName:o,hookName:n,durationMs:Math.round(performance.now()-c),ok:l,errorCode:h,sampleRate:a,recordedAt:r})}catch{}}})}async invokeCommand(e,n,t,o){const i=this.getCommand(e);if(!i)return{found:!1};const a=this.findCommandOwner(e),r=a?this.deriveExtensionName(a):"unknown",c=new Date,l=performance.now();let h="ok",u=null,m;const d=v(n),y={extensionName:r,callerContext:d==="no-args"?`command:/${e}`:`command:/${e} ${d}`,isUserInitiated:!0,sessionId:o?.sessionId??null,runId:o?.runId??null,variant:o?.variant??null};try{await x(y,()=>i.handler(n,t))}catch(g){h="error",u=g instanceof Error?g.constructor.name:"unknown",m=g,this.emitError({extensionPath:`command:${e}`,event:"command",error:g instanceof Error?g.message:String(g)})}const S=Math.round(performance.now()-l),w=new Date;try{this.telemetrySink?.writeCommandEvent({extensionName:r,commandName:e,argsSignature:d,argsLength:n.length,outcome:h,errorCode:u,durationMs:S,startedAt:c,endedAt:w,sessionId:o?.sessionId??null,runId:o?.runId??null,variant:o?.variant??null})}catch{}return{found:!0}}shutdown(){this.shutdownHandler()}createContext(){const e=this.getModel;return{ui:this.uiContext,hasUI:this.hasUI(),cwd:this.cwd,agentDir:this.agentDir,sessionManager:this.sessionManager,modelRegistry:this.modelRegistry,get model(){return e()},completeSimple:s((n,t)=>this.completeSimpleFn(n,t),"completeSimple"),completeSimpleWithUsage:s((n,t)=>this.completeSimpleWithUsageFn(n,t),"completeSimpleWithUsage"),completeJson:this.completeJsonFn?(n,t,o,i)=>this.completeJsonFn(n,t,o,i):void 0,isIdle:s(()=>this.isIdleFn(),"isIdle"),abort:s(()=>this.abortFn(),"abort"),hasPendingMessages:s(()=>this.hasPendingMessagesFn(),"hasPendingMessages"),shutdown:s(()=>this.shutdownHandler(),"shutdown"),getContextUsage:s(()=>this.getContextUsageFn(),"getContextUsage"),compact:s(n=>this.compactFn(n),"compact"),getSystemPrompt:s(()=>this.getSystemPromptFn(),"getSystemPrompt"),getSoulManager:s(()=>this.getSoulManagerFn(),"getSoulManager"),getSettings:s(()=>this.getSettingsFn(),"getSettings"),getSkills:s(()=>this.getSkillsFn(),"getSkills")}}createCommandContext(){return{...this.createContext(),waitForIdle:s(()=>this.waitForIdleFn(),"waitForIdle"),newSession:s(e=>this.newSessionHandler(e),"newSession"),fork:s(e=>this.forkHandler(e),"fork"),navigateTree:s((e,n)=>this.navigateTreeHandler(e,n),"navigateTree"),switchSession:s(e=>this.switchSessionHandler(e),"switchSession"),reload:s(()=>this.reloadHandler(),"reload")}}isSessionBeforeEvent(e){return e.type==="session_before_switch"||e.type==="session_before_fork"||e.type==="session_before_compact"||e.type==="session_before_tree"}async emit(e){const n=this.createContext();let t;for(const o of this.extensions){const i=o.handlers.get(e.type);if(!(!i||i.length===0))for(const a of i)try{const r=await this.invokeHookHandler(o,e.type,()=>a(e,n));if(this.isSessionBeforeEvent(e)&&r&&(t=r,t.cancel))return t}catch(r){const c=r instanceof Error?r.message:String(r),l=r instanceof Error?r.stack:void 0;this.emitError({extensionPath:o.path,event:e.type,error:c,stack:l})}}return t}async emitToolResult(e){const n=this.createContext(),t={...e};let o=!1;for(const i of this.extensions){const a=i.handlers.get("tool_result");if(!(!a||a.length===0))for(const r of a)try{const c=await this.invokeHookHandler(i,"tool_result",()=>r(t,n));if(!c)continue;c.content!==void 0&&(t.content=c.content,o=!0),c.details!==void 0&&(t.details=c.details,o=!0),c.isError!==void 0&&(t.isError=c.isError,o=!0)}catch(c){const l=c instanceof Error?c.message:String(c),h=c instanceof Error?c.stack:void 0;this.emitError({extensionPath:i.path,event:"tool_result",error:l,stack:h})}}if(o)return{content:t.content,details:t.details,isError:t.isError}}async emitToolCall(e){const n=this.createContext();let t;for(const o of this.extensions){const i=o.handlers.get("tool_call");if(!(!i||i.length===0))for(const a of i){const r=await this.invokeHookHandler(o,"tool_call",()=>a(e,n));if(r&&(t=r,t.block))return t}}return t}async emitUserBash(e){const n=this.createContext();for(const t of this.extensions){const o=t.handlers.get("user_bash");if(!(!o||o.length===0))for(const i of o)try{const a=await this.invokeHookHandler(t,"user_bash",()=>i(e,n));if(a)return a}catch(a){const r=a instanceof Error?a.message:String(a),c=a instanceof Error?a.stack:void 0;this.emitError({extensionPath:t.path,event:"user_bash",error:r,stack:c})}}}async emitContext(e){const n=this.createContext();let t=structuredClone(e);for(const o of this.extensions){const i=o.handlers.get("context");if(!(!i||i.length===0))for(const a of i)try{const r={type:"context",messages:t},c=await this.invokeHookHandler(o,"context",()=>a(r,n));c&&c.messages&&(t=c.messages)}catch(r){const c=r instanceof Error?r.message:String(r),l=r instanceof Error?r.stack:void 0;this.emitError({extensionPath:o.path,event:"context",error:c,stack:l})}}return t}async emitBeforeAgentStart(e,n,t){const o=this.createContext(),i=[];let a=t,r=!1;for(const c of this.extensions){const l=c.handlers.get("before_agent_start");if(!(!l||l.length===0))for(const h of l)try{const u={type:"before_agent_start",prompt:e,images:n,systemPrompt:a},m=await this.withTimeout(this.invokeHookHandler(c,"before_agent_start",()=>h(u,o)),this.beforeAgentStartTimeoutMs);if(m===this.beforeAgentStartTimeoutSentinel){this.reportBeforeAgentStartTimeout(c.path);continue}if(m){const d=m;d.message&&i.push(d.message),d.systemPrompt!==void 0&&(a=d.systemPrompt,r=!0),d.appendSystemPrompt!==void 0&&(a=this.appendSystemPrompt(a,d.appendSystemPrompt),r=!0)}}catch(u){const m=u instanceof Error?u.message:String(u),d=u instanceof Error?u.stack:void 0;this.emitError({extensionPath:c.path,event:"before_agent_start",error:m,stack:d})}}if(i.length>0||r)return{messages:i.length>0?i:void 0,systemPrompt:r?a:void 0}}appendSystemPrompt(e,n){const t=n.trim();return t?`${e}
+var k=Object.defineProperty;var s=(f,e)=>k(f,"name",{value:e,configurable:!0});import{theme as E}from"../../modes/interactive/theme/theme.js";import{classifyArgsSignature as F,HOOK_SAMPLE_RATES as v,runWithExtCallerContext as x}from"../platform/telemetry/index.js";const C=["interrupt","clear","exit","suspend","cycleThinkingLevel","cycleModelForward","cycleModelBackward","selectModel","selectProviderThenModel","expandTools","toggleThinking","externalEditor","followUp","submit","selectConfirm","selectCancel","copy","deleteToLineEnd"],T=s(f=>{const e={};for(const[n,t]of Object.entries(f)){const o=n,i=Array.isArray(t)?t:[t],a=C.includes(o);for(const r of i){const l=r.toLowerCase();e[l]={action:o,restrictOverride:a}}}return e},"buildBuiltinKeybindings");async function _(f){return f?.hasHandlers("session_shutdown")?(await f.emit({type:"session_shutdown"}),!0):!1}s(_,"emitSessionShutdownEvent");const p={select:s(async()=>{},"select"),confirm:s(async()=>!1,"confirm"),input:s(async()=>{},"input"),notify:s(()=>{},"notify"),onTerminalInput:s(()=>()=>{},"onTerminalInput"),setStatus:s(()=>{},"setStatus"),setWorkingMessage:s(()=>{},"setWorkingMessage"),setWidget:s(()=>{},"setWidget"),setFooter:s(()=>{},"setFooter"),setHeader:s(()=>{},"setHeader"),setTitle:s(()=>{},"setTitle"),custom:s(async()=>{},"custom"),pasteToEditor:s(()=>{},"pasteToEditor"),setEditorText:s(()=>{},"setEditorText"),getEditorText:s(()=>"","getEditorText"),editor:s(async()=>{},"editor"),openExternalEditor:s(async()=>!1,"openExternalEditor"),setEditorComponent:s(()=>{},"setEditorComponent"),get theme(){return E},getAllThemes:s(()=>[],"getAllThemes"),getTheme:s(()=>{},"getTheme"),setTheme:s(f=>({success:!1,error:"UI not available"}),"setTheme"),getToolsExpanded:s(()=>!1,"getToolsExpanded"),setToolsExpanded:s(()=>{},"setToolsExpanded")};class H{static{s(this,"ExtensionRunner")}extensions;runtime;uiContext;cwd;agentDir;sessionManager;modelRegistry;errorListeners=new Set;getModel=s(()=>{},"getModel");completeSimpleFn=s(async()=>{},"completeSimpleFn");completeSimpleWithUsageFn=s(async()=>{},"completeSimpleWithUsageFn");completeJsonFn;isIdleFn=s(()=>!0,"isIdleFn");waitForIdleFn=s(async()=>{},"waitForIdleFn");abortFn=s(()=>{},"abortFn");clearFollowUpQueueFn=s(()=>{},"clearFollowUpQueueFn");hasPendingMessagesFn=s(()=>!1,"hasPendingMessagesFn");getContextUsageFn=s(()=>{},"getContextUsageFn");compactFn=s(()=>{},"compactFn");getSystemPromptFn=s(()=>"","getSystemPromptFn");getSoulManagerFn=s(()=>{},"getSoulManagerFn");getSettingsFn=s(()=>({}),"getSettingsFn");getSkillsFn=s(()=>[],"getSkillsFn");newSessionHandler=s(async()=>({cancelled:!1}),"newSessionHandler");forkHandler=s(async()=>({cancelled:!1}),"forkHandler");navigateTreeHandler=s(async()=>({cancelled:!1}),"navigateTreeHandler");switchSessionHandler=s(async()=>({cancelled:!1}),"switchSessionHandler");reloadHandler=s(async()=>{},"reloadHandler");shutdownHandler=s(()=>{},"shutdownHandler");shortcutDiagnostics=[];commandDiagnostics=[];telemetrySink;_beforeAgentStartTimeoutMs=1500;get beforeAgentStartTimeoutMs(){return this._beforeAgentStartTimeoutMs??1500}beforeAgentStartTimeoutSentinel=Symbol("before_agent_start_timeout");beforeAgentStartTimeoutLastReported=new Map;constructor(e,n,t,o,i,a){this.extensions=e,this.runtime=n,this.uiContext=p,this.cwd=t,this.agentDir=o,this.sessionManager=i,this.modelRegistry=a}async withTimeout(e,n){return new Promise(t=>{const o=setTimeout(()=>t(this.beforeAgentStartTimeoutSentinel),n);Promise.resolve(e).then(i=>{clearTimeout(o),t(i)}).catch(()=>{clearTimeout(o),t(this.beforeAgentStartTimeoutSentinel)})})}reportBeforeAgentStartTimeout(e){const n=Date.now(),t=this.beforeAgentStartTimeoutLastReported.get(e)??0;n-t<6e4||(this.beforeAgentStartTimeoutLastReported.set(e,n),console.warn(`Extension before_agent_start timed out (${this.beforeAgentStartTimeoutMs}ms): ${e}`))}bindCore(e,n){this.runtime.sendMessage=e.sendMessage,this.runtime.sendUserMessage=e.sendUserMessage,this.runtime.executeCommand=e.executeCommand,this.runtime.appendEntry=e.appendEntry,this.runtime.setSessionName=e.setSessionName,this.runtime.getSessionName=e.getSessionName,this.runtime.setLabel=e.setLabel,this.runtime.getActiveTools=e.getActiveTools,this.runtime.getAllTools=e.getAllTools,this.runtime.setActiveTools=e.setActiveTools,this.runtime.getCommands=e.getCommands,this.runtime.setModel=e.setModel,this.runtime.getThinkingLevel=e.getThinkingLevel,this.runtime.setThinkingLevel=e.setThinkingLevel,this.getModel=n.getModel,this.completeSimpleFn=n.completeSimple,this.completeSimpleWithUsageFn=n.completeSimpleWithUsage,this.completeJsonFn=n.completeJson,this.isIdleFn=n.isIdle,this.runtime.isIdle=()=>this.isIdleFn(),this.abortFn=n.abort,this.clearFollowUpQueueFn=n.clearFollowUpQueue,this.runtime.clearFollowUpQueue=()=>this.clearFollowUpQueueFn(),this.hasPendingMessagesFn=n.hasPendingMessages,this.shutdownHandler=n.shutdown,this.getContextUsageFn=n.getContextUsage,this.compactFn=n.compact,this.getSystemPromptFn=n.getSystemPrompt,this.getSoulManagerFn=n.getSoulManager,this.getSettingsFn=n.getSettings,this.getSkillsFn=n.getSkills;for(const{name:t,config:o}of this.runtime.pendingProviderRegistrations)this.modelRegistry.registerProvider(t,o);this.runtime.pendingProviderRegistrations=[]}bindCommandContext(e){if(e){this.waitForIdleFn=e.waitForIdle,this.newSessionHandler=e.newSession,this.forkHandler=e.fork,this.navigateTreeHandler=e.navigateTree,this.switchSessionHandler=e.switchSession,this.reloadHandler=e.reload;return}this.waitForIdleFn=async()=>{},this.newSessionHandler=async()=>({cancelled:!1}),this.forkHandler=async()=>({cancelled:!1}),this.navigateTreeHandler=async()=>({cancelled:!1}),this.switchSessionHandler=async()=>({cancelled:!1}),this.reloadHandler=async()=>{}}setUIContext(e){this.uiContext=e??p}getUIContext(){return this.uiContext}hasUI(){return this.uiContext===p?!1:this.uiContext.__nonInteractive!==!0}getExtensionPaths(){return this.extensions.map(e=>e.path)}getAllRegisteredTools(){const e=new Map;for(const n of this.extensions)for(const t of n.tools.values())e.has(t.definition.name)||e.set(t.definition.name,t);return Array.from(e.values())}getToolDefinition(e){for(const n of this.extensions){const t=n.tools.get(e);if(t)return t.definition}}getFlags(){const e=new Map;for(const n of this.extensions)for(const[t,o]of n.flags)e.has(t)||e.set(t,o);return e}setFlagValue(e,n){this.runtime.flagValues.set(e,n)}getFlagValues(){return new Map(this.runtime.flagValues)}getShortcuts(e){this.shortcutDiagnostics=[];const n=T(e),t=new Map,o=s((i,a)=>{this.shortcutDiagnostics.push({type:"warning",message:i,path:a}),this.hasUI()||console.warn(i)},"addDiagnostic");for(const i of this.extensions)for(const[a,r]of i.shortcuts){const l=a.toLowerCase(),c=n[l];if(c?.restrictOverride===!0){o(`Extension shortcut '${a}' from ${r.extensionPath} conflicts with built-in shortcut. Skipping.`,r.extensionPath);continue}c?.restrictOverride===!1&&o(`Extension shortcut conflict: '${a}' is built-in shortcut for ${c.action} and ${r.extensionPath}. Using ${r.extensionPath}.`,r.extensionPath);const h=t.get(l);h&&o(`Extension shortcut conflict: '${a}' registered by both ${h.extensionPath} and ${r.extensionPath}. Using ${r.extensionPath}.`,r.extensionPath),t.set(l,r)}return t}getShortcutDiagnostics(){return this.shortcutDiagnostics}onError(e){return this.errorListeners.add(e),()=>this.errorListeners.delete(e)}emitError(e){for(const n of this.errorListeners)n(e)}setBeforeAgentStartTimeout(e){this._beforeAgentStartTimeoutMs=e}dispose(){this.errorListeners.clear(),this.beforeAgentStartTimeoutLastReported.clear(),this.shortcutDiagnostics=[],this.commandDiagnostics=[];for(const e of this.extensions)e.handlers.clear()}hasHandlers(e){for(const n of this.extensions){const t=n.handlers.get(e);if(t&&t.length>0)return!0}return!1}getMessageRenderer(e){for(const n of this.extensions){const t=n.messageRenderers.get(e);if(t)return t}}getRegisteredCommands(e){this.commandDiagnostics=[];const n=[],t=new Map;for(const o of this.extensions)for(const i of o.commands.values()){if(e?.has(i.name)){const r=`Extension command '${i.name}' from ${o.path} conflicts with built-in commands. Skipping.`;this.commandDiagnostics.push({type:"warning",message:r,path:o.path}),this.hasUI()||console.warn(r);continue}const a=t.get(i.name);if(a){const r=`Extension command '${i.name}' from ${o.path} conflicts with ${a}. Skipping.`;this.commandDiagnostics.push({type:"warning",message:r,path:o.path}),this.hasUI()||console.warn(r);continue}t.set(i.name,o.path),n.push(i)}return n}getCommandDiagnostics(){return this.commandDiagnostics}getRegisteredCommandsWithPaths(){const e=[];for(const n of this.extensions)for(const t of n.commands.values())e.push({command:t,extensionPath:n.path});return e}getCommand(e){for(const n of this.extensions){const t=n.commands.get(e);if(t)return t}}findCommandOwner(e){for(const n of this.extensions)if(n.commands.has(e))return n}deriveExtensionName(e){const t=(e.path||e.resolvedPath||"").replace(/\/+$/,"").split(/[\\/]/);return t[t.length-1]||"unknown"}setTelemetrySink(e){this.telemetrySink=e}writeLlmCallEvent(e){try{this.telemetrySink?.writeLlmCallEvent(e)}catch{}}invokeHookHandler(e,n,t){const o=this.deriveExtensionName(e),i={extensionName:o,callerContext:`hook:${n}`,isUserInitiated:!1};return x(i,async()=>{const a=v[n]??1;if(a<1&&Math.random()>=a)return await t();const r=new Date,l=performance.now();let c=!0,h=null;try{return await t()}catch(u){throw c=!1,h=u instanceof Error?u.constructor.name:"unknown",u}finally{try{this.telemetrySink?.writeHookEvent({extensionName:o,hookName:n,durationMs:Math.round(performance.now()-l),ok:c,errorCode:h,sampleRate:a,recordedAt:r})}catch{}}})}async invokeCommand(e,n,t,o){const i=this.getCommand(e);if(!i)return{found:!1};const a=this.findCommandOwner(e),r=a?this.deriveExtensionName(a):"unknown",l=new Date,c=performance.now();let h="ok",u=null,m;const d=F(n),y={extensionName:r,callerContext:d==="no-args"?`command:/${e}`:`command:/${e} ${d}`,isUserInitiated:!0,sessionId:o?.sessionId??null,runId:o?.runId??null,variant:o?.variant??null};try{await x(y,()=>i.handler(n,t))}catch(g){h="error",u=g instanceof Error?g.constructor.name:"unknown",m=g,this.emitError({extensionPath:`command:${e}`,event:"command",error:g instanceof Error?g.message:String(g)})}const S=Math.round(performance.now()-c),w=new Date;try{this.telemetrySink?.writeCommandEvent({extensionName:r,commandName:e,argsSignature:d,argsLength:n.length,outcome:h,errorCode:u,durationMs:S,startedAt:l,endedAt:w,sessionId:o?.sessionId??null,runId:o?.runId??null,variant:o?.variant??null})}catch{}return{found:!0}}shutdown(){this.shutdownHandler()}createContext(){const e=this.getModel;return{ui:this.uiContext,hasUI:this.hasUI(),cwd:this.cwd,agentDir:this.agentDir,sessionManager:this.sessionManager,modelRegistry:this.modelRegistry,get model(){return e()},completeSimple:s((n,t)=>this.completeSimpleFn(n,t),"completeSimple"),completeSimpleWithUsage:s((n,t)=>this.completeSimpleWithUsageFn(n,t),"completeSimpleWithUsage"),completeJson:this.completeJsonFn?(n,t,o,i)=>this.completeJsonFn(n,t,o,i):void 0,isIdle:s(()=>this.isIdleFn(),"isIdle"),abort:s(()=>this.abortFn(),"abort"),clearFollowUpQueue:s(()=>this.clearFollowUpQueueFn(),"clearFollowUpQueue"),hasPendingMessages:s(()=>this.hasPendingMessagesFn(),"hasPendingMessages"),shutdown:s(()=>this.shutdownHandler(),"shutdown"),getContextUsage:s(()=>this.getContextUsageFn(),"getContextUsage"),compact:s(n=>this.compactFn(n),"compact"),getSystemPrompt:s(()=>this.getSystemPromptFn(),"getSystemPrompt"),getSoulManager:s(()=>this.getSoulManagerFn(),"getSoulManager"),getSettings:s(()=>this.getSettingsFn(),"getSettings"),getSkills:s(()=>this.getSkillsFn(),"getSkills")}}createCommandContext(){return{...this.createContext(),waitForIdle:s(()=>this.waitForIdleFn(),"waitForIdle"),newSession:s(e=>this.newSessionHandler(e),"newSession"),fork:s(e=>this.forkHandler(e),"fork"),navigateTree:s((e,n)=>this.navigateTreeHandler(e,n),"navigateTree"),switchSession:s(e=>this.switchSessionHandler(e),"switchSession"),reload:s(()=>this.reloadHandler(),"reload")}}isSessionBeforeEvent(e){return e.type==="session_before_switch"||e.type==="session_before_fork"||e.type==="session_before_compact"||e.type==="session_before_tree"}async emit(e){const n=this.createContext();let t;for(const o of this.extensions){const i=o.handlers.get(e.type);if(!(!i||i.length===0))for(const a of i)try{const r=await this.invokeHookHandler(o,e.type,()=>a(e,n));if(this.isSessionBeforeEvent(e)&&r&&(t=r,t.cancel))return t}catch(r){const l=r instanceof Error?r.message:String(r),c=r instanceof Error?r.stack:void 0;this.emitError({extensionPath:o.path,event:e.type,error:l,stack:c})}}return t}async emitToolResult(e){const n=this.createContext(),t={...e};let o=!1;for(const i of this.extensions){const a=i.handlers.get("tool_result");if(!(!a||a.length===0))for(const r of a)try{const l=await this.invokeHookHandler(i,"tool_result",()=>r(t,n));if(!l)continue;l.content!==void 0&&(t.content=l.content,o=!0),l.details!==void 0&&(t.details=l.details,o=!0),l.isError!==void 0&&(t.isError=l.isError,o=!0)}catch(l){const c=l instanceof Error?l.message:String(l),h=l instanceof Error?l.stack:void 0;this.emitError({extensionPath:i.path,event:"tool_result",error:c,stack:h})}}if(o)return{content:t.content,details:t.details,isError:t.isError}}async emitToolCall(e){const n=this.createContext();let t;for(const o of this.extensions){const i=o.handlers.get("tool_call");if(!(!i||i.length===0))for(const a of i){const r=await this.invokeHookHandler(o,"tool_call",()=>a(e,n));if(r&&(t=r,t.block))return t}}return t}async emitUserBash(e){const n=this.createContext();for(const t of this.extensions){const o=t.handlers.get("user_bash");if(!(!o||o.length===0))for(const i of o)try{const a=await this.invokeHookHandler(t,"user_bash",()=>i(e,n));if(a)return a}catch(a){const r=a instanceof Error?a.message:String(a),l=a instanceof Error?a.stack:void 0;this.emitError({extensionPath:t.path,event:"user_bash",error:r,stack:l})}}}async emitContext(e){const n=this.createContext();let t=structuredClone(e);for(const o of this.extensions){const i=o.handlers.get("context");if(!(!i||i.length===0))for(const a of i)try{const r={type:"context",messages:t},l=await this.invokeHookHandler(o,"context",()=>a(r,n));l&&l.messages&&(t=l.messages)}catch(r){const l=r instanceof Error?r.message:String(r),c=r instanceof Error?r.stack:void 0;this.emitError({extensionPath:o.path,event:"context",error:l,stack:c})}}return t}async emitBeforeAgentStart(e,n,t){const o=this.createContext(),i=[];let a=t,r=!1;for(const l of this.extensions){const c=l.handlers.get("before_agent_start");if(!(!c||c.length===0))for(const h of c)try{const u={type:"before_agent_start",prompt:e,images:n,systemPrompt:a},m=await this.withTimeout(this.invokeHookHandler(l,"before_agent_start",()=>h(u,o)),this.beforeAgentStartTimeoutMs);if(m===this.beforeAgentStartTimeoutSentinel){this.reportBeforeAgentStartTimeout(l.path);continue}if(m){const d=m;d.message&&i.push(d.message),d.systemPrompt!==void 0&&(a=d.systemPrompt,r=!0),d.appendSystemPrompt!==void 0&&(a=this.appendSystemPrompt(a,d.appendSystemPrompt),r=!0)}}catch(u){const m=u instanceof Error?u.message:String(u),d=u instanceof Error?u.stack:void 0;this.emitError({extensionPath:l.path,event:"before_agent_start",error:m,stack:d})}}if(i.length>0||r)return{messages:i.length>0?i:void 0,systemPrompt:r?a:void 0}}appendSystemPrompt(e,n){const t=n.trim();return t?`${e}
-${t}`:e}async emitResourcesDiscover(e,n){const t=this.createContext(),o=[],i=[],a=[];for(const r of this.extensions){const c=r.handlers.get("resources_discover");if(!(!c||c.length===0))for(const l of c)try{const h={type:"resources_discover",cwd:e,reason:n},m=await this.invokeHookHandler(r,"resources_discover",()=>l(h,t));m?.skillPaths?.length&&o.push(...m.skillPaths.map(d=>({path:d,extensionPath:r.path}))),m?.promptPaths?.length&&i.push(...m.promptPaths.map(d=>({path:d,extensionPath:r.path}))),m?.themePaths?.length&&a.push(...m.themePaths.map(d=>({path:d,extensionPath:r.path})))}catch(h){const u=h instanceof Error?h.message:String(h),m=h instanceof Error?h.stack:void 0;this.emitError({extensionPath:r.path,event:"resources_discover",error:u,stack:m})}}return{skillPaths:o,promptPaths:i,themePaths:a}}async emitInput(e,n,t){const o=this.createContext();let i=e,a=n;for(const r of this.extensions)for(const c of r.handlers.get("input")??[])try{const l={type:"input",text:i,images:a,source:t},h=await this.invokeHookHandler(r,"input",()=>c(l,o));if(h?.action==="handled")return h;h?.action==="transform"&&(i=h.text,a=h.images??a)}catch(l){this.emitError({extensionPath:r.path,event:"input",error:l instanceof Error?l.message:String(l),stack:l instanceof Error?l.stack:void 0})}return i!==e||a!==n?{action:"transform",text:i,images:a}:{action:"continue"}}}export{H as ExtensionRunner,_ as emitSessionShutdownEvent};
+${t}`:e}async emitResourcesDiscover(e,n){const t=this.createContext(),o=[],i=[],a=[];for(const r of this.extensions){const l=r.handlers.get("resources_discover");if(!(!l||l.length===0))for(const c of l)try{const h={type:"resources_discover",cwd:e,reason:n},m=await this.invokeHookHandler(r,"resources_discover",()=>c(h,t));m?.skillPaths?.length&&o.push(...m.skillPaths.map(d=>({path:d,extensionPath:r.path}))),m?.promptPaths?.length&&i.push(...m.promptPaths.map(d=>({path:d,extensionPath:r.path}))),m?.themePaths?.length&&a.push(...m.themePaths.map(d=>({path:d,extensionPath:r.path})))}catch(h){const u=h instanceof Error?h.message:String(h),m=h instanceof Error?h.stack:void 0;this.emitError({extensionPath:r.path,event:"resources_discover",error:u,stack:m})}}return{skillPaths:o,promptPaths:i,themePaths:a}}async emitInput(e,n,t){const o=this.createContext();let i=e,a=n;for(const r of this.extensions)for(const l of r.handlers.get("input")??[])try{const c={type:"input",text:i,images:a,source:t},h=await this.invokeHookHandler(r,"input",()=>l(c,o));if(h?.action==="handled")return h;h?.action==="transform"&&(i=h.text,a=h.images??a)}catch(c){this.emitError({extensionPath:r.path,event:"input",error:c instanceof Error?c.message:String(c),stack:c instanceof Error?c.stack:void 0})}return i!==e||a!==n?{action:"transform",text:i,images:a}:{action:"continue"}}}export{H as ExtensionRunner,_ as emitSessionShutdownEvent};

package/dist/core/extensions-host/types.d.ts CHANGED Viewed

@@ -10,7 +10,7 @@ import type { AssistantMessageEventStream } from "@pencil-agent/ai/events";
 import type { OAuthCredentials, OAuthLoginCallbacks } from "@pencil-agent/ai/oauth";
 import type { AutocompleteItem, Component, EditorComponent, EditorTheme, KeyId, OverlayHandle, OverlayOptions, TUI } from "@pencil-agent/tui";
 import type { Static, TSchema } from "@sinclair/typebox";
-import type { ToolRuntimeDescriptor } from "@pencil-agent/extension-sdk";
+import type { ArgumentCompletionContext as ProtocolArgumentCompletionContext, ExtensionCommand as ProtocolExtensionCommand, ExtensionFlag, ExtensionFlagOptions, ExtensionFlagValue, HookEventName as ProtocolHookEventName, ToolRuntimeDescriptor } from "@pencil-agent/protocol";
 import type { Theme } from "../theme-contract.js";
 import type { BashResult } from "../platform/exec/bash-executor.js";
 import type { CompactionPreparation, CompactionResult } from "../session/compaction/index.js";
@@ -222,6 +222,10 @@ export interface ExtensionContext {
     isIdle(): boolean;
     /** Abort the current agent operation */
     abort(): void;
+    /** Clear the agent's follow-up message queue. Useful when an extension
+     *  transitions to a terminal state and wants to prevent stale follow-ups
+     *  from triggering additional turns. */
+    clearFollowUpQueue(): void;
     /** Whether there are queued messages waiting */
     hasPendingMessages(): boolean;
     /** Gracefully shutdown NanoPencil and exit. Available in all contexts. */
@@ -640,6 +644,7 @@ export declare function isToolCallEventType<TName extends string, TInput extends
 };
 /** Union of all event types */
 export type ExtensionEvent = ResourcesDiscoverEvent | SessionEvent | ContextEvent | BeforeAgentStartEvent | AgentStartEvent | AgentEndEvent | AgentResultEvent | AgentAbortEvent | TurnStartEvent | TurnEndEvent | MessageStartEvent | MessageUpdateEvent | MessageEndEvent | ToolExecutionStartEvent | ToolExecutionUpdateEvent | ToolExecutionEndEvent | ModelSelectEvent | UserBashEvent | InputEvent | ToolCallEvent | ToolResultEvent;
+export type ExtensionEventName = ProtocolHookEventName;
 export interface ContextEventResult {
     messages?: AgentMessage[];
 }
@@ -696,14 +701,8 @@ export interface MessageRenderOptions {
     expanded: boolean;
 }
 export type MessageRenderer<T = unknown> = (message: CustomMessage<T>, options: MessageRenderOptions, theme: Theme) => Component | undefined;
-export interface ArgumentCompletionContext {
-    commandName: string;
-    argumentText: string;
-    argumentPrefix: string;
-    tokenIndex: number;
-    previousTokens: string[];
-}
-export interface RegisteredCommand {
+export type ArgumentCompletionContext = ProtocolArgumentCompletionContext;
+export interface RegisteredCommand extends ProtocolExtensionCommand<ExtensionCommandContext> {
     name: string;
     description?: string;
     getArgumentCompletions?: (argumentPrefix: string, context?: ArgumentCompletionContext) => AutocompleteItem[] | null;
@@ -760,13 +759,9 @@ export interface ExtensionAPI {
         handler: (ctx: ExtensionContext) => Promise<void> | void;
     }): void;
     /** Register a CLI flag. */
-    registerFlag(name: string, options: {
-        description?: string;
-        type: "boolean" | "string";
-        default?: boolean | string;
-    }): void;
+    registerFlag(name: string, options: ExtensionFlagOptions): void;
     /** Get the value of a registered CLI flag. */
-    getFlag(name: string): boolean | string | undefined;
+    getFlag(name: string): ExtensionFlagValue | undefined;
     /** Register a custom renderer for CustomMessageEntry. */
     registerMessageRenderer<T = unknown>(customType: string, renderer: MessageRenderer<T>): void;
     /** Send a custom message to the session. */
@@ -785,6 +780,9 @@ export interface ExtensionAPI {
     executeCommand(text: string): Promise<boolean>;
     /** Whether the agent is idle (not streaming). For queueing loop tasks use with sendUserMessage. */
     isIdle(): boolean;
+    /** Clear the agent's follow-up message queue. Prevents stale follow-ups
+     *  from triggering additional turns after a terminal state transition. */
+    clearFollowUpQueue(): void;
     /** Append a custom entry to the session for state persistence (not sent to LLM). */
     appendEntry<T = unknown>(customType: string, data?: T): void;
     /** Set the session display name (shown in session selector). */
@@ -927,13 +925,7 @@ export interface RegisteredTool {
     definition: ToolDefinition;
     extensionPath: string;
 }
-export interface ExtensionFlag {
-    name: string;
-    description?: string;
-    type: "boolean" | "string";
-    default?: boolean | string;
-    extensionPath: string;
-}
+export type { ExtensionFlag, ExtensionFlagOptions, ExtensionFlagValue } from "@pencil-agent/protocol";
 export interface ExtensionShortcut {
     shortcut: KeyId;
     description?: string;
@@ -1010,6 +1002,7 @@ export interface ExtensionContextActions {
     }) => Promise<string | undefined>;
     isIdle: () => boolean;
     abort: () => void;
+    clearFollowUpQueue: () => void;
     hasPendingMessages: () => boolean;
     shutdown: () => void;
     getContextUsage: () => ContextUsage | undefined;
@@ -1054,6 +1047,8 @@ export interface ExtensionCommandContextActions {
 export interface ExtensionRuntime extends ExtensionRuntimeState, ExtensionActions {
     /** Bound in runner.bindCore from ExtensionContextActions.isIdle */
     isIdle: () => boolean;
+    /** Bound in runner.bindCore from ExtensionContextActions.clearFollowUpQueue */
+    clearFollowUpQueue: () => void;
 }
 /** Loaded extension with all registered items. */
 export interface Extension {

package/dist/core/lib/ai/src/types.d.ts CHANGED Viewed

@@ -25,14 +25,22 @@ export interface StreamOptions {
     signal?: AbortSignal;
     /** Retry configuration for automatic retry on retriable errors */
     retry?: {
-        /** Maximum number of retry attempts (default: 3) */
+        /** Maximum number of retry attempts (default: 3). Ignored when persistentRetry is true. */
         maxRetries?: number;
         /** Base delay in ms for exponential backoff (default: 1000) */
         baseDelayMs?: number;
-        /** Maximum delay cap in ms (default: 30000) */
+        /** Maximum delay cap in ms (default: 30000, or 300000 for persistentRetry) */
         maxDelayMs?: number;
         /** Whether to add jitter to avoid thundering herd (default: true) */
         jitter?: boolean;
+        /**
+         * Persistent/unattended retry mode for 429/529 overload errors.
+         * When true, retries indefinitely with capped backoff and periodic heartbeat yields.
+         * Only applies to overload errors (429, 529); other retriable errors still respect maxRetries.
+         */
+        persistentRetry?: boolean;
+        /** Heartbeat interval in ms during persistent retry to prevent idle timeouts (default: 30000) */
+        heartbeatMs?: number;
     };
     apiKey?: string;
     /**
@@ -148,6 +156,8 @@ export interface AssistantMessage {
     usage: Usage;
     stopReason: StopReason;
     errorMessage?: string;
+    /** HTTP response headers from the error response, if available. Used for retry-after extraction. */
+    errorHeaders?: Record<string, string>;
     timestamp: number;
 }
 export interface ToolResultMessage<TDetails = any> {

package/dist/core/persona/persona-manager.js CHANGED Viewed

@@ -1,2 +1,5 @@
-var p=Object.defineProperty;var e=(r,t)=>p(r,"name",{value:t,configurable:!0});import{existsSync as c,readdirSync as D,readFileSync as g,mkdirSync as l,statSync as m,writeFileSync as h}from"node:fs";import{join as o,resolve as d}from"node:path";import{defaultAgentDirContext as S}from"../agent-dir/agent-dir-context.js";const f={default:"pencil"};function P(r){const t=r.trim();return t?t.replace(/[^a-zA-Z0-9_-]/g,"_"):"general"}e(P,"normalizePersonaId");class x{static{e(this,"PersonaManager")}ctx;constructor(t=S()){this.ctx=t}get personasDir(){return o(this.ctx.path,"personas")}get activePersonaStatePath(){return o(this.ctx.path,"persona.json")}ensurePersonasDir(){c(this.personasDir)||l(this.personasDir,{recursive:!0})}getPersonasDir(){return this.personasDir}getActivePersonaId(){try{if(!c(this.activePersonaStatePath))return"vex";const t=g(this.activePersonaStatePath,"utf-8"),n=JSON.parse(t);if(!n?.activePersonaId)return"vex";let i=P(String(n.activePersonaId));return f[i]&&(i=f[i],this.setActivePersonaId(i)),i}catch{return"vex"}}setActivePersonaId(t){if(this.ensurePersonasDir(),!t){try{h(this.activePersonaStatePath,JSON.stringify({},null,2),"utf-8")}catch{}return}const n=P(t),i=this.getPersonaDir(n);c(i)||l(i,{recursive:!0});const u={activePersonaId:n};h(this.activePersonaStatePath,JSON.stringify(u,null,2),"utf-8")}listPersonas(){try{return this.ensurePersonasDir(),D(this.personasDir).map(t=>P(t)).filter(t=>{const n=this.getPersonaDir(t);try{return m(n).isDirectory()}catch{return!1}}).sort((t,n)=>t.localeCompare(n))}catch{return[]}}getPersonaDir(t){const n=P(t);return o(this.personasDir,n)}getPersonaPencilPath(t){return o(this.getPersonaDir(t),"PENCIL.md")}getPersonaSkillsDir(t){return o(this.getPersonaDir(t),"skills")}getPersonaSoulDir(t){return o(this.getPersonaDir(t),"soul")}getPersonaMemoryDir(t){return o(this.getPersonaDir(t),"memory")}getPersonaMcpConfigPath(t){return o(this.getPersonaDir(t),"mcp.json")}getPersonaDescription(t){try{const n=this.getPersonaPencilPath(t);if(!c(n))return"";const i=g(n,"utf-8");for(const u of i.split(`
-`)){const a=u.trim();if(!(!a||a.startsWith("#")))return a.length>60?a.slice(0,57)+"...":a}return""}catch{return""}}}const s=new x;function M(){return s.getPersonasDir()}e(M,"getPersonasDir");function C(){return s.getActivePersonaId()}e(C,"getActivePersonaId");function N(r){s.setActivePersonaId(r)}e(N,"setActivePersonaId");function k(){return s.listPersonas()}e(k,"listPersonas");function w(r){return s.getPersonaDir(r)}e(w,"getPersonaDir");function z(r){return s.getPersonaPencilPath(r)}e(z,"getPersonaPencilPath");function E(r){return s.getPersonaSkillsDir(r)}e(E,"getPersonaSkillsDir");function O(r){return s.getPersonaSoulDir(r)}e(O,"getPersonaSoulDir");function j(r){return s.getPersonaMemoryDir(r)}e(j,"getPersonaMemoryDir");function J(r){return s.getPersonaMcpConfigPath(r)}e(J,"getPersonaMcpConfigPath");function _(r){return s.getPersonaDescription(r)}e(_,"getPersonaDescription");function F(r){const t=r.trim();return t&&d(t)}e(F,"toAbsolutePath");export{x as PersonaManager,C as getActivePersonaId,_ as getPersonaDescription,w as getPersonaDir,J as getPersonaMcpConfigPath,j as getPersonaMemoryDir,z as getPersonaPencilPath,E as getPersonaSkillsDir,O as getPersonaSoulDir,M as getPersonasDir,k as listPersonas,N as setActivePersonaId,F as toAbsolutePath};
+var p=Object.defineProperty;var n=(e,t)=>p(e,"name",{value:t,configurable:!0});import{existsSync as a,readdirSync as D,readFileSync as f,mkdirSync as l,statSync as m,writeFileSync as g}from"node:fs";import{join as o,resolve as d}from"node:path";import{defaultAgentDirContext as v}from"../agent-dir/agent-dir-context.js";const h={default:"pencil"};function P(e){const t=e.trim();return t?t.replace(/[^a-zA-Z0-9_-]/g,"_"):"general"}n(P,"normalizePersonaId");class x{static{n(this,"PersonaManager")}ctx;constructor(t=v()){this.ctx=t}get personasDir(){return o(this.ctx.path,"personas")}get activePersonaStatePath(){return o(this.ctx.path,"persona.json")}ensurePersonasDir(){a(this.personasDir)||l(this.personasDir,{recursive:!0});const t=o(this.personasDir,"vex");if(!a(t)){l(t,{recursive:!0});const r=o(t,"PENCIL.md");a(r)||g(r,`# vex
+Default persona for nanoPencil.
+`,"utf-8")}}getPersonasDir(){return this.personasDir}getActivePersonaId(){try{if(!a(this.activePersonaStatePath))return"vex";const t=f(this.activePersonaStatePath,"utf-8"),r=JSON.parse(t);if(!r?.activePersonaId)return"vex";let i=P(String(r.activePersonaId));return h[i]&&(i=h[i],this.setActivePersonaId(i)),i}catch{return"vex"}}setActivePersonaId(t){if(this.ensurePersonasDir(),!t){try{g(this.activePersonaStatePath,JSON.stringify({},null,2),"utf-8")}catch{}return}const r=P(t),i=this.getPersonaDir(r);a(i)||l(i,{recursive:!0});const u={activePersonaId:r};g(this.activePersonaStatePath,JSON.stringify(u,null,2),"utf-8")}listPersonas(){try{return this.ensurePersonasDir(),D(this.personasDir).map(t=>P(t)).filter(t=>{const r=this.getPersonaDir(t);try{return m(r).isDirectory()}catch{return!1}}).sort((t,r)=>t.localeCompare(r))}catch{return[]}}getPersonaDir(t){const r=P(t);return o(this.personasDir,r)}getPersonaPencilPath(t){return o(this.getPersonaDir(t),"PENCIL.md")}getPersonaSkillsDir(t){return o(this.getPersonaDir(t),"skills")}getPersonaSoulDir(t){return o(this.getPersonaDir(t),"soul")}getPersonaMemoryDir(t){return o(this.getPersonaDir(t),"memory")}getPersonaMcpConfigPath(t){return o(this.getPersonaDir(t),"mcp.json")}getPersonaDescription(t){try{const r=this.getPersonaPencilPath(t);if(!a(r))return"";const i=f(r,"utf-8");for(const u of i.split(`
+`)){const c=u.trim();if(!(!c||c.startsWith("#")))return c.length>60?c.slice(0,57)+"...":c}return""}catch{return""}}}const s=new x;function M(){return s.getPersonasDir()}n(M,"getPersonasDir");function C(){return s.getActivePersonaId()}n(C,"getActivePersonaId");function N(e){s.setActivePersonaId(e)}n(N,"setActivePersonaId");function k(){return s.listPersonas()}n(k,"listPersonas");function E(e){return s.getPersonaDir(e)}n(E,"getPersonaDir");function w(e){return s.getPersonaPencilPath(e)}n(w,"getPersonaPencilPath");function z(e){return s.getPersonaSkillsDir(e)}n(z,"getPersonaSkillsDir");function O(e){return s.getPersonaSoulDir(e)}n(O,"getPersonaSoulDir");function j(e){return s.getPersonaMemoryDir(e)}n(j,"getPersonaMemoryDir");function J(e){return s.getPersonaMcpConfigPath(e)}n(J,"getPersonaMcpConfigPath");function _(e){return s.getPersonaDescription(e)}n(_,"getPersonaDescription");function F(e){const t=e.trim();return t&&d(t)}n(F,"toAbsolutePath");export{x as PersonaManager,C as getActivePersonaId,_ as getPersonaDescription,E as getPersonaDir,J as getPersonaMcpConfigPath,j as getPersonaMemoryDir,w as getPersonaPencilPath,z as getPersonaSkillsDir,O as getPersonaSoulDir,M as getPersonasDir,k as listPersonas,N as setActivePersonaId,F as toAbsolutePath};