agent-afk 2.26.3 → 2.26.4
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- package/dist/cli.mjs +4 -4
- package/dist/index.mjs +14 -14
- package/dist/telegram.mjs +69 -69
- package/package.json +1 -1
package/dist/index.mjs
CHANGED
|
@@ -927,7 +927,7 @@ Don't refuse the flow; just clarify where the wizard runs.
|
|
|
927
927
|
Be terse and operational. The user is doing one-time setup; they want it done, not narrated. Confirm each step in one line, don't over-explain. Use \`\u2713\` for success, \`\u2717\` for failure, and code fences for any command they should run.
|
|
928
928
|
`}};function $(t){let e=fr[t];if(!e){let n=Object.keys(fr).sort(),r=n.length>0?"Available: "+n.join(", "):"";throw new Error("Unknown skill: "+t+". "+r)}return e}var Ze=new Map;function X(t){Ze.set(t.name,t)}function he(t){let e=Ze.get(t);if(e)return e;let n=Array.from(Ze.keys()).sort(),r=n.length>0?`
|
|
929
929
|
Available skills: ${n.join(", ")}`:"";throw new Error(`Skill not found: ${t}${r}`)}function mr(){return Array.from(Ze.keys()).sort()}var Oe=class{nodes=new Map;traceWriter;constructor(e){this.traceWriter=e}register(e,n){this.nodes.has(e)||this.nodes.set(e,{controller:n,children:new Set,listeners:new Set,cascading:!1})}has(e){return this.nodes.has(e)}getController(e){return this.nodes.get(e)?.controller}linkChild(e,n){let r=this.nodes.get(e),o=this.nodes.get(n);if(!r)throw new Error(`AbortGraph: parent ${e} not registered`);if(!o)throw new Error(`AbortGraph: child ${n} not registered`);if(o.parentId=e,r.children.add(n),r.controller.signal.aborted){o.controller.signal.aborted||(o.cascading=!0,o.controller.abort(r.controller.signal.reason));return}r.controller.signal.addEventListener("abort",()=>{let s=this.nodes.get(n);!s||s.parentId!==e||s.controller.signal.aborted||(s.cascading=!0,s.controller.abort(r.controller.signal.reason))},{once:!0}),o.controller.signal.addEventListener("abort",()=>{let s=this.nodes.get(n);if(!s||s.parentId!==e||s.cascading)return;let i=this.nodes.get(e);if(!i)return;let a={parentId:e,childId:n,reason:s.controller.signal.reason};for(let c of i.listeners)try{c(a)}catch{}},{once:!0})}onChildAborted(e,n){let r=this.nodes.get(e);if(!r)throw new Error(`AbortGraph: ${e} not registered`);return r.listeners.add(n),()=>{r.listeners.delete(n)}}abort(e,n,r="user_signal"){let o=this.nodes.get(e);if(!o||o.controller.signal.aborted)return;let s=[],i=[...o.children],a=new Set;for(;i.length;){let c=i.shift();if(a.has(c))continue;a.add(c);let l=this.nodes.get(c);if(l){l.cascading=!0,s.push(c);for(let d of l.children)i.push(d)}}Un(this.traceWriter,{origin:r,cascadedTo:s,...n!==void 0?{reason:Si(n)}:{}}),o.controller.abort(n);for(let c of s){let l=this.nodes.get(c);l&&!l.controller.signal.aborted&&l.controller.abort(n)}}dispose(e){let n=this.nodes.get(e);if(n){n.parentId&&this.nodes.get(n.parentId)?.children.delete(e);for(let r of n.children){let o=this.nodes.get(r);o&&(o.parentId=void 0)}this.nodes.delete(e)}}};function Si(t){if(typeof t=="string")return t;if(t instanceof Error)return t.message;try{return JSON.stringify(t)}catch{return String(t)}}var Fe=0,et=5e3;async function $e(t,e,n={}){if(!Number.isFinite(e)||e<=0)return t;let r,o=new Promise((s,i)=>{r=setTimeout(()=>{let a=n.label?` (${n.label})`:"",c=new ie(`Operation timed out after ${e}ms${a}`,e);n.controller&&!n.controller.signal.aborted&&n.controller.abort(c),i(c)},e)});try{return await Promise.race([t,o])}finally{r!==void 0&&clearTimeout(r)}}async function ae(t,e,n,r){if(!t)return;if(r.kind==="blocked"){await Se(t,{hookEvent:e,decision:"block",...r.err.reason!==void 0?{reason:r.err.reason}:{},...e==="PreToolUse"&&n.toolName!==void 0?{blockedTool:n.toolName}:{}});return}let o=r.decision;await Se(t,{hookEvent:e,decision:o.decision,...o.reason!==void 0?{reason:o.reason}:{},...o.injectContext!==void 0?{injectedContextBytes:Buffer.byteLength(o.injectContext,"utf8")}:{}})}async function hr(t,e,n={}){if(t)try{let r=await t.dispatch(e,n.signal);await ae(n.traceWriter,"SubagentStart",{},{kind:"decision",decision:r})}catch(r){throw r instanceof U&&await ae(n.traceWriter,"SubagentStart",{},{kind:"blocked",err:r}),r}}async function gr(t,e,n={}){if(!t)return{};try{let r=await t.dispatch(e,n.signal);return await ae(n.traceWriter,"SubagentStop",{},{kind:"decision",decision:r}),r}catch(r){return r instanceof U&&await ae(n.traceWriter,"SubagentStop",{},{kind:"blocked",err:r}),r instanceof U||r instanceof Y?(_(`SubagentStop hook swallowed ${r.name}: ${r.message}`),n.onError?.(r),{}):(_(`SubagentStop hook unexpected error: ${String(r)}`),n.onError?.(r instanceof Error?r:new Error(String(r))),{})}}async function Wt(t,e,n={}){if(t)try{let r=await t.dispatch(e,n.signal);await ae(n.traceWriter,"PreToolUse",{toolName:e.toolName},{kind:"decision",decision:r})}catch(r){throw r instanceof U&&await ae(n.traceWriter,"PreToolUse",{toolName:e.toolName},{kind:"blocked",err:r}),r}}async function Gt(t,e,n={}){if(t)try{let r=await t.dispatch(e,n.signal);await ae(n.traceWriter,"PostToolUse",{toolName:e.toolName},{kind:"decision",decision:r})}catch(r){if(r instanceof U&&await ae(n.traceWriter,"PostToolUse",{toolName:e.toolName},{kind:"blocked",err:r}),r instanceof U||r instanceof Y){_(`PostToolUse hook swallowed ${r.name}: ${r.message}`),n.onError?.(r);return}_(`PostToolUse hook unexpected error: ${String(r)}`),n.onError?.(r instanceof Error?r:new Error(String(r)))}}import{mkdir as _i,writeFile as Ai}from"fs/promises";import{dirname as Ri,join as Ci}from"path";import{join as H,dirname as xi}from"path";import{homedir as Ti}from"os";import{fileURLToPath as Ei}from"url";function Q(){return process.env.AFK_HOME||H(Ti(),".afk")}function ge(){return H(Q(),"agent-framework")}function yr(){return H(ge(),"forge-telemetry.jsonl")}function xe(){return H(ge(),"briefs")}function tt(){return H(ge(),"ceiling-ledger")}function Kt(){return H(Q(),"skills")}function Te(){return H(Q(),"plugins")}function Pi(){return H(process.cwd(),".afk")}function qt(){return H(Pi(),"plugins")}function nt(){return H(Te(),".index.json")}function zt(){let t=Ei(import.meta.url),e=xi(t);return H(e,"bundled-plugins")}function Vt(){return H(Q(),"state")}function br(){return H(Vt(),"sessions")}function rt(){return H(Vt(),"memory")}function ot(){return H(Vt(),"session-grants.jsonl")}function wr(){return H(Q(),"farms")}function kr(t){return H(wr(),t)}function Ii(){return Ci(ge(),"routing-decisions.jsonl")}async function st(t){if(!(process.env.VITEST||process.env.NODE_ENV==="test"))try{let e=Ii();await _i(Ri(e),{recursive:!0});let r={ts:new Date().toISOString().split(".")[0]+"Z",surface:"afk"};for(let[s,i]of Object.entries(t))i!==void 0&&(r[s]=i);let o=JSON.stringify(r)+`
|
|
930
|
-
`;await Ai(e,o,{flag:"a"})}catch{}}import{AsyncLocalStorage as Mi}from"node:async_hooks";var vr=new Mi;function Sr(t,e){return vr.run(t,e)}function Ee(){return vr.getStore()}function Jt(t){let e=Di(t);return e!==void 0?e:Oi(t)}function Di(t){let e=/```(?:json)?\s*([\s\S]*?)```/gi,n,r;for(;(r=e.exec(t))!==null;)n=r[1];if(n)return xr(n.trim())}function Oi(t){for(let e=t.length-1;e>=0;e--){if(t[e]!=="}")continue;let n=Fi(t,e);if(n===-1)continue;let r=t.slice(n,e+1),o=xr(r);if(o!==void 0)return o}}function Fi(t,e){let n=0,r=!1,o=!1;for(let s=e;s>=0;s--){let i=t[s];if(o){o=!1;continue}if(r){if(i==="\\"){o=!0;continue}i==='"'&&(r=!1);continue}if(i==='"'){r=!0;continue}if(i==="}")n++;else if(i==="{"&&(n--,n===0))return s}return-1}function xr(t){try{return JSON.parse(t)}catch{return}}function Yt(){return{toolCalls:[],toolResults:[],thinkingPresent:!1,turnCount:0}}function Tr(t,e,n,r,o){if(!r)return{id:t,status:e,message:n,trace:o};let s=Jt(n.content),i=r.safeParse(s);return i.success?{id:t,status:e,message:n,output:i.data,trace:o}:{id:t,status:"failed",message:n,error:new Error(`structured output did not match schema: ${i.error.message}`,{cause:i.error}),schemaError:i.error,trace:o}}function Er(t,e,n,r){let o=n instanceof Error?n:new Error(String(n));return{id:t,status:e,error:o,trace:r}}function O(t){return`${t.status}${t.error?`: ${t.error.message}`:""}`}var it=class{constructor(e,n,r,o,s,i,a,c,l,d,u,p,g,b){this.id=e;this.session=n;this.controller=r;this.abortGraph=o;this.outputSchema=s;this.timeoutMs=i;this.hookRegistry=a;this.onTerminal=c;this.parentInputStreamRef=l;this.parentAbortSignal=d;this.agentType=u;this.traceWriter=b;this.progressSink=p,this.parentId=g}id;session;controller;abortGraph;outputSchema;timeoutMs;hookRegistry;onTerminal;parentInputStreamRef;parentAbortSignal;agentType;traceWriter;currentStatus="idle";inFlight=null;lastMessage;lastDurationMs;latestTerminalStatus;stopDispatched=!1;progressSink;parentId;currentTrace=Yt();lastStreamedContent="";get status(){return this.currentStatus}async run(e){if(this.currentStatus==="running")throw new Error(`Subagent ${this.id} is already running`);if(this.currentStatus==="cancelled")throw new Error(`Subagent ${this.id} is cancelled`);this.currentStatus="running";let n=Date.now(),r=$e(this.streamToFinalMessage(e),this.timeoutMs,{controller:this.controller,label:this.id});this.inFlight=r;try{let o=await r;return this.lastMessage=o.content,this.lastDurationMs=Date.now()-n,this.currentStatus="succeeded",this.latestTerminalStatus="succeeded",pe(this.traceWriter,{transition:"succeeded",subagentId:this.id,durationMs:this.lastDurationMs,turnCount:this.currentTrace.turnCount,outputBytes:Buffer.byteLength(this.lastMessage,"utf8")}),this.onTerminal(),o}catch(o){throw this.lastDurationMs=Date.now()-n,this.currentStatus!=="cancelled"&&(this.controller.signal.aborted?pe(this.traceWriter,{transition:"cancelled",subagentId:this.id,source:"cascade"}):pe(this.traceWriter,{transition:"failed",subagentId:this.id,errorClass:o instanceof Error?o.constructor.name:"Unknown",errorMessage:o instanceof Error?o.message:String(o),partialOutputBytes:Buffer.byteLength(this.lastStreamedContent,"utf8")}),this.currentStatus="failed",this.latestTerminalStatus="failed"),this.onTerminal(),o}finally{this.inFlight=null}}async streamToFinalMessage(e){let n,r;this.lastStreamedContent="",this.currentTrace=Yt();let o=this.progressSink??Ee(),s={subagentId:this.id,...this.parentId!==void 0&&{parentId:this.parentId},...this.agentType!==void 0&&{agentType:this.agentType}};for await(let i of this.session.sendMessageStream(e)){if(o&&o(i,s),i.type==="chunk"){let a=i.chunk;a.type==="content"?this.lastStreamedContent+=a.content:a.type==="tool_use_detail"?this.currentTrace.toolCalls.push({id:a.toolUseId,name:a.toolName,inputBytes:Buffer.byteLength(a.toolInput,"utf8")}):a.type==="tool_result"?this.currentTrace.toolResults.push({toolUseId:a.toolUseId,isError:a.isError,truncated:a.truncated,sizeBytes:a.sizeBytes}):a.type==="thinking"&&(this.currentTrace.thinkingPresent=!0)}if(i.type==="message")n=i.message,this.currentTrace.turnCount++;else if(i.type==="error"){r=i.error;break}else if(i.type==="done"){if(typeof i.metadata?.usage=="object"&&i.metadata.usage!==null){let a=i.metadata.usage;this.currentTrace.usage={inputTokens:typeof a.input_tokens=="number"?a.input_tokens:void 0,outputTokens:typeof a.output_tokens=="number"?a.output_tokens:void 0,cacheReadTokens:typeof a.cache_read_input_tokens=="number"?a.cache_read_input_tokens:void 0,cacheCreationTokens:typeof a.cache_creation_input_tokens=="number"?a.cache_creation_input_tokens:void 0}}break}}if(r)throw r;if(n)return n;if(this.lastStreamedContent.length>0)return{role:"assistant",content:this.lastStreamedContent,timestamp:new Date};throw new Error(`Subagent ${this.id} produced no terminal message`)}async runToResult(e){try{let n=await this.run(e);return Tr(this.id,this.currentStatus,n,this.outputSchema,this.currentTrace)}catch(n){let r=Er(this.id,this.currentStatus,n,this.currentTrace);return this.lastStreamedContent.length>0&&(r.partialOutput=this.lastStreamedContent),r}}runInBackground(e,n){this.runToResult(e).then(r=>{n?.(r)})}async cancel(){if(this.currentStatus==="cancelled"||this.stopDispatched)return;let e=this.latestTerminalStatus??"cancelled";this.currentStatus="cancelled",pe(this.traceWriter,{transition:"cancelled",subagentId:this.id,source:"explicit"});try{this.abortGraph.abort(this.id,"cancelled")}catch{}try{this.inFlight&&await this.session.interrupt()}catch{}try{await this.session.close()}finally{await this.dispatchStopAndRelease(e)}}async teardown(){if(this.stopDispatched)return;let e=this.latestTerminalStatus??"cancelled";try{this.inFlight&&await this.session.interrupt()}catch{}try{await this.session.close()}finally{await this.dispatchStopAndRelease(e)}}async dispatchStopAndRelease(e){if(this.stopDispatched){this.onTerminal();return}this.stopDispatched=!0;let n=await gr(this.hookRegistry,{event:"SubagentStop",subagentId:this.id,status:e,lastMessage:this.lastMessage,agentType:this.agentType,durationMs:this.lastDurationMs,trace:this.currentTrace},this.traceWriter?{traceWriter:this.traceWriter}:{});if(n.injectContext&&this.parentInputStreamRef)if(this.parentAbortSignal?.aborted)_(`Skipping SubagentStop injectContext for ${this.id}: parent is aborted`);else try{this.parentInputStreamRef.pushUserMessage(n.injectContext)}catch(r){_(`Failed to inject context from SubagentStop handler: ${String(r)}`)}this.onTerminal()}};var P=class{active=new Map;parentCanUseTool;hookRegistry;progressSink;parentApiKey;parentCwd;abortGraph;rootId;rootController;counter=0;constructor(e={}){if(this.parentCanUseTool=e.canUseTool,this.hookRegistry=e.hookRegistry,this.progressSink=e.progressSink,this.parentApiKey=e.apiKey,this.parentCwd=e.cwd,this.abortGraph=new Oe(e.traceWriter),this.rootId=`manager-root-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,this.rootController=new AbortController,this.abortGraph.register(this.rootId,this.rootController),e.parentAbortSignal){let n=e.parentAbortSignal;n.aborted?this.rootController.abort(n.reason):n.addEventListener("abort",()=>{this.rootController.signal.aborted||this.rootController.abort(n.reason)},{once:!0})}}list(){return[...this.active.values()].map(e=>({id:e.id,status:e.status}))}get(e){return this.active.get(e)}onChildAborted(e){return this.abortGraph.onChildAborted(this.rootId,e)}abortAll(e,n="user_signal"){this.abortGraph.abort(this.rootId,e,n)}async forkSubagent(e){let n=`${e.idPrefix??"subagent"}-${Date.now()}-${++this.counter}`,r=e.parent.sessionId,o=e.config.hookRegistry??this.hookRegistry;o&&await hr(o,{event:"SubagentStart",subagentId:n,parentSessionId:e.parent.sessionId},{signal:this.rootController.signal,...e.config.traceWriter?{traceWriter:e.config.traceWriter}:{}});let s=new AbortController;this.abortGraph.register(n,s),this.abortGraph.linkChild(this.rootId,n);let i={...e.config,resume:r,forkSession:r?!0:e.config.forkSession,abortSignal:s.signal,apiKey:e.config.apiKey||this.parentApiKey,...e.config.cwd===void 0&&this.parentCwd!==void 0?{cwd:this.parentCwd}:{},hookRegistry:e.config.hookRegistry??this.hookRegistry,permissionBubbler:e.config.permissionBubbler??(this.parentCanUseTool!==void 0&&e.config.canUseTool===void 0?{canUseTool:this.parentCanUseTool}:void 0)},a=new ye(i),c=e.parent.getInputStreamRef?.(),l=e.parent.abortSignal,d=this.progressSink??Ee(),u=e.agentType?.trim()||void 0,p=e.parentId?.trim()||void 0,g=new it(n,a,s,this.abortGraph,e.outputSchema,e.config.timeoutMs??Fe,o,()=>{this.active.delete(n),this.abortGraph.dispose(n)},c,l,u??e.idPrefix,d,p??e.parent.sessionId,e.config.traceWriter);this.active.set(n,g);let b=typeof e.config.model=="string"?e.config.model:JSON.stringify(e.config.model);return pe(e.config.traceWriter,{transition:"started",subagentId:n,parentId:e.parent.sessionId??this.rootId,model:b,...i.tools?.allowedTools?{allowedTools:[...i.tools.allowedTools]}:{}}),await st({event:"subagent.dispatched",subagent_id:n,id_prefix:e.idPrefix,parent_session_id:e.parent.sessionId}),g}async kill(e){let n=this.active.get(e);return n?(await n.cancel(),!0):!1}async killAll(){await Promise.allSettled([...this.active.values()].map(e=>e.cancel()))}async teardownAll(){await Promise.allSettled([...this.active.values()].map(e=>e.teardown()))}};async function at(t,e={}){let{failFast:n=!0,teardown:r=!0}=e;if(t.length===0)return[];let o=new Array(t.length),s=new Set(t.map((a,c)=>c)),i=t.map((a,c)=>a.handle.runToResult(a.prompt).then(l=>{if(o[c]=l,s.delete(c),n&&l.status!=="succeeded")for(let d of s){let u=t[d];u&&u.handle.status==="running"&&u.handle.cancel().catch(()=>{})}}));return await Promise.all(i),r&&await Promise.allSettled(t.map(a=>a.handle.teardown())),o}import{fileURLToPath as $i}from"node:url";import{dirname as Li}from"node:path";var Ni=$i(import.meta.url),
|
|
930
|
+
`;await Ai(e,o,{flag:"a"})}catch{}}import{AsyncLocalStorage as Mi}from"node:async_hooks";var vr=new Mi;function Sr(t,e){return vr.run(t,e)}function Ee(){return vr.getStore()}function Jt(t){let e=Di(t);return e!==void 0?e:Oi(t)}function Di(t){let e=/```(?:json)?\s*([\s\S]*?)```/gi,n,r;for(;(r=e.exec(t))!==null;)n=r[1];if(n)return xr(n.trim())}function Oi(t){for(let e=t.length-1;e>=0;e--){if(t[e]!=="}")continue;let n=Fi(t,e);if(n===-1)continue;let r=t.slice(n,e+1),o=xr(r);if(o!==void 0)return o}}function Fi(t,e){let n=0,r=!1,o=!1;for(let s=e;s>=0;s--){let i=t[s];if(o){o=!1;continue}if(r){if(i==="\\"){o=!0;continue}i==='"'&&(r=!1);continue}if(i==='"'){r=!0;continue}if(i==="}")n++;else if(i==="{"&&(n--,n===0))return s}return-1}function xr(t){try{return JSON.parse(t)}catch{return}}function Yt(){return{toolCalls:[],toolResults:[],thinkingPresent:!1,turnCount:0}}function Tr(t,e,n,r,o){if(!r)return{id:t,status:e,message:n,trace:o};let s=Jt(n.content),i=r.safeParse(s);return i.success?{id:t,status:e,message:n,output:i.data,trace:o}:{id:t,status:"failed",message:n,error:new Error(`structured output did not match schema: ${i.error.message}`,{cause:i.error}),schemaError:i.error,trace:o}}function Er(t,e,n,r){let o=n instanceof Error?n:new Error(String(n));return{id:t,status:e,error:o,trace:r}}function O(t){return`${t.status}${t.error?`: ${t.error.message}`:""}`}var it=class{constructor(e,n,r,o,s,i,a,c,l,d,u,p,g,b){this.id=e;this.session=n;this.controller=r;this.abortGraph=o;this.outputSchema=s;this.timeoutMs=i;this.hookRegistry=a;this.onTerminal=c;this.parentInputStreamRef=l;this.parentAbortSignal=d;this.agentType=u;this.traceWriter=b;this.progressSink=p,this.parentId=g}id;session;controller;abortGraph;outputSchema;timeoutMs;hookRegistry;onTerminal;parentInputStreamRef;parentAbortSignal;agentType;traceWriter;currentStatus="idle";inFlight=null;lastMessage;lastDurationMs;latestTerminalStatus;stopDispatched=!1;progressSink;parentId;currentTrace=Yt();lastStreamedContent="";get status(){return this.currentStatus}async run(e){if(this.currentStatus==="running")throw new Error(`Subagent ${this.id} is already running`);if(this.currentStatus==="cancelled")throw new Error(`Subagent ${this.id} is cancelled`);this.currentStatus="running";let n=Date.now(),r=$e(this.streamToFinalMessage(e),this.timeoutMs,{controller:this.controller,label:this.id});this.inFlight=r;try{let o=await r;return this.lastMessage=o.content,this.lastDurationMs=Date.now()-n,this.currentStatus="succeeded",this.latestTerminalStatus="succeeded",pe(this.traceWriter,{transition:"succeeded",subagentId:this.id,durationMs:this.lastDurationMs,turnCount:this.currentTrace.turnCount,outputBytes:Buffer.byteLength(this.lastMessage,"utf8")}),this.onTerminal(),o}catch(o){throw this.lastDurationMs=Date.now()-n,this.currentStatus!=="cancelled"&&(this.controller.signal.aborted?pe(this.traceWriter,{transition:"cancelled",subagentId:this.id,source:"cascade"}):pe(this.traceWriter,{transition:"failed",subagentId:this.id,errorClass:o instanceof Error?o.constructor.name:"Unknown",errorMessage:o instanceof Error?o.message:String(o),partialOutputBytes:Buffer.byteLength(this.lastStreamedContent,"utf8")}),this.currentStatus="failed",this.latestTerminalStatus="failed"),this.onTerminal(),o}finally{this.inFlight=null}}async streamToFinalMessage(e){let n,r;this.lastStreamedContent="",this.currentTrace=Yt();let o=this.progressSink??Ee(),s={subagentId:this.id,...this.parentId!==void 0&&{parentId:this.parentId},...this.agentType!==void 0&&{agentType:this.agentType}};for await(let i of this.session.sendMessageStream(e)){if(o&&o(i,s),i.type==="chunk"){let a=i.chunk;a.type==="content"?this.lastStreamedContent+=a.content:a.type==="tool_use_detail"?this.currentTrace.toolCalls.push({id:a.toolUseId,name:a.toolName,inputBytes:Buffer.byteLength(a.toolInput,"utf8")}):a.type==="tool_result"?this.currentTrace.toolResults.push({toolUseId:a.toolUseId,isError:a.isError,truncated:a.truncated,sizeBytes:a.sizeBytes}):a.type==="thinking"&&(this.currentTrace.thinkingPresent=!0)}if(i.type==="message")n=i.message,this.currentTrace.turnCount++;else if(i.type==="error"){r=i.error;break}else if(i.type==="done"){if(typeof i.metadata?.usage=="object"&&i.metadata.usage!==null){let a=i.metadata.usage;this.currentTrace.usage={inputTokens:typeof a.input_tokens=="number"?a.input_tokens:void 0,outputTokens:typeof a.output_tokens=="number"?a.output_tokens:void 0,cacheReadTokens:typeof a.cache_read_input_tokens=="number"?a.cache_read_input_tokens:void 0,cacheCreationTokens:typeof a.cache_creation_input_tokens=="number"?a.cache_creation_input_tokens:void 0}}break}}if(r)throw r;if(n)return n;if(this.lastStreamedContent.length>0)return{role:"assistant",content:this.lastStreamedContent,timestamp:new Date};throw new Error(`Subagent ${this.id} produced no terminal message`)}async runToResult(e){try{let n=await this.run(e);return Tr(this.id,this.currentStatus,n,this.outputSchema,this.currentTrace)}catch(n){let r=Er(this.id,this.currentStatus,n,this.currentTrace);return this.lastStreamedContent.length>0&&(r.partialOutput=this.lastStreamedContent),r}}runInBackground(e,n){this.runToResult(e).then(r=>{n?.(r)})}async cancel(){if(this.currentStatus==="cancelled"||this.stopDispatched)return;let e=this.latestTerminalStatus??"cancelled";this.currentStatus="cancelled",pe(this.traceWriter,{transition:"cancelled",subagentId:this.id,source:"explicit"});try{this.abortGraph.abort(this.id,"cancelled")}catch{}try{this.inFlight&&await this.session.interrupt()}catch{}try{await this.session.close()}finally{await this.dispatchStopAndRelease(e)}}async teardown(){if(this.stopDispatched)return;let e=this.latestTerminalStatus??"cancelled";try{this.inFlight&&await this.session.interrupt()}catch{}try{await this.session.close()}finally{await this.dispatchStopAndRelease(e)}}async dispatchStopAndRelease(e){if(this.stopDispatched){this.onTerminal();return}this.stopDispatched=!0;let n=await gr(this.hookRegistry,{event:"SubagentStop",subagentId:this.id,status:e,lastMessage:this.lastMessage,agentType:this.agentType,durationMs:this.lastDurationMs,trace:this.currentTrace},this.traceWriter?{traceWriter:this.traceWriter}:{});if(n.injectContext&&this.parentInputStreamRef)if(this.parentAbortSignal?.aborted)_(`Skipping SubagentStop injectContext for ${this.id}: parent is aborted`);else try{this.parentInputStreamRef.pushUserMessage(n.injectContext)}catch(r){_(`Failed to inject context from SubagentStop handler: ${String(r)}`)}this.onTerminal()}};var P=class{active=new Map;parentCanUseTool;hookRegistry;progressSink;parentApiKey;parentCwd;abortGraph;rootId;rootController;counter=0;constructor(e={}){if(this.parentCanUseTool=e.canUseTool,this.hookRegistry=e.hookRegistry,this.progressSink=e.progressSink,this.parentApiKey=e.apiKey,this.parentCwd=e.cwd,this.abortGraph=new Oe(e.traceWriter),this.rootId=`manager-root-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,this.rootController=new AbortController,this.abortGraph.register(this.rootId,this.rootController),e.parentAbortSignal){let n=e.parentAbortSignal;n.aborted?this.rootController.abort(n.reason):n.addEventListener("abort",()=>{this.rootController.signal.aborted||this.rootController.abort(n.reason)},{once:!0})}}list(){return[...this.active.values()].map(e=>({id:e.id,status:e.status}))}get(e){return this.active.get(e)}onChildAborted(e){return this.abortGraph.onChildAborted(this.rootId,e)}abortAll(e,n="user_signal"){this.abortGraph.abort(this.rootId,e,n)}async forkSubagent(e){let n=`${e.idPrefix??"subagent"}-${Date.now()}-${++this.counter}`,r=e.parent.sessionId,o=e.config.hookRegistry??this.hookRegistry;o&&await hr(o,{event:"SubagentStart",subagentId:n,parentSessionId:e.parent.sessionId},{signal:this.rootController.signal,...e.config.traceWriter?{traceWriter:e.config.traceWriter}:{}});let s=new AbortController;this.abortGraph.register(n,s),this.abortGraph.linkChild(this.rootId,n);let i={...e.config,resume:r,forkSession:r?!0:e.config.forkSession,abortSignal:s.signal,apiKey:e.config.apiKey||this.parentApiKey,...e.config.cwd===void 0&&this.parentCwd!==void 0?{cwd:this.parentCwd}:{},hookRegistry:e.config.hookRegistry??this.hookRegistry,permissionBubbler:e.config.permissionBubbler??(this.parentCanUseTool!==void 0&&e.config.canUseTool===void 0?{canUseTool:this.parentCanUseTool}:void 0)},a=new ye(i),c=e.parent.getInputStreamRef?.(),l=e.parent.abortSignal,d=this.progressSink??Ee(),u=e.agentType?.trim()||void 0,p=e.parentId?.trim()||void 0,g=new it(n,a,s,this.abortGraph,e.outputSchema,e.config.timeoutMs??Fe,o,()=>{this.active.delete(n),this.abortGraph.dispose(n)},c,l,u??e.idPrefix,d,p??e.parent.sessionId,e.config.traceWriter);this.active.set(n,g);let b=typeof e.config.model=="string"?e.config.model:JSON.stringify(e.config.model);return pe(e.config.traceWriter,{transition:"started",subagentId:n,parentId:e.parent.sessionId??this.rootId,model:b,...i.tools?.allowedTools?{allowedTools:[...i.tools.allowedTools]}:{}}),await st({event:"subagent.dispatched",subagent_id:n,id_prefix:e.idPrefix,parent_session_id:e.parent.sessionId}),g}async kill(e){let n=this.active.get(e);return n?(await n.cancel(),!0):!1}async killAll(){await Promise.allSettled([...this.active.values()].map(e=>e.cancel()))}async teardownAll(){await Promise.allSettled([...this.active.values()].map(e=>e.teardown()))}};async function at(t,e={}){let{failFast:n=!0,teardown:r=!0}=e;if(t.length===0)return[];let o=new Array(t.length),s=new Set(t.map((a,c)=>c)),i=t.map((a,c)=>a.handle.runToResult(a.prompt).then(l=>{if(o[c]=l,s.delete(c),n&&l.status!=="succeeded")for(let d of s){let u=t[d];u&&u.handle.status==="running"&&u.handle.cancel().catch(()=>{})}}));return await Promise.all(i),r&&await Promise.allSettled(t.map(a=>a.handle.teardown())),o}import{fileURLToPath as $i}from"node:url";import{dirname as Li}from"node:path";var Ni=$i(import.meta.url),Ip=Li(Ni),q={name:"research-agent",systemPrompt:`---
|
|
931
931
|
name: research-agent
|
|
932
932
|
description: Read-only sub-agent for research, validation, verification, and codebase inspection. Mechanically locked to Read, Grep, Glob, WebFetch, WebSearch \u2014 cannot Edit, Write, Bash, commit, or push. Delegates git queries to \`git-investigator\`. Use when the dispatched task is findings-only.
|
|
933
933
|
model: sonnet
|
|
@@ -980,7 +980,7 @@ Unless the dispatcher specifies a different schema, return:
|
|
|
980
980
|
**\`boundary_flag\` is required.** If nothing applies, emit \`"none"\` \u2014 do not omit the field. Treat missing as \`"none"\` is acceptable on the orchestrator side, but emit the field explicitly so downstream synthesizers and validators do not see \`null\`.
|
|
981
981
|
|
|
982
982
|
If \`scope_check\` flags implementation (non-git), the orchestrator should dispatch a different sub-agent type for follow-up. Do not re-dispatch the same task through \`research-agent\`.
|
|
983
|
-
`,sourcePath:"agent-framework-private/agents/research-agent.md",allowedTools:["Read","Grep","Glob","WebFetch","WebSearch"],description:"Read-only sub-agent for research, validation, verification, and codebase inspection. Mechanically locked to Read, Grep, Glob, WebFetch, WebSearch \u2014 cannot Edit, Write, Bash, commit, or push. Delegates git queries to `git-investigator`. Use when the dispatched task is findings-only."};import{existsSync as ce,readdirSync as qi,readFileSync as zi}from"fs";import{join as ne}from"path";import{existsSync as Xt,readFileSync as Hi,readdirSync as Bi,statSync as Wi}from"fs";import{join as Ne,resolve as _r}from"path";import{existsSync as Ui,mkdirSync as
|
|
983
|
+
`,sourcePath:"agent-framework-private/agents/research-agent.md",allowedTools:["Read","Grep","Glob","WebFetch","WebSearch"],description:"Read-only sub-agent for research, validation, verification, and codebase inspection. Mechanically locked to Read, Grep, Glob, WebFetch, WebSearch \u2014 cannot Edit, Write, Bash, commit, or push. Delegates git queries to `git-investigator`. Use when the dispatched task is findings-only."};import{existsSync as ce,readdirSync as qi,readFileSync as zi}from"fs";import{join as ne}from"path";import{existsSync as Xt,readFileSync as Hi,readdirSync as Bi,statSync as Wi}from"fs";import{join as Ne,resolve as _r}from"path";import{existsSync as Ui,mkdirSync as Op,readFileSync as ji,renameSync as Fp,writeFileSync as $p,unlinkSync as Lp}from"fs";function Pr(t=nt()){if(!Ui(t))return ct();try{let e=ji(t,"utf8"),n=JSON.parse(e);if(!n||typeof n!="object")return ct();let r=n,o=r.plugins&&typeof r.plugins=="object"?r.plugins:{};if(r.version===1)return{version:2,plugins:o,marketplaces:{}};if(r.version===2){let s=r.marketplaces&&typeof r.marketplaces=="object"?r.marketplaces:{};return{version:2,plugins:o,marketplaces:s}}return ct()}catch{return ct()}}function ct(){return{version:2,plugins:{},marketplaces:{}}}var Gi=5,Ar="cache",Le;function te(t=Te()){Le||(Le=new Map);let e=Le.get(t);if(e)return[...e];if(!Xt(t))return Le.set(t,[]),[];let n=t===Te()?nt():Ne(t,".index.json"),r=Pr(n),o=[];return Rr(t,t,0,o,new Set,r.plugins),Le.set(t,o),[...o]}function Rr(t,e,n,r,o,s){if(n>Gi||o.has(e))return;if(o.add(e),Xt(Ne(e,".claude-plugin","plugin.json"))){let a=Qt(t,e);if(a===null){r.push({type:"local",path:e});return}if(a.layout==="cache"){let l=s[a.key];if(!l||l.enabled===!1)return;r.push({type:"local",path:e});return}let c=s[a.key];if(c&&c.enabled===!1)return;r.push({type:"local",path:e});return}let i;try{i=Bi(e)}catch{return}for(let a of i){if(a.startsWith("."))continue;let c=Ne(e,a),l;try{l=Wi(c)}catch{continue}l.isDirectory()&&Rr(t,c,n+1,r,o,s)}}function Qt(t,e){if(!e.startsWith(t))return null;let n=e.slice(t.length).replace(/^\/+/,"");if(!n)return null;let r=n.split("/").filter(s=>s.length>0);if(r.length===0)return null;if(r[0]===Ar&&r.length>=3){let s=r[1];if(s){let i=Ne(t,Ar,s),c=Ki(i,e)??r[2];if(c)return{layout:"cache",key:`${s}:${c}`}}}let o=r[0];return o?{layout:"flat",key:o}:null}function Ki(t,e){let n=Ne(t,".claude-plugin","marketplace.json");if(!Xt(n))return null;let r;try{r=JSON.parse(Hi(n,"utf8"))}catch{return null}if(!r||typeof r!="object")return null;let o=r.plugins;if(!Array.isArray(o))return null;let s=_r(e);for(let i of o){if(!i||typeof i!="object")continue;let a=i;if(!(typeof a.name!="string"||typeof a.source!="string")&&!(!a.source.startsWith("./")&&!a.source.startsWith("../"))&&_r(t,a.source)===s)return a.name}return null}var Cr=["command","agent"];function Ir(t=Q()){let e=[],n=ne(t,"skills");if(ce(n))for(let r of lt(n)){let o=ne(n,r,"SKILL.md");ce(o)&&e.push({path:o,type:"skill",source:"user"})}for(let r of Cr){let o=ne(t,`${r}s`);if(ce(o))for(let s of lt(o))s.endsWith(".md")&&e.push({path:ne(o,s),type:r,source:"user"})}return e}function Mr(t=Te()){if(!ce(t))return[];let e=[],n=te(t);for(let r of n){let s=Qt(t,r.path)?.key,i=ne(r.path,"skills");if(ce(i))for(let a of lt(i)){let c=ne(i,a,"SKILL.md");if(!ce(c))continue;let l={path:c,type:"skill",source:"plugin"};s&&(l.plugin_key=s),e.push(l)}for(let a of Cr){let c=ne(r.path,`${a}s`);if(ce(c))for(let l of lt(c)){if(!l.endsWith(".md"))continue;let d={path:ne(c,l),type:a,source:"plugin"};s&&(d.plugin_key=s),e.push(d)}}}return e}function Dr(t=ne(Q(),"settings.json")){if(!ce(t))return[];try{let e=zi(t,"utf8"),r=JSON.parse(e).hooks;if(!r||typeof r!="object")return[];let o=[];for(let[s,i]of Object.entries(r))if(Array.isArray(i))for(let a=0;a<i.length;a++)o.push({event:s,index:a,raw:i[a]});return o}catch{return[]}}function lt(t){try{return qi(t).filter(e=>!e.startsWith("."))}catch{return[]}}var Lr=D.object({path:D.string(),type:D.enum(["skill","command","agent","hook"]),source:D.enum(["user","plugin"]),plugin_key:D.string().optional(),verdict:D.enum(["correct","misfit","outlier"]),recommended_type:D.string(),rationale:D.string(),confidence:D.enum(["high","med","low"])}),$r=D.record(D.string(),D.record(D.string(),D.number())),af=D.object({inventory:D.object({user:$r,plugin:$r}),misfits:D.array(Lr),briefs_written:D.number(),total_artifacts:D.number()}),Vi=D.object({writeBriefs:D.boolean().optional(),scope:D.enum(["user","plugin","all"]).optional()}),Ji=["skill","command","agent"],Nr=["skill","command","agent","hook"];function Yi(t){return{runUserDiscovery:t!=="plugin",runPluginDiscovery:t!=="user",runHookInspector:t!=="plugin"}}function Xi(t){let e=()=>{let s={};for(let i of Nr)s[i]={correct:0,misfit:0,outlier:0};return s},n={user:e(),plugin:e()};for(let s of t)n[s.source][s.type][s.verdict]+=1;let r={high:0,med:1,low:2},o=t.filter(s=>s.verdict==="misfit").slice().sort((s,i)=>r[s.confidence]-r[i.confidence]);return{inventory:n,misfits:o}}function Qi(t){return t.verdict==="misfit"&&t.confidence==="high"&&t.source==="user"}function Zi(t){let e=t.filter(o=>o.source==="user"),n=t.filter(o=>o.source==="plugin"),r=["","## Discovered artifacts (audit only these)",""];if(r.push('### User-scope artifacts (set `"source": "user"`, omit `plugin_key`)'),e.length===0)r.push("(none discovered)");else for(let o of e)r.push(`- ${o.path}`);if(r.push(""),r.push('### Plugin-scope artifacts (set `"source": "plugin"`, copy `plugin_key` from each entry)'),n.length===0)r.push("(none discovered)");else for(let o of n){let s=o.plugin_key??"<unknown>";r.push(`- ${o.path} (plugin_key: ${s})`)}return r.join(`
|
|
984
984
|
`)}function ea(t,e){let n=["","## Discovered hooks (audit only these)",""];if(n.push(`Settings file (use this absolute path verbatim in each verdict's \`path\` field): \`${t}\``),n.push(""),e.length===0)return n.push("(no hooks discovered)"),n.join(`
|
|
985
985
|
`);for(let r of e){let o=`${r.event}-${r.index}`;n.push(`### Hook \`${o}\``),n.push(""),n.push("```json"),n.push(JSON.stringify(r.raw,null,2)),n.push("```"),n.push("")}return n.join(`
|
|
986
986
|
`)}function ta(t,e){if(!e)return{kind:"failure",message:`${t}: no result`};if(e.schemaError)return{kind:"failure",message:`${t}: schema mismatch \u2014 ${e.schemaError.message}`};if(e.status!=="succeeded"){let n=e.error?` \u2014 ${e.error.message}`:"";return{kind:"failure",message:`${t}: ${e.status}${n}`}}return e.output?{kind:"success",output:e.output}:{kind:"failure",message:`${t}: no output`}}async function na(t,e,n){let r=n?.apiKey,o=typeof t=="object"&&t!==null?t:{},s=Vi.parse(o),i=s.writeBriefs??!0,a=s.scope??"all",c=Yi(a);if(!e?.sessionId)throw new Error("audit-fit requires a parent session with sessionId");let l=e.sessionId,d=$("audit-fit"),u={skill:d["01-skill-inspector.md"],command:d["02-command-inspector.md"],agent:d["03-agent-inspector.md"],hook:d["04-hook-inspector.md"]};for(let v of Nr)if(!u[v])throw new Error(`audit-fit skill missing inspector prompt for ${v}`);let p=c.runUserDiscovery?Ir():[],g=c.runPluginDiscovery?Mr():[],b={skill:[],command:[],agent:[]};for(let v of[...p,...g])b[v.type].push(v);let m=new P({apiKey:r}),S=()=>async v=>q.allowedTools.includes(v)?{behavior:"allow"}:{behavior:"deny",message:`Tool ${v} not allowed for audit-fit inspectors. Allowed tools: ${q.allowedTools.join(", ")}`},f=[];for(let v of Ji){let E=b[v];if(E.length===0)continue;let C=u[v];C&&f.push({type:v,prompt:`${C}
|
|
@@ -1012,7 +1012,7 @@ ${E.rationale}
|
|
|
1012
1012
|
---
|
|
1013
1013
|
Generated by audit-fit on ${new Date().toISOString().split(".")[0]}Z
|
|
1014
1014
|
`;await Fr(R,K),A++}}let F=ge();await Or(F,{recursive:!0});let x=v=>{let E=0;for(let C of Object.values(v))for(let R of Object.values(C))E+=R;return E},T=v=>{let E=y.user[v]??{},C=y.plugin[v]??{},R=K=>Object.values(K).reduce((L,J)=>L+J,0);return R(E)+R(C)},M={timestamp:new Date().toISOString(),surface:"afk",scope:a,total_artifacts:h.length,misfits_count:k.length,briefs_written:A,by_source:{user:x(y.user),plugin:x(y.plugin)},by_type:{skill:T("skill"),command:T("command"),agent:T("agent"),hook:T("hook")}},j=Zt(F,"audit-fit-telemetry.jsonl");return await Fr(j,JSON.stringify(M)+`
|
|
1015
|
-
`),{inventory:y,misfits:k,briefs_written:A,total_artifacts:h.length}}var ra={name:"audit-fit",description:"Audit ~/.afk artifacts (skills, commands, agents, hooks) for correct type categorization. Walks user-scope dirs (~/.afk/{skills,commands,agents}/) and every plugin installed under ~/.afk/plugins/ (flat and marketplace-cache layouts), plus ~/.afk/settings.json for hooks. Dispatches per-type inspectors in parallel, applies decision heuristics (progressive-disclosure value, isolation need, deterministic vs. reasoning), flags misfits. Generates migration briefs only for user-scope misfits (plugin misfits are inventory-only \u2014 refactoring vendored plugin code is the maintainer's job). Optional `scope` input filters to `user`, `plugin`, or `all` (default). Use for inventory audits after bulk authoring, imports, or periodic hygiene.",handler:na,argumentHint:"[--write-briefs]",whenToUse:"When the user wants ~/.afk artifacts (skills, commands, agents, hooks) audited for correct type categorization.",flags:["--write-briefs"]};X(ra);import{z as w}from"zod";import{execFile as aa}from"node:child_process";import{promisify as ca}from"node:util";import{tmpdir as la}from"node:os";import{join as jr}from"node:path";function Ur(t){return t.confidence<.5?{verify:!0,reason:`low confidence (${t.confidence.toFixed(2)} < ${.5})`}:t.boundary_flag&&t.boundary_flag.length>0?{verify:!0,reason:`boundary flag set: ${t.boundary_flag}`}:t.coverage_gaps&&t.coverage_gaps.length>0?{verify:!0,reason:`coverage gap${t.coverage_gaps.length===1?"":"s"}: ${t.coverage_gaps.length} unresolved`}:{verify:!1,reason:`confidence ${t.confidence.toFixed(2)} with no gaps or boundary`}}import{fileURLToPath as oa}from"node:url";import{dirname as sa}from"node:path";var ia=oa(import.meta.url),
|
|
1015
|
+
`),{inventory:y,misfits:k,briefs_written:A,total_artifacts:h.length}}var ra={name:"audit-fit",description:"Audit ~/.afk artifacts (skills, commands, agents, hooks) for correct type categorization. Walks user-scope dirs (~/.afk/{skills,commands,agents}/) and every plugin installed under ~/.afk/plugins/ (flat and marketplace-cache layouts), plus ~/.afk/settings.json for hooks. Dispatches per-type inspectors in parallel, applies decision heuristics (progressive-disclosure value, isolation need, deterministic vs. reasoning), flags misfits. Generates migration briefs only for user-scope misfits (plugin misfits are inventory-only \u2014 refactoring vendored plugin code is the maintainer's job). Optional `scope` input filters to `user`, `plugin`, or `all` (default). Use for inventory audits after bulk authoring, imports, or periodic hygiene.",handler:na,argumentHint:"[--write-briefs]",whenToUse:"When the user wants ~/.afk artifacts (skills, commands, agents, hooks) audited for correct type categorization.",flags:["--write-briefs"]};X(ra);import{z as w}from"zod";import{execFile as aa}from"node:child_process";import{promisify as ca}from"node:util";import{tmpdir as la}from"node:os";import{join as jr}from"node:path";function Ur(t){return t.confidence<.5?{verify:!0,reason:`low confidence (${t.confidence.toFixed(2)} < ${.5})`}:t.boundary_flag&&t.boundary_flag.length>0?{verify:!0,reason:`boundary flag set: ${t.boundary_flag}`}:t.coverage_gaps&&t.coverage_gaps.length>0?{verify:!0,reason:`coverage gap${t.coverage_gaps.length===1?"":"s"}: ${t.coverage_gaps.length} unresolved`}:{verify:!1,reason:`confidence ${t.confidence.toFixed(2)} with no gaps or boundary`}}import{fileURLToPath as oa}from"node:url";import{dirname as sa}from"node:path";var ia=oa(import.meta.url),ff=sa(ia),en={name:"git-investigator",systemPrompt:'---\nname: git-investigator\ndescription: Read-only git specialist. Dispatched by research-agent (or any research-shaped caller) when a finding requires git history, reflog, diff, blame, branch/remote state, or merge-base analysis. Runs git commands only \u2014 no mutations, no shell escapes.\nmodel: sonnet\ntools: Bash, Read, Grep, Glob\n---\n\nYou are `git-investigator`, a leaf sub-agent specialized for read-only git queries.\n\nYou have Bash, Read, Grep, and Glob. You do not dispatch other sub-agents. You do not Edit or Write. Your Bash surface is restricted **by this prompt** to `git ...` invocations and benign output-shaping pipes.\n\n## Allowed commands\n\nRead-only git only:\n\n- `git status`, `git log`, `git diff`, `git show`\n- `git rev-parse`, `git rev-list`, `git reflog`\n- `git branch -v / -vv / -a` (list only)\n- `git remote -v`, `git ls-remote`\n- `git ls-files`, `git blame`\n- `git merge-base`, `git for-each-ref`, `git describe`\n- `git cat-file`, `git shortlog`\n- `git tag` (list/show only)\n- `git stash list`, `git stash show`\n- `git config --get`, `git config --get-all`, `git config --list`\n- `git worktree list` (read only)\n\nOutput-shaping pipes are fine: `| head`, `| tail`, `| wc`, `| grep`, `| jq`, `| awk \'NR==...\'` (for formatting only \u2014 no mutations).\n\n## Forbidden\n\nAnything that mutates repo or working tree state:\n\n- `commit`, `push`, `pull`, `fetch --prune`\n- `reset`, `revert`, `rebase`, `merge`, `cherry-pick`\n- `checkout` (except `checkout -- <path>` file-restore, and even that is mutation \u2014 avoid it, just report the need)\n- `restore`, `switch`\n- `branch -d / -D / -m / -M`, `branch <new>`\n- `stash push / pop / drop / apply / clear`\n- `tag -d`, creating a new tag\n- `remote add / remove / set-url`\n- `config --set`, `config --unset`\n- `gc`, `fsck`, `prune`, `reflog delete`, `reflog expire`\n- `filter-branch`, `filter-repo`\n- `worktree add / remove / move`\n- `hooks install`, `submodule add / update`\n- Any non-`git` command that mutates: `rm`, `mv`, `cp` (writes), `sed -i`, `> file`, `>> file`, `tee`, `curl`, `wget`, `pip install`, shell builtins that change state.\n\nIf the caller asks for any of the above, do not run it. Return `scope_check: "requires mutation: <reason>"` and stop.\n\n## Behavior\n\n- Run the minimum set of commands needed. Prefer `git log -n 5 --oneline -- <path>` over `git log -- <path>` when a count is fine.\n- Cite concrete evidence: commit SHAs (short form OK), ref names, `path:line` references from blame, diff hunks trimmed to the relevant range.\n- Use `Read`/`Grep`/`Glob` for follow-up inspection of files the git output identifies (e.g., `git show SHA:path | head` then `Read` the current file to diff mentally).\n- Do not speculate beyond what the commands show. If a question needs history the commands don\'t surface (deleted-file recovery, ancient reflog that has expired), say so in `caveats`.\n- Keep output compact \u2014 dispatchers merge your findings into a larger response. No preamble, no ceremony.\n\n## Return shape\n\n```\n{\n "findings": "<summary of what the git data shows>",\n "evidence": ["<SHA>", "<ref>", "<path:line>", ...],\n "git_commands_run": ["git log ...", "git diff ...", ...],\n "caveats": "<gaps, ambiguity, or \'none\'>",\n "scope_check": "pure git research" | "requires mutation: <reason>"\n}\n```\n\nBegin your response with the first schema field. No preamble.\n',sourcePath:"agent-framework-private/agents/git-investigator.md",allowedTools:["Bash","Read","Grep","Glob"],description:"Read-only git specialist. Dispatched by research-agent (or any research-shaped caller) when a finding requires git history, reflog, diff, blame, branch/remote state, or merge-base analysis. Runs git commands only \u2014 no mutations, no shell escapes.",model:"sonnet"};function tn(t){let e={description:t.description,prompt:t.systemPrompt};return t.allowedTools&&(e.tools=[...t.allowedTools]),t.model&&(e.model=t.model),e}var dt=ca(aa),Wr=w.object({id:w.string(),claim:w.string(),confidence:w.number().min(0).max(1),evidence_sources:w.array(w.string()),location:w.string().optional(),proposed_fix:w.string().optional(),coverage_gaps:w.array(w.string()).optional(),boundary_flag:w.string().optional()}),da=w.object({hypothesis_id:w.string(),claim:w.string(),verdict:w.enum(["VERIFIED","REFUTED","INCONCLUSIVE"]),evidence:w.string(),gate_reason:w.string()}),Gr=w.object({hypothesis_id:w.string(),reproducer_passed:w.boolean(),regressions:w.array(w.string()),confidence:w.number().min(0).max(1),verification_log:w.string()}),ua=w.enum(["crash","regression","logic-error","flaky","environment","unknown"]),pa=w.object({failure_type:ua,error_signature:w.string(),affected_area:w.string()}),fa=w.enum(["clear_winner","multiple_plausible","dissent","all_inconclusive","no_hypotheses"]),Cf=w.object({reproducer:w.string().optional(),triage:pa.optional(),hypotheses:w.array(Wr),premise_verifications:w.array(da).optional(),winner:w.object({hypothesis_id:w.string(),verification_log:w.string(),proposed_fix:w.string()}).optional(),verification_results:w.array(Gr).optional(),outcome:fa.optional(),recommended_next_skill:w.enum(["spec"]).optional()});async function ma(t,e){let n=t.map(c=>({hypothesis:c,decision:Ur(c)})).filter(c=>c.decision.verify);if(n.length===0)return{premise_verifications:[],hypotheses_to_test:t};let r=[],o;try{r=await e(n.map(c=>c.hypothesis.claim))}catch(c){o=c instanceof Error?c.message:String(c)}let s=n.map((c,l)=>{let d=r[l];return o!==void 0?{hypothesis_id:c.hypothesis.id,claim:c.hypothesis.claim,verdict:"INCONCLUSIVE",evidence:`shadow-verify dispatch failed: ${o}`,gate_reason:c.decision.reason}:d?{hypothesis_id:c.hypothesis.id,claim:c.hypothesis.claim,verdict:d.verdict,evidence:d.evidence,gate_reason:c.decision.reason}:{hypothesis_id:c.hypothesis.id,claim:c.hypothesis.claim,verdict:"INCONCLUSIVE",evidence:"no verifier result for this claim",gate_reason:c.decision.reason}}),i=new Set(s.filter(c=>c.verdict==="REFUTED").map(c=>c.hypothesis_id)),a=i.size===0?t:t.filter(c=>!i.has(c.id));return{premise_verifications:s,hypotheses_to_test:a}}async function ha(t,e,n){let r=n?.apiKey,o=(()=>{if(typeof t=="string")return{failure:t,repoPath:process.cwd(),context:"",maxHypotheses:4};if(typeof t=="object"&&t!==null){let I=t;if(typeof I.failure=="string")return{failure:I.failure,repoPath:I.repoPath||process.cwd(),context:I.context||"",maxHypotheses:Math.min(I.maxHypotheses||4,4)}}throw new Error("diagnose handler requires input.failure (string) or a string argument")})();if(!e?.sessionId)throw new Error("diagnose requires a parent session with sessionId");let s=e.sessionId,i=$("diagnose"),a=i["system.md"],c=i["research.md"],l=i["hypothesis.md"],d=i["verify.md"];if(!a||!c||!l||!d)throw new Error("diagnose skill missing required prompts (system.md, research.md, hypothesis.md, verify.md)");let u=new P({apiKey:r}),p=wa(o.context),g=ga(o.failure,o.context),b=`Triage:
|
|
1016
1016
|
failure_type: ${g.failure_type}
|
|
1017
1017
|
error_signature: ${g.error_signature}
|
|
1018
1018
|
affected_area: ${g.affected_area}`,m=`${q.systemPrompt}
|
|
@@ -1055,7 +1055,7 @@ Location: ${t.location||"unknown"}
|
|
|
1055
1055
|
Proposed fix: ${t.proposed_fix||"unknown"}
|
|
1056
1056
|
Reproducer: ${e}
|
|
1057
1057
|
|
|
1058
|
-
Working directory (isolated): ${i}`,l=await a.runToResult(c);return l.status!=="succeeded"||!l.output?{hypothesis_id:t.id,reproducer_passed:!1,regressions:[],confidence:0,verification_log:`Verification failed: ${O(l)}`}:l.output}catch(c){return{hypothesis_id:t.id,reproducer_passed:!1,regressions:[],confidence:0,verification_log:`Error during verification: ${c instanceof Error?c.message:String(c)}`}}finally{if(a)try{await a.teardown()}catch{}try{await dt("git",["worktree","remove","--force",i],{cwd:n})}catch{}}}function Sa(){let t=["Edit","Write","Bash","Agent","Task"];return async e=>t.includes(e)?{behavior:"deny",message:`Tool ${e} not allowed in worktree verification. Verification is read-only.`}:q.allowedTools.includes(e)?{behavior:"allow"}:{behavior:"deny",message:`Tool ${e} not allowed. Allowed tools: ${q.allowedTools.join(", ")}`}}var xa={name:"diagnose",description:"Parallel root-cause analysis for bugs and failing tests \u2014 forks research subagents, synthesizes hypotheses, and validates each in isolated worktrees",handler:ha,argumentHint:"<bug-or-failing-test>",whenToUse:"When a test is failing, a bug is reported, or behavior is unexplained \u2014 runs parallel root-cause analysis with hypothesis sub-agents."};X(xa);import{z as Z}from"zod";import{execFile as La}from"child_process";import{promisify as Na}from"util";import{mkdir as Qr,writeFile as Zr}from"fs/promises";import{existsSync as on}from"fs";import{dirname as eo,join as le}from"path";import{fileURLToPath as Ua}from"url";import{fileURLToPath as Ta}from"node:url";import{dirname as Ea}from"node:path";var Pa=Ta(import.meta.url),
|
|
1058
|
+
Working directory (isolated): ${i}`,l=await a.runToResult(c);return l.status!=="succeeded"||!l.output?{hypothesis_id:t.id,reproducer_passed:!1,regressions:[],confidence:0,verification_log:`Verification failed: ${O(l)}`}:l.output}catch(c){return{hypothesis_id:t.id,reproducer_passed:!1,regressions:[],confidence:0,verification_log:`Error during verification: ${c instanceof Error?c.message:String(c)}`}}finally{if(a)try{await a.teardown()}catch{}try{await dt("git",["worktree","remove","--force",i],{cwd:n})}catch{}}}function Sa(){let t=["Edit","Write","Bash","Agent","Task"];return async e=>t.includes(e)?{behavior:"deny",message:`Tool ${e} not allowed in worktree verification. Verification is read-only.`}:q.allowedTools.includes(e)?{behavior:"allow"}:{behavior:"deny",message:`Tool ${e} not allowed. Allowed tools: ${q.allowedTools.join(", ")}`}}var xa={name:"diagnose",description:"Parallel root-cause analysis for bugs and failing tests \u2014 forks research subagents, synthesizes hypotheses, and validates each in isolated worktrees",handler:ha,argumentHint:"<bug-or-failing-test>",whenToUse:"When a test is failing, a bug is reported, or behavior is unexplained \u2014 runs parallel root-cause analysis with hypothesis sub-agents."};X(xa);import{z as Z}from"zod";import{execFile as La}from"child_process";import{promisify as Na}from"util";import{mkdir as Qr,writeFile as Zr}from"fs/promises";import{existsSync as on}from"fs";import{dirname as eo,join as le}from"path";import{fileURLToPath as Ua}from"url";import{fileURLToPath as Ta}from"node:url";import{dirname as Ea}from"node:path";var Pa=Ta(import.meta.url),Ff=Ea(Pa),nn={name:"qualify",systemPrompt:`---
|
|
1059
1059
|
name: qualify
|
|
1060
1060
|
description: Gate proposed plugin skills. Approve only real force multipliers. Reject reminders, checklists, best-practice nudges, and generic execution advice. Invoke when evaluating whether a proposed skill deserves top-level status in this plugin.
|
|
1061
1061
|
model: sonnet
|
|
@@ -1288,7 +1288,7 @@ If the append fails (permissions, disk full, unwritable path), do not retry and
|
|
|
1288
1288
|
- Stage 1 alone would land at SALVAGE (rule 8). Rule 6 fires because Stage 2 \u22648 \u2192 downgrade one tier \u2192 **REJECT**. Rewrite target: raise Bounded Damage (dry-run/draft-PR instead of push), Default Reversibility (require confirmation), Assumption Exposure (surface what tests assume before acting).
|
|
1289
1289
|
|
|
1290
1290
|
Be skeptical. Protect the plugin from fluff. Stage 2 catches patterns that are strong when they work and catastrophic when they don't.
|
|
1291
|
-
`,sourcePath:"agent-framework-local/agents/qualify.md"};import{fileURLToPath as _a}from"node:url";import{dirname as Aa}from"node:path";var Ra=_a(import.meta.url),
|
|
1291
|
+
`,sourcePath:"agent-framework-local/agents/qualify.md"};import{fileURLToPath as _a}from"node:url";import{dirname as Aa}from"node:path";var Ra=_a(import.meta.url),jf=Aa(Ra);import{mkdir as Kr,writeFile as qr}from"fs/promises";import{dirname as Ca,join as Ia}from"path";async function re(t){let e=yr();await Kr(Ca(e),{recursive:!0});let n=new Date().toISOString().split(".")[0]+"Z",r={timestamp:n,surface:"afk",...t},o=JSON.stringify(r)+`
|
|
1292
1292
|
`;return await qr(e,o,{flag:"a"}),n}async function zr(){let t=tt(),e=Ia(t,"forge-thaw-history.jsonl");await Kr(t,{recursive:!0});let r={timestamp:new Date().toISOString().split(".")[0]+"Z",surface:"afk",event:"forge.thaw_override",thaw_triggered:!0},o=JSON.stringify(r)+`
|
|
1293
1293
|
`;await qr(e,o,{flag:"a"})}import{readFile as Vr,readdir as Ma,writeFile as Da,mkdir as Oa,unlink as Fa}from"fs/promises";import{join as ut}from"path";import{existsSync as $a}from"fs";async function Jr(t){let e=ut(xe(),t+".md"),n=await Vr(e,"utf-8");return{id:t,content:n}}async function Yr(){let t=xe();return $a(t)?(await Ma(t,{withFileTypes:!0})).filter(r=>r.isFile()&&r.name.endsWith(".md")).map(r=>r.name.slice(0,-3)):[]}async function rn(t,e){let n=xe(),r=ut(n,t+".md"),o=ut(n,e),s=ut(o,t+".md");await Oa(o,{recursive:!0});let i=await Vr(r,"utf-8");await Da(s,i,"utf-8"),await Fa(r)}function Xr(t){let e=/^\*{0,2}(APPROVE|SALVAGE|REJECT)\*{0,2}/,n=t.split(`
|
|
1294
1294
|
`).map(l=>l.trim()).filter(l=>l),r=n.find(l=>e.test(l));if(!r)return{verdict:"REJECT",feedback:t};let o=r.match(e)?.[1];if(!o)return{verdict:"REJECT",feedback:t};let s=t.match(/score:\s*(\d+)/i),i=s&&s[1]?parseInt(s[1],10):void 0,a=n.indexOf(r),c=n.slice(a+1).join(`
|
|
@@ -1296,7 +1296,7 @@ Be skeptical. Protect the plugin from fluff. Stage 2 catches patterns that are s
|
|
|
1296
1296
|
- ${e.join(`
|
|
1297
1297
|
- `)}`)}function Ba(){let t=eo(Ua(import.meta.url));return Ha(t)}function Wa(t){return le(t,"..","..","..","plugins","awa-private")}function Ga(){return tt()}function Ka(t){let e=t.split(`
|
|
1298
1298
|
`),n=[],r=/^\s+✗\s+(\S+):/;for(let o of e){let s=o.match(r);s&&s[1]&&n.push(s[1])}return n}async function qa(t){let e=Ga(),n=le(e,"qualifications.jsonl");await Qr(e,{recursive:!0});let o=new Date().toISOString().split(".")[0]+"Z",i=JSON.stringify({timestamp:o,surface:"afk",refers_to_run_id:t,source:"forge-gate-check-ts"})+`
|
|
1299
|
-
`;return await Zr(n,i,{flag:"a"}),o}async function za(){let t;try{t=Ba()}catch(c){throw new Error(`Failed to resolve eval-harness runner.py: ${c instanceof Error?c.message:String(c)}`)}let e=Wa(t),n="",r="",o=0;try{let c=await ja("python3",[t,"--plugin-root",e],{timeout:6e4});n=c.stdout||"",r=c.stderr||"",o=0}catch(c){let l=c;if(n=l.stdout||"",r=l.stderr||"",o=typeof l.code=="number"?l.code:1,l.code==="ENOENT"||r&&r.includes("No such file"))throw new Error(`eval-harness runner.py not found at ${t}.`)}let s=o===0?"OPEN":"CLOSED",i=s==="CLOSED"?Ka(n):void 0,a;if(s==="OPEN"){let c=new Date().toISOString().split(".")[0]+"Z";a=await qa(c)}return{gate_status:s,exit_code:o,stdout:n,stderr:r||void 0,tasks_failed:i,ledger_entry_ref:a}}var Va=Z.object({iteration:Z.number().int().positive(),verdict:Z.enum(["APPROVE","SALVAGE","REJECT"]),score:Z.number().optional(),feedback:Z.string()}),
|
|
1299
|
+
`;return await Zr(n,i,{flag:"a"}),o}async function za(){let t;try{t=Ba()}catch(c){throw new Error(`Failed to resolve eval-harness runner.py: ${c instanceof Error?c.message:String(c)}`)}let e=Wa(t),n="",r="",o=0;try{let c=await ja("python3",[t,"--plugin-root",e],{timeout:6e4});n=c.stdout||"",r=c.stderr||"",o=0}catch(c){let l=c;if(n=l.stdout||"",r=l.stderr||"",o=typeof l.code=="number"?l.code:1,l.code==="ENOENT"||r&&r.includes("No such file"))throw new Error(`eval-harness runner.py not found at ${t}.`)}let s=o===0?"OPEN":"CLOSED",i=s==="CLOSED"?Ka(n):void 0,a;if(s==="OPEN"){let c=new Date().toISOString().split(".")[0]+"Z";a=await qa(c)}return{gate_status:s,exit_code:o,stdout:n,stderr:r||void 0,tasks_failed:i,ledger_entry_ref:a}}var Va=Z.object({iteration:Z.number().int().positive(),verdict:Z.enum(["APPROVE","SALVAGE","REJECT"]),score:Z.number().optional(),feedback:Z.string()}),vm=Z.object({status:Z.enum(["APPROVED","REJECTED","GATE_CLOSED","MAX_ITERATIONS"]),skill_path:Z.string().optional(),qualify_verdicts:Z.array(Va),brief_id:Z.string().optional(),telemetry_ref:Z.string()});async function Ja(t,e,n){let r=n?.apiKey,o=typeof t=="string"?{brief:t}:typeof t=="object"&&t!==null?t:{},s=o.brief,i=o.forceThaw??!1,a=o.maxIterations??3,c="",l=[],d="REJECTED",u,p;try{let g=await za();if(g.gate_status==="CLOSED"&&!i)return c=await re({event:"forge.gate_check",gate_status:"CLOSED"}),{status:"GATE_CLOSED",qualify_verdicts:[],telemetry_ref:c};i&&g.gate_status==="CLOSED"&&(await zr(),c=await re({event:"forge.thaw_override",gate_status:"CLOSED"})),c=await re({event:"forge.gate_check",gate_status:"OPEN"});let b="",m=!1;if(s)b=s,m=!0;else{let x=await Yr();if(x.length>0){let T=x[0],M=await Jr(T);b=M.content,p=M.id,m=!0}else{if(!e?.sessionId)throw new Error("forge requires parent session for gap discovery");let M=$("forge")["gap-discovery.md"];if(!M)throw new Error("forge skill missing gap-discovery.md prompt");let E=await(await new P({apiKey:r}).forkSubagent({parent:{sessionId:e.sessionId},config:{model:"sonnet",systemPrompt:M},idPrefix:"forge-gap-discovery"})).runToResult("Identify the most impactful skill gap.");if(E.status!=="succeeded")throw new Error(`gap discovery failed: ${O(E)}`);if(b=E.message?.content||"",!b)throw new Error("gap discovery returned no concept")}}if(c=await re({event:"forge.brief_loaded",used_brief:m,brief_id:p||null}),!e?.sessionId)throw new Error("forge requires parent session for skill generation");let S=$("forge"),f=S["generate.md"],h=S["system.md"];if(!f)throw new Error("forge skill missing generate.md prompt");if(!h)throw new Error("forge skill missing system.md prompt");let A=await(await new P({apiKey:r}).forkSubagent({parent:{sessionId:e.sessionId},config:{model:"sonnet",systemPrompt:h},idPrefix:"forge-generate"})).runToResult(`Generate a new amplifier skill based on this concept:
|
|
1300
1300
|
|
|
1301
1301
|
${b}`);if(A.status!=="succeeded")throw new Error(`skill generation failed: ${O(A)}`);let F=A.message?.content||"";if(!F)throw new Error("skill generation returned no output");for(let x=1;x<=a;x++){let T=nn.systemPrompt;if(!T)throw new Error("qualify agent missing system prompt");let v=await(await new P({apiKey:r}).forkSubagent({parent:{sessionId:e.sessionId},config:{model:"sonnet",systemPrompt:T},idPrefix:`forge-qualify-${x}`})).runToResult(`Evaluate this amplifier skill against the force-multiplier criteria:
|
|
1302
1302
|
|
|
@@ -1429,7 +1429,7 @@ ${e.replace(/<\/?cross-session-memory\b[^>]*>/gi,"")}
|
|
|
1429
1429
|
|
|
1430
1430
|
${o}`};if(o&&typeof o=="object"&&"type"in o&&o.type==="preset"){let s=o.append??"";return{...t,systemPrompt:{...o,append:`${r}
|
|
1431
1431
|
|
|
1432
|
-
${s}`}}}return{...t,systemPrompt:r}}var bo={name:"memory_search",description:'Search cross-session memory for facts and procedures. Returns results ranked by relevance. Use this to recall information from prior sessions. Supports FTS5 match syntax: AND, OR, NOT, "exact phrase", prefix*',input_schema:{type:"object",properties:{query:{type:"string",description:'Search query (supports FTS5 match syntax: AND, OR, NOT, "exact phrase", prefix*)'},category:{type:"string",enum:["preference","convention","decision","learning"],description:"Optional: filter by fact category"},since:{type:"string",description:"Optional: ISO date \u2014 only return facts created after this date"},limit:{type:"number",description:"Max results (default 10)"}},required:["query"]}},wo={name:"memory_update",description:'Store a fact in cross-session memory or update hot memory. Hot memory (target: "hot") persists in the system prompt across all future sessions. Facts (target: "fact") are stored in the searchable archive.',input_schema:{type:"object",properties:{target:{type:"string",enum:["hot","fact"],description:'"hot" writes to HOT.md (system prompt), "fact" writes to the searchable archive'},action:{type:"string",enum:["set","supersede","remove"],description:"Operation: set (create/overwrite), supersede (replace while keeping history), remove (delete)"},content:{type:"string",description:"The content to store (for set/supersede)"},category:{type:"string",enum:["preference","convention","decision","learning"],description:"Required for fact target"},supersedes:{type:"number",description:"Fact ID being superseded (for supersede action)"},id:{type:"number",description:"Fact ID to remove (for remove action)"}},required:["target","action"]}},ko={name:"procedure_write",description:"Write a reusable procedure to memory. Procedures are markdown files describing how to perform recurring tasks. They persist across sessions and are searchable via memory_search.",input_schema:{type:"object",properties:{name:{type:"string",description:"Procedure name (kebab-case, becomes the filename)"},content:{type:"string",description:"Procedure content (markdown)"}},required:["name","content"]}},ht=[bo,wo,ko],vo=ht.map(t=>t.name);function an(t,e,n){let r=async i=>{try{let a=vc(i),c=t.search(a.query,{category:a.category,since:a.since,limit:a.limit??10});return{content:JSON.stringify(c)}}catch(a){return{content:`memory_search error: ${a instanceof Error?a.message:String(a)}`,isError:!0}}},o=async i=>{try{let a=Sc(i);if(a.target==="hot")return a.action!=="set"?{content:'Hot memory only supports action: "set". Use supersede/remove only for facts.',isError:!0}:a.content?(t.saveHot(a.content),{content:JSON.stringify({saved:!0,target:"hot"})}):{content:'content is required for action: "set"',isError:!0};if(a.action==="set"){if(!a.category)return{content:"category is required for fact storage",isError:!0};if(!a.content)return{content:'content is required for action: "set"',isError:!0};let c=t.storeFact({session_id:e,category:a.category,content:a.content,source_surface:n??"cli"});return{content:JSON.stringify({id:c,action:"set",target:"fact"})}}if(a.action==="supersede"){if(!a.supersedes)return{content:'supersedes (fact ID) is required for action: "supersede"',isError:!0};if(!a.content)return{content:'content is required for action: "supersede"',isError:!0};let c=t.supersedeFact(a.supersedes,a.content,a.category??void 0);return{content:JSON.stringify({id:c,action:"supersede",target:"fact",supersedes:a.supersedes})}}if(a.action==="remove"){if(!a.id)return{content:'id (fact ID) is required for action: "remove"',isError:!0};let c=t.removeFact(a.id);return{content:JSON.stringify({removed:c,action:"remove",target:"fact"})}}return{content:`Unknown action: ${a.action}`,isError:!0}}catch(a){return{content:`memory_update error: ${a instanceof Error?a.message:String(a)}`,isError:!0}}},s=async i=>{try{let a=xc(i);return t.writeProcedure(a.name,a.content,e),{content:JSON.stringify({name:a.name,written:!0})}}catch(a){return{content:`procedure_write error: ${a instanceof Error?a.message:String(a)}`,isError:!0}}};return new Map([["memory_search",r],["memory_update",o],["procedure_write",s]])}function vc(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t;if(typeof e.query!="string")throw new Error("query (string) is required");let n={query:e.query};if(e.category!==void 0){if(typeof e.category!="string")throw new Error("category must be a string");let r=["preference","convention","decision","learning"];if(!r.includes(e.category))throw new Error(`category must be one of: ${r.join(", ")}`);n.category=e.category}if(e.since!==void 0){if(typeof e.since!="string")throw new Error("since must be a string (ISO date)");n.since=e.since}if(e.limit!==void 0){if(typeof e.limit!="number"||e.limit<=0)throw new Error("limit must be a positive number");n.limit=e.limit}return n}function Sc(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t,n=["hot","fact"];if(typeof e.target!="string"||!n.includes(e.target))throw new Error(`target must be one of: ${n.join(", ")}`);let r=["set","supersede","remove"];if(typeof e.action!="string"||!r.includes(e.action))throw new Error(`action must be one of: ${r.join(", ")}`);let o={target:e.target,action:e.action};if(e.content!==void 0){if(typeof e.content!="string")throw new Error("content must be a string");o.content=e.content}if(e.category!==void 0){if(typeof e.category!="string")throw new Error("category must be a string");let s=["preference","convention","decision","learning"];if(!s.includes(e.category))throw new Error(`category must be one of: ${s.join(", ")}`);o.category=e.category}if(e.supersedes!==void 0){if(typeof e.supersedes!="number"||e.supersedes<=0)throw new Error("supersedes must be a positive fact ID");o.supersedes=e.supersedes}if(e.id!==void 0){if(typeof e.id!="number"||e.id<=0)throw new Error("id must be a positive fact ID");o.id=e.id}return o}function xc(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t;if(typeof e.name!="string")throw new Error("name (string) is required");if(typeof e.content!="string")throw new Error("content (string) is required");return{name:e.name,content:e.content}}import{config as
|
|
1432
|
+
${s}`}}}return{...t,systemPrompt:r}}var bo={name:"memory_search",description:'Search cross-session memory for facts and procedures. Returns results ranked by relevance. Use this to recall information from prior sessions. Supports FTS5 match syntax: AND, OR, NOT, "exact phrase", prefix*',input_schema:{type:"object",properties:{query:{type:"string",description:'Search query (supports FTS5 match syntax: AND, OR, NOT, "exact phrase", prefix*)'},category:{type:"string",enum:["preference","convention","decision","learning"],description:"Optional: filter by fact category"},since:{type:"string",description:"Optional: ISO date \u2014 only return facts created after this date"},limit:{type:"number",description:"Max results (default 10)"}},required:["query"]}},wo={name:"memory_update",description:'Store a fact in cross-session memory or update hot memory. Hot memory (target: "hot") persists in the system prompt across all future sessions. Facts (target: "fact") are stored in the searchable archive.',input_schema:{type:"object",properties:{target:{type:"string",enum:["hot","fact"],description:'"hot" writes to HOT.md (system prompt), "fact" writes to the searchable archive'},action:{type:"string",enum:["set","supersede","remove"],description:"Operation: set (create/overwrite), supersede (replace while keeping history), remove (delete)"},content:{type:"string",description:"The content to store (for set/supersede)"},category:{type:"string",enum:["preference","convention","decision","learning"],description:"Required for fact target"},supersedes:{type:"number",description:"Fact ID being superseded (for supersede action)"},id:{type:"number",description:"Fact ID to remove (for remove action)"}},required:["target","action"]}},ko={name:"procedure_write",description:"Write a reusable procedure to memory. Procedures are markdown files describing how to perform recurring tasks. They persist across sessions and are searchable via memory_search.",input_schema:{type:"object",properties:{name:{type:"string",description:"Procedure name (kebab-case, becomes the filename)"},content:{type:"string",description:"Procedure content (markdown)"}},required:["name","content"]}},ht=[bo,wo,ko],vo=ht.map(t=>t.name);function an(t,e,n){let r=async i=>{try{let a=vc(i),c=t.search(a.query,{category:a.category,since:a.since,limit:a.limit??10});return{content:JSON.stringify(c)}}catch(a){return{content:`memory_search error: ${a instanceof Error?a.message:String(a)}`,isError:!0}}},o=async i=>{try{let a=Sc(i);if(a.target==="hot")return a.action!=="set"?{content:'Hot memory only supports action: "set". Use supersede/remove only for facts.',isError:!0}:a.content?(t.saveHot(a.content),{content:JSON.stringify({saved:!0,target:"hot"})}):{content:'content is required for action: "set"',isError:!0};if(a.action==="set"){if(!a.category)return{content:"category is required for fact storage",isError:!0};if(!a.content)return{content:'content is required for action: "set"',isError:!0};let c=t.storeFact({session_id:e,category:a.category,content:a.content,source_surface:n??"cli"});return{content:JSON.stringify({id:c,action:"set",target:"fact"})}}if(a.action==="supersede"){if(!a.supersedes)return{content:'supersedes (fact ID) is required for action: "supersede"',isError:!0};if(!a.content)return{content:'content is required for action: "supersede"',isError:!0};let c=t.supersedeFact(a.supersedes,a.content,a.category??void 0);return{content:JSON.stringify({id:c,action:"supersede",target:"fact",supersedes:a.supersedes})}}if(a.action==="remove"){if(!a.id)return{content:'id (fact ID) is required for action: "remove"',isError:!0};let c=t.removeFact(a.id);return{content:JSON.stringify({removed:c,action:"remove",target:"fact"})}}return{content:`Unknown action: ${a.action}`,isError:!0}}catch(a){return{content:`memory_update error: ${a instanceof Error?a.message:String(a)}`,isError:!0}}},s=async i=>{try{let a=xc(i);return t.writeProcedure(a.name,a.content,e),{content:JSON.stringify({name:a.name,written:!0})}}catch(a){return{content:`procedure_write error: ${a instanceof Error?a.message:String(a)}`,isError:!0}}};return new Map([["memory_search",r],["memory_update",o],["procedure_write",s]])}function vc(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t;if(typeof e.query!="string")throw new Error("query (string) is required");let n={query:e.query};if(e.category!==void 0){if(typeof e.category!="string")throw new Error("category must be a string");let r=["preference","convention","decision","learning"];if(!r.includes(e.category))throw new Error(`category must be one of: ${r.join(", ")}`);n.category=e.category}if(e.since!==void 0){if(typeof e.since!="string")throw new Error("since must be a string (ISO date)");n.since=e.since}if(e.limit!==void 0){if(typeof e.limit!="number"||e.limit<=0)throw new Error("limit must be a positive number");n.limit=e.limit}return n}function Sc(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t,n=["hot","fact"];if(typeof e.target!="string"||!n.includes(e.target))throw new Error(`target must be one of: ${n.join(", ")}`);let r=["set","supersede","remove"];if(typeof e.action!="string"||!r.includes(e.action))throw new Error(`action must be one of: ${r.join(", ")}`);let o={target:e.target,action:e.action};if(e.content!==void 0){if(typeof e.content!="string")throw new Error("content must be a string");o.content=e.content}if(e.category!==void 0){if(typeof e.category!="string")throw new Error("category must be a string");let s=["preference","convention","decision","learning"];if(!s.includes(e.category))throw new Error(`category must be one of: ${s.join(", ")}`);o.category=e.category}if(e.supersedes!==void 0){if(typeof e.supersedes!="number"||e.supersedes<=0)throw new Error("supersedes must be a positive fact ID");o.supersedes=e.supersedes}if(e.id!==void 0){if(typeof e.id!="number"||e.id<=0)throw new Error("id must be a positive fact ID");o.id=e.id}return o}function xc(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t;if(typeof e.name!="string")throw new Error("name (string) is required");if(typeof e.content!="string")throw new Error("content (string) is required");return{name:e.name,content:e.content}}import{config as Vm}from"dotenv";var So={opus:"claude-opus-4-7",opus_1m:"claude-opus-4-7",sonnet:"claude-sonnet-4-6",sonnet_1m:"claude-sonnet-4-6",haiku:"claude-haiku-4-5-20251001"};function xo(t){return t in So}function To(t){let e=So[t];if(!e)throw new Error(`Invalid model: ${t}`);return e}function He(t){if(t!==void 0)return typeof t=="string"&&xo(t)?To(t):t}function Eo(){return process.env.ANTHROPIC_API_KEY||process.env.CLAUDE_CODE_OAUTH_TOKEN||me()}function B(){return Eo()}async function Po(t,e,n){let o=$("mint")["spec.md"];if(!o)throw new Error("mint skill missing spec.md prompt");let a=await(await new P(n!==void 0?{cwd:n}:{}).forkSubagent({parent:{sessionId:e},config:{model:"sonnet",systemPrompt:o,apiKey:B()},idPrefix:"mint-spec"})).runToResult(`Create a detailed specification for: ${t}`);if(a.status!=="succeeded"||!a.message)throw new Error(`spec phase failed: ${O(a)}`);return a.message.content}async function _o(t,e,n){let o=$("mint")["research.md"];if(!o)throw new Error("mint skill missing research.md prompt");let a=await(await new P(n!==void 0?{cwd:n}:{}).forkSubagent({parent:{sessionId:e},config:{model:"sonnet",systemPrompt:o,apiKey:B()},idPrefix:"mint-research"})).runToResult(`Gather context and research for this specification:
|
|
1433
1433
|
|
|
1434
1434
|
${t}`);if(a.status!=="succeeded"||!a.message)throw new Error(`research phase failed: ${O(a)}`);return a.message.content}async function Ao(t,e,n,r){let s=$("mint")["plan.md"];if(!s)throw new Error("mint skill missing plan.md prompt");let a=await new P(r!==void 0?{cwd:r}:{}).forkSubagent({parent:{sessionId:n},config:{model:"sonnet",systemPrompt:s,apiKey:B()},idPrefix:"mint-plan"}),c=`Specification:
|
|
1435
1435
|
${t}
|
|
@@ -1484,7 +1484,7 @@ Create a ship-ready summary with next steps.`,c=await i.runToResult(a);if(c.stat
|
|
|
1484
1484
|
`))return{};let n=e.slice(4),r=n.indexOf(`
|
|
1485
1485
|
---`);if(r===-1)return{};let o=n.slice(0,r),s=n.slice(r+4).trim(),i={},a=o.split(`
|
|
1486
1486
|
`);for(let c of a){if(!c)continue;let l=c.indexOf(":");if(l===-1)continue;let d=c.slice(0,l).trim(),u=c.slice(l+1).trim();d==="name"?i.name=u.replace(/^["']|["']$/g,""):d==="description"?i.description=u.replace(/^["']|["']$/g,""):d==="argumentHint"&&(i.argumentHint=u.replace(/^["']|["']$/g,""))}return s.length>0&&(i.body=s),i}catch{return{}}}function Wo(t){let e=pn(t);if(e.length===0)return"";let n=[];for(let r of e){let o=r.argumentHint?`${r.argumentHint}`:"",s=o?`- \`${r.name} ${o}\`: ${r.description}`:`- ${r.name}: ${r.description}`;n.push(s),r.whenToUse&&n.push(` When to use: ${r.whenToUse}`)}return["Available skills (invoke via the `skill` tool):","","Each skill dispatches one or more context-isolated subagents internally. Calling `skill` is a delegation primitive \u2014 it preserves the main session's context. Prefer a skill over inline investigation when the task shape matches.","",...n].join(`
|
|
1487
|
-
`)}function pn(t){let e=[],n=new Set;for(let o of mr()){let s=he(o);e.push({name:o,description:s.description,source:s.origin==="user"?"user":s.origin==="project"?"project":"builtin",argumentHint:s.argumentHint,whenToUse:s.whenToUse}),n.add(o)}let r=t??[...te(qt()),...te(),...te(zt())];for(let o of r){if(o.type!=="local")continue;let s=un(o.path);for(let i of s)!i.name||n.has(i.name)||(e.push({name:i.name,description:i.description??`Skill from plugin at ${o.path}`,source:"plugin"}),n.add(i.name))}return e}function Co(t){let e=new Map,n=t??[...te(qt()),...te(),...te(zt())];for(let r of n){if(r.type!=="local")continue;let o=un(r.path);for(let s of o)s.name&&s.body&&s.body.length>0&&e.set(s.name,s.body)}return e}var zc={opus:128e3,opus_1m:128e3,sonnet:64e3,sonnet_1m:64e3,haiku:64e3,"claude-opus-4-7":128e3,"claude-opus-4-6":128e3,"claude-sonnet-4-6":64e3,"claude-haiku-4-5-20251001":64e3},Vc=64e3;function Go(t){return zc[t]??Vc}var Jc={opus:2e5,opus_1m:1e6,sonnet:2e5,sonnet_1m:1e6,haiku:2e5},Yc=2e5;function Ko(t){return Jc[t]??Yc}var Xc=3,Qc="claude-haiku-4-5-20251001",Zc=1024,el=[{value:"claude-sonnet-4-5-20250929",displayName:"Claude Sonnet 4.5",description:"Latest balanced Claude \u2014 recommended default"},{value:"claude-opus-4-5-20250929",displayName:"Claude Opus 4.5",description:"Highest-capability Claude"},{value:"claude-haiku-4-5-20250929",displayName:"Claude Haiku 4.5",description:"Fastest, cheapest Claude"}],yt=class{client;authMode;initSessionId;promptStream;toolDispatcher;maxTokens;tools;systemPrefix;userSystem;tokenRefresher;thinking;traceWriter;currentModel;currentPermissionMode;messages;closed=!1;abortController=null;pendingAbortReason=null;closedPromise;closeResolve=null;lastUsage=null;refreshPromise=null;usageLimitWaitPromise=null;autoResumeOnUsageLimit;cwdDependentsFactory;constructor(e){this.client=e.client,this.authMode=e.authMode,this.initSessionId=e.sessionId??We(),this.promptStream=e.promptStream,this.toolDispatcher=e.toolDispatcher,this.maxTokens=e.maxTokens,this.tools=e.tools,this.systemPrefix=e.systemPrefix,this.userSystem=e.userSystem,this.currentModel=e.model,this.currentPermissionMode=e.permissionMode??"default",this.messages=e.initialMessages?[...e.initialMessages]:[],this.tokenRefresher=e.tokenRefresher,this.thinking=e.thinking,this.traceWriter=e.traceWriter,this.autoResumeOnUsageLimit=e.autoResumeOnUsageLimit??!0,this.cwdDependentsFactory=e.cwdDependentsFactory,this.closedPromise=new Promise(n=>{this.closeResolve=()=>n("__closed__")})}async*[Symbol.asyncIterator](){yield{type:"session.init",info:{sessionId:this.initSessionId,model:this.currentModel,permissionMode:this.currentPermissionMode,cwd:process.cwd(),tools:[],slashCommands:[],skills:[],plugins:[],mcpServers:[],apiKeySource:this.authMode,version:"anthropic-direct-v1"}};let n=this.promptStream[Symbol.asyncIterator]();try{for(;!this.closed;){let r=await Promise.race([n.next(),this.closedPromise]);if(r==="__closed__")break;let o=r;if(o.done)break;let s=o.value,i=new AbortController;if(this.abortController=i,this.pendingAbortReason!==null&&!i.signal.aborted&&(i.abort(this.pendingAbortReason),this.pendingAbortReason=null),i.signal.aborted){this.abortController===i&&(this.abortController=null);return}this.messages.push({role:"user",content:s.content});let a=this.composeSystem(),c=De(this.authMode,this.initSessionId,We()),l={client:this.client,messages:this.messages,system:a,tools:this.tools,toolDispatcher:this.toolDispatcher,model:this.currentModel,maxTokens:this.maxTokens,headers:c,signal:i.signal,ctx:{sessionId:this.initSessionId},...this.thinking!==void 0?{thinking:this.thinking}:{},...this.traceWriter?{traceWriter:this.traceWriter}:{}};try{for await(let d of this.turnWithUsageLimitRetry(l)){if(this.closed)return;d.type==="turn.completed"&&(this.lastUsage=d.usage,this.abortController===i&&(this.abortController=null)),yield d}}catch(d){if(i.signal.aborted)return;yield{type:"error",error:d instanceof Error?d:new Error(String(d))};return}finally{this.abortController===i&&(this.abortController=null)}}}catch(r){yield{type:"error",error:r instanceof Error?r:new Error(String(r))}}finally{try{await n.return?.()}catch{}}}async*turnWithUsageLimitRetry(e){let n=null,r=null;for await(let l of this.turnWithAuthRetry(e)){if(l.type==="error"){let d=Ht(l.error);if(d&&d.kind==="oauth-limit"){r=d.resetsAt,n=l;break}}yield l}if(!r||!n)return;let o=7200*1e3;if(r.getTime()-Date.now()>o){yield n;return}let s=jt(me()??"");if(yield{type:"paused",reason:"usage-limit",resetsAt:r,accountId:s},!this.autoResumeOnUsageLimit){yield n;return}let i;if(this.usageLimitWaitPromise)i=await this.usageLimitWaitPromise;else{this.usageLimitWaitPromise=Xn({resetsAt:r,signal:e.signal});try{i=await this.usageLimitWaitPromise}finally{this.usageLimitWaitPromise=null}}if(i==="aborted")return;let a=me();e.headers=De(this.authMode,this.initSessionId,We());let c=i==="hot-swap"?jt(a??""):s;yield{type:"resumed",hotSwapped:i==="hot-swap",accountId:c},yield*this.turnWithAuthRetry(e)}async*turnWithAuthRetry(e){let n=null;for await(let o of Bt(e)){if(this.closed)return;if(o.type==="error"&&this.isRetryableAuth(o.error)){n=o;break}yield o}if(!n)return;let r=null;try{if(this.refreshPromise)r=await this.refreshPromise;else{this.refreshPromise=this.tokenRefresher();try{r=await this.refreshPromise??null}finally{this.refreshPromise=null}}}catch{this.refreshPromise=null}if(!r){yield n;return}this.client=r,e.client=this.client,e.headers=De(this.authMode,this.initSessionId,We()),yield*Bt(e)}isRetryableAuth(e){return this.authMode==="oauth"&&this.tokenRefresher!==void 0&&"status"in e&&e.status===401}composeSystem(){let e=this.systemPrefix,n=this.userSystem,r=[];e&&e.length>0&&r.push(...e),n&&n.length>0&&r.push({type:"text",text:n});let o=tr(this.currentPermissionMode);return o!==null&&r.push(o),r.length===0?null:Xe()?Qn(r,Qe()):r}async interrupt(){let e=this.abortController;if(e&&!e.signal.aborted){e.abort("interrupted");return}this.pendingAbortReason="interrupted"}async setModel(e){e!==void 0&&e.length>0&&(this.currentModel=e)}async setPermissionMode(e){this.currentPermissionMode=e}setCwd(e){if(!this.cwdDependentsFactory)return;let{userSystem:n,dispatcher:r}=this.cwdDependentsFactory(e);this.userSystem=n,this.toolDispatcher=r}async supportedCommands(){try{return pn().map(n=>{let r={name:n.name,description:n.description};return n.argumentHint&&(r.argumentHint=n.argumentHint),r})}catch{return[]}}async supportedModels(){return el.map(e=>({...e}))}async supportedAgents(){return[]}async getContextUsage(){let e=this.lastUsage,n=Ko(this.currentModel),r;if(e&&n>0){let o=(e.inputTokens??0)+(e.outputTokens??0)+(e.cachedInputTokens??0)+(e.cacheCreationTokens??0);r=Math.min(100,Math.max(0,o/n*100))}return{tools:[],agents:[],isAutoCompactEnabled:!1,apiUsage:this.lastUsage,...r!==void 0?{percentage:r}:{},maxTokens:n}}async mcpServerStatus(){return[]}async accountInfo(){return{subscriptionType:this.authMode==="oauth"?"claude-subscription":"api-key"}}async rewindFiles(e,n){return{canRewind:!1,error:"anthropic-direct provider does not support file checkpoint rewind"}}async compact(){let e=this.messages.length;if(this.closed)return{compacted:!1,reason:"session-closed",messagesBefore:e,messagesAfter:e};if(this.abortController!==null)return{compacted:!1,reason:"turn-in-flight",messagesBefore:e,messagesAfter:e};let n=tl(),r=ar(this.messages,n);if(r<=0)return{compacted:!1,reason:"history-too-short",messagesBefore:e,messagesAfter:e};let o=this.messages.slice(0,r),s=nl(),i=cr(o,s,Zc),a=new AbortController;this.abortController=a,this.pendingAbortReason!==null&&!a.signal.aborted&&(a.abort(this.pendingAbortReason),this.pendingAbortReason=null);let c;try{if(a.signal.aborted)return{compacted:!1,reason:"aborted",messagesBefore:e,messagesAfter:e};let p=De(this.authMode,this.initSessionId,We()),g=this.client,b=await Promise.resolve(g.messages.create(i,{headers:p,signal:a.signal}));c=await rl(b)}catch(p){return a.signal.aborted?{compacted:!1,reason:"aborted",messagesBefore:e,messagesAfter:e}:{compacted:!1,reason:"summarization-failed: "+(p instanceof Error?p.message:String(p)),messagesBefore:e,messagesAfter:e}}finally{this.abortController===a&&(this.abortController=null)}if(c.trim().length===0)return{compacted:!1,reason:"empty-summary",messagesBefore:e,messagesAfter:e};let l=dr(this.messages,r,c),d=lr(this.messages,r,c);this.messages.splice(0,this.messages.length,...d);let u=this.messages.length;return jn(this.traceWriter,{trigger:"manual",preCompactionMessages:o,summary:c,keptTailCount:e-r,keepLastNConfig:n,messagesBefore:e,messagesAfter:u,tokensSavedEstimate:l}),{compacted:!0,messagesBefore:e,messagesAfter:u,tokensSavedEstimate:l}}close(){this.closed=!0;let e=this.abortController;e&&!e.signal.aborted?e.abort("closed"):this.pendingAbortReason="closed",this.closeResolve?.()}};function tl(){let t=process.env.AFK_COMPACT_KEEP_LAST_TURNS;if(t!==void 0&&t.length>0){let e=Number.parseInt(t,10);if(Number.isFinite(e)&&e>0)return e}return Xc}function nl(){let t=process.env.AFK_COMPACT_MODEL;return t!==void 0&&t.length>0?t:Qc}async function rl(t){let e="";for await(let n of t)if(n.type==="content_block_delta"){let r=n.delta;r.type==="text_delta"&&typeof r.text=="string"&&(e+=r.text)}return e}import fn from"path";import{appendFileSync as ol,mkdirSync as sl}from"fs";import{dirname as il}from"path";function Ge(t,e){return e?.allowedTools?e.allowedTools.includes(t)?{allowed:!0}:{allowed:!1,reason:`Tool "${t}" is not in the configured allowlist`}:{allowed:!0}}var al=new Set(["agent","compose","read_file","glob","grep","list_directory","memory_search"]);function qo(t){return al.has(t)}function cl(t,e){return t.reduce((n,r,o)=>{let s=e(r.name,r.input),i=n[n.length-1];return i&&s&&i.isConcurrencySafe?i.indices.push(o):n.push({isConcurrencySafe:s,indices:[o]}),n},[])}var ke=class{handlers;schemas;hookRegistry;permissions;subagentExecutor;skillExecutor;composeExecutor;classifier;resolveBase;_readRoots;_writeRoots;sessionId;traceWriter;constructor(e){this.handlers=e.handlers,this.schemas=e.schemas,this.hookRegistry=e.hookRegistry,this.permissions=e.permissions,this.subagentExecutor=e.subagentExecutor,this.skillExecutor=e.skillExecutor,this.composeExecutor=e.composeExecutor,this.classifier=e.concurrencyClassifier??qo,this.resolveBase=e.cwd,this.sessionId=e.sessionId,this.traceWriter=e.traceWriter;let n=e.cwd?[e.cwd]:[];this._readRoots=e.readRoots??n.slice(),this._writeRoots=e.writeRoots??n.slice()}get handlerContext(){return{cwd:this.resolveBase,resolveBase:this.resolveBase,readRoots:this._readRoots.slice(),writeRoots:this._writeRoots.slice()}}addReadRoot(e,n="slash"){let r=fn.resolve(e);this._readRoots.includes(r)||this._readRoots.push(r),this.appendAuditLog({action:"grant-read",path:r,source:n})}addWriteRoot(e,n="slash"){let r=fn.resolve(e);this._readRoots.includes(r)||this._readRoots.push(r),this._writeRoots.includes(r)||this._writeRoots.push(r),this.appendAuditLog({action:"grant-write",path:r,source:n})}revokeRoot(e,n="slash"){let r=fn.resolve(e);if(r===this.resolveBase)return;let o=this._readRoots.indexOf(r);o!==-1&&this._readRoots.splice(o,1);let s=this._writeRoots.indexOf(r);s!==-1&&this._writeRoots.splice(s,1),this.appendAuditLog({action:"revoke",path:r,source:n})}getGrants(){return{resolveBase:this.resolveBase,readRoots:this._readRoots.slice(),writeRoots:this._writeRoots.slice()}}appendAuditLog(e){try{let n=ot();sl(il(n),{recursive:!0});let r=JSON.stringify({timestamp:new Date().toISOString(),sessionId:this.sessionId??null,action:e.action,path:e.path,source:e.source});ol(n,r+`
|
|
1487
|
+
`)}function pn(t){let e=[],n=new Set;for(let o of mr()){let s=he(o);e.push({name:o,description:s.description,source:s.origin==="user"?"user":s.origin==="project"?"project":"builtin",argumentHint:s.argumentHint,whenToUse:s.whenToUse}),n.add(o)}let r=t??[...te(qt()),...te(),...te(zt())];for(let o of r){if(o.type!=="local")continue;let s=un(o.path);for(let i of s)!i.name||n.has(i.name)||(e.push({name:i.name,description:i.description??`Skill from plugin at ${o.path}`,source:"plugin"}),n.add(i.name))}return e}function Co(t){let e=new Map,n=t??[...te(qt()),...te(),...te(zt())];for(let r of n){if(r.type!=="local")continue;let o=un(r.path);for(let s of o)s.name&&s.body&&s.body.length>0&&e.set(s.name,s.body)}return e}var zc={opus:128e3,opus_1m:128e3,sonnet:64e3,sonnet_1m:64e3,haiku:64e3,"claude-opus-4-7":128e3,"claude-opus-4-6":128e3,"claude-sonnet-4-6":64e3,"claude-haiku-4-5-20251001":64e3},Vc=64e3;function Go(t){return zc[t]??Vc}var Jc={opus:2e5,opus_1m:1e6,sonnet:2e5,sonnet_1m:1e6,haiku:2e5},Yc=2e5;function Ko(t){return Jc[t]??Yc}var Xc=2,Qc="claude-haiku-4-5-20251001",Zc=1024,el=[{value:"claude-sonnet-4-5-20250929",displayName:"Claude Sonnet 4.5",description:"Latest balanced Claude \u2014 recommended default"},{value:"claude-opus-4-5-20250929",displayName:"Claude Opus 4.5",description:"Highest-capability Claude"},{value:"claude-haiku-4-5-20250929",displayName:"Claude Haiku 4.5",description:"Fastest, cheapest Claude"}],yt=class{client;authMode;initSessionId;promptStream;toolDispatcher;maxTokens;tools;systemPrefix;userSystem;tokenRefresher;thinking;traceWriter;currentModel;currentPermissionMode;messages;closed=!1;abortController=null;pendingAbortReason=null;closedPromise;closeResolve=null;lastUsage=null;refreshPromise=null;usageLimitWaitPromise=null;autoResumeOnUsageLimit;cwdDependentsFactory;constructor(e){this.client=e.client,this.authMode=e.authMode,this.initSessionId=e.sessionId??We(),this.promptStream=e.promptStream,this.toolDispatcher=e.toolDispatcher,this.maxTokens=e.maxTokens,this.tools=e.tools,this.systemPrefix=e.systemPrefix,this.userSystem=e.userSystem,this.currentModel=e.model,this.currentPermissionMode=e.permissionMode??"default",this.messages=e.initialMessages?[...e.initialMessages]:[],this.tokenRefresher=e.tokenRefresher,this.thinking=e.thinking,this.traceWriter=e.traceWriter,this.autoResumeOnUsageLimit=e.autoResumeOnUsageLimit??!0,this.cwdDependentsFactory=e.cwdDependentsFactory,this.closedPromise=new Promise(n=>{this.closeResolve=()=>n("__closed__")})}async*[Symbol.asyncIterator](){yield{type:"session.init",info:{sessionId:this.initSessionId,model:this.currentModel,permissionMode:this.currentPermissionMode,cwd:process.cwd(),tools:[],slashCommands:[],skills:[],plugins:[],mcpServers:[],apiKeySource:this.authMode,version:"anthropic-direct-v1"}};let n=this.promptStream[Symbol.asyncIterator]();try{for(;!this.closed;){let r=await Promise.race([n.next(),this.closedPromise]);if(r==="__closed__")break;let o=r;if(o.done)break;let s=o.value,i=new AbortController;if(this.abortController=i,this.pendingAbortReason!==null&&!i.signal.aborted&&(i.abort(this.pendingAbortReason),this.pendingAbortReason=null),i.signal.aborted){this.abortController===i&&(this.abortController=null);return}this.messages.push({role:"user",content:s.content});let a=this.composeSystem(),c=De(this.authMode,this.initSessionId,We()),l={client:this.client,messages:this.messages,system:a,tools:this.tools,toolDispatcher:this.toolDispatcher,model:this.currentModel,maxTokens:this.maxTokens,headers:c,signal:i.signal,ctx:{sessionId:this.initSessionId},...this.thinking!==void 0?{thinking:this.thinking}:{},...this.traceWriter?{traceWriter:this.traceWriter}:{}};try{for await(let d of this.turnWithUsageLimitRetry(l)){if(this.closed)return;d.type==="turn.completed"&&(this.lastUsage=d.usage,this.abortController===i&&(this.abortController=null)),yield d}}catch(d){if(i.signal.aborted)return;yield{type:"error",error:d instanceof Error?d:new Error(String(d))};return}finally{this.abortController===i&&(this.abortController=null)}}}catch(r){yield{type:"error",error:r instanceof Error?r:new Error(String(r))}}finally{try{await n.return?.()}catch{}}}async*turnWithUsageLimitRetry(e){let n=null,r=null;for await(let l of this.turnWithAuthRetry(e)){if(l.type==="error"){let d=Ht(l.error);if(d&&d.kind==="oauth-limit"){r=d.resetsAt,n=l;break}}yield l}if(!r||!n)return;let o=7200*1e3;if(r.getTime()-Date.now()>o){yield n;return}let s=jt(me()??"");if(yield{type:"paused",reason:"usage-limit",resetsAt:r,accountId:s},!this.autoResumeOnUsageLimit){yield n;return}let i;if(this.usageLimitWaitPromise)i=await this.usageLimitWaitPromise;else{this.usageLimitWaitPromise=Xn({resetsAt:r,signal:e.signal});try{i=await this.usageLimitWaitPromise}finally{this.usageLimitWaitPromise=null}}if(i==="aborted")return;let a=me();e.headers=De(this.authMode,this.initSessionId,We());let c=i==="hot-swap"?jt(a??""):s;yield{type:"resumed",hotSwapped:i==="hot-swap",accountId:c},yield*this.turnWithAuthRetry(e)}async*turnWithAuthRetry(e){let n=null;for await(let o of Bt(e)){if(this.closed)return;if(o.type==="error"&&this.isRetryableAuth(o.error)){n=o;break}yield o}if(!n)return;let r=null;try{if(this.refreshPromise)r=await this.refreshPromise;else{this.refreshPromise=this.tokenRefresher();try{r=await this.refreshPromise??null}finally{this.refreshPromise=null}}}catch{this.refreshPromise=null}if(!r){yield n;return}this.client=r,e.client=this.client,e.headers=De(this.authMode,this.initSessionId,We()),yield*Bt(e)}isRetryableAuth(e){return this.authMode==="oauth"&&this.tokenRefresher!==void 0&&"status"in e&&e.status===401}composeSystem(){let e=this.systemPrefix,n=this.userSystem,r=[];e&&e.length>0&&r.push(...e),n&&n.length>0&&r.push({type:"text",text:n});let o=tr(this.currentPermissionMode);return o!==null&&r.push(o),r.length===0?null:Xe()?Qn(r,Qe()):r}async interrupt(){let e=this.abortController;if(e&&!e.signal.aborted){e.abort("interrupted");return}this.pendingAbortReason="interrupted"}async setModel(e){e!==void 0&&e.length>0&&(this.currentModel=e)}async setPermissionMode(e){this.currentPermissionMode=e}setCwd(e){if(!this.cwdDependentsFactory)return;let{userSystem:n,dispatcher:r}=this.cwdDependentsFactory(e);this.userSystem=n,this.toolDispatcher=r}async supportedCommands(){try{return pn().map(n=>{let r={name:n.name,description:n.description};return n.argumentHint&&(r.argumentHint=n.argumentHint),r})}catch{return[]}}async supportedModels(){return el.map(e=>({...e}))}async supportedAgents(){return[]}async getContextUsage(){let e=this.lastUsage,n=Ko(this.currentModel),r;if(e&&n>0){let o=(e.inputTokens??0)+(e.outputTokens??0)+(e.cachedInputTokens??0)+(e.cacheCreationTokens??0);r=Math.min(100,Math.max(0,o/n*100))}return{tools:[],agents:[],isAutoCompactEnabled:!1,apiUsage:this.lastUsage,...r!==void 0?{percentage:r}:{},maxTokens:n}}async mcpServerStatus(){return[]}async accountInfo(){return{subscriptionType:this.authMode==="oauth"?"claude-subscription":"api-key"}}async rewindFiles(e,n){return{canRewind:!1,error:"anthropic-direct provider does not support file checkpoint rewind"}}async compact(){let e=this.messages.length;if(this.closed)return{compacted:!1,reason:"session-closed",messagesBefore:e,messagesAfter:e};if(this.abortController!==null)return{compacted:!1,reason:"turn-in-flight",messagesBefore:e,messagesAfter:e};let n=tl(),r=ar(this.messages,n);if(r<0)return{compacted:!1,reason:"history-too-short",messagesBefore:e,messagesAfter:e};if(r===0)return{compacted:!1,reason:"nothing-to-summarize",messagesBefore:e,messagesAfter:e};let o=this.messages.slice(0,r),s=nl(),i=cr(o,s,Zc),a=new AbortController;this.abortController=a,this.pendingAbortReason!==null&&!a.signal.aborted&&(a.abort(this.pendingAbortReason),this.pendingAbortReason=null);let c;try{if(a.signal.aborted)return{compacted:!1,reason:"aborted",messagesBefore:e,messagesAfter:e};let p=De(this.authMode,this.initSessionId,We()),g=this.client,b=await Promise.resolve(g.messages.create(i,{headers:p,signal:a.signal}));c=await rl(b)}catch(p){return a.signal.aborted?{compacted:!1,reason:"aborted",messagesBefore:e,messagesAfter:e}:{compacted:!1,reason:"summarization-failed: "+(p instanceof Error?p.message:String(p)),messagesBefore:e,messagesAfter:e}}finally{this.abortController===a&&(this.abortController=null)}if(c.trim().length===0)return{compacted:!1,reason:"empty-summary",messagesBefore:e,messagesAfter:e};let l=dr(this.messages,r,c),d=lr(this.messages,r,c);this.messages.splice(0,this.messages.length,...d);let u=this.messages.length;return jn(this.traceWriter,{trigger:"manual",preCompactionMessages:o,summary:c,keptTailCount:e-r,keepLastNConfig:n,messagesBefore:e,messagesAfter:u,tokensSavedEstimate:l}),{compacted:!0,messagesBefore:e,messagesAfter:u,tokensSavedEstimate:l}}close(){this.closed=!0;let e=this.abortController;e&&!e.signal.aborted?e.abort("closed"):this.pendingAbortReason="closed",this.closeResolve?.()}};function tl(){let t=process.env.AFK_COMPACT_KEEP_LAST_TURNS;if(t!==void 0&&t.length>0){let e=Number.parseInt(t,10);if(Number.isFinite(e)&&e>0)return e}return Xc}function nl(){let t=process.env.AFK_COMPACT_MODEL;return t!==void 0&&t.length>0?t:Qc}async function rl(t){let e="";for await(let n of t)if(n.type==="content_block_delta"){let r=n.delta;r.type==="text_delta"&&typeof r.text=="string"&&(e+=r.text)}return e}import fn from"path";import{appendFileSync as ol,mkdirSync as sl}from"fs";import{dirname as il}from"path";function Ge(t,e){return e?.allowedTools?e.allowedTools.includes(t)?{allowed:!0}:{allowed:!1,reason:`Tool "${t}" is not in the configured allowlist`}:{allowed:!0}}var al=new Set(["agent","compose","read_file","glob","grep","list_directory","memory_search"]);function qo(t){return al.has(t)}function cl(t,e){return t.reduce((n,r,o)=>{let s=e(r.name,r.input),i=n[n.length-1];return i&&s&&i.isConcurrencySafe?i.indices.push(o):n.push({isConcurrencySafe:s,indices:[o]}),n},[])}var ke=class{handlers;schemas;hookRegistry;permissions;subagentExecutor;skillExecutor;composeExecutor;classifier;resolveBase;_readRoots;_writeRoots;sessionId;traceWriter;constructor(e){this.handlers=e.handlers,this.schemas=e.schemas,this.hookRegistry=e.hookRegistry,this.permissions=e.permissions,this.subagentExecutor=e.subagentExecutor,this.skillExecutor=e.skillExecutor,this.composeExecutor=e.composeExecutor,this.classifier=e.concurrencyClassifier??qo,this.resolveBase=e.cwd,this.sessionId=e.sessionId,this.traceWriter=e.traceWriter;let n=e.cwd?[e.cwd]:[];this._readRoots=e.readRoots??n.slice(),this._writeRoots=e.writeRoots??n.slice()}get handlerContext(){return{cwd:this.resolveBase,resolveBase:this.resolveBase,readRoots:this._readRoots.slice(),writeRoots:this._writeRoots.slice()}}addReadRoot(e,n="slash"){let r=fn.resolve(e);this._readRoots.includes(r)||this._readRoots.push(r),this.appendAuditLog({action:"grant-read",path:r,source:n})}addWriteRoot(e,n="slash"){let r=fn.resolve(e);this._readRoots.includes(r)||this._readRoots.push(r),this._writeRoots.includes(r)||this._writeRoots.push(r),this.appendAuditLog({action:"grant-write",path:r,source:n})}revokeRoot(e,n="slash"){let r=fn.resolve(e);if(r===this.resolveBase)return;let o=this._readRoots.indexOf(r);o!==-1&&this._readRoots.splice(o,1);let s=this._writeRoots.indexOf(r);s!==-1&&this._writeRoots.splice(s,1),this.appendAuditLog({action:"revoke",path:r,source:n})}getGrants(){return{resolveBase:this.resolveBase,readRoots:this._readRoots.slice(),writeRoots:this._writeRoots.slice()}}appendAuditLog(e){try{let n=ot();sl(il(n),{recursive:!0});let r=JSON.stringify({timestamp:new Date().toISOString(),sessionId:this.sessionId??null,action:e.action,path:e.path,source:e.source});ol(n,r+`
|
|
1488
1488
|
`)}catch{}}get toolDefs(){return this.schemas}async execute(e){if(e.signal.aborted)return{content:"Tool call aborted",isError:!0};if(this.hookRegistry){let s={event:"PreToolUse",toolName:e.name,input:e.input};try{await Wt(this.hookRegistry,s,{signal:e.signal,...this.traceWriter?{traceWriter:this.traceWriter}:{}})}catch(i){if(i instanceof U)return{content:`Tool "${e.name}" blocked by PreToolUse hook: ${i.message}`,isError:!0};throw i}}let n=Ge(e.name,this.permissions);if(!n.allowed)return{content:n.reason??`Tool "${e.name}" is not permitted`,isError:!0};if(e.name==="agent"){if(!this.subagentExecutor)return{content:"Agent tool is not available in this session configuration",isError:!0};let s;try{s=await this.subagentExecutor.execute(e)}catch(i){s={content:`Agent tool error: ${i instanceof Error?i.message:String(i)}`,isError:!0}}return await this.dispatchPostToolUseAndAwait(e.name,s.content,e.signal),s}if(e.name==="skill"){if(!this.skillExecutor)return{content:"Skill tool is not available in this session configuration",isError:!0};let s;try{s=await this.skillExecutor.execute(e)}catch(i){s={content:`Skill tool error: ${i instanceof Error?i.message:String(i)}`,isError:!0}}return await this.dispatchPostToolUseAndAwait(e.name,s.content,e.signal),s}if(e.name==="compose"){let s=await this.executeCompose(e);return this.firePostToolUse(e.name,s.content,e.signal),s}let r=this.handlers.get(e.name);if(!r)return{content:`Unknown tool "${e.name}". Available tools: ${[...this.handlers.keys()].join(", ")}`,isError:!0};let o;try{o=await r(e.input,e.signal,this.handlerContext)}catch(s){o={content:`Tool execution error: ${s instanceof Error?s.message:String(s)}`,isError:!0}}return await this.dispatchPostToolUseAndAwait(e.name,o.content,e.signal),o}async executeBatch(e){if(e.length===0)return[];if(e.length===1)return[await this.execute(e[0])];let n=new Array(e.length),r=new Set;for(let i=0;i<e.length;i++){let a=e[i];if(a.signal.aborted){n[i]={content:"Tool call aborted",isError:!0},r.add(i);continue}if(this.hookRegistry){let l={event:"PreToolUse",toolName:a.name,input:a.input};try{await Wt(this.hookRegistry,l,{signal:a.signal,...this.traceWriter?{traceWriter:this.traceWriter}:{}})}catch(d){if(d instanceof U){n[i]={content:`Tool "${a.name}" blocked by PreToolUse hook: ${d.message}`,isError:!0},r.add(i);continue}throw d}}let c=Ge(a.name,this.permissions);c.allowed||(n[i]={content:c.reason??`Tool "${a.name}" is not permitted`,isError:!0},r.add(i))}let o=e.map((i,a)=>({call:i,originalIndex:a})).filter((i,a)=>!r.has(a));if(o.length===0)return n;let s=cl(o.map(i=>i.call),this.classifier);for(let i of s)if(i.isConcurrencySafe){let a=await Promise.allSettled(i.indices.map(async c=>{let{call:l,originalIndex:d}=o[c];return l.signal.aborted?{result:{content:"Tool call aborted",isError:!0},originalIndex:d}:{result:await this.executeCore(l),originalIndex:d}}));for(let c of a)if(c.status==="fulfilled")n[c.value.originalIndex]=c.value.result;else{let l=c.reason instanceof Error?c.reason.message:String(c.reason),d=i.indices[a.indexOf(c)];n[o[d].originalIndex]={content:`Tool execution error: ${l}`,isError:!0}}}else for(let a of i.indices){let{call:c,originalIndex:l}=o[a];if(c.signal.aborted){n[l]={content:"Tool call aborted",isError:!0};continue}n[l]=await this.executeCore(c)}return n}async executeCore(e){if(e.name==="agent"){if(!this.subagentExecutor)return{content:"Agent tool is not available in this session configuration",isError:!0};let o;try{o=await this.subagentExecutor.execute(e)}catch(s){o={content:`Agent tool error: ${s instanceof Error?s.message:String(s)}`,isError:!0}}return this.firePostToolUse(e.name,o.content,e.signal),o}if(e.name==="skill"){if(!this.skillExecutor)return{content:"Skill tool is not available in this session configuration",isError:!0};let o;try{o=await this.skillExecutor.execute(e)}catch(s){o={content:`Skill tool error: ${s instanceof Error?s.message:String(s)}`,isError:!0}}return this.firePostToolUse(e.name,o.content,e.signal),o}if(e.name==="compose"){let o=await this.executeCompose(e);return this.firePostToolUse(e.name,o.content,e.signal),o}let n=this.handlers.get(e.name);if(!n)return{content:`Unknown tool "${e.name}". Available tools: ${[...this.handlers.keys()].join(", ")}`,isError:!0};let r;try{r=await n(e.input,e.signal,this.handlerContext)}catch(o){r={content:`Tool execution error: ${o instanceof Error?o.message:String(o)}`,isError:!0}}return this.firePostToolUse(e.name,r.content,e.signal),r}async executeCompose(e){if(!this.composeExecutor)return{content:"Compose tool is not available in this session configuration",isError:!0};try{return await this.composeExecutor.execute(e)}catch(n){return{content:`Compose tool error: ${n instanceof Error?n.message:String(n)}`,isError:!0}}}firePostToolUse(e,n,r){if(!this.hookRegistry)return;let o={event:"PostToolUse",toolName:e,output:n};Gt(this.hookRegistry,o,{signal:r,...this.traceWriter?{traceWriter:this.traceWriter}:{}}).catch(()=>{})}async dispatchPostToolUseAndAwait(e,n,r){if(!this.hookRegistry)return;let o={event:"PostToolUse",toolName:e,output:n};await Gt(this.hookRegistry,o,{signal:r,...this.traceWriter?{traceWriter:this.traceWriter}:{}})}};import{spawn as ll}from"child_process";function dl(t){if(typeof t!="object"||t===null)throw new Error("Input must be an object");let e=t;if(typeof e.command!="string")throw new Error('Input must have a "command" field of type string');let n=12e4;if(e.timeout_ms!==void 0){if(typeof e.timeout_ms!="number")throw new Error("timeout_ms must be a number");if(e.timeout_ms<0||e.timeout_ms>6e5)throw new Error("timeout_ms must be between 0 and 600000");n=e.timeout_ms}return{command:e.command,timeout_ms:n}}function ul(t){return t.replace(/\x1b\[[0-9;]*[a-zA-Z]/g,"")}function bt(t,e){let n=!1;function r(){n||t==="bypassPermissions"&&(n=!0,console.warn("[security] bash handler: shell=true with bypassPermissions \u2014 all shell metacharacters are interpreted without confirmation. Migrate to execFile to eliminate this risk (tracked: C4)."))}return async(o,s,i)=>{let{command:a,timeout_ms:c}=dl(o);return s.aborted?{content:"Command aborted",isError:!0}:(r(),new Promise(l=>{let d=!1;function u(f){d||(d=!0,clearTimeout(g),s.removeEventListener("abort",S),l(f))}let p=ll(a,{shell:!0,stdio:["ignore","pipe","pipe"],...(i?.resolveBase??i?.cwd??e)!==void 0?{cwd:i?.resolveBase??i?.cwd??e}:{}}),g=setTimeout(()=>{p.kill(),u({content:`Command timed out after ${c}ms`,isError:!0})},c),b="",m="";p.stdout.on("data",f=>{b+=f.toString()}),p.stderr.on("data",f=>{m+=f.toString()});let S=()=>{p.kill(),u({content:"Command aborted",isError:!0})};s.addEventListener("abort",S),p.on("close",()=>{let f=(b+m).trimEnd();f=ul(f);let h=1e5;f.length>h&&(f=f.slice(0,h)+`
|
|
1489
1489
|
[output truncated \u2014 exceeded 100KB]`),u({content:f})}),p.on("error",f=>{u({content:`Failed to execute: ${f.message}`,isError:!0})})}))}}var zo=bt("default");import{promises as pl}from"fs";import mn from"path";function V(t,e,n="read"){let r=e?.resolveBase??e?.cwd,o=mn.isAbsolute(t)?t:mn.resolve(r??process.cwd(),t);if(r===void 0)return o;let s=n==="read"?e?.readRoots??[r]:e?.writeRoots??[r];for(let c of s)if(!mn.relative(c,o).startsWith(".."))return o;let i=s.map(c=>`\`${c}\``).join(", "),a=n==="read"?"read roots":"write roots";throw new Error(`Path \`${t}\` is outside the allowed ${a} [${i}].`)}var Vo=async(t,e,n)=>{if(!t||typeof t!="object")return{content:"Invalid input: expected an object",isError:!0};let r=t,o=r.file_path,s=r.offset??1,i=r.limit??2e3;if(typeof o!="string")return{content:"Invalid input: file_path must be a string",isError:!0};if(typeof s!="number"||s<1)return{content:"Invalid input: offset must be a positive number",isError:!0};if(typeof i!="number"||i<1)return{content:"Invalid input: limit must be a positive number",isError:!0};let a;try{a=V(o,n,"read")}catch(c){return{content:c instanceof Error?c.message:String(c),isError:!0}}try{let c=await pl.readFile(a),l=Math.min(8192,c.length);for(let h=0;h<l;h++)if(c[h]===0)return{content:`File appears to be binary: ${a}`,isError:!0};let d=c.toString("utf-8");if(d.length===0)return{content:""};let u=d.split(`
|
|
1490
1490
|
`),p=Math.max(0,s-1),g=Math.min(u.length,p+i),b=u.slice(p,g),m=u.length;if(b.length===0)return{content:`... (offset ${s} is past end of file \u2014 file has ${m} lines)`};let S=String(m).length,f=b.map((h,y)=>{let k=p+y+1;return`${String(k).padStart(S," ")} ${h}`}).join(`
|
|
@@ -1554,8 +1554,8 @@ Save reusable multi-step workflows the user teaches you or that you discover wor
|
|
|
1554
1554
|
`+(t.aggregated_output??"");yield{type:"tool.output",toolUseId:t.id,content:a,isError:s,...n!==void 0?{sessionId:n}:{}}}return}if(t.type==="file_change"){if(e){let s=t.changes.map(i=>`${i.kind} ${i.path}`).join(`
|
|
1555
1555
|
`);yield{type:"tool.output",toolUseId:t.id,content:s,isError:t.status==="failed",...n!==void 0?{sessionId:n}:{}}}return}if(t.type==="mcp_tool_call"){if(e){let s=t.status==="failed",i;t.error?i=t.error.message:t.result?i=JSON.stringify(t.result,null,2):i="",yield{type:"tool.output",toolUseId:t.id,content:i,isError:s,...n!==void 0?{sessionId:n}:{}}}return}if(t.type==="web_search"){e&&(yield{type:"tool.output",toolUseId:t.id,content:`web_search: ${t.query}`,...n!==void 0?{sessionId:n}:{}});return}if(t.type!=="todo_list"&&t.type==="error"){e&&(yield{type:"error",error:new Error(t.message)});return}}var xn=class{startOpts;promptStream;codex;thread;currentModel;currentSandbox;currentApproval;abortController=null;pendingAbortReason=null;closed=!1;initSessionId;dispose;closeResolve=null;closedPromise;constructor(e,n,r){this.startOpts=e,this.promptStream=n,this.initSessionId=r,this.codex=new ps(e.codexOptions),this.thread=e.resumeId?this.codex.resumeThread(e.resumeId,e.threadOptions):this.codex.startThread(e.threadOptions),this.currentModel=e.threadOptions.model,this.currentSandbox=e.threadOptions.sandboxMode??"workspace-write",this.currentApproval=e.threadOptions.approvalPolicy??"never",e.instructionsDispose!==void 0&&(this.dispose=e.instructionsDispose),this.closedPromise=new Promise(o=>{this.closeResolve=()=>o("__closed__")})}async*[Symbol.asyncIterator](){yield{type:"session.init",info:{sessionId:this.initSessionId,...this.currentModel!==void 0?{model:this.currentModel}:{},permissionMode:this.sandboxToPermissionMode(),cwd:this.startOpts.threadOptions.workingDirectory??process.cwd(),tools:["Bash","Read","Write","Edit"],slashCommands:[],skills:[],plugins:[],mcpServers:[],apiKeySource:this.startOpts.codexOptions.apiKey!==void 0?"apiKey":"codex-cli",version:"codex-sdk"}};let n=new Map,r=new Map,o=this.initSessionId,s=this.promptStream[Symbol.asyncIterator]();try{for(;!this.closed;){let i=await Promise.race([s.next(),this.closedPromise]);if(i==="__closed__")break;let a=i;if(a.done)break;let c=a.value,l=new AbortController;if(this.abortController=l,this.pendingAbortReason!==null&&!l.signal.aborted&&(l.abort(this.pendingAbortReason),this.pendingAbortReason=null),l.signal.aborted)return;let d;try{let u=typeof c.content=="string"?c.content:c.content.map(p=>{if(typeof p=="object"&&p&&"type"in p){if(p.type==="text")return p.text;if(p.type==="image")return"[image omitted]"}return""}).join(`
|
|
1556
1556
|
`);d=await this.thread.runStreamed(u,{signal:l.signal})}catch(u){if(l.signal.aborted)return;yield{type:"error",error:u instanceof Error?u:new Error(String(u))};return}try{for await(let u of d.events){if(this.closed)return;u.type==="thread.started"&&(o=u.thread_id),yield*wd(u,o,n,r)}}catch(u){if(l.signal.aborted)return;yield{type:"error",error:u instanceof Error?u:new Error(String(u))};return}finally{this.abortController===l&&(this.abortController=null)}}}catch(i){yield{type:"error",error:i instanceof Error?i:new Error(String(i))}}finally{try{await s.return?.()}catch{}}}sandboxToPermissionMode(){return this.currentSandbox==="read-only"||this.currentApproval==="untrusted"?"plan":"bypassPermissions"}async interrupt(){let e=this.abortController;if(e&&!e.signal.aborted){e.abort("interrupted");return}this.pendingAbortReason="interrupted"}async setModel(e){this.currentModel=e;let n={...this.startOpts.threadOptions,...e!==void 0?{model:e}:{},sandboxMode:this.currentSandbox,approvalPolicy:this.currentApproval},r=this.thread.id;this.thread=r?this.codex.resumeThread(r,n):this.codex.startThread(n),this.startOpts.threadOptions=n}async setPermissionMode(e){let{sandboxMode:n,approvalPolicy:r}=fs(e);this.currentSandbox=n,this.currentApproval=r;let o={...this.startOpts.threadOptions,sandboxMode:n,approvalPolicy:r},s=this.thread.id;this.thread=s?this.codex.resumeThread(s,o):this.codex.startThread(o),this.startOpts.threadOptions=o}async supportedCommands(){return[]}async supportedModels(){return fd.map(e=>({...e}))}async supportedAgents(){return[]}async getContextUsage(){return{tools:[],agents:[],isAutoCompactEnabled:!1,apiUsage:null}}async mcpServerStatus(){return[]}async accountInfo(){return{}}async rewindFiles(e,n){throw new Me(Ke,"rewindFiles",`${Ke} provider does not support file checkpoint rewind.`)}close(){this.closed=!0;let e=this.abortController;e&&!e.signal.aborted?e.abort("closed"):this.pendingAbortReason="closed",this.closeResolve?.(),this.dispose?.()}getThread(){return this.thread}},vd=null;var Re=class{name=Ke;query(e){md(e.config);let n=bd(e.config),r=hd(e.config.effort),{sandboxMode:o,approvalPolicy:s}=fs(e.config.permissionMode),i={...e.config.model!==void 0?{model:e.config.model}:{},sandboxMode:o,approvalPolicy:s,...r!==void 0?{modelReasoningEffort:r}:{},skipGitRepoCheck:!0,workingDirectory:process.cwd()},a=gd(e.config),c={};n!==void 0&&(c.apiKey=n);let l;if(a!==void 0){let{path:b,dispose:m}=yd(a);c.config={...c.config??{},model_instructions_file:b},l=m}_(`\u{1F7E2} OpenAICodexProvider: creating Codex thread (model=${String(e.config.model)}, sandbox=${o}, approval=${s})`);let d=vd,u=d??(b=>new ps(b)),p=e.config.resume??`codex-pending-${Date.now()}-${Math.random().toString(36).slice(2,8)}`,g=new xn({threadOptions:i,codexOptions:c,...e.config.resume!==void 0?{resumeId:e.config.resume}:{},...l!==void 0?{instructionsDispose:l}:{}},e.prompt,p);if(d){let b=u(c);g.codex=b,g.thread=e.config.resume?b.resumeThread(e.config.resume,i):b.startThread(i)}return g}},Tt=new Re;var Sd=new Set(["opus","opus_1m","sonnet","sonnet_1m","haiku","auto"]);function Be(t){if(!t)return"anthropic-direct";let e=t.trim().toLowerCase();return!e||Sd.has(e)||e.startsWith("claude-")||e.startsWith("claude_")?"anthropic-direct":e.startsWith("gpt-")||e.startsWith("gpt_")||e.startsWith("o1")||e.startsWith("o3")||e.startsWith("o4")||e.startsWith("codex-")||e.startsWith("codex_")||e==="codex"?"openai-codex":"anthropic-direct"}function Et(t){return Be(t)==="openai-codex"?Tt:new we}async function Pt(t,e,n){if(!t)return;if(n.kind==="blocked"){await Se(t,{hookEvent:e,decision:"block",...n.err.reason!==void 0?{reason:n.err.reason}:{}});return}let r=n.decision;await Se(t,{hookEvent:e,decision:r.decision,...r.reason!==void 0?{reason:r.reason}:{},...r.injectContext!==void 0?{injectedContextBytes:Buffer.byteLength(r.injectContext,"utf8")}:{}})}async function ms(t,e,n={}){if(t)try{let r=await t.dispatch(e,n.signal);await Pt(n.traceWriter,"SessionStart",{kind:"decision",decision:r})}catch(r){throw r instanceof U&&await Pt(n.traceWriter,"SessionStart",{kind:"blocked",err:r}),r}}async function hs(t,e,n={}){if(t)try{let r=await t.dispatch(e,n.signal);await Pt(n.traceWriter,"SessionEnd",{kind:"decision",decision:r})}catch(r){if(r instanceof U&&await Pt(n.traceWriter,"SessionEnd",{kind:"blocked",err:r}),r instanceof U||r instanceof Y){_(`SessionEnd hook swallowed ${r.name}: ${r.message}`),n.onError?.(r);return}_(`SessionEnd hook unexpected error: ${String(r)}`),n.onError?.(r instanceof Error?r:new Error(String(r)))}}var _t=class{pendingResolve=null;bufferedMessage=null;getSessionId;constructor(e){this.getSessionId=e}pushUserMessage(e){if(this.pendingResolve){let n=this.pendingResolve;this.pendingResolve=null;let r=this.getSessionId();n({content:e,...r!==void 0?{sessionId:r}:{}});return}this.bufferedMessage=e}createIterable(){let e=this;return{[Symbol.asyncIterator](){return{next(){if(e.bufferedMessage!==null){let n=e.bufferedMessage;e.bufferedMessage=null;let r=e.getSessionId();return Promise.resolve({value:{content:n,...r!==void 0?{sessionId:r}:{}},done:!1})}return new Promise(n=>{e.pendingResolve=r=>n({value:r,done:!1})})},return(){return Promise.resolve({value:void 0,done:!0})}}}}}};function gs(t,e,n){t&&(t.aborted?e.abort(t.reason):t.addEventListener("abort",()=>{e.signal.aborted||e.abort(t.reason)},{once:!0})),e.signal.addEventListener("abort",n,{once:!0})}function ys(t,e){let n=t.permissionMode??"default",r=t.persistSession??!0,o={sessionId:t.sessionId,configuredSessionId:t.sessionId,resume:t.resume,resumeSessionAt:t.resumeSessionAt,continue:t.continue,forkSession:t.forkSession,persistSession:r},s={sessionId:t.sessionId,model:e,permissionMode:n};return{sessionIdentity:o,metadata:s}}var At=class{initializationPromise;resolveInitialization;rejectInitialization;initializationSettled=!1;sessionMetadata;sessionIdentity;constructor(e,n){this.sessionIdentity=e,this.sessionMetadata=n,this.initializationPromise=new Promise((r,o)=>{this.resolveInitialization=r,this.rejectInitialization=o})}waitForInitialization(){return this.initializationPromise}getSessionIdentity(){return{...this.sessionIdentity,sessionId:this.getSessionId()}}getSessionMetadata(){return{...this.sessionMetadata,sessionId:this.getSessionId()}}getSessionId(){return this.sessionMetadata.sessionId??this.sessionIdentity.sessionId}updateSessionIdentity(e){e&&(this.sessionIdentity={...this.sessionIdentity,sessionId:e},this.sessionMetadata={...this.sessionMetadata,sessionId:e})}setSessionMetadata(e){this.sessionMetadata=e(this.sessionMetadata)}resolveInitializationIfNeeded(){this.initializationSettled||(this.initializationSettled=!0,this.resolveInitialization(this.getSessionMetadata()))}resolveInitializationOnce(){this.initializationSettled||(this.initializationSettled=!0,this.resolveInitialization(this.getSessionMetadata()))}rejectInitializationOnce(e){this.initializationSettled||(this.initializationSettled=!0,this.rejectInitialization(e))}isInitializationSettled(){return this.initializationSettled}};function bs(t){try{let e=JSON.parse(t);if(!Array.isArray(e))return null;let n=e.length;if(n===0)return"no results";let r=0,o=0;for(let a of e)if(a&&typeof a=="object"){let c=a.type;c==="fact"?r++:c==="procedure"&&o++}let s=`${n} result${n===1?"":"s"}`;if(r+o!==n)return s;let i=[];return r>0&&i.push(`${r} fact${r===1?"":"s"}`),o>0&&i.push(`${o} procedure${o===1?"":"s"}`),i.length===0?s:`${s} (${i.join(", ")})`}catch{return null}}function ws(t){try{let e=JSON.parse(t);if(!e||typeof e!="object")return null;let n=e;if(n.target==="hot"&&n.saved===!0)return"hot memory saved";if(n.target==="fact"){if(n.action==="remove")return n.removed===!0?"fact removed":"fact not found";if(n.action==="set"&&typeof n.id=="number")return`fact #${n.id} set`;if(n.action==="supersede"&&typeof n.id=="number"&&typeof n.supersedes=="number")return`fact #${n.id} supersedes #${n.supersedes}`}return null}catch{return null}}function ks(t){try{let e=JSON.parse(t);if(!e||typeof e!="object")return null;let n=e;return typeof n.name=="string"&&n.written===!0?`wrote procedure '${n.name}'`:null}catch{return null}}function Tn(t){let e=t.trim();if(e.length===0)return null;let n=e[0];if(n!=="{"&&n!=="[")return null;let r=e[e.length-1];if(n==="{"&&r!=="}"||n==="["&&r!=="]")return null;let o;try{o=JSON.parse(e)}catch{return null}return Array.isArray(o)?vs(xd(o)):o!==null&&typeof o=="object"?vs(Td(o)):null}function xd(t){return t.length===0?"[empty array]":t.length===1?"[1 item]":`[${t.length} items]`}function Td(t){let e=Object.keys(t);if(e.length===0)return"{empty object}";let n=e.slice(0,4),r=e.length>4?", \u2026":"";return`{${n.join(", ")}${r}}`}function vs(t){return t.length<=80?t:t.slice(0,79)+"\u2026"}var Ed=new Map([["memory_search",bs],["memory_update",ws],["procedure_write",ks],["bash",Tn],["Bash",Tn]]);function Pd(t){return t.replace(/\x1b\[[0-?]*[ -/]*[@-~]/g,"").replace(/\x1b[@-_]/g,"").replace(/[\x00-\x1f\x7f]/g," ").trim()}function Ss(t,e){if(!t)return null;let n=Ed.get(t);if(!n)return null;try{let r=n(e);if(r===null)return null;let o=Pd(r);return o.length>0?o:null}catch{return null}}function _d(t){let e=/Output too large \((\d+(?:\.\d+)?)\s*(B|KB|MB|GB)\)\.\s*Full output saved to:\s*(\/[^\n]+)/,n=t.match(e);if(!n||!n[1]||!n[2]||!n[3])return null;let r=n[1],o=n[2],s=n[3],i=parseFloat(r),a=i;o==="KB"?a=i*1024:o==="MB"?a=i*1024*1024:o==="GB"&&(a=i*1024*1024*1024);let c=r;return i%1===0&&(c=String(Math.floor(i))),c+=o,{sizeLabel:c,sizeBytes:Math.round(a),absolutePath:s.trim()}}function Ad(t){if(t<1024)return`${t}B`;let e=t/1024;if(e<1024)return e%1===0?`${Math.floor(e)}KB`:`${e.toFixed(1)}KB`;let n=e/1024;if(n<1024)return n%1===0?`${Math.floor(n)}MB`:`${n.toFixed(1)}MB`;let r=n/1024;return r%1===0?`${Math.floor(r)}GB`:`${r.toFixed(1)}GB`}function Rd(t){let e=Buffer.byteLength(t,"utf8"),n=Ad(e),r=t.split(`
|
|
1557
|
-
`);if(r.length<=1&&t.length<=80)return{content:t,truncated:!1,sizeBytes:e,sizeLabel:n};if(r.length<=1)return t.length<=80?{content:t,truncated:!1,sizeBytes:e,sizeLabel:n}:{content:t.substring(0,80)+"\u2026",truncated:!0,sizeBytes:e,sizeLabel:n};if(t.length<=80)return{content:t,truncated:!1,sizeBytes:e,sizeLabel:n};let o=r[0]??"",s=o;return o.length>80&&(s=o.substring(0,80)+"\u2026"),{content:s+`\u2026+${r.length} lines`,truncated:!0,lineCount:r.length,sizeBytes:e,sizeLabel:n}}function Cd(t,e){let n={...t.raw??{}};return t.inputTokens!==void 0&&(n.input_tokens=t.inputTokens),t.outputTokens!==void 0&&(n.output_tokens=t.outputTokens),t.cachedInputTokens!==void 0&&(n.cache_read_input_tokens=t.cachedInputTokens),t.cacheCreationTokens!==void 0&&(n.cache_creation_input_tokens=t.cacheCreationTokens),t.totalTokens!==void 0&&(n.total_tokens=t.totalTokens),{sessionId:e,stopReason:t.stopReason??void 0,resultSubtype:t.resultSubtype,durationMs:t.durationMs,durationApiMs:t.durationApiMs,totalCostUsd:t.totalCostUsd,isError:t.isError,usage:Object.keys(n).length>0?n:void 0,modelUsage:t.modelUsage,permissionDenials:t.permissionDenials,errors:t.errors}}function Id(t){let e=t.isError===!0?null:Ss(t.toolName,t.content),n=e!==null?{display:e}:{},r=_d(t.content);if(r)return{type:"chunk",chunk:{type:"tool_result",toolUseId:t.toolUseId,content:`Output persisted (${r.sizeLabel}) \u2192 ${r.absolutePath}`,isError:t.isError===!0,persistedPath:r.absolutePath,sizeBytes:r.sizeBytes,sizeLabel:r.sizeLabel,...n}};let{content:o,truncated:s,lineCount:i,sizeBytes:a,sizeLabel:c}=Rd(t.content);return{type:"chunk",chunk:{type:"tool_result",toolUseId:t.toolUseId,content:o,isError:t.isError===!0,sizeBytes:a,sizeLabel:c,...s&&{truncated:s},...i!==void 0&&{lineCount:i},...n}}}function En(t,e){switch(t.type){case"session.init":{let n=t.info;return e.setSessionMetadata(r=>({...r,sessionId:n.sessionId,model:n.model??r.model,...n.permissionMode!==void 0?{permissionMode:n.permissionMode}:{},...n.cwd!==void 0?{cwd:n.cwd}:{},tools:n.tools?[...n.tools]:r.tools,slashCommands:n.slashCommands?[...n.slashCommands]:r.slashCommands,skills:n.skills?[...n.skills]:r.skills,plugins:n.plugins?n.plugins.map(o=>({...o})):r.plugins,mcpServers:n.mcpServers?n.mcpServers.map(o=>({...o})):r.mcpServers,...n.apiKeySource!==void 0?{apiKeySource:n.apiKeySource}:{},...n.version!==void 0?{claudeCodeVersion:n.version}:{},...n.outputStyle!==void 0?{outputStyle:n.outputStyle}:{}})),e.updateSessionIdentity(n.sessionId),e.resolveInitialization(),null}case"session.status":return e.setSessionMetadata(n=>({...n,sessionId:t.sessionId,...t.permissionMode!==void 0?{permissionMode:t.permissionMode}:{permissionMode:n.permissionMode},...t.status!==void 0?{status:t.status}:{}})),null;case"delta.text":return{type:"chunk",chunk:{type:"content",content:t.text,metadata:{eventType:"delta",deltaType:"text_delta"}}};case"delta.reasoning":return{type:"chunk",chunk:{type:"thinking",content:t.text,metadata:{eventType:"delta",deltaType:"thinking_delta"}}};case"assistant.message":if(t.sessionId&&e.updateSessionIdentity(t.sessionId),t.text){let n={role:"assistant",content:t.text,timestamp:new Date};return e.conversationHistory.push(n),{type:"message",message:n}}return null;case"tool.use.start":return{type:"chunk",chunk:{type:"tool_use_detail",toolUseId:t.toolUseId,toolName:t.toolName,toolInput:t.toolInput}};case"tool.use":return{type:"chunk",chunk:{type:"tool_use",content:t.summary,metadata:{eventType:"tool_use_summary",precedingToolUseIds:t.toolUseIds}}};case"tool.output":return Id(t);case"progress":return{type:"progress",progress:{taskId:t.progress.taskId,description:t.progress.description,...t.progress.summary!==void 0?{summary:t.progress.summary}:{},...t.progress.lastToolName!==void 0?{lastToolName:t.progress.lastToolName}:{},totalTokens:t.progress.totalTokens,toolUses:t.progress.toolUses,durationMs:t.progress.durationMs}};case"suggestion":return{type:"suggestion",suggestion:t.suggestion};case"turn.completed":{let n=Cd(t.usage,t.sessionId??e.getSessionMetadata().sessionId);e.setLastResponseMetadata(n);for(let r=e.conversationHistory.length-1;r>=0;r--){let o=e.conversationHistory[r];if(o?.role==="assistant"){o.metadata=n;break}}if(e.maxBudgetUsd!==void 0&&e.abortBudget!==void 0&&typeof n.totalCostUsd=="number"&&(e._runningCostUsd=(e._runningCostUsd??0)+n.totalCostUsd,e._runningCostUsd>=e.maxBudgetUsd)){Nn(e.traceWriter,{kind:"monetary",runningCostUsd:e._runningCostUsd,maxBudgetUsd:e.maxBudgetUsd,lastTurnCostUsd:n.totalCostUsd});let r=new ve(e._runningCostUsd,e.maxBudgetUsd);return e.abortBudget(r.message),{type:"error",error:r}}return{type:"done",metadata:n}}case"error":return{type:"error",error:t.error};case"paused":return{type:"paused",reason:t.reason,...t.resetsAt!==void 0?{resetsAt:t.resetsAt}:{},...t.accountId!==void 0?{accountId:t.accountId}:{}};case"resumed":return{type:"resumed",hotSwapped:t.hotSwapped,...t.accountId!==void 0?{accountId:t.accountId}:{}};default:return null}}var ye=class{config;currentState="idle";providerQuery;providerIterator;conversationHistory=[];turnCount=0;lastResponseMetadata=null;initPromise=null;inputStream;abortController;hookRegistry;sessionEndDispatched=!1;stateManager;sessionRunningCostUsd=0;sessionRunningTokens={input:0,output:0,cacheRead:0,cacheCreation:0};lastStopReason;constructor(e){this.config=e,this.abortController=new AbortController,this.hookRegistry=e.hookRegistry,gs(e.abortSignal,this.abortController,()=>{this.onAbort()}),this.initSdkLifecycle()}initSdkLifecycle(){let e=He(this.config.model)??this.config.model,{sessionIdentity:n,metadata:r}=ys(this.config,e);this.stateManager=new At(n,r),this.inputStream=new _t(()=>this.sessionId);let o=this.config.provider??Et(e);_(`\u{1F7E2} AgentSession: Creating query session via provider=${o.name}`),this.providerQuery=o.query({prompt:this.inputStream.createIterable(),config:this.config}),this.conversationHistory=[],this.turnCount=0,this.lastResponseMetadata=null,this.sessionRunningCostUsd=0,this.sessionRunningTokens={input:0,output:0,cacheRead:0,cacheCreation:0},this.lastStopReason=void 0,this.sessionEndDispatched=!1,this.currentState="idle";let s=this.providerQuery;this.providerIterator=s[Symbol.asyncIterator](),this.initPromise=this.pullInitialization()}async pullInitialization(){try{for(await ms(this.hookRegistry,{event:"SessionStart",sessionId:this.sessionId},{signal:this.abortController.signal,...this.config.traceWriter?{traceWriter:this.config.traceWriter}:{}});;){let e=await this.providerIterator.next();if(e.done){this.stateManager.resolveInitializationIfNeeded();return}let n=e.value,r=En(n,this.buildTransformDeps());if(n.type==="session.init"||r&&r.type==="error")return}}catch(e){let n=e instanceof Error?e:new Error(String(e));this.stateManager.isInitializationSettled()||this.stateManager.rejectInitializationOnce(n),await this.dispatchSessionEndOnce("error").catch(()=>{})}}buildTransformDeps(){return{conversationHistory:this.conversationHistory,getSessionMetadata:()=>this.stateManager.getSessionMetadata(),setSessionMetadata:e=>this.stateManager.setSessionMetadata(e),updateSessionIdentity:e=>this.stateManager.updateSessionIdentity(e),resolveInitialization:()=>this.stateManager.resolveInitializationOnce(),setLastResponseMetadata:e=>{this.lastResponseMetadata=e,typeof e.totalCostUsd=="number"&&Number.isFinite(e.totalCostUsd)&&(this.sessionRunningCostUsd+=e.totalCostUsd);let n=e.usage;if(n&&typeof n=="object"){let r=n,o=(s,i)=>{let a=r[s];typeof a=="number"&&Number.isFinite(a)&&(this.sessionRunningTokens[i]+=a)};o("input_tokens","input"),o("output_tokens","output"),o("cache_read_input_tokens","cacheRead"),o("cache_creation_input_tokens","cacheCreation")}typeof e.stopReason=="string"&&(this.lastStopReason=e.stopReason)},maxBudgetUsd:this.config.maxBudgetUsd,abortBudget:e=>{this.abortController.signal.aborted||this.abortController.abort(e)},...this.config.traceWriter?{traceWriter:this.config.traceWriter}:{}}}get state(){return this.currentState}get sessionId(){return this.stateManager.getSessionId()}get cwd(){return this.config.cwd}get abortSignal(){return this.abortController.signal}async sendMessage(e,n={}){this.assertCanSend();let r=this.config.timeoutMs??Fe,o=async()=>{let s=null,i="";this.currentState=n.stream?"streaming":"processing";for await(let a of this.sendMessageStreamInternal(e)){if(a.type==="chunk"&&a.chunk.type==="content"&&(i+=a.chunk.content),a.type==="message"&&a.message.role==="assistant"&&(s=a.message),a.type==="error")throw a.error;if(a.type==="done"){if(s)return{...s,metadata:a.metadata};if(i)return{role:"assistant",content:i,metadata:a.metadata,timestamp:new Date}}}if(s)return s;if(i)return{role:"assistant",content:i,timestamp:new Date};throw new Error("No assistant response received")};try{return await $e(o(),r,{controller:this.abortController,label:this.sessionId??"session"})}finally{this.state!=="closed"&&(this.currentState="idle")}}async*sendMessageStream(e){this.assertCanSend(),this.currentState="streaming",yield*this.sendMessageStreamInternal(e)}async*sendMessageStreamInternal(e){this.initPromise&&await this.initPromise;let r={role:"user",content:typeof e=="string"?e:this.summarizeContentBlocks(e),timestamp:new Date};this.conversationHistory.push(r),this.inputStream.pushUserMessage(e);let o=this.buildTransformDeps();try{for(;;){let s=await this.providerIterator.next();if(s.done)break;let i=s.value,a=En(i,o);if(a&&(a.type==="done"&&this.turnCount++,yield a,a.type==="done"||a.type==="error"))break}}finally{this.state!=="closed"&&(this.currentState="idle")}}summarizeContentBlocks(e){let n=[],r=0;for(let s of e)s.type==="text"?n.push(s.text):s.type==="image"&&r++;let o=n.join(" ");return r>0&&(o=o?`${o} [+ ${r} image(s)]`:`[+ ${r} image(s)]`),o||"[content block(s)]"}async interrupt(){this.currentState!=="streaming"&&this.currentState!=="processing"||(this.currentState="idle",await this.providerQuery.interrupt())}async reset(){if(this.currentState==="closed")throw new Error("Cannot reset: session is closed");if(this.abortController.signal.aborted)throw new Y("Cannot reset: session aborted");if(this.currentState==="processing"||this.currentState==="streaming")try{await this.providerQuery.interrupt()}catch{}await this.dispatchSessionEndOnce("reset");try{await this.providerQuery.close()}catch{}await this.providerIterator.return?.(),this.initPromise&&await Promise.race([this.initPromise,new Promise(e=>setTimeout(e,et))]).catch(()=>{}),this.stateManager.resolveInitializationIfNeeded();try{this.initSdkLifecycle()}catch(e){throw this.currentState="closed",new Error(`Session reset failed during lifecycle rebuild: ${e instanceof Error?e.message:String(e)}`,{cause:e})}}async onAbort(){try{await this.providerQuery.interrupt()}catch{}}async setModel(e){let n=He(e),r=this.stateManager.getSessionMetadata();await this.providerQuery.setModel(n??r.model??""),n&&this.stateManager.setSessionMetadata(o=>({...o,model:n}))}async setPermissionMode(e){await this.providerQuery.setPermissionMode(e),this.stateManager.setSessionMetadata(n=>({...n,permissionMode:e}))}setCwd(e){this.config={...this.config,cwd:e},this.providerQuery.setCwd?.(e)}waitForInitialization(){return this.stateManager.waitForInitialization()}getSessionIdentity(){return this.stateManager.getSessionIdentity()}getSessionMetadata(){return this.stateManager.getSessionMetadata()}getQuery(){return this.providerQuery}supportedCommands(){return this.providerQuery.supportedCommands()}supportedModels(){return this.providerQuery.supportedModels()}supportedAgents(){return this.providerQuery.supportedAgents()}getContextUsage(){return this.providerQuery.getContextUsage()}mcpServerStatus(){return this.providerQuery.mcpServerStatus()}accountInfo(){return this.providerQuery.accountInfo()}rewindFiles(e,n){return this.providerQuery.rewindFiles(e,n)}async compact(){if(this.currentState==="closed")throw new Error("Cannot compact: session is closed");if(this.currentState!=="idle")return{compacted:!1,reason:"session-busy",messagesBefore:0,messagesAfter:0};let e=this.providerQuery.compact?.bind(this.providerQuery);return e?e():{compacted:!1,reason:"not-supported",messagesBefore:0,messagesAfter:0}}getLastResponseMetadata(){return this.lastResponseMetadata}getOutputStream(){throw new Error("getOutputStream() is not supported \u2014 use sendMessageStream() instead")}getInputStreamRef(){return{pushUserMessage:e=>this.inputStream.pushUserMessage(e)}}getHistory(){return[...this.conversationHistory]}getTurnCount(){return this.turnCount}async close(){if(this.currentState!=="closed"){this.currentState="closed",this.abortController.signal.aborted||this.abortController.abort("closed"),this.stateManager.resolveInitializationIfNeeded();try{this.providerQuery.close()}catch{}if(await this.providerIterator.return?.(),this.initPromise)try{await Promise.race([this.initPromise,new Promise(e=>setTimeout(e,et))])}catch{}await this.dispatchSessionEndOnce("close")}}async dispatchSessionEndOnce(e){this.sessionEndDispatched||(this.sessionEndDispatched=!0,await this.emitClosure(e).catch(()=>{}),await this.sealTraceWriter(e).catch(()=>{}),await hs(this.hookRegistry,{event:"SessionEnd",sessionId:this.sessionId,reason:e},this.config.traceWriter?{traceWriter:this.config.traceWriter}:{}))}async emitClosure(e){let n=this.config.traceWriter;if(!n)return;let r=this.deriveClosureReason(e),o={};this.sessionRunningTokens.input>0&&(o.input=this.sessionRunningTokens.input),this.sessionRunningTokens.output>0&&(o.output=this.sessionRunningTokens.output),this.sessionRunningTokens.cacheRead>0&&(o.cacheRead=this.sessionRunningTokens.cacheRead),this.sessionRunningTokens.cacheCreation>0&&(o.cacheCreation=this.sessionRunningTokens.cacheCreation),await Hn(n,{reason:r,finalTurnCount:this.turnCount,finalCostUsd:this.sessionRunningCostUsd,finalTokens:o,...this.lastStopReason!==void 0?{lastStopReason:this.lastStopReason}:{}})}deriveClosureReason(e){if(e==="error")return"abort";let n=this.abortController.signal;if(n.aborted&&n.reason!=="closed"){let r=n.reason;if(r instanceof ve)return"budget_exceeded";if(r instanceof ie)return"timeout";if(typeof r=="string"){if(r.startsWith("Budget "))return"budget_exceeded";if(r.includes("timed out"))return"timeout"}return"abort"}return"model_end_turn"}async sealTraceWriter(e){let n=this.config.traceWriter;if(!n)return;let r=this.deriveSealStatus(e);await n.seal({status:r,finalCostUsd:this.sessionRunningCostUsd,finalTurnCount:this.turnCount,closedAt:new Date().toISOString()})}deriveSealStatus(e){if(e==="error")return"failed";let n=this.abortController.signal;return n.aborted&&n.reason!=="closed"?"cancelled":"succeeded"}assertCanSend(){if(this.currentState==="closed")throw new Error("Cannot send message: session is closed");if(this.abortController.signal.aborted)throw new Y("Cannot send message: session aborted");if(this.currentState==="processing"||this.currentState==="streaming")throw new Error("Cannot send message: session is busy");if(this.config.maxTurns&&this.turnCount>=this.config.maxTurns)throw new Error(`Maximum turns (${this.config.maxTurns}) exceeded`)}};function Md(t){let e=new Map,n=t.defaultMode??"ask";if(t.tools)for(let[r,o]of Object.entries(t.tools))typeof o=="string"?e.set(r,{mode:o}):e.set(r,{mode:o.mode,reason:o.reason});if(t.list)for(let r of t.list)e.set(r.tool,{mode:r.mode,reason:r.reason});return{defaultMode:n,byTool:e}}function Dd(t){return t.behavior==="allow"?{behavior:"allow"}:{behavior:"deny",message:t.reason??"Tool denied by permission rules"}}function xs(t){let{defaultMode:e,byTool:n}=Md(t.rules);return async(r,o)=>{let s={toolName:r,input:o},i=n.get(r),a=i?.mode??e,c=i?.reason,l;return a==="ask"&&t.onAsk?l=await t.onAsk(s):a==="deny"?l={behavior:"deny",reason:c}:l={behavior:"allow"},t.onDecision?.(s,l),Dd(l)}}import{Telegraf as hu}from"telegraf";import{promises as qe}from"fs";import{join as Ts}from"path";var Ce=class{sessions=new Map;pendingSessions=new Map;sessionData=new Map;options;constructor(e){this.options={dataDir:e.dataDir||"./data/telegram-sessions",defaultModel:e.defaultModel||"sonnet",apiKey:e.apiKey,settingSources:e.settingSources,thinking:e.thinking,effort:e.effort,createSession:e.createSession}}getSessionIfExists(e){return this.sessions.get(e)}async getSession(e){let n=this.sessions.get(e);if(n)return this._touchActivity(e),n;let r=this.pendingSessions.get(e);if(r){let i=await r;return this._touchActivity(e),i}let o=this.sessionData.get(e)??{chatId:e,model:this.options.defaultModel,createdAt:new Date().toISOString(),lastActivity:new Date().toISOString()},s=(async()=>{let i={model:o.model,apiKey:this.options.apiKey};this.options.settingSources?.length&&(i.settingSources=this.options.settingSources),this.options.thinking!==void 0&&(i.thinking=this.options.thinking),this.options.effort!==void 0&&(i.effort=this.options.effort);let a=await this.options.createSession(sn(i));return this.sessions.set(e,a),this.sessionData.set(e,o),a})();this.pendingSessions.set(e,s);try{let i=await s;return this._touchActivity(e),i}finally{this.pendingSessions.delete(e)}}_touchActivity(e){let n=this.sessionData.get(e);n&&(n.lastActivity=new Date().toISOString())}async resetSession(e){let n=this.sessions.get(e);n&&(await n.close(),this.sessions.delete(e));let r=this.sessionData.get(e);r&&(r.lastActivity=new Date().toISOString())}async switchModel(e,n){let r=this.sessions.get(e);r&&(await r.close(),this.sessions.delete(e));let o=this.sessionData.get(e);o?(o.model=n,o.lastActivity=new Date().toISOString()):(o={chatId:e,model:n,createdAt:new Date().toISOString(),lastActivity:new Date().toISOString()},this.sessionData.set(e,o))}getModel(e){return this.sessionData.get(e)?.model||this.options.defaultModel}async loadSessions(){try{await qe.mkdir(this.options.dataDir,{recursive:!0});let e=await qe.readdir(this.options.dataDir);for(let n of e)if(n.endsWith(".json")){let r=Ts(this.options.dataDir,n),o=await qe.readFile(r,"utf-8"),s=JSON.parse(o);this.sessionData.set(s.chatId,s)}}catch(e){e.code!=="ENOENT"&&console.error("Failed to load sessions:",e)}}async saveSessions(){try{await qe.mkdir(this.options.dataDir,{recursive:!0});for(let[e,n]of this.sessionData.entries()){let r=Ts(this.options.dataDir,`${e}.json`);await qe.writeFile(r,JSON.stringify(n,null,2))}}catch(e){console.error("Failed to save sessions:",e)}}async closeAll(){await this.saveSessions();let e=Array.from(this.sessions.values()).map(n=>n.close().catch(r=>console.error("Error closing session:",r)));await Promise.all(e),this.sessions.clear()}getSessionCount(){return this.sessions.size}getChatCount(){return this.sessionData.size}};function ze(t,e=4096){if(t.length<=e)return[t];let n=[],r=t;for(;r.length>0;){if(r.length<=e){n.push(r);break}let o=e,s=r.lastIndexOf(`
|
|
1558
|
-
`,e);if(s>e-500&&s>0)o=s+1;else{let i=r.slice(0,e).match(/[.!?]\s+(?=[A-Z])/g);if(i&&i.length>0){let a=i[i.length-1];if(a){let c=r.lastIndexOf(a,e);c>e-200&&c>0&&(o=c+2)}}else{let a=r.lastIndexOf(" ",e);a>e-100&&a>0&&(o=a+1)}}n.push(r.slice(0,o).trim()),r=r.slice(o).trim()}return n}function Pn(t){let e=t.replace(/&/g,"&").replace(/</g,"<").replace(/>/g,">");
|
|
1557
|
+
`);if(r.length<=1&&t.length<=80)return{content:t,truncated:!1,sizeBytes:e,sizeLabel:n};if(r.length<=1)return t.length<=80?{content:t,truncated:!1,sizeBytes:e,sizeLabel:n}:{content:t.substring(0,80)+"\u2026",truncated:!0,sizeBytes:e,sizeLabel:n};if(t.length<=80)return{content:t,truncated:!1,sizeBytes:e,sizeLabel:n};let o=r[0]??"",s=o;return o.length>80&&(s=o.substring(0,80)+"\u2026"),{content:s+`\u2026+${r.length} lines`,truncated:!0,lineCount:r.length,sizeBytes:e,sizeLabel:n}}function Cd(t,e){let n={...t.raw??{}};return t.inputTokens!==void 0&&(n.input_tokens=t.inputTokens),t.outputTokens!==void 0&&(n.output_tokens=t.outputTokens),t.cachedInputTokens!==void 0&&(n.cache_read_input_tokens=t.cachedInputTokens),t.cacheCreationTokens!==void 0&&(n.cache_creation_input_tokens=t.cacheCreationTokens),t.totalTokens!==void 0&&(n.total_tokens=t.totalTokens),{sessionId:e,stopReason:t.stopReason??void 0,resultSubtype:t.resultSubtype,durationMs:t.durationMs,durationApiMs:t.durationApiMs,totalCostUsd:t.totalCostUsd,isError:t.isError,usage:Object.keys(n).length>0?n:void 0,modelUsage:t.modelUsage,permissionDenials:t.permissionDenials,errors:t.errors}}function Id(t){let e=t.isError===!0?null:Ss(t.toolName,t.content),n=e!==null?{display:e}:{},r=_d(t.content);if(r)return{type:"chunk",chunk:{type:"tool_result",toolUseId:t.toolUseId,content:`Output persisted (${r.sizeLabel}) \u2192 ${r.absolutePath}`,isError:t.isError===!0,persistedPath:r.absolutePath,sizeBytes:r.sizeBytes,sizeLabel:r.sizeLabel,...n}};let{content:o,truncated:s,lineCount:i,sizeBytes:a,sizeLabel:c}=Rd(t.content);return{type:"chunk",chunk:{type:"tool_result",toolUseId:t.toolUseId,content:o,isError:t.isError===!0,sizeBytes:a,sizeLabel:c,...s&&{truncated:s},...i!==void 0&&{lineCount:i},...n}}}function En(t,e){switch(t.type){case"session.init":{let n=t.info;return e.setSessionMetadata(r=>({...r,sessionId:n.sessionId,model:n.model??r.model,...n.permissionMode!==void 0?{permissionMode:n.permissionMode}:{},...n.cwd!==void 0?{cwd:n.cwd}:{},tools:n.tools?[...n.tools]:r.tools,slashCommands:n.slashCommands?[...n.slashCommands]:r.slashCommands,skills:n.skills?[...n.skills]:r.skills,plugins:n.plugins?n.plugins.map(o=>({...o})):r.plugins,mcpServers:n.mcpServers?n.mcpServers.map(o=>({...o})):r.mcpServers,...n.apiKeySource!==void 0?{apiKeySource:n.apiKeySource}:{},...n.version!==void 0?{claudeCodeVersion:n.version}:{},...n.outputStyle!==void 0?{outputStyle:n.outputStyle}:{}})),e.updateSessionIdentity(n.sessionId),e.resolveInitialization(),null}case"session.status":return e.setSessionMetadata(n=>({...n,sessionId:t.sessionId,...t.permissionMode!==void 0?{permissionMode:t.permissionMode}:{permissionMode:n.permissionMode},...t.status!==void 0?{status:t.status}:{}})),null;case"delta.text":return{type:"chunk",chunk:{type:"content",content:t.text,metadata:{eventType:"delta",deltaType:"text_delta"}}};case"delta.reasoning":return{type:"chunk",chunk:{type:"thinking",content:t.text,metadata:{eventType:"delta",deltaType:"thinking_delta"}}};case"assistant.message":if(t.sessionId&&e.updateSessionIdentity(t.sessionId),t.text){let n={role:"assistant",content:t.text,timestamp:new Date};return e.conversationHistory.push(n),{type:"message",message:n}}return null;case"tool.use.start":return{type:"chunk",chunk:{type:"tool_use_detail",toolUseId:t.toolUseId,toolName:t.toolName,toolInput:t.toolInput}};case"tool.use":return{type:"chunk",chunk:{type:"tool_use",content:t.summary,metadata:{eventType:"tool_use_summary",precedingToolUseIds:t.toolUseIds}}};case"tool.output":return Id(t);case"progress":return{type:"progress",progress:{taskId:t.progress.taskId,description:t.progress.description,...t.progress.summary!==void 0?{summary:t.progress.summary}:{},...t.progress.lastToolName!==void 0?{lastToolName:t.progress.lastToolName}:{},totalTokens:t.progress.totalTokens,toolUses:t.progress.toolUses,durationMs:t.progress.durationMs}};case"suggestion":return{type:"suggestion",suggestion:t.suggestion};case"turn.completed":{let n=Cd(t.usage,t.sessionId??e.getSessionMetadata().sessionId);e.setLastResponseMetadata(n);for(let r=e.conversationHistory.length-1;r>=0;r--){let o=e.conversationHistory[r];if(o?.role==="assistant"){o.metadata=n;break}}if(e.maxBudgetUsd!==void 0&&e.abortBudget!==void 0&&typeof n.totalCostUsd=="number"&&(e._runningCostUsd=(e._runningCostUsd??0)+n.totalCostUsd,e._runningCostUsd>=e.maxBudgetUsd)){Nn(e.traceWriter,{kind:"monetary",runningCostUsd:e._runningCostUsd,maxBudgetUsd:e.maxBudgetUsd,lastTurnCostUsd:n.totalCostUsd});let r=new ve(e._runningCostUsd,e.maxBudgetUsd);return e.abortBudget(r.message),{type:"error",error:r}}return{type:"done",metadata:n}}case"error":return{type:"error",error:t.error};case"paused":return{type:"paused",reason:t.reason,...t.resetsAt!==void 0?{resetsAt:t.resetsAt}:{},...t.accountId!==void 0?{accountId:t.accountId}:{}};case"resumed":return{type:"resumed",hotSwapped:t.hotSwapped,...t.accountId!==void 0?{accountId:t.accountId}:{}};default:return null}}var ye=class{config;currentState="idle";providerQuery;providerIterator;conversationHistory=[];turnCount=0;lastResponseMetadata=null;initPromise=null;inputStream;abortController;hookRegistry;sessionEndDispatched=!1;stateManager;sessionRunningCostUsd=0;sessionRunningTokens={input:0,output:0,cacheRead:0,cacheCreation:0};lastStopReason;constructor(e){this.config=e,this.abortController=new AbortController,this.hookRegistry=e.hookRegistry,gs(e.abortSignal,this.abortController,()=>{this.onAbort()}),this.initSdkLifecycle()}initSdkLifecycle(){let e=He(this.config.model)??this.config.model,{sessionIdentity:n,metadata:r}=ys(this.config,e);this.stateManager=new At(n,r),this.inputStream=new _t(()=>this.sessionId);let o=this.config.provider??Et(e);_(`\u{1F7E2} AgentSession: Creating query session via provider=${o.name}`),this.providerQuery=o.query({prompt:this.inputStream.createIterable(),config:this.config}),this.conversationHistory=[],this.turnCount=0,this.lastResponseMetadata=null,this.sessionRunningCostUsd=0,this.sessionRunningTokens={input:0,output:0,cacheRead:0,cacheCreation:0},this.lastStopReason=void 0,this.sessionEndDispatched=!1,this.currentState="idle";let s=this.providerQuery;this.providerIterator=s[Symbol.asyncIterator](),this.initPromise=this.pullInitialization()}async pullInitialization(){try{for(await ms(this.hookRegistry,{event:"SessionStart",sessionId:this.sessionId},{signal:this.abortController.signal,...this.config.traceWriter?{traceWriter:this.config.traceWriter}:{}});;){let e=await this.providerIterator.next();if(e.done){this.stateManager.resolveInitializationIfNeeded();return}let n=e.value,r=En(n,this.buildTransformDeps());if(n.type==="session.init"||r&&r.type==="error")return}}catch(e){let n=e instanceof Error?e:new Error(String(e));this.stateManager.isInitializationSettled()||this.stateManager.rejectInitializationOnce(n),await this.dispatchSessionEndOnce("error").catch(()=>{})}}buildTransformDeps(){return{conversationHistory:this.conversationHistory,getSessionMetadata:()=>this.stateManager.getSessionMetadata(),setSessionMetadata:e=>this.stateManager.setSessionMetadata(e),updateSessionIdentity:e=>this.stateManager.updateSessionIdentity(e),resolveInitialization:()=>this.stateManager.resolveInitializationOnce(),setLastResponseMetadata:e=>{this.lastResponseMetadata=e,typeof e.totalCostUsd=="number"&&Number.isFinite(e.totalCostUsd)&&(this.sessionRunningCostUsd+=e.totalCostUsd);let n=e.usage;if(n&&typeof n=="object"){let r=n,o=(s,i)=>{let a=r[s];typeof a=="number"&&Number.isFinite(a)&&(this.sessionRunningTokens[i]+=a)};o("input_tokens","input"),o("output_tokens","output"),o("cache_read_input_tokens","cacheRead"),o("cache_creation_input_tokens","cacheCreation")}typeof e.stopReason=="string"&&(this.lastStopReason=e.stopReason)},maxBudgetUsd:this.config.maxBudgetUsd,abortBudget:e=>{this.abortController.signal.aborted||this.abortController.abort(e)},...this.config.traceWriter?{traceWriter:this.config.traceWriter}:{}}}get state(){return this.currentState}get sessionId(){return this.stateManager.getSessionId()}get cwd(){return this.config.cwd}get abortSignal(){return this.abortController.signal}async sendMessage(e,n={}){this.assertCanSend();let r=this.config.timeoutMs??Fe,o=async()=>{let s=null,i="";this.currentState=n.stream?"streaming":"processing";for await(let a of this.sendMessageStreamInternal(e)){if(a.type==="chunk"&&a.chunk.type==="content"&&(i+=a.chunk.content),a.type==="message"&&a.message.role==="assistant"&&(s=a.message),a.type==="error")throw a.error;if(a.type==="done"){if(s)return{...s,metadata:a.metadata};if(i)return{role:"assistant",content:i,metadata:a.metadata,timestamp:new Date}}}if(s)return s;if(i)return{role:"assistant",content:i,timestamp:new Date};throw new Error("No assistant response received")};try{return await $e(o(),r,{controller:this.abortController,label:this.sessionId??"session"})}finally{this.state!=="closed"&&(this.currentState="idle")}}async*sendMessageStream(e){this.assertCanSend(),this.currentState="streaming";try{yield*this.sendMessageStreamInternal(e)}finally{this.state!=="closed"&&(this.currentState="idle")}}async*sendMessageStreamInternal(e){this.initPromise&&await this.initPromise;let r={role:"user",content:typeof e=="string"?e:this.summarizeContentBlocks(e),timestamp:new Date};this.conversationHistory.push(r),this.inputStream.pushUserMessage(e);let o=this.buildTransformDeps();try{for(;;){let s=await this.providerIterator.next();if(s.done)break;let i=s.value,a=En(i,o);if(a&&(a.type==="done"&&this.turnCount++,yield a,a.type==="done"||a.type==="error"))break}}finally{this.state!=="closed"&&(this.currentState="idle")}}summarizeContentBlocks(e){let n=[],r=0;for(let s of e)s.type==="text"?n.push(s.text):s.type==="image"&&r++;let o=n.join(" ");return r>0&&(o=o?`${o} [+ ${r} image(s)]`:`[+ ${r} image(s)]`),o||"[content block(s)]"}async interrupt(){this.currentState!=="streaming"&&this.currentState!=="processing"||(this.currentState="idle",await this.providerQuery.interrupt())}async reset(){if(this.currentState==="closed")throw new Error("Cannot reset: session is closed");if(this.abortController.signal.aborted)throw new Y("Cannot reset: session aborted");if(this.currentState==="processing"||this.currentState==="streaming")try{await this.providerQuery.interrupt()}catch{}await this.dispatchSessionEndOnce("reset");try{await this.providerQuery.close()}catch{}await this.providerIterator.return?.(),this.initPromise&&await Promise.race([this.initPromise,new Promise(e=>setTimeout(e,et))]).catch(()=>{}),this.stateManager.resolveInitializationIfNeeded();try{this.initSdkLifecycle()}catch(e){throw this.currentState="closed",new Error(`Session reset failed during lifecycle rebuild: ${e instanceof Error?e.message:String(e)}`,{cause:e})}}async onAbort(){try{await this.providerQuery.interrupt()}catch{}}async setModel(e){let n=He(e),r=this.stateManager.getSessionMetadata();await this.providerQuery.setModel(n??r.model??""),n&&this.stateManager.setSessionMetadata(o=>({...o,model:n}))}async setPermissionMode(e){await this.providerQuery.setPermissionMode(e),this.stateManager.setSessionMetadata(n=>({...n,permissionMode:e}))}setCwd(e){this.config={...this.config,cwd:e},this.providerQuery.setCwd?.(e)}waitForInitialization(){return this.stateManager.waitForInitialization()}getSessionIdentity(){return this.stateManager.getSessionIdentity()}getSessionMetadata(){return this.stateManager.getSessionMetadata()}getQuery(){return this.providerQuery}supportedCommands(){return this.providerQuery.supportedCommands()}supportedModels(){return this.providerQuery.supportedModels()}supportedAgents(){return this.providerQuery.supportedAgents()}getContextUsage(){return this.providerQuery.getContextUsage()}mcpServerStatus(){return this.providerQuery.mcpServerStatus()}accountInfo(){return this.providerQuery.accountInfo()}rewindFiles(e,n){return this.providerQuery.rewindFiles(e,n)}async compact(){if(this.currentState==="closed")throw new Error("Cannot compact: session is closed");if(this.currentState!=="idle")return{compacted:!1,reason:"session-busy",messagesBefore:0,messagesAfter:0};let e=this.providerQuery.compact?.bind(this.providerQuery);return e?e():{compacted:!1,reason:"not-supported",messagesBefore:0,messagesAfter:0}}getLastResponseMetadata(){return this.lastResponseMetadata}getOutputStream(){throw new Error("getOutputStream() is not supported \u2014 use sendMessageStream() instead")}getInputStreamRef(){return{pushUserMessage:e=>this.inputStream.pushUserMessage(e)}}getHistory(){return[...this.conversationHistory]}getTurnCount(){return this.turnCount}async close(){if(this.currentState!=="closed"){this.currentState="closed",this.abortController.signal.aborted||this.abortController.abort("closed"),this.stateManager.resolveInitializationIfNeeded();try{this.providerQuery.close()}catch{}if(await this.providerIterator.return?.(),this.initPromise)try{await Promise.race([this.initPromise,new Promise(e=>setTimeout(e,et))])}catch{}await this.dispatchSessionEndOnce("close")}}async dispatchSessionEndOnce(e){this.sessionEndDispatched||(this.sessionEndDispatched=!0,await this.emitClosure(e).catch(()=>{}),await this.sealTraceWriter(e).catch(()=>{}),await hs(this.hookRegistry,{event:"SessionEnd",sessionId:this.sessionId,reason:e},this.config.traceWriter?{traceWriter:this.config.traceWriter}:{}))}async emitClosure(e){let n=this.config.traceWriter;if(!n)return;let r=this.deriveClosureReason(e),o={};this.sessionRunningTokens.input>0&&(o.input=this.sessionRunningTokens.input),this.sessionRunningTokens.output>0&&(o.output=this.sessionRunningTokens.output),this.sessionRunningTokens.cacheRead>0&&(o.cacheRead=this.sessionRunningTokens.cacheRead),this.sessionRunningTokens.cacheCreation>0&&(o.cacheCreation=this.sessionRunningTokens.cacheCreation),await Hn(n,{reason:r,finalTurnCount:this.turnCount,finalCostUsd:this.sessionRunningCostUsd,finalTokens:o,...this.lastStopReason!==void 0?{lastStopReason:this.lastStopReason}:{}})}deriveClosureReason(e){if(e==="error")return"abort";let n=this.abortController.signal;if(n.aborted&&n.reason!=="closed"){let r=n.reason;if(r instanceof ve)return"budget_exceeded";if(r instanceof ie)return"timeout";if(typeof r=="string"){if(r.startsWith("Budget "))return"budget_exceeded";if(r.includes("timed out"))return"timeout"}return"abort"}return"model_end_turn"}async sealTraceWriter(e){let n=this.config.traceWriter;if(!n)return;let r=this.deriveSealStatus(e);await n.seal({status:r,finalCostUsd:this.sessionRunningCostUsd,finalTurnCount:this.turnCount,closedAt:new Date().toISOString()})}deriveSealStatus(e){if(e==="error")return"failed";let n=this.abortController.signal;return n.aborted&&n.reason!=="closed"?"cancelled":"succeeded"}assertCanSend(){if(this.currentState==="closed")throw new Error("Cannot send message: session is closed");if(this.abortController.signal.aborted)throw new Y("Cannot send message: session aborted");if(this.currentState==="processing"||this.currentState==="streaming")throw new Error("Cannot send message: session is busy");if(this.config.maxTurns&&this.turnCount>=this.config.maxTurns)throw new Error(`Maximum turns (${this.config.maxTurns}) exceeded`)}};function Md(t){let e=new Map,n=t.defaultMode??"ask";if(t.tools)for(let[r,o]of Object.entries(t.tools))typeof o=="string"?e.set(r,{mode:o}):e.set(r,{mode:o.mode,reason:o.reason});if(t.list)for(let r of t.list)e.set(r.tool,{mode:r.mode,reason:r.reason});return{defaultMode:n,byTool:e}}function Dd(t){return t.behavior==="allow"?{behavior:"allow"}:{behavior:"deny",message:t.reason??"Tool denied by permission rules"}}function xs(t){let{defaultMode:e,byTool:n}=Md(t.rules);return async(r,o)=>{let s={toolName:r,input:o},i=n.get(r),a=i?.mode??e,c=i?.reason,l;return a==="ask"&&t.onAsk?l=await t.onAsk(s):a==="deny"?l={behavior:"deny",reason:c}:l={behavior:"allow"},t.onDecision?.(s,l),Dd(l)}}import{Telegraf as gu}from"telegraf";import{promises as qe}from"fs";import{join as Ts}from"path";var Ce=class{sessions=new Map;pendingSessions=new Map;sessionData=new Map;options;constructor(e){this.options={dataDir:e.dataDir||"./data/telegram-sessions",defaultModel:e.defaultModel||"sonnet",apiKey:e.apiKey,settingSources:e.settingSources,thinking:e.thinking,effort:e.effort,createSession:e.createSession}}getSessionIfExists(e){return this.sessions.get(e)}async getSession(e){let n=this.sessions.get(e);if(n)return this._touchActivity(e),n;let r=this.pendingSessions.get(e);if(r){let i=await r;return this._touchActivity(e),i}let o=this.sessionData.get(e)??{chatId:e,model:this.options.defaultModel,createdAt:new Date().toISOString(),lastActivity:new Date().toISOString()},s=(async()=>{let i={model:o.model,apiKey:this.options.apiKey};this.options.settingSources?.length&&(i.settingSources=this.options.settingSources),this.options.thinking!==void 0&&(i.thinking=this.options.thinking),this.options.effort!==void 0&&(i.effort=this.options.effort);let a=await this.options.createSession(sn(i));return this.sessions.set(e,a),this.sessionData.set(e,o),a})();this.pendingSessions.set(e,s);try{let i=await s;return this._touchActivity(e),i}finally{this.pendingSessions.delete(e)}}_touchActivity(e){let n=this.sessionData.get(e);n&&(n.lastActivity=new Date().toISOString())}async resetSession(e){let n=this.sessions.get(e);n&&(await n.close(),this.sessions.delete(e));let r=this.sessionData.get(e);r&&(r.lastActivity=new Date().toISOString())}async switchModel(e,n){let r=this.sessions.get(e);r&&(await r.close(),this.sessions.delete(e));let o=this.sessionData.get(e);o?(o.model=n,o.lastActivity=new Date().toISOString()):(o={chatId:e,model:n,createdAt:new Date().toISOString(),lastActivity:new Date().toISOString()},this.sessionData.set(e,o))}getModel(e){return this.sessionData.get(e)?.model||this.options.defaultModel}async loadSessions(){try{await qe.mkdir(this.options.dataDir,{recursive:!0});let e=await qe.readdir(this.options.dataDir);for(let n of e)if(n.endsWith(".json")){let r=Ts(this.options.dataDir,n),o=await qe.readFile(r,"utf-8"),s=JSON.parse(o);this.sessionData.set(s.chatId,s)}}catch(e){e.code!=="ENOENT"&&console.error("Failed to load sessions:",e)}}async saveSessions(){try{await qe.mkdir(this.options.dataDir,{recursive:!0});for(let[e,n]of this.sessionData.entries()){let r=Ts(this.options.dataDir,`${e}.json`);await qe.writeFile(r,JSON.stringify(n,null,2))}}catch(e){console.error("Failed to save sessions:",e)}}async closeAll(){await this.saveSessions();let e=Array.from(this.sessions.values()).map(n=>n.close().catch(r=>console.error("Error closing session:",r)));await Promise.all(e),this.sessions.clear()}getSessionCount(){return this.sessions.size}getChatCount(){return this.sessionData.size}};function ze(t,e=4096){if(t.length<=e)return[t];let n=[],r=t;for(;r.length>0;){if(r.length<=e){n.push(r);break}let o=e,s=r.lastIndexOf(`
|
|
1558
|
+
`,e);if(s>e-500&&s>0)o=s+1;else{let i=r.slice(0,e).match(/[.!?]\s+(?=[A-Z])/g);if(i&&i.length>0){let a=i[i.length-1];if(a){let c=r.lastIndexOf(a,e);c>e-200&&c>0&&(o=c+2)}}else{let a=r.lastIndexOf(" ",e);a>e-100&&a>0&&(o=a+1)}}n.push(r.slice(0,o).trim()),r=r.slice(o).trim()}return n}function Pn(t){let e=t.replace(/&/g,"&").replace(/</g,"<").replace(/>/g,">"),n=[];e=e.replace(/^```[\w]*\n?([\s\S]*?)```/gm,(o,s)=>{let i=n.length;return n.push(`<pre>${s}</pre>`),`FENCED${i}`});let r=[];return e=e.replace(/`([^`]+)`/g,(o,s)=>{let i=r.length;return r.push(`<code>${s}</code>`),`CODE${i}`}),e=e.replace(/\*\*([^*]+)\*\*/g,"<b>$1</b>"),e=e.replace(/__([^_]+)__/g,"<b>$1</b>"),e=e.replace(/\*([^*]+)\*/g,"<i>$1</i>"),e=e.replace(/_([^_]+)_/g,"<i>$1</i>"),e=e.replace(/~~([^~]+)~~/g,"<s>$1</s>"),e=e.replace(/\[([^\]]+)\]\(([^)]+)\)/g,(o,s,i)=>'<a href="'+i.replace(/&/g,"&").replace(/"/g,""")+'">'+s+"</a>"),e=e.replace(/^#{1,6}\s+/gm,""),e=e.replace(/\x02CODE(\d+)\x03/g,(o,s)=>r[Number(s)]??""),e=e.replace(/\x02FENCED(\d+)\x03/g,(o,s)=>n[Number(s)]??""),e}function W(t){return`\u274C Error: ${t instanceof Error?t.message:t}`}function _n(){return"\u26A0\uFE0F An internal error occurred. Please try again."}var Es=[{cmd:"/start",desc:"Show welcome and this command list"},{cmd:"/help",desc:"Show this command list"},{cmd:"/clear",desc:"Clear conversation history (SDK /clear)"},{cmd:"/compact",desc:"Compact conversation history (summarize older messages)"},{cmd:"/model [opus|sonnet|haiku]",desc:"Switch Claude model"}];function Ps(t){let e=["\u{1F4CB} Bot commands (aligned with agent-afk CLI):","",...Es.map(n=>` ${n.cmd}
|
|
1559
1559
|
${n.desc}`)];return t&&t.length>0&&e.push("","\u{1F4CB} Session commands (from SDK, when using settingSources):","",...t.map(n=>` /${n.replace(/^\//,"")}`)),e.push("","Just send a message to chat with Claude."),e.join(`
|
|
1560
1560
|
`)}function An(){return`\u{1F44B} Welcome to Agent AFK Bot!
|
|
1561
1561
|
|
|
@@ -1565,10 +1565,10 @@ Available commands:
|
|
|
1565
1565
|
${Es.map(e=>`${e.cmd} - ${e.desc}`).join(`
|
|
1566
1566
|
`)}
|
|
1567
1567
|
|
|
1568
|
-
Just send me a message to get started!`}function Rn(t){return`${{opus:"\u{1F680}",sonnet:"\u26A1",haiku:"\u{1F338}"}[t]||"\u{1F916}"} Switched to Claude ${t.toUpperCase()}`}function Ve(){return"\u{1F504} Conversation history cleared!"}function Cn(t){if(!t)return"\u{1F4E6} Conversation compacted (older messages summarized).";let e=t.tokensSavedEstimate!==void 0&&t.tokensSavedEstimate>0?` (~${Fd(t.tokensSavedEstimate)} input tokens saved)`:"";return`\u{1F4E6} Compacted ${t.before} \u2192 ${t.after} messages${e}.`}function In(t){return t==="aborted"?"\u{1F4E6} Compaction cancelled.":t.startsWith("summarization-failed")?`\u26A0\uFE0F Compaction failed: ${t}. History unchanged
|
|
1568
|
+
Just send me a message to get started!`}function Rn(t){return`${{opus:"\u{1F680}",sonnet:"\u26A1",haiku:"\u{1F338}"}[t]||"\u{1F916}"} Switched to Claude ${t.toUpperCase()}`}function Ve(){return"\u{1F504} Conversation history cleared!"}function Cn(t){if(!t)return"\u{1F4E6} Conversation compacted (older messages summarized).";let e=t.tokensSavedEstimate!==void 0&&t.tokensSavedEstimate>0?` (~${Fd(t.tokensSavedEstimate)} input tokens saved)`:"";return`\u{1F4E6} Compacted ${t.before} \u2192 ${t.after} messages${e}.`}function In(t){return t==="aborted"?"\u{1F4E6} Compaction cancelled.":t.startsWith("summarization-failed")?`\u26A0\uFE0F Compaction failed: ${t}. History unchanged.`:t==="nothing-to-summarize"?"\u{1F4E6} Nothing to compact \u2014 all history is within the keep window.":`\u{1F4E6} Nothing to compact (${t}).`}function Fd(t){return t>=1e3?`${Math.round(t/100)/10}k`:String(t)}async function Mn(t){await t.reply(An())}async function Dn(t,e){let n=t.chat?.id,r,o=n?e.getSessionIfExists(n):void 0;if(o)try{await Promise.race([o.waitForInitialization(),new Promise((i,a)=>setTimeout(()=>a(new Error("timeout")),2e3))]);let s=o.getSessionMetadata();s.slashCommands?.length&&(r=s.slashCommands)}catch{}await t.reply(Ps(r))}async function On(t,e,n,r){let o=t.chat?.id;if(!o){await t.reply(W("Could not identify chat"));return}try{await e.resetSession(o),n.delete(o),await t.reply(Ve())}catch(s){r("Clear error:",s),await t.reply(W(s))}}async function _s(t,e,n){let r=t.chat?.id;if(!r){await t.reply(W("Could not identify chat"));return}try{let o=await e.getSession(r);await t.sendChatAction("typing").catch(()=>{});let s=await o.compact();s.compacted?await t.reply(Cn({before:s.messagesBefore,after:s.messagesAfter,...s.tokensSavedEstimate!==void 0?{tokensSavedEstimate:s.tokensSavedEstimate}:{}})):await t.reply(In(s.reason??"unknown"))}catch(o){n("Compact error:",o),await t.reply(W(o))}}async function Fn(t,e,n){let r=t.chat?.id;if(!r){await t.reply(W("Could not identify chat"));return}let s=t.message.text.split(/\s+/).slice(1);if(s.length===0){let l=e.getModel(r);await t.reply(`Current model: ${l.toUpperCase()}
|
|
1569
1569
|
|
|
1570
1570
|
Usage: /model [opus|sonnet|haiku]`);return}let i=s[0];if(!i){await t.reply(W("Please specify a model: opus, sonnet, or haiku"));return}let a=i.toLowerCase();if(!["opus","sonnet","haiku"].includes(a)){await t.reply(W(`Invalid model: ${i}
|
|
1571
|
-
Valid options: opus, sonnet, haiku`));return}try{await e.switchModel(r,a),await t.reply(Rn(a))}catch(l){n("Model switch error:",l),await t.reply(W(l))}}import{execFile as Zd,spawn as eu}from"node:child_process";import{promisify as tu}from"node:util";import{execFile as $d}from"node:child_process";import{randomBytes as Ld}from"node:crypto";import{promises as Rt}from"node:fs";import{join as Ct}from"node:path";import{promisify as Nd}from"node:util";var
|
|
1571
|
+
Valid options: opus, sonnet, haiku`));return}try{await e.switchModel(r,a),await t.reply(Rn(a))}catch(l){n("Model switch error:",l),await t.reply(W(l))}}import{execFile as Zd,spawn as eu}from"node:child_process";import{promisify as tu}from"node:util";import{execFile as $d}from"node:child_process";import{randomBytes as Ld}from"node:crypto";import{promises as Rt}from"node:fs";import{join as Ct}from"node:path";import{promisify as Nd}from"node:util";var gk=Nd($d);var de=class extends Error{cause;code;constructor(e,n,r){super(e),this.name="WorktreeError",this.cause=n,this.code=r}};function Ud(t,e=40){return t.toLowerCase().trim().replace(/[^a-z0-9]+/g,"-").replace(/^-+|-+$/g,"").slice(0,e).replace(/-+$/g,"")||"task"}function jd(){return Ld(4).toString("hex").slice(0,4)}function Hd(t){let e=n=>String(n).padStart(2,"0");return`${t.getUTCFullYear()}${e(t.getUTCMonth()+1)}${e(t.getUTCDate())}T${e(t.getUTCHours())}${e(t.getUTCMinutes())}${e(t.getUTCSeconds())}`}function As(t,e={}){let n=(e.now??(()=>new Date))(),r=(e.randomSuffix??jd)();return`${Hd(n)}-${Ud(t,32)}-${r}`}function Bd(t){let e=t;return e.respawnedAt===void 0&&(e.respawnedAt=void 0),e.respawnedAs===void 0&&(e.respawnedAs=void 0),e.prUrl===void 0&&(e.prUrl=void 0),e.prCreatedAt===void 0&&(e.prCreatedAt=void 0),e}async function Ie(t){let e=Ct(kr(t),"farm.json");try{let n=await Rt.readFile(e,"utf8"),r=JSON.parse(n);if(r.schemaVersion!==1&&r.schemaVersion!==2&&r.schemaVersion!==3)throw new de(`unsupported farm manifest schema: ${r.schemaVersion} (expected 1, 2, or 3)`,void 0,"unsupported-schema");return Bd(r)}catch(n){if(n.code==="ENOENT")return null;throw n instanceof de?n:new de(`failed to load farm manifest ${e}`,n,"invalid")}}async function Rs(t,e){let n=await Ie(t);if(!n)throw new de(`farm not found: ${t}`);return n.human_decision=e,n.decidedAt=new Date().toISOString(),n.schemaVersion=3,await Rt.writeFile(Ct(n.farmDir,"farm.json"),JSON.stringify(n,null,2)+`
|
|
1572
1572
|
`,"utf8"),n}async function Cs(t,e){let n=await Ie(t);if(!n)throw new de(`farm not found: ${t}`);return n.respawnedAt=new Date().toISOString(),n.respawnedAs=e,n.schemaVersion=3,await Rt.writeFile(Ct(n.farmDir,"farm.json"),JSON.stringify(n,null,2)+`
|
|
1573
1573
|
`,"utf8"),n}async function Is(t,e){let n=await Ie(t);if(!n)throw new de(`farm not found: ${t}`);return n.prUrl=e,n.prCreatedAt=new Date().toISOString(),n.schemaVersion=3,await Rt.writeFile(Ct(n.farmDir,"farm.json"),JSON.stringify(n,null,2)+`
|
|
1574
1574
|
`,"utf8"),n}import{execFile as Wd}from"node:child_process";import{promisify as Gd}from"node:util";var Kd=Gd(Wd),qd=2e4,Ye=class extends Error{kind;exitCode;stderr;constructor(e,n,r,o){super(e),this.name="GhError",this.kind=n,this.exitCode=r,this.stderr=o}},It=null,Je=null;function zd(t,e,n){return n===!0?"timeout":e==="ENOENT"?"not-found":/already exists/i.test(t)?"already-exists":/authentication|please log in|HTTP 40[13]|bad credentials|token|scope/i.test(t)?"unauthed":/ETIMEDOUT|ECONNRESET|ECONNREFUSED/i.test(t+(e??""))?"network":"unknown"}function Ms(t,e){return Kd(t,e,{timeout:qd,killSignal:"SIGTERM"}).then(n=>({stdout:n.stdout,stderr:n.stderr}))}async function Ds(t={}){let e=(t._now??(()=>Date.now()))(),n=t.ttlMs??6e4,r=t.execFn??Ms,o=t.log??(()=>{});return It&&It.expiresAt>e?(o("[gh] checkGhReady cache hit"),It.result):(o("[gh] checkGhReady cache miss \u2014 probing"),Je||(Je=(async()=>{try{await r("gh",["--version"])}catch(a){let c=a;return c.killed?{ok:!1,hint:"`gh` timed out \u2014 check connectivity"}:c.code==="ENOENT"?{ok:!1,hint:"`gh` CLI not found \u2014 install with: brew install gh"}:{ok:!1,hint:"`gh --version` failed unexpectedly \u2014 check gh installation"}}try{await r("gh",["auth","status"])}catch(a){let c=a;return c.killed||/ETIMEDOUT|ECONNRESET|ENOTFOUND/i.test(String(c.code??""))?{ok:!1,hint:"check network \u2014 cannot reach GitHub"}:{ok:!1,hint:"`gh` is not authenticated \u2014 run: gh auth login"}}let i={ok:!0};return n>0&&(It={result:i,expiresAt:e+n}),i})().finally(()=>{Je=null}),Je))}async function Os(t,e){let n=e??Ms,r=["pr","create","--base",t.base,"--head",t.head,"--title",t.title,"--body",t.body];try{let{stdout:o}=await n("gh",r);return o.trim()}catch(o){let s=o,i=s.stderr??"",a=s.code,c=s.exitCode??1,l=zd(i,a,s.killed);throw new Ye(`gh pr create failed (${l}): ${i.trim()}`,l,c,i)}}function Fs(t,e){let n;try{n=e?._store??new be}catch(r){return{skipped:!0,reason:r instanceof Error?r.message:String(r)}}try{let r={type:"farm-decision",taskSlug:t.taskSlug,decision:t.decision,decidedAt:t.decidedAt,via:t.via};return{factId:n.storeFact({category:"decision",content:JSON.stringify(r),source_surface:"afk"})}}catch(r){return{skipped:!0,reason:r instanceof Error?r.message:String(r)}}}import{promises as Vd}from"node:fs";import{join as Jd}from"node:path";function Ns(t){let e=t.filter(r=>r.score!==null),n=t.filter(r=>r.score===null).map(r=>r.index).sort((r,o)=>r-o);return e.sort((r,o)=>{let s=r.score,i=o.score,a=$s(s),c=$s(i);if(a!==c)return c-a;let l=Ls(s.lint_ok),d=Ls(i.lint_ok);return l!==d?d-l:s.loc_delta!==i.loc_delta?s.loc_delta-i.loc_delta:r.index-o.index}),[...e.map(r=>r.index),...n]}function $s(t){let e=t.pass+t.fail;return e===0?0:t.pass/e}function Ls(t){return t===!0?2:t===!1?1:0}async function Mt(t,e={}){if(t.branches.length===0)throw new Error(`resolveWinnerBranch: farm ${t.taskSlug} has no branches`);let n=e.loadScore??Yd,r=await Promise.all(t.branches.map(async i=>({index:i.index,score:await n(t.farmDir,i.index)}))),o=Ns(r),s=new Map(t.branches.map(i=>[i.index,i]));for(let i of o){let a=r.find(c=>c.index===i)?.score;if(a&&a.pass>0&&a.fail===0){let c=s.get(i);if(c)return{branch:c,source:"winner"}}}for(let i of o)if(r.find(c=>c.index===i)?.score){let c=s.get(i);if(c)return{branch:c,source:"top-scored"}}return{branch:t.branches[0],source:"fallback-first-branch"}}async function Yd(t,e){let n=Jd(t,"scores",`branch-${e}.json`);try{let r=await Vd.readFile(n,"utf8");return JSON.parse(r)}catch(r){return r.code==="ENOENT",null}}var Dt="afk:f:";var Xd=new Set(["p","d","r","x"]),Qd=/^[a-z0-9T][a-z0-9T-]{0,62}$/;function Us(t){if(!t||!t.startsWith(Dt)||Buffer.byteLength(t,"utf8")>64)return null;let e=t.slice(Dt.length),n=e.indexOf(":");if(n<1)return null;let r=e.slice(0,n),o=e.slice(n+1);return!Xd.has(r)||!Qd.test(o)?null:{action:r,taskSlug:o}}var nu=tu(Zd),$n=new Map;async function Hs(t,e={}){let n=e.log??(()=>{}),r=ru(t),o=Us(r);if(!o){await N(t,"Unknown action",n);return}if(t.chat?.id===void 0){await N(t,"No chat context",n);return}let s=e.loadFarm??Ie,i;try{i=await s(o.taskSlug)}catch(a){n("[farm-callback] loadFarm failed:",a),await N(t,"Farm load failed",n);return}if(!i){await N(t,"Farm not found (already GC\u2019d?)",n);return}try{await ou(o.action,t,i,e,n)}catch(a){n("[farm-callback] dispatch error:",a),await N(t,"Internal error",n)}}function ru(t){return t.callbackQuery?.data}async function N(t,e,n){try{await t.answerCbQuery(e)}catch(r){n("[farm-callback] answerCbQuery failed:",r)}}async function ou(t,e,n,r,o){switch(t){case"x":return cu(e,n,r,o);case"d":return lu(e,n,r,o);case"p":return js(`p:${n.taskSlug}`,e,n,r,o,su);case"r":return js(`r:${n.taskSlug}`,e,n,r,o,au)}}async function js(t,e,n,r,o,s){let i=$n.get(t);if(i){o(`[farm-callback] ${t} \u2014 second tap, awaiting in-flight lock`);try{await i}catch{}let c=r.loadFarm??Ie,l;try{l=await c(n.taskSlug)}catch{await N(e,"Farm load failed",o);return}if(!l){await N(e,"Farm not found",o);return}return s(e,l,r,o)}let a=s(e,n,r,o);return $n.set(t,a),a.finally(()=>{$n.delete(t)}),a}async function su(t,e,n,r){if(e.prUrl){await N(t,`PR already open: ${e.prUrl}`,r);return}await N(t,"Opening PR\u2026",r);let o=n.checkGhReady??Ds,s;try{s=await o()}catch(m){r("[farm-callback] checkGhReady threw:",m);try{await t.reply("gh readiness check failed \u2014 see daemon logs")}catch{}return}if(!s.ok){try{await t.reply(s.hint)}catch{}return}let i=n.resolveWinnerBranch??Mt,a;try{a=await i(e)}catch(m){r("[farm-callback] resolveWinnerBranch failed:",m);try{await t.reply("Winner lookup failed")}catch{}return}let c=a.branch.branch,l=e.baseBranch??"main",d=`Auto PR: ${e.taskName}`,u=`Auto-generated by afk farm ${e.taskSlug} | winner: ${c} | created: ${new Date().toISOString()}`,p=n.createPr??Os,g;try{g=await p({base:l,head:c,title:d,body:u})}catch(m){if(m instanceof Ye){let S={"not-found":"gh CLI not found \u2014 install with: brew install gh","already-exists":"PR already exists for this branch",unauthed:"gh is not authenticated \u2014 run: gh auth login",network:"Network error \u2014 check gh connectivity",timeout:"gh timed out \u2014 check connectivity",unknown:"gh pr create failed \u2014 see daemon logs"};try{await t.reply(S[m.kind])}catch{}return}r("[farm-callback] createPr failed:",m);try{await t.reply("gh pr create failed \u2014 see daemon logs")}catch{}return}let b=n.recordPrCreated??Is;try{await b(e.taskSlug,g)}catch(m){r("[farm-callback] recordPrCreated failed:",m)}try{await t.reply(`PR opened \u2713
|
|
@@ -1582,7 +1582,7 @@ Commits:
|
|
|
1582
1582
|
${l.stdout.trim()||"(none)"}
|
|
1583
1583
|
|
|
1584
1584
|
Stat:
|
|
1585
|
-
${d.stdout.trim()||"(no changes)"}`;await t.reply(u.slice(0,4e3))}catch(l){r("[farm-callback] diff failed:",l);try{await t.reply("Diff failed \u2014 see daemon logs.")}catch{}}}async function du(t,e){let n=await nu("git",e,{cwd:t,maxBuffer:4194304});return{stdout:n.stdout,stderr:n.stderr}}function Ot(t){let e=t instanceof Error?t.message:String(t);return e.toLowerCase().includes("rate limit")||e.toLowerCase().includes("too many requests")}function Ft(t){let e=t instanceof Error?t.message:String(t);return e.toLowerCase().includes("network")||e.toLowerCase().includes("connect")||e.toLowerCase().includes("timeout")}var
|
|
1585
|
+
${d.stdout.trim()||"(no changes)"}`;await t.reply(u.slice(0,4e3))}catch(l){r("[farm-callback] diff failed:",l);try{await t.reply("Diff failed \u2014 see daemon logs.")}catch{}}}async function du(t,e){let n=await nu("git",e,{cwd:t,maxBuffer:4194304});return{stdout:n.stdout,stderr:n.stderr}}function Ot(t){let e=t instanceof Error?t.message:String(t);return e.toLowerCase().includes("rate limit")||e.toLowerCase().includes("too many requests")}function Ft(t){let e=t instanceof Error?t.message:String(t);return e.toLowerCase().includes("network")||e.toLowerCase().includes("connect")||e.toLowerCase().includes("timeout")}import{TelegramError as uu}from"telegraf";var pu=300,fu=9e4,Bs=6e4,mu=300*1e3,hu=9e4;async function Ln(t,e,n,r){let o="",s=null,i=0,a=null,c=null,l=!1,d=-1,u=async(p,g=!1)=>{let b=Pn(p||"\u2026"),m=Date.now();if(!s){let f=ze(b);s=await t.reply(f[0]??"\u2026",{parse_mode:"HTML"});return}if(!g&&m-i<pu&&p.length<100)return;i=m;let S=ze(b);try{await t.telegram.editMessageText(t.chat?.id,s.message_id,void 0,S[0]??b,{parse_mode:"HTML"})}catch(f){if(f instanceof uu&&f.code===400&&/can't parse entities/i.test(f.description))try{await t.telegram.editMessageText(t.chat?.id,s.message_id,void 0,p)}catch{}}};try{let p="sendMessageStream"in e&&typeof e.sendMessageStream=="function"?e.sendMessageStream(n):(async function*(){let h=await e.sendMessage(n,{stream:!1});yield{type:"message",message:h},yield{type:"done",metadata:h.metadata}})();await u("Thinking\u2026");let g=p[Symbol.asyncIterator](),b=!1,m=null,S=()=>{let h=a!==null?Math.max(Bs,a.getTime()-Date.now()+hu):b?Bs:fu;return new Promise((y,k)=>{m=setTimeout(()=>{m=null,k(new Error(b?"Response timed out. Try sending a shorter message or try again.":"Request timed out. The agent may still be starting (first message can take a minute). Try again in a moment."))},h),g.next().then(A=>{m!=null&&(clearTimeout(m),m=null),y(A)},A=>{m!=null&&(clearTimeout(m),m=null),k(A)})})};if(await Sr((h,y)=>{let k=y.agentType??y.subagentId;if(h.type==="chunk"&&h.chunk.type==="tool_use_detail"){let A=h.chunk.toolInput.length>60?h.chunk.toolInput.slice(0,57)+"...":h.chunk.toolInput;o+=`
|
|
1586
1586
|
\u25E6 ${k}: ${h.chunk.toolName} ${A}`,u(o)}else h.type==="done"&&(o+=`
|
|
1587
1587
|
\u25E6 ${k}: Done`,u(o))},async()=>{for(;;){process.env.AFK_TELEGRAM_TRACE&&console.log("[trace] awaiting next event");let h=await S();if(process.env.AFK_TELEGRAM_TRACE&&console.log("[trace] event arrived:",h.done?"DONE":h.value.type),h.done)break;let y=h.value;if(b||(b=!0,console.log("\u{1F4E1} First stream event received:",y.type),r?.("First stream event received:",y.type)),y.type==="chunk"&&y.chunk.type==="content"&&(o+=y.chunk.content,await u(o)),y.type==="message"&&y.message.role==="assistant"&&(o=y.message.content,await u(o)),y.type==="progress"){let{description:k,summary:A,lastToolName:F}=y.progress,x=F?`
|
|
1588
1588
|
\u25E6 ${k} (${F})`:`
|
|
@@ -1599,4 +1599,4 @@ I'll auto-resume when the limit resets \u2014 no need to retype.`;await u(F,!0),
|
|
|
1599
1599
|
|
|
1600
1600
|
Resets at ${a.toLocaleTimeString(void 0,{hour:"numeric",minute:"2-digit",hour12:!0})} (in ~${x} min).
|
|
1601
1601
|
|
|
1602
|
-
I'll auto-resume when the limit resets \u2014 no need to retype.`;l=!0,u(j,!0).finally(()=>{l=!1})}},
|
|
1602
|
+
I'll auto-resume when the limit resets \u2014 no need to retype.`;l=!0,u(j,!0).finally(()=>{l=!1})}},mu));continue}if(y.type==="resumed"){c!==null&&(clearInterval(c),c=null),a=null;let k=y.hotSwapped&&y.accountId?`\u25B6 **Resumed on ${y.accountId}**`:"\u25B6 **Resumed**";await u(k,!0);continue}if(y.type==="done"){c!==null&&(clearInterval(c),c=null),o.trim()&&await u(o,!0);break}if(y.type==="error")throw c!==null&&(clearInterval(c),c=null),y.error}}),await Promise.resolve(g.return?.(void 0)).catch(()=>{}),o&&s){let h=ze(Pn(o));if(h.length>1)for(let y=1;y<h.length;y++){let k=h[y];k&&await t.reply(k,{parse_mode:"HTML"})}}}catch(p){throw r?.("Streaming error:",p),p}}async function Ws(t,e,n,r,o){if(!r.has(e))try{await Promise.race([n.waitForInitialization(),new Promise((a,c)=>setTimeout(()=>c(new Error("timeout")),5e3))]);let s=n.getSessionMetadata(),i=[{command:"start",description:"Show welcome and command list"},{command:"help",description:"Show this command list"},{command:"clear",description:"Clear conversation history"},{command:"compact",description:"Compact conversation history"},{command:"model",description:"Switch Claude model (opus/sonnet/haiku)"}];if(s.slashCommands?.length)for(let a of s.slashCommands){let c=a.replace(/^\//,"");i.push({command:c,description:`SDK command: ${c}`})}if(s.skills?.length)for(let a of s.skills)i.push({command:a,description:`Run ${a} skill`});await t.telegram.setMyCommands(i,{scope:{type:"chat",chat_id:e}}),r.add(e),o(`Registered ${i.length} commands for chat ${e}`)}catch(s){o(`Could not register dynamic commands for chat ${e}:`,s)}}var $t=class{sessionManager;messageQueues=new Map;registeredCommandChats;log;bot;constructor(e,n,r,o){this.bot=e,this.sessionManager=n,this.registeredCommandChats=r,this.log=o}async handle(e){let n=e.chat?.id,r=e.message.text;if(!(!n||!r)&&(console.log(`\u{1F4EC} Message from chat ID: ${n}`),!r.startsWith("/")))try{let o=await this.sessionManager.getSession(n);if(Ws(this.bot,n,o,this.registeredCommandChats,this.log).catch(s=>this.log("Failed to register chat commands:",s)),o.state!=="idle"){this.enqueueMessage(n,e,r),await e.reply("Message queued.");return}await this.processOne(n,e,r)}catch(o){if(console.error("\u274C Message handling error:",o),this.log("Message handling error:",o),(o?.message??"").includes("session is busy")){this.enqueueMessage(n,e,r),await e.reply("Message queued.");return}Ot(o)?await e.reply("\u23F3 Rate limit reached. Please wait a moment and try again."):Ft(o)?await e.reply("\u{1F310} Network error. Please check your connection and try again."):await e.reply(_n())}}async processClearDirect(e,n){try{await this.sessionManager.resetSession(e),this.registeredCommandChats.delete(e),await n.reply(Ve())}catch(r){this.log("Clear error:",r),await n.reply(W(r))}}enqueueMessage(e,n,r){let o=this.messageQueues.get(e);o||(o=[],this.messageQueues.set(e,o)),o.push({type:"message",ctx:n,text:r})}enqueueClear(e,n){let r=this.messageQueues.get(e);r||(r=[],this.messageQueues.set(e,r)),r.push({type:"clear",ctx:n})}async processOne(e,n,r){try{let o=await this.sessionManager.getSession(e);await n.sendChatAction("typing").catch(()=>{}),await Ln(n,o,r,this.log)}catch(o){console.error("\u274C Message handling error:",o),this.log("Message handling error:",o),Ot(o)?await n.reply("\u23F3 Rate limit reached. Please wait a moment and try again."):Ft(o)?await n.reply("\u{1F310} Network error. Please check your connection and try again."):await n.reply(_n())}finally{this.drainQueue(e).catch(o=>this.log("Drain error:",o))}}async drainQueue(e){let n=this.messageQueues.get(e);if(!n?.length)return;let r=n.shift();r.type==="message"?await this.processOne(e,r.ctx,r.text):await this.processClearDirect(e,r.ctx)}};var Lt=class{bot;sessionManager;options;running=!1;registeredCommandChats=new Set;messageHandler;constructor(e){this.options=e,this.bot=new gu(e.botToken),this.sessionManager=new Ce(e),this.messageHandler=new $t(this.bot,this.sessionManager,this.registeredCommandChats,this.log.bind(this)),this.setupHandlers()}setupHandlers(){this.bot.use(bn(this.options.allowedChatIds,this.log.bind(this))),this.bot.command("start",n=>Mn(n)),this.bot.command("help",n=>Dn(n,this.sessionManager)),this.bot.command("clear",async n=>{let r=n.chat?.id;if(!r){await n.reply(W("Could not identify chat"));return}(await this.sessionManager.getSession(r)).state!=="idle"?(this.messageHandler.enqueueClear(r,n),await n.reply("Clear queued.")):await On(n,this.sessionManager,this.registeredCommandChats,this.log.bind(this))}),this.bot.command("compact",n=>_s(n,this.sessionManager,this.log.bind(this))),this.bot.command("model",n=>Fn(n,this.sessionManager,this.log.bind(this))),this.bot.on("text",n=>this.messageHandler.handle(n));let e=new RegExp(`^${yu(Dt)}`);this.bot.action(e,n=>Hs(n,{log:this.log.bind(this)})),this.bot.catch((n,r)=>{this.log("Bot error:",n),r.reply(W("An unexpected error occurred. Please try again.")).catch(o=>this.log("Failed to send error message:",o))})}async start(){if(this.running)throw new Error("Bot is already running");this.log("Loading sessions..."),await this.sessionManager.loadSessions(),this.log("Starting bot..."),await this.bot.launch(),this.log("Registering bot commands..."),await this.bot.telegram.setMyCommands([{command:"start",description:"Show welcome and command list"},{command:"help",description:"Show this command list"},{command:"clear",description:"Clear conversation history"},{command:"compact",description:"Compact conversation history"},{command:"model",description:"Switch Claude model (opus/sonnet/haiku)"}]),this.running=!0,this.log("Bot started successfully");let e=async n=>{this.log(`Received ${n}, shutting down...`),await this.stop(),process.exit(0)};process.once("SIGINT",()=>e("SIGINT")),process.once("SIGTERM",()=>e("SIGTERM"))}async stop(){if(this.running){this.log("Stopping bot..."),this.running=!1,this.log("Closing sessions..."),await this.sessionManager.closeAll(),this.log("Stopping bot polling...");try{this.bot.stop()}catch(e){this.log("Error stopping bot (may not have been started):",e)}this.log("Bot stopped")}}getStats(){return{running:this.running,activeSessions:this.sessionManager.getSessionCount(),totalChats:this.sessionManager.getChatCount()}}async handleStart(e){return Mn(e)}async handleHelp(e){return Dn(e,this.sessionManager)}async handleClear(e){let n=e.chat?.id;if(!n){await e.reply(W("Could not identify chat"));return}if((await this.sessionManager.getSession(n)).state!=="idle")this.messageHandler.enqueueClear(n,e),await e.reply("Clear queued.");else return On(e,this.sessionManager,this.registeredCommandChats,this.log.bind(this))}async handleMessage(e){return this.messageHandler.handle(e)}async handleModelSwitch(e){return Fn(e,this.sessionManager,this.log.bind(this))}log(...e){this.options.verbose&&console.log("[TelegramBot]",...e)}};function yu(t){return t.replace(/[.*+?^${}()|[\]\\]/g,"\\$&")}export{ye as AgentSession,Re as OpenAICodexProvider,Ce as SessionManager,P as SubagentManager,Lt as TelegramBot,xs as createCanUseToolHook,Tt as openaiCodexProvider,Be as providerForModel,Et as resolveProvider};
|