@readmagic/workmate 0.0.1

package/dist/chunks/QueryEngine-D2HUnwmU.js

@@ -0,0 +1 @@
+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-4KQ3kaqz.js";import{$ as m,GT as h,Gn as g,Gp as _,Gy as v,Hy as ee,KT as y,Kn as te,ND as b,Nl as x,PD as S,Pl as ne,Qi as C,Sn as re,Wp as w,Xi as ie,_C as T,aw as ae,dT as E,dw as D,et as O,fs as k,gC as oe,iw as A,jr as se,lO as ce,lo as le,lr as j,lt as M,ms as ue,nT as N,oO as P,oo as de,op as F,pw as I,qT as fe,rt as pe,si as L,tT as me,ut as R,vC as he,xn as z,yC as B,zo as ge,zy as _e}from"./loadAgentsDir-nzr38hai.js";import{n as V,t as ve}from"./cwd-BuJhkTrc.js";import{N as ye,P as be,c as xe,g as Se}from"./config-C7YHKCPV.js";import{Y as Ce,i as we,o as Te}from"./log-B3eB4EFv.js";import{C as Ee,X as H,_ as U,tt as De,u as Oe}from"./model-DfFA3qMO.js";import{Cn as ke,ct as Ae,dt as W,wn as je}from"./prompt-Bef63FhB.js";import{i as G,o as Me}from"./fileStateCache-DDW5toIs.js";import{P as Ne,lt as Pe,st as Fe}from"./rcDebugLog-BUuD-X1D.js";import{a as Ie,n as Le,r as Re}from"./mappers-CWykjGfP.js";import{c as ze,n as Be,o as Ve,r as K,s as He}from"./MessageSelector-_UZB8cGb.js";import{n as Ue,t as We}from"./systemTheme-Ceakddth.js";import{randomUUID as q}from"crypto";n(),i(),o(),l(),V(),Ce(),Te(),be(),Se(),De(),ke(),b(),W(),Me(),Ne(),N(),I(),v(),F(),x(),k(),ge(),C(),se(),g(),z(),O(),R(),fe(),ae(),U(),E(),P();async function Ge({tools:e,mainLoopModel:t,additionalWorkingDirectories:n,mcpClients:r,customSystemPrompt:i}){let[a,o,s]=await Promise.all([i===void 0?M(e,t,n,r):Promise.resolve([]),y(),i===void 0?h():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=Oe(),f=a(),{defaultSystemPrompt:p,userContext:m,systemContext:h}=await Ge({tools:e,mainLoopModel:d,additionalWorkingDirectories:Array.from(f.toolPermissionContext.additionalWorkingDirectories.keys()),mcpClients:n,customSystemPrompt:s}),g=Fe([...s===void 0?p:[s],...c?[c]:[]]),_=r.at(-1),v=_?.type===`assistant`&&_.message.stop_reason===null?r.slice(0,-1):r;return{systemPrompt:g,userContext:m,systemContext:h,toolUseContext:{options:{commands:t,debug:!1,mainLoopModel:d,tools:e,verbose:!1,thinkingConfig:l??(ce()===!1?{type:`disabled`}:{type:`adaptive`}),mcpClients:n,mcpResources:{},isNonInteractiveSession:!0,agentDefinitions:{activeAgents:u,allAgents:[]},customSystemPrompt:s,appendSystemPrompt:c},abortController:A(),readFileState:i,getAppState:a,setAppState:o,messages:v,setInProgressToolUseIDs:()=>{},setResponseLength:()=>{},updateFileHistoryState:()=>{},updateAttributionState:()=>{}},forkContextMessages:v}}ae(),U(),w(),E(),We(),P(),Le(),T();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??A(),this.permissionDenials=[],this.readFileState=e.readFileCache,this.totalUsage=Pe}async*submitMessage(e,n){let{cwd:i,commands:a,tools:o,mcpClients:l,verbose:h=!1,thinkingConfig:g,maxTurns:v,maxBudgetUsd:y,taskBudget:b,canUseTool:x,customSystemPrompt:C,appendSystemPrompt:w,userSpecifiedModel:T,fallbackModel:ae,jsonSchema:E,getAppState:O,setAppState:k,replayUserMessages:A=!1,includePartialMessages:se=!1,agents:M=[],setSDKStatus:N,orphanedPermission:P}=this.config;this.discoveredSkillNames.clear(),this.permissionDenials=[],_(i);let F=!p(),I=Date.now(),fe=async(e,t,n,r,i,a)=>{let o=await x(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},R=O(),z=T?Ee(T):Oe(),ge=g||(ce()===!1?{type:`disabled`}:{type:`adaptive`});D(`before_getSystemPrompt`);let V=typeof C==`string`?C:void 0,{defaultSystemPrompt:be,userContext:Se,systemContext:Ce}=await Ge({tools:o,mainLoopModel:z,additionalWorkingDirectories:Array.from(R.toolPermissionContext.additionalWorkingDirectories.keys()),mcpClients:l,customSystemPrompt:V});D(`after_getSystemPrompt`);let Te={...Se},U=V!==void 0&&ye()?await te():null,De=Fe([...V===void 0?be:[V],...U?[U]:[],...w?[w]:[]]),ke=o.some(e=>je(e,Ae));E&&ke&&re(k,d());let W={messages:this.mutableMessages,setMessages:e=>{this.mutableMessages=e(this.mutableMessages)},onChangeAPIKey:()=>{},handleElicitation:this.config.handleElicitation,options:{commands:a,debug:!1,tools:o,verbose:h,mainLoopModel:z,thinkingConfig:ge,mcpClients:l,mcpResources:{},ideInstallationStatus:null,isNonInteractiveSession:!0,customSystemPrompt:C,appendSystemPrompt:w,agentDefinitions:{activeAgents:M,allAgents:[]},theme:Ue(xe().theme),maxBudgetUsd:y},getAppState:O,setAppState:k,abortController:this.abortController,readFileState:this.readFileState,nestedMemoryAttachmentTriggers:new Set,loadedNestedMemoryPaths:this.loadedNestedMemoryPaths,dynamicSkillDirTriggers:new Set,discoveredSkillNames:this.discoveredSkillNames,setInProgressToolUseIDs:()=>{},setResponseLength:()=>{},updateFileHistoryState:e=>{k(t=>{let n=e(t.fileHistory);return n===t.fileHistory?t:{...t,fileHistory:n}})},updateAttributionState:e=>{k(t=>{let n=e(t.attribution);return n===t.attribution?t:{...t,attribution:n}})},setSDKStatus:N};if(P&&!this.hasHandledOrphanedPermission){this.hasHandledOrphanedPermission=!0;for await(let e of oe(P,o,this.mutableMessages,W))yield e}let{messages:G,shouldQuery:Me,allowedTools:Ne,model:Le,resultText:Be}=await ze({input:e,mode:`prompt`,setToolJSX:()=>{},context:{...W,messages:this.mutableMessages},messages:this.mutableMessages,uuid:n?.uuid,isMeta:n?.isMeta,querySource:`sdk`});this.mutableMessages.push(...G);let K=[...this.mutableMessages];if(F&&G.length>0){let e=L(K);r()||(await e,(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await j())}let We=qe(),Ke=G.filter(e=>e.type===`user`&&!e.isMeta&&!e.toolUseResult&&(We?.selectableUserMessagesFilter(e)??!0)||e.type===`system`&&e.subtype===`compact_boundary`),J=A?Ke:[];k(e=>({...e,toolPermissionContext:{...e.toolPermissionContext,alwaysAllowRules:{...e.toolPermissionContext.alwaysAllowRules,command:Ne}}}));let Y=Le??z;W={messages:K,setMessages:()=>{},onChangeAPIKey:()=>{},handleElicitation:this.config.handleElicitation,options:{commands:a,debug:!1,tools:o,verbose:h,mainLoopModel:Y,thinkingConfig:ge,mcpClients:l,mcpResources:{},ideInstallationStatus:null,isNonInteractiveSession:!0,customSystemPrompt:C,appendSystemPrompt:w,theme:Ue(xe().theme),agentDefinitions:{activeAgents:M,allAgents:[]},maxBudgetUsd:y},getAppState:O,setAppState:k,abortController:this.abortController,readFileState:this.readFileState,nestedMemoryAttachmentTriggers:new Set,loadedNestedMemoryPaths:this.loadedNestedMemoryPaths,dynamicSkillDirTriggers:new Set,discoveredSkillNames:this.discoveredSkillNames,setInProgressToolUseIDs:()=>{},setResponseLength:()=>{},updateFileHistoryState:W.updateFileHistoryState,updateAttributionState:W.updateAttributionState,setSDKStatus:N},D(`before_skills_plugins`);let[Je,{enabled:Ye}]=await Promise.all([ie(ve()),ue()]);if(D(`after_skills_plugins`),yield Ve({tools:o,mcpClients:l,model:Y,permissionMode:R.toolPermissionContext.mode,commands:a,agents:M,skills:Je,plugins:Ye,fastMode:R.fastMode}),D(`system_message_yielded`),!Me){for(let e of G)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)}}F&&(await L(K),(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await j()),yield{type:`result`,subtype:`success`,is_error:!1,duration_ms:Date.now()-I,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:H(Y,R.fastMode),uuid:q()};return}if(_e()&&F){let e=qe()?.selectableUserMessagesFilter??(e=>!0);G.filter(e).forEach(e=>{ee(e=>{k(t=>({...t,fileHistory:e(t.fileHistory)}))},e.uuid)})}let X=Pe,Z=1,Xe=!1,Ze,Q=null,Qe=we().at(-1),$e=E?le(this.mutableMessages,Ae):0;for await(let e of ne({messages:K,systemPrompt:De,userContext:Te,systemContext:Ce,canUseTool:fe,toolUseContext:W,fallbackModel:ae,querySource:`sdk`,maxTurns:v,taskBudget:b})){if(e.type===`assistant`||e.type===`user`||e.type===`system`&&e.subtype===`compact_boundary`){if(F&&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 L(this.mutableMessages.slice(0,e+1))}}if(K.push(e),F&&(e.type===`assistant`?L(K):await L(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*B(t);break}case`progress`:{let t=e;this.mutableMessages.push(t),F&&(K.push(t),L(K)),yield*B(t);break}case`user`:{let t=e;this.mutableMessages.push(t),yield*B(t);break}case`stream_event`:{let t=e.event;if(t.type===`message_start`){X=Pe;let e=t.message;X=pe(X,e.usage)}if(t.type===`message_delta`){X=pe(X,t.usage);let e=t.delta;e.stop_reason!=null&&(Q=e.stop_reason)}t.type===`message_stop`&&(this.totalUsage=m(this.totalUsage,X)),se&&(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),F&&(K.push(n),L(K));let r=n.attachment;if(r.type===`structured_output`)Ze=r.data;else if(r.type===`max_turns_reached`){F&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await j(),yield{type:`result`,subtype:`error_max_turns`,duration_ms:Date.now()-I,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:H(Y,R.fastMode),uuid:q(),errors:[`Reached maximum number of turns (${r.maxTurns})`]};return}else A&&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:me(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(y!==void 0&&f()>=y){F&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await j(),yield{type:`result`,subtype:`error_max_budget_usd`,duration_ms:Date.now()-I,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:H(Y,R.fastMode),uuid:q(),errors:[`Reached maximum budget ($${y}). Increase the limit with --max-budget-usd or start a new session.`]};return}if(e.type===`user`&&E){let e=le(this.mutableMessages,Ae)-$e,n=parseInt(process.env.MAX_STRUCTURED_OUTPUT_RETRIES||`5`,10);if(e>=n){F&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await j(),yield{type:`result`,subtype:`error_max_structured_output_retries`,duration_ms:Date.now()-I,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:H(Y,R.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`?S($.message.content)?.type??`none`:`n/a`;if(F&&(t(process.env.CLAUDE_CODE_EAGER_FLUSH)||t(process.env.CLAUDE_CODE_IS_COWORK))&&await j(),!he($,Q)){yield{type:`result`,subtype:`error_during_execution`,duration_ms:Date.now()-I,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:H(Y,R.fastMode),uuid:q(),errors:(()=>{let e=we(),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=S($.message.content);e?.type===`text`&&!de.has(e.text)&&(nt=e.text),rt=!!$.isApiErrorMessage}yield{type:`result`,subtype:`success`,is_error:rt,duration_ms:Date.now()-I,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:H(Y,R.fastMode),uuid:q()}}interrupt(){this.abortController.abort()}resetAbortController(){this.abortController=A()}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:m,setReadFileCache:h,customSystemPrompt:g,appendSystemPrompt:_,userSpecifiedModel:v,fallbackModel:ee,jsonSchema:y,getAppState:te,setAppState:b,abortController:x,replayUserMessages:S=!1,includePartialMessages:ne=!1,handleElicitation:C,agents:re=[],setSDKStatus:w,orphanedPermission:ie}){let T=new J({cwd:i,tools:a,commands:e,mcpClients:o,agents:re??[],canUseTool:f,getAppState:te,setAppState:b,initialMessages:p,readFileCache:G(m()),customSystemPrompt:g,appendSystemPrompt:_,userSpecifiedModel:v,fallbackModel:ee,thinkingConfig:c,maxTurns:l,maxBudgetUsd:u,taskBudget:d,jsonSchema:y,verbose:s,handleElicitation:C,replayUserMessages:S,includePartialMessages:ne,setSDKStatus:w,abortController:x,orphanedPermission:ie});try{yield*T.submitMessage(t,{uuid:n,isMeta:r})}finally{h(T.getReadFileState())}}export{Y as n,Ke as r,J as t};