@openadapter/koda 1.0.0-beta.3

package/dist/core/agent-session.js

@@ -0,0 +1,32 @@
+var J=Object.defineProperty;var f=(E,e)=>J(E,"name",{value:e,configurable:!0});import{existsSync as k,mkdirSync as D,readFileSync as T,unlinkSync as Q,writeFileSync as L}from"node:fs";import{homedir as Y}from"node:os";import{basename as Z,dirname as U,join as S,relative as X,resolve as K}from"node:path";import{clampThinkingLevel as ee,cleanupSessionResources as te,getSupportedThinkingLevels as se,isContextOverflow as $,modelsAreEqual as C,resetApiProviders as ne,streamSimple as B}from"@openadapter/koda-ai";import{theme as ie}from"../modes/interactive/theme/theme.js";import{stripFrontmatter as F}from"../utils/frontmatter.js";import{resolvePath as re}from"../utils/paths.js";import{sleep as oe}from"../utils/sleep.js";import{formatNoApiKeyFoundMessage as O,formatNoModelSelectedMessage as H}from"./auth-guidance.js";import{executeBashWithOperations as ae}from"./bash-executor.js";import{calculateContextTokens as z,collectEntriesForBranchSummary as le,compact as j,estimateContextTokens as G,generateBranchSummary as ce,prepareCompaction as q,shouldCompact as he}from"./compaction/index.js";import{DEFAULT_THINKING_LEVEL as de}from"./defaults.js";import{execCommand as ue}from"./exec.js";import{exportSessionToHtml as me}from"./export-html/index.js";import{createToolHtmlRenderer as ge}from"./export-html/tool-renderer.js";import{ExtensionRunner as pe,wrapRegisteredTools as W}from"./extensions/index.js";import{emitSessionShutdownEvent as fe}from"./extensions/runner.js";import{expandPromptTemplate as N}from"./prompt-templates.js";import{CURRENT_SESSION_VERSION as _e,getLatestCompactionEntry as V}from"./session-manager.js";import{createSyntheticSourceInfo as I}from"./source-info.js";import{buildSystemPrompt as ye}from"./system-prompt.js";import{createLocalBashOperations as we}from"./tools/bash.js";import{createAllToolDefinitions as xe}from"./tools/index.js";import{createToolDefinitionFromAgentTool as be}from"./tools/tool-definition-wrapper.js";function Ve(E){const e=E.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}f(Ve,"parseSkillBlock");const ve=["off","minimal","low","medium","high"];class Je{static{f(this,"AgentSession")}agent;sessionManager;settingsManager;_scopedModels;_unsubscribeAgent;_eventListeners=[];_steeringMessages=[];_followUpMessages=[];_pendingNextTurnMessages=[];_compactionAbortController=void 0;_autoCompactionAbortController=void 0;_overflowRecoveryAttempted=!1;_branchSummaryAbortController=void 0;_retryAbortController=void 0;_retryAttempt=0;_overseerAbortController=void 0;_failoverTried=new Set;get _zenitsu(){return process.env.KODA_ZENITSU==="1"}_zenitsuPreferred;_zenitsuCooldown=new Map;_zenitsuHealthCache;_overseerRounds=0;_currentGoal;_tierCache;_overseerLastError="";_escalationsDone=0;_escalateTried=new Set;_sessionEscalations=0;_stickyModel;_homeModel;_sessionGoal;_pinnedSkills=new Map;_codeChangedThisTask=!1;_verifyCmd;_taskStartTs=0;_taskHadActivity=!1;_suggestShown=new Set;_suggestedThisTask=!1;_harnessTrace=[];_gatewayCircuit={failures:0,openUntil:0};_bashAbortController=void 0;_pendingBashMessages=[];_extensionRunner;_turnIndex=0;_resourceLoader;_customTools;_baseToolDefinitions=new Map;_cwd;_protectedSnap=new Map;_extensionRunnerRef;_initialActiveToolNames;_allowedToolNames;_excludedToolNames;_baseToolsOverride;_sessionStartEvent;_extensionUIContext;_extensionMode="print";_extensionCommandContextActions;_extensionAbortHandler;_extensionShutdownHandler;_extensionErrorListener;_extensionErrorUnsubscriber;_modelRegistry;_toolRegistry=new Map;_toolDefinitions=new Map;_toolPromptSnippets=new Map;_toolPromptGuidelines=new Map;_baseSystemPrompt="";_baseSystemPromptOptions;constructor(e){this.agent=e.agent,this.sessionManager=e.sessionManager,this.settingsManager=e.settingsManager,this._scopedModels=e.scopedModels??[],this._resourceLoader=e.resourceLoader,this._customTools=e.customTools??[],this._cwd=e.cwd,this._modelRegistry=e.modelRegistry,this._extensionRunnerRef=e.extensionRunnerRef,this._initialActiveToolNames=e.initialActiveToolNames,this._allowedToolNames=e.allowedToolNames?new Set(e.allowedToolNames):void 0,this._excludedToolNames=e.excludedToolNames?new Set(e.excludedToolNames):void 0,this._baseToolsOverride=e.baseToolsOverride,this._sessionStartEvent=e.sessionStartEvent??{type:"session_start",reason:"startup"},this._unsubscribeAgent=this.agent.subscribe(this._handleAgentEvent),this._installAgentToolHooks(),this._buildRuntime({activeToolNames:this._initialActiveToolNames,includeAllExtensionTools:!0})}get modelRegistry(){return this._modelRegistry}async _getRequiredRequestAuth(e){const t=await this._modelRegistry.getApiKeyAndHeaders(e);if(!t.ok)throw t.error.startsWith("No API key found")?new Error(O(e.provider)):new Error(t.error);if(t.apiKey)return{apiKey:t.apiKey,headers:t.headers};throw this._modelRegistry.isUsingOAuth(e)?new Error(`Authentication failed for "${e.provider}". Credentials may have expired or network is unavailable. Run '/login ${e.provider}' to re-authenticate.`):new Error(O(e.provider))}async _getCompactionRequestAuth(e){if(this.agent.streamFn===B)return this._getRequiredRequestAuth(e);const t=await this._modelRegistry.getApiKeyAndHeaders(e);return t.ok?{apiKey:t.apiKey,headers:t.headers}:{}}_installAgentToolHooks(){const e=process.env.KODA_UNSAFE==="1",t=process.cwd(),s=[/(^|\/)test_[^/]*\.[A-Za-z0-9]+$/,/(^|\/)[^/]*_test\.[A-Za-z0-9]+$/,/\.(test|spec)\.[A-Za-z0-9]+$/,/(^|\/)(tests?|__tests__|spec)\//,/(^|\/)\.git(\/|$)/,/(^|\/)\.env(\.[^/]*)?$/,/(package-lock\.json|yarn\.lock|pnpm-lock\.yaml|poetry\.lock|Cargo\.lock)$/],n=(process.env.KODA_PROTECT||"").split(",").map(g=>g.trim()).filter(Boolean),i=f(g=>{const p=g.replace(/\\/g,"/");return s.some(_=>_.test(p))||n.some(_=>p.includes(_))},"isProtected"),o=f(g=>{for(const p of g.matchAll(/(?:>>?|\btee\s+(?:-a\s+)?)\s*([^\s;|&>]+)/g))if(i(p[1]))return`output redirect onto protected file '${p[1]}'`;return/\brm\s+(?:-\w+\s+)*[^\n;|&]*(?:test_|tests?\/|__tests__|\.git|\.env)/.test(g)?"rm targeting a protected path":/\bgit\s+(?:reset\s+--hard|clean\b|checkout\s+(?:--\s+)?\.)/.test(g)?"destructive git command":null},"dangerousBash"),r=S(process.env.KODA_CODING_AGENT_DIR||S(Y(),".koda","agent"),"checkpoints",String(this.sessionId||"session")),a=new Set,l=f(g=>{if(!a.has(g)){a.add(g);try{D(r,{recursive:!0});const p=S(r,`${g.replace(/[^A-Za-z0-9]/g,"_")}.json`),_=k(g);L(p,JSON.stringify({path:g,existed:_,content:_?T(g,"utf-8"):null}))}catch{}}},"backup");this.agent.beforeToolCall=async({toolCall:g,args:p})=>{if((g.name==="write"||g.name==="edit")&&(this._codeChangedThisTask=!0),this._taskHadActivity=!0,g.name==="bash"){const b=String(p?.command??"");/\b(npm|pnpm|yarn|bun)\s+(run\s+)?(dev|start|serve|preview|watch)\b|\bvite\b|\bnext\s+(dev|start)\b|\bnuxt\s+dev\b|\bnodemon\b|http[-.]server\b|\bserve\b|flask\s+run|rails\s+s(erver)?\b|manage\.py\s+runserver|\b(uvicorn|gunicorn)\b/i.test(b)&&this._suggest("bg","That looks like a long-running server. Use run_background (or /bg) so it keeps running without blocking Koda.")}if(process.env.KODA_MODE==="plan"){const b=p,v=g.name;if(v==="write"||v==="edit")return{block:!0,reason:"Koda is in PLAN MODE \u2014 file changes are disabled. Investigate and present a step-by-step plan instead; the user will run /build to execute it."};if(v==="bash"&&this._isMutatingBash(String(b?.command??"")))return{block:!0,reason:"Koda is in PLAN MODE \u2014 commands that change state are disabled. Use read-only commands to investigate, then present a plan; the user will run /build to execute it."}}if(!e){const b=p,v=g.name;if(v==="write"||v==="edit"){const y=String(b?.path??b?.file_path??"");if(y){const M=K(t,y);if(X(t,M).startsWith(".."))return{block:!0,reason:`Refused: '${y}' is outside the workspace. Only modify files under the project directory.`};if(i(y))return{block:!0,reason:`Refused: '${y}' is a protected file (tests / .git / secrets / lockfiles). Do not modify it \u2014 change the source code instead.`};l(M)}}else if(v==="bash"){const y=o(String(b?.command??""));if(y)return{block:!0,reason:`Refused: ${y}. Use the edit/write tools on source files instead of shell redirects, and never touch tests/.git/secrets.`}}const A=[];if(v==="write"||v==="edit"){const y=String(b?.path??b?.file_path??"");y&&A.push(y)}else if(v==="bash")for(const y of String(b?.command??"").matchAll(/[\w./-]*[\w]\.[A-Za-z0-9]+/g))A.push(y[0]);for(const y of A){if(!i(y))continue;const M=K(t,y);if(!this._protectedSnap.has(M))try{this._protectedSnap.set(M,k(M)?T(M,"utf-8"):null)}catch{}}}const _=this._extensionRunner;if(_.hasHandlers("tool_call"))try{return await _.emitToolCall({type:"tool_call",toolName:g.name,toolCallId:g.id,input:p})}catch(b){throw b instanceof Error?b:new Error(`Extension failed, blocking execution: ${String(b)}`)}};const c=process.env.KODA_CASCADE_MODEL,h=Number(process.env.KODA_CASCADE_ERRORS)||3,u=Number(process.env.KODA_CASCADE_STEPS)||16;let m=0,w=0,d=!1,x;this.agent.afterToolCall=async({toolCall:g,args:p,result:_,isError:b})=>{for(const[y,M]of this._protectedSnap){let P=null;try{P=k(y)?T(y,"utf-8"):null}catch{continue}if(P!==M)try{M===null?k(y)&&Q(y):L(y,M,"utf-8"),console.error(`[koda safety] reverted unauthorized change to protected file ${y}`)}catch{}}if(c&&!d&&(w++,b?m++:m=0,m>=h||w>=u)){const y=this.agent.state.model,[M,P]=c.includes("/")?[c.slice(0,c.indexOf("/")),c.slice(c.indexOf("/")+1)]:[y.provider,c],R=this._modelRegistry.find(M,P);R&&!C(R,y)&&(x=R,this.agent.state.model=R,this.sessionManager.appendModelChange(R.provider,R.id),this._emitModelSelect(R,y,"cycle"),d=!0,console.error(`[koda cascade] escalated ${y.provider}/${y.id} \u2192 ${R.provider}/${R.id} (errors=${m}, toolCalls=${w})`))}const v=this._extensionRunner;if(!v.hasHandlers("tool_result"))return;const A=await v.emitToolResult({type:"tool_result",toolName:g.name,toolCallId:g.id,input:p,content:_.content,details:_.details,isError:b});if(A)return{content:A.content,details:A.details,isError:A.isError??b}},this.agent.prepareNextTurn=async()=>{if(this._zenitsu){const g=await this._zenitsuPickModel();if(g&&!C(g,this.agent.state.model))return{model:g}}return d&&x?{model:x}:void 0}}_emit(e){for(const t of this._eventListeners)t(e)}_emitQueueUpdate(){this._emit({type:"queue_update",steering:[...this._steeringMessages],followUp:[...this._followUpMessages]})}_lastAssistantMessage=void 0;_handleAgentEvent=f(async e=>{if(e.type==="message_start"&&e.message.role==="user"){this._overflowRecoveryAttempted=!1;const t=this._getUserMessageText(e.message);if(t){const s=this._steeringMessages.indexOf(t);if(s!==-1)this._steeringMessages.splice(s,1),this._emitQueueUpdate();else{const n=this._followUpMessages.indexOf(t);n!==-1&&(this._followUpMessages.splice(n,1),this._emitQueueUpdate())}}}if(await this._emitExtensionEvent(e),this._emit(e.type==="agent_end"?{...e,willRetry:this._willRetryAfterAgentEnd(e)}:e),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;const t=e.message;t.stopReason!=="error"&&(this._overflowRecoveryAttempted=!1),t.stopReason!=="error"&&this._retryAttempt>0&&(this._emit({type:"auto_retry_end",success:!0,attempt:this._retryAttempt}),this._retryAttempt=0)}},"_handleAgentEvent");_willRetryAfterAgentEnd(e){const t=this.settingsManager.getRetrySettings();if(!t.enabled||this._retryAttempt>=t.maxRetries)return!1;for(let s=e.messages.length-1;s>=0;s--){const n=e.messages[s];if(n.role==="assistant")return this._isRetryableError(n)}return!1}_getUserMessageText(e){if(e.role!=="user")return"";const t=e.content;return typeof t=="string"?t:t.filter(n=>n.type==="text").map(n=>n.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}}_replaceMessageInPlace(e,t){if(e===t)return;const s=e;for(const n of Object.keys(s))delete s[n];Object.assign(s,t)}async _emitExtensionEvent(e){if(e.type==="agent_start")this._turnIndex=0,await this._extensionRunner.emit({type:"agent_start"});else if(e.type==="agent_end")await this._extensionRunner.emit({type:"agent_end",messages:e.messages});else if(e.type==="turn_start"){const t={type:"turn_start",turnIndex:this._turnIndex,timestamp:Date.now()};await this._extensionRunner.emit(t)}else if(e.type==="turn_end"){const t={type:"turn_end",turnIndex:this._turnIndex,message:e.message,toolResults:e.toolResults};await this._extensionRunner.emit(t),this._turnIndex++}else if(e.type==="message_start"){const t={type:"message_start",message:e.message};await this._extensionRunner.emit(t)}else if(e.type==="message_update"){const t={type:"message_update",message:e.message,assistantMessageEvent:e.assistantMessageEvent};await this._extensionRunner.emit(t)}else if(e.type==="message_end"){const t={type:"message_end",message:e.message},s=await this._extensionRunner.emitMessageEnd(t);s&&this._replaceMessageInPlace(e.message,s)}else if(e.type==="tool_execution_start"){const t={type:"tool_execution_start",toolCallId:e.toolCallId,toolName:e.toolName,args:e.args};await this._extensionRunner.emit(t)}else if(e.type==="tool_execution_update"){const t={type:"tool_execution_update",toolCallId:e.toolCallId,toolName:e.toolName,args:e.args,partialResult:e.partialResult};await this._extensionRunner.emit(t)}else if(e.type==="tool_execution_end"){const t={type:"tool_execution_end",toolCallId:e.toolCallId,toolName:e.toolName,result:e.result,isError:e.isError};await this._extensionRunner.emit(t)}}subscribe(e){return this._eventListeners.push(e),()=>{const t=this._eventListeners.indexOf(e);t!==-1&&this._eventListeners.splice(t,1)}}_disconnectFromAgent(){this._unsubscribeAgent&&(this._unsubscribeAgent(),this._unsubscribeAgent=void 0)}_reconnectToAgent(){this._unsubscribeAgent||(this._unsubscribeAgent=this.agent.subscribe(this._handleAgentEvent))}dispose(){try{this.abortRetry(),this.abortCompaction(),this.abortBranchSummary(),this.abortBash(),this.agent.abort()}catch{}this._extensionRunner.invalidate("This extension ctx is stale after session replacement or reload. Do not use a captured pi or command ctx after ctx.newSession(), ctx.fork(), ctx.switchSession(), or ctx.reload(). For newSession, fork, and switchSession, move post-replacement work into withSession and use the ctx passed to withSession. For reload, do not use the old ctx after await ctx.reload()."),this._disconnectFromAgent(),this._eventListeners=[],te(this.sessionId)}get state(){return this.agent.state}get model(){return this.agent.state.model}get thinkingLevel(){return this.agent.state.thinkingLevel}get isStreaming(){return this.agent.state.isStreaming}get systemPrompt(){return this.agent.state.systemPrompt}get retryAttempt(){return this._retryAttempt}getActiveToolNames(){return this.agent.state.tools.map(e=>e.name)}getAllTools(){return Array.from(this._toolDefinitions.values()).map(({definition:e,sourceInfo:t})=>({name:e.name,description:e.description,parameters:e.parameters,promptGuidelines:e.promptGuidelines,sourceInfo:t}))}getToolDefinition(e){return this._toolDefinitions.get(e)?.definition}setActiveToolsByName(e){const t=[],s=[];for(const n of e){const i=this._toolRegistry.get(n);i&&(t.push(i),s.push(n))}this.agent.state.tools=t,this._baseSystemPrompt=this._rebuildSystemPrompt(s),this.agent.state.systemPrompt=this._baseSystemPrompt}get isCompacting(){return this._autoCompactionAbortController!==void 0||this._compactionAbortController!==void 0||this._branchSummaryAbortController!==void 0}get messages(){return this.agent.state.messages}get steeringMode(){return this.agent.steeringMode}get followUpMode(){return this.agent.followUpMode}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}_normalizePromptSnippet(e){if(!e)return;const t=e.replace(/[\r\n]+/g," ").replace(/\s+/g," ").trim();return t.length>0?t:void 0}_normalizePromptGuidelines(e){if(!e||e.length===0)return[];const t=new Set;for(const s of e){const n=s.trim();n.length>0&&t.add(n)}return Array.from(t)}_rebuildSystemPrompt(e){const t=e.filter(h=>this._toolRegistry.has(h)),s={},n=[];for(const h of t){const u=this._toolPromptSnippets.get(h);u&&(s[h]=u);const m=this._toolPromptGuidelines.get(h);m&&n.push(...m)}const i=this._resourceLoader.getSystemPrompt(),r=[...this._resourceLoader.getAppendSystemPrompt(),...this._pinnedSkills.values()],a=r.length>0?r.join(`
+
+`):void 0,l=this._resourceLoader.getSkills().skills,c=this._resourceLoader.getAgentsFiles().agentsFiles;return this._baseSystemPromptOptions={cwd:this._cwd,skills:l,contextFiles:c,customPrompt:i,appendSystemPrompt:a,selectedTools:t,toolSnippets:s,promptGuidelines:n},ye(this._baseSystemPromptOptions)}async _runAgentPrompt(e){this._failoverTried.clear(),this._overseerRounds=0,this._escalationsDone=0,this._escalateTried.clear(),this._codeChangedThisTask=!1,this._taskStartTs=Date.now(),this._taskHadActivity=!1,this._suggestedThisTask=!1;const t=this._goalFromMessages(e);if(t&&(this._currentGoal=t),!this._homeModel)this._homeModel=this.agent.state.model;else{const s=this._stickyModel??this._homeModel;s.id!==this.agent.state.model.id&&this._modelRegistry.hasConfiguredAuth(s)&&(this.agent.state.model=s,this._zenitsuPreferred=s,this._stickyModel||(this._trace("revert",`\u2192 ${s.id} (new task)`),this._emit({type:"model_switch_notice",message:`\u2190 Back on your pick \u2014 ${s.id}`})))}try{for(await this.agent.prompt(e);await this._handlePostAgentRun();)try{await this.agent.continue()}catch(s){const n=s?.message||String(s);if(this._trace("recover",`continue failed: ${n.slice(0,90)}`),/cannot continue from message role/i.test(n)){this.agent.followUp({role:"user",content:[{type:"text",text:"Continue from where you left off and finish the task."}],timestamp:Date.now()});try{await this.agent.continue();continue}catch{}}this._emit({type:"model_switch_notice",message:`Koda hit a snag and stopped cleanly (${n.slice(0,60)}). Your work is saved \u2014 send another message to continue.`});break}if(this._taskHadActivity&&process.env.KODA_RECAP!=="0"){const s=await this._generateRecap();this._emit({type:"task_recap",elapsedMs:Date.now()-this._taskStartTs,recap:s||""})}this._taskHadActivity&&!this._sessionGoal&&Date.now()-this._taskStartTs>18e4&&this._suggest("goal","Working on something big? Set a goal with /goal \u2014 Koda tracks the milestones and won't stop until they're done.")}finally{this._flushPendingBashMessages()}}async _generateRecap(){const e=this.agent.state.model,t=process.env.KODA_RECAP_MODEL?[process.env.KODA_RECAP_MODEL]:["free/qwen3-coder","free/qwen3-next-80b-a3b-instruct","GLM-4.7-Flash","GLM-4.7"];for(const s of t){const n=this._modelRegistry.find(e.provider,s);if(!n)continue;const i=await this._recapOnce(n);if(i)return i}return null}async _recapOnce(e){try{if(!e.baseUrl)return null;const t=await this._modelRegistry.getApiKeyAndHeaders(e);if(!t?.apiKey)return null;const s=new AbortController,n=setTimeout(()=>s.abort(),2e4);try{const i=await fetch(`${e.baseUrl.replace(/\/$/,"")}/chat/completions`,{method:"POST",headers:{Authorization:`Bearer ${t.apiKey}`,"Content-Type":"application/json",...t.headers||{}},body:JSON.stringify({model:e.id,temperature:.2,max_tokens:150,messages:[{role:"system",content:'Output ONLY a 1-2 sentence, past-tense recap of what was done this turn (and the next step if any). No analysis, no numbered lists, no markdown, no preamble \u2014 just the recap. Example: "Added a /health endpoint to server.ts and confirmed the build passes. Next: add a test for it."'},{role:"user",content:`REQUEST:
+${this._currentGoal||"(none)"}
+
+WORK LOG:
+${this._overseerContext()}`}]}),signal:s.signal});return i.ok&&((await i.json())?.choices?.[0]?.message?.content??"").replace(/<think>[\s\S]*?<\/think>/gi,"").trim().slice(0,400)||null}finally{clearTimeout(n)}}catch{return null}}async _handlePostAgentRun(){const e=this._lastAssistantMessage;return this._lastAssistantMessage=void 0,e?(e.stopReason!=="error"&&this._maybeReturnHome(),this._isDegenerateOutput(e)&&await this._escalate("output broke down \u2014 bringing in a stronger model",{stripLast:!0})||this._zenitsu&&await this._zenitsuHandle429(e)||(!this._zenitsu&&this._isRateLimited(e)&&this._suggest("zenitsu","Getting rate limited \u2014 zenitsu mode reroutes to a faster model instantly instead of waiting. Turn it on with /zenitsu."),this._isRetryableError(e)&&await this._prepareRetry(e))||e.stopReason==="error"&&await this._failoverModel(e)||(e.stopReason==="error"&&this._retryAttempt>0&&(this._emit({type:"auto_retry_end",success:!1,attempt:this._retryAttempt,finalError:e.errorMessage}),this._retryAttempt=0),await this._checkCompaction(e))||await this._overseerCheck(e)?!0:this.agent.hasQueuedMessages()):!1}async prompt(e,t){const s=t?.expandPromptTemplates??!0,n=t?.preflightResult;let i;try{if(s&&e.startsWith("/")&&await this._tryExecuteExtensionCommand(e)){n?.(!0);return}let o=e,r=t?.images;if(this._extensionRunner.hasHandlers("input")){const u=await this._extensionRunner.emitInput(o,r,t?.source??"interactive",this.isStreaming?t?.streamingBehavior:void 0);if(u.action==="handled"){n?.(!0);return}u.action==="transform"&&(o=u.text,r=u.images??r)}let a=o;if(s&&(a=this._expandSkillCommand(a),a=N(a,[...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(a,r):await this._queueSteer(a,r),n?.(!0);return}if(this._flushPendingBashMessages(),!this.model)throw new Error(H());if(!this._modelRegistry.hasConfiguredAuth(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(O(this.model.provider));const l=this._findLastAssistantMessage();if(l&&await this._checkCompaction(l,!1))try{for(await this.agent.continue();await this._handlePostAgentRun();)await this.agent.continue()}finally{this._flushPendingBashMessages()}i=[];const c=[{type:"text",text:a}];r&&c.push(...r),i.push({role:"user",content:c,timestamp:Date.now()});for(const u of this._pendingNextTurnMessages)i.push(u);this._pendingNextTurnMessages=[];const h=await this._extensionRunner.emitBeforeAgentStart(a,r,this._baseSystemPrompt,this._baseSystemPromptOptions);if(h?.messages)for(const u of h.messages)i.push({role:"custom",customType:u.customType,content:u.content,display:u.display,details:u.details,timestamp:Date.now()});h?.systemPrompt?this.agent.state.systemPrompt=h.systemPrompt:this.agent.state.systemPrompt=this._baseSystemPrompt}catch(o){throw n?.(!1),o}i&&(n?.(!0),await this._runAgentPrompt(i))}async _tryExecuteExtensionCommand(e){const t=e.indexOf(" "),s=t===-1?e.slice(1):e.slice(1,t),n=t===-1?"":e.slice(t+1),i=this._extensionRunner.getCommand(s);if(!i)return!1;const o=this._extensionRunner.createCommandContext();try{return await i.handler(n,o),!0}catch(r){return this._extensionRunner.emitError({extensionPath:`command:${s}`,event:"command",error:r instanceof Error?r.message:String(r)}),!0}}_expandSkillCommand(e){if(!e.startsWith("/skill:"))return e;const t=e.indexOf(" "),s=t===-1?e.slice(7):e.slice(7,t),n=t===-1?"":e.slice(t+1).trim(),i=this.resourceLoader.getSkills().skills.find(o=>o.name===s);if(!i)return e;try{const o=T(i.filePath,"utf-8"),r=F(o).trim(),a=`<skill name="${i.name}" location="${i.filePath}">
+References are relative to ${i.baseDir}.
+
+${r}
+</skill>`;return n?`${a}
+
+${n}`:a}catch(o){return this._extensionRunner.emitError({extensionPath:i.filePath,event:"skill_expansion",error:o instanceof Error?o.message:String(o)}),e}}async steer(e,t){e.startsWith("/")&&this._throwIfExtensionCommand(e);let s=this._expandSkillCommand(e);s=N(s,[...this.promptTemplates]),await this._queueSteer(s,t)}async followUp(e,t){e.startsWith("/")&&this._throwIfExtensionCommand(e);let s=this._expandSkillCommand(e);s=N(s,[...this.promptTemplates]),await this._queueFollowUp(s,t)}async _queueSteer(e,t){this._steeringMessages.push(e),this._emitQueueUpdate();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),this._emitQueueUpdate();const s=[{type:"text",text:e}];t&&s.push(...t),this.agent.followUp({role:"user",content:s,timestamp:Date.now()})}_throwIfExtensionCommand(e){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._runAgentPrompt(s):(this.agent.state.messages.push(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,n;if(typeof e=="string")s=e;else{const i=[];n=[];for(const o of e)o.type==="text"?i.push(o.text):n.push(o);s=i.join(`
+`),n.length===0&&(n=void 0)}await this.prompt(s,{expandPromptTemplates:!1,streamingBehavior:t?.deliverAs,images:n,source:"extension"})}clearQueue(){const e=[...this._steeringMessages],t=[...this._followUpMessages];return this._steeringMessages=[],this._followUpMessages=[],this.agent.clearAllQueues(),this._emitQueueUpdate(),{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()}async _emitModelSelect(e,t,s){C(t,e)||await this._extensionRunner.emit({type:"model_select",model:e,previousModel:t,source:s})}async setModel(e){if(!this._modelRegistry.hasConfiguredAuth(e))throw new Error(`No API key for ${e.provider}/${e.id}`);const t=this.model,s=this._getThinkingLevelForModelSwitch();this.agent.state.model=e,this._homeModel=e,this._stickyModel=void 0,this._sessionEscalations=0,this.sessionManager.appendModelChange(e.provider,e.id),this.settingsManager.setDefaultModelAndProvider(e.provider,e.id),this.setThinkingLevel(s),await this._emitModelSelect(e,t,"set")}async cycleModel(e="forward"){return this._scopedModels.length>0?this._cycleScopedModel(e):this._cycleAvailableModel(e)}async _cycleScopedModel(e){const t=this._scopedModels.filter(l=>this._modelRegistry.hasConfiguredAuth(l.model));if(t.length<=1)return;const s=this.model;let n=t.findIndex(l=>C(l.model,s));n===-1&&(n=0);const i=t.length,o=e==="forward"?(n+1)%i:(n-1+i)%i,r=t[o],a=this._getThinkingLevelForModelSwitch(r.thinkingLevel);return this.agent.state.model=r.model,this._homeModel=r.model,this._stickyModel=void 0,this._sessionEscalations=0,this.sessionManager.appendModelChange(r.model.provider,r.model.id),this.settingsManager.setDefaultModelAndProvider(r.model.provider,r.model.id),this.setThinkingLevel(a),await this._emitModelSelect(r.model,s,"cycle"),{model:r.model,thinkingLevel:this.thinkingLevel,isScoped:!0}}async _cycleAvailableModel(e){const t=await this._modelRegistry.getAvailable();if(t.length<=1)return;const s=this.model;let n=t.findIndex(l=>C(l,s));n===-1&&(n=0);const i=t.length,o=e==="forward"?(n+1)%i:(n-1+i)%i,r=t[o],a=this._getThinkingLevelForModelSwitch();return this.agent.state.model=r,this._homeModel=r,this._stickyModel=void 0,this._sessionEscalations=0,this.sessionManager.appendModelChange(r.provider,r.id),this.settingsManager.setDefaultModelAndProvider(r.provider,r.id),this.setThinkingLevel(a),await this._emitModelSelect(r,s,"cycle"),{model:r,thinkingLevel:this.thinkingLevel,isScoped:!1}}setThinkingLevel(e){const t=this.getAvailableThinkingLevels(),s=t.includes(e)?e:this._clampThinkingLevel(e,t),n=this.agent.state.thinkingLevel,i=s!==n;this.agent.state.thinkingLevel=s,i&&(this.sessionManager.appendThinkingLevelChange(s),(this.supportsThinking()||s!=="off")&&this.settingsManager.setDefaultThinkingLevel(s),this._emit({type:"thinking_level_changed",level:s}),this._extensionRunner.emit({type:"thinking_level_select",level:s,previousLevel:n}))}cycleThinkingLevel(){if(!this.supportsThinking())return;const e=this.getAvailableThinkingLevels(),s=(e.indexOf(this.thinkingLevel)+1)%e.length,n=e[s];return this.setThinkingLevel(n),n}getAvailableThinkingLevels(){return this.model?se(this.model):ve}supportsThinking(){return!!this.model?.reasoning}_getThinkingLevelForModelSwitch(e){return e!==void 0?e:this.supportsThinking()?this.thinkingLevel:this.settingsManager.getDefaultThinkingLevel()??de}_clampThinkingLevel(e,t){return this.model?ee(this.model,e):"off"}setSteeringMode(e){this.agent.steeringMode=e,this.settingsManager.setSteeringMode(e)}setFollowUpMode(e){this.agent.followUpMode=e,this.settingsManager.setFollowUpMode(e)}async compact(e){this._disconnectFromAgent(),await this.abort(),this._compactionAbortController=new AbortController,this._emit({type:"compaction_start",reason:"manual"});try{if(!this.model)throw new Error(H());const{apiKey:t,headers:s}=await this._getCompactionRequestAuth(this.model),n=this.sessionManager.getBranch(),i=this.settingsManager.getCompactionSettings(),o=q(n,i);if(!o)throw n[n.length-1]?.type==="compaction"?new Error("Already compacted"):new Error("Nothing to compact (session too small)");let r,a=!1;if(this._extensionRunner.hasHandlers("session_before_compact")){const g=await this._extensionRunner.emit({type:"session_before_compact",preparation:o,branchEntries:n,customInstructions:e,signal:this._compactionAbortController.signal});if(g?.cancel)throw new Error("Compaction cancelled");g?.compaction&&(r=g.compaction,a=!0)}let l,c,h,u;if(r)l=r.summary,c=r.firstKeptEntryId,h=r.tokensBefore,u=r.details;else{const g=await j(o,this.model,t,s,e,this._compactionAbortController.signal,this.thinkingLevel,this.agent.streamFn);l=g.summary,c=g.firstKeptEntryId,h=g.tokensBefore,u=g.details}if(this._compactionAbortController.signal.aborted)throw new Error("Compaction cancelled");this.sessionManager.appendCompaction(l,c,h,u,a);const m=this.sessionManager.getEntries(),w=this.sessionManager.buildSessionContext();this.agent.state.messages=w.messages;const d=m.find(g=>g.type==="compaction"&&g.summary===l);this._extensionRunner&&d&&await this._extensionRunner.emit({type:"session_compact",compactionEntry:d,fromExtension:a});const x={summary:l,firstKeptEntryId:c,tokensBefore:h,details:u};return this._emit({type:"compaction_end",reason:"manual",result:x,aborted:!1,willRetry:!1}),x}catch(t){const s=t instanceof Error?t.message:String(t),n=s==="Compaction cancelled"||t instanceof Error&&t.name==="AbortError";throw this._emit({type:"compaction_end",reason:"manual",result:void 0,aborted:n,willRetry:!1,errorMessage:n?void 0:`Compaction failed: ${s}`}),t}finally{this._compactionAbortController=void 0,this._reconnectToAgent()}}abortCompaction(){this._compactionAbortController?.abort(),this._autoCompactionAbortController?.abort()}abortBranchSummary(){this._branchSummaryAbortController?.abort()}abortOverseer(){this._overseerAbortController?.abort()}async _checkCompaction(e,t=!0){const s=this.settingsManager.getCompactionSettings();if(!s.enabled||t&&e.stopReason==="aborted")return!1;const n=this.model?.contextWindow??0,i=this.model&&e.provider===this.model.provider&&e.model===this.model.id,o=V(this.sessionManager.getBranch());if(o!==null&&e.timestamp<=new Date(o.timestamp).getTime())return!1;if(i&&$(e,n)){if(this._overflowRecoveryAttempted)return this._emit({type:"compaction_end",reason:"overflow",result:void 0,aborted:!1,willRetry:!1,errorMessage:"Context overflow recovery failed after one compact-and-retry attempt. Try reducing context or switching to a larger-context model."}),!1;this._overflowRecoveryAttempted=!0;const l=this.agent.state.messages;return l.length>0&&l[l.length-1].role==="assistant"&&(this.agent.state.messages=l.slice(0,-1)),await this._runAutoCompaction("overflow",!0)}let a;if(e.stopReason==="error"){const l=this.agent.state.messages,c=G(l);if(c.lastUsageIndex===null)return!1;const h=l[c.lastUsageIndex];if(o&&h.role==="assistant"&&h.timestamp<=new Date(o.timestamp).getTime())return!1;a=c.tokens}else a=z(e.usage);return he(a,n,s)?await this._runAutoCompaction("threshold",!1):!1}async _runAutoCompaction(e,t){const s=this.settingsManager.getCompactionSettings();this._emit({type:"compaction_start",reason:e}),this._autoCompactionAbortController=new AbortController;try{if(!this.model)return this._emit({type:"compaction_end",reason:e,result:void 0,aborted:!1,willRetry:!1}),!1;let n,i;if(this.agent.streamFn===B){const p=await this._modelRegistry.getApiKeyAndHeaders(this.model);if(!p.ok||!p.apiKey)return this._emit({type:"compaction_end",reason:e,result:void 0,aborted:!1,willRetry:!1}),!1;n=p.apiKey,i=p.headers}else({apiKey:n,headers:i}=await this._getCompactionRequestAuth(this.model));const o=this.sessionManager.getBranch(),r=q(o,s);if(!r)return this._emit({type:"compaction_end",reason:e,result:void 0,aborted:!1,willRetry:!1}),!1;let a,l=!1;if(this._extensionRunner.hasHandlers("session_before_compact")){const p=await this._extensionRunner.emit({type:"session_before_compact",preparation:r,branchEntries:o,customInstructions:void 0,signal:this._autoCompactionAbortController.signal});if(p?.cancel)return this._emit({type:"compaction_end",reason:e,result:void 0,aborted:!0,willRetry:!1}),!1;p?.compaction&&(a=p.compaction,l=!0)}let c,h,u,m;if(a)c=a.summary,h=a.firstKeptEntryId,u=a.tokensBefore,m=a.details;else{const p=await j(r,this.model,n,i,void 0,this._autoCompactionAbortController.signal,this.thinkingLevel,this.agent.streamFn);c=p.summary,h=p.firstKeptEntryId,u=p.tokensBefore,m=p.details}if(this._autoCompactionAbortController.signal.aborted)return this._emit({type:"compaction_end",reason:e,result:void 0,aborted:!0,willRetry:!1}),!1;this.sessionManager.appendCompaction(c,h,u,m,l);const w=this.sessionManager.getEntries(),d=this.sessionManager.buildSessionContext();this.agent.state.messages=d.messages;const x=w.find(p=>p.type==="compaction"&&p.summary===c);this._extensionRunner&&x&&await this._extensionRunner.emit({type:"session_compact",compactionEntry:x,fromExtension:l});const g={summary:c,firstKeptEntryId:h,tokensBefore:u,details:m};if(this._emit({type:"compaction_end",reason:e,result:g,aborted:!1,willRetry:t}),t){const p=this.agent.state.messages,_=p[p.length-1];return _?.role==="assistant"&&_.stopReason==="error"&&(this.agent.state.messages=p.slice(0,-1)),!0}return this.agent.hasQueuedMessages()}catch(n){const i=n instanceof Error?n.message:"compaction failed";return this._emit({type:"compaction_end",reason:e,result:void 0,aborted:!1,willRetry:!1,errorMessage:e==="overflow"?`Context overflow recovery failed: ${i}`:`Auto-compaction failed: ${i}`}),!1}finally{this._autoCompactionAbortController=void 0}}setAutoCompactionEnabled(e){this.settingsManager.setCompactionEnabled(e)}get autoCompactionEnabled(){return this.settingsManager.getCompactionEnabled()}async bindExtensions(e){e.uiContext!==void 0&&(this._extensionUIContext=e.uiContext),e.mode!==void 0&&(this._extensionMode=e.mode),e.commandContextActions!==void 0&&(this._extensionCommandContextActions=e.commandContextActions),e.abortHandler!==void 0&&(this._extensionAbortHandler=e.abortHandler),e.shutdownHandler!==void 0&&(this._extensionShutdownHandler=e.shutdownHandler),e.onError!==void 0&&(this._extensionErrorListener=e.onError),this._applyExtensionBindings(this._extensionRunner),await this._extensionRunner.emit(this._sessionStartEvent),await this.extendResourcesFromExtensions(this._sessionStartEvent.reason==="reload"?"reload":"startup")}async extendResourcesFromExtensions(e){if(!this._extensionRunner.hasHandlers("resources_discover"))return;const{skillPaths:t,promptPaths:s,themePaths:n}=await this._extensionRunner.emitResourcesDiscover(this._cwd,e);if(t.length===0&&s.length===0&&n.length===0)return;const i={skillPaths:this.buildExtensionResourcePaths(t),promptPaths:this.buildExtensionResourcePaths(s),themePaths:this.buildExtensionResourcePaths(n)};this._resourceLoader.extendResources(i),this._baseSystemPrompt=this._rebuildSystemPrompt(this.getActiveToolNames()),this.agent.state.systemPrompt=this._baseSystemPrompt}buildExtensionResourcePaths(e){return e.map(t=>{const s=this.getExtensionSourceLabel(t.extensionPath),n=t.extensionPath.startsWith("<")?void 0:U(t.extensionPath);return{path:t.path,metadata:{source:s,scope:"temporary",origin:"top-level",baseDir:n}}})}getExtensionSourceLabel(e){return e.startsWith("<")?`extension:${e.replace(/[<>]/g,"")}`:`extension:${Z(e).replace(/\.(ts|js)$/,"")}`}_applyExtensionBindings(e){e.setUIContext(this._extensionUIContext,this._extensionMode),e.bindCommandContext(this._extensionCommandContextActions),this._extensionErrorUnsubscriber?.(),this._extensionErrorUnsubscriber=this._extensionErrorListener?e.onError(this._extensionErrorListener):void 0}_refreshCurrentModelFromRegistry(){const e=this.model;if(!e)return;const t=this._modelRegistry.find(e.provider,e.id);!t||t===e||(this.agent.state.model=t)}_bindExtensionCore(e){const t=f(()=>{const s=e.getRegisteredCommands().map(o=>({name:o.invocationName,description:o.description,source:"extension",sourceInfo:o.sourceInfo})),n=this.promptTemplates.map(o=>({name:o.name,description:o.description,source:"prompt",sourceInfo:o.sourceInfo})),i=this._resourceLoader.getSkills().skills.map(o=>({name:`skill:${o.name}`,description:o.description,source:"skill",sourceInfo:o.sourceInfo}));return[...s,...n,...i]},"getCommands");e.bindCore({sendMessage:f((s,n)=>{this.sendCustomMessage(s,n).catch(i=>{e.emitError({extensionPath:"<runtime>",event:"send_message",error:i instanceof Error?i.message:String(i)})})},"sendMessage"),sendUserMessage:f((s,n)=>{this.sendUserMessage(s,n).catch(i=>{e.emitError({extensionPath:"<runtime>",event:"send_user_message",error:i instanceof Error?i.message:String(i)})})},"sendUserMessage"),appendEntry:f((s,n)=>{this.sessionManager.appendCustomEntry(s,n)},"appendEntry"),setSessionName:f(s=>{this.setSessionName(s)},"setSessionName"),getSessionName:f(()=>this.sessionManager.getSessionName(),"getSessionName"),setLabel:f((s,n)=>{this.sessionManager.appendLabelChange(s,n)},"setLabel"),getActiveTools:f(()=>this.getActiveToolNames(),"getActiveTools"),getAllTools:f(()=>this.getAllTools(),"getAllTools"),setActiveTools:f(s=>this.setActiveToolsByName(s),"setActiveTools"),refreshTools:f(()=>this._refreshToolRegistry(),"refreshTools"),getCommands:t,setModel:f(async s=>this.modelRegistry.hasConfiguredAuth(s)?(await this.setModel(s),!0):!1,"setModel"),getThinkingLevel:f(()=>this.thinkingLevel,"getThinkingLevel"),setThinkingLevel:f(s=>this.setThinkingLevel(s),"setThinkingLevel")},{getModel:f(()=>this.model,"getModel"),isIdle:f(()=>!this.isStreaming,"isIdle"),getSignal:f(()=>this.agent.signal,"getSignal"),abort:f(()=>{if(this._extensionAbortHandler){this._extensionAbortHandler();return}this.abort()},"abort"),hasPendingMessages:f(()=>this.pendingMessageCount>0,"hasPendingMessages"),shutdown:f(()=>{this._extensionShutdownHandler?.()},"shutdown"),getContextUsage:f(()=>this.getContextUsage(),"getContextUsage"),compact:f(s=>{(async()=>{try{const n=await this.compact(s?.customInstructions);s?.onComplete?.(n)}catch(n){const i=n instanceof Error?n:new Error(String(n));s?.onError?.(i)}})()},"compact"),getSystemPrompt:f(()=>this.systemPrompt,"getSystemPrompt"),getSystemPromptOptions:f(()=>this._baseSystemPromptOptions,"getSystemPromptOptions")},{registerProvider:f((s,n)=>{this._modelRegistry.registerProvider(s,n),this._refreshCurrentModelFromRegistry()},"registerProvider"),unregisterProvider:f(s=>{this._modelRegistry.unregisterProvider(s),this._refreshCurrentModelFromRegistry()},"unregisterProvider")})}_refreshToolRegistry(e){const t=new Set(this._toolRegistry.keys()),s=this.getActiveToolNames(),n=this._allowedToolNames,i=this._excludedToolNames,o=f(d=>(!n||n.has(d))&&!i?.has(d),"isAllowedTool"),a=[...this._extensionRunner.getAllRegisteredTools(),...this._customTools.map(d=>({definition:d,sourceInfo:I(`<sdk:${d.name}>`,{source:"sdk"})}))].filter(d=>o(d.definition.name)),l=new Map(Array.from(this._baseToolDefinitions.entries()).filter(([d])=>o(d)).map(([d,x])=>[d,{definition:x,sourceInfo:I(`<builtin:${d}>`,{source:"builtin"})}]));for(const d of a)l.set(d.definition.name,{definition:d.definition,sourceInfo:d.sourceInfo});this._toolDefinitions=l,this._toolPromptSnippets=new Map(Array.from(l.values()).map(({definition:d})=>{const x=this._normalizePromptSnippet(d.promptSnippet);return x?[d.name,x]:void 0}).filter(d=>d!==void 0)),this._toolPromptGuidelines=new Map(Array.from(l.values()).map(({definition:d})=>{const x=this._normalizePromptGuidelines(d.promptGuidelines);return x.length>0?[d.name,x]:void 0}).filter(d=>d!==void 0));const c=this._extensionRunner,h=W(a,c),u=W(Array.from(this._baseToolDefinitions.values()).filter(d=>o(d.name)).map(d=>({definition:d,sourceInfo:I(`<builtin:${d.name}>`,{source:"builtin"})})),c),m=new Map(u.map(d=>[d.name,d]));for(const d of h)m.set(d.name,d);this._toolRegistry=m;const w=(e?.activeToolNames?[...e.activeToolNames]:[...s]).filter(d=>o(d));if(n)for(const d of this._toolRegistry.keys())n.has(d)&&w.push(d);else if(e?.includeAllExtensionTools)for(const d of h)w.push(d.name);else if(!e?.activeToolNames)for(const d of this._toolRegistry.keys())t.has(d)||w.push(d);this.setActiveToolsByName([...new Set(w)])}_buildRuntime(e){const t=this.settingsManager.getImageAutoResize(),s=this.settingsManager.getShellCommandPrefix(),n=this.settingsManager.getShellPath(),i=this._baseToolsOverride?Object.fromEntries(Object.entries(this._baseToolsOverride).map(([l,c])=>[l,be(c)])):xe(this._cwd,{read:{autoResizeImages:t},bash:{commandPrefix:s,shellPath:n}});this._baseToolDefinitions=new Map(Object.entries(i).map(([l,c])=>[l,c]));const o=this._resourceLoader.getExtensions();if(e.flagValues)for(const[l,c]of e.flagValues)o.runtime.flagValues.set(l,c);this._extensionRunner=new pe(o.extensions,o.runtime,this._cwd,this.sessionManager,this._modelRegistry),this._extensionRunnerRef&&(this._extensionRunnerRef.current=this._extensionRunner),this._bindExtensionCore(this._extensionRunner),this._applyExtensionBindings(this._extensionRunner);const r=this._baseToolsOverride?Object.keys(this._baseToolsOverride):["read","bash","edit","write"],a=e.activeToolNames??r;this._refreshToolRegistry({activeToolNames:a,includeAllExtensionTools:e.includeAllExtensionTools})}async reload(){const e=this._extensionRunner.getFlagValues();await fe(this._extensionRunner,{type:"session_shutdown",reason:"reload"}),await this.settingsManager.reload(),ne(),await this._resourceLoader.reload(),this._buildRuntime({activeToolNames:this.getActiveToolNames(),flagValues:e,includeAllExtensionTools:!0}),(this._extensionUIContext||this._extensionCommandContextActions||this._extensionShutdownHandler||this._extensionErrorListener)&&(await this._extensionRunner.emit({type:"session_start",reason:"reload"}),await this.extendResourcesFromExtensions("reload"))}_isNonRetryableProviderLimitError(e){return/GoUsageLimitError|FreeUsageLimitError|Monthly usage limit reached|available balance|insufficient_quota|out of budget|quota exceeded|billing/i.test(e)}_isRateLimited(e){return e.stopReason==="error"&&!!e.errorMessage&&/rate.?limit|429|too many requests|retry delay|quota/i.test(e.errorMessage)}_suggest(e,t){process.env.KODA_TIPS!=="0"&&(this._suggestedThisTask||this._suggestShown.has(e)||(this._suggestShown.add(e),this._suggestedThisTask=!0,this._emit({type:"suggest_feature",id:e,message:t})))}_isRetryableError(e){if(e.stopReason!=="error"||!e.errorMessage)return!1;const t=this.model?.contextWindow??0;if($(e,t))return!1;const s=e.errorMessage;return this._isNonRetryableProviderLimitError(s)?!1:/overloaded|provider.?returned.?error|rate.?limit|too many requests|429|500|502|503|504|service.?unavailable|server.?error|internal.?error|network.?error|connection.?error|connection.?refused|connection.?lost|websocket.?closed|websocket.?error|other side closed|fetch failed|upstream.?connect|reset before headers|socket hang up|ended without|stream ended before message_stop|http2 request did not get a response|timed? out|timeout|terminated|retry delay/i.test(s)}async _prepareRetry(e){const t=this.settingsManager.getRetrySettings();if(!t.enabled)return!1;if(this._retryAttempt++,this._retryAttempt>t.maxRetries)return this._retryAttempt--,!1;const s=t.baseDelayMs*2**(this._retryAttempt-1);this._emit({type:"auto_retry_start",attempt:this._retryAttempt,maxAttempts:t.maxRetries,delayMs:s,errorMessage:e.errorMessage||"Unknown error"});const n=this.agent.state.messages;n.length>0&&n[n.length-1].role==="assistant"&&(this.agent.state.messages=n.slice(0,-1)),this._retryAbortController=new AbortController;try{await oe(s,this._retryAbortController.signal)}catch{const i=this._retryAttempt;return this._retryAttempt=0,this._emit({type:"auto_retry_end",success:!1,attempt:i,finalError:"Retry cancelled"}),!1}finally{this._retryAbortController=void 0}return!0}async _failoverModel(e){if(e.stopReason!=="error"||$(e,this.model?.contextWindow??0))return!1;const t=this.agent.state.model;this._failoverTried.add(`${t.provider}/${t.id}`);const s=(process.env.KODA_FAILOVER||"openadapter/glm-5.1,openadapter/GLM-5,openadapter/Kimi-K2.6,openadapter/Kimi-K2.5,openadapter/MiniMax-M2.7,openadapter/DeepSeek-V4-Pro").split(",").map(a=>a.trim()).filter(Boolean),n=[];for(const a of s){const l=a.indexOf("/"),[c,h]=l>=0?[a.slice(0,l),a.slice(l+1)]:[t.provider,a];if(this._failoverTried.has(`${c}/${h}`))continue;const u=this._modelRegistry.find(c,h);!u||C(u,t)||n.push({key:`${c}/${h}`,model:u})}if(n.length===0)return!1;const i=await this._fetchModelHealth(n[0].model);if(i){const a=f(c=>c?.health!=null&&c.health<=3,"isHealthy"),l=f(c=>c?.latency??6,"latKey");n.sort((c,h)=>{const u=i[c.model.id],m=i[h.model.id],w=a(u)?1:0,d=a(m)?1:0;return w!==d?d-w:l(u)-l(m)})}const o=n[0].model;this._failoverTried.add(n[0].key);const r=this.agent.state.messages;return r.length>0&&r[r.length-1].role==="assistant"&&(this.agent.state.messages=r.slice(0,-1)),this.agent.state.model=o,this._retryAttempt=0,this.sessionManager.appendModelChange(o.provider,o.id),this._emitModelSelect(o,t,"cycle"),this._trace("failover",`${t.id} \u2192 ${o.id} (provider error)`),this._emit({type:"model_switch_notice",message:`\u2191 ${t.id} hit a wall \u2014 Koda beefed up to ${o.id}`}),!0}async _fetchModelHealth(e){try{if(!e.baseUrl)return null;const t=await this._modelRegistry.getApiKeyAndHeaders(e);if(!t?.apiKey)return null;const s=new AbortController,n=setTimeout(()=>s.abort(),2500);try{const i=await fetch(`${e.baseUrl.replace(/\/$/,"")}/model-health`,{headers:{Authorization:`Bearer ${t.apiKey}`,...t.headers||{}},signal:s.signal});return i.ok?(await i.json())?.models??null:null}finally{clearTimeout(n)}}catch{return null}}async _zenitsuGetHealth(){const e=Date.now();if(this._zenitsuHealthCache&&e-this._zenitsuHealthCache.at<3e4)return this._zenitsuHealthCache.models;const t=this._zenitsuPreferred||this.agent.state.model,s=await this._fetchModelHealth(t);return s&&(this._zenitsuHealthCache={at:e,models:s}),this._zenitsuHealthCache?.models}async _zenitsuPickModel(){this._zenitsuPreferred||(this._zenitsuPreferred=this.agent.state.model);const e=this._zenitsuPreferred,t=e.provider,s=Date.now(),n=await this._zenitsuGetHealth()||{},i=f(m=>this._zenitsuCooldown.get(`${t}/${m}`)??0,"cooldownOf"),o=(process.env.KODA_ZENITSU_LADDER||"glm-5.1,Kimi-K2.6,MiniMax-M2.7,GLM-4.7,GLM-5,Kimi-K2.5,DeepSeek-V4-Pro").split(",").map(m=>m.trim()).filter(Boolean),r=[],a=new Set;for(const m of[e.id,...o]){if(a.has(m)||(a.add(m),i(m)>s))continue;const w=this._modelRegistry.find(t,m);if(!w)continue;const d=n[m]||{};r.push({id:m,model:w,healthy:d.health!=null&&d.health<=3,lat:d.latency??6})}if(r.length===0)return e;const l=i(e.id)<=s,c=r.find(m=>m.id===e.id),h=n[e.id]?.health!=null&&n[e.id].health>3;if(l&&c&&!h){const w=c.lat<=5?r.filter(d=>d.id!==e.id&&d.healthy&&d.lat<c.lat).sort((d,x)=>d.lat-x.lat)[0]:void 0;return w?w.model:e}const u=r.filter(m=>m.healthy).sort((m,w)=>m.lat-w.lat);return u.length?u[0].model:r.sort((m,w)=>m.lat-w.lat)[0].model}_messageText(e){const t=e?.content;return typeof t=="string"?t:Array.isArray(t)?t.filter(s=>s?.type==="text").map(s=>s.text).join(`
+`):""}_goalFromMessages(e){const t=Array.isArray(e)?e:[e];for(let s=t.length-1;s>=0;s--)if(t[s]?.role==="user"){const n=this._messageText(t[s]).trim();if(n)return n}}_overseerContext(){const e=[];for(const t of this.agent.state.messages.slice(-12)){const s=t?.role;if(s==="assistant"){const n=this._messageText(t).trim();n&&e.push(`assistant: ${n.slice(0,600)}`);for(const i of Array.isArray(t.content)?t.content:[])(i?.type==="toolUse"||i?.type==="tool_use")&&e.push(`  \u2192 tool ${i.name}(${JSON.stringify(i.input??i.args??{}).slice(0,160)})`)}else(s==="toolResult"||s==="tool")&&e.push(`  result(${t.toolName??""}): ${this._messageText(t).slice(0,300)}`)}return e.join(`
+`).slice(-4e3)}async _getModelTier(e){if(!this._tierCache){this._tierCache=new Map;try{if(e.baseUrl){const t=await this._modelRegistry.getApiKeyAndHeaders(e);if(t?.apiKey){const s=new AbortController,n=setTimeout(()=>s.abort(),3e3);try{const i=await fetch(`${e.baseUrl.replace(/\/$/,"")}/models`,{headers:{Authorization:`Bearer ${t.apiKey}`,...t.headers||{}},signal:s.signal});if(i.ok){const o=await i.json();for(const r of o.data??[])r.tier&&this._tierCache.set(r.id,r.tier)}}finally{clearTimeout(n)}}}}catch{}}return this._tierCache.get(e.id)}_isDegenerateOutput(e){const t=this._messageText(e);if(t.length<200)return!1;const s=t.match(/(.{3,60}?)\1{6,}/s);if(s&&s[0].length>150)return!0;const n=t.split(/\s+/).filter(Boolean);return n.length>60&&new Set(n).size/n.length<.15}async _escalate(e,t){if(this._escalationsDone>=2)return!1;const s=this.agent.state.model,n=(process.env.KODA_ESCALATE||"GLM-4.7,glm-5.1").split(",").map(r=>r.trim()).filter(Boolean);let i;for(const r of n){if(r===s.id||this._escalateTried.has(r))continue;const a=this._modelRegistry.find(s.provider,r);if(a){i=a;break}}if(!i)return!1;if(this._escalationsDone++,this._sessionEscalations++,this._escalateTried.add(i.id),this._sessionEscalations>=2&&(this._stickyModel=i),t?.stripLast){const r=this.agent.state.messages;r.length>0&&r[r.length-1].role==="assistant"&&(this.agent.state.messages=r.slice(0,-1))}this.agent.state.model=i,this._retryAttempt=0,this._overseerRounds=0,this.sessionManager.appendModelChange(i.provider,i.id),this._emitModelSelect(i,s,"cycle");const o=this._stickyModel?" (staying on it \u2014 this session needs the muscle)":"";return this._trace("escalate",`${s.id} \u2192 ${i.id}${this._stickyModel?" [sticky]":""} \u2014 ${e}`),this._emit({type:"model_switch_notice",message:`\u25B2 Koda escalated to ${i.id} \u2014 ${e}${o}`}),!0}listSkillNames(){return this._resourceLoader.getSkills().skills.map(e=>e.name)}getPinnedSkills(){return[...this._pinnedSkills.keys()]}pinSkill(e){const t=this._resourceLoader.getSkills().skills.find(s=>s.name===e);if(!t)return!1;try{const s=F(T(t.filePath,"utf-8")).trim();return this._pinnedSkills.set(e,`<active_skill name="${t.name}" location="${t.filePath}">
+References are relative to ${t.baseDir}. Follow these instructions for this session:
+
+${s}
+</active_skill>`),this._refreshSystemPrompt(),!0}catch{return!1}}unpinSkill(e){const t=this._pinnedSkills.delete(e);return t&&this._refreshSystemPrompt(),t}_refreshSystemPrompt(){this._baseSystemPrompt=this._rebuildSystemPrompt(this.getActiveToolNames()),this.agent.state.systemPrompt=this._baseSystemPrompt}getSessionGoal(){return this._sessionGoal}clearSessionGoal(){this._sessionGoal=void 0,this._emit({type:"goal_cleared"})}async setSessionGoal(e){const t=e.trim();if(!t)return;const s=await this._generateMilestones(t);this._sessionGoal={text:t,milestones:s.map(n=>({text:n,done:!1})),achieved:!1},this._emit({type:"goal_set",text:t,milestones:this._sessionGoal.milestones})}async _generateMilestones(e){const t=[e];try{const s=this.agent.state.model,n=this._modelRegistry.find(s.provider,process.env.KODA_OVERSEER_MODEL||"GLM-4.7")||s;if(!n.baseUrl)return t;const i=await this._modelRegistry.getApiKeyAndHeaders(n);if(!i?.apiKey)return t;const o=await fetch(`${n.baseUrl.replace(/\/$/,"")}/chat/completions`,{method:"POST",headers:{Authorization:`Bearer ${i.apiKey}`,"Content-Type":"application/json",...i.headers||{}},body:JSON.stringify({model:n.id,temperature:0,messages:[{role:"system",content:"Break the user's session goal into 3-6 concrete, verifiable milestones (short imperative phrases, in order). Reply with ONLY a JSON array of strings."},{role:"user",content:e}]})});if(!o.ok)return t;let a=((await o.json())?.choices?.[0]?.message?.content??"").trim();a=a.replace(/<think>[\s\S]*?<\/think>/gi,"");const l=a.match(/\[[\s\S]*\]/);l&&(a=l[0]);const c=JSON.parse(a),h=Array.isArray(c)?c.filter(u=>typeof u=="string"&&u.trim()).map(u=>u.trim()).slice(0,6):[];return h.length?h:t}catch{return t}}_trace(e,t){this._harnessTrace.push({ts:Date.now(),kind:e,detail:t}),this._harnessTrace.length>60&&this._harnessTrace.shift()}getHarnessTrace(){return[...this._harnessTrace]}_gatewayOpen(){return Date.now()<this._gatewayCircuit.openUntil}_gatewayResult(e){e?this._gatewayCircuit.failures=0:++this._gatewayCircuit.failures>=3&&(this._gatewayCircuit.openUntil=Date.now()+3e4,this._gatewayCircuit.failures=0,this._trace("circuit","gateway calls paused 30s after repeated failures"))}_detectVerify(){if(this._verifyCmd!==void 0)return this._verifyCmd;const e=f(s=>(this._verifyCmd=s,s),"set"),t=process.env.KODA_VERIFY;if(t==="0")return e(null);if(t)return e(t);try{const s=this._cwd,n=S(s,"package.json");if(k(n)){const i=JSON.parse(T(n,"utf-8")),o=i.scripts||{},r=Object.keys({...i.dependencies||{},...i.devDependencies||{}}).join(" ");if(/\b(react|vue|svelte|next|nuxt|vite|@angular\/core|solid-js|preact|astro)\b/i.test(r)){const l=[];if(o.typecheck?l.push("npm run typecheck"):o["type-check"]?l.push("npm run type-check"):k(S(s,"tsconfig.json"))&&l.push("npx tsc --noEmit"),o.lint&&l.push("npm run lint"),l.length)return e(l.join(" && "));if(o.build)return e("npm run build")}for(const l of["typecheck","type-check","build","lint"])if(o[l])return e(`npm run ${l}`)}if(k(S(s,"tsconfig.json")))return e("npx tsc --noEmit");if(k(S(s,"go.mod")))return e("go build ./...");if(k(S(s,"Cargo.toml")))return e("cargo check")}catch{}return e(null)}async _runVerification(e,t){try{const s=await ue("sh",["-c",e],this._cwd,{timeout:12e4,signal:t}),n=`${s.stdout}
+${s.stderr}`.trim();return{passed:s.code===0&&!s.killed,output:n.slice(-1500)}}catch(s){return{passed:!0,output:`(verify could not run: ${s.message})`}}}async _overseerCheck(e){if(e.stopReason==="error")return!1;const t=this.agent.state.model,s=this._sessionGoal,n=!!s&&!s.achieved,i=await this._getModelTier(t),o=process.env.KODA_OVERSEER!=="0"&&this._overseerRounds<2&&!!this._currentGoal&&!!i&&i!=="max",r=this._codeChangedThisTask?this._detectVerify():null;if(!r&&!o&&!n)return!1;this._emit({type:"overseer_start",model:t.id}),this._overseerAbortController=new AbortController;const a=this._overseerAbortController.signal;try{let l;if(r){const h=await this._runVerification(r,a);if(a.aborted)return this._trace("overseer","cancelled during verify"),this._emit({type:"overseer_end",outcome:"skipped",detail:"cancelled"}),!1;if(this._trace("verify",`${r} \u2192 ${h.passed?"pass":"FAIL"}`),l=h.passed,!h.passed){const u=this._overseerRounds>=1&&await this._escalate("build/tests still failing");return u||this._overseerRounds++,this.agent.followUp({role:"user",content:[{type:"text",text:`Not done \u2014 \`${r}\` failed. Fix the root cause and re-run it until it passes:
+
+${h.output}`}],timestamp:Date.now()}),this._emit({type:"overseer_end",outcome:"nudging",detail:u?"escalating: verification failing":`verification failed: ${r}`}),!0}}if(!o&&!n)return this._emit({type:"overseer_end",outcome:"complete"}),!1;const c=await this._taskEndReview(o?this._currentGoal:void 0,this._overseerContext(),this.getActiveToolNames(),n?s:void 0,l,a);if(a.aborted)return this._trace("overseer","cancelled during review"),this._emit({type:"overseer_end",outcome:"skipped",detail:"cancelled"}),!1;if(!c)return this._emit({type:"overseer_end",outcome:"skipped",detail:this._overseerLastError||"unknown"}),!1;if(n&&s&&c.goal&&(c.goal.milestonesDone.forEach((h,u)=>{s.milestones[u]&&h&&(s.milestones[u].done=!0)}),s.achieved=c.goal.achieved||s.milestones.every(h=>h.done),this._emit({type:"goal_progress",text:s.text,milestones:s.milestones.map(h=>({...h})),achieved:s.achieved,drift:c.goal.drift||void 0})),o&&c.task&&!c.task.done){const h=this._overseerRounds>=1&&await this._escalate(`stuck \u2014 ${c.task.missing}`);h||this._overseerRounds++;const u=n&&c.goal?.drift?` You've also drifted from the session goal: ${c.goal.drift}.`:"";return this.agent.followUp({role:"user",content:[{type:"text",text:`Not done yet \u2014 keep going. ${c.task.missing}${u} Do exactly that, then verify it works before you stop. If you're unsure how, call ask_for_help to get a step-by-step plan from a senior model and follow it.`}],timestamp:Date.now()}),this._emit({type:"overseer_end",outcome:"nudging",detail:h?`escalating: ${c.task.missing}`:c.task.missing}),!0}return this._emit({type:"overseer_end",outcome:"complete"}),!1}finally{this._overseerAbortController=void 0}}async _taskEndReview(e,t,s,n,i,o){try{if(this._gatewayOpen())return this._overseerLastError="gateway circuit open (recent failures)",null;const r=this.agent.state.model,a=[process.env.KODA_OVERSEER_MODEL||"GLM-4.7","GLM-5","glm-5.1"],l=a.map(c=>this._modelRegistry.find(r.provider,c)).filter(c=>!!c&&c.id!==r.id);if(l.length===0)return this._overseerLastError=`no judge model registered (tried ${a.join(", ")})`,null;for(const c of l){if(o?.aborted)return this._overseerLastError="cancelled",null;const h=await this._reviewOnce(c,e,t,s,n,i,o);if(this._gatewayResult(!h.error||h.error.startsWith("unparseable")),h.review)return h.review;this._overseerLastError=`${c.id}: ${h.error}`}return null}catch(r){return this._overseerLastError=r?.message?.slice(0,80)||"error",null}}async _reviewOnce(e,t,s,n,i,o,r){try{if(r?.aborted)return{error:"cancelled"};if(!e.baseUrl)return{error:"no baseUrl"};const a=await this._modelRegistry.getApiKeyAndHeaders(e);if(!a?.apiKey)return{error:"no api key"};const l=["You are a meticulous reviewer for a coding agent. Ground every judgement in the WORK LOG (a record of tool calls and their results); be skeptical of any claim with no matching evidence."],c=[];if(t&&(l.push("TASK REVIEW \u2014 decide whether EVERY explicit deliverable of the REQUEST is present AND works. Judge only its explicit asks. If it says to run/test/verify, it is done only if the log shows that ran and succeeded. Completion requires EVIDENCE in the work log \u2014 files actually written, commands that actually ran and succeeded. The assistant's reasoning, plans, intentions, or confident claims are NOT evidence; if the log shows only thinking or failed/partial attempts with no successful results producing the deliverables, it is NOT done. "+(o===!0?"NOTE: the project's build/tests were just run by the harness and PASSED, so the code works \u2014 judge only whether the request's scope is fully addressed. ":"")+`The assistant has these TOOLS: ${n.join(", ")||"(none)"}; if it guessed or gave up without using a tool that could have found the answer, it is NOT done \u2014 name the tool to call. "taskMissing" must name ONLY what is left (imperative; exact file/function/command/tool), never restate done work.`),c.push('"taskDone": true|false, "taskMissing": "<only what is left; empty if done>"')),i){const w=i.milestones.map((d,x)=>`${x}: ${d.text}`).join("; ");l.push(`GOAL REVIEW \u2014 the session goal is: "${i.text}". Milestones by index: ${w}. From the WORK LOG, return "milestonesDone" as an array of booleans (one per index; true only if clearly accomplished in the log). Set "goalAchieved" true only if the whole goal is genuinely met. If recent work clearly wandered onto something unrelated to the goal, put a one-sentence note in "drift" (else "").`),c.push('"milestonesDone": [booleans], "goalAchieved": true|false, "drift": "<note or empty>"')}l.push(`Output ONLY a JSON object, no prose or fences: { ${c.join(", ")} }.`);const h=new AbortController,u=setTimeout(()=>h.abort(),45e3),m=f(()=>h.abort(),"onAbort");r?.addEventListener("abort",m,{once:!0});try{const w=await fetch(`${e.baseUrl.replace(/\/$/,"")}/chat/completions`,{method:"POST",headers:{Authorization:`Bearer ${a.apiKey}`,"Content-Type":"application/json",...a.headers||{}},body:JSON.stringify({model:e.id,temperature:0,messages:[{role:"system",content:l.join(`
+
+`)},{role:"user",content:`${t?`REQUEST:
+${t}
+
+`:""}WORK LOG:
+${s}`}]}),signal:h.signal});if(!w.ok)return{error:`HTTP ${w.status}`};const d=await w.json();let x=(d?.choices?.[0]?.message?.content??d?.choices?.[0]?.message?.reasoning_content??"").trim();x=x.replace(/<think>[\s\S]*?<\/think>/gi,"").replace(/^```(?:json)?/i,"").replace(/```\s*$/,"").trim();const g=x.match(/\{[\s\S]*\}/);if(g&&(x=g[0]),!x)return{error:"empty response"};let p;try{p=JSON.parse(x)}catch{return{error:"unparseable JSON"}}const _={};if(t){const b=p?.taskDone===!0||p?.taskDone==="true",v=String(Array.isArray(p?.taskMissing)?p.taskMissing.join("; "):p?.taskMissing??"").trim();_.task=!b&&!v?{done:!0,missing:""}:{done:b,missing:v}}return i&&(_.goal={milestonesDone:Array.isArray(p?.milestonesDone)?p.milestonesDone.map(b=>b===!0||b==="true"):[],achieved:p?.goalAchieved===!0||p?.goalAchieved==="true",drift:String(p?.drift??"").trim()}),{review:_}}finally{clearTimeout(u),r?.removeEventListener("abort",m)}}catch(a){return{error:a?.name==="AbortError"?"timeout 45s":(a?.message||"error").slice(0,80)}}}_isMutatingBash(e){return/>>?\s*(?!\/dev\/null|&?\d)/.test(e)?!0:/\b(rm|rmdir|mv|cp|mkdir|touch|truncate|dd|chmod|chown|ln|tee|patch|sed\s+-i|install|npm\s+(i|install|add|remove|uninstall|ci)|pnpm\s+(install|add|remove)|yarn\s+(add|remove|install)|pip\d?\s+install|cargo\s+(add|install)|go\s+install|apt(-get)?\s+(install|remove)|brew\s+(install|uninstall)|git\s+(commit|add|push|checkout|reset|merge|rebase|clean|stash|apply|rm|mv))\b/.test(e)}_parseRetrySeconds(e){const t=/(?:retry|again)[^0-9]{0,16}?(\d+(?:\.\d+)?)\s*(ms|s\b|sec|second)/i.exec(e);if(t){const s=parseFloat(t[1]);return/ms/i.test(t[2])?Math.max(1,Math.ceil(s/1e3)):Math.ceil(s)}}async _zenitsuHandle429(e){if(e.stopReason!=="error"||!e.errorMessage||!/rate.?limit|429|too many requests|retry delay|quota/i.test(e.errorMessage))return!1;const t=this.agent.state.model,s=this._parseRetrySeconds(e.errorMessage)??30;this._zenitsuCooldown.set(`${t.provider}/${t.id}`,Date.now()+s*1e3);const n=await this._zenitsuPickModel();if(!n||C(n,t))return!1;const i=this.agent.state.messages;return i.length>0&&i[i.length-1].role==="assistant"&&(this.agent.state.messages=i.slice(0,-1)),this.agent.state.model=n,this._retryAttempt=0,this.sessionManager.appendModelChange(n.provider,n.id),this._emitModelSelect(n,t,"cycle"),this._trace("zenitsu",`${t.id} \u2192 ${n.id} (429)`),this._emit({type:"model_switch_notice",message:`\xBB ${t.id} rate-limited \u2014 Koda zipped over to ${n.id} (no waiting)`}),!0}_maybeReturnHome(){const e=this._stickyModel??this._homeModel;if(!e)return!1;const t=this.agent.state.model;return t.id===e.id&&t.provider===e.provider||!this._modelRegistry.hasConfiguredAuth(e)||(this._zenitsuCooldown.get(`${e.provider}/${e.id}`)??0)>Date.now()?!1:(this.agent.state.model=e,this._zenitsuPreferred=e,this._retryAttempt=0,this.sessionManager.appendModelChange(e.provider,e.id),this._emitModelSelect(e,t,"cycle"),this._trace("revert",`\u2192 ${e.id} (preferred available)`),this._emit({type:"model_switch_notice",message:`\u2190 Back on your pick \u2014 ${e.id}`}),!0)}abortRetry(){this._retryAbortController?.abort()}get isRetrying(){return this._retryAbortController!==void 0}get autoRetryEnabled(){return this.settingsManager.getRetryEnabled()}setAutoRetryEnabled(e){this.settingsManager.setRetryEnabled(e)}async executeBash(e,t,s){this._bashAbortController=new AbortController;const n=this.settingsManager.getShellCommandPrefix(),i=this.settingsManager.getShellPath(),o=n?`${n}
+${e}`:e;try{const r=await ae(o,this.sessionManager.getCwd(),s?.operations??we({shellPath:i}),{onChunk:t,signal:this._bashAbortController.signal});return this.recordBashResult(e,r,s),r}finally{this._bashAbortController=void 0}}recordBashResult(e,t,s){const n={role:"bashExecution",command:e,output:t.output,exitCode:t.exitCode,cancelled:t.cancelled,truncated:t.truncated,fullOutputPath:t.fullOutputPath,timestamp:Date.now(),excludeFromContext:s?.excludeFromContext};this.isStreaming?this._pendingBashMessages.push(n):(this.agent.state.messages.push(n),this.sessionManager.appendMessage(n))}abortBash(){this._bashAbortController?.abort()}get isBashRunning(){return this._bashAbortController!==void 0}get hasPendingBashMessages(){return this._pendingBashMessages.length>0}_flushPendingBashMessages(){if(this._pendingBashMessages.length!==0){for(const e of this._pendingBashMessages)this.agent.state.messages.push(e),this.sessionManager.appendMessage(e);this._pendingBashMessages=[]}}setSessionName(e){this.sessionManager.appendSessionInfo(e),this._emit({type:"session_info_changed",name:this.sessionManager.getSessionName()})}async navigateTree(e,t={}){const s=this.sessionManager.getLeafId();if(e===s)return{cancelled:!1};if(t.summarize&&!this.model)throw new Error("No model available for summarization");const n=this.sessionManager.getEntry(e);if(!n)throw new Error(`Entry ${e} not found`);const{entries:i,commonAncestorId:o}=le(this.sessionManager,s,e);let r=t.customInstructions,a=t.replaceInstructions,l=t.label;const c={targetId:e,oldLeafId:s,commonAncestorId:o,entriesToSummarize:i,userWantsSummary:t.summarize??!1,customInstructions:r,replaceInstructions:a,label:l};this._branchSummaryAbortController=new AbortController;try{let h,u=!1;if(this._extensionRunner.hasHandlers("session_before_tree")){const _=await this._extensionRunner.emit({type:"session_before_tree",preparation:c,signal:this._branchSummaryAbortController.signal});if(_?.cancel)return{cancelled:!0};_?.summary&&t.summarize&&(h=_.summary,u=!0),_?.customInstructions!==void 0&&(r=_.customInstructions),_?.replaceInstructions!==void 0&&(a=_.replaceInstructions),_?.label!==void 0&&(l=_.label)}let m,w;if(t.summarize&&i.length>0&&!h){const _=this.model,{apiKey:b,headers:v}=await this._getRequiredRequestAuth(_),A=this.settingsManager.getBranchSummarySettings(),y=await ce(i,{model:_,apiKey:b,headers:v,signal:this._branchSummaryAbortController.signal,customInstructions:r,replaceInstructions:a,reserveTokens:A.reserveTokens,streamFn:this.agent.streamFn});if(y.aborted)return{cancelled:!0,aborted:!0};if(y.error)throw new Error(y.error);m=y.summary,w={readFiles:y.readFiles||[],modifiedFiles:y.modifiedFiles||[]}}else h&&(m=h.summary,w=h.details);let d,x;n.type==="message"&&n.message.role==="user"?(d=n.parentId,x=this._extractUserMessageText(n.message.content)):n.type==="custom_message"?(d=n.parentId,x=typeof n.content=="string"?n.content:n.content.filter(_=>_.type==="text").map(_=>_.text).join("")):d=e;let g;if(m){const _=this.sessionManager.branchWithSummary(d,m,w,u);g=this.sessionManager.getEntry(_),l&&this.sessionManager.appendLabelChange(_,l)}else d===null?this.sessionManager.resetLeaf():this.sessionManager.branch(d);l&&!m&&this.sessionManager.appendLabelChange(e,l);const p=this.sessionManager.buildSessionContext();return this.agent.state.messages=p.messages,await this._extensionRunner.emit({type:"session_tree",newLeafId:this.sessionManager.getLeafId(),oldLeafId:s,summaryEntry:g,fromExtension:m?u:void 0}),{editorText:x,cancelled:!1,summaryEntry:g}}finally{this._branchSummaryAbortController=void 0}}getUserMessagesForForking(){const e=this.sessionManager.getEntries(),t=[];for(const s of e){if(s.type!=="message"||s.message.role!=="user")continue;const n=this._extractUserMessageText(s.message.content);n&&t.push({entryId:s.id,text:n})}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,n=e.messages.filter(h=>h.role==="toolResult").length;let i=0,o=0,r=0,a=0,l=0,c=0;for(const h of e.messages)if(h.role==="assistant"){const u=h;i+=u.content.filter(m=>m.type==="toolCall").length,o+=u.usage.input,r+=u.usage.output,a+=u.usage.cacheRead,l+=u.usage.cacheWrite,c+=u.usage.cost.total}return{sessionFile:this.sessionFile,sessionId:this.sessionId,userMessages:t,assistantMessages:s,toolCalls:i,toolResults:n,totalMessages:e.messages.length,tokens:{input:o,output:r,cacheRead:a,cacheWrite:l,total:o+r+a+l},cost:c,contextUsage:this.getContextUsage()}}getContextUsage(){const e=this.model;if(!e)return;const t=e.contextWindow??0;if(t<=0)return;const s=this.sessionManager.getBranch(),n=V(s);if(n){const r=s.lastIndexOf(n);let a=!1;for(let l=s.length-1;l>r;l--){const c=s[l];if(c.type==="message"&&c.message.role==="assistant"){const h=c.message;if(h.stopReason!=="aborted"&&h.stopReason!=="error"){z(h.usage)>0&&(a=!0);break}}}if(!a)return{tokens:null,contextWindow:t,percent:null}}const i=G(this.messages),o=i.tokens/t*100;return{tokens:i.tokens,contextWindow:t,percent:o}}async exportToHtml(e){const t=this.settingsManager.getTheme(),s=ge({getToolDefinition:f(n=>this.getToolDefinition(n),"getToolDefinition"),theme:ie,cwd:this.sessionManager.getCwd()});return await me(this.sessionManager,this.state,{outputPath:e,themeName:t,toolRenderer:s})}exportToJsonl(e){const t=re(e??`session-${new Date().toISOString().replace(/[:.]/g,"-")}.jsonl`,process.cwd()),s=U(t);k(s)||D(s,{recursive:!0});const n={type:"session",version:_e,id:this.sessionManager.getSessionId(),timestamp:new Date().toISOString(),cwd:this.sessionManager.getCwd()},i=this.sessionManager.getBranch(),o=[JSON.stringify(n)];let r=null;for(const a of i){const l={...a,parentId:r};o.push(JSON.stringify(l)),r=a.id}return L(t,`${o.join(`
+`)}
+`),t}getLastAssistantText(){const e=this.messages.slice().reverse().find(s=>{if(s.role!=="assistant")return!1;const n=s;return!(n.stopReason==="aborted"&&n.content.length===0)});if(!e)return;let t="";for(const s of e.content)s.type==="text"&&(t+=s.text);return t.trim()||void 0}createReplacedSessionContext(){const e=Object.defineProperties({},Object.getOwnPropertyDescriptors(this._extensionRunner.createCommandContext()));return e.sendMessage=(t,s)=>this.sendCustomMessage(t,s),e.sendUserMessage=(t,s)=>this.sendUserMessage(t,s),e}hasExtensionHandlers(e){return this._extensionRunner.hasHandlers(e)}get extensionRunner(){return this._extensionRunner}}export{Je as AgentSession,Ve as parseSkillBlock};

package/dist/core/auth-guidance.d.ts

@@ -0,0 +1,4 @@
+export declare function getProviderLoginHelp(): string;
+export declare function formatNoModelsAvailableMessage(): string;
+export declare function formatNoModelSelectedMessage(): string;
+export declare function formatNoApiKeyFoundMessage(provider: string): string;

package/dist/core/auth-guidance.js

@@ -0,0 +1,8 @@
+var l=Object.defineProperty;var e=(o,t)=>l(o,"name",{value:t,configurable:!0});import{join as r}from"node:path";import{getDocsPath as i}from"../config.js";const s="unknown";function n(){return["Use /login to log into a provider via OAuth or API key. See:",`  ${r(i(),"providers.md")}`,`  ${r(i(),"models.md")}`].join(`
+`)}e(n,"getProviderLoginHelp");function p(){return`No models available. ${n()}`}e(p,"formatNoModelsAvailableMessage");function u(){return`No model selected.
+
+${n()}
+
+Then use /model to select a model.`}e(u,"formatNoModelSelectedMessage");function c(o){return`No API key found for ${o===s?"the selected model":o}.
+
+${n()}`}e(c,"formatNoApiKeyFoundMessage");export{c as formatNoApiKeyFoundMessage,u as formatNoModelSelectedMessage,p as formatNoModelsAvailableMessage,n as getProviderLoginHelp};

package/dist/core/auth-storage.d.ts

@@ -0,0 +1,140 @@
+/**
+ * Credential storage for API keys and OAuth tokens.
+ * Handles loading, saving, and refreshing credentials from auth.json.
+ *
+ * Uses file locking to prevent race conditions when multiple pi instances
+ * try to refresh tokens simultaneously.
+ */
+import { type OAuthCredentials, type OAuthLoginCallbacks, type OAuthProviderId } from "@openadapter/koda-ai";
+export type ApiKeyCredential = {
+    type: "api_key";
+    key: string;
+};
+export type OAuthCredential = {
+    type: "oauth";
+} & OAuthCredentials;
+export type AuthCredential = ApiKeyCredential | OAuthCredential;
+export type AuthStorageData = Record<string, AuthCredential>;
+export type AuthStatus = {
+    configured: boolean;
+    source?: "stored" | "runtime" | "environment" | "fallback" | "models_json_key" | "models_json_command";
+    label?: string;
+};
+type LockResult<T> = {
+    result: T;
+    next?: string;
+};
+export interface AuthStorageBackend {
+    withLock<T>(fn: (current: string | undefined) => LockResult<T>): T;
+    withLockAsync<T>(fn: (current: string | undefined) => Promise<LockResult<T>>): Promise<T>;
+}
+export declare class FileAuthStorageBackend implements AuthStorageBackend {
+    private authPath;
+    constructor(authPath?: string);
+    private ensureParentDir;
+    private ensureFileExists;
+    private acquireLockSyncWithRetry;
+    withLock<T>(fn: (current: string | undefined) => LockResult<T>): T;
+    withLockAsync<T>(fn: (current: string | undefined) => Promise<LockResult<T>>): Promise<T>;
+}
+export declare class InMemoryAuthStorageBackend implements AuthStorageBackend {
+    private value;
+    withLock<T>(fn: (current: string | undefined) => LockResult<T>): T;
+    withLockAsync<T>(fn: (current: string | undefined) => Promise<LockResult<T>>): Promise<T>;
+}
+/**
+ * Credential storage backed by a JSON file.
+ */
+export declare class AuthStorage {
+    private data;
+    private runtimeOverrides;
+    private fallbackResolver?;
+    private loadError;
+    private errors;
+    private storage;
+    private constructor();
+    static create(authPath?: string): AuthStorage;
+    static fromStorage(storage: AuthStorageBackend): AuthStorage;
+    static inMemory(data?: AuthStorageData): AuthStorage;
+    /**
+     * Set a runtime API key override (not persisted to disk).
+     * Used for CLI --api-key flag.
+     */
+    setRuntimeApiKey(provider: string, apiKey: string): void;
+    /**
+     * Remove a runtime API key override.
+     */
+    removeRuntimeApiKey(provider: string): void;
+    /**
+     * Set a fallback resolver for API keys not found in auth.json or env vars.
+     * Used for custom provider keys from models.json.
+     */
+    setFallbackResolver(resolver: (provider: string) => string | undefined): void;
+    private recordError;
+    private parseStorageData;
+    /**
+     * Reload credentials from storage.
+     */
+    reload(): void;
+    private persistProviderChange;
+    /**
+     * Get credential for a provider.
+     */
+    get(provider: string): AuthCredential | undefined;
+    /**
+     * Set credential for a provider.
+     */
+    set(provider: string, credential: AuthCredential): void;
+    /**
+     * Remove credential for a provider.
+     */
+    remove(provider: string): void;
+    /**
+     * List all providers with credentials.
+     */
+    list(): string[];
+    /**
+     * Check if credentials exist for a provider in auth.json.
+     */
+    has(provider: string): boolean;
+    /**
+     * Check if any form of auth is configured for a provider.
+     * Unlike getApiKey(), this doesn't refresh OAuth tokens.
+     */
+    hasAuth(provider: string): boolean;
+    /**
+     * Return auth status without exposing credential values or refreshing tokens.
+     */
+    getAuthStatus(provider: string): AuthStatus;
+    /**
+     * Get all credentials (for passing to getOAuthApiKey).
+     */
+    getAll(): AuthStorageData;
+    drainErrors(): Error[];
+    /**
+     * Login to an OAuth provider.
+     */
+    login(providerId: OAuthProviderId, callbacks: OAuthLoginCallbacks): Promise<void>;
+    /**
+     * Logout from a provider.
+     */
+    logout(provider: string): void;
+    private refreshOAuthTokenWithLock;
+    /**
+     * Get API key for a provider.
+     * Priority:
+     * 1. Runtime override (CLI --api-key)
+     * 2. API key from auth.json
+     * 3. OAuth token from auth.json (auto-refreshed with locking)
+     * 4. Environment variable
+     * 5. Fallback resolver (models.json custom providers)
+     */
+    getApiKey(providerId: string, options?: {
+        includeFallback?: boolean;
+    }): Promise<string | undefined>;
+    /**
+     * Get all registered OAuth providers
+     */
+    getOAuthProviders(): import("@openadapter/koda-ai").OAuthProviderInterface[];
+}
+export {};

package/dist/core/auth-storage.js

@@ -0,0 +1 @@
+var x=Object.defineProperty;var h=(f,t)=>x(f,"name",{value:t,configurable:!0});import{findEnvKeys as S,getEnvApiKey as w}from"@openadapter/koda-ai";import{getOAuthApiKey as O,getOAuthProvider as d,getOAuthProviders as v}from"@openadapter/koda-ai/oauth";import{chmodSync as m,existsSync as l,mkdirSync as D,readFileSync as p,writeFileSync as y}from"fs";import{dirname as K,join as k}from"path";import A from"proper-lockfile";import{getAgentDir as E}from"../config.js";import{normalizePath as b}from"../utils/paths.js";import{resolveConfigValue as C}from"./resolve-config-value.js";class L{static{h(this,"FileAuthStorageBackend")}authPath;constructor(t=k(E(),"auth.json")){this.authPath=b(t)}ensureParentDir(){const t=K(this.authPath);l(t)||D(t,{recursive:!0,mode:448})}ensureFileExists(){l(this.authPath)||(y(this.authPath,"{}","utf-8"),m(this.authPath,384))}acquireLockSyncWithRetry(t){let i;for(let s=1;s<=10;s++)try{return A.lockSync(t,{realpath:!1})}catch(n){if((typeof n=="object"&&n!==null&&"code"in n?String(n.code):void 0)!=="ELOCKED"||s===10)throw n;i=n;const a=Date.now();for(;Date.now()-a<20;);}throw i??new Error("Failed to acquire auth storage lock")}withLock(t){this.ensureParentDir(),this.ensureFileExists();let e;try{e=this.acquireLockSyncWithRetry(this.authPath);const r=l(this.authPath)?p(this.authPath,"utf-8"):void 0,{result:i,next:s}=t(r);return s!==void 0&&(y(this.authPath,s,"utf-8"),m(this.authPath,384)),i}finally{e&&e()}}async withLockAsync(t){this.ensureParentDir(),this.ensureFileExists();let e,r=!1,i;const s=h(()=>{if(r)throw i??new Error("Auth storage lock was compromised")},"throwIfCompromised");try{e=await A.lock(this.authPath,{retries:{retries:10,factor:2,minTimeout:100,maxTimeout:1e4,randomize:!0},stale:3e4,onCompromised:h(o=>{r=!0,i=o},"onCompromised")}),s();const n=l(this.authPath)?p(this.authPath,"utf-8"):void 0,{result:u,next:a}=await t(n);return s(),a!==void 0&&(y(this.authPath,a,"utf-8"),m(this.authPath,384)),s(),u}finally{if(e)try{await e()}catch{}}}}class R{static{h(this,"InMemoryAuthStorageBackend")}value;withLock(t){const{result:e,next:r}=t(this.value);return r!==void 0&&(this.value=r),e}async withLockAsync(t){const{result:e,next:r}=await t(this.value);return r!==void 0&&(this.value=r),e}}class c{static{h(this,"AuthStorage")}data={};runtimeOverrides=new Map;fallbackResolver;loadError=null;errors=[];storage;constructor(t){this.storage=t,this.reload()}static create(t){return new c(new L(t??k(E(),"auth.json")))}static fromStorage(t){return new c(t)}static inMemory(t={}){const e=new R;return e.withLock(()=>({result:void 0,next:JSON.stringify(t,null,2)})),c.fromStorage(e)}setRuntimeApiKey(t,e){this.runtimeOverrides.set(t,e)}removeRuntimeApiKey(t){this.runtimeOverrides.delete(t)}setFallbackResolver(t){this.fallbackResolver=t}recordError(t){const e=t instanceof Error?t:new Error(String(t));this.errors.push(e)}parseStorageData(t){return t?JSON.parse(t):{}}reload(){let t;try{this.storage.withLock(e=>(t=e,{result:void 0})),this.data=this.parseStorageData(t),this.loadError=null}catch(e){this.loadError=e,this.recordError(e)}}persistProviderChange(t,e){if(!this.loadError)try{this.storage.withLock(r=>{const s={...this.parseStorageData(r)};return e?s[t]=e:delete s[t],{result:void 0,next:JSON.stringify(s,null,2)}})}catch(r){this.recordError(r)}}get(t){return this.data[t]??void 0}set(t,e){this.data[t]=e,this.persistProviderChange(t,e)}remove(t){delete this.data[t],this.persistProviderChange(t,void 0)}list(){return Object.keys(this.data)}has(t){return t in this.data}hasAuth(t){return!!(this.runtimeOverrides.has(t)||this.data[t]||w(t)||this.fallbackResolver?.(t))}getAuthStatus(t){if(this.data[t])return{configured:!0,source:"stored"};if(this.runtimeOverrides.has(t))return{configured:!1,source:"runtime",label:"--api-key"};const e=S(t);return e?.[0]?{configured:!1,source:"environment",label:e[0]}:this.fallbackResolver?.(t)?{configured:!1,source:"fallback",label:"custom provider config"}:{configured:!1}}getAll(){return{...this.data}}drainErrors(){const t=[...this.errors];return this.errors=[],t}async login(t,e){const r=d(t);if(!r)throw new Error(`Unknown OAuth provider: ${t}`);const i=await r.login(e);this.set(t,{type:"oauth",...i})}logout(t){this.remove(t)}async refreshOAuthTokenWithLock(t){const e=d(t);return e?await this.storage.withLockAsync(async i=>{const s=this.parseStorageData(i);this.data=s,this.loadError=null;const n=s[t];if(n?.type!=="oauth")return{result:null};if(Date.now()<n.expires)return{result:{apiKey:e.getApiKey(n),newCredentials:n}};const u={};for(const[P,g]of Object.entries(s))g.type==="oauth"&&(u[P]=g);const a=await O(t,u);if(!a)return{result:null};const o={...s,[t]:{type:"oauth",...a.newCredentials}};return this.data=o,this.loadError=null,{result:a,next:JSON.stringify(o,null,2)}}):null}async getApiKey(t,e){const r=this.runtimeOverrides.get(t);if(r)return r;const i=this.data[t];if(i?.type==="api_key")return C(i.key);if(i?.type==="oauth"){const n=d(t);if(!n)return;if(Date.now()>=i.expires)try{const a=await this.refreshOAuthTokenWithLock(t);if(a)return a.apiKey}catch(a){this.recordError(a),this.reload();const o=this.data[t];return o?.type==="oauth"&&Date.now()<o.expires?n.getApiKey(o):void 0}else return n.getApiKey(i)}const s=w(t);if(s)return s;if(e?.includeFallback!==!1)return this.fallbackResolver?.(t)??void 0}getOAuthProviders(){return v()}}export{c as AuthStorage,L as FileAuthStorageBackend,R as InMemoryAuthStorageBackend};

package/dist/core/bash-executor.d.ts

@@ -0,0 +1,31 @@
+/**
+ * Bash command execution with streaming support and cancellation.
+ *
+ * This module provides a unified bash execution implementation used by:
+ * - AgentSession.executeBash() for interactive and RPC modes
+ * - Direct calls from modes that need bash execution
+ */
+import type { BashOperations } from "./tools/bash.ts";
+export interface BashExecutorOptions {
+    /** Callback for streaming output chunks (already sanitized) */
+    onChunk?: (chunk: string) => void;
+    /** AbortSignal for cancellation */
+    signal?: AbortSignal;
+}
+export interface BashResult {
+    /** Combined stdout + stderr output (sanitized, possibly truncated) */
+    output: string;
+    /** Process exit code (undefined if killed/cancelled) */
+    exitCode: number | undefined;
+    /** Whether the command was cancelled via signal */
+    cancelled: boolean;
+    /** Whether the output was truncated */
+    truncated: boolean;
+    /** Path to temp file containing full output (if output exceeded truncation threshold) */
+    fullOutputPath?: string;
+}
+/**
+ * Execute a bash command using custom BashOperations.
+ * Used for remote execution (SSH, containers, etc.).
+ */
+export declare function executeBashWithOperations(command: string, cwd: string, operations: BashOperations, options?: BashExecutorOptions): Promise<BashResult>;

package/dist/core/bash-executor.js

@@ -0,0 +1 @@
+var O=Object.defineProperty;var c=(a,l)=>O(a,"name",{value:l,configurable:!0});import{randomBytes as w}from"node:crypto";import{createWriteStream as C}from"node:fs";import{tmpdir as k}from"node:os";import{join as T}from"node:path";import{stripAnsi as F}from"../utils/ansi.js";import{sanitizeBinaryOutput as S}from"../utils/shell.js";import{DEFAULT_MAX_BYTES as p,truncateTail as h}from"./tools/truncate.js";async function _(a,l,g,u){const o=[];let s=0;const x=p*2;let i,n,f=0;const d=c(()=>{if(i)return;const r=w(8).toString("hex");i=T(k(),`koda-bash-${r}.log`),n=C(i);for(const t of o)n.write(t)},"ensureTempFile"),y=new TextDecoder,B=c(r=>{f+=r.length;const t=S(F(y.decode(r,{stream:!0}))).replace(/\r/g,"");for(f>p&&d(),n&&n.write(t),o.push(t),s+=t.length;s>x&&o.length>1;){const e=o.shift();s-=e.length}u?.onChunk&&u.onChunk(t)},"onData");try{const r=await g.exec(a,l,{onData:B,signal:u?.signal}),t=o.join(""),e=h(t);e.truncated&&d(),n&&n.end();const m=u?.signal?.aborted??!1;return{output:e.truncated?e.content:t,exitCode:m?void 0:r.exitCode??void 0,cancelled:m,truncated:e.truncated,fullOutputPath:i}}catch(r){if(u?.signal?.aborted){const t=o.join(""),e=h(t);return e.truncated&&d(),n&&n.end(),{output:e.truncated?e.content:t,exitCode:void 0,cancelled:!0,truncated:e.truncated,fullOutputPath:i}}throw n&&n.end(),r}}c(_,"executeBashWithOperations");export{_ as executeBashWithOperations};

package/dist/core/compaction/branch-summarization.d.ts

@@ -0,0 +1,89 @@
+/**
+ * Branch summarization for tree navigation.
+ *
+ * When navigating to a different point in the session tree, this generates
+ * a summary of the branch being left so context isn't lost.
+ */
+import type { AgentMessage, StreamFn } from "@openadapter/koda-agent-core";
+import type { Model } from "@openadapter/koda-ai";
+import type { ReadonlySessionManager, SessionEntry } from "../session-manager.ts";
+import { type FileOperations } from "./utils.ts";
+export interface BranchSummaryResult {
+    summary?: string;
+    readFiles?: string[];
+    modifiedFiles?: string[];
+    aborted?: boolean;
+    error?: string;
+}
+/** Details stored in BranchSummaryEntry.details for file tracking */
+export interface BranchSummaryDetails {
+    readFiles: string[];
+    modifiedFiles: string[];
+}
+export type { FileOperations } from "./utils.ts";
+export interface BranchPreparation {
+    /** Messages extracted for summarization, in chronological order */
+    messages: AgentMessage[];
+    /** File operations extracted from tool calls */
+    fileOps: FileOperations;
+    /** Total estimated tokens in messages */
+    totalTokens: number;
+}
+export interface CollectEntriesResult {
+    /** Entries to summarize, in chronological order */
+    entries: SessionEntry[];
+    /** Common ancestor between old and new position, if any */
+    commonAncestorId: string | null;
+}
+export interface GenerateBranchSummaryOptions {
+    /** Model to use for summarization */
+    model: Model<any>;
+    /** API key for the model */
+    apiKey: string;
+    /** Request headers for the model */
+    headers?: Record<string, string>;
+    /** Abort signal for cancellation */
+    signal: AbortSignal;
+    /** Optional custom instructions for summarization */
+    customInstructions?: string;
+    /** If true, customInstructions replaces the default prompt instead of being appended */
+    replaceInstructions?: boolean;
+    /** Tokens reserved for prompt + LLM response (default 16384) */
+    reserveTokens?: number;
+    /** Optional session stream function. Used to preserve SDK request behavior without mutating agent state. */
+    streamFn?: StreamFn;
+}
+/**
+ * Collect entries that should be summarized when navigating from one position to another.
+ *
+ * Walks from oldLeafId back to the common ancestor with targetId, collecting entries
+ * along the way. Does NOT stop at compaction boundaries - those are included and their
+ * summaries become context.
+ *
+ * @param session - Session manager (read-only access)
+ * @param oldLeafId - Current position (where we're navigating from)
+ * @param targetId - Target position (where we're navigating to)
+ * @returns Entries to summarize and the common ancestor
+ */
+export declare function collectEntriesForBranchSummary(session: ReadonlySessionManager, oldLeafId: string | null, targetId: string): CollectEntriesResult;
+/**
+ * Prepare entries for summarization with token budget.
+ *
+ * Walks entries from NEWEST to OLDEST, adding messages until we hit the token budget.
+ * This ensures we keep the most recent context when the branch is too long.
+ *
+ * Also collects file operations from:
+ * - Tool calls in assistant messages
+ * - Existing branch_summary entries' details (for cumulative tracking)
+ *
+ * @param entries - Entries in chronological order
+ * @param tokenBudget - Maximum tokens to include (0 = no limit)
+ */
+export declare function prepareBranchEntries(entries: SessionEntry[], tokenBudget?: number): BranchPreparation;
+/**
+ * Generate a summary of abandoned branch entries.
+ *
+ * @param entries - Session entries to summarize (chronological order)
+ * @param options - Generation options
+ */
+export declare function generateBranchSummary(entries: SessionEntry[], options: GenerateBranchSummaryOptions): Promise<BranchSummaryResult>;

package/dist/core/compaction/branch-summarization.js

@@ -0,0 +1,38 @@
+var S=Object.defineProperty;var m=(e,o)=>S(e,"name",{value:o,configurable:!0});import{completeSimple as _}from"@openadapter/koda-ai";import{convertToLlm as v,createBranchSummaryMessage as P,createCompactionSummaryMessage as B,createCustomMessage as R}from"../messages.js";import{estimateTokens as O}from"./compaction.js";import{computeFileLists as C,createFileOps as E,extractFileOpsFromMessage as I,formatFileOperations as N,SUMMARIZATION_SYSTEM_PROMPT as W,serializeConversation as z}from"./utils.js";function X(e,o,i){if(!o)return{entries:[],commonAncestorId:null};const c=new Set(e.getBranch(o).map(r=>r.id)),a=e.getBranch(i);let n=null;for(let r=a.length-1;r>=0;r--)if(c.has(a[r].id)){n=a[r].id;break}const t=[];let s=o;for(;s&&s!==n;){const r=e.getEntry(s);if(!r)break;t.push(r),s=r.parentId}return t.reverse(),{entries:t,commonAncestorId:n}}m(X,"collectEntriesForBranchSummary");function D(e){switch(e.type){case"message":return e.message.role==="toolResult"?void 0:e.message;case"custom_message":return R(e.customType,e.content,e.display,e.details,e.timestamp);case"branch_summary":return P(e.summary,e.fromId,e.timestamp);case"compaction":return B(e.summary,e.tokensBefore,e.timestamp);case"thinking_level_change":case"model_change":case"custom":case"label":case"session_info":return}}m(D,"getMessageFromEntry");function U(e,o=0){const i=[],c=E();let a=0;for(const n of e)if(n.type==="branch_summary"&&!n.fromHook&&n.details){const t=n.details;if(Array.isArray(t.readFiles))for(const s of t.readFiles)c.read.add(s);if(Array.isArray(t.modifiedFiles))for(const s of t.modifiedFiles)c.edited.add(s)}for(let n=e.length-1;n>=0;n--){const t=e[n],s=D(t);if(!s)continue;I(s,c);const r=O(s);if(o>0&&a+r>o){(t.type==="compaction"||t.type==="branch_summary")&&a<o*.9&&(i.unshift(s),a+=r);break}i.unshift(s),a+=r}return{messages:i,fileOps:c,totalTokens:a}}m(U,"prepareBranchEntries");const $=`The user explored a different conversation branch before returning here.
+Summary of that exploration:
+
+`,A=`Create a structured summary of this conversation branch for context when returning later.
+
+Use this EXACT format:
+
+## Goal
+[What was the user trying to accomplish in this branch?]
+
+## Constraints & Preferences
+- [Any constraints, preferences, or requirements mentioned]
+- [Or "(none)" if none were mentioned]
+
+## Progress
+### Done
+- [x] [Completed tasks/changes]
+
+### In Progress
+- [ ] [Work that was started but not finished]
+
+### Blocked
+- [Issues preventing progress, if any]
+
+## Key Decisions
+- **[Decision]**: [Brief rationale]
+
+## Next Steps
+1. [What should happen next to continue this work]
+
+Keep each section concise. Preserve exact file paths, function names, and error messages.`;async function Z(e,o){const{model:i,apiKey:c,headers:a,signal:n,customInstructions:t,replaceInstructions:s,reserveTokens:r=16384,streamFn:d}=o,w=(i.contextWindow||128e3)-r,{messages:h,fileOps:T}=U(e,w);if(h.length===0)return{summary:"No content to summarize"};const b=v(h),k=z(b);let u;s&&t?u=t:t?u=`${A}
+
+Additional focus: ${t}`:u=A;const F=[{role:"user",content:[{type:"text",text:`<conversation>
+${k}
+</conversation>
+
+${u}`}],timestamp:Date.now()}],g={systemPrompt:W,messages:F},y={apiKey:c,headers:a,signal:n,maxTokens:2048},l=d?await(await d(i,g,y)).result():await _(i,g,y);if(l.stopReason==="aborted")return{aborted:!0};if(l.stopReason==="error")return{error:l.errorMessage||"Summarization failed"};let f=l.content.filter(p=>p.type==="text").map(p=>p.text).join(`
+`);f=$+f;const{readFiles:x,modifiedFiles:M}=C(T);return f+=N(x,M),{summary:f||"No summary generated",readFiles:x,modifiedFiles:M}}m(Z,"generateBranchSummary");export{X as collectEntriesForBranchSummary,Z as generateBranchSummary,U as prepareBranchEntries};