@readmagic/workmate 0.0.7 → 0.0.9

package/dist/chunks/QueryEngine-CGYaShXD.js

@@ -1 +0,0 @@
-import{a as e}from"./rolldown-runtime-DaDcL9Zw.js";import{l as t,o as n,s as r}from"./envUtils-DSxRz_kt.js";import{n as i,t as a}from"./defineProperty-C-QyHrDq.js";import{Kr as o,qr as s}from"./src-CDCSSo0U2.js";import{Tt as c,en as l,et as u,ht as d,kt as f,tn as p}from"./state-ix0xIunV.js";import{$ as ee,$C as m,$a as h,$f as g,CE as _,Dl as te,El as v,Fw as y,Gn as b,HC as x,HT as S,Hw as C,Iw as w,Kn as ne,Lp as T,MT as re,Mo as E,NT as D,Qi as O,Rp as ie,SE as k,Sn as ae,UC as oe,VC as se,WC as ce,Ww as le,Xi as ue,as as A,et as j,gb as M,jr as N,lO as P,lr as F,lt as de,no as fe,nw as pe,pb as me,rt as he,si as I,ss as ge,uO as _e,ub as ve,ut as L,wE as R,xn as ye}from"./loadAgentsDir-x3rlgED4.js";import{n as z,t as be}from"./cwd-CYmmnQVS.js";import{N as xe,P as Se,c as Ce,g as we}from"./config-DRDuVyzE.js";import{Y as Te,i as Ee,o as De}from"./log-BMCygSzM.js";import{C as Oe,Y as B,_ as V,et as ke,u as Ae}from"./model-CZViyZnE.js";import{En as je,Tn as Me,ct as H,dt as U}from"./prompt-CE3sFmZV.js";import{i as W,o as Ne}from"./fileStateCache-DDW5toIs.js";import{P as Pe,lt as G,st as Fe}from"./rcDebugLog-DBB9r8aw.js";import{a as Ie,n as Le,r as Re}from"./mappers-04gNXJ3w.js";import{c as ze,n as Be,o as Ve,r as K,s as He}from"./MessageSelector-iIroHvXo.js";import{n as Ue,t as We}from"./systemTheme-Ceakddth.js";import{randomUUID as q}from"crypto";n(),i(),o(),l(),z(),Te(),De(),Se(),Me(),we(),ke(),P(),U(),Ne(),Pe(),D(),le(),M(),g(),v(),A(),E(),O(),N(),b(),ye(),j(),L(),R(),w(),V(),S(),m();async function Ge({tools:e,mainLoopModel:t,additionalWorkingDirectories:n,mcpClients:r,customSystemPrompt:i}){let[a,o,s]=await Promise.all([i===void 0?de(e,t,n,r):Promise.resolve([]),_(),i===void 0?k():Promise.resolve({})]);return{defaultSystemPrompt:a,userContext:o,systemContext:s}}async function Ke({tools:e,commands:t,mcpClients:n,messages:r,readFileState:i,getAppState:a,setAppState:o,customSystemPrompt:s,appendSystemPrompt:c,thinkingConfig:l,agents:u}){let d=Ae(),f=a(),{defaultSystemPrompt:p,userContext:ee,systemContext:m}=await Ge({tools:e,mainLoopModel:d,additionalWorkingDirectories:Array.from(f.toolPermissionContext.additionalWorkingDirectories.keys()),mcpClients:n,customSystemPrompt:s}),h=Fe([...s===void 0?p:[s],...c?[c]:[]]),g=r.at(-1),_=g?.type===`assistant`&&g.message.stop_reason===null?r.slice(0,-1):r;return{systemPrompt:h,userContext:ee,systemContext:m,toolUseContext:{options:{commands:t,debug:!1,mainLoopModel:d,tools:e,verbose:!1,thinkingConfig:l??(pe()===!1?{type:`disabled`}:{type:`adaptive`}),mcpClients:n,mcpResources:{},isNonInteractiveSession:!0,agentDefinitions:{activeAgents:u,allAgents:[]},customSystemPrompt:s,appendSystemPrompt:c},abortController:y(),readFileState:i,getAppState:a,setAppState:o,messages:_,setInProgressToolUseIDs:()=>{},setResponseLength:()=>{},updateFileHistoryState:()=>{},updateAttributionState:()=>{}},forkContextMessages:_}}w(),V(),T(),S(),We(),m(),Le(),x();var qe=()=>{try{return K(),e(Be)}catch{return null}},J=class{constructor(e){a(this,`config`,void 0),a(this,`mutableMessages`,void 0),a(this,`abortController`,void 0),a(this,`permissionDenials`,void 0),a(this,`totalUsage`,void 0),a(this,`hasHandledOrphanedPermission`,!1),a(this,`readFileState`,void 0),a(this,`discoveredSkillNames`,new Set),a(this,`loadedNestedMemoryPaths`,new Set),this.config=e,this.mutableMessages=e.initialMessages??[],this.abortController=e.abortController??y(),this.permissionDenials=[],this.readFileState=e.readFileCache,this.totalUsage=G}async*submitMessage(e,n){let{cwd:i,commands:a,tools:o,mcpClients:l,verbose:m=!1,thinkingConfig:g,maxTurns:_,maxBudgetUsd:v,taskBudget:y,canUseTool:b,customSystemPrompt:x,appendSystemPrompt:S,userSpecifiedModel:w,fallbackModel:T,jsonSchema:E,getAppState:D,setAppState:O,replayUserMessages:k=!1,includePartialMessages:le=!1,agents:A=[],setSDKStatus:j,orphanedPermission:M}=this.config;this.discoveredSkillNames.clear(),this.permissionDenials=[],ie(i);let N=!p(),P=Date.now(),de=async(e,t,n,r,i,a)=>{let o=await b(e,t,n,r,i,a);return o.behavior!==`allow`&&this.permissionDenials.push({type:`permission_denial`,tool_name:He(e.name),tool_use_id:i,tool_input:t}),o},L=D(),R=w?Oe(w):Ae(),ye=g||(pe()===!1?{type:`disabled`}:{type:`adaptive`});C(`before_getSystemPrompt`);let z=typeof x==`string`?x:void 0,{defaultSystemPrompt:Se,userContext:we,systemContext:Te}=await Ge({tools:o,mainLoopModel:R,additionalWorkingDirectories:Array.from(L.toolPermissionContext.additionalWorkingDirectories.keys()),mcpClients:l,customSystemPrompt:z});C(`after_getSystemPrompt`);let De={...we},V=z!==void 0&&xe()?await ne():null,ke=Fe([...z===void 0?Se:[z],...V?[V]:[],...S?[S]:[]]),Me=o.some(e=>je(e,H));E&&Me&&ae(O,d());let U={messages:this.mutableMessages,setMessages:e=>{this.mutableMessages=e(this.mutableMessages)},onChangeAPIKey:()=>{},handleElicitation:this.config.handleElicitation,options:{commands:a,debug:!1,tools:o,verbose:m,mainLoopModel:R,thinkingConfig:ye,mcpClients:l,mcpResources:{},ideInstallationStatus:null,isNonInteractiveSession:!0,customSystemPrompt:x,appendSystemPrompt:S,agentDefinitions:{activeAgents:A,allAgents:[]},theme:Ue(Ce().theme),maxBudgetUsd:v},getAppState:D,setAppState:O,abortController:this.abortController,readFileState:this.readFileState,nestedMemoryAttachmentTriggers:new Set,loadedNestedMemoryPaths:this.loadedNestedMemoryPaths,dynamicSkillDirTriggers:new Set,discoveredSkillNames:this.discoveredSkillNames,setInProgressToolUseIDs:()=>{},setResponseLength:()=>{},updateFileHistoryState:e=>{O(t=>{let n=e(t.fileHistory);return n===t.fileHistory?t:{...t,fileHistory:n}})},updateAttributionState:e=>{O(t=>{let n=e(t.attribution);return n===t.attribution?t:{...t,attribution:n}})},setSDKStatus:j};if(M&&!this.hasHandledOrphanedPermission){this.hasHandledOrphanedPermission=!0;for await(let e of se(M,o,this.mutableMessages,U))yield e}let{messages:W,shouldQuery:Ne,allowedTools:Pe,model:Le,resultText:Be}=await ze({input:e,mode:`prompt`,setToolJSX:()=>{},context:{...U,messages:this.mutableMessages},messages:this.mutableMessages,uuid:n?.uuid,isMeta:n?.isMeta,querySource:`sdk`});this.mutableMessages.push(...W);let K=[...this.mutableMessages];if(N&&W.length>0){let e=I(K);r()||(await e,(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await F())}let We=qe(),Ke=W.filter(e=>e.type===`user`&&!e.isMeta&&!e.toolUseResult&&(We?.selectableUserMessagesFilter(e)??!0)||e.type===`system`&&e.subtype===`compact_boundary`),J=k?Ke:[];O(e=>({...e,toolPermissionContext:{...e.toolPermissionContext,alwaysAllowRules:{...e.toolPermissionContext.alwaysAllowRules,command:Pe}}}));let Y=Le??R;U={messages:K,setMessages:()=>{},onChangeAPIKey:()=>{},handleElicitation:this.config.handleElicitation,options:{commands:a,debug:!1,tools:o,verbose:m,mainLoopModel:Y,thinkingConfig:ye,mcpClients:l,mcpResources:{},ideInstallationStatus:null,isNonInteractiveSession:!0,customSystemPrompt:x,appendSystemPrompt:S,theme:Ue(Ce().theme),agentDefinitions:{activeAgents:A,allAgents:[]},maxBudgetUsd:v},getAppState:D,setAppState:O,abortController:this.abortController,readFileState:this.readFileState,nestedMemoryAttachmentTriggers:new Set,loadedNestedMemoryPaths:this.loadedNestedMemoryPaths,dynamicSkillDirTriggers:new Set,discoveredSkillNames:this.discoveredSkillNames,setInProgressToolUseIDs:()=>{},setResponseLength:()=>{},updateFileHistoryState:U.updateFileHistoryState,updateAttributionState:U.updateAttributionState,setSDKStatus:j},C(`before_skills_plugins`);let[Je,{enabled:Ye}]=await Promise.all([ue(be()),ge()]);if(C(`after_skills_plugins`),yield Ve({tools:o,mcpClients:l,model:Y,permissionMode:L.toolPermissionContext.mode,commands:a,agents:A,skills:Je,plugins:Ye,fastMode:L.fastMode}),C(`system_message_yielded`),!Ne){for(let e of W)if(e.type===`user`&&typeof e.message.content==`string`&&(e.message.content.includes(`<local-command-stdout>`)||e.message.content.includes(`<local-command-stderr>`)||e.isCompactSummary)&&(yield{type:`user`,message:{...e.message,content:s(e.message.content)},session_id:d(),parent_tool_use_id:null,uuid:e.uuid,timestamp:e.timestamp,isReplay:!e.isCompactSummary,isSynthetic:e.isMeta||e.isVisibleInTranscriptOnly}),e.type===`system`&&e.subtype===`local_command`&&typeof e.content==`string`&&(e.content.includes(`<local-command-stdout>`)||e.content.includes(`<local-command-stderr>`))&&(yield Re(e.content,e.uuid)),e.type===`system`&&e.subtype===`compact_boundary`){let t=e;yield{type:`system`,subtype:`compact_boundary`,session_id:d(),uuid:e.uuid,compact_metadata:Ie(t.compactMetadata)}}N&&(await I(K),(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await F()),yield{type:`result`,subtype:`success`,is_error:!1,duration_ms:Date.now()-P,duration_api_ms:c(),num_turns:K.length-1,result:Be??``,stop_reason:null,session_id:d(),total_cost_usd:f(),usage:this.totalUsage,modelUsage:u(),permission_denials:this.permissionDenials,fast_mode_state:B(Y,L.fastMode),uuid:q()};return}if(ve()&&N){let e=qe()?.selectableUserMessagesFilter??(e=>!0);W.filter(e).forEach(e=>{me(e=>{O(t=>({...t,fileHistory:e(t.fileHistory)}))},e.uuid)})}let X=G,Z=1,Xe=!1,Ze,Q=null,Qe=Ee().at(-1),$e=E?fe(this.mutableMessages,H):0;for await(let e of te({messages:K,systemPrompt:ke,userContext:De,systemContext:Te,canUseTool:de,toolUseContext:U,fallbackModel:T,querySource:`sdk`,maxTurns:_,taskBudget:y})){if(e.type===`assistant`||e.type===`user`||e.type===`system`&&e.subtype===`compact_boundary`){if(N&&e.type===`system`&&e.subtype===`compact_boundary`){let t=e.compactMetadata?.preservedSegment?.tailUuid;if(t){let e=this.mutableMessages.findLastIndex(e=>e.uuid===t);e!==-1&&await I(this.mutableMessages.slice(0,e+1))}}if(K.push(e),N&&(e.type===`assistant`?I(K):await I(K)),!Xe&&J.length>0){Xe=!0;for(let e of J)e.type===`user`&&(yield{type:`user`,message:e.message,session_id:d(),parent_tool_use_id:null,uuid:e.uuid,timestamp:e.timestamp,isReplay:!0})}}switch(e.type===`user`&&Z++,e.type){case`tombstone`:break;case`assistant`:{let t=e,n=t.message?.stop_reason;n!=null&&(Q=n),this.mutableMessages.push(t),yield*ce(t);break}case`progress`:{let t=e;this.mutableMessages.push(t),N&&(K.push(t),I(K)),yield*ce(t);break}case`user`:{let t=e;this.mutableMessages.push(t),yield*ce(t);break}case`stream_event`:{let t=e.event;if(t.type===`message_start`){X=G;let e=t.message;X=he(X,e.usage)}if(t.type===`message_delta`){X=he(X,t.usage);let e=t.delta;e.stop_reason!=null&&(Q=e.stop_reason)}t.type===`message_stop`&&(this.totalUsage=ee(this.totalUsage,X)),le&&(yield{type:`stream_event`,event:t,session_id:d(),parent_tool_use_id:null,uuid:q()});break}case`attachment`:{let n=e;this.mutableMessages.push(n),N&&(K.push(n),I(K));let r=n.attachment;if(r.type===`structured_output`)Ze=r.data;else if(r.type===`max_turns_reached`){N&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await F(),yield{type:`result`,subtype:`error_max_turns`,duration_ms:Date.now()-P,duration_api_ms:c(),is_error:!0,num_turns:r.turnCount,stop_reason:Q,session_id:d(),total_cost_usd:f(),usage:this.totalUsage,modelUsage:u(),permission_denials:this.permissionDenials,fast_mode_state:B(Y,L.fastMode),uuid:q(),errors:[`Reached maximum number of turns (${r.maxTurns})`]};return}else k&&r.type===`queued_command`&&(yield{type:`user`,message:{role:`user`,content:r.prompt},session_id:d(),parent_tool_use_id:null,uuid:r.source_uuid||n.uuid,timestamp:n.timestamp,isReplay:!0});break}case`stream_request_start`:break;case`system`:{let t=e,n=this.config.snipReplay?.(t,this.mutableMessages);if(n!==void 0){n.executed&&(this.mutableMessages.length=0,this.mutableMessages.push(...n.messages));break}if(this.mutableMessages.push(t),t.subtype===`compact_boundary`&&t.compactMetadata){let e=t,n=this.mutableMessages.length-1;n>0&&this.mutableMessages.splice(0,n);let r=K.length-1;r>0&&K.splice(0,r),yield{type:`system`,subtype:`compact_boundary`,session_id:d(),uuid:t.uuid,compact_metadata:Ie(e.compactMetadata)}}if(t.subtype===`api_error`){let e=t;yield{type:`system`,subtype:`api_retry`,attempt:e.retryAttempt,max_retries:e.maxRetries,retry_delay_ms:e.retryInMs,error_status:e.error.status??null,error:re(e.error),session_id:d(),uuid:t.uuid}}break}case`tool_use_summary`:{let t=e;yield{type:`tool_use_summary`,summary:t.summary,preceding_tool_use_ids:t.precedingToolUseIds,session_id:d(),uuid:t.uuid};break}}if(v!==void 0&&f()>=v){N&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await F(),yield{type:`result`,subtype:`error_max_budget_usd`,duration_ms:Date.now()-P,duration_api_ms:c(),is_error:!0,num_turns:Z,stop_reason:Q,session_id:d(),total_cost_usd:f(),usage:this.totalUsage,modelUsage:u(),permission_denials:this.permissionDenials,fast_mode_state:B(Y,L.fastMode),uuid:q(),errors:[`Reached maximum budget ($${v}). Increase the limit with --max-budget-usd or start a new session.`]};return}if(e.type===`user`&&E){let e=fe(this.mutableMessages,H)-$e,n=parseInt(process.env.MAX_STRUCTURED_OUTPUT_RETRIES||`5`,10);if(e>=n){N&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await F(),yield{type:`result`,subtype:`error_max_structured_output_retries`,duration_ms:Date.now()-P,duration_api_ms:c(),is_error:!0,num_turns:Z,stop_reason:Q,session_id:d(),total_cost_usd:f(),usage:this.totalUsage,modelUsage:u(),permission_denials:this.permissionDenials,fast_mode_state:B(Y,L.fastMode),uuid:q(),errors:[`Failed to provide valid structured output after ${n} attempts`]};return}}}let $=K.findLast(e=>e.type===`assistant`||e.type===`user`),et=$?.type??`undefined`,tt=$?.type===`assistant`?_e($.message.content)?.type??`none`:`n/a`;if(N&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await F(),!oe($,Q)){yield{type:`result`,subtype:`error_during_execution`,duration_ms:Date.now()-P,duration_api_ms:c(),is_error:!0,num_turns:Z,stop_reason:Q,session_id:d(),total_cost_usd:f(),usage:this.totalUsage,modelUsage:u(),permission_denials:this.permissionDenials,fast_mode_state:B(Y,L.fastMode),uuid:q(),errors:(()=>{let e=Ee(),t=Qe?e.lastIndexOf(Qe)+1:0;return[`[ede_diagnostic] result_type=${et} last_content_type=${tt} stop_reason=${Q}`,...e.slice(t).map(e=>e.error)]})()};return}let nt=``,rt=!1;if($.type===`assistant`){let e=_e($.message.content);e?.type===`text`&&!h.has(e.text)&&(nt=e.text),rt=!!$.isApiErrorMessage}yield{type:`result`,subtype:`success`,is_error:rt,duration_ms:Date.now()-P,duration_api_ms:c(),num_turns:Z,result:nt,stop_reason:Q,session_id:d(),total_cost_usd:f(),usage:this.totalUsage,modelUsage:u(),permission_denials:this.permissionDenials,structured_output:Ze,fast_mode_state:B(Y,L.fastMode),uuid:q()}}interrupt(){this.abortController.abort()}resetAbortController(){this.abortController=y()}getAbortSignal(){return this.abortController.signal}getMessages(){return this.mutableMessages}getReadFileState(){return this.readFileState}getSessionId(){return d()}setModel(e){this.config.userSpecifiedModel=e}};async function*Y({commands:e,prompt:t,promptUuid:n,isMeta:r,cwd:i,tools:a,mcpClients:o,verbose:s=!1,thinkingConfig:c,maxTurns:l,maxBudgetUsd:u,taskBudget:d,canUseTool:f,mutableMessages:p=[],getReadFileCache:ee,setReadFileCache:m,customSystemPrompt:h,appendSystemPrompt:g,userSpecifiedModel:_,fallbackModel:te,jsonSchema:v,getAppState:y,setAppState:b,abortController:x,replayUserMessages:S=!1,includePartialMessages:C=!1,handleElicitation:w,agents:ne=[],setSDKStatus:T,orphanedPermission:re}){let E=new J({cwd:i,tools:a,commands:e,mcpClients:o,agents:ne??[],canUseTool:f,getAppState:y,setAppState:b,initialMessages:p,readFileCache:W(ee()),customSystemPrompt:h,appendSystemPrompt:g,userSpecifiedModel:_,fallbackModel:te,thinkingConfig:c,maxTurns:l,maxBudgetUsd:u,taskBudget:d,jsonSchema:v,verbose:s,handleElicitation:w,replayUserMessages:S,includePartialMessages:C,setSDKStatus:T,abortController:x,orphanedPermission:re});try{yield*E.submitMessage(t,{uuid:n,isMeta:r})}finally{m(E.getReadFileState())}}export{Y as n,Ke as r,J as t};