@pencil-agent/nano-pencil 2.0.0 → 2.0.2
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.
- package/dist/build-meta.json +3 -3
- package/dist/core/mcp/mcp-client.d.ts +3 -1
- package/dist/core/mcp/mcp-client.js +6 -6
- package/dist/core/mcp/mcp-config.d.ts +3 -3
- package/dist/core/mcp/mcp-config.js +1 -1
- package/dist/core/mcp/mcp-manager.d.ts +5 -1
- package/dist/core/mcp/mcp-manager.js +1 -1
- package/dist/core/model/custom-providers.js +1 -1
- package/dist/core/model-registry.js +5 -5
- package/dist/core/platform/config/resource-loader.d.ts +2 -0
- package/dist/core/platform/config/resource-loader.js +2 -2
- package/dist/core/runtime/agent-session.d.ts +12 -0
- package/dist/core/runtime/agent-session.js +8 -8
- package/dist/core/runtime/sdk.d.ts +8 -0
- package/dist/core/runtime/sdk.js +1 -1
- package/dist/modes/interactive/controllers/input-submit-controller.js +2 -2
- package/dist/modes/interactive/controllers/stream-render-controller.js +2 -2
- package/dist/modes/interactive/interactive-mode.js +41 -41
- package/dist/node_modules/@pencil-agent/agent-core/dist/agent-loop.js +3 -2
- package/dist/node_modules/@pencil-agent/agent-core/dist/structured-adaptive-agent-loop.js +2 -1
- package/dist/node_modules/@pencil-agent/ai/dist/models.generated.js +1 -1
- package/package.json +1 -1
|
@@ -1,15 +1,15 @@
|
|
|
1
|
-
var
|
|
2
|
-
`)}catch{}}_customTools;_staticCustomTools;_mcpToolsFactory;_soulManagerFactory;_baseToolRegistry=new Map;_agentTool;_createSessionFactory;_cwd;_extensionRunnerRef;_soulManager;_lastSoulInjection;_initialActiveToolNames;_baseToolsOverride;_extensionUIContext;_extensionCommandContextActions;_extensionShutdownHandler;_extensionErrorListener;_extensionErrorUnsubscriber;_modelRegistry;_agentDir;_baseSystemPrompt="";_modelController;_compactionController;_sessionTreeController;_lifecycleController;_toolOrchestrator;_toolRuntimeController;constructor(e){if(this.agent=e.agent,this.sessionManager=e.sessionManager,this.settingsManager=e.settingsManager,this.agentCtx=e.agentCtx,this._scopedModels=e.scopedModels??[],this._resourceLoader=e.resourceLoader,this._theme=e.theme,this._bashRunner=new K({getCwd:n(()=>this._cwd,"getCwd"),getShellCommandPrefix:n(()=>this.settingsManager.getShellCommandPrefix(),"getShellCommandPrefix"),appendToAgent:n(t=>this.agent.appendMessage(t),"appendToAgent"),appendToSession:n(t=>this.sessionManager.appendMessage(t),"appendToSession"),isStreaming:n(()=>this.isStreaming,"isStreaming")}),this._staticCustomTools=e.customTools??[],this._mcpToolsFactory=e.mcpToolsFactory,this._soulManagerFactory=e.soulManagerFactory,this._customTools=[...this._staticCustomTools,...e.initialMcpTools??[]],this._initialActiveToolNames=e.initialActiveToolNames,this._toolOrchestrator=new O({customTools:this._customTools,initialActiveToolNames:this._initialActiveToolNames,getExtensionTools:n(()=>new Map((this._extensionRunner?.getAllRegisteredTools()??[]).map(t=>[t.definition.name,t.definition])),"getExtensionTools")}),this._toolRuntimeController=new z(this._toolOrchestrator),this._cwd=e.cwd,this._agentDir=e.agentDir,this._modelRegistry=e.modelRegistry,this._extensionRunnerRef=e.extensionRunnerRef,this._soulManager=e.soulManager,this._baseToolsOverride=e.baseToolsOverride,this._createSessionFactory=e.createSession,this._extensionEventBridge=new te({getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner")}),this._modelController=new q({getModel:n(()=>this.model,"getModel"),getThinkingLevel:n(()=>this.thinkingLevel,"getThinkingLevel"),getScopedModels:n(()=>this._scopedModels,"getScopedModels"),setAgentModel:n(t=>this.agent.setModel(t),"setAgentModel"),setAgentThinkingLevel:n(t=>this.agent.setThinkingLevel(t),"setAgentThinkingLevel"),setAgentLoopFramework:n(t=>this.agent.setAgentLoopFramework(t),"setAgentLoopFramework"),setLoopPolicy:n(t=>this.agent.setLoopPolicy(t),"setLoopPolicy"),getApiKey:n(t=>this._modelRegistry.getApiKey(t),"getApiKey"),getApiKeyForProvider:n(t=>this._modelRegistry.getApiKeyForProvider(t),"getApiKeyForProvider"),getAvailableModels:n(()=>this._modelRegistry.getAvailableAsync(),"getAvailableModels"),getAuthCredential:n(t=>this._modelRegistry.authStorage.get(t),"getAuthCredential"),appendModelChange:n((t,s)=>this.sessionManager.appendModelChange(t,s),"appendModelChange"),appendThinkingLevelChange:n(t=>this.sessionManager.appendThinkingLevelChange(t),"appendThinkingLevelChange"),setDefaultModelAndProvider:n((t,s)=>this.settingsManager.setDefaultModelAndProvider(t,s),"setDefaultModelAndProvider"),setDefaultThinkingLevel:n(t=>this.settingsManager.setDefaultThinkingLevel(t),"setDefaultThinkingLevel"),emitModelSelect:n(async({model:t,previousModel:s,source:o})=>{this._extensionRunner&&await this._extensionRunner.emit({type:"model_select",model:t,previousModel:s,source:o})},"emitModelSelect")}),this._compactionController=new Q({getModel:n(()=>this.model,"getModel"),getApiKey:n(t=>this._modelRegistry.getApiKey(t),"getApiKey"),getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner"),getBranch:n(()=>this.sessionManager.getBranch(),"getBranch"),getEntries:n(()=>this.sessionManager.getEntries(),"getEntries"),getCompactionSettings:n(()=>this.settingsManager.getCompactionSettings(),"getCompactionSettings"),appendCompaction:n((t,s,o,r,i)=>this.sessionManager.appendCompaction(t,s,o,r,i),"appendCompaction"),applyCompactedMessages:n(()=>{const t=this.sessionManager.buildSessionContext();return this.agent.replaceMessages(t.messages),t.messages},"applyCompactedMessages"),logInfo:n((t,s)=>this._logger.info(t,s),"logInfo"),disconnectFromAgent:n(()=>this._disconnectFromAgent(),"disconnectFromAgent"),reconnectToAgent:n(()=>this._reconnectToAgent(),"reconnectToAgent"),abortAgent:n(()=>this.abort(),"abortAgent"),emitAutoCompactionStart:n(t=>this._emit({type:"auto_compaction_start",reason:t}),"emitAutoCompactionStart"),emitAutoCompactionEnd:n(t=>this._emit({type:"auto_compaction_end",...t}),"emitAutoCompactionEnd"),getAutoCompactionEnabled:n(()=>this.settingsManager.getCompactionEnabled(),"getAutoCompactionEnabled"),setAutoCompactionEnabled:n(t=>this.settingsManager.setCompactionEnabled(t),"setAutoCompactionEnabled")}),this._sessionTreeController=new X({getModel:n(()=>this.model,"getModel"),getApiKey:n(t=>this._modelRegistry.getApiKey(t),"getApiKey"),getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner"),getLeafId:n(()=>this.sessionManager.getLeafId(),"getLeafId"),getEntry:n(t=>this.sessionManager.getEntry(t),"getEntry"),collectBranchSummaryEntries:n((t,s)=>N(this.sessionManager,t,s),"collectBranchSummaryEntries"),getBranchSummaryReserveTokens:n(()=>this.settingsManager.getBranchSummarySettings().reserveTokens,"getBranchSummaryReserveTokens"),branchWithSummary:n((t,s,o,r)=>this.sessionManager.branchWithSummary(t,s,o,r),"branchWithSummary"),appendLabelChange:n((t,s)=>this.sessionManager.appendLabelChange(t,s),"appendLabelChange"),resetLeaf:n(()=>this.sessionManager.resetLeaf(),"resetLeaf"),branch:n(t=>this.sessionManager.branch(t),"branch"),rebuildAgentMessages:n(()=>{const t=this.sessionManager.buildSessionContext();this.agent.replaceMessages(t.messages)},"rebuildAgentMessages"),extractUserMessageText:n(t=>this._extractUserMessageText(t),"extractUserMessageText")}),this._lifecycleController=new G({getSessionFile:n(()=>this.sessionManager.getSessionFile(),"getSessionFile"),getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner"),disconnectFromAgent:n(()=>this._disconnectFromAgent(),"disconnectFromAgent"),reconnectToAgent:n(()=>this._reconnectToAgent(),"reconnectToAgent"),abortAgent:n(()=>this.abort(),"abortAgent"),resetAgent:n(()=>this.agent.reset(),"resetAgent"),syncAgentSessionId:n(()=>{this.agent.sessionId=this.sessionManager.getSessionId()},"syncAgentSessionId"),clearPendingQueues:n(()=>{this._steeringMessages=[],this._followUpMessages=[],this._pendingNextTurnMessages=[]},"clearPendingQueues"),clearPendingNextTurnMessages:n(()=>{this._pendingNextTurnMessages=[]},"clearPendingNextTurnMessages"),sessionNewSession:n(t=>this.sessionManager.newSession({parentSession:t}),"sessionNewSession"),sessionSetFile:n(t=>this.sessionManager.setSessionFile(t),"sessionSetFile"),sessionCreateBranchedSession:n(t=>this.sessionManager.createBranchedSession(t),"sessionCreateBranchedSession"),getEntry:n(t=>this.sessionManager.getEntry(t),"getEntry"),buildSessionContext:n(()=>this.sessionManager.buildSessionContext(),"buildSessionContext"),replaceAgentMessages:n(t=>this.agent.replaceMessages(t),"replaceAgentMessages"),appendThinkingLevelChange:n(t=>this.sessionManager.appendThinkingLevelChange(t),"appendThinkingLevelChange"),getThinkingLevel:n(()=>this.thinkingLevel,"getThinkingLevel"),getBranch:n(()=>this.sessionManager.getBranch(),"getBranch"),getDefaultThinkingLevel:n(()=>this.settingsManager.getDefaultThinkingLevel()??D,"getDefaultThinkingLevel"),getAvailableModels:n(()=>this._modelRegistry.getAvailable(),"getAvailableModels"),restoreModel:n(t=>this._modelController.restoreModel(t),"restoreModel"),restoreThinkingLevel:n(t=>this._modelController.restoreThinkingLevel(t),"restoreThinkingLevel"),runSetup:n(t=>t(this.sessionManager),"runSetup"),extractUserMessageText:n(t=>this._extractUserMessageText(t),"extractUserMessageText")}),this.agent.setModelErrorRecovery(t=>this._recoverModelErrorInLoop(t)),this._unsubscribeAgent=this.agent.subscribe(this._handleAgentEvent),e.signal){const t=n(()=>{this.abort()},"externalAbortHandler");e.signal.addEventListener("abort",t,{once:!0}),this._detachExternalAbort=()=>{e.signal?.removeEventListener("abort",t)}}this._buildRuntime({activeToolNames:this._initialActiveToolNames,includeAllExtensionTools:!0}),this._retryCoordinator=new oe(this._createRetryHost()),this._logger=ie({sessionId:this.sessionManager.getSessionId(),component:"agent-session"})}get modelRegistry(){return this._modelRegistry}get cwd(){return this._cwd}get agentDir(){return this._agentDir}getSlashCommands(){return ne({promptTemplates:this.promptTemplates,resourceLoader:this._resourceLoader,extensionRunner:this._extensionRunner},W)}async tryExecuteExtensionCommand(e){return this._tryExecuteExtensionCommand(e)}async executeSlashCommand(e){return e.startsWith("/")?this._slashCommandExecutor&&await this._slashCommandExecutor(e)?!0:this._tryExecuteExtensionCommand(e):!1}setSlashCommandExecutor(e){this._slashCommandExecutor=e}_emit(e){this._listeners.emit(e)}_lastAssistantMessage=void 0;_handleAgentEvent=n(async e=>{if(e.type==="message_start"&&e.message.role==="user"){const t=this._getUserMessageText(e.message);if(t){const s=this._steeringMessages.indexOf(t);if(s!==-1)this._steeringMessages.splice(s,1);else{const o=this._followUpMessages.indexOf(t);o!==-1&&this._followUpMessages.splice(o,1)}}}if(e.type==="message_update")this._emit(e),this._extensionEventBridge.emitExtensionEvent(e).catch(t=>{this._logger.error("[extension] message_update event error",{error:t})});else{const t=this._extensionEventBridge.emitExtensionEvent(e);this._emit(e),await t}if(e.type==="message_end"&&(e.message.role==="custom"?this.sessionManager.appendCustomMessageEntry(e.message.customType,e.message.content,e.message.display,e.message.details):(e.message.role==="user"||e.message.role==="assistant"||e.message.role==="toolResult")&&this.sessionManager.appendMessage(e.message),e.message.role==="assistant"&&(this._lastAssistantMessage=e.message,e.message.stopReason!=="error"&&this._retryCoordinator.onSuccess())),e.type==="agent_end"&&this._lastAssistantMessage){const t=this._lastAssistantMessage;if(this._lastAssistantMessage=void 0,this._retryCoordinator.isRetryableError(t)&&await this._retryCoordinator.handleError(t))return;if(await this._checkCompaction(t),this._soulManager){const s=t.stopReason==="error"?"failure":"success",o=this._cwd.split(/[/\\]/).pop()||"unknown",{tags:r,complexity:i,toolUsage:c}=S(this.state.messages,this._cwd),g=w(o,r,i,c),l=r[0]||o;(async()=>{try{await this._soulManager.recordInteraction(g,s,"turn"),await this._soulManager.updateExpertise(l,r,s==="success")}catch(m){this._logger.warn("[soul] recordInteraction/updateExpertise failed",{error:m})}})()}}e.type==="agent_end"&&this._extensionRunner&&this._extensionRunner.emit({type:"agent_end",messages:e.messages}).catch(t=>{this._logger.error("[extension] agent_end event error",{error:t})})},"_handleAgentEvent");_getUserMessageText(e){if(e.role!=="user")return"";const t=e.content;return typeof t=="string"?t:t.filter(o=>o.type==="text").map(o=>o.text).join("")}_findLastAssistantMessage(){const e=this.agent.state.messages;for(let t=e.length-1;t>=0;t--){const s=e[t];if(s.role==="assistant")return s}}subscribe(e){return this._listeners.add(e)}_disconnectFromAgent(){this._unsubscribeAgent&&(this._unsubscribeAgent(),this._unsubscribeAgent=void 0)}_reconnectToAgent(){this._unsubscribeAgent||(this._unsubscribeAgent=this.agent.subscribe(this._handleAgentEvent))}dispose(){this._disconnectFromAgent(),this._extensionRunner?.dispose(),this._listeners.clear(),this._detachExternalAbort&&(this._detachExternalAbort(),this._detachExternalAbort=void 0)}get state(){return this.agent.state}get model(){return this.agent.state.model}get thinkingLevel(){return this.agent.state.thinkingLevel}get agentLoopFramework(){return this.agent.agentLoopFramework}get isStreaming(){return this.agent.state.isStreaming}get systemPrompt(){return this.agent.state.systemPrompt}get soulManager(){return this._soulManager}get retryAttempt(){return this._retryCoordinator.attempt}getActiveToolNames(){return this.agent.state.tools.map(e=>e.name)}getAllTools(){return this._toolOrchestrator.getAllTools()}setActiveToolsByName(e){const{tools:t,validToolNames:s}=this._toolOrchestrator.setActiveToolsByName(e);this.agent.setTools(t),this._baseSystemPrompt=this._rebuildSystemPrompt(s),this.agent.setSystemPrompt(this._baseSystemPrompt)}get isCompacting(){return this._compactionController.isCompacting}get messages(){return this.agent.state.messages}get steeringMode(){return this.agent.getSteeringMode()}get followUpMode(){return this.agent.getFollowUpMode()}get sessionFile(){return this.sessionManager.getSessionFile()}get sessionId(){return this.sessionManager.getSessionId()}get sessionName(){return this.sessionManager.getSessionName()}get scopedModels(){return this._scopedModels}setScopedModels(e){this._scopedModels=e}get promptTemplates(){return this._resourceLoader.getPrompts().prompts}_rebuildSystemPrompt(e,t){return J({cwd:this._cwd,resourceLoader:this._resourceLoader,toolNames:e,baseToolRegistry:this._baseToolRegistry,soulInjection:t?.soulInjection??this._lastSoulInjection})}_getActiveBaseToolNames(){return Y(this.getActiveToolNames(),this._baseToolRegistry)}async _generateSoulInjection(){if(!this._soulManager){this._lastSoulInjection=void 0;return}try{const e=this._cwd.split(/[/\\]/).pop()||"unknown",{tags:t,complexity:s,toolUsage:o}=S(this.state.messages,this._cwd),r=await this._soulManager.generateInjection(w(e,t,s,o));return this._lastSoulInjection=typeof r=="string"&&r.trim().length>0?r:void 0,this._lastSoulInjection}catch(e){return this._emit({type:"sdk:error",source:"soul",error:e}),this._lastSoulInjection}}async prompt(e,t){const s=t?.expandPromptTemplates??!0;if(this._dbg(`prompt: "${e.slice(0,80)}" isStreaming=${this.isStreaming} hasModel=${!!this.model}`),s&&e.startsWith("/")){const u=await this.executeSlashCommand(e);if(this._dbg(`prompt: slash handled=${u}`),u)return}let o=e,r=t?.images;if(this._extensionRunner?.hasHandlers("input")){const u=await this._extensionRunner.emitInput(o,r,t?.source??"interactive");if(u.action==="handled")return;u.action==="transform"&&(o=u.text,r=u.images??r)}let i=o;if(s&&(i=this._expandSkillCommand(i),i=f(i,[...this.promptTemplates])),this.isStreaming){if(!t?.streamingBehavior)throw new Error("Agent is already processing. Specify streamingBehavior ('steer' or 'followUp') to queue the message.");t.streamingBehavior==="followUp"?await this._queueFollowUp(i,r):await this._queueSteer(i,r);return}if(this._bashRunner.flushPending(),!this.model)throw new Error(`No model selected.
|
|
1
|
+
var E=Object.defineProperty;var n=(a,e)=>E(a,"name",{value:e,configurable:!0});import{appendFileSync as v,readFileSync as k}from"node:fs";import{homedir as L}from"node:os";import{basename as P,dirname as I,join as f}from"node:path";import{isContextOverflow as b}from"@pencil-agent/ai/overflow";import{resetApiProviders as F}from"@pencil-agent/ai/registry";import{getDocsPath as C}from"../../config.js";import{stripFrontmatter as N}from"../../utils/frontmatter.js";import{calculateContextTokens as w,collectEntriesForBranchSummary as U,estimateContextTokens as B,shouldCompact as D}from"../session/compaction/index.js";import{ToolOrchestrator as O}from"../tools/orchestrator.js";import{DEFAULT_THINKING_LEVEL as $}from"../platform/config/defaults.js";import{createExtensionTelemetrySink as j}from"../platform/telemetry/index.js";import{ExtensionRunner as W}from"../extensions-host/index.js";import{expandPromptTemplate as y}from"../prompt/prompt-templates.js";import{getLatestCompactionEntry as x}from"../session/session-manager.js";import{t as H}from"../platform/i18n/index.js";import{toSoulContext as S,extractSessionContext as R}from"../soul-integration.js";import{createDefaultRuntimeTools as K}from"./default-tools.js";import{BashRunner as V}from"./bash-runner.js";import{Listeners as q}from"../platform/listeners.js";import{ModelController as Q}from"./model-controller.js";import{CompactionController as G}from"./compaction-controller.js";import{SessionLifecycleController as X}from"./session-lifecycle-controller.js";import{SessionTreeController as z}from"./session-tree-controller.js";import{ToolRuntimeController as J}from"./tool-runtime-controller.js";import{buildRuntimeSystemPrompt as Y,getActiveBaseToolNames as Z}from"./prompt-assembly.js";import{exportSessionHtml as ee,getLastAssistantText as te}from"./export-bridge.js";import{ExtensionEventBridge as se}from"./event-bridge.js";import{bindExtensionCore as ne}from"./extension-core-bindings.js";import{buildSessionSlashCommands as oe}from"./slash-command-catalog.js";import{RetryCoordinator as ie}from"./retry-coordinator.js";import{createLogger as re}from"../platform/utils/logger.js";import{createAgentTool as ae,createTaskToolAlias as le,createSendMessageTool as he,AGENT_TOOL_NAME as M,TASK_TOOL_NAME as ge,SEND_MESSAGE_TOOL_NAME as T}from"../sub-agent/index.js";import{CycleModelError as Je}from"./model-controller.js";function Qe(a){const e=a.match(/^<skill name="([^"]+)" location="([^"]+)">\n([\s\S]*?)\n<\/skill>(?:\n\n([\s\S]+))?$/);return e?{name:e[1],location:e[2],content:e[3],userMessage:e[4]?.trim()||void 0}:null}n(Qe,"parseSkillBlock");function A(a,e){const t=new Set(e.content.filter(o=>o.type==="toolCall").map(o=>o.id));let s=a.length;for(;s>0&&ce(a[s-1],t);)s--;return s>0&&me(a[s-1],e)&&s--,a.slice(0,s)}n(A,"pruneRecoverableErrorTail");function ce(a,e){return a.role==="toolResult"&&e.has(a.toolCallId)}n(ce,"isRecoverableTailToolResult");function me(a,e){return a.role==="assistant"&&a.stopReason===e.stopReason&&a.timestamp===e.timestamp&&a.provider===e.provider&&a.model===e.model&&a.api===e.api&&a.errorMessage===e.errorMessage}n(me,"isSameRecoverableAssistantMessage");function ue(a){switch(a.type){case"agent_start":return{type:"sub_agent_start",subAgentId:a.subAgentId,agentType:a.agentType,description:a.description,isAsync:a.isAsync};case"tool_start":return{type:"sub_agent_tool_start",subAgentId:a.subAgentId,toolName:a.toolName};case"tool_end":return{type:"sub_agent_tool_end",subAgentId:a.subAgentId,toolName:a.toolName,isError:a.isError};case"agent_end":return{type:"sub_agent_end",subAgentId:a.subAgentId,success:a.success};default:return}}n(ue,"mapSubAgentEvent");class Ge{static{n(this,"AgentSession")}agent;sessionManager;settingsManager;agentCtx;_scopedModels;_unsubscribeAgent;_detachExternalAbort;_listeners=new q;_steeringMessages=[];_followUpMessages=[];_pendingNextTurnMessages=[];_retryCoordinator;_logger;_bashRunner;_extensionRunner=void 0;_slashCommandExecutor=void 0;_extensionEventBridge;_resourceLoader;_theme;_debugLevel="off";_dbg(e){if(process.env.NANOPENCIL_DEBUG_SESSION==="1")try{const t=f(L(),".nanopencil","agent","nanopencil-debug.log");v(t,`[${new Date().toISOString()}] [session] ${e}
|
|
2
|
+
`)}catch{}}_customTools;_staticCustomTools;_mcpToolsFactory;_soulManagerFactory;_baseToolRegistry=new Map;_agentTool;_createSessionFactory;_cwd;_extensionRunnerRef;_soulManager;_lastSoulInjection;_initialActiveToolNames;_baseToolsOverride;_extensionUIContext;_extensionCommandContextActions;_extensionShutdownHandler;_extensionErrorListener;_extensionErrorUnsubscriber;_modelRegistry;_agentDir;_baseSystemPrompt="";_modelController;_compactionController;_sessionTreeController;_lifecycleController;_toolOrchestrator;_toolRuntimeController;constructor(e){if(this.agent=e.agent,this.sessionManager=e.sessionManager,this.settingsManager=e.settingsManager,this.agentCtx=e.agentCtx,this._scopedModels=e.scopedModels??[],this._resourceLoader=e.resourceLoader,this._theme=e.theme,this._debugLevel=e.debugLevel??"off",this._bashRunner=new V({getCwd:n(()=>this._cwd,"getCwd"),getShellCommandPrefix:n(()=>this.settingsManager.getShellCommandPrefix(),"getShellCommandPrefix"),appendToAgent:n(t=>this.agent.appendMessage(t),"appendToAgent"),appendToSession:n(t=>this.sessionManager.appendMessage(t),"appendToSession"),isStreaming:n(()=>this.isStreaming,"isStreaming")}),this._staticCustomTools=e.customTools??[],this._mcpToolsFactory=e.mcpToolsFactory,this._soulManagerFactory=e.soulManagerFactory,this._customTools=[...this._staticCustomTools,...e.initialMcpTools??[]],this._initialActiveToolNames=e.initialActiveToolNames,this._toolOrchestrator=new O({customTools:this._customTools,initialActiveToolNames:this._initialActiveToolNames,getExtensionTools:n(()=>new Map((this._extensionRunner?.getAllRegisteredTools()??[]).map(t=>[t.definition.name,t.definition])),"getExtensionTools")}),this._toolRuntimeController=new J(this._toolOrchestrator),this._cwd=e.cwd,this._agentDir=e.agentDir,this._modelRegistry=e.modelRegistry,this._extensionRunnerRef=e.extensionRunnerRef,this._soulManager=e.soulManager,this._baseToolsOverride=e.baseToolsOverride,this._createSessionFactory=e.createSession,this._extensionEventBridge=new se({getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner")}),this._modelController=new Q({getModel:n(()=>this.model,"getModel"),getThinkingLevel:n(()=>this.thinkingLevel,"getThinkingLevel"),getScopedModels:n(()=>this._scopedModels,"getScopedModels"),setAgentModel:n(t=>this.agent.setModel(t),"setAgentModel"),setAgentThinkingLevel:n(t=>this.agent.setThinkingLevel(t),"setAgentThinkingLevel"),setAgentLoopFramework:n(t=>this.agent.setAgentLoopFramework(t),"setAgentLoopFramework"),setLoopPolicy:n(t=>this.agent.setLoopPolicy(t),"setLoopPolicy"),getApiKey:n(t=>this._modelRegistry.getApiKey(t),"getApiKey"),getApiKeyForProvider:n(t=>this._modelRegistry.getApiKeyForProvider(t),"getApiKeyForProvider"),getAvailableModels:n(()=>this._modelRegistry.getAvailableAsync(),"getAvailableModels"),getAuthCredential:n(t=>this._modelRegistry.authStorage.get(t),"getAuthCredential"),appendModelChange:n((t,s)=>this.sessionManager.appendModelChange(t,s),"appendModelChange"),appendThinkingLevelChange:n(t=>this.sessionManager.appendThinkingLevelChange(t),"appendThinkingLevelChange"),setDefaultModelAndProvider:n((t,s)=>this.settingsManager.setDefaultModelAndProvider(t,s),"setDefaultModelAndProvider"),setDefaultThinkingLevel:n(t=>this.settingsManager.setDefaultThinkingLevel(t),"setDefaultThinkingLevel"),emitModelSelect:n(async({model:t,previousModel:s,source:o})=>{this._extensionRunner&&await this._extensionRunner.emit({type:"model_select",model:t,previousModel:s,source:o})},"emitModelSelect")}),this._compactionController=new G({getModel:n(()=>this.model,"getModel"),getApiKey:n(t=>this._modelRegistry.getApiKey(t),"getApiKey"),getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner"),getBranch:n(()=>this.sessionManager.getBranch(),"getBranch"),getEntries:n(()=>this.sessionManager.getEntries(),"getEntries"),getCompactionSettings:n(()=>this.settingsManager.getCompactionSettings(),"getCompactionSettings"),appendCompaction:n((t,s,o,r,i)=>this.sessionManager.appendCompaction(t,s,o,r,i),"appendCompaction"),applyCompactedMessages:n(()=>{const t=this.sessionManager.buildSessionContext();return this.agent.replaceMessages(t.messages),t.messages},"applyCompactedMessages"),logInfo:n((t,s)=>this._logger.info(t,s),"logInfo"),disconnectFromAgent:n(()=>this._disconnectFromAgent(),"disconnectFromAgent"),reconnectToAgent:n(()=>this._reconnectToAgent(),"reconnectToAgent"),abortAgent:n(()=>this.abort(),"abortAgent"),emitAutoCompactionStart:n(t=>this._emit({type:"auto_compaction_start",reason:t}),"emitAutoCompactionStart"),emitAutoCompactionEnd:n(t=>this._emit({type:"auto_compaction_end",...t}),"emitAutoCompactionEnd"),getAutoCompactionEnabled:n(()=>this.settingsManager.getCompactionEnabled(),"getAutoCompactionEnabled"),setAutoCompactionEnabled:n(t=>this.settingsManager.setCompactionEnabled(t),"setAutoCompactionEnabled")}),this._sessionTreeController=new z({getModel:n(()=>this.model,"getModel"),getApiKey:n(t=>this._modelRegistry.getApiKey(t),"getApiKey"),getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner"),getLeafId:n(()=>this.sessionManager.getLeafId(),"getLeafId"),getEntry:n(t=>this.sessionManager.getEntry(t),"getEntry"),collectBranchSummaryEntries:n((t,s)=>U(this.sessionManager,t,s),"collectBranchSummaryEntries"),getBranchSummaryReserveTokens:n(()=>this.settingsManager.getBranchSummarySettings().reserveTokens,"getBranchSummaryReserveTokens"),branchWithSummary:n((t,s,o,r)=>this.sessionManager.branchWithSummary(t,s,o,r),"branchWithSummary"),appendLabelChange:n((t,s)=>this.sessionManager.appendLabelChange(t,s),"appendLabelChange"),resetLeaf:n(()=>this.sessionManager.resetLeaf(),"resetLeaf"),branch:n(t=>this.sessionManager.branch(t),"branch"),rebuildAgentMessages:n(()=>{const t=this.sessionManager.buildSessionContext();this.agent.replaceMessages(t.messages)},"rebuildAgentMessages"),extractUserMessageText:n(t=>this._extractUserMessageText(t),"extractUserMessageText")}),this._lifecycleController=new X({getSessionFile:n(()=>this.sessionManager.getSessionFile(),"getSessionFile"),getExtensionRunner:n(()=>this._extensionRunner,"getExtensionRunner"),disconnectFromAgent:n(()=>this._disconnectFromAgent(),"disconnectFromAgent"),reconnectToAgent:n(()=>this._reconnectToAgent(),"reconnectToAgent"),abortAgent:n(()=>this.abort(),"abortAgent"),resetAgent:n(()=>this.agent.reset(),"resetAgent"),syncAgentSessionId:n(()=>{this.agent.sessionId=this.sessionManager.getSessionId()},"syncAgentSessionId"),clearPendingQueues:n(()=>{this._steeringMessages=[],this._followUpMessages=[],this._pendingNextTurnMessages=[]},"clearPendingQueues"),clearPendingNextTurnMessages:n(()=>{this._pendingNextTurnMessages=[]},"clearPendingNextTurnMessages"),sessionNewSession:n(t=>this.sessionManager.newSession({parentSession:t}),"sessionNewSession"),sessionSetFile:n(t=>this.sessionManager.setSessionFile(t),"sessionSetFile"),sessionCreateBranchedSession:n(t=>this.sessionManager.createBranchedSession(t),"sessionCreateBranchedSession"),getEntry:n(t=>this.sessionManager.getEntry(t),"getEntry"),buildSessionContext:n(()=>this.sessionManager.buildSessionContext(),"buildSessionContext"),replaceAgentMessages:n(t=>this.agent.replaceMessages(t),"replaceAgentMessages"),appendThinkingLevelChange:n(t=>this.sessionManager.appendThinkingLevelChange(t),"appendThinkingLevelChange"),getThinkingLevel:n(()=>this.thinkingLevel,"getThinkingLevel"),getBranch:n(()=>this.sessionManager.getBranch(),"getBranch"),getDefaultThinkingLevel:n(()=>this.settingsManager.getDefaultThinkingLevel()??$,"getDefaultThinkingLevel"),getAvailableModels:n(()=>this._modelRegistry.getAvailable(),"getAvailableModels"),restoreModel:n(t=>this._modelController.restoreModel(t),"restoreModel"),restoreThinkingLevel:n(t=>this._modelController.restoreThinkingLevel(t),"restoreThinkingLevel"),runSetup:n(t=>t(this.sessionManager),"runSetup"),extractUserMessageText:n(t=>this._extractUserMessageText(t),"extractUserMessageText")}),this.agent.setModelErrorRecovery(t=>this._recoverModelErrorInLoop(t)),this._unsubscribeAgent=this.agent.subscribe(this._handleAgentEvent),e.signal){const t=n(()=>{this.abort()},"externalAbortHandler");e.signal.addEventListener("abort",t,{once:!0}),this._detachExternalAbort=()=>{e.signal?.removeEventListener("abort",t)}}this._buildRuntime({activeToolNames:this._initialActiveToolNames,includeAllExtensionTools:!0}),this._retryCoordinator=new ie(this._createRetryHost()),this._logger=re({sessionId:this.sessionManager.getSessionId(),component:"agent-session"}),this._emitDebug("basic","session","session_created",{sessionId:this.sessionManager.getSessionId()})}get modelRegistry(){return this._modelRegistry}get cwd(){return this._cwd}get agentDir(){return this._agentDir}getSlashCommands(){return oe({promptTemplates:this.promptTemplates,resourceLoader:this._resourceLoader,extensionRunner:this._extensionRunner},H)}async tryExecuteExtensionCommand(e){return this._tryExecuteExtensionCommand(e)}async executeSlashCommand(e){return e.startsWith("/")?this._slashCommandExecutor&&await this._slashCommandExecutor(e)?!0:this._tryExecuteExtensionCommand(e):!1}setSlashCommandExecutor(e){this._slashCommandExecutor=e}_emit(e){this._listeners.emit(e)}_emitDebug(e,t,s,o){this._debugLevel!=="off"&&(e==="verbose"&&this._debugLevel!=="verbose"||this._emit({type:"debug",level:e,source:t,message:s,data:o,timestamp:Date.now()}))}_lastAssistantMessage=void 0;_handleAgentEvent=n(async e=>{if(e.type==="message_start"&&e.message.role==="user"){const t=this._getUserMessageText(e.message);if(t){const s=this._steeringMessages.indexOf(t);if(s!==-1)this._steeringMessages.splice(s,1);else{const o=this._followUpMessages.indexOf(t);o!==-1&&this._followUpMessages.splice(o,1)}}}if(e.type==="message_update")this._emit(e),this._extensionEventBridge.emitExtensionEvent(e).catch(t=>{this._logger.error("[extension] message_update event error",{error:t})});else{const t=this._extensionEventBridge.emitExtensionEvent(e);this._emit(e),await t}if(e.type==="tool_execution_start"?this._emitDebug("verbose","tool","tool_start",{toolName:e.toolName,toolCallId:e.toolCallId}):e.type==="tool_execution_end"&&this._emitDebug("verbose","tool","tool_end",{toolName:e.toolName,isError:e.isError}),e.type==="message_end"&&(e.message.role==="custom"?this.sessionManager.appendCustomMessageEntry(e.message.customType,e.message.content,e.message.display,e.message.details):(e.message.role==="user"||e.message.role==="assistant"||e.message.role==="toolResult")&&this.sessionManager.appendMessage(e.message),e.message.role==="assistant"&&(this._lastAssistantMessage=e.message,e.message.stopReason!=="error"&&this._retryCoordinator.onSuccess())),e.type==="agent_end"&&this._lastAssistantMessage){const t=this._lastAssistantMessage;if(this._lastAssistantMessage=void 0,this._retryCoordinator.isRetryableError(t)&&await this._retryCoordinator.handleError(t))return;if(await this._checkCompaction(t),this._soulManager){const s=t.stopReason==="error"?"failure":"success",o=this._cwd.split(/[/\\]/).pop()||"unknown",{tags:r,complexity:i,toolUsage:l}=R(this.state.messages,this._cwd),m=S(o,r,i,l),g=r[0]||o;(async()=>{try{await this._soulManager.recordInteraction(m,s,"turn"),await this._soulManager.updateExpertise(g,r,s==="success")}catch(c){this._logger.warn("[soul] recordInteraction/updateExpertise failed",{error:c})}})()}}e.type==="agent_end"&&this._extensionRunner&&this._extensionRunner.emit({type:"agent_end",messages:e.messages}).catch(t=>{this._logger.error("[extension] agent_end event error",{error:t})})},"_handleAgentEvent");_getUserMessageText(e){if(e.role!=="user")return"";const t=e.content;return typeof t=="string"?t:t.filter(o=>o.type==="text").map(o=>o.text).join("")}_findLastAssistantMessage(){const e=this.agent.state.messages;for(let t=e.length-1;t>=0;t--){const s=e[t];if(s.role==="assistant")return s}}subscribe(e){return this._listeners.add(e)}_disconnectFromAgent(){this._unsubscribeAgent&&(this._unsubscribeAgent(),this._unsubscribeAgent=void 0)}_reconnectToAgent(){this._unsubscribeAgent||(this._unsubscribeAgent=this.agent.subscribe(this._handleAgentEvent))}dispose(){this._disconnectFromAgent(),this._extensionRunner?.dispose(),this._listeners.clear(),this._detachExternalAbort&&(this._detachExternalAbort(),this._detachExternalAbort=void 0)}get state(){return this.agent.state}get model(){return this.agent.state.model}get thinkingLevel(){return this.agent.state.thinkingLevel}get agentLoopFramework(){return this.agent.agentLoopFramework}get isStreaming(){return this.agent.state.isStreaming}get systemPrompt(){return this.agent.state.systemPrompt}get soulManager(){return this._soulManager}get retryAttempt(){return this._retryCoordinator.attempt}getActiveToolNames(){return this.agent.state.tools.map(e=>e.name)}getAllTools(){return this._toolOrchestrator.getAllTools()}setActiveToolsByName(e){const{tools:t,validToolNames:s}=this._toolOrchestrator.setActiveToolsByName(e);this.agent.setTools(t),this._baseSystemPrompt=this._rebuildSystemPrompt(s),this.agent.setSystemPrompt(this._baseSystemPrompt)}get isCompacting(){return this._compactionController.isCompacting}get messages(){return this.agent.state.messages}get steeringMode(){return this.agent.getSteeringMode()}get followUpMode(){return this.agent.getFollowUpMode()}get sessionFile(){return this.sessionManager.getSessionFile()}get sessionId(){return this.sessionManager.getSessionId()}get sessionName(){return this.sessionManager.getSessionName()}get scopedModels(){return this._scopedModels}setScopedModels(e){this._scopedModels=e}get promptTemplates(){return this._resourceLoader.getPrompts().prompts}_rebuildSystemPrompt(e,t){return Y({cwd:this._cwd,resourceLoader:this._resourceLoader,toolNames:e,baseToolRegistry:this._baseToolRegistry,soulInjection:t?.soulInjection??this._lastSoulInjection})}_getActiveBaseToolNames(){return Z(this.getActiveToolNames(),this._baseToolRegistry)}async _generateSoulInjection(){if(!this._soulManager){this._lastSoulInjection=void 0;return}try{const e=this._cwd.split(/[/\\]/).pop()||"unknown",{tags:t,complexity:s,toolUsage:o}=R(this.state.messages,this._cwd),r=await this._soulManager.generateInjection(S(e,t,s,o));return this._lastSoulInjection=typeof r=="string"&&r.trim().length>0?r:void 0,this._lastSoulInjection}catch(e){return this._emit({type:"sdk:error",source:"soul",error:e}),this._lastSoulInjection}}async prompt(e,t){const s=performance.now(),o=t?.expandPromptTemplates??!0;if(this._dbg(`prompt: "${e.slice(0,80)}" isStreaming=${this.isStreaming} hasModel=${!!this.model}`),o&&e.startsWith("/")){const u=await this.executeSlashCommand(e);if(this._dbg(`prompt: slash handled=${u}`),u)return}let r=e,i=t?.images;if(this._extensionRunner?.hasHandlers("input")){const u=await this._extensionRunner.emitInput(r,i,t?.source??"interactive");if(u.action==="handled")return;u.action==="transform"&&(r=u.text,i=u.images??i)}let l=r;if(o&&(l=this._expandSkillCommand(l),l=y(l,[...this.promptTemplates])),this.isStreaming){if(!t?.streamingBehavior)throw new Error("Agent is already processing. Specify streamingBehavior ('steer' or 'followUp') to queue the message.");t.streamingBehavior==="followUp"?await this._queueFollowUp(l,i):await this._queueSteer(l,i);return}if(this._bashRunner.flushPending(),!this.model)throw new Error(`No model selected.
|
|
3
3
|
|
|
4
|
-
Use /login or set an API key environment variable. See ${
|
|
4
|
+
Use /login or set an API key environment variable. See ${f(C(),"providers.md")}
|
|
5
5
|
|
|
6
6
|
Then use /model to select a model.`);if(!await this._modelRegistry.getApiKey(this.model))throw this._modelRegistry.isUsingOAuth(this.model)?new Error(`Authentication failed for "${this.model.provider}". Credentials may have expired or network is unavailable. Run '/login ${this.model.provider}' to re-authenticate.`):new Error(`No API key found for ${this.model.provider}.
|
|
7
7
|
|
|
8
|
-
Use /login or set an API key environment variable. See ${
|
|
8
|
+
Use /login or set an API key environment variable. See ${f(C(),"providers.md")}`);const g=this._findLastAssistantMessage();g&&await this._checkCompaction(g,!1);const c=[],h=[{type:"text",text:l}];i&&h.push(...i),c.push({role:"user",content:h,timestamp:Date.now()});for(const u of this._pendingNextTurnMessages)c.push(u);this._pendingNextTurnMessages=[];const d=this._getActiveBaseToolNames(),_=await this._generateSoulInjection();if(this._baseSystemPrompt=this._rebuildSystemPrompt(d,{soulInjection:_}),this._extensionRunner){const u=await this._extensionRunner.emitBeforeAgentStart(l,i,this._baseSystemPrompt);if(u?.messages)for(const p of u.messages)c.push({role:"custom",customType:p.customType,content:p.content,display:p.display,details:p.details,timestamp:Date.now()});u?.systemPrompt?this.agent.setSystemPrompt(u.systemPrompt):this.agent.setSystemPrompt(this._baseSystemPrompt)}else this.agent.setSystemPrompt(this._baseSystemPrompt);this._dbg(`calling agent.prompt with ${c.length} message(s)`),await this.agent.prompt(c),this._dbg(`agent.prompt returned (${(performance.now()-s).toFixed(0)}ms)`),await this.waitForRetry()}async _tryExecuteExtensionCommand(e){if(!this._extensionRunner)return!1;const t=e.indexOf(" "),s=t===-1?e.slice(1):e.slice(1,t),o=t===-1?"":e.slice(t+1),r=this._extensionRunner.createCommandContext();return(await this._extensionRunner.invokeCommand(s,o,r,{sessionId:this.sessionManager.getSessionId()})).found}_expandSkillCommand(e){if(!e.startsWith("/skill:"))return e;const t=e.indexOf(" "),s=t===-1?e.slice(7):e.slice(7,t),o=t===-1?"":e.slice(t+1).trim(),r=this.resourceLoader.getSkills().skills.find(i=>i.name===s);if(!r)return e;try{const i=k(r.filePath,"utf-8"),l=N(i).trim(),m=`<skill name="${r.name}" location="${r.filePath}">
|
|
9
9
|
References are relative to ${r.baseDir}.
|
|
10
10
|
|
|
11
|
-
${
|
|
12
|
-
</skill>`;return o?`${
|
|
11
|
+
${l}
|
|
12
|
+
</skill>`;return o?`${m}
|
|
13
13
|
|
|
14
|
-
${o}`:
|
|
15
|
-
`),o.length===0&&(o=void 0)}await this.prompt(s,{expandPromptTemplates:!1,streamingBehavior:t?.deliverAs,images:o,source:"extension"})}clearQueue(){const e=[...this._steeringMessages],t=[...this._followUpMessages];return this._steeringMessages=[],this._followUpMessages=[],this.agent.clearAllQueues(),{steering:e,followUp:t}}get pendingMessageCount(){return this._steeringMessages.length+this._followUpMessages.length}getSteeringMessages(){return this._steeringMessages}getFollowUpMessages(){return this._followUpMessages}get resourceLoader(){return this._resourceLoader}async abort(){this.abortRetry(),this.agent.abort(),await this.agent.waitForIdle(),this._extensionRunner?.emit({type:"agent_abort"})}async newSession(e){return this._lifecycleController.newSession(e)}async setModel(e){await this._modelController.setModel(e)}async cycleModel(e="forward"){return this._modelController.cycleModel(e)}setThinkingLevel(e){this._modelController.setThinkingLevel(e)}setAgentLoopFramework(e){this._modelController.setAgentLoopFramework(e)}setLoopPolicy(e){this._modelController.setLoopPolicy(e)}cycleThinkingLevel(){return this._modelController.cycleThinkingLevel()}getAvailableThinkingLevels(){return this._modelController.getAvailableThinkingLevels()}supportsXhighThinking(){return this._modelController.supportsXhighThinking()}supportsThinking(){return this._modelController.supportsThinking()}setSteeringMode(e){this.agent.setSteeringMode(e),this.settingsManager.setSteeringMode(e)}setFollowUpMode(e){this.agent.setFollowUpMode(e),this.settingsManager.setFollowUpMode(e)}async compact(e){return this._compactionController.compact(e)}abortCompaction(){this._compactionController.abort()}abortBranchSummary(){this._sessionTreeController.abortBranchSummary()}async _checkCompaction(e,t=!0){const s=this.settingsManager.getCompactionSettings();if(!s.enabled||t&&e.stopReason==="aborted")return;const o=this.model?.contextWindow??0,r=this.model&&e.provider===this.model.provider&&e.model===this.model.id,i=y(this.sessionManager.getBranch()),c=i!==null&&e.timestamp<new Date(i.timestamp).getTime();if(r&&!c&&T(e,o)){const l=this.agent.state.messages;l.length>0&&l[l.length-1].role==="assistant"&&this.agent.replaceMessages(l.slice(0,-1)),await this._runAutoCompaction("overflow",!0);return}if(e.stopReason==="error")return;const g=b(e.usage);B(g,o,s)&&await this._runAutoCompaction("threshold",!1)}async _recoverModelErrorInLoop(e){const t=this.settingsManager.getCompactionSettings();if(e.message.role!=="assistant")return{action:"stop"};const s=e.message;if(e.errorSubtype!=="context_overflow"){if(!this._retryCoordinator.isRetryableError(s))return{action:"stop"};if(!await this._retryCoordinator.handleErrorInLoop(s))return{action:"stop"};const h=R(this.agent.state.messages,s);return this.agent.replaceMessages(h),{action:"retry",messages:h,transition:{reason:"model_error_recovery",subtype:e.errorSubtype,attempt:e.attempt}}}if(!t.enabled)return{action:"stop"};const o=this.model?.contextWindow??0;if(!(this.model&&s.provider===this.model.provider&&s.model===this.model.id)||!T(s,o))return{action:"stop"};const i=y(this.sessionManager.getBranch());if(i!==null&&s.timestamp<new Date(i.timestamp).getTime())return{action:"stop"};const g=this.agent.state.messages;this.agent.replaceMessages(R(g,s));const l=await this._runAutoCompaction("overflow",!0,{triggerContinue:!1});return l?{action:"retry",messages:l,transition:{reason:"model_error_recovery",subtype:e.errorSubtype,attempt:e.attempt}}:{action:"stop"}}async _runAutoCompaction(e,t,s){const o=s?.triggerContinue??!0,r=await this._compactionController.runAuto(e,t);if(r!==void 0){if(t&&o){const i=this.agent.state.messages,c=i[i.length-1];c?.role==="assistant"&&c.stopReason==="error"&&this.agent.replaceMessages(i.slice(0,-1)),setTimeout(()=>{this.agent.continue().catch(()=>{})},100)}else!t&&this.agent.hasQueuedMessages()&&setTimeout(()=>{this.agent.continue().catch(()=>{})},100);return r}}setAutoCompactionEnabled(e){this._compactionController.setAutoCompactionEnabled(e)}get autoCompactionEnabled(){return this._compactionController.autoCompactionEnabled}async bindExtensions(e){e.uiContext!==void 0&&(this._extensionUIContext=e.uiContext),e.commandContextActions!==void 0&&(this._extensionCommandContextActions=e.commandContextActions),e.shutdownHandler!==void 0&&(this._extensionShutdownHandler=e.shutdownHandler),e.onError!==void 0&&(this._extensionErrorListener=e.onError),this._extensionRunner&&(this._applyExtensionBindings(this._extensionRunner),await this._extensionRunner.emit({type:"session_start"}),await this.extendResourcesFromExtensions("startup"))}async extendResourcesFromExtensions(e){if(!this._extensionRunner?.hasHandlers("resources_discover"))return;const{skillPaths:t,promptPaths:s,themePaths:o}=await this._extensionRunner.emitResourcesDiscover(this._cwd,e);if(t.length===0&&s.length===0&&o.length===0)return;const r={skillPaths:this.buildExtensionResourcePaths(t),promptPaths:this.buildExtensionResourcePaths(s),themePaths:this.buildExtensionResourcePaths(o)};this._resourceLoader.extendResources(r),this._baseSystemPrompt=this._rebuildSystemPrompt(this.getActiveToolNames()),this.agent.setSystemPrompt(this._baseSystemPrompt)}buildExtensionResourcePaths(e){return e.map(t=>{const s=this.getExtensionSourceLabel(t.extensionPath),o=t.extensionPath.startsWith("<")?void 0:P(t.extensionPath);return{path:t.path,metadata:{source:s,scope:"temporary",origin:"top-level",baseDir:o}}})}getExtensionSourceLabel(e){return e.startsWith("<")?`extension:${e.replace(/[<>]/g,"")}`:`extension:${L(e).replace(/\.(ts|js)$/,"")}`}_applyExtensionBindings(e){e.setUIContext(this._extensionUIContext),e.bindCommandContext(this._extensionCommandContextActions),this._extensionErrorUnsubscriber?.(),this._extensionErrorUnsubscriber=this._extensionErrorListener?e.onError(this._extensionErrorListener):void 0}_bindExtensionCore(e){const t=this;se(e,{promptTemplates:this.promptTemplates,resourceLoader:this._resourceLoader,modelRegistry:this.modelRegistry,sessionManager:this.sessionManager,settingsManager:this.settingsManager,shutdownHandler:this._extensionShutdownHandler,soulManager:this._soulManager,get model(){return t.model},get thinkingLevel(){return t.thinkingLevel},get isStreaming(){return t.isStreaming},get pendingMessageCount(){return t.pendingMessageCount},get systemPrompt(){return t.systemPrompt},sendCustomMessage:n((s,o)=>this.sendCustomMessage(s,o),"sendCustomMessage"),sendUserMessage:n((s,o)=>this.sendUserMessage(s,o),"sendUserMessage"),executeSlashCommand:n(s=>this.executeSlashCommand(s),"executeSlashCommand"),getActiveToolNames:n(()=>this.getActiveToolNames(),"getActiveToolNames"),getAllTools:n(()=>this.getAllTools(),"getAllTools"),setActiveToolsByName:n(s=>this.setActiveToolsByName(s),"setActiveToolsByName"),setModel:n(s=>this.setModel(s),"setModel"),setThinkingLevel:n(s=>this.setThinkingLevel(s),"setThinkingLevel"),abort:n(()=>this.abort(),"abort"),clearFollowUpQueue:n(()=>{this.agent.clearFollowUpQueue(),this._followUpMessages=[]},"clearFollowUpQueue"),getContextUsage:n(()=>this.getContextUsage(),"getContextUsage"),compact:n(s=>this.compact(s),"compact")})}_buildRuntime(e){const t=this._baseToolsOverride?this._baseToolsOverride:H(this._cwd,this.settingsManager);this._baseToolRegistry=new Map(Object.entries(t).map(([m,h])=>[m,h]));const s=this._resourceLoader.getExtensions();if(e.flagValues)for(const[m,h]of e.flagValues)s.runtime.flagValues.set(m,h);const o=s.extensions.length>0,r=this._customTools.length>0;this._extensionRunner=o||r?new j(s.extensions,s.runtime,this._cwd,this._agentDir,this.sessionManager,this._modelRegistry):void 0,this._extensionRunnerRef&&(this._extensionRunnerRef.current=this._extensionRunner),this._extensionRunner&&(this._bindExtensionCore(this._extensionRunner),this._applyExtensionBindings(this._extensionRunner),this._extensionRunner.setTelemetrySink($({workspaceRoot:this._cwd})));const i=this._toolRuntimeController.build({baseTools:this._baseToolRegistry,baseToolsOverride:this._baseToolsOverride,customTools:this._customTools,activeToolNames:e.activeToolNames,includeAllExtensionTools:e.includeAllExtensionTools,extensionRunner:this._extensionRunner}),c=n(m=>{const h=me(m);h&&this._emit(h)},"onSubAgentEvent");this._agentTool=re({parentSession:this,parentPermissionMode:"default",parentModel:this.model,createSession:this._createSessionFactory,onSubAgentEvent:c}),i.activeTools.push(this._agentTool),this._toolOrchestrator.registerTool(x,this._agentTool);const g=ae({parentSession:this,parentPermissionMode:"default",parentModel:this.model,createSession:this._createSessionFactory,onSubAgentEvent:c});i.activeTools.push(g),this._toolOrchestrator.registerTool(he,g);const l=le({parentSession:this,parentPermissionMode:"default",parentModel:this.model,createSession:this._createSessionFactory});i.activeTools.push(l),this._toolOrchestrator.registerTool(M,l),i.systemPromptToolNames.includes(x)||i.systemPromptToolNames.push(x),i.systemPromptToolNames.includes(M)||i.systemPromptToolNames.push(M),this.agent.setTools(i.activeTools),this._baseSystemPrompt=this._rebuildSystemPrompt(i.systemPromptToolNames),this.agent.setSystemPrompt(this._baseSystemPrompt)}async _refreshMcpTools(){if(!this._mcpToolsFactory)return 0;try{const e=await this._mcpToolsFactory();return this._customTools=[...this._staticCustomTools,...e],e.length}catch(e){this._emit({type:"sdk:error",source:"mcp",error:e});const t=this._customTools.filter(s=>s.name.startsWith("mcp_"));return this._customTools=[...this._staticCustomTools,...t],t.length}}async warmupMcpTools(){if(this._mcpToolsFactory)try{const e=await this._refreshMcpTools();this._buildRuntime({activeToolNames:this.getActiveToolNames(),flagValues:this._extensionRunner?.getFlagValues(),includeAllExtensionTools:!0}),this._emit({type:"sdk:mcp_ready",toolCount:e})}catch(e){this._emit({type:"sdk:error",source:"mcp",error:e})}}async reload(){const e=this._extensionRunner?.getFlagValues();if(await this._extensionRunner?.emit({type:"session_shutdown"}),this.settingsManager.reload(),I(),await this._resourceLoader.reload(),await this._refreshMcpTools(),this._soulManagerFactory)try{this._soulManager=await this._soulManagerFactory(),this._lastSoulInjection=void 0}catch(s){this._emit({type:"sdk:error",source:"soul",error:s})}this._buildRuntime({activeToolNames:this.getActiveToolNames(),flagValues:e,includeAllExtensionTools:!0});const t=this._extensionUIContext||this._extensionCommandContextActions||this._extensionShutdownHandler||this._extensionErrorListener;this._extensionRunner&&t&&(await this._extensionRunner.emit({type:"session_start"}),await this.extendResourcesFromExtensions("reload"))}_createRetryHost(){return{getContextWindow:n(()=>this.model?.contextWindow??0,"getContextWindow"),getRetrySettings:n(()=>this.settingsManager.getRetrySettings(),"getRetrySettings"),removeLastAssistantMessage:n(()=>{const e=this.agent.state.messages;e.length>0&&e[e.length-1].role==="assistant"&&this.agent.replaceMessages(e.slice(0,-1))},"removeLastAssistantMessage"),triggerContinue:n(()=>{setTimeout(()=>{this.agent.continue().catch(()=>{})},0)},"triggerContinue"),emitEvent:n(e=>{this._emit(e)},"emitEvent")}}abortRetry(){this._retryCoordinator.abort()}async waitForRetry(){await this._retryCoordinator.waitForCompletion()}get isRetrying(){return this._retryCoordinator.isActive}get autoRetryEnabled(){return this.settingsManager.getRetryEnabled()}setAutoRetryEnabled(e){this.settingsManager.setRetryEnabled(e)}async executeBash(e,t,s){return this._bashRunner.execute(e,t,s)}recordBashResult(e,t,s){this._bashRunner.recordResult(e,t,s)}abortBash(){this._bashRunner.abort()}get isBashRunning(){return this._bashRunner.isRunning}get hasPendingBashMessages(){return this._bashRunner.hasPending}async switchSession(e){return this._lifecycleController.switchSession(e)}setSessionName(e){this.sessionManager.appendSessionInfo(e)}async fork(e){return this._lifecycleController.fork(e)}async navigateTree(e,t={}){return this._sessionTreeController.navigateTree(e,t)}getUserMessagesForForking(){const e=this.sessionManager.getEntries(),t=[];for(const s of e){if(s.type!=="message"||s.message.role!=="user")continue;const o=this._extractUserMessageText(s.message.content);o&&t.push({entryId:s.id,text:o})}return t}_extractUserMessageText(e){return typeof e=="string"?e:Array.isArray(e)?e.filter(t=>t.type==="text").map(t=>t.text).join(""):""}getSessionStats(){const e=this.state,t=e.messages.filter(h=>h.role==="user").length,s=e.messages.filter(h=>h.role==="assistant").length,o=e.messages.filter(h=>h.role==="toolResult").length;let r=0,i=0,c=0,g=0,l=0,m=0;for(const h of e.messages)if(h.role==="assistant"){const d=h;r+=d.content.filter(u=>u.type==="toolCall").length,i+=d.usage.input,c+=d.usage.output,g+=d.usage.cacheRead,l+=d.usage.cacheWrite,m+=d.usage.cost.total}return{sessionFile:this.sessionFile,sessionId:this.sessionId,userMessages:t,assistantMessages:s,toolCalls:r,toolResults:o,totalMessages:e.messages.length,tokens:{input:i,output:c,cacheRead:g,cacheWrite:l,total:i+c+g+l},cost:m}}getContextUsage(){const e=this.model;if(!e)return;const t=e.contextWindow??0;if(t<=0)return;const s=this.sessionManager.getBranch(),o=y(s);if(o){const c=s.lastIndexOf(o);let g=!1;for(let l=s.length-1;l>c;l--){const m=s[l];if(m.type==="message"&&m.message.role==="assistant"){const h=m.message;if(h.stopReason!=="aborted"&&h.stopReason!=="error"){b(h.usage)>0&&(g=!0);break}}}if(!g)return{tokens:null,contextWindow:t,percent:null}}const r=U(this.messages),i=r.tokens/t*100;return{tokens:r.tokens,contextWindow:t,percent:i}}async exportToHtml(e){return await Z({sessionManager:this.sessionManager,state:this.state,outputPath:e,themeName:this.settingsManager.getTheme(),extensionRunner:this._extensionRunner,theme:this._theme})}getLastAssistantText(){return ee(this.messages)}hasExtensionHandlers(e){return this._extensionRunner?.hasHandlers(e)??!1}get extensionRunner(){return this._extensionRunner}}export{Qe as AgentSession,ze as CycleModelError,qe as parseSkillBlock,R as pruneRecoverableErrorTail};
|
|
14
|
+
${o}`:m}catch(i){return this._extensionRunner?.emitError({extensionPath:r.filePath,event:"skill_expansion",error:i instanceof Error?i.message:String(i)}),e}}async steer(e,t){e.startsWith("/")&&this._throwIfExtensionCommand(e);let s=this._expandSkillCommand(e);s=y(s,[...this.promptTemplates]),await this._queueSteer(s,t)}async followUp(e,t){e.startsWith("/")&&this._throwIfExtensionCommand(e);let s=this._expandSkillCommand(e);s=y(s,[...this.promptTemplates]),await this._queueFollowUp(s,t)}async _queueSteer(e,t){this._steeringMessages.push(e);const s=[{type:"text",text:e}];t&&s.push(...t),this.agent.steer({role:"user",content:s,timestamp:Date.now()})}async _queueFollowUp(e,t){this._followUpMessages.push(e);const s=[{type:"text",text:e}];t&&s.push(...t),this.agent.followUp({role:"user",content:s,timestamp:Date.now()})}_throwIfExtensionCommand(e){if(!this._extensionRunner)return;const t=e.indexOf(" "),s=t===-1?e.slice(1):e.slice(1,t);if(this._extensionRunner.getCommand(s))throw new Error(`Extension command "/${s}" cannot be queued. Use prompt() or execute the command when not streaming.`)}async sendCustomMessage(e,t){const s={role:"custom",customType:e.customType,content:e.content,display:e.display,details:e.details,timestamp:Date.now()};t?.deliverAs==="nextTurn"?this._pendingNextTurnMessages.push(s):this.isStreaming?t?.deliverAs==="followUp"?this.agent.followUp(s):this.agent.steer(s):t?.triggerTurn?await this.agent.prompt(s):(this.agent.appendMessage(s),this.sessionManager.appendCustomMessageEntry(e.customType,e.content,e.display,e.details),this._emit({type:"message_start",message:s}),this._emit({type:"message_end",message:s}))}async sendUserMessage(e,t){let s,o;if(typeof e=="string")s=e;else{const r=[];o=[];for(const i of e)i.type==="text"?r.push(i.text):i.type==="image"&&o.push(i);s=r.join(`
|
|
15
|
+
`),o.length===0&&(o=void 0)}await this.prompt(s,{expandPromptTemplates:!1,streamingBehavior:t?.deliverAs,images:o,source:"extension"})}clearQueue(){const e=[...this._steeringMessages],t=[...this._followUpMessages];return this._steeringMessages=[],this._followUpMessages=[],this.agent.clearAllQueues(),{steering:e,followUp:t}}get pendingMessageCount(){return this._steeringMessages.length+this._followUpMessages.length}getSteeringMessages(){return this._steeringMessages}getFollowUpMessages(){return this._followUpMessages}get resourceLoader(){return this._resourceLoader}async abort(){this.abortRetry(),this.agent.abort(),await this.agent.waitForIdle(),this._extensionRunner?.emit({type:"agent_abort"})}async newSession(e){return this._lifecycleController.newSession(e)}async setModel(e){await this._modelController.setModel(e),this._emitDebug("basic","model","model_change",{provider:e.api,modelId:e.id})}async cycleModel(e="forward"){const t=await this._modelController.cycleModel(e);return t&&this._emitDebug("basic","model","model_cycle",{direction:e,provider:t.model.api,modelId:t.model.id}),t}setThinkingLevel(e){this._modelController.setThinkingLevel(e)}setAgentLoopFramework(e){this._modelController.setAgentLoopFramework(e)}setLoopPolicy(e){this._modelController.setLoopPolicy(e)}cycleThinkingLevel(){return this._modelController.cycleThinkingLevel()}getAvailableThinkingLevels(){return this._modelController.getAvailableThinkingLevels()}supportsXhighThinking(){return this._modelController.supportsXhighThinking()}supportsThinking(){return this._modelController.supportsThinking()}setSteeringMode(e){this.agent.setSteeringMode(e),this.settingsManager.setSteeringMode(e)}setFollowUpMode(e){this.agent.setFollowUpMode(e),this.settingsManager.setFollowUpMode(e)}async compact(e){return this._compactionController.compact(e)}abortCompaction(){this._compactionController.abort()}abortBranchSummary(){this._sessionTreeController.abortBranchSummary()}async _checkCompaction(e,t=!0){const s=this.settingsManager.getCompactionSettings();if(!s.enabled||t&&e.stopReason==="aborted")return;const o=this.model?.contextWindow??0,r=this.model&&e.provider===this.model.provider&&e.model===this.model.id,i=x(this.sessionManager.getBranch()),l=i!==null&&e.timestamp<new Date(i.timestamp).getTime();if(r&&!l&&b(e,o)){const g=this.agent.state.messages;g.length>0&&g[g.length-1].role==="assistant"&&this.agent.replaceMessages(g.slice(0,-1)),await this._runAutoCompaction("overflow",!0);return}if(e.stopReason==="error")return;const m=w(e.usage);D(m,o,s)&&await this._runAutoCompaction("threshold",!1)}async _recoverModelErrorInLoop(e){const t=this.settingsManager.getCompactionSettings();if(e.message.role!=="assistant")return{action:"stop"};const s=e.message;if(e.errorSubtype!=="context_overflow"){if(!this._retryCoordinator.isRetryableError(s))return{action:"stop"};if(!await this._retryCoordinator.handleErrorInLoop(s))return{action:"stop"};const h=A(this.agent.state.messages,s);return this.agent.replaceMessages(h),{action:"retry",messages:h,transition:{reason:"model_error_recovery",subtype:e.errorSubtype,attempt:e.attempt}}}if(!t.enabled)return{action:"stop"};const o=this.model?.contextWindow??0;if(!(this.model&&s.provider===this.model.provider&&s.model===this.model.id)||!b(s,o))return{action:"stop"};const i=x(this.sessionManager.getBranch());if(i!==null&&s.timestamp<new Date(i.timestamp).getTime())return{action:"stop"};const m=this.agent.state.messages;this.agent.replaceMessages(A(m,s));const g=await this._runAutoCompaction("overflow",!0,{triggerContinue:!1});return g?{action:"retry",messages:g,transition:{reason:"model_error_recovery",subtype:e.errorSubtype,attempt:e.attempt}}:{action:"stop"}}async _runAutoCompaction(e,t,s){const o=s?.triggerContinue??!0,r=await this._compactionController.runAuto(e,t);if(r!==void 0){if(t&&o){const i=this.agent.state.messages,l=i[i.length-1];l?.role==="assistant"&&l.stopReason==="error"&&this.agent.replaceMessages(i.slice(0,-1)),setTimeout(()=>{this.agent.continue().catch(()=>{})},100)}else!t&&this.agent.hasQueuedMessages()&&setTimeout(()=>{this.agent.continue().catch(()=>{})},100);return r}}setAutoCompactionEnabled(e){this._compactionController.setAutoCompactionEnabled(e)}get autoCompactionEnabled(){return this._compactionController.autoCompactionEnabled}async bindExtensions(e){e.uiContext!==void 0&&(this._extensionUIContext=e.uiContext),e.commandContextActions!==void 0&&(this._extensionCommandContextActions=e.commandContextActions),e.shutdownHandler!==void 0&&(this._extensionShutdownHandler=e.shutdownHandler),e.onError!==void 0&&(this._extensionErrorListener=e.onError),this._extensionRunner&&(this._applyExtensionBindings(this._extensionRunner),await this._extensionRunner.emit({type:"session_start"}),await this.extendResourcesFromExtensions("startup")),this._emitDebug("basic","extension","extensions_bound")}async extendResourcesFromExtensions(e){if(!this._extensionRunner?.hasHandlers("resources_discover"))return;const{skillPaths:t,promptPaths:s,themePaths:o}=await this._extensionRunner.emitResourcesDiscover(this._cwd,e);if(t.length===0&&s.length===0&&o.length===0)return;const r={skillPaths:this.buildExtensionResourcePaths(t),promptPaths:this.buildExtensionResourcePaths(s),themePaths:this.buildExtensionResourcePaths(o)};this._resourceLoader.extendResources(r),this._baseSystemPrompt=this._rebuildSystemPrompt(this.getActiveToolNames()),this.agent.setSystemPrompt(this._baseSystemPrompt)}buildExtensionResourcePaths(e){return e.map(t=>{const s=this.getExtensionSourceLabel(t.extensionPath),o=t.extensionPath.startsWith("<")?void 0:I(t.extensionPath);return{path:t.path,metadata:{source:s,scope:"temporary",origin:"top-level",baseDir:o}}})}getExtensionSourceLabel(e){return e.startsWith("<")?`extension:${e.replace(/[<>]/g,"")}`:`extension:${P(e).replace(/\.(ts|js)$/,"")}`}_applyExtensionBindings(e){e.setUIContext(this._extensionUIContext),e.bindCommandContext(this._extensionCommandContextActions),this._extensionErrorUnsubscriber?.(),this._extensionErrorUnsubscriber=this._extensionErrorListener?e.onError(this._extensionErrorListener):void 0}_bindExtensionCore(e){const t=this;ne(e,{promptTemplates:this.promptTemplates,resourceLoader:this._resourceLoader,modelRegistry:this.modelRegistry,sessionManager:this.sessionManager,settingsManager:this.settingsManager,shutdownHandler:this._extensionShutdownHandler,soulManager:this._soulManager,get model(){return t.model},get thinkingLevel(){return t.thinkingLevel},get isStreaming(){return t.isStreaming},get pendingMessageCount(){return t.pendingMessageCount},get systemPrompt(){return t.systemPrompt},sendCustomMessage:n((s,o)=>this.sendCustomMessage(s,o),"sendCustomMessage"),sendUserMessage:n((s,o)=>this.sendUserMessage(s,o),"sendUserMessage"),executeSlashCommand:n(s=>this.executeSlashCommand(s),"executeSlashCommand"),getActiveToolNames:n(()=>this.getActiveToolNames(),"getActiveToolNames"),getAllTools:n(()=>this.getAllTools(),"getAllTools"),setActiveToolsByName:n(s=>this.setActiveToolsByName(s),"setActiveToolsByName"),setModel:n(s=>this.setModel(s),"setModel"),setThinkingLevel:n(s=>this.setThinkingLevel(s),"setThinkingLevel"),abort:n(()=>this.abort(),"abort"),clearFollowUpQueue:n(()=>{this.agent.clearFollowUpQueue(),this._followUpMessages=[]},"clearFollowUpQueue"),getContextUsage:n(()=>this.getContextUsage(),"getContextUsage"),compact:n(s=>this.compact(s),"compact")})}_buildRuntime(e){const t=this._baseToolsOverride?this._baseToolsOverride:K(this._cwd,this.settingsManager);this._baseToolRegistry=new Map(Object.entries(t).map(([c,h])=>[c,h]));const s=this._resourceLoader.getExtensions();if(e.flagValues)for(const[c,h]of e.flagValues)s.runtime.flagValues.set(c,h);const o=s.extensions.length>0,r=this._customTools.length>0;this._extensionRunner=o||r?new W(s.extensions,s.runtime,this._cwd,this._agentDir,this.sessionManager,this._modelRegistry):void 0,this._extensionRunnerRef&&(this._extensionRunnerRef.current=this._extensionRunner),this._extensionRunner&&(this._bindExtensionCore(this._extensionRunner),this._applyExtensionBindings(this._extensionRunner),this._extensionRunner.setTelemetrySink(j({workspaceRoot:this._cwd})));const i=this._toolRuntimeController.build({baseTools:this._baseToolRegistry,baseToolsOverride:this._baseToolsOverride,customTools:this._customTools,activeToolNames:e.activeToolNames,includeAllExtensionTools:e.includeAllExtensionTools,extensionRunner:this._extensionRunner}),l=n(c=>{const h=ue(c);h&&this._emit(h)},"onSubAgentEvent");this._agentTool=ae({parentSession:this,parentPermissionMode:"default",parentModel:this.model,createSession:this._createSessionFactory,onSubAgentEvent:l}),i.activeTools.push(this._agentTool),this._toolOrchestrator.registerTool(M,this._agentTool);const m=le({parentSession:this,parentPermissionMode:"default",parentModel:this.model,createSession:this._createSessionFactory,onSubAgentEvent:l});i.activeTools.push(m),this._toolOrchestrator.registerTool(ge,m);const g=he({parentSession:this,parentPermissionMode:"default",parentModel:this.model,createSession:this._createSessionFactory});i.activeTools.push(g),this._toolOrchestrator.registerTool(T,g),i.systemPromptToolNames.includes(M)||i.systemPromptToolNames.push(M),i.systemPromptToolNames.includes(T)||i.systemPromptToolNames.push(T),this.agent.setTools(i.activeTools),this._baseSystemPrompt=this._rebuildSystemPrompt(i.systemPromptToolNames),this.agent.setSystemPrompt(this._baseSystemPrompt)}async _refreshMcpTools(){if(!this._mcpToolsFactory)return 0;try{const e=await this._mcpToolsFactory();return this._customTools=[...this._staticCustomTools,...e],e.length}catch(e){this._emit({type:"sdk:error",source:"mcp",error:e});const t=this._customTools.filter(s=>s.name.startsWith("mcp_"));return this._customTools=[...this._staticCustomTools,...t],t.length}}async warmupMcpTools(){if(this._mcpToolsFactory)try{const e=await this._refreshMcpTools();this._buildRuntime({activeToolNames:this.getActiveToolNames(),flagValues:this._extensionRunner?.getFlagValues(),includeAllExtensionTools:!0}),this._emit({type:"sdk:mcp_ready",toolCount:e}),this._emitDebug("basic","mcp","mcp_warmup_complete",{toolCount:e})}catch(e){this._emit({type:"sdk:error",source:"mcp",error:e})}}async reload(){this._emitDebug("basic","resource","reload_start");const e=this._extensionRunner?.getFlagValues();if(await this._extensionRunner?.emit({type:"session_shutdown"}),this.settingsManager.reload(),F(),await this._resourceLoader.reload(),await this._refreshMcpTools(),this._soulManagerFactory)try{this._soulManager=await this._soulManagerFactory(),this._lastSoulInjection=void 0}catch(s){this._emit({type:"sdk:error",source:"soul",error:s})}this._buildRuntime({activeToolNames:this.getActiveToolNames(),flagValues:e,includeAllExtensionTools:!0});const t=this._extensionUIContext||this._extensionCommandContextActions||this._extensionShutdownHandler||this._extensionErrorListener;this._extensionRunner&&t&&(await this._extensionRunner.emit({type:"session_start"}),await this.extendResourcesFromExtensions("reload")),this._emitDebug("basic","resource","reload_end")}_createRetryHost(){return{getContextWindow:n(()=>this.model?.contextWindow??0,"getContextWindow"),getRetrySettings:n(()=>this.settingsManager.getRetrySettings(),"getRetrySettings"),removeLastAssistantMessage:n(()=>{const e=this.agent.state.messages;e.length>0&&e[e.length-1].role==="assistant"&&this.agent.replaceMessages(e.slice(0,-1))},"removeLastAssistantMessage"),triggerContinue:n(()=>{setTimeout(()=>{this.agent.continue().catch(()=>{})},0)},"triggerContinue"),emitEvent:n(e=>{this._emit(e)},"emitEvent")}}abortRetry(){this._retryCoordinator.abort()}async waitForRetry(){await this._retryCoordinator.waitForCompletion()}get isRetrying(){return this._retryCoordinator.isActive}get autoRetryEnabled(){return this.settingsManager.getRetryEnabled()}setAutoRetryEnabled(e){this.settingsManager.setRetryEnabled(e)}async executeBash(e,t,s){return this._bashRunner.execute(e,t,s)}recordBashResult(e,t,s){this._bashRunner.recordResult(e,t,s)}abortBash(){this._bashRunner.abort()}get isBashRunning(){return this._bashRunner.isRunning}get hasPendingBashMessages(){return this._bashRunner.hasPending}async switchSession(e){return this._lifecycleController.switchSession(e)}setSessionName(e){this.sessionManager.appendSessionInfo(e)}async fork(e){return this._lifecycleController.fork(e)}async navigateTree(e,t={}){return this._sessionTreeController.navigateTree(e,t)}getUserMessagesForForking(){const e=this.sessionManager.getEntries(),t=[];for(const s of e){if(s.type!=="message"||s.message.role!=="user")continue;const o=this._extractUserMessageText(s.message.content);o&&t.push({entryId:s.id,text:o})}return t}_extractUserMessageText(e){return typeof e=="string"?e:Array.isArray(e)?e.filter(t=>t.type==="text").map(t=>t.text).join(""):""}getSessionStats(){const e=this.state,t=e.messages.filter(h=>h.role==="user").length,s=e.messages.filter(h=>h.role==="assistant").length,o=e.messages.filter(h=>h.role==="toolResult").length;let r=0,i=0,l=0,m=0,g=0,c=0;for(const h of e.messages)if(h.role==="assistant"){const d=h;r+=d.content.filter(_=>_.type==="toolCall").length,i+=d.usage.input,l+=d.usage.output,m+=d.usage.cacheRead,g+=d.usage.cacheWrite,c+=d.usage.cost.total}return{sessionFile:this.sessionFile,sessionId:this.sessionId,userMessages:t,assistantMessages:s,toolCalls:r,toolResults:o,totalMessages:e.messages.length,tokens:{input:i,output:l,cacheRead:m,cacheWrite:g,total:i+l+m+g},cost:c}}getContextUsage(){const e=this.model;if(!e)return;const t=e.contextWindow??0;if(t<=0)return;const s=this.sessionManager.getBranch(),o=x(s);if(o){const l=s.lastIndexOf(o);let m=!1;for(let g=s.length-1;g>l;g--){const c=s[g];if(c.type==="message"&&c.message.role==="assistant"){const h=c.message;if(h.stopReason!=="aborted"&&h.stopReason!=="error"){w(h.usage)>0&&(m=!0);break}}}if(!m)return{tokens:null,contextWindow:t,percent:null}}const r=B(this.messages),i=r.tokens/t*100;return{tokens:r.tokens,contextWindow:t,percent:i}}async exportToHtml(e){return await ee({sessionManager:this.sessionManager,state:this.state,outputPath:e,themeName:this.settingsManager.getTheme(),extensionRunner:this._extensionRunner,theme:this._theme})}getLastAssistantText(){return te(this.messages)}hasExtensionHandlers(e){return this._extensionRunner?.hasHandlers(e)??!1}get extensionRunner(){return this._extensionRunner}}export{Ge as AgentSession,Je as CycleModelError,Qe as parseSkillBlock,A as pruneRecoverableErrorTail};
|
|
@@ -84,6 +84,14 @@ export interface CreateAgentSessionOptions extends SoulOptionsContract {
|
|
|
84
84
|
deferMcpInit?: boolean;
|
|
85
85
|
/** Resource loader. When omitted, DefaultResourceLoader is used. */
|
|
86
86
|
resourceLoader?: ResourceLoader;
|
|
87
|
+
/** Additional directories to search for skills. Appended to default paths. */
|
|
88
|
+
additionalSkillPaths?: string[];
|
|
89
|
+
/** Additional directories to search for AGENT.md/CLAUDE.md context files. */
|
|
90
|
+
additionalAgentDirs?: string[];
|
|
91
|
+
/** Custom MCP config file path. Overrides default agentDir/mcp.json. */
|
|
92
|
+
mcpConfigPath?: string;
|
|
93
|
+
/** Debug event verbosity level. "off" = none, "basic" = lifecycle, "verbose" = all. Default: "off" */
|
|
94
|
+
debugLevel?: "off" | "basic" | "verbose";
|
|
87
95
|
/** Session manager. Default: SessionManager.create(cwd) */
|
|
88
96
|
sessionManager?: SessionManager;
|
|
89
97
|
/** Settings manager. Default: SettingsManager.create(cwd, agentDir) */
|
package/dist/core/runtime/sdk.js
CHANGED
|
@@ -1 +1 @@
|
|
|
1
|
-
var Q=Object.defineProperty;var n=(e,s)=>Q(e,"name",{value:s,configurable:!0});import{join as E}from"node:path";import{Agent as X}from"@pencil-agent/agent-core";import{getAgentDir as Y,getDocsPath as Z}from"../../config.js";import{AgentSession as ee}from"./agent-session.js";import{AuthStorage as te}from"../platform/config/auth-storage.js";import{DEFAULT_THINKING_LEVEL as B}from"../platform/config/defaults.js";import{convertToLlm as oe}from"../messages.js";import{MCPManager as re}from"../mcp/mcp-manager.js";import{registerFigmaMcpOAuthProvider as ne}from"../mcp/figma-auth.js";import{ModelRegistry as le}from"../model-registry.js";import{findInitialModel as ae}from"../model-resolver.js";import{DefaultResourceLoader as ie}from"../platform/config/resource-loader.js";import{SessionManager as se}from"../session/session-manager.js";import{SettingsManager as F}from"../platform/config/settings-manager.js";import{defaultAgentDirContext as ce}from"../agent-dir/agent-dir-context.js";import{time as
|
|
1
|
+
var Q=Object.defineProperty;var n=(e,s)=>Q(e,"name",{value:s,configurable:!0});import{join as E}from"node:path";import{Agent as X}from"@pencil-agent/agent-core";import{getAgentDir as Y,getDocsPath as Z}from"../../config.js";import{AgentSession as ee}from"./agent-session.js";import{AuthStorage as te}from"../platform/config/auth-storage.js";import{DEFAULT_THINKING_LEVEL as B}from"../platform/config/defaults.js";import{convertToLlm as oe}from"../messages.js";import{MCPManager as re}from"../mcp/mcp-manager.js";import{registerFigmaMcpOAuthProvider as ne}from"../mcp/figma-auth.js";import{ModelRegistry as le}from"../model-registry.js";import{findInitialModel as ae}from"../model-resolver.js";import{DefaultResourceLoader as ie}from"../platform/config/resource-loader.js";import{SessionManager as se}from"../session/session-manager.js";import{SettingsManager as F}from"../platform/config/settings-manager.js";import{defaultAgentDirContext as ce}from"../agent-dir/agent-dir-context.js";import{time as v}from"../platform/timings.js";import{isSoulEnabled as de,createSoulManager as z}from"../soul-integration.js";import{allTools as O,bashTool as ge,codingTools as me,createBashTool as ue,createCodingTools as fe,createEditTool as pe,createFindTool as he,createGrepTool as Te,createLsTool as ve,createReadOnlyTools as ye,createReadTool as xe,createWriteTool as Se,editTool as Me,findTool as Pe,grepTool as we,lsTool as ke,readOnlyTools as Ce,readTool as Ae,writeTool as Le}from"../tools/index.js";const be={error:n(()=>{},"error"),warn:n(()=>{},"warn"),info:n(()=>{},"info")},Re={error:n((e,...s)=>console.error(e,...s),"error"),warn:n((e,...s)=>console.warn(e,...s),"warn"),info:n((e,...s)=>console.log(e,...s),"info")};function $e(e,s,w){if(e instanceof F)return e;const a=F.create(s,w);if(!e||typeof e!="object"||typeof e.getSettings!="function")return a;const g=a;for(const c of Object.keys(e)){const p=e[c];typeof p=="function"&&(g[c]=p.bind(e))}return a}n($e,"normalizeSettingsManager");function Ye(){return Y()}n(Ye,"getDefaultAgentDir");async function De(e={}){ne();const s=typeof import.meta.url=="string"&&import.meta.url.includes("node_modules"),w=process.env.NODE_ENV==="production"||process.env.NODE_ENV!=="development"&&s,a=e.silent?be:e.logger??Re,g=e.cwd??process.cwd(),c=e.agentCtx??ce(),p=e.agentDir??c.path;let y=e.resourceLoader;const r=$e(e.settingsManager,g,c),N=r.getSettings().locale??"en",{setLocale:j}=await import("../platform/i18n/index.js");j(N);const _=e.authStorage??te.create(c),U=E(p,"models.json"),h=e.modelRegistry??new le(_,U),m=e.sessionManager??se.create(g,void 0,c);y||(y=new ie({cwd:g,agentDir:p,settingsManager:r,agentCtx:c,additionalSkillPaths:e.additionalSkillPaths,additionalAgentDirs:e.additionalAgentDirs}),await y.reload(),v("resourceLoader.reload"));const d=m.buildSessionContext(),P=d.messages.length>0,L=m.getBranch().some(t=>t.type==="thinking_level_change");let l=e.model,x;if(!l&&P&&d.model){const t=h.find(d.model.provider,d.model.modelId);t&&await h.getApiKey(t)&&(l=t),l||(x=`Could not restore model ${d.model.provider}/${d.model.modelId}`)}l||(l=(await ae({scopedModels:[],isContinuing:P,defaultProvider:r.getDefaultProvider(),defaultModelId:r.getDefaultModel(),defaultThinkingLevel:r.getDefaultThinkingLevel(),modelRegistry:h})).model,l?x&&(x+=`. Using ${l.provider}/${l.id}`):x=`No models available. Use /login or set an API key environment variable. See ${E(Z(),"providers.md")}. Then use /model to select a model.`);let u;e.thinkingLevel!==void 0?u=e.thinkingLevel:P?u=L?d.thinkingLevel:r.getDefaultThinkingLevel()??B:u=r.getDefaultThinkingLevel()??B,(!l||!l.reasoning)&&(u="off");const H=["read","bash","edit","write"],K=e.tools?e.tools.map(t=>t.name).filter(t=>t in O):H;let S;const V=n(t=>{const o=oe(t);return r.getBlockImages()?o.map(i=>{if(i.role==="user"||i.role==="toolResult"){const T=i.content;if(Array.isArray(T)&&T.some(C=>C.type==="image")){const C=T.map(M=>M.type==="image"?{type:"text",text:"Image reading is disabled."}:M).filter((M,A,I)=>!(M.type==="text"&&M.text==="Image reading is disabled."&&A>0&&I[A-1].type==="text"&&I[A-1].text==="Image reading is disabled."));return{...i,content:C}}}return i}):o},"convertToLlmWithBlockImages"),b={};S=new X({initialState:{systemPrompt:"",model:l,thinkingLevel:u,tools:[]},convertToLlm:V,sessionId:m.getSessionId(),transformContext:n(async t=>{const o=b.current;return o?o.emitContext(t):t},"transformContext"),steeringMode:r.getSteeringMode(),followUpMode:r.getFollowUpMode(),transport:r.getTransport(),agentLoopFramework:e.agentLoopFramework??r.getAgentLoopFramework(),thinkingBudgets:r.getThinkingBudgets(),maxRetryDelayMs:r.getRetrySettings().maxDelayMs,maxToolResultBatchSizeChars:e.maxToolResultBatchSizeChars??e.loopPolicy?.maxToolResultBatchSizeChars??r.getAgentLoopSettings().maxToolResultBatchSizeChars,maxToolConcurrency:e.maxToolConcurrency??e.loopPolicy?.maxToolConcurrency,maxTurnsPerPrompt:e.maxTurnsPerPrompt??e.loopPolicy?.maxTurnsPerPrompt,maxToolCallsPerPrompt:e.maxToolCallsPerPrompt??e.loopPolicy?.maxToolCallsPerPrompt,outputTokenBudget:e.outputTokenBudget??e.loopPolicy?.outputTokenBudget,maxOutputTokenRecoveryAttempts:e.maxOutputTokenRecoveryAttempts??e.loopPolicy?.maxOutputTokenRecoveryAttempts,maxModelErrorRecoveryAttempts:e.maxModelErrorRecoveryAttempts??e.loopPolicy?.maxModelErrorRecoveryAttempts,maxStopHookContinuations:e.maxStopHookContinuations??e.loopPolicy?.maxStopHookContinuations,getApiKey:n(async t=>{const o=t||S.state.model?.provider;if(!o)throw new Error("No model selected");const i=await h.getApiKeyForProvider(o);if(!i){const T=S.state.model;throw T&&h.isUsingOAuth(T)?new Error(`Authentication failed for "${o}". Credentials may have expired or network is unavailable. Run '/login ${o}' to re-authenticate.`):new Error(`No API key found for "${o}". Set an API key environment variable or run '/login ${o}'.`)}return i},"getApiKey")}),P?(S.replaceMessages(d.messages),L||m.appendThinkingLevelChange(u)):(l&&m.appendModelChange(l.provider,l.id),m.appendThinkingLevelChange(u)),v("agent.construct");let f;const W=[],G=e.customTools??[];let R;e.enableMCP&&(process.once("exit",()=>f?.dispose()),R=n(async()=>{try{f?.dispose()}catch{}f=new re({mcpConfigPath:e.mcpConfigPath}),f.setWorkingDir(g),await f.initialize(),v("mcp.initialize");const t=f.getStatus();if(w){const o=t.startedServers,i=t.failedServers;o.length>0&&a.info(`MCP: ${o.length} server(s) ready (${o.join(", ")})`),i.length>0&&a.warn(`MCP: ${i.length} failed (${i.join(", ")})`)}else if(t.toolCount===0){const o=t.failedServers.length>0?` failed=${t.failedServers.join(",")}`:"";a.warn(`MCP enabled but no tools loaded (enabled=${t.enabledServers.length}, started=${t.startedServers.length}, tools=0).${o}`)}else a.info(`MCP tools loaded: ${t.toolCount}`),t.failedServers.length>0&&a.warn(`MCP: ${t.failedServers.length} server(s) failed to start (${t.failedServers.join(", ")}); tools from other servers are still available.`);return f.getTools()},"mcpToolsFactory"));let k,$;if(de(e)){try{const t=await z(c);t?(k=t,await t.initialize(),v("soul.initialize")):a.warn("Soul not available (nanosoul package not installed). Skipping...")}catch(t){a.warn(`Failed to initialize Soul: ${t}`)}$=n(async()=>{try{const t=await z(c);return t?(await t.initialize(),v("soul.initialize"),t):null}catch(t){return a.warn(`Failed to refresh Soul: ${t}`),null}},"soulManagerFactory")}const D=new ee({agent:S,sessionManager:m,settingsManager:r,cwd:g,agentDir:p,agentCtx:c,scopedModels:e.scopedModels,resourceLoader:y,customTools:G,initialMcpTools:W,mcpToolsFactory:R,modelRegistry:h,initialActiveToolNames:K,extensionRunnerRef:b,soulManager:k,soulManagerFactory:$,signal:e.signal,theme:e.theme,createSession:De,debugLevel:e.debugLevel});v("session.construct"),e.enableMCP&&!e.deferMcpInit&&await D.warmupMcpTools();const q=y.getExtensions();return{session:D,extensionsResult:q,modelFallbackMessage:x,soulManager:k}}n(De,"createAgentSession");export{O as allBuiltInTools,ge as bashTool,me as codingTools,De as createAgentSession,ue as createBashTool,fe as createCodingTools,pe as createEditTool,he as createFindTool,Te as createGrepTool,ve as createLsTool,ye as createReadOnlyTools,xe as createReadTool,Se as createWriteTool,Re as defaultLogger,Me as editTool,Pe as findTool,we as grepTool,ke as lsTool,Ce as readOnlyTools,Ae as readTool,be as silentLogger,Le as writeTool};
|
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
var
|
|
2
|
-
`)}h(a,"dbg");class
|
|
1
|
+
var x=Object.defineProperty;var h=(d,e)=>x(d,"name",{value:e,configurable:!0});import*as g from"node:fs";import*as f from"node:os";import*as l from"node:path";const S=process.env.NANOPENCIL_DEBUG==="1",p=l.join(f.homedir(),".nanopencil","agent","nanopencil-debug.log");function a(d){if(S)try{g.mkdirSync(l.dirname(p),{recursive:!0}),g.appendFileSync(p,`[${new Date().toISOString()}] [submit] ${d}
|
|
2
|
+
`)}catch{}}h(a,"dbg");class I{static{h(this,"InputSubmitController")}ctx;constructor(e){this.ctx=e}async handleSubmit(e){const t=e.trim();if(!t)return;if(a(`handleSubmit: "${t.slice(0,80)}"`),await this.ctx.image.awaitPendingPaste(),await this.ctx.slash.execute(t)){a("handleSubmit \u2192 built-in slash handled");return}const s=t.match(/\s+\/persona\b/);if(s){const n=t.slice(s.index+1),r=t.slice(0,s.index).trim();this.ctx.editor.setText(""),await this.ctx.commands.handlePersonaCommand(n),r&&await this.ctx.session.promptAfterRender(r);return}if(t.startsWith("!")){const n=t.startsWith("!!"),r=n?t.slice(2).trim():t.slice(1).trim();if(r){if(this.ctx.session.isBashRunning()){this.ctx.render.showWarning("A bash command is already running. Press Esc to cancel it first."),this.ctx.editor.setText(t);return}this.ctx.editor.addToHistory(t),await this.ctx.commands.handleBashCommand(r,n),this.ctx.editor.setBashMode(!1),this.ctx.editor.updateBorderColor();return}}if(this.ctx.session.isCompacting()){this.ctx.commands.isExtensionCommand(t)?(this.ctx.editor.addToHistory(t),this.ctx.editor.setText(""),await this.ctx.session.promptAfterRender(t)):this.ctx.session.queueCompactionMessage(t,"steer");return}if(this.ctx.session.isStreaming()){await this.handleStreamingSubmit(t);return}await this.handleIdleSubmit(t)}async handleStreamingSubmit(e){this.ctx.editor.addToHistory(e),this.ctx.editor.setText("");const t=[{type:"text",text:e}];this.ctx.render.addOptimisticUserMessage(e,t),this.ctx.render.requestRender();const s=await this.ctx.image.extractImagesFromText(e),n=s.images;let r=[];const i=this.ctx.image.takePendingAttachments();i.length>0&&(r=i.map(m=>m.path));const o=this.ctx.session.getModel();(n.length>0||r.length>0)&&o&&!o.input.includes("image")&&(n.length=0,r=[],this.ctx.render.showStatus(`Images dropped: ${o.name} does not support images.${this.imageSupportSuggestion(o,!1)}`),this.ctx.render.requestRender());let c=s.text;if(r.length>0){const m=this.ctx.session.getCwd();c=r.map(u=>`@${l.relative(m,u).replace(/\\/g,"/")}`).join(" ")+" "+c}await this.ctx.session.promptAfterRender(c,{streamingBehavior:"steer",images:n.length>0?n:void 0}),this.ctx.render.updatePendingMessagesDisplay(),this.ctx.render.requestRender()}async handleIdleSubmit(e){a(`handleIdleSubmit: "${e.slice(0,80)}"`),this.ctx.render.flushPendingBashComponents(),this.ctx.editor.addToHistory(e),this.ctx.editor.setText("");const{text:t,images:s}=await this.ctx.image.extractImagesFromText(e),n=this.ctx.image.takePendingAttachments();if(n.length>0){const i=await this.ctx.image.processAttachmentFiles(n);s.push(...i)}if(s.length>0){const i=this.ctx.session.getModel();i&&!i.input.includes("image")&&(this.ctx.render.showWarning(`Model "${i.name}" does not support image input. Images have been removed from this message.${this.imageSupportSuggestion(i,!0)}`),s.length=0)}if(this.ctx.commands.isExtensionCommand(t))a("handleIdleSubmit \u2192 skipping optimistic message for extension command");else{const i=[{type:"text",text:t}];s.length>0&&i.push(...s),this.ctx.render.addOptimisticUserMessage(t,i),this.ctx.render.requestRender(),a("handleIdleSubmit \u2192 optimistic message added")}if(this.ctx.editor.handleExternalInput(e)){a("handleIdleSubmit \u2192 handed off to main loop via handleExternalInput");return}try{delete process.env.NANOPENCIL_JUST_SWITCHED_PERSONA,await this.ctx.session.promptAfterRender(t,{images:s.length>0?s:void 0}),a("handleIdleSubmit \u2192 promptAfterRender returned normally")}catch(i){a(`handleIdleSubmit \u2192 promptAfterRender threw: ${i}`),this.ctx.render.rollbackFirstOptimisticUserMessageIfMatches(t);const o=i instanceof Error?i.message:"Unknown error occurred";this.ctx.render.showError(o)}this.ctx.render.updatePendingMessagesDisplay(),this.ctx.render.requestRender(),this.ctx.image.cleanupClipboardImages()}imageSupportSuggestion(e,t){const s=t?" Try using":" Try";return e.id==="glm-5"||e.id==="glm-5-turbo"?`${s} glm-5v-turbo for image support.`:e.id==="glm-4.5"||e.id==="glm-4.5-air"?`${s} glm-4.5v for image support.`:""}}export{I as InputSubmitController};
|
|
@@ -1,2 +1,2 @@
|
|
|
1
|
-
var
|
|
2
|
-
`)}u(y,"dbg");const H=5e3;class z{static{u(this,"StreamRenderController")}ctx;autoCompactionLoader=void 0;autoCompactionEscapeHandler=void 0;retryLoader=void 0;retryEscapeHandler=void 0;taskUpdateUnsubscribe=void 0;taskAutoHideTimer=void 0;constructor(t){this.ctx=t}async handle(t){y(`handle event: ${t.type}`),await this.ctx.surface.ensureInitialized(),this.ctx.layout.invalidateFooter();const e=this.ctx.state.get(),n=this.ctx.layout.getUi(),a=this.ctx.layout.getChatContainer(),o=this.ctx.layout.getStatusContainer();switch(t.type){case"agent_start":this.retryEscapeHandler&&(this.ctx.escape.setHandler(this.retryEscapeHandler),this.retryEscapeHandler=void 0),this.retryLoader&&(this.retryLoader.stop(),this.retryLoader=void 0),e.loadingAnimation&&e.loadingAnimation.stop(),o.clear(),e.loadingAnimation=new g(n,l,this.ctx.loaders.getDefaultWorkingMessage(),this.ctx.loaders.getSessionId()),o.addChild(e.loadingAnimation),this.ctx.loaders.setBuddyPetState("working","Working..."),e.pendingWorkingMessage!==void 0&&(e.workingMessageOverride=e.pendingWorkingMessage||void 0,e.pendingWorkingMessage=void 0),this.ctx.loaders.startAgentRunTimer(),this.ctx.loaders.updateWorkingMessage({resetStallTimer:!1}),this.ctx.loaders.isInPlanMode()&&(e.planProgress=D(),e.planProgressPanel=new M(n,l),o.addChild(e.planProgressPanel),e.planProgressPanel.update(e.planProgress)),this.refreshTaskPanel(e,n,o).catch(()=>{}),this.taskUpdateUnsubscribe?.(),this.taskUpdateUnsubscribe=_(()=>{this.refreshTaskPanel(e,n,o).catch(()=>{})}),this.ctx.surface.restoreEditorFocusIfPossible(),this.ctx.layout.requestRender();break;case"message_start":if(t.message.role==="custom")this.ctx.layout.addMessageToChat(t.message),this.ctx.layout.requestRender();else if(t.message.role==="user"){const s=this.ctx.surface.getUserMessageText(t.message);if(e.optimisticUserMessages.length>0&&e.optimisticUserMessages[0]?.text===s){e.optimisticUserMessages.shift(),this.ctx.layout.updatePendingMessagesDisplay(),this.ctx.layout.requestRender();break}this.ctx.layout.addMessageToChat(t.message),this.ctx.layout.updatePendingMessagesDisplay(),this.ctx.layout.requestRender()}else t.message.role==="assistant"&&(e.streamingComponent=new b(void 0,e.hideThinkingBlock,this.ctx.surface.getMarkdownThemeWithSettings()),e.streamingMessage=t.message,a.addChild(e.streamingComponent),e.streamingComponent.updateContent(e.streamingMessage),this.ctx.layout.requestRender());break;case"message_update":if(e.streamingComponent&&t.message.role==="assistant"){e.loadingAnimation&&e.loadingAnimation.resetStallTimer(),e.streamingMessage=t.message,e.streamingComponent.updateContent(e.streamingMessage),a.markDirty(e.streamingComponent);for(const s of e.streamingMessage.content)if(s.type==="toolCall"){if(!this.ctx.toolTrace.shouldRenderToolTrace(s.name))continue;if(e.pendingTools.has(s.id)){const i=e.pendingTools.get(s.id);i&&(i.updateArgs(s.arguments),a.markDirty(i))}else{a.addChild(new h("",0,0));const i=new f(s.name,s.arguments,{showImages:this.ctx.toolTrace.getShowImages()},this.ctx.toolTrace.getRegisteredToolDefinition(s.name),n);i.setExpanded(e.toolOutputExpanded),a.addChild(i),e.pendingTools.set(s.id,i)}}this.ctx.layout.requestRender()}break;case"message_end":if(t.message.role==="user")break;if(e.streamingComponent&&t.message.role==="assistant"){e.streamingMessage=t.message;let s;if(e.streamingMessage.stopReason==="aborted"){const i=this.ctx.runtime.getRetryAttempt();s=i>0?`Aborted after ${i} retry attempt${i>1?"s":""}`:"Operation aborted",e.streamingMessage.errorMessage=s}if(e.streamingComponent.updateContent(e.streamingMessage),a.markDirty(e.streamingComponent),e.streamingMessage.stopReason==="aborted"||e.streamingMessage.stopReason==="error"){s||(s=e.streamingMessage.errorMessage||"Error");for(const[,i]of e.pendingTools.entries())i.updateResult({content:[{type:"text",text:s}],isError:!0}),a.markDirty(i);e.pendingTools.clear()}else for(const[,i]of e.pendingTools.entries())i.setArgsComplete(),a.markDirty(i);e.streamingComponent=void 0,e.streamingMessage=void 0,this.ctx.layout.invalidateFooter()}this.ctx.layout.requestRender();break;case"tool_execution_start":{if(e.planProgress&&e.planProgressPanel&&(L(e.planProgress,t.toolName,t.args),e.planProgressPanel.update(e.planProgress),this.ctx.layout.requestRender()),!this.ctx.toolTrace.shouldRenderToolTrace(t.toolName))break;if(!e.pendingTools.has(t.toolCallId)){const s=new f(t.toolName,t.args,{showImages:this.ctx.toolTrace.getShowImages()},this.ctx.toolTrace.getRegisteredToolDefinition(t.toolName),n);s.setExpanded(e.toolOutputExpanded),a.addChild(s),e.pendingTools.set(t.toolCallId,s),this.ctx.layout.requestRender()}break}case"tool_execution_update":{const s=e.pendingTools.get(t.toolCallId);s&&(s.updateResult({...t.partialResult,isError:!1},!0),a.markDirty(s),this.ctx.layout.requestRender());break}case"tool_execution_end":{const s=e.pendingTools.get(t.toolCallId);s&&(s.updateResult({...t.result,isError:t.isError}),a.markDirty(s),e.pendingTools.delete(t.toolCallId),this.ctx.layout.requestRender());break}case"agent_end":{y("agent_end received");const s=e.agentRunStartMs!==void 0?this.ctx.loaders.formatElapsedSeconds(Date.now()-e.agentRunStartMs):void 0;this.ctx.loaders.stopAgentRunTimer(),e.agentRunStartMs=void 0,e.workingMessageOverride=void 0,e.loadingAnimation&&(e.loadingAnimation.stop(),e.loadingAnimation=void 0,o.clear()),e.streamingComponent&&(a.removeChild(e.streamingComponent),e.streamingComponent=void 0,e.streamingMessage=void 0),e.pendingTools.clear(),e.subAgentPanelComponent&&(o.removeChild(e.subAgentPanelComponent),e.subAgentPanelComponent=void 0),e.subAgentStates.clear(),e.planProgressPanel&&(o.removeChild(e.planProgressPanel),e.planProgressPanel=void 0),e.planProgress=void 0,e.taskStatusPanel&&(o.removeChild(e.taskStatusPanel),e.taskStatusPanel=void 0),this.taskAutoHideTimer&&(clearTimeout(this.taskAutoHideTimer),this.taskAutoHideTimer=void 0),this.ctx.runtime.clearAttachments(),this.ctx.loaders.setBuddyPetState("happy","Done!",{resetTo:"idle",afterMs:1800}),s&&this.ctx.surface.showStatus(`Completed in ${s}`),await this.ctx.runtime.checkShutdownRequested(),this.ctx.surface.restoreEditorFocusIfPossible(),this.ctx.layout.requestRender();break}case"sub_agent_start":{const s={id:t.subAgentId,agentType:t.agentType,description:t.description,isAsync:t.isAsync,isResolved:!1,isError:!1,toolUseCount:0,lastToolName:null,startTime:Date.now()};e.subAgentStates.set(t.subAgentId,s),e.subAgentPanelComponent||(e.subAgentPanelComponent=new S(n,l),o.addChild(e.subAgentPanelComponent)),e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender();break}case"sub_agent_tool_start":{const s=e.subAgentStates.get(t.subAgentId);s&&(s.lastToolName=t.toolName,e.subAgentPanelComponent&&e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender());break}case"sub_agent_tool_end":{const s=e.subAgentStates.get(t.subAgentId);s&&(s.toolUseCount+=1,s.lastToolName=null,e.subAgentPanelComponent&&e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender());break}case"sub_agent_end":{const s=e.subAgentStates.get(t.subAgentId);s&&(s.isResolved=!0,s.isError=!t.success,s.lastToolName=null,e.subAgentPanelComponent&&e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender());break}case"auto_compaction_start":{this.autoCompactionEscapeHandler=this.ctx.escape.getHandler(),this.ctx.escape.setHandler(()=>{this.ctx.runtime.abortCompaction()}),o.clear();const s=t.reason==="overflow"?"Context overflow detected, ":"";this.autoCompactionLoader=new g(n,l,`${s}Auto-compacting... (${this.ctx.loaders.getInterruptKeyHint()} to cancel)`),o.addChild(this.autoCompactionLoader),this.ctx.layout.requestRender();break}case"auto_compaction_end":{this.autoCompactionEscapeHandler&&(this.ctx.escape.setHandler(this.autoCompactionEscapeHandler),this.autoCompactionEscapeHandler=void 0),this.autoCompactionLoader&&(this.autoCompactionLoader.stop(),this.autoCompactionLoader=void 0,o.clear()),t.aborted?this.ctx.surface.showStatus("Auto-compaction cancelled"):t.result?(a.clear(),this.ctx.layout.rebuildChatFromMessages(),this.ctx.layout.addMessageToChat({role:"compactionSummary",tokensBefore:t.result.tokensBefore,summary:t.result.summary,timestamp:Date.now()}),this.ctx.layout.invalidateFooter()):t.errorMessage&&(a.addChild(new A(1)),a.addChild(new h(l.fg("error",t.errorMessage),1,0))),this.ctx.runtime.flushCompactionQueue({willRetry:t.willRetry}),this.ctx.layout.requestRender();break}case"auto_retry_start":{this.retryEscapeHandler=this.ctx.escape.getHandler(),this.ctx.escape.setHandler(()=>{this.ctx.runtime.abortRetry()}),o.clear();const s=Math.round(t.delayMs/1e3);this.retryLoader=new g(n,l,`Retrying (${t.attempt}/${t.maxAttempts}) in ${s}s... (${this.ctx.loaders.getInterruptKeyHint()} to cancel)`),o.addChild(this.retryLoader),this.ctx.layout.requestRender();break}case"auto_retry_end":{this.retryEscapeHandler&&(this.ctx.escape.setHandler(this.retryEscapeHandler),this.retryEscapeHandler=void 0),this.retryLoader&&(this.retryLoader.stop(),this.retryLoader=void 0,o.clear()),t.success||this.ctx.surface.showError(`Retry failed after ${t.attempt} attempts: ${t.finalError||"Unknown error"}`),this.ctx.layout.requestRender();break}}}static TASK_TOOL_NAMES=new Set(["TaskCreate","TaskUpdate","TaskList","TaskGet","TaskStop","TaskDelete"]);async refreshTaskPanel(t,e,n){const a=this.ctx.runtime.getAgentDir();if(a)try{const o=await x(a,m),s=o.filter(d=>!d.metadata?._internal).map(d=>({id:d.id,subject:d.subject,status:d.status,activeForm:d.activeForm,blockedBy:d.blockedBy.filter(p=>o.some(c=>c.id===p&&c.status!=="completed"))}));if(s.length===0){t.taskStatusPanel&&(n.removeChild(t.taskStatusPanel),t.taskStatusPanel=void 0),this.taskAutoHideTimer&&(clearTimeout(this.taskAutoHideTimer),this.taskAutoHideTimer=void 0);return}t.taskStatusPanel||(t.taskStatusPanel=new R(e,l),n.addChild(t.taskStatusPanel)),t.taskStatusPanel.update(s),this.ctx.layout.requestRender(),s.every(d=>d.status==="completed")?this.taskAutoHideTimer||(this.taskAutoHideTimer=setTimeout(async()=>{this.taskAutoHideTimer=void 0;const d=await x(a,m).catch(()=>[]);d.length>0&&d.every(c=>c.status==="completed")&&await w(a,m).catch(()=>{})},H)):this.taskAutoHideTimer&&(clearTimeout(this.taskAutoHideTimer),this.taskAutoHideTimer=void 0)}catch{}}}const I=["Phase 1: Initial Understanding","Phase 2: Design","Phase 3: Review","Phase 4: Final Plan","Phase 5: Call ExitPlanMode"];function D(){return{phases:I.map((r,t)=>({label:r,status:t===0?"in_progress":"pending"})),currentPhaseIndex:0,startTime:Date.now()}}u(D,"createInitialPlanProgress");function L(r,t,e){let n;if(t==="ExitPlanMode"){for(const a of r.phases)a.status="completed";r.currentPhaseIndex=r.phases.length;return}if(t==="Agent")(e.subagent_type??"").toLowerCase()==="plan"?n=1:n=0;else if(t==="Read"||t==="Grep"||t==="Find")n=2;else if(t==="Write"||t==="Edit")n=3;else return void 0;if(n>r.currentPhaseIndex){for(let a=r.currentPhaseIndex;a<n;a++)r.phases[a].status="completed";r.currentPhaseIndex=n,r.phases[n].status="in_progress"}}u(L,"detectPlanPhaseTransition");export{z as StreamRenderController};
|
|
1
|
+
var P=Object.defineProperty;var u=(i,t)=>P(i,"name",{value:t,configurable:!0});import*as h from"node:fs";import*as A from"node:os";import*as f from"node:path";import{Spacer as b,Text as y}from"@pencil-agent/tui";import{AssistantMessageComponent as S}from"../components/assistant-message.js";import{PencilLoader as g}from"../components/pencil-loader.js";import{ToolExecutionComponent as x}from"../components/tool-execution.js";import{SubAgentPanelComponent as M}from"../components/sub-agent-panel.js";import{PlanProgressPanelComponent as R}from"../components/plan-progress-panel.js";import{TaskStatusPanelComponent as _}from"../components/task-status-panel.js";import{theme as l}from"../theme/theme.js";import{listTasks as T,onTasksUpdated as w,resetTaskList as E}from"../../../extensions/builtin/task/task-store.js";import{DEFAULT_TASK_LIST_ID as m}from"../../../extensions/builtin/task/task-types.js";const z=new Set(["TaskCreate","TaskUpdate","TaskList","TaskGet","TaskStop","TaskDelete"]),H=process.env.NANOPENCIL_DEBUG==="1",k=f.join(A.homedir(),".nanopencil","agent","nanopencil-debug.log");function C(i){if(H)try{h.mkdirSync(f.dirname(k),{recursive:!0}),h.appendFileSync(k,`[${new Date().toISOString()}] [render] ${i}
|
|
2
|
+
`)}catch{}}u(C,"dbg");const I=5e3;class Y{static{u(this,"StreamRenderController")}ctx;autoCompactionLoader=void 0;autoCompactionEscapeHandler=void 0;retryLoader=void 0;retryEscapeHandler=void 0;taskUpdateUnsubscribe=void 0;taskAutoHideTimer=void 0;constructor(t){this.ctx=t}async handle(t){C(`handle event: ${t.type}`),await this.ctx.surface.ensureInitialized(),this.ctx.layout.invalidateFooter();const e=this.ctx.state.get(),n=this.ctx.layout.getUi(),a=this.ctx.layout.getChatContainer(),o=this.ctx.layout.getStatusContainer();switch(t.type){case"agent_start":this.retryEscapeHandler&&(this.ctx.escape.setHandler(this.retryEscapeHandler),this.retryEscapeHandler=void 0),this.retryLoader&&(this.retryLoader.stop(),this.retryLoader=void 0),e.loadingAnimation&&e.loadingAnimation.stop(),o.clear(),e.loadingAnimation=new g(n,l,this.ctx.loaders.getDefaultWorkingMessage(),this.ctx.loaders.getSessionId()),o.addChild(e.loadingAnimation),this.ctx.loaders.setBuddyPetState("working","Working..."),e.pendingWorkingMessage!==void 0&&(e.workingMessageOverride=e.pendingWorkingMessage||void 0,e.pendingWorkingMessage=void 0),this.ctx.loaders.startAgentRunTimer(),this.ctx.loaders.updateWorkingMessage({resetStallTimer:!1}),this.ctx.loaders.isInPlanMode()&&(e.planProgress=L(),e.planProgressPanel=new R(n,l),o.addChild(e.planProgressPanel),e.planProgressPanel.update(e.planProgress)),this.refreshTaskPanel(e,n,o).catch(()=>{}),this.taskUpdateUnsubscribe?.(),this.taskUpdateUnsubscribe=w(()=>{this.refreshTaskPanel(e,n,o).catch(()=>{})}),this.ctx.surface.restoreEditorFocusIfPossible(),this.ctx.layout.requestRender();break;case"message_start":if(t.message.role==="custom")this.ctx.layout.addMessageToChat(t.message),this.ctx.layout.requestRender();else if(t.message.role==="user"){const s=this.ctx.surface.getUserMessageText(t.message);if(e.optimisticUserMessages.length>0&&e.optimisticUserMessages[0]?.text===s){e.optimisticUserMessages.shift(),this.ctx.layout.updatePendingMessagesDisplay(),this.ctx.layout.requestRender();break}this.ctx.layout.addMessageToChat(t.message),this.ctx.layout.updatePendingMessagesDisplay(),this.ctx.layout.requestRender()}else t.message.role==="assistant"&&(e.streamingComponent=new S(void 0,e.hideThinkingBlock,this.ctx.surface.getMarkdownThemeWithSettings()),e.streamingMessage=t.message,a.addChild(e.streamingComponent),e.streamingComponent.updateContent(e.streamingMessage),this.ctx.layout.requestRender());break;case"message_update":if(e.streamingComponent&&t.message.role==="assistant"){e.loadingAnimation&&e.loadingAnimation.resetStallTimer(),e.streamingMessage=t.message,e.streamingComponent.updateContent(e.streamingMessage),a.markDirty(e.streamingComponent);for(const s of e.streamingMessage.content)if(s.type==="toolCall"){if(!this.ctx.toolTrace.shouldRenderToolTrace(s.name))continue;if(e.pendingTools.has(s.id)){const r=e.pendingTools.get(s.id);r&&(r.updateArgs(s.arguments),a.markDirty(r))}else{a.addChild(new y("",0,0));const r=new x(s.name,s.arguments,{showImages:this.ctx.toolTrace.getShowImages()},this.ctx.toolTrace.getRegisteredToolDefinition(s.name),n);r.setExpanded(e.toolOutputExpanded),a.addChild(r),e.pendingTools.set(s.id,r)}}this.ctx.layout.requestRender()}break;case"message_end":if(t.message.role==="user")break;if(e.streamingComponent&&t.message.role==="assistant"){e.streamingMessage=t.message;let s;if(e.streamingMessage.stopReason==="aborted"){const r=this.ctx.runtime.getRetryAttempt();s=r>0?`Aborted after ${r} retry attempt${r>1?"s":""}`:"Operation aborted",e.streamingMessage.errorMessage=s}if(e.streamingComponent.updateContent(e.streamingMessage),a.markDirty(e.streamingComponent),e.streamingMessage.stopReason==="aborted"||e.streamingMessage.stopReason==="error"){s||(s=e.streamingMessage.errorMessage||"Error");for(const[,r]of e.pendingTools.entries())r.updateResult({content:[{type:"text",text:s}],isError:!0}),a.markDirty(r);e.pendingTools.clear()}else for(const[,r]of e.pendingTools.entries())r.setArgsComplete(),a.markDirty(r);e.streamingComponent=void 0,e.streamingMessage=void 0,this.ctx.layout.invalidateFooter()}this.ctx.layout.requestRender();break;case"tool_execution_start":{if(e.planProgress&&e.planProgressPanel&&(q(e.planProgress,t.toolName,t.args),e.planProgressPanel.update(e.planProgress),this.ctx.layout.requestRender()),!this.ctx.toolTrace.shouldRenderToolTrace(t.toolName))break;if(!e.pendingTools.has(t.toolCallId)){const s=new x(t.toolName,t.args,{showImages:this.ctx.toolTrace.getShowImages()},this.ctx.toolTrace.getRegisteredToolDefinition(t.toolName),n);s.setExpanded(e.toolOutputExpanded),a.addChild(s),e.pendingTools.set(t.toolCallId,s),this.ctx.layout.requestRender()}break}case"tool_execution_update":{const s=e.pendingTools.get(t.toolCallId);s&&(s.updateResult({...t.partialResult,isError:!1},!0),a.markDirty(s),this.ctx.layout.requestRender());break}case"tool_execution_end":{const s=e.pendingTools.get(t.toolCallId);s&&(s.updateResult({...t.result,isError:t.isError}),a.markDirty(s),e.pendingTools.delete(t.toolCallId),this.ctx.layout.requestRender());break}case"agent_end":{C("agent_end received");const s=e.agentRunStartMs!==void 0?this.ctx.loaders.formatElapsedSeconds(Date.now()-e.agentRunStartMs):void 0;this.ctx.loaders.stopAgentRunTimer(),e.agentRunStartMs=void 0,e.workingMessageOverride=void 0,e.loadingAnimation&&(e.loadingAnimation.stop(),e.loadingAnimation=void 0,o.clear()),e.streamingComponent&&(a.removeChild(e.streamingComponent),e.streamingComponent=void 0,e.streamingMessage=void 0),e.pendingTools.clear(),e.subAgentPanelComponent&&(o.removeChild(e.subAgentPanelComponent),e.subAgentPanelComponent=void 0),e.subAgentStates.clear(),e.planProgressPanel&&(o.removeChild(e.planProgressPanel),e.planProgressPanel=void 0),e.planProgress=void 0,e.taskStatusPanel&&(o.removeChild(e.taskStatusPanel),e.taskStatusPanel=void 0),this.taskAutoHideTimer&&(clearTimeout(this.taskAutoHideTimer),this.taskAutoHideTimer=void 0),this.ctx.runtime.clearAttachments(),this.ctx.loaders.setBuddyPetState("happy","Done!",{resetTo:"idle",afterMs:1800}),s&&this.ctx.surface.showStatus(`Completed in ${s}`),await this.ctx.runtime.checkShutdownRequested(),this.ctx.surface.restoreEditorFocusIfPossible(),this.ctx.layout.requestRender();break}case"sub_agent_start":{const s={id:t.subAgentId,agentType:t.agentType,description:t.description,isAsync:t.isAsync,isResolved:!1,isError:!1,toolUseCount:0,lastToolName:null,startTime:Date.now()};e.subAgentStates.set(t.subAgentId,s),e.subAgentPanelComponent||(e.subAgentPanelComponent=new M(n,l),o.addChild(e.subAgentPanelComponent)),e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender();break}case"sub_agent_tool_start":{const s=e.subAgentStates.get(t.subAgentId);s&&(s.lastToolName=t.toolName,e.subAgentPanelComponent&&e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender());break}case"sub_agent_tool_end":{const s=e.subAgentStates.get(t.subAgentId);s&&(s.toolUseCount+=1,s.lastToolName=null,e.subAgentPanelComponent&&e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender());break}case"sub_agent_end":{const s=e.subAgentStates.get(t.subAgentId);s&&(s.isResolved=!0,s.isError=!t.success,s.lastToolName=null,e.subAgentPanelComponent&&e.subAgentPanelComponent.update(e.subAgentStates),this.ctx.layout.requestRender());break}case"auto_compaction_start":{this.autoCompactionEscapeHandler=this.ctx.escape.getHandler(),this.ctx.escape.setHandler(()=>{this.ctx.runtime.abortCompaction()}),o.clear();const s=t.reason==="overflow"?"Context overflow detected, ":"";this.autoCompactionLoader=new g(n,l,`${s}Auto-compacting... (${this.ctx.loaders.getInterruptKeyHint()} to cancel)`),o.addChild(this.autoCompactionLoader),this.ctx.layout.requestRender();break}case"auto_compaction_end":{this.autoCompactionEscapeHandler&&(this.ctx.escape.setHandler(this.autoCompactionEscapeHandler),this.autoCompactionEscapeHandler=void 0),this.autoCompactionLoader&&(this.autoCompactionLoader.stop(),this.autoCompactionLoader=void 0,o.clear()),t.aborted?this.ctx.surface.showStatus("Auto-compaction cancelled"):t.result?(a.clear(),this.ctx.layout.rebuildChatFromMessages(),this.ctx.layout.addMessageToChat({role:"compactionSummary",tokensBefore:t.result.tokensBefore,summary:t.result.summary,timestamp:Date.now()}),this.ctx.layout.invalidateFooter()):t.errorMessage&&(a.addChild(new b(1)),a.addChild(new y(l.fg("error",t.errorMessage),1,0))),this.ctx.runtime.flushCompactionQueue({willRetry:t.willRetry}),this.ctx.layout.requestRender();break}case"auto_retry_start":{this.retryEscapeHandler=this.ctx.escape.getHandler(),this.ctx.escape.setHandler(()=>{this.ctx.runtime.abortRetry()}),o.clear();const s=Math.round(t.delayMs/1e3);this.retryLoader=new g(n,l,`Retrying (${t.attempt}/${t.maxAttempts}) in ${s}s... (${this.ctx.loaders.getInterruptKeyHint()} to cancel)`),o.addChild(this.retryLoader),this.ctx.layout.requestRender();break}case"auto_retry_end":{this.retryEscapeHandler&&(this.ctx.escape.setHandler(this.retryEscapeHandler),this.retryEscapeHandler=void 0),this.retryLoader&&(this.retryLoader.stop(),this.retryLoader=void 0,o.clear()),t.success||this.ctx.surface.showError(`Retry failed after ${t.attempt} attempts: ${t.finalError||"Unknown error"}`),this.ctx.layout.requestRender();break}}}static TASK_TOOL_NAMES=new Set(["TaskCreate","TaskUpdate","TaskList","TaskGet","TaskStop","TaskDelete"]);async refreshTaskPanel(t,e,n){const a=this.ctx.runtime.getAgentDir();if(a)try{const o=await T(a,m),s=o.filter(d=>!d.metadata?._internal).map(d=>({id:d.id,subject:d.subject,status:d.status,activeForm:d.activeForm,blockedBy:d.blockedBy.filter(p=>o.some(c=>c.id===p&&c.status!=="completed"))}));if(s.length===0){t.taskStatusPanel&&(n.removeChild(t.taskStatusPanel),t.taskStatusPanel=void 0),this.taskAutoHideTimer&&(clearTimeout(this.taskAutoHideTimer),this.taskAutoHideTimer=void 0);return}t.taskStatusPanel||(t.taskStatusPanel=new _(e,l),n.addChild(t.taskStatusPanel)),t.taskStatusPanel.update(s),this.ctx.layout.requestRender(),s.every(d=>d.status==="completed")?this.taskAutoHideTimer||(this.taskAutoHideTimer=setTimeout(async()=>{this.taskAutoHideTimer=void 0;const d=await T(a,m).catch(()=>[]);d.length>0&&d.every(c=>c.status==="completed")&&await E(a,m).catch(()=>{})},I)):this.taskAutoHideTimer&&(clearTimeout(this.taskAutoHideTimer),this.taskAutoHideTimer=void 0)}catch{}}}const D=["Phase 1: Initial Understanding","Phase 2: Design","Phase 3: Review","Phase 4: Final Plan","Phase 5: Call ExitPlanMode"];function L(){return{phases:D.map((i,t)=>({label:i,status:t===0?"in_progress":"pending"})),currentPhaseIndex:0,startTime:Date.now()}}u(L,"createInitialPlanProgress");function q(i,t,e){let n;if(t==="ExitPlanMode"){for(const a of i.phases)a.status="completed";i.currentPhaseIndex=i.phases.length;return}if(t==="Agent")(e.subagent_type??"").toLowerCase()==="plan"?n=1:n=0;else if(t==="Read"||t==="Grep"||t==="Find")n=2;else if(t==="Write"||t==="Edit")n=3;else return void 0;if(n>i.currentPhaseIndex){for(let a=i.currentPhaseIndex;a<n;a++)i.phases[a].status="completed";i.currentPhaseIndex=n,i.phases[n].status="in_progress"}}u(q,"detectPlanPhaseTransition");export{Y as StreamRenderController};
|