@zibby/core 0.3.10 → 0.3.12
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/index.js +66 -66
- package/dist/package.json +1 -1
- package/dist/register-built-in-strategies.js +27 -27
- package/dist/strategies/claude-strategy.js +2 -2
- package/dist/strategies/index.js +30 -30
- package/package.json +1 -1
package/dist/package.json
CHANGED
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import{registerStrategy as pe}from"@zibby/agent-workflow";import{AgentStrategy as
|
|
1
|
+
import{registerStrategy as pe}from"@zibby/agent-workflow";import{AgentStrategy as kt,DEFAULT_OUTPUT_BASE as Pt,SESSION_INFO_FILE as Nt,STOP_REQUEST_FILE as Rt,getAllSkills as Mt,getSkill as Ze}from"@zibby/agent-workflow";import{spawn as Lt,execSync as Z}from"node:child_process";import{writeFileSync as We,readFileSync as qe,mkdirSync as Ve,existsSync as le,accessSync as Qe,constants as et,unlinkSync as Dt}from"node:fs";import{join as j,resolve as Ut}from"node:path";import{homedir as ae}from"node:os";import ie from"chalk";var F={debug:0,info:1,warn:2,error:3,silent:4},Ce=class{constructor(){this._level=this._getLogLevel()}_getLogLevel(){if(process.env.ZIBBY_DEBUG==="true")return F.debug;if(process.env.ZIBBY_VERBOSE==="true")return F.info;let e=process.env.LOG_LEVEL?.toLowerCase();return e&&e in F?F[e]:F.info}_shouldLog(e){return F[e]>=this._level}_formatMessage(e,t,r={}){let o=new Date().toISOString(),i=`${this._getPrefix(e)} ${t}`;return Object.keys(r).length>0&&(i+=ie.dim(` ${JSON.stringify(r)}`)),i}_getPrefix(e){return{debug:ie.gray("[DEBUG]"),info:ie.cyan("[INFO]"),warn:ie.yellow("[WARN]"),error:ie.red("\u274C [ERROR]")}[e]||""}debug(e,t){this._shouldLog("debug")&&console.log(this._formatMessage("debug",e,t))}info(e,t){this._shouldLog("info")&&console.log(this._formatMessage("info",e,t))}warn(e,t){this._shouldLog("warn")&&console.warn(this._formatMessage("warn",e,t))}error(e,t){this._shouldLog("error")&&console.error(this._formatMessage("error",e,t))}setLevel(e){e in F&&(this._level=F[e])}getLevel(){return Object.keys(F).find(e=>F[e]===this._level)}},a=new Ce;var K={ASSISTANT:"gpt-5.4-nano-2026-03-17",CLAUDE:"claude-sonnet-4-6",CURSOR:"auto",CODEX:"o4-mini",GEMINI:"gemini-2.5-pro",OPENAI_POSTPROCESSING:"gpt-4o-mini"};var ke={auto:"claude-sonnet-4-6","sonnet-4.6":"claude-sonnet-4-6","sonnet-4-6":"claude-sonnet-4-6","opus-4.6":"claude-opus-4-6","opus-4-6":"claude-opus-4-6","sonnet-4.5":"claude-sonnet-4-5-20250929","sonnet-4-5":"claude-sonnet-4-5-20250929","opus-4.5":"claude-opus-4-20250514","opus-4-5":"claude-opus-4-20250514","claude-sonnet-4-6":"claude-sonnet-4-6","claude-opus-4-6":"claude-opus-4-6","claude-sonnet-4-5-20250929":"claude-sonnet-4-5-20250929","claude-opus-4-20250514":"claude-opus-4-20250514"},Pe={auto:"o4-mini","o4-mini":"o4-mini",o3:"o3","o3-mini":"o3-mini","codex-mini":"codex-mini-latest","gpt-4o":"gpt-4o","gpt-4o-mini":"gpt-4o-mini","gpt-5.2-codex":"gpt-5.2-codex","gpt-5.2":"gpt-5.2","gpt-5.3":"gpt-5.3","gpt-5.4":"gpt-5.4"},je={auto:"gemini-2.5-pro","gemini-2.5-pro":"gemini-2.5-pro","gemini-2.5-flash":"gemini-2.5-flash"},me={CURSOR_AGENT_DEFAULT:1200*1e3,OPENAI_REQUEST:18e4};var q=class n{constructor(){this.buffer="",this.extractedResult=null,this.rawText="",this.zodSchema=null,this.lastOutputLength=0,this.onToolCall=null,this._lastToolEmit=null}processChunk(e){if(!e)return null;this.buffer+=e;let t=this.buffer.split(`
|
|
2
2
|
`);this.buffer=t.pop()||"";let r="";for(let o of t)if(o.trim())try{let s=JSON.parse(o);this._emitToolCalls(s);let i=this.extractText(s);if(i){if(this.rawText&&i.startsWith(this.rawText)){let l=i.substring(this.rawText.length);this.rawText=i,r+=l}else(!this.rawText.includes(i)||i.length<20)&&(this.rawText+=i,r+=i);this.tryExtractResult(this.rawText)}else this.isValidResult(s)&&(this.rawText+=`${o}
|
|
3
3
|
`,r+=`${o}
|
|
4
4
|
`,this.extractedResult=s)}catch{if(o.includes('"text"')||o.includes('"content"')){let i=o.match(/"text"\s*:\s*"([^"]*)/),l=o.match(/"content"\s*:\s*"([^"]*)/),c=i?i[1]:l?l[1]:null;c&&!this.rawText.includes(c)&&(r+=c,this.rawText+=c)}}return r||null}flush(){if(!this.buffer.trim())return null;let e="";try{let t=JSON.parse(this.buffer);this._emitToolCalls(t);let r=this.extractText(t);r&&(this.rawText+=r,e+=r,this.tryExtractResult(r))}catch{this.rawText+=this.buffer,e+=this.buffer,this.tryExtractResult(this.buffer)}return this.buffer="",e||null}_emitToolCalls(e){if(!this.onToolCall)return;let t=(i,l)=>{if(!i)return;let c=`${i}:${JSON.stringify(l??{})}`;this._lastToolEmit!==c&&(this._lastToolEmit=c,this.onToolCall(i,l??void 0))},r=i=>{if(i!=null){if(typeof i=="object"&&!Array.isArray(i))return i;if(typeof i=="string")try{return JSON.parse(i)}catch{return}}};if(e.type==="tool_use"||e.type==="tool_call"){if(e.name){t(e.name,r(e.input??e.arguments));return}let i=e.tool_call;if(i&&typeof i=="object"&&!Array.isArray(i)){let l=Object.keys(i);if(l.length===1){let c=l[0],p=i[c],f=p&&typeof p=="object"?p.args??p.input??p:void 0;t(c,r(f))}return}return}if(Array.isArray(e.tool_calls)){for(let i of e.tool_calls)t(i.name,r(i.input??i.arguments));return}let o=e.message??e;if(Array.isArray(o?.tool_calls)){for(let i of o.tool_calls)t(i.name,r(i.input??i.arguments));return}let s=o?.content??e.content;if(Array.isArray(s))for(let i of s)(i.type==="tool_use"||i.type==="tool_call")&&i.name&&t(i.name,r(i.input??i.arguments))}extractText(e){if(e.type==="assistant"&&e.message?.content){let t=e.message.content;if(Array.isArray(t))return t.filter(r=>r.type==="text"&&r.text).map(r=>r.text).join("")}return e.type==="thinking"&&e.text||e.text?e.text:e.content&&typeof e.content=="string"?e.content:e.delta?e.delta:null}tryExtractResult(e){if(!e||typeof e!="string")return;let t=[],r=/```json\s*\n?([\s\S]*?)\n?```/g,o;for(;(o=r.exec(e))!==null;){let _=o[1].trim();try{JSON.parse(_),t.push({text:_,source:"markdown"})}catch{}}let s=0,i=0;for(;s<e.length&&(s=e.indexOf("{",s),s!==-1);){let _=0,m=s;for(let h=s;h<e.length;h++)if(e[h]==="{")_++;else if(e[h]==="}"&&(_--,_===0)){m=h,t.push({text:e.substring(s,m+1),source:"brace"}),i++;break}s=m+1}let l=this.extractedResult,c=l?JSON.stringify(l).length:0,p=0,f=-1;for(let _=0;_<t.length;_++){let m=t[_];try{let h=m.text.replace(/,(\s*[}\]])/g,"$1"),O=JSON.parse(h);this.isValidResult(O)&&(p++,c=JSON.stringify(O).length,l=O,f=_)}catch{}}l&&(this.extractedResult=l)}isValidResult(e){if(!e||typeof e!="object"||Array.isArray(e)||e.session_id||e.timestamp_ms||e.type||e.call_id||e.tool_call||e.result&&typeof e.result=="object"&&(e.result.success&&typeof e.result.success=="object"||e.result.error&&typeof e.result.error=="object")||e.args&&typeof e.args=="object")return!1;if(this.zodSchema)try{return this.zodSchema.parse(e),!0}catch{return!1}return!0}getResult(){return this.extractedResult}getRawText(){return this.rawText}static extractResult(e,t=null){let r=new n;r.zodSchema=t,r.processChunk(e),r.flush();let o=r.getResult();return!o&&process.env.LOG_LEVEL==="debug"&&console.error("[StreamingParser] No result extracted from",e?.length||0,"chars"),o}};import{zodToJsonSchema as mt}from"zod-to-json-schema";var ge=class{static generateFileOutputInstructions(e,t){let r;typeof e?.parse=="function"?r=mt(e,{target:"openApi3"}):r=e;let o=this._buildExample(r);return`
|
|
@@ -37,9 +37,9 @@ ${n.slice(-i)}`);let c=`Extract and format the following information into struct
|
|
|
37
37
|
RAW CONTENT:
|
|
38
38
|
${l}
|
|
39
39
|
|
|
40
|
-
Extract all relevant information and format it according to the schema. If any required fields are missing, do your best to infer them from the content.`,p={model:
|
|
40
|
+
Extract all relevant information and format it according to the schema. If any required fields are missing, do your best to infer them from the content.`,p={model:K.OPENAI_POSTPROCESSING,messages:[{role:"user",content:c}],response_format:{type:"json_schema",json_schema:{name:"extract",schema:s,strict:!0}}};a.info(`\u{1F4E4} Sending to OpenAI proxy: model=${K.OPENAI_POSTPROCESSING}, schema keys=${Object.keys(s.properties||{}).join(", ")}`),a.debug(` Schema size: ${JSON.stringify(s).length} chars`),a.debug(` Prompt size: ${c.length} chars`);try{let f={"Content-Type":"application/json"};process.env.OPENAI_PROXY_TOKEN?(f["x-proxy-token"]=t,f["x-execution-id"]=process.env.EXECUTION_ID||""):(f.Authorization=`Bearer ${t}`,f["x-api-key"]=process.env.ZIBBY_API_KEY||"",f["x-execution-id"]=process.env.EXECUTION_ID||"");let m=(await gt.post(r,p,{headers:f,timeout:me.OPENAI_REQUEST})).data?.choices?.[0]?.message?.content;if(!m)throw new Error("OpenAI proxy returned empty response");let h=JSON.parse(m);return a.info("\u2705 Successfully formatted with OpenAI proxy"),{structured:h,raw:n}}catch(f){if(f.response){let _=f.response.status,m=f.response.data;throw a.error(`\u274C OpenAI proxy request failed: ${_}`),a.error(` Status: ${_}`),a.error(` Response: ${JSON.stringify(m,null,2)}`),_===401||_===403?new Error(`Authentication failed for OpenAI proxy.
|
|
41
41
|
Run \`zibby login\` or set ZIBBY_USER_TOKEN environment variable.
|
|
42
|
-
Response: ${JSON.stringify(m)}`,{cause:f}):new Error(`Failed to format Cursor output: ${m?.error?.message||"Unknown error"}`,{cause:f})}throw a.error(`\u274C OpenAI proxy request failed: ${f.message}`),new Error(`Failed to format output: ${f.message}`,{cause:f})}}function G(n){let e=`${n}_POOL`,t=process.env[e];if(!t||typeof t!="string")return{picked:null,count:0,dispose:()=>{}};let r=t.split(/[,\n]+/).map(i=>i.trim()).filter(Boolean);if(r.length===0)return{picked:null,count:0,dispose:()=>{}};let o=r[Math.floor(Math.random()*r.length)],s=process.env[n];return process.env[n]=o,a.debug(`[auth-pool] ${n}: picked 1 of ${r.length} from pool (***${o.slice(-4)})`),{picked:o,count:r.length,dispose(){s===void 0?delete process.env[n]:process.env[n]=s}}}import{timeline as Y,Timeline as jo,WORKFLOW_GRAPH_LOG_MARKER_PREFIX as Fo}from"@zibby/agent-workflow";import{copyFileSync as bt,existsSync as Ne,lstatSync as wt,mkdirSync as Ye,rmSync as Tt,symlinkSync as xt,unlinkSync as
|
|
42
|
+
Response: ${JSON.stringify(m)}`,{cause:f}):new Error(`Failed to format Cursor output: ${m?.error?.message||"Unknown error"}`,{cause:f})}throw a.error(`\u274C OpenAI proxy request failed: ${f.message}`),new Error(`Failed to format output: ${f.message}`,{cause:f})}}function G(n){let e=`${n}_POOL`,t=process.env[e];if(!t||typeof t!="string")return{picked:null,count:0,dispose:()=>{}};let r=t.split(/[,\n]+/).map(i=>i.trim()).filter(Boolean);if(r.length===0)return{picked:null,count:0,dispose:()=>{}};let o=r[Math.floor(Math.random()*r.length)],s=process.env[n];return process.env[n]=o,a.debug(`[auth-pool] ${n}: picked 1 of ${r.length} from pool (***${o.slice(-4)})`),{picked:o,count:r.length,dispose(){s===void 0?delete process.env[n]:process.env[n]=s}}}import{timeline as Y,Timeline as jo,WORKFLOW_GRAPH_LOG_MARKER_PREFIX as Fo}from"@zibby/agent-workflow";import{copyFileSync as bt,existsSync as Ne,lstatSync as wt,mkdirSync as Ye,rmSync as Tt,symlinkSync as xt,unlinkSync as $t}from"node:fs";import{join as z}from"node:path";import{homedir as vt}from"node:os";import{randomBytes as It}from"node:crypto";var Ct=["cli-config.json","config.json","auth.json","argv.json"];function ze(n){return!(!n||typeof n!="string"||process.env.ZIBBY_CURSOR_USE_GLOBAL_MCP==="1"||process.env.ZIBBY_CURSOR_USE_GLOBAL_MCP==="true")}function He(n){let e=z(n||process.cwd(),".zibby","tmp");Ye(e,{recursive:!0});let t=`${process.pid}-${Date.now()}-${It(4).toString("hex")}`,r=z(e,`cursor-agent-home-${t}`),o=z(r,".cursor");Ye(o,{recursive:!0});let s=vt(),i=z(s,".cursor");if(Ne(i))for(let l of Ct){let c=z(i,l);if(Ne(c))try{bt(c,z(o,l))}catch{}}if(process.platform==="darwin"){let l=z(s,"Library");if(Ne(l))try{xt(l,z(r,"Library"))}catch{}}return r}function Xe(n){if(!(!n||typeof n!="string"))try{let e=z(n,"Library");try{wt(e).isSymbolicLink()&&$t(e)}catch{}Tt(n,{recursive:!0,force:!0})}catch{}}var he=class extends kt{constructor(){super("cursor","Cursor (CLI)",100)}canHandle(e){let t=[j(ae(),".local","bin","cursor-agent"),j(ae(),".cursor","bin","cursor-agent"),"/usr/local/bin/cursor-agent","/usr/local/bin/agent","/Applications/Cursor.app/Contents/Resources/app/bin/cursor","agent","cursor-agent"];for(let r of t)try{if(r.startsWith("/")){Qe(r,et.X_OK);let o=Z(`"${r}" --version 2>&1`,{encoding:"utf-8",timeout:3e3,stdio:"pipe"});if(o&&o.length>0)return a.debug(`[Cursor] Found agent at: ${r} (version: ${o.trim().slice(0,50)})`),!0}else{let o=Z(`which ${r}`,{encoding:"utf-8",timeout:2e3,stdio:"pipe"}).trim();if(!o)continue;let s=Z(`${r} --version 2>&1`,{encoding:"utf-8",timeout:3e3,stdio:"pipe"});if(s&&s.length>0)return a.debug(`[Cursor] Found '${r}' in PATH at ${o} (version: ${s.trim().slice(0,50)})`),!0}}catch{continue}return a.warn("[Cursor] \u274C Cursor Agent CLI not found or not working. Run: agent --version"),!1}async invoke(e,t={}){let{workspace:r=process.cwd(),print:o=!1,schema:s=null,skills:i=null,sessionPath:l=null,nodeName:c=null,timeout:p=me.CURSOR_AGENT_DEFAULT,config:f={},signal:_=null}=t,m=f?.agent?.strictMode||!1,h=t.model??f?.agent?.cursor?.model??K.CURSOR,O=G("CURSOR_API_KEY");try{a.debug(`[Cursor] Invoking (model: ${h}, timeout: ${p/1e3}s, skills: ${JSON.stringify(i)})`);let y=(this._setupMcpConfig(l,r,f,i,c)||{}).isolatedMcpHome??null,x=[j(ae(),".local","bin","cursor-agent"),j(ae(),".cursor","bin","cursor-agent"),"/usr/local/bin/cursor-agent","/usr/local/bin/agent","/Applications/Cursor.app/Contents/Resources/app/bin/cursor","agent","cursor-agent"],$=null;for(let w of x)try{if(w.startsWith("/"))Qe(w,et.X_OK),Z(`"${w}" --version 2>&1`,{encoding:"utf-8",timeout:3e3,stdio:"pipe"});else{if(!Z(`which ${w}`,{encoding:"utf-8",timeout:2e3}).trim())throw new Error("not in PATH");Z(`${w} --version 2>&1`,{encoding:"utf-8",timeout:3e3,stdio:"pipe"})}$=w,a.debug(`[Agent] Using binary: ${w}`);break}catch(u){a.debug(`[Agent] Binary '${w}' check failed: ${u.message}`);continue}if(!$)throw new Error(`Cursor Agent CLI not found or not working.
|
|
43
43
|
|
|
44
44
|
Checked paths:
|
|
45
45
|
${x.map(w=>` - ${w}`).join(`
|
|
@@ -51,52 +51,52 @@ Install cursor-agent:
|
|
|
51
51
|
Then add to PATH:
|
|
52
52
|
echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.zshrc && source ~/.zshrc
|
|
53
53
|
|
|
54
|
-
Test with: agent --version`);let
|
|
54
|
+
Test with: agent --version`);let v=new V(e),g=null;if(s){let w=`zibby-result-${Date.now()}.json`;g=j(r,".zibby","tmp",w);let u=j(r,".zibby","tmp");le(u)||Ve(u,{recursive:!0}),v.addStructuredOutput(s,g)}let R=process.env.CURSOR_API_KEY,C=R?` | key: ***${R.slice(-4)}`:" | key: not set";console.log(`
|
|
55
55
|
\u25C6 Model: ${h||"auto"}${C}
|
|
56
|
-
`);let
|
|
57
|
-
${
|
|
58
|
-
`)}catch{}let
|
|
56
|
+
`);let k=(await import("chalk")).default,E=v.getUserPrompt();console.log(`
|
|
57
|
+
${k.bold("Prompt sent to LLM:")}`),console.log(k.dim("\u2500".repeat(60))),console.log(k.dim(E)),console.log(k.dim("\u2500".repeat(60)));let S=v.build(),M=v.getStats(),b=["--print","--force","--approve-mcps","--output-format","stream-json","--stream-partial-output","--model",h||"auto"];if(process.env.CURSOR_API_KEY&&b.push("--api-key",process.env.CURSOR_API_KEY),b.push(S),a.debug(`[Agent] User prompt: ${M.userPromptLength} chars, System instructions: ${M.systemInstructionsLength} chars (${M.instructionCount} blocks), Full: ${M.fullPromptLength} chars, model: ${h||"auto"}`),a.debug(`[Agent] Workspace: ${r}`),process.env.LOG_LEVEL==="debug"||process.env.ZIBBY_LOG_CURSOR_CLI==="1")try{console.log(`\u{1F527} Cursor CLI --model ${h||"auto"} (from .zibby.config.js agent.cursor.model)
|
|
58
|
+
`)}catch{}let P,U=null;try{let w=l||(process.env.ZIBBY_SESSION_PATH?String(process.env.ZIBBY_SESSION_PATH).trim():null);P=await this._spawnWithStreaming($,b,r,p,null,w,y,_)}catch(w){U=w}let I=P?.stdout||"";if(s){let w=typeof s.parse=="function",u=null,T=!!(g&&le(g));if(g&&a.info(`[Agent] Result file: ${T?"present":"missing"} at ${g}`),T)try{let d=qe(g,"utf-8").trim();u=JSON.parse(d),a.info(`[Agent] Parsed JSON from result file OK (${d.length} chars) \u2192 object ready for validation`),U&&a.debug("[Agent] Agent exited non-zero but result file was written \u2014 recovering")}catch(d){a.warn(`\u26A0\uFE0F [Agent] Result file exists on disk but is not valid JSON: ${d.message}`)}else if(U)a.warn(`[Agent] Result file missing at ${g} (agent process error \u2014 may still recover if strictMode repairs)`);else throw a.error(`\u274C [Agent] Result file was never created at ${g}`),new Error(`Agent did not write required result file at ${g}`);if(u&&w)try{let d=s.parse(u);return a.info("\u2705 [Agent] Zod validation passed for structured result file"),m&&a.debug("[Agent] strictMode enabled but not needed \u2014 agent wrote valid file"),{raw:I,structured:d}}catch(d){let D=d.issues?d.issues.map(N=>` - ${N.path.join(".")}: ${N.message}`).join(`
|
|
59
59
|
`):d.message?.slice(0,400);if(a.error(`\u274C [Agent] Zod validation FAILED - invalid output schema:
|
|
60
60
|
${D}`),a.error(`\u{1F4C4} [Agent] Invalid JSON written to file:
|
|
61
61
|
${JSON.stringify(u,null,2).slice(0,500)}`),!m)throw new Error(`Agent output failed schema validation:
|
|
62
62
|
${D}
|
|
63
63
|
|
|
64
64
|
The agent wrote invalid data that doesn't match the required outputSchema.
|
|
65
|
-
Enable strictMode in .zibby.config.js for automatic repair.`,{cause:d})}else{if(u)return a.info("\u2705 [Agent] File-based output extracted (no Zod parse fn) \u2014 accepting as structured"),m&&a.debug("[Agent] strictMode enabled but not needed \u2014 agent wrote valid file"),{raw:I,structured:u};T&&a.error("\u274C [Agent] Result file exists but produced no in-memory JSON (parse failed earlier)")}if(m&&!U){let d=
|
|
66
|
-
`),r=null;for(let o of t){let s=o.trim();if(s)try{let i=JSON.parse(s);if(i.type==="assistant"&&i.message?.content){let l=i.message.content;if(Array.isArray(l)){let c=l.filter(p=>p.type==="text"&&p.text).map(p=>p.text).join("");c&&(r=c)}else typeof l=="string"&&l&&(r=l)}}catch{}}return r?.trim()||null}_setupMcpConfig(e,t,r,o=null,s=null){let i=r?.headless,l=j(ae(),".cursor"),c=j(l,"mcp.json"),p={};if(le(c))try{p=JSON.parse(qe(c,"utf-8"))}catch{}let f=p.mcpServers||{},_=r?.paths?.output||
|
|
67
|
-
`;if(ze(e)){let y=He(t||process.cwd()),x=j(y,".cursor","mcp.json");return We(x,L,"utf8"),a.debug(`[MCP] Isolated cursor-agent HOME (session-scoped mcp.json): ${y} | servers: ${Object.keys(f).join(", ")}`),{isolatedMcpHome:y}}return le(l)||Ve(l,{recursive:!0}),We(c,L,"utf8"),a.debug(`[MCP] Global ~/.cursor/mcp.json | servers: ${Object.keys(f).join(", ")}`),{isolatedMcpHome:null}}_ensureSkillConfigured(e,t,r,o,s=null,i){let l=t.cursorKey||t.serverName,c=e[l]?l:e[t.serverName]?t.serverName:null;if(c&&r){let f=typeof t.resolve=="function"?t.resolve({sessionPath:r,nodeName:s,headless:i}):null;f?.args?e[c].args=f.args:e[c].args=(e[c].args||[]).map(h=>h.startsWith("--output-dir=")?`--output-dir=${r}`:h);let _=f?.env||{},m=t.sessionEnvKey?{[t.sessionEnvKey]:o}:{};e[c].env={...e[c].env||{},..._,...m},a.debug(`[MCP] Updated ${c} session \u2192 ${r}`);return}if(c)return;let p=t.resolve({sessionPath:r,nodeName:s,headless:i});p&&(e[l]={...p,...t.sessionEnvKey&&{env:{...p.env||{},[t.sessionEnvKey]:o}}},a.debug(`[MCP] Configured ${l}`))}_spawnWithStreaming(e,t,r,o,s=null,i=null,l=null,c=null){return new Promise((p,f)=>{let _=Date.now(),m="",h="",O=Date.now(),L=0,y=!1,x=null,v=!1
|
|
68
|
-
`)){let X=J.replace(`${r}/`,"");
|
|
69
|
-
${m.slice(-2e3)}`),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},5e3)},o),w=new q;w.onToolCall=(u,T)=>{let A=u,d=T;if(u==="mcpToolCall"&&T?.name)A=T.name.replace(/^mcp_+[^_]+_+/,""),A.includes("-")&&A.split("-")[0]===A.split("-")[1]&&(A=A.split("-")[0]),d=T.args??T.input??T;else{if(u==="readToolCall"||u==="editToolCall"||u==="writeToolCall")return;(u.startsWith("mcp__")||u.includes("ToolCall"))&&(A=u.replace(/^mcp_+[^_]+_+/,"").replace(/ToolCall$/,""))}if(A.includes("memory")?Y.stepMemory(`Tool: ${A}`):Y.stepTool(`Tool: ${A}`),d!=null&&typeof d=="object"&&Object.keys(d).length>0
|
|
70
|
-
`).filter(D=>D.trim());L+=d.length}),E.stderr.on("data",u=>{let T=u.toString();h+=T,O=Date.now()
|
|
71
|
-
`).filter(d=>d.trim());for(let d of A)a.warn(`\u26A0\uFE0F [Agent stderr] ${d}`)}),E.on("close",(u,T)=>{if(
|
|
65
|
+
Enable strictMode in .zibby.config.js for automatic repair.`,{cause:d})}else{if(u)return a.info("\u2705 [Agent] File-based output extracted (no Zod parse fn) \u2014 accepting as structured"),m&&a.debug("[Agent] strictMode enabled but not needed \u2014 agent wrote valid file"),{raw:I,structured:u};T&&a.error("\u274C [Agent] Result file exists but produced no in-memory JSON (parse failed earlier)")}if(m&&!U){let d=P.parsedText,D=u?JSON.stringify(u):d;a.info(`[Agent] strictMode: calling OpenAI proxy to fix structured output (${D.length} chars in)`);try{let N=await Fe(D,s);if(w){let J=s.parse(N.structured);return a.info("\u2705 [Agent] Proxy output passed Zod validation"),{raw:I,structured:J}}return{raw:I,...N}}catch(N){if(a.warn(`\u26A0\uFE0F [Agent] strictMode proxy failed: ${N.message}`),u)return a.warn("[Agent] Using agent's original result file as fallback"),{raw:I,structured:u}}}if(U)throw U;let A=T?u==null?"file existed but JSON.parse failed \u2014 see WARN log above":w?"JSON was valid but Zod validation failed \u2014 see WARN log above":"no structured object after read (unexpected)":"file never appeared (agent may not have run Write tool to the path above)";throw a.error(`\u274C [Agent] No validated structured output: ${A}`),a.error("\u{1F4A1} Tip: Set strictMode=true in .zibby.config.js for OpenAI proxy fallback"),new Error(`Agent did not produce a valid result file at ${g}. Enable strictMode for proxy fallback.`)}if(U)throw U;return this._extractFinalResult(I)||P?.parsedText||I}finally{O.dispose()}}_extractFinalResult(e){if(!e)return null;let t=e.split(`
|
|
66
|
+
`),r=null;for(let o of t){let s=o.trim();if(s)try{let i=JSON.parse(s);if(i.type==="assistant"&&i.message?.content){let l=i.message.content;if(Array.isArray(l)){let c=l.filter(p=>p.type==="text"&&p.text).map(p=>p.text).join("");c&&(r=c)}else typeof l=="string"&&l&&(r=l)}}catch{}}return r?.trim()||null}_setupMcpConfig(e,t,r,o=null,s=null){let i=r?.headless,l=j(ae(),".cursor"),c=j(l,"mcp.json"),p={};if(le(c))try{p=JSON.parse(qe(c,"utf-8"))}catch{}let f=p.mcpServers||{},_=r?.paths?.output||Pt,m=j(t||process.cwd(),_,Nt),h=Array.isArray(o)?o.map(y=>Ze(y)).filter(Boolean):[...Mt()].map(([,y])=>y),O=new Set;for(let y of h)typeof y.resolve=="function"&&(O.has(y.serverName)||(O.add(y.serverName),this._ensureSkillConfigured(f,y,e,m,s,i)));if(e){let y=Ze("browser");y&&typeof y.resolve=="function"&&!O.has(y.serverName)&&this._ensureSkillConfigured(f,y,e,m,"execute_live",i)}if(Object.keys(f).length===0)return a.debug("[MCP] No MCP servers configured - agent will run without tool access"),{isolatedMcpHome:null};let L=`${JSON.stringify({mcpServers:f},null,2)}
|
|
67
|
+
`;if(ze(e)){let y=He(t||process.cwd()),x=j(y,".cursor","mcp.json");return We(x,L,"utf8"),a.debug(`[MCP] Isolated cursor-agent HOME (session-scoped mcp.json): ${y} | servers: ${Object.keys(f).join(", ")}`),{isolatedMcpHome:y}}return le(l)||Ve(l,{recursive:!0}),We(c,L,"utf8"),a.debug(`[MCP] Global ~/.cursor/mcp.json | servers: ${Object.keys(f).join(", ")}`),{isolatedMcpHome:null}}_ensureSkillConfigured(e,t,r,o,s=null,i){let l=t.cursorKey||t.serverName,c=e[l]?l:e[t.serverName]?t.serverName:null;if(c&&r){let f=typeof t.resolve=="function"?t.resolve({sessionPath:r,nodeName:s,headless:i}):null;f?.args?e[c].args=f.args:e[c].args=(e[c].args||[]).map(h=>h.startsWith("--output-dir=")?`--output-dir=${r}`:h);let _=f?.env||{},m=t.sessionEnvKey?{[t.sessionEnvKey]:o}:{};e[c].env={...e[c].env||{},..._,...m},a.debug(`[MCP] Updated ${c} session \u2192 ${r}`);return}if(c)return;let p=t.resolve({sessionPath:r,nodeName:s,headless:i});p&&(e[l]={...p,...t.sessionEnvKey&&{env:{...p.env||{},[t.sessionEnvKey]:o}}},a.debug(`[MCP] Configured ${l}`))}_spawnWithStreaming(e,t,r,o,s=null,i=null,l=null,c=null){return new Promise((p,f)=>{let _=Date.now(),m="",h="",O=Date.now(),L=0,y=!1,x=null,$=!1,v=!1,g=null;if(i)try{g=j(Ut(String(i)),Rt)}catch{g=null}let R=!1,C=()=>{R||(R=!0,Xe(l))},k={...process.env};l&&(k.HOME=l,process.platform==="win32"&&(k.USERPROFILE=l),a.debug(`[Agent] cursor-agent HOME=${l} (isolated MCP config)`));let E=Lt(e,t,{cwd:r,shell:!1,stdio:["pipe","pipe","pipe"],env:k,...c&&{signal:c}});a.debug(`[Agent] PID: ${E.pid}`);let S=null,M=()=>{y||v||(y=!0,x="aborted",a.warn("\u{1F6D1} Abort signal \u2014 terminating cursor-agent (SIGKILL in 5s if SIGTERM trapped)"),S=setTimeout(()=>{if(!v){a.warn("\u{1F6D1} SIGTERM not honored \u2014 sending SIGKILL");try{E.kill("SIGKILL")}catch{}}},5e3))};c&&(c.aborted?M():c.addEventListener("abort",M,{once:!0})),E.stdin.on("error",u=>{u.code!=="EPIPE"&&a.warn(`[Agent] stdin error: ${u.message}`)}),E.stdout.on("error",u=>{u.code!=="EPIPE"&&a.warn(`[Agent] stdout error: ${u.message}`)}),E.stderr.on("error",u=>{u.code!=="EPIPE"&&a.warn(`[Agent] stderr error: ${u.message}`)}),s?(E.stdin.write(s,u=>{u&&u.code!=="EPIPE"&&a.warn(`[Agent] Failed to write to stdin: ${u.message}`),E.stdin.end()}),a.debug(`[Agent] Prompt also piped to stdin (${s.length} chars)`)):E.stdin.end();let b=null;g&&(b=setInterval(()=>{if(!(y||v))try{if(le(g)){y=!0,x="aborted";try{Dt(g)}catch{}a.warn("\u{1F6D1} External stop requested \u2014 terminating Cursor agent (and MCP browser session)"),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},2e3)}}catch{}},600));let P=new Set,U=new Date(_).toISOString().replace(/\.\d+Z$/,""),I=setInterval(()=>{let u=Math.round((Date.now()-_)/1e3),T=Math.round((Date.now()-O)/1e3),A=[];try{let D=Math.ceil(u/60)+1,N=Z(`find "${r}" -type f -mmin -${D} -not -path '*/node_modules/*' -not -path '*/.git/*' -not -path '*/target/*' 2>/dev/null | head -20`,{encoding:"utf-8",timeout:5e3}).trim();if(N)for(let J of N.split(`
|
|
68
|
+
`)){let X=J.replace(`${r}/`,"");P.has(X)||(P.add(X),A.push(X))}}catch{}let d="";A.length>0&&(d=` | \u{1F4C1} new: ${A.map(N=>N.split("/").pop()).join(", ")}`),P.size>0&&(d+=` | \u{1F4E6} total: ${P.size} files`),a.debug(`\u{1F493} [Agent] Running for ${u}s | ${L} lines output${d}`),L===0&&u>=30&&P.size===0&&(u<35&&a.warn(`\u26A0\uFE0F [Agent] No output after ${u}s \u2014 agent may be stuck. Check your CURSOR_API_KEY.`),u>=60&&(y=!0,x=x||"stall",a.error(`\u274C [Agent] No response after ${u}s \u2014 killing. Verify CURSOR_API_KEY is valid and agent CLI works: agent --version`),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},3e3)))},3e4),B=setTimeout(()=>{y=!0,x=x||"timeout";let u=Math.round((Date.now()-_)/1e3);a.error(`\u23F1\uFE0F [Agent] Timeout after ${u}s \u2014 killing process (PID: ${E.pid})`),m.trim()&&a.warn(`\u{1F4E4} [Agent] Partial output (${m.length} chars) before timeout:
|
|
69
|
+
${m.slice(-2e3)}`),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},5e3)},o),w=new q;w.onToolCall=(u,T)=>{let A=u,d=T;if(u==="mcpToolCall"&&T?.name)A=T.name.replace(/^mcp_+[^_]+_+/,""),A.includes("-")&&A.split("-")[0]===A.split("-")[1]&&(A=A.split("-")[0]),d=T.args??T.input??T;else{if(u==="readToolCall"||u==="editToolCall"||u==="writeToolCall")return;(u.startsWith("mcp__")||u.includes("ToolCall"))&&(A=u.replace(/^mcp_+[^_]+_+/,"").replace(/ToolCall$/,""))}if(A.includes("memory")?Y.stepMemory(`Tool: ${A}`):Y.stepTool(`Tool: ${A}`),d!=null&&typeof d=="object"&&Object.keys(d).length>0&&!v){let N=JSON.stringify(d),J=N.length>100?`${N.substring(0,100)}...`:N;console.log(` Input: ${J}`)}},E.stdout.on("data",u=>{let T=u.toString();m+=T,O=Date.now(),$||($=!0);let A=w.processChunk(T);A&&!v&&process.stdout.write(A);let d=T.split(`
|
|
70
|
+
`).filter(D=>D.trim());L+=d.length}),E.stderr.on("data",u=>{let T=u.toString();h+=T,O=Date.now(),$||($=!0);let A=T.split(`
|
|
71
|
+
`).filter(d=>d.trim());for(let d of A)a.warn(`\u26A0\uFE0F [Agent stderr] ${d}`)}),E.on("close",(u,T)=>{if(v=!0,C(),clearTimeout(B),clearInterval(I),b&&clearInterval(b),S&&clearTimeout(S),c&&!c.aborted)try{c.removeEventListener("abort",M)}catch{}w.flush();let A=Math.round((Date.now()-_)/1e3);if(a.debug(`[Agent] Exited: code=${u}, signal=${T}, elapsed=${A}s, output=${m.length} chars`),y){if(x==="aborted"){let N=new Error("Aborted via signal");N.name="AbortError",f(N);return}f(new Error(`Cursor Agent timed out after ${A}s (limit: ${o/1e3}s). ${L} lines produced. Last output ${Math.round((Date.now()-O)/1e3)}s ago. ${m.trim()?`
|
|
72
72
|
Partial output (last 500 chars):
|
|
73
73
|
${m.slice(-500)}`:"No output captured."}`));return}if(u!==0){f(new Error(`Cursor Agent failed: exit code ${u}, signal ${c}. ${h.trim()?`
|
|
74
74
|
Stderr: ${h.slice(-1e3)}`:""}${m.trim()?`
|
|
75
|
-
Stdout (last 500 chars): ${m.slice(-500)}`:""}`));return}let d=w.getResult(),D=d?JSON.stringify(d,null,2):w.getRawText()||m||"";p({stdout:m||h||"",parsedText:D})}),E.on("error",u=>{if(C(),clearTimeout(
|
|
75
|
+
Stdout (last 500 chars): ${m.slice(-500)}`:""}`));return}let d=w.getResult(),D=d?JSON.stringify(d,null,2):w.getRawText()||m||"";p({stdout:m||h||"",parsedText:D})}),E.on("error",u=>{if(C(),clearTimeout(B),clearInterval(I),b&&clearInterval(b),S&&clearTimeout(S),c&&!c.aborted)try{c.removeEventListener("abort",M)}catch{}f(new Error(`Cursor Agent spawn error: ${u.message}
|
|
76
76
|
Binary: ${e}
|
|
77
77
|
This usually means the binary is not in PATH. Try:
|
|
78
|
-
echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.zshrc && source ~/.zshrc`))})})}};import{AgentStrategy as
|
|
78
|
+
echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.zshrc && source ~/.zshrc`))})})}};import{AgentStrategy as Bt,getSkill as Kt}from"@zibby/agent-workflow";import{query as Jt}from"@anthropic-ai/claude-agent-sdk";import{zodToJsonSchema as Gt}from"zod-to-json-schema";import{writeFileSync as Re,mkdirSync as Me,existsSync as tt,readFileSync as ot}from"fs";import{homedir as rt}from"os";import{join as ee}from"path";var ye=class extends Bt{constructor(){super("claude","Claude (Anthropic API)",50)}canHandle(e){let t=!!process.env.ANTHROPIC_API_KEY||!!process.env.ANTHROPIC_API_KEY_POOL,r=!!process.env.CLAUDE_CODE_OAUTH_TOKEN||!!process.env.CLAUDE_CODE_OAUTH_TOKEN_POOL,o=!!process.env.ANTHROPIC_AUTH_TOKEN||!!process.env.ANTHROPIC_AUTH_TOKEN_POOL,s=t||r||o;return s||a.debug("ClaudeAgentStrategy: no credentials. Set one of ANTHROPIC_API_KEY / CLAUDE_CODE_OAUTH_TOKEN / ANTHROPIC_AUTH_TOKEN (or a *_POOL variant for multi-token rotation)."),s}async invoke(e,t={}){let{model:r,workspace:o=process.cwd(),schema:s=null,images:i=[],skills:l=null,sessionPath:c=null,nodeName:p=null,timeout:f,config:_={},signal:m=null}=t,h=r;(!h||h==="auto")&&(a.debug(`Model is '${h||"undefined"}', using default: ${K.CLAUDE}`),h=K.CLAUDE);let O=ke[h]||h;ke[h]&&h!==O&&a.debug(`Mapped model: ${h} \u2192 ${O}`);let L=[G("ANTHROPIC_API_KEY"),G("CLAUDE_CODE_OAUTH_TOKEN"),G("ANTHROPIC_AUTH_TOKEN")];try{a.debug(`Invoking Claude Agent SDK with model: ${O}, skills: ${JSON.stringify(l)}`);let y=process.env.CLAUDE_CODE_OAUTH_TOKEN,x=process.env.ANTHROPIC_API_KEY,$=process.env.ANTHROPIC_AUTH_TOKEN;process.env.CLAUDE_CODE_CREDENTIALS_JSON?Yt(process.env.CLAUDE_CODE_CREDENTIALS_JSON):y&&Ft(y,process.env.__ZIBBY_CLAUDE_PLAN);let v=process.env.__ZIBBY_CLAUDE_PLAN,g=v?v.charAt(0).toUpperCase()+v.slice(1):null,R;y?R=g?`Claude ${g} plan \xB7 ***${y.slice(-4)}`:`Subscription \xB7 ***${y.slice(-4)}`:x?R=`API key \xB7 ***${x.slice(-4)}`:$?R=`Bearer \xB7 ***${$.slice(-4)}`:R="NONE \u2014 request will fail";let C=(await import("chalk")).default;console.log(""),console.log(` ${C.cyan("\u25C6")} ${C.bold(O)}`),console.log(` ${C.dim("Auth: ")}${R}`),console.log(` ${C.dim("Workdir: ")}${o}`),console.log(""),console.log(C.bold("Prompt sent to LLM:")),console.log(C.dim("\u2500".repeat(60))),console.log(C.dim(e)),console.log(C.dim("\u2500".repeat(60)));let{allowedTools:k,mcpServers:E}=this._resolveSkills(l,{sessionPath:c,workspace:o,nodeName:p});try{let S={cwd:o,allowedTools:k,permissionMode:"bypassPermissions",allowDangerouslySkipPermissions:!0,settings:{permissions:{allow:["Bash(*)","Read(*)","Write(*)","Edit(*)","MultiEdit(*)","NotebookEdit(*)","Grep(*)","Glob(*)","WebFetch(*)","WebSearch(*)","Task(*)","TodoWrite(*)","Skill(*)"],defaultMode:"bypassPermissions"}},model:O,...Object.keys(E).length>0&&{mcpServers:E}};if(s){let T=typeof s.parse=="function"?Gt(s,{target:"openApi3"}):s;S.outputFormat={type:"json_schema",schema:T},a.debug("Structured output enforced via SDK outputFormat")}a.debug(`Agent SDK options: ${JSON.stringify({cwd:S.cwd,toolCount:k.length,permissionMode:S.permissionMode,model:S.model,hasOutputFormat:!!S.outputFormat})}`);let M="",b=0,P=[];a.debug("Starting Claude Agent SDK query stream");let U;try{U=Jt({prompt:e,options:S})}catch(u){throw a.error(`Failed to initialize Claude Agent SDK: ${u.message}`),u}let I=null,B=0,w=3;if(m?.aborted){let u=new Error("Aborted via signal");throw u.name="AbortError",u}try{for await(let u of U){if(m?.aborted){let A=new Error("Aborted via signal");throw A.name="AbortError",A}if(P.push(u),u.type==="error"||u.error){let A=u.error?.message||u.error||u.message||"Unknown API error";try{a.error("SDK error event raw payload",{fullMessage:JSON.stringify(u,null,2)})}catch{}throw new Error(typeof A=="string"?A:JSON.stringify(A))}let T=JSON.stringify(u.message?.content||u.text||"").slice(0,200);if(T===I){if(B++,B>=w){let A=(u.message?.content?.[0]?.text||u.text||"unknown").slice(0,100);throw new Error(`API stuck in loop (${B}x repeated): ${A}`)}}else I=T,B=1;if(u.type==="assistant"||u.constructor?.name==="AssistantMessage"){let A=u.message?.content||u.content||[];for(let d of A)if(d.type==="thinking"&&d.thinking)console.log(`${d.thinking.substring(0,200)}${d.thinking.length>200?"...":""}`);else if(d.type==="text"&&d.text)M+=d.text,d.text.length<500?console.log(`${d.text}`):console.log(`${d.text.substring(0,200)}... (${d.text.length} chars)`);else if(d.type==="tool_use"){b++,d.name.includes("memory")?Y.stepMemory(`Tool: ${d.name}`):Y.stepTool(`Tool: ${d.name}`);let N=JSON.stringify(d.input).substring(0,100);console.log(` Input: ${N}${JSON.stringify(d.input).length>100?"...":""}`)}}else if(!(u.type==="user"&&u.tool_use_result)){if(u.type==="result"||u.constructor?.name==="ResultMessage"){let A=u.result||u.text||u.content||M;try{jt({message:u,sessionPath:c,nodeName:p,model:h,toolCallCount:b})}catch(d){a.debug(`usage.json write failed: ${d.message}`)}if(s){if(u.structured_output){a.debug("Using SDK native structured_output");let D=typeof s.parse=="function"?s.parse(u.structured_output):u.structured_output;return{raw:A,structured:D}}if(A){let d=this._extractJson(A,s);if(d)return{raw:A,structured:d}}a.warn(`Could not extract structured output \u2014 returning raw text (${(A||"").length} chars)`)}return A||""}}}if(a.warn(`Agent SDK ended without result. Collected ${P.length} messages`),M.length>0)return a.debug("Returning accumulated text from messages"),M;throw new Error("Claude Agent SDK query ended without result")}catch(u){let T={message:u?.message,name:u?.name,code:u?.code,cause:u?.cause?.message||u?.cause,stack:u?.stack?.split(`
|
|
79
79
|
`).slice(0,5).join(`
|
|
80
|
-
`),stringified:(()=>{try{return JSON.stringify(u,Object.getOwnPropertyNames(Object(u)))}catch{return String(u)}})()};throw a.error(`Error during query stream: ${u?.message||u}`,T),u}}catch(S){throw a.error("Claude Agent SDK call failed",{message:S?.message,name:S?.name,code:S?.code,cause:S?.cause?.message||S?.cause}),S}}finally{for(let y of L)y.dispose()}}_resolveSkills(e,t){let r=["Read","Write","Edit","Bash","Grep","Glob"];if(e===null)return a.debug("No skills \u2014 pure LLM mode"),{allowedTools:[],mcpServers:{}};if(!Array.isArray(e)||e.length===0)return a.debug("Default IDE skills for code generation"),{allowedTools:[...r],mcpServers:{}};let o=[...r],s={};for(let i of e){let l=
|
|
80
|
+
`),stringified:(()=>{try{return JSON.stringify(u,Object.getOwnPropertyNames(Object(u)))}catch{return String(u)}})()};throw a.error(`Error during query stream: ${u?.message||u}`,T),u}}catch(S){throw a.error("Claude Agent SDK call failed",{message:S?.message,name:S?.name,code:S?.code,cause:S?.cause?.message||S?.cause}),S}}finally{for(let y of L)y.dispose()}}_resolveSkills(e,t){let r=["Read","Write","Edit","Bash","Grep","Glob"];if(e===null)return a.debug("No skills \u2014 pure LLM mode"),{allowedTools:[],mcpServers:{}};if(!Array.isArray(e)||e.length===0)return a.debug("Default IDE skills for code generation"),{allowedTools:[...r],mcpServers:{}};let o=[...r],s={};for(let i of e){let l=Kt(i);if(!l){a.warn(`Unknown skill "${i}" \u2014 skipping`);continue}if(l.allowedTools&&o.push(...l.allowedTools),typeof l.resolve=="function"){let c=l.resolve(t);c&&(s[l.serverName]=c,a.debug(`MCP: ${l.serverName} \u2192 ${c.command} ${c.args[0]}`))}}return{allowedTools:o,mcpServers:s}}_extractJson(e,t){let r=[()=>{if(e.includes("===JSON_START===")){let o=e.indexOf("===JSON_START===")+16,s=e.indexOf("===JSON_END===");return e.substring(o,s).trim()}},()=>e.match(/```json\s*\n([\s\S]*?)\n```/)?.[1]?.trim(),()=>{if(!e.startsWith("{"))return e.match(/```\s*\n([\s\S]*?)\n```/)?.[1]?.trim()},()=>e.trim(),()=>{let o=e.indexOf("{"),s=e.lastIndexOf("}");if(o!==-1&&s>o)return e.substring(o,s+1)}];for(let o of r)try{let s=o();if(!s)continue;let i=JSON.parse(s);if(typeof i!="object"||i===null)continue;return typeof t.parse=="function"?t.parse(i):i}catch{}return null}};function jt({message:n,sessionPath:e,nodeName:t,model:r,toolCallCount:o}){if(!e||!t)return;let s=n?.usage||n?.message?.usage||{},i=ce(s.input_tokens,s.inputTokens),l=ce(s.output_tokens,s.outputTokens),c=ce(s.cache_read_input_tokens,s.cacheReadInputTokens),p=ce(s.cache_creation_input_tokens,s.cacheCreationInputTokens),f=ce(n?.duration_ms,n?.durationMs),_={model:r||null,input_tokens:i,output_tokens:l,cache_read_tokens:c,cache_creation_tokens:p,tool_call_count:typeof o=="number"?o:null,duration_ms:f,captured_at:new Date().toISOString()},m=ee(e,t);try{Me(m,{recursive:!0})}catch{}Re(ee(m,"usage.json"),JSON.stringify(_,null,2),"utf-8")}function ce(...n){for(let e of n){if(typeof e=="number"&&Number.isFinite(e))return e;if(typeof e=="string"&&e.trim()!==""&&!isNaN(Number(e)))return Number(e)}return null}function Ft(n,e){let t=ee(rt(),".claude"),r=ee(t,".credentials.json"),o={claudeAiOauth:{accessToken:n,...e?{subscriptionType:String(e).toLowerCase()}:{}}};try{if(tt(r)&&JSON.parse(ot(r,"utf-8"))?.claudeAiOauth?.accessToken===n)return;Me(t,{recursive:!0}),Re(r,JSON.stringify(o,null,2),"utf-8"),a.debug(`Wrote Claude Code OAuth credentials to ${r}`)}catch(s){a.warn(`Could not write ${r}: ${s.message} \u2014 Claude SDK may fail to authenticate.`)}}function Yt(n){let e=ee(rt(),".claude"),t=ee(e,".credentials.json"),r;try{r=JSON.parse(n)}catch(o){a.warn(`CLAUDE_CODE_CREDENTIALS_JSON not valid JSON: ${o.message}`);return}if(!r?.claudeAiOauth?.accessToken){a.warn("CLAUDE_CODE_CREDENTIALS_JSON missing claudeAiOauth.accessToken \u2014 skipping write.");return}try{if(tt(t)){let o=JSON.parse(ot(t,"utf-8"));if(o?.claudeAiOauth?.accessToken===r.claudeAiOauth.accessToken&&o?.claudeAiOauth?.expiresAt===r.claudeAiOauth.expiresAt)return}Me(e,{recursive:!0}),Re(t,JSON.stringify(r,null,2),"utf-8"),a.debug(`Wrote full Claude Code credentials blob to ${t}`)}catch(o){a.warn(`Could not write ${t}: ${o.message} \u2014 Claude SDK may fail to authenticate.`)}}import{AgentStrategy as zt,getSkill as Ht}from"@zibby/agent-workflow";import{execSync as Xt}from"node:child_process";import{zodToJsonSchema as Zt}from"zod-to-json-schema";var _e=class extends zt{constructor(){super("codex","Codex (OpenAI)",75)}canHandle(e){if(!!!(process.env.OPENAI_API_KEY||process.env.OPENAI_API_KEY_POOL||process.env.CODEX_API_KEY||process.env.CODEX_API_KEY_POOL))return a.debug("CodexAgentStrategy: OPENAI_API_KEY or CODEX_API_KEY not set"),!1;try{return Xt("codex --version",{encoding:"utf-8",timeout:5e3,stdio:"pipe"}),!0}catch{return a.warn("[Codex] codex CLI not found. Install: npm install -g @openai/codex"),!1}}async invoke(e,t={}){let{model:r,workspace:o=process.cwd(),schema:s=null,skills:i=null,sessionPath:l=null,nodeName:c=null,timeout:p,config:f={},signal:_=null}=t,m=[G("OPENAI_API_KEY"),G("CODEX_API_KEY")];try{let{Codex:h}=await import("@openai/codex-sdk"),O=r;(!O||O==="auto")&&(a.debug(`Model is '${O||"undefined"}', using default: ${K.CODEX}`),O=K.CODEX);let L=Pe[O]||O;Pe[O]&&O!==L&&a.debug(`Mapped model: ${O} \u2192 ${L}`),a.debug(`Invoking Codex SDK with model: ${L}, skills: ${JSON.stringify(i)}`);let y=process.env.CODEX_API_KEY||process.env.OPENAI_API_KEY;y&&!process.env.CODEX_API_KEY&&(process.env.CODEX_API_KEY=y);let x=y?` | key: ***${y.slice(-4)}`:" | key: not set";console.log(`
|
|
81
81
|
\u25C6 Model: ${L}${x}
|
|
82
|
-
`);let
|
|
83
|
-
${
|
|
84
|
-
${d}`,"schema_instruction","append");let D=`zibby-result-${Date.now()}.json`,N=W(o,".zibby","tmp");g=W(N,D),nt(N,{recursive:!0})
|
|
85
|
-
${d}`,"json_instruction","append")}let R
|
|
82
|
+
`);let $=(await import("chalk")).default;console.log(`
|
|
83
|
+
${$.bold("Prompt sent to LLM:")}`),console.log($.dim("\u2500".repeat(60))),console.log($.dim(e)),console.log($.dim("\u2500".repeat(60)));let v=this._resolveSkillsToMcp(i,{sessionPath:l,workspace:o,nodeName:c}),g={};Object.keys(v).length>0&&(g.mcp_servers=v,a.debug(`[Codex] MCP servers: ${Object.keys(v).join(", ")}`));let C=new h({...Object.keys(g).length>0&&{config:g}}).startThread({workingDirectory:o,skipGitRepoCheck:!0,approvalPolicy:"never",sandboxMode:"danger-full-access",networkAccessEnabled:!0}),k=s&&typeof s.parse=="function",E={};if(s)try{let S=k?Zt(s,{target:"openAi"}):s;E.outputSchema=S,a.debug("Structured output via SDK outputSchema")}catch(S){a.warn(`[Codex] Schema conversion failed, will extract from text: ${S.message}`)}if(_?.aborted){let S=new Error("Aborted via signal");throw S.name="AbortError",S}try{let{events:S}=await C.runStreamed(e,E),M=0,b="";for await(let P of S){if(_?.aborted){let I=new Error("Aborted via signal");throw I.name="AbortError",I}let U=P.type;if(U==="item.completed"){let I=P.item,B=I?.type;if(B==="mcp_tool_call"){M++;let w=`${I.server}/${I.tool}`;if(Y.stepTool(`Tool: ${w}`),I.arguments){let u=JSON.stringify(I.arguments),T=u.length>100?`${u.substring(0,100)}...`:u;console.log(` Input: ${T}`)}}else if(B==="tool_call"||B==="function_call"||B==="command_execution"){M++;let w=I.name||I.tool||I.command||"unknown";Y.stepTool(`Tool: ${w}`)}else B==="agent_message"&&(b=I.text||"",b.length<500?console.log(b):console.log(`${b.substring(0,200)}... (${b.length} chars)`))}else U==="turn.completed"?a.debug(`[Codex] Turn completed. Usage: ${JSON.stringify(P.usage||{})}`):a.debug(`[Codex] Event: ${U} ${JSON.stringify(P).slice(0,300)}`)}if(a.debug(`[Codex] Last agent message (${b.length} chars): ${b.slice(0,500)}`),s){if(!b)throw new Error("Codex agent returned no response");let P=JSON.parse(b),U=k?s.parse(P):P;return a.debug("\u2705 [Codex] Structured output validated"),{raw:b,structured:U}}return b||""}catch(S){let M=S.message||String(S);throw a.error(`\u274C [Codex] SDK call failed: ${M}`),M.includes("exited with code")&&(a.error("\u{1F4A1} [Codex] Verify: codex --version && echo $OPENAI_API_KEY"),a.error("\u{1F4A1} [Codex] If codex is missing: npm install -g @openai/codex")),S}}finally{for(let h of m)h.dispose()}}_resolveSkillsToMcp(e,t={}){if(!Array.isArray(e)||e.length===0)return{};let r={};for(let o of e){let s=Ht(o);if(!s){a.warn(`[Codex] Unknown skill "${o}" \u2014 skipping`);continue}if(typeof s.resolve!="function")continue;let i=s.resolve(t);if(!i)continue;let l=s.serverName||o,c={command:i.command};i.args?.length&&(c.args=i.args),i.env&&Object.keys(i.env).length>0&&(c.env=i.env),r[l]=c,a.debug(`[Codex] MCP: ${l} \u2192 ${i.command} ${(i.args||[]).join(" ")}`)}return r}};import{AgentStrategy as Wt,getSkill as qt}from"@zibby/agent-workflow";import{execSync as Vt,spawn as Qt}from"node:child_process";import{zodToJsonSchema as eo}from"zod-to-json-schema";import{existsSync as st,mkdirSync as nt,readFileSync as it,rmSync as to,writeFileSync as lt}from"node:fs";import{join as W}from"node:path";function oo(n){if(!n)return null;let e=String(n),t=e.match(/```(?:json)?\s*([\s\S]*?)```/i);if(t?.[1])try{return JSON.parse(t[1].trim())}catch{}let r=e.indexOf("{");if(r<0)return null;let o=0,s=!1,i=!1,l=-1;for(let c=r;c<e.length;c++){let p=e[c];if(s){i?i=!1:p==="\\"?i=!0:p==='"'&&(s=!1);continue}if(p==='"'){s=!0;continue}if(p==="{"){o===0&&(l=c),o+=1;continue}if(p==="}"){if(o===0)continue;if(o-=1,o===0&&l>=0){let f=e.slice(l,c+1);try{return JSON.parse(f)}catch{l=-1}}}}return null}function ro(n){let e=String(n||"").trim();if(!e)return null;try{return JSON.parse(e)}catch{return oo(e)}}function so(n){try{let e=JSON.parse(n);if(typeof e=="string")return e;if(typeof e?.response=="string")return e.response;if(typeof e?.text=="string")return e.text;if(typeof e?.output=="string")return e.output;if(Array.isArray(e?.candidates)&&e.candidates.length>0){let t=e.candidates[0];if(typeof t?.content=="string")return t.content;if(Array.isArray(t?.content?.parts)){let r=t.content.parts.map(o=>typeof o?.text=="string"?o.text:"").join("");if(r.trim())return r}}}catch{}return n}var Se=class extends Wt{constructor(){super("gemini","Gemini (Google)",70)}canHandle(e){if(!!!(process.env.GEMINI_API_KEY||process.env.GEMINI_API_KEY_POOL||process.env.GOOGLE_API_KEY||process.env.GOOGLE_API_KEY_POOL))return a.debug("GeminiAgentStrategy: GEMINI_API_KEY or GOOGLE_API_KEY not set"),!1;try{return Vt("gemini --version",{encoding:"utf-8",timeout:5e3,stdio:"pipe"}),!0}catch{return a.warn("[Gemini] gemini CLI not found. Install: npm install -g @google/gemini-cli"),!1}}async invoke(e,t={}){let{model:r,workspace:o=process.cwd(),schema:s=null,skills:i=null,sessionPath:l=null,nodeName:c=null,timeout:p=600*1e3,signal:f=null}=t,_=[G("GEMINI_API_KEY"),G("GOOGLE_API_KEY")];try{let m=r;(!m||m==="auto")&&(m=K.GEMINI);let h=je[m]||m,O=String(process.env.GEMINI_API_KEY||"").trim(),L=String(process.env.GOOGLE_API_KEY||"").trim(),y=this._resolveSkillsToMcp(i,{sessionPath:l,workspace:o,nodeName:c}),x=Object.keys(y).length>0,$=new V(e),v=s&&typeof s.parse=="function",g=null;if(s){let d;try{let D=v?eo(s,{target:"openAi"}):s;d=JSON.stringify(D,null,2)}catch{d="{}"}if(x){$.addSystemInstruction(`Write valid JSON that matches this schema:
|
|
84
|
+
${d}`,"schema_instruction","append");let D=`zibby-result-${Date.now()}.json`,N=W(o,".zibby","tmp");g=W(N,D),nt(N,{recursive:!0}),$.addStructuredOutput(s,g)}else $.addSystemInstruction(`Return ONLY valid JSON (no markdown, no commentary) that matches this schema:
|
|
85
|
+
${d}`,"json_instruction","append")}let R=$.build(),C=$.getUserPrompt(),k=$.getStats(),E=String(process.env.GEMINI_API_KEY||process.env.GOOGLE_API_KEY||"").trim(),S=E?` | key: ***${E.slice(-4)}`:" | key: not set";console.log(`
|
|
86
86
|
\u25C6 Model: ${h||"auto"}${S}
|
|
87
87
|
`);let M=(await import("chalk")).default;console.log(`
|
|
88
|
-
${M.bold("Prompt sent to LLM:")}`),console.log(M.dim("\u2500".repeat(60))),console.log(M.dim(C)),console.log(M.dim("\u2500".repeat(60)));let b=this._createGeminiConfigDir(o,y),
|
|
88
|
+
${M.bold("Prompt sent to LLM:")}`),console.log(M.dim("\u2500".repeat(60))),console.log(M.dim(C)),console.log(M.dim("\u2500".repeat(60)));let b=this._createGeminiConfigDir(o,y),P=["--output-format","json"];h&&h!=="auto"&&P.push("--model",h);let U=Object.keys(y);if(U.length>0){P.push("--approval-mode","yolo");for(let d of U)P.push("--allowed-mcp-server-names",d);a.info(`[Gemini] Enabling MCP servers: ${U.join(", ")}`)}else i&&i.length>0&&a.warn(`[Gemini] Skills requested but no MCP servers configured: ${i.join(", ")}`);P.push("-p",R);let I={...process.env,GEMINI_CLI_HOME:b};O?(I.GEMINI_API_KEY=O,delete I.GOOGLE_API_KEY):L&&(I.GOOGLE_API_KEY=L,delete I.GEMINI_API_KEY),a.debug(`[Gemini] Command: gemini ${P.slice(0,8).join(" ")}... (${P.length} total args)`),a.debug(`[Gemini] User prompt: ${k.userPromptLength} chars, System instructions: ${k.systemInstructionsLength} chars (${k.instructionCount} blocks), Full: ${k.fullPromptLength} chars`),a.debug(`[Gemini] Config home: ${b}`),a.debug(`[Gemini] GEMINI_CLI_HOME env: ${I.GEMINI_CLI_HOME}`);let B="",w=null;try{B=await new Promise((D,N)=>{let J=Qt("gemini",P,{cwd:o,env:I,stdio:["ignore","pipe","pipe"],...f&&{signal:f}}),X="",Ke="",ve=!1,ne=null,Ie=!1,Je=setTimeout(()=>{try{J.kill("SIGTERM")}catch{}},p),de=()=>{Ie||ve||(Ie=!0,ne=setTimeout(()=>{if(!ve)try{J.kill("SIGKILL")}catch{}},5e3))};f&&(f.aborted?de():f.addEventListener("abort",de,{once:!0})),J.stdout.on("data",H=>{X+=H.toString()}),J.stderr.on("data",H=>{Ke+=H.toString()}),J.on("error",H=>{if(clearTimeout(Je),ne&&clearTimeout(ne),f&&!f.aborted)try{f.removeEventListener("abort",de)}catch{}N(H)}),J.on("close",H=>{if(ve=!0,clearTimeout(Je),ne&&clearTimeout(ne),f&&!f.aborted)try{f.removeEventListener("abort",de)}catch{}if(Ie){let Ge=new Error("Aborted via signal");return Ge.name="AbortError",N(Ge)}if(H===0)return D(X.trim());N(new Error(`gemini failed with code ${H}: ${(Ke||X).trim()}`))})})}catch(d){w=d}finally{try{to(b,{recursive:!0,force:!0})}catch{}}let u=so(B).trim();if(!s){if(w)throw w;return u}if(g){let d=st(g);if(a.info(`[Gemini] Result file: ${d?"present":"missing"} at ${g}`),d)try{let D=it(g,"utf-8").trim(),N=JSON.parse(D),J=v?s.parse(N):N;return a.info("[Gemini] Structured output recovered from result file"),{raw:u,structured:J}}catch(D){a.warn(`[Gemini] Result file parse/validation failed: ${D.message}`)}else w||a.warn("[Gemini] Result file missing; falling back to stream-parsed JSON")}let T=null;if(s){let d=new q;d.zodSchema=s,d.processChunk(u),d.flush(),T=d.getResult()}if(a.info(`[Gemini] Raw stdout length: ${B.length} chars`),a.info(`[Gemini] Extracted text length: ${u.length} chars`),a.info(`[Gemini] StreamParser result: ${T?"extracted":"null"}`),T||(u.length<2e3?a.info(`[Gemini] Raw text preview:
|
|
89
89
|
${u}`):a.info(`[Gemini] Raw text preview (first 1000 chars):
|
|
90
|
-
${u.slice(0,1e3)}`),T=ro(u)),!T)throw w||(a.error("[Gemini] Failed to extract valid JSON from output"),a.error("\u{1F4A1} Tip: Set strictMode=true in .zibby.config.js for OpenAI proxy fallback"),new Error("Gemini did not return valid JSON for structured output. Enable strictMode for proxy fallback."));let A
|
|
90
|
+
${u.slice(0,1e3)}`),T=ro(u)),!T)throw w||(a.error("[Gemini] Failed to extract valid JSON from output"),a.error("\u{1F4A1} Tip: Set strictMode=true in .zibby.config.js for OpenAI proxy fallback"),new Error("Gemini did not return valid JSON for structured output. Enable strictMode for proxy fallback."));let A=v?s.parse(T):T;return{raw:u,structured:A}}finally{for(let m of _)m.dispose()}}_resolveSkillsToMcp(e,t={}){if(!Array.isArray(e)||e.length===0)return{};let r={};for(let o of e){let s=qt(o);if(!s||typeof s.resolve!="function")continue;let i=s.resolve(t);if(!i)continue;let l=s.cursorKey||s.serverName||o,c={command:i.command};i.args?.length&&(c.args=i.args),i.env&&Object.keys(i.env).length>0&&(c.env=i.env),i.cwd&&(c.cwd=i.cwd),r[l]=c}return r}_createGeminiConfigDir(e,t){let r=`${Date.now()}-${Math.random().toString(16).slice(2,10)}`,o=W(e||process.cwd(),".zibby","tmp",`gemini-home-${r}`),s=W(o,".gemini");nt(s,{recursive:!0});let i=W(s,"settings.json"),l={},c=W(process.env.HOME||"",".gemini","settings.json");if(st(c))try{l=JSON.parse(it(c,"utf-8"))}catch{l={}}let p={...l,mcpServers:{...l.mcpServers&&typeof l.mcpServers=="object"?l.mcpServers:{},...t||{}}};lt(i,`${JSON.stringify(p,null,2)}
|
|
91
91
|
`,"utf-8");let f=W(e||process.cwd(),".zibby","tmp","gemini-settings-debug.json");try{lt(f,`${JSON.stringify(p,null,2)}
|
|
92
92
|
`,"utf-8")}catch{}return a.debug(`[Gemini] Created isolated config with ${Object.keys(p.mcpServers||{}).length} MCP servers`),a.debug(`[Gemini] MCP servers: ${JSON.stringify(Object.keys(p.mcpServers||{}),null,2)}`),o}};import{AgentStrategy as _o,getSkill as Te}from"@zibby/agent-workflow";var ue=class{formatTools(e){throw new Error("formatTools() must be implemented")}hasToolCalls(e){throw new Error("hasToolCalls() must be implemented")}parseToolCalls(e){throw new Error("parseToolCalls() must be implemented")}getTextContent(e){throw new Error("getTextContent() must be implemented")}buildAssistantMessage(e){throw new Error("buildAssistantMessage() must be implemented")}buildToolResultMessage(e,t){throw new Error("buildToolResultMessage() must be implemented")}injectToolsIntoBody(e,t){throw new Error("injectToolsIntoBody() must be implemented")}};var te=class extends ue{formatTools(e){return e.map(t=>({type:"function",function:{name:t.name,description:t.description,parameters:t.parameters||t.input_schema||{type:"object",properties:{}}}}))}hasToolCalls(e){let t=e.choices?.[0]?.message;return!!(t?.tool_calls&&t.tool_calls.length>0)}parseToolCalls(e){return(e.choices?.[0]?.message?.tool_calls||[]).map(r=>({id:r.id,name:r.function.name,args:JSON.parse(r.function.arguments||"{}")}))}getTextContent(e){return e.choices?.[0]?.message?.content||""}buildAssistantMessage(e){return e.choices?.[0]?.message}buildToolResultMessage(e,t){return{role:"tool",tool_call_id:e,content:typeof t=="string"?t:JSON.stringify(t)}}injectToolsIntoBody(e,t){return t.length>0&&(e.tools=t),e}};var fe=class{async fetchCompletion(e,t,r={}){throw new Error("fetchCompletion() must be implemented")}async fetchStreamingCompletion(e,t,r={}){throw new Error("fetchStreamingCompletion() must be implemented")}};function Oe(n){return Buffer.byteLength(JSON.stringify(n),"utf8")}function Ee(n,e){let t=String(n||"");if(t.length<=e)return t;let r=Math.max(0,e-28);return`${t.slice(0,r)}
|
|
93
93
|
|
|
94
|
-
[truncated for size budget]`}function Le(n,e=0){if(!n||typeof n!="object"||e>8)return n;if(Array.isArray(n))return n.map(r=>Le(r,e+1));let t={};for(let[r,o]of Object.entries(n))r==="description"||r==="title"||r==="examples"||r==="default"||(t[r]=Le(o,e+1));return t}function no(n=[]){return n.map(e=>({...e,function:{...e.function,description:Ee(e.function?.description||"",180),parameters:Le(e.function?.parameters||{type:"object",properties:{}})}}))}function at(n){let e=new Set;for(let o of n)if(o.role==="assistant"&&Array.isArray(o.tool_calls))for(let s of o.tool_calls)e.add(s.id);let t=n.filter(o=>o.role==="tool"?e.has(o.tool_call_id):!0),r=new Set;for(let o of t)o.role==="tool"&&r.add(o.tool_call_id);return t.map(o=>{if(o.role!=="assistant"||!Array.isArray(o.tool_calls)||o.tool_calls.every(c=>r.has(c.id)))return o;let{tool_calls:i,...l}=o;return{...l,content:l.content||""}})}function De(n,e={}){let t=e.maxBytes||49e3,r=e.systemMaxChars||12e3,o={...n,messages:Array.isArray(n.messages)?[...n.messages]:[],tools:Array.isArray(n.tools)?no(n.tools):n.tools};o.messages.length>0&&o.messages[0]?.role==="system"&&(o.messages[0]={...o.messages[0],content:Ee(o.messages[0].content,r)});let s=!1;for(;Oe(o)>t&&o.messages.length>2;)o.messages.splice(1,1),s=!0;if(s&&(o.messages=at(o.messages)),Oe(o)>t&&o.messages.length>0&&(o.messages[0]={...o.messages[0],content:Ee(o.messages[0].content,6e3)},s=!0),Oe(o)>t){let i=o.messages.find(c=>c.role==="system")||o.messages[0],l=o.messages.slice(-2);o.messages=at([i,...l].filter(Boolean).map((c,p)=>({...c,content:Ee(c.content,p===0?4e3:8e3)}))),s=!0}return{body:o,meta:{bytes:Oe(o),trimmed:s,maxBytes:t,messageCount:o.messages.length}}}var ct=n=>Buffer.byteLength(JSON.stringify(n),"utf8"),oe=class extends fe{async fetchCompletion(e,t,r={}){let o=ct(e),{body:s,meta:i}=De(e,r.payloadCompaction);r.onBudget?.({streaming:!1,beforeBytes:o,meta:i});let l=this.#e(r),c=`${t.baseUrl}${r.chatCompletionsPath||"/v1/chat/completions"}`,p=await fetch(c,{method:"POST",headers:t.headers,body:JSON.stringify(s),signal:l});if(!p.ok){let f=await p.text();throw p.status===401||p.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${p.status}: ${f}`)}return p.json()}async fetchStreamingCompletion(e,t,r={}){let o={...e,stream:!0},s=ct(o),{body:i,meta:l}=De(o,r.payloadCompaction);r.onBudget?.({streaming:!0,beforeBytes:s,meta:l});let c=this.#e(r),p=`${t.baseUrl}${r.chatCompletionsPath||"/v1/chat/completions"}`,f=await fetch(p,{method:"POST",headers:t.headers,body:JSON.stringify(i),signal:c});if(!f.ok){let y=await f.text();throw f.status===401||f.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${f.status}: ${y}`)}let _=f.body.getReader(),m=new TextDecoder,h="",O="",L=new Map;for(;;){let{done:y,value:x}=await _.read();if(y)break;h+=m.decode(x,{stream:!0});let
|
|
95
|
-
`);h
|
|
94
|
+
[truncated for size budget]`}function Le(n,e=0){if(!n||typeof n!="object"||e>8)return n;if(Array.isArray(n))return n.map(r=>Le(r,e+1));let t={};for(let[r,o]of Object.entries(n))r==="description"||r==="title"||r==="examples"||r==="default"||(t[r]=Le(o,e+1));return t}function no(n=[]){return n.map(e=>({...e,function:{...e.function,description:Ee(e.function?.description||"",180),parameters:Le(e.function?.parameters||{type:"object",properties:{}})}}))}function at(n){let e=new Set;for(let o of n)if(o.role==="assistant"&&Array.isArray(o.tool_calls))for(let s of o.tool_calls)e.add(s.id);let t=n.filter(o=>o.role==="tool"?e.has(o.tool_call_id):!0),r=new Set;for(let o of t)o.role==="tool"&&r.add(o.tool_call_id);return t.map(o=>{if(o.role!=="assistant"||!Array.isArray(o.tool_calls)||o.tool_calls.every(c=>r.has(c.id)))return o;let{tool_calls:i,...l}=o;return{...l,content:l.content||""}})}function De(n,e={}){let t=e.maxBytes||49e3,r=e.systemMaxChars||12e3,o={...n,messages:Array.isArray(n.messages)?[...n.messages]:[],tools:Array.isArray(n.tools)?no(n.tools):n.tools};o.messages.length>0&&o.messages[0]?.role==="system"&&(o.messages[0]={...o.messages[0],content:Ee(o.messages[0].content,r)});let s=!1;for(;Oe(o)>t&&o.messages.length>2;)o.messages.splice(1,1),s=!0;if(s&&(o.messages=at(o.messages)),Oe(o)>t&&o.messages.length>0&&(o.messages[0]={...o.messages[0],content:Ee(o.messages[0].content,6e3)},s=!0),Oe(o)>t){let i=o.messages.find(c=>c.role==="system")||o.messages[0],l=o.messages.slice(-2);o.messages=at([i,...l].filter(Boolean).map((c,p)=>({...c,content:Ee(c.content,p===0?4e3:8e3)}))),s=!0}return{body:o,meta:{bytes:Oe(o),trimmed:s,maxBytes:t,messageCount:o.messages.length}}}var ct=n=>Buffer.byteLength(JSON.stringify(n),"utf8"),oe=class extends fe{async fetchCompletion(e,t,r={}){let o=ct(e),{body:s,meta:i}=De(e,r.payloadCompaction);r.onBudget?.({streaming:!1,beforeBytes:o,meta:i});let l=this.#e(r),c=`${t.baseUrl}${r.chatCompletionsPath||"/v1/chat/completions"}`,p=await fetch(c,{method:"POST",headers:t.headers,body:JSON.stringify(s),signal:l});if(!p.ok){let f=await p.text();throw p.status===401||p.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${p.status}: ${f}`)}return p.json()}async fetchStreamingCompletion(e,t,r={}){let o={...e,stream:!0},s=ct(o),{body:i,meta:l}=De(o,r.payloadCompaction);r.onBudget?.({streaming:!0,beforeBytes:s,meta:l});let c=this.#e(r),p=`${t.baseUrl}${r.chatCompletionsPath||"/v1/chat/completions"}`,f=await fetch(p,{method:"POST",headers:t.headers,body:JSON.stringify(i),signal:c});if(!f.ok){let y=await f.text();throw f.status===401||f.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${f.status}: ${y}`)}let _=f.body.getReader(),m=new TextDecoder,h="",O="",L=new Map;for(;;){let{done:y,value:x}=await _.read();if(y)break;h+=m.decode(x,{stream:!0});let $=h.split(`
|
|
95
|
+
`);h=$.pop();for(let v of $){if(!v.startsWith("data: "))continue;let g=v.slice(6).trim();if(g==="[DONE]")continue;let R;try{R=JSON.parse(g)}catch{continue}let C=R.choices?.[0]?.delta;if(C&&(C.content&&(O+=C.content,r.onToken&&r.onToken(C.content)),C.tool_calls))for(let k of C.tool_calls){let E=k.index??0;L.has(E)||L.set(E,{id:"",name:"",args:""});let S=L.get(E);k.id&&(S.id=k.id),k.function?.name&&(S.name=k.function.name),k.function?.arguments!=null&&(S.args+=k.function.arguments)}}}if(L.size>0){let y=[...L.entries()].sort(([x],[$])=>x-$).map(([,x])=>({id:x.id,type:"function",function:{name:x.name,arguments:x.args}}));return{choices:[{message:{role:"assistant",content:O||null,tool_calls:y}}]}}return{choices:[{message:{role:"assistant",content:O}}]}}#e(e={}){let t=[e.signal,e.timeout?AbortSignal.timeout(e.timeout):null].filter(Boolean);return t.length>1?AbortSignal.any(t):t[0]||void 0}};import{Client as io}from"@modelcontextprotocol/sdk/client/index.js";import{StdioClientTransport as lo}from"@modelcontextprotocol/sdk/client/stdio.js";var Ae=class{#e=new Map;async ensureServer(e,t){if(this.#e.has(e))return this.#e.get(e);let{command:r,args:o=[],env:s={}}=t;a.debug(`[MCP] Starting ${e}: ${r} ${o.join(" ")}`);let i=new lo({command:r,args:o,env:{...process.env,...s}}),l=new io({name:`zibby-chat-${e}`,version:"1.0.0"},{capabilities:{}});await l.connect(i);let c={client:l,transport:i,serverConfig:t};return this.#e.set(e,c),c}async callTool(e,t,r={}){let o=this.#e.get(e);if(!o)throw new Error(`MCP server "${e}" not running`);a.debug(`[MCP] ${e}.${t}(${JSON.stringify(r).slice(0,200)})`);let s=await o.client.callTool({name:t,arguments:r});return{text:s.content?.filter(l=>l.type==="text").map(l=>l.text).join(`
|
|
96
96
|
`)||"",isError:s.isError||!1}}isRunning(e){return this.#e.has(e)}async stopServer(e){let t=this.#e.get(e);if(t){try{await t.client.close()}catch(r){a.debug(`[MCP] Error closing ${e}: ${r.message}`)}this.#e.delete(e)}}async stopAll(){let e=[...this.#e.keys()];await Promise.allSettled(e.map(t=>this.stopServer(t)))}};import{existsSync as ao,readFileSync as co}from"node:fs";import{join as uo}from"node:path";import{homedir as fo}from"node:os";function po(){try{let n=uo(fo(),".zibby","config.json");return ao(n)?JSON.parse(co(n,"utf-8")):{}}catch{return{}}}function be(n){return String(n||"").replace(/\/v1\/?$/,"")}function mo(n,e){let t=process.env.OPENAI_PROXY_URL;if(t)return be(t);if(n==="session")return e.proxyUrl?be(e.proxyUrl):`${(process.env.ZIBBY_API_URL||"https://api-prod.zibby.app").replace(/\/$/,"")}/openai-proxy`;if(n==="byok"){let r=process.env.OPENAI_BASE_URL;return be(r||"https://api.openai.com")}return be(t||"")}function Ue(){let n=po(),e=process.env.ZIBBY_USER_TOKEN||n.sessionToken||null,t=process.env.OPENAI_API_KEY||null,r=(process.env.ASSISTANT_AUTH_MODE||"").trim().toLowerCase(),o=process.env.OPENAI_PROXY_NO_AUTH==="true",s=process.env.OPENAI_PROXY_URL,i="session";r==="byok"||r==="local"||r==="session"?i=r:e?i="session":t?i="byok":s&&o&&(i="local");let l=mo(i,n),c={"Content-Type":"application/json"};if(i==="session"){if(!e)return{ok:!1,mode:i,reason:"missing_session_token"};c.Authorization=`Bearer ${e}`}else if(i==="byok"){if(!t)return{ok:!1,mode:i,reason:"missing_openai_api_key"};c.Authorization=`Bearer ${t}`}else if(i==="local"){if(!l)return{ok:!1,mode:i,reason:"missing_openai_proxy_url"};!o&&t&&(c.Authorization=`Bearer ${t}`)}return l?{ok:!0,mode:i,baseUrl:l,headers:c,tokenPreview:c.Authorization?`***${c.Authorization.slice(-4)}`:"none"}:{ok:!1,mode:i,reason:"missing_base_url"}}function we(n,e){let t=String(n??"");return t.length<=e?t:`${t.slice(0,Math.max(0,e-30))}
|
|
97
97
|
|
|
98
|
-
[tool result truncated for size]`}function go(n,e){if(typeof n=="string")return we(n,e);try{return we(JSON.stringify(n),e)}catch{return we(String(n),e)}}function ut(n){let e=new Set;for(let o of n)if(o.role==="assistant"&&Array.isArray(o.tool_calls))for(let s of o.tool_calls)e.add(s.id);let t=n.filter(o=>o.role==="tool"?e.has(o.tool_call_id):!0),r=new Set;for(let o of t)o.role==="tool"&&r.add(o.tool_call_id);return t.map(o=>{if(o.role!=="assistant"||!Array.isArray(o.tool_calls)||o.tool_calls.every(c=>r.has(c.id)))return o;let{tool_calls:i,...l}=o;return{...l,content:l.content||""}})}function ho(n){let e=Array.isArray(n?.messages)?n.messages:[],t=e.find(s=>s.role==="system"),r=e.slice(-4).map(s=>({...s,content:we(s.content,s.role==="tool"?1200:2500)}));r=ut(r);let o={...n,messages:[t,...r].filter(Boolean)};return delete o.tools,o}async function yo({body:n,streaming:e,auth:t,options:r,fetchCompletion:o,fetchStreamingCompletion:s,onFallbackLog:i}){try{return e?await s(n,t,r):await o(n,t,r)}catch(l){let c=String(l?.message||l||"");if(!/proxy error 413|payload too large/i.test(c))throw l;let p=ho(n);return typeof i=="function"&&i(n,p),{data:e?await s(p,t,r):await o(p,t,r),fallback:p}}}async function ft({body:n,auth:e,options:t,streaming:r,toolContext:o,activeSkills:s,round:i,verbose:l,dependencies:c,config:p={}}){let f=p.maxToolResultChars||3e3,{fetchCompletion:_,fetchStreamingCompletion:m,onFallbackLog:h,hasToolCalls:O,getTextContent:L,parseToolCalls:y,buildAssistantMessage:x,buildToolResultMessage:v,
|
|
98
|
+
[tool result truncated for size]`}function go(n,e){if(typeof n=="string")return we(n,e);try{return we(JSON.stringify(n),e)}catch{return we(String(n),e)}}function ut(n){let e=new Set;for(let o of n)if(o.role==="assistant"&&Array.isArray(o.tool_calls))for(let s of o.tool_calls)e.add(s.id);let t=n.filter(o=>o.role==="tool"?e.has(o.tool_call_id):!0),r=new Set;for(let o of t)o.role==="tool"&&r.add(o.tool_call_id);return t.map(o=>{if(o.role!=="assistant"||!Array.isArray(o.tool_calls)||o.tool_calls.every(c=>r.has(c.id)))return o;let{tool_calls:i,...l}=o;return{...l,content:l.content||""}})}function ho(n){let e=Array.isArray(n?.messages)?n.messages:[],t=e.find(s=>s.role==="system"),r=e.slice(-4).map(s=>({...s,content:we(s.content,s.role==="tool"?1200:2500)}));r=ut(r);let o={...n,messages:[t,...r].filter(Boolean)};return delete o.tools,o}async function yo({body:n,streaming:e,auth:t,options:r,fetchCompletion:o,fetchStreamingCompletion:s,onFallbackLog:i}){try{return e?await s(n,t,r):await o(n,t,r)}catch(l){let c=String(l?.message||l||"");if(!/proxy error 413|payload too large/i.test(c))throw l;let p=ho(n);return typeof i=="function"&&i(n,p),{data:e?await s(p,t,r):await o(p,t,r),fallback:p}}}async function ft({body:n,auth:e,options:t,streaming:r,toolContext:o,activeSkills:s,round:i,verbose:l,dependencies:c,config:p={}}){let f=p.maxToolResultChars||3e3,{fetchCompletion:_,fetchStreamingCompletion:m,onFallbackLog:h,hasToolCalls:O,getTextContent:L,parseToolCalls:y,buildAssistantMessage:x,buildToolResultMessage:$,executeTool:v,onToolCallLog:g,injectTools:R}=c;Array.isArray(n?.messages)&&(n.messages=ut(n.messages));let C=await yo({body:n,streaming:r,auth:e,options:t,fetchCompletion:_,fetchStreamingCompletion:m,onFallbackLog:h}),k=C?.data||C;if(!O(k))return{done:!0,text:L(k),body:C?.fallback||n};let E=y(k),S=C?.fallback||n;S.messages.push(x(k)),l&&typeof g=="function"&&g(E);let M=await Promise.all(E.map((b,P)=>(typeof t.onToolCall=="function"&&t.onToolCall(b.name,b.args,{round:i,index:P,total:E.length}),v(b,o))));for(let b=0;b<E.length;b++){let U=E[b].name==="get_skill_context"?typeof M[b]=="string"?M[b]:JSON.stringify(M[b]):go(M[b],f);S.messages.push($(E[b].id,U))}return typeof t.onToolCall=="function"&&t.onToolCall(null),R(S,s),{done:!1,body:S}}var So=K.ASSISTANT,Oo=15,Be="get_skill_context";function se(n){!n||typeof n!="object"||(n.type==="object"&&n.properties&&(n.required=Object.keys(n.properties),n.additionalProperties=!1,Object.values(n.properties).forEach(se)),n.type==="array"&&n.items&&se(n.items),n.anyOf&&n.anyOf.forEach(se),n.oneOf&&n.oneOf.forEach(se),n.allOf&&n.allOf.forEach(se))}var pt={maxBytes:49e3,systemMaxChars:12e3},re=()=>process.env.ZIBBY_VERBOSE==="true"||process.env.ZIBBY_DEBUG==="true",Eo=n=>Buffer.byteLength(JSON.stringify(n),"utf8");function xe(n){return Array.isArray(n)?new Set(n.map(e=>String(e||"").trim()).filter(Boolean)):new Set}function Ao(n){return Array.isArray(n)?n.map(e=>String(e||"").trim()).filter(Boolean):[]}var $e=class extends _o{#e;#t;#o=new Ae;constructor(e=null){super("assistant","Zibby Assistant",200),e&&typeof e=="object"&&(e.toolProvider||e.completionProvider)?(this.#e=e.toolProvider||new te,this.#t=e.completionProvider||new oe):(this.#e=e||new te,this.#t=new oe)}canHandle(e){return Ue().ok}async invoke(e,t={}){let r=G("OPENAI_API_KEY");try{let o=t.model&&t.model!=="auto"?t.model:So,s=Ue();if(!s.ok)throw s.reason==="missing_session_token"?new Error("Login required. Run `zibby login` to authenticate."):s.reason==="missing_openai_api_key"?new Error("Missing OPENAI_API_KEY for BYOK mode."):s.reason==="missing_openai_proxy_url"?new Error("Missing OPENAI_PROXY_URL for local mode."):new Error(`Assistant auth unavailable (${s.reason}).`);let i=t.messages||[{role:"user",content:e}],l=s.baseUrl,c=re();if(c?await this.#m(i,o,l,s.tokenPreview||"none"):a.debug(`[Assistant] ${l} | model: ${o} | messages: ${i.length}`),t.schema)return this.#c(o,i,s,t);let p=t.activeSkills||[],f=this.#f(t),_=this.#p(p,f),m=this.#d(_),h=this.#u(t),O={...t,payloadCompaction:h},L={activeSkills:_,options:O,executionRegistry:m,capabilityPolicy:f},y=!!t.stream,x={model:o,messages:[...i],stream:!1};this.#s(x,_,f);for(let $=0;$<Oo;$++){if(t.signal?.aborted){let g=new Error("Aborted via signal");throw g.name="AbortError",g}let v=await ft({body:x,auth:s,options:O,streaming:y,toolContext:L,activeSkills:_,round:$,verbose:c,dependencies:{fetchCompletion:this.#n.bind(this),fetchStreamingCompletion:this.#a.bind(this),onFallbackLog:(g,R)=>{re()&&console.log(`413 fallback: messages ${g.messages.length} -> ${R.messages.length}, bytes=${Eo(R)}`)},hasToolCalls:g=>this.#e.hasToolCalls(g),getTextContent:g=>this.#e.getTextContent(g),parseToolCalls:g=>this.#e.parseToolCalls(g),buildAssistantMessage:g=>this.#e.buildAssistantMessage(g),buildToolResultMessage:(g,R)=>this.#e.buildToolResultMessage(g,R),executeTool:(g,R)=>this.#l(g,R),onToolCallLog:async g=>{let R=(await import("chalk")).default;console.log(R.dim(` ${g.map(C=>`${C.name}(${JSON.stringify(C.args).slice(0,80)})`).join(", ")}`))},injectTools:(g,R)=>this.#s(g,R,f)},config:{maxToolResultChars:t.maxToolResultChars||3e3}});if(v.done)return v.text;x.messages=v.body.messages,v.body.tools?x.tools=v.body.tools:delete x.tools}return"I hit my tool-calling limit for this turn. Please try again."}finally{r.dispose()}}async cleanup(){await this.#o.stopAll()}#s(e,t,r=null){let o=this.#i(t,r),s=this.#e.formatTools(o);this.#e.injectToolsIntoBody(e,s)}#i(e,t=null){let r=[],o=new Set;for(let s of e){let i=Te(s);if(i?.tools?.length)for(let l of i.tools)this.#r(l.name,t)&&(o.has(l.name)||(o.add(l.name),r.push({name:l.name,description:l.description,parameters:l.parameters||l.input_schema||{type:"object",properties:{}}})))}return!t?.disableSkillContextTool&&this.#r(Be,t)&&r.push({name:Be,description:"Fetch full prompt guidance/instructions for one installed skill on demand. Use this before making complex tool decisions if guidance is needed.",parameters:{type:"object",properties:{skillId:{type:"string",description:"Installed skill id to inspect (e.g. jira, github, runner, chat-memory)"}},required:["skillId"]}}),r}async#l(e,t){let{activeSkills:r,options:o,executionRegistry:s,capabilityPolicy:i}=t;if(!this.#r(e.name,i))return`Tool "${e.name}" blocked by policy`;if(e.name===Be){let p=String(e.args?.skillId||"").trim();if(!p)return JSON.stringify({error:"skillId is required"});if(!r.includes(p))return JSON.stringify({error:`Skill "${p}" is not active`,activeSkills:r});let f=Te(p);if(!f)return JSON.stringify({error:`Skill "${p}" not found`});let _=typeof f.promptFragment=="function"?f.promptFragment():f.promptFragment||"",m=(f.tools||[]).map(O=>O.name),h=JSON.stringify({skillId:p,description:f.description||"",toolNames:m,promptFragment:_||""});return re()&&(console.log(`
|
|
99
99
|
\u{1F4D6} get_skill_context("${p}") \u2192 ${h.length} chars (fragment: ${_.length} chars)`),console.log(` tools: [${m.join(", ")}]`),console.log(` fragment preview: ${_.slice(0,200).replace(/\n/g,"\\n")}\u2026
|
|
100
100
|
`)),h}let l=s?.get(e.name)||null;if(!l)return`Unknown tool: ${e.name}`;let c=Te(l.skillId);if(!c)return`Skill "${l.skillId}" not found for tool "${e.name}"`;if(l.mode==="handler")try{return c.handleToolCall(e.name,e.args,t)}catch(p){return`Error in ${e.name}: ${p.message}`}if(l.mode==="mcp")try{if(!this.#o.isRunning(c.serverName)){let f=c.resolve(o);if(!f)return`Skill "${l.skillId}" is not available (cannot start server)`;await this.#o.ensureServer(c.serverName,f)}let p=await this.#o.callTool(c.serverName,e.name,e.args);return p.text||(p.isError?"Tool call failed":"Done")}catch(p){return`MCP error (${c.serverName}): ${p.message}`}return`Skill "${l.skillId}" owns tool "${e.name}" but has no execution mode`}async#a(e,t,r){return this.#t.fetchStreamingCompletion(e,t,{...r,onBudget:({beforeBytes:o,meta:s})=>{re()&&console.log(`payload bytes (stream) before=${o} after=${s.bytes} trimmed=${s.trimmed} messages=${s.messageCount}`)}})}async#n(e,t,r){return this.#t.fetchCompletion(e,t,{...r,onBudget:({beforeBytes:o,meta:s})=>{re()&&console.log(`payload bytes before=${o} after=${s.bytes} trimmed=${s.trimmed} messages=${s.messageCount}`)}})}async#c(e,t,r,o){let{zodToJsonSchema:s}=await import("zod-to-json-schema"),i=typeof o.schema?.parse=="function",l=i?s(o.schema):o.schema;delete l.$schema,se(l);let c={model:e,messages:t,stream:!1,response_format:{type:"json_schema",json_schema:{name:"extract",schema:l,strict:!0}}},p=await this.#n(c,r,o),f=this.#e.getTextContent(p),_=JSON.parse(f),m=i?o.schema.parse(_):_;return{raw:f,structured:m}}#u(e={}){let t=e?.config?.agent?.assistant?.payloadCompaction||{};return{maxBytes:Number(t.maxBytes||e.maxPayloadBytes||pt.maxBytes),systemMaxChars:Number(t.systemMaxChars||pt.systemMaxChars)}}#f(e={}){let t=e?.config?.agent?.assistant?.toolPolicy||{};return{allowTools:xe(t.allowTools||e.allowTools),denyTools:xe(t.denyTools||e.denyTools),denyPrefixes:Ao(t.denyPrefixes||e.denyToolPrefixes),includeSkills:xe(t.includeSkills||e.includeSkills),excludeSkills:xe(t.excludeSkills||e.excludeSkills),disableSkillContextTool:!!(t.disableSkillContextTool||e.disableSkillContextTool)}}#p(e,t){let r=t?.includeSkills||new Set,o=t?.excludeSkills||new Set;return r.size===0&&o.size===0?e:e.filter(s=>!(r.size>0&&!r.has(s)||o.has(s)))}#r(e,t){let r=String(e||"").trim();if(!r)return!1;let o=t?.allowTools;if(o&&o.size>0&&!o.has(r))return!1;let s=t?.denyTools;return!(s&&s.has(r)||(t?.denyPrefixes||[]).some(l=>r.startsWith(l)))}#d(e){let t=new Map,r=[];for(let o of e){let s=Te(o);if(!s?.tools?.length)continue;let i=typeof s.handleToolCall=="function"?"handler":s.serverName&&typeof s.resolve=="function"?"mcp":null;if(i)for(let l of s.tools){let c=String(l?.name||"").trim();if(c){if(t.has(c)){r.push({tool:c,winner:t.get(c).skillId,skipped:o});continue}t.set(c,{skillId:o,mode:i})}}}if(r.length>0&&re()){let o=r.slice(0,5).map(s=>`${s.tool}:${s.winner}>${s.skipped}`).join(", ");console.log(`tool registry collisions: ${o}${r.length>5?" ...":""}`)}return t}async#m(e,t,r,o){console.log(`
|
|
101
101
|
\u25C6 Model: ${t} | proxy: ${r} | token: ${o||"none"}
|
|
102
|
-
`);let s=(await import("chalk")).default;console.log(s.bold("Prompt sent to LLM:")),console.log(s.dim("\u2500".repeat(60)));let i=!1;for(let l of e)if(l.role==="system")console.log(s.dim(`[System] ${l.content||""}`));else{i||(console.log(s.dim("\u2500\u2500\u2500 chat history \u2500\u2500\u2500")),i=!0);let c=l.role==="user"?"Human":"AI",p=l.content?.length>200?`${l.content.slice(0,200)}...`:l.content||"";console.log(s.dim(`[${c}] ${p}`))}console.log(s.dim("\u2500".repeat(60)))}};var dt=!1;function bs(){dt||(pe(new
|
|
102
|
+
`);let s=(await import("chalk")).default;console.log(s.bold("Prompt sent to LLM:")),console.log(s.dim("\u2500".repeat(60)));let i=!1;for(let l of e)if(l.role==="system")console.log(s.dim(`[System] ${l.content||""}`));else{i||(console.log(s.dim("\u2500\u2500\u2500 chat history \u2500\u2500\u2500")),i=!0);let c=l.role==="user"?"Human":"AI",p=l.content?.length>200?`${l.content.slice(0,200)}...`:l.content||"";console.log(s.dim(`[${c}] ${p}`))}console.log(s.dim("\u2500".repeat(60)))}};var dt=!1;function bs(){dt||(pe(new $e),pe(new he),pe(new ye),pe(new _e),pe(new Se),dt=!0)}export{bs as registerBuiltInStrategies};
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import{AgentStrategy as te,getSkill as oe}from"@zibby/agent-workflow";import{query as
|
|
1
|
+
import{AgentStrategy as te,getSkill as oe}from"@zibby/agent-workflow";import{query as se}from"@anthropic-ai/claude-agent-sdk";import{zodToJsonSchema as ne}from"zod-to-json-schema";import{writeFileSync as P,mkdirSync as x,existsSync as q,readFileSync as X}from"fs";import{homedir as z}from"os";import{join as A}from"path";import y from"chalk";var f={debug:0,info:1,warn:2,error:3,silent:4},v=class{constructor(){this._level=this._getLogLevel()}_getLogLevel(){if(process.env.ZIBBY_DEBUG==="true")return f.debug;if(process.env.ZIBBY_VERBOSE==="true")return f.info;let e=process.env.LOG_LEVEL?.toLowerCase();return e&&e in f?f[e]:f.info}_shouldLog(e){return f[e]>=this._level}_formatMessage(e,s,r={}){let n=new Date().toISOString(),a=`${this._getPrefix(e)} ${s}`;return Object.keys(r).length>0&&(a+=y.dim(` ${JSON.stringify(r)}`)),a}_getPrefix(e){return{debug:y.gray("[DEBUG]"),info:y.cyan("[INFO]"),warn:y.yellow("[WARN]"),error:y.red("\u274C [ERROR]")}[e]||""}debug(e,s){this._shouldLog("debug")&&console.log(this._formatMessage("debug",e,s))}info(e,s){this._shouldLog("info")&&console.log(this._formatMessage("info",e,s))}warn(e,s){this._shouldLog("warn")&&console.warn(this._formatMessage("warn",e,s))}error(e,s){this._shouldLog("error")&&console.error(this._formatMessage("error",e,s))}setLevel(e){e in f&&(this._level=f[e])}getLevel(){return Object.keys(f).find(e=>f[e]===this._level)}},i=new v;import{timeline as b,Timeline as pe,WORKFLOW_GRAPH_LOG_MARKER_PREFIX as ge}from"@zibby/agent-workflow";var I={ASSISTANT:"gpt-5.4-nano-2026-03-17",CLAUDE:"claude-sonnet-4-6",CURSOR:"auto",CODEX:"o4-mini",GEMINI:"gemini-2.5-pro",OPENAI_POSTPROCESSING:"gpt-4o-mini"};var $={auto:"claude-sonnet-4-6","sonnet-4.6":"claude-sonnet-4-6","sonnet-4-6":"claude-sonnet-4-6","opus-4.6":"claude-opus-4-6","opus-4-6":"claude-opus-4-6","sonnet-4.5":"claude-sonnet-4-5-20250929","sonnet-4-5":"claude-sonnet-4-5-20250929","opus-4.5":"claude-opus-4-20250514","opus-4-5":"claude-opus-4-20250514","claude-sonnet-4-6":"claude-sonnet-4-6","claude-opus-4-6":"claude-opus-4-6","claude-sonnet-4-5-20250929":"claude-sonnet-4-5-20250929","claude-opus-4-20250514":"claude-opus-4-20250514"};var fe={CURSOR_AGENT_DEFAULT:1200*1e3,OPENAI_REQUEST:18e4};function D(l){let e=`${l}_POOL`,s=process.env[e];if(!s||typeof s!="string")return{picked:null,count:0,dispose:()=>{}};let r=s.split(/[,\n]+/).map(a=>a.trim()).filter(Boolean);if(r.length===0)return{picked:null,count:0,dispose:()=>{}};let n=r[Math.floor(Math.random()*r.length)],o=process.env[l];return process.env[l]=n,i.debug(`[auth-pool] ${l}: picked 1 of ${r.length} from pool (***${n.slice(-4)})`),{picked:n,count:r.length,dispose(){o===void 0?delete process.env[l]:process.env[l]=o}}}var Z=class extends te{constructor(){super("claude","Claude (Anthropic API)",50)}canHandle(e){let s=!!process.env.ANTHROPIC_API_KEY||!!process.env.ANTHROPIC_API_KEY_POOL,r=!!process.env.CLAUDE_CODE_OAUTH_TOKEN||!!process.env.CLAUDE_CODE_OAUTH_TOKEN_POOL,n=!!process.env.ANTHROPIC_AUTH_TOKEN||!!process.env.ANTHROPIC_AUTH_TOKEN_POOL,o=s||r||n;return o||i.debug("ClaudeAgentStrategy: no credentials. Set one of ANTHROPIC_API_KEY / CLAUDE_CODE_OAUTH_TOKEN / ANTHROPIC_AUTH_TOKEN (or a *_POOL variant for multi-token rotation)."),o}async invoke(e,s={}){let{model:r,workspace:n=process.cwd(),schema:o=null,images:a=[],skills:d=null,sessionPath:m=null,nodeName:N=null,timeout:M,config:R={},signal:E=null}=s,g=r;(!g||g==="auto")&&(i.debug(`Model is '${g||"undefined"}', using default: ${I.CLAUDE}`),g=I.CLAUDE);let S=$[g]||g;$[g]&&g!==S&&i.debug(`Mapped model: ${g} \u2192 ${S}`);let Q=[D("ANTHROPIC_API_KEY"),D("CLAUDE_CODE_OAUTH_TOKEN"),D("ANTHROPIC_AUTH_TOKEN")];try{i.debug(`Invoking Claude Agent SDK with model: ${S}, skills: ${JSON.stringify(d)}`);let O=process.env.CLAUDE_CODE_OAUTH_TOKEN,U=process.env.ANTHROPIC_API_KEY,K=process.env.ANTHROPIC_AUTH_TOKEN;process.env.CLAUDE_CODE_CREDENTIALS_JSON?ce(process.env.CLAUDE_CODE_CREDENTIALS_JSON):O&&ie(O,process.env.__ZIBBY_CLAUDE_PLAN);let k=process.env.__ZIBBY_CLAUDE_PLAN,J=k?k.charAt(0).toUpperCase()+k.slice(1):null,T;O?T=J?`Claude ${J} plan \xB7 ***${O.slice(-4)}`:`Subscription \xB7 ***${O.slice(-4)}`:U?T=`API key \xB7 ***${U.slice(-4)}`:K?T=`Bearer \xB7 ***${K.slice(-4)}`:T="NONE \u2014 request will fail";let _=(await import("chalk")).default;console.log(""),console.log(` ${_.cyan("\u25C6")} ${_.bold(S)}`),console.log(` ${_.dim("Auth: ")}${T}`),console.log(` ${_.dim("Workdir: ")}${n}`),console.log(""),console.log(_.bold("Prompt sent to LLM:")),console.log(_.dim("\u2500".repeat(60))),console.log(_.dim(e)),console.log(_.dim("\u2500".repeat(60)));let{allowedTools:F,mcpServers:H}=this._resolveSkills(d,{sessionPath:m,workspace:n,nodeName:N});try{let p={cwd:n,allowedTools:F,permissionMode:"bypassPermissions",allowDangerouslySkipPermissions:!0,settings:{permissions:{allow:["Bash(*)","Read(*)","Write(*)","Edit(*)","MultiEdit(*)","NotebookEdit(*)","Grep(*)","Glob(*)","WebFetch(*)","WebSearch(*)","Task(*)","TodoWrite(*)","Skill(*)"],defaultMode:"bypassPermissions"}},model:S,...Object.keys(H).length>0&&{mcpServers:H}};if(o){let h=typeof o.parse=="function"?ne(o,{target:"openApi3"}):o;p.outputFormat={type:"json_schema",schema:h},i.debug("Structured output enforced via SDK outputFormat")}i.debug(`Agent SDK options: ${JSON.stringify({cwd:p.cwd,toolCount:F.length,permissionMode:p.permissionMode,model:p.model,hasOutputFormat:!!p.outputFormat})}`);let L="",B=0,G=[];i.debug("Starting Claude Agent SDK query stream");let j;try{j=se({prompt:e,options:p})}catch(t){throw i.error(`Failed to initialize Claude Agent SDK: ${t.message}`),t}let W=null,w=0,V=3;if(E?.aborted){let t=new Error("Aborted via signal");throw t.name="AbortError",t}try{for await(let t of j){if(E?.aborted){let u=new Error("Aborted via signal");throw u.name="AbortError",u}if(G.push(t),t.type==="error"||t.error){let u=t.error?.message||t.error||t.message||"Unknown API error";try{i.error("SDK error event raw payload",{fullMessage:JSON.stringify(t,null,2)})}catch{}throw new Error(typeof u=="string"?u:JSON.stringify(u))}let h=JSON.stringify(t.message?.content||t.text||"").slice(0,200);if(h===W){if(w++,w>=V){let u=(t.message?.content?.[0]?.text||t.text||"unknown").slice(0,100);throw new Error(`API stuck in loop (${w}x repeated): ${u}`)}}else W=h,w=1;if(t.type==="assistant"||t.constructor?.name==="AssistantMessage"){let u=t.message?.content||t.content||[];for(let c of u)if(c.type==="thinking"&&c.thinking)console.log(`${c.thinking.substring(0,200)}${c.thinking.length>200?"...":""}`);else if(c.type==="text"&&c.text)L+=c.text,c.text.length<500?console.log(`${c.text}`):console.log(`${c.text.substring(0,200)}... (${c.text.length} chars)`);else if(c.type==="tool_use"){B++,c.name.includes("memory")?b.stepMemory(`Tool: ${c.name}`):b.stepTool(`Tool: ${c.name}`);let ee=JSON.stringify(c.input).substring(0,100);console.log(` Input: ${ee}${JSON.stringify(c.input).length>100?"...":""}`)}}else if(!(t.type==="user"&&t.tool_use_result)){if(t.type==="result"||t.constructor?.name==="ResultMessage"){let u=t.result||t.text||t.content||L;try{re({message:t,sessionPath:m,nodeName:N,model:g,toolCallCount:B})}catch(c){i.debug(`usage.json write failed: ${c.message}`)}if(o){if(t.structured_output){i.debug("Using SDK native structured_output");let Y=typeof o.parse=="function"?o.parse(t.structured_output):t.structured_output;return{raw:u,structured:Y}}if(u){let c=this._extractJson(u,o);if(c)return{raw:u,structured:c}}i.warn(`Could not extract structured output \u2014 returning raw text (${(u||"").length} chars)`)}return u||""}}}if(i.warn(`Agent SDK ended without result. Collected ${G.length} messages`),L.length>0)return i.debug("Returning accumulated text from messages"),L;throw new Error("Claude Agent SDK query ended without result")}catch(t){let h={message:t?.message,name:t?.name,code:t?.code,cause:t?.cause?.message||t?.cause,stack:t?.stack?.split(`
|
|
2
2
|
`).slice(0,5).join(`
|
|
3
|
-
`),stringified:(()=>{try{return JSON.stringify(t,Object.getOwnPropertyNames(Object(t)))}catch{return String(t)}})()};throw i.error(`Error during query stream: ${t?.message||t}`,
|
|
3
|
+
`),stringified:(()=>{try{return JSON.stringify(t,Object.getOwnPropertyNames(Object(t)))}catch{return String(t)}})()};throw i.error(`Error during query stream: ${t?.message||t}`,h),t}}catch(p){throw i.error("Claude Agent SDK call failed",{message:p?.message,name:p?.name,code:p?.code,cause:p?.cause?.message||p?.cause}),p}}finally{for(let O of Q)O.dispose()}}_resolveSkills(e,s){let r=["Read","Write","Edit","Bash","Grep","Glob"];if(e===null)return i.debug("No skills \u2014 pure LLM mode"),{allowedTools:[],mcpServers:{}};if(!Array.isArray(e)||e.length===0)return i.debug("Default IDE skills for code generation"),{allowedTools:[...r],mcpServers:{}};let n=[...r],o={};for(let a of e){let d=oe(a);if(!d){i.warn(`Unknown skill "${a}" \u2014 skipping`);continue}if(d.allowedTools&&n.push(...d.allowedTools),typeof d.resolve=="function"){let m=d.resolve(s);m&&(o[d.serverName]=m,i.debug(`MCP: ${d.serverName} \u2192 ${m.command} ${m.args[0]}`))}}return{allowedTools:n,mcpServers:o}}_extractJson(e,s){let r=[()=>{if(e.includes("===JSON_START===")){let n=e.indexOf("===JSON_START===")+16,o=e.indexOf("===JSON_END===");return e.substring(n,o).trim()}},()=>e.match(/```json\s*\n([\s\S]*?)\n```/)?.[1]?.trim(),()=>{if(!e.startsWith("{"))return e.match(/```\s*\n([\s\S]*?)\n```/)?.[1]?.trim()},()=>e.trim(),()=>{let n=e.indexOf("{"),o=e.lastIndexOf("}");if(n!==-1&&o>n)return e.substring(n,o+1)}];for(let n of r)try{let o=n();if(!o)continue;let a=JSON.parse(o);if(typeof a!="object"||a===null)continue;return typeof s.parse=="function"?s.parse(a):a}catch{}return null}};function re({message:l,sessionPath:e,nodeName:s,model:r,toolCallCount:n}){if(!e||!s)return;let o=l?.usage||l?.message?.usage||{},a=C(o.input_tokens,o.inputTokens),d=C(o.output_tokens,o.outputTokens),m=C(o.cache_read_input_tokens,o.cacheReadInputTokens),N=C(o.cache_creation_input_tokens,o.cacheCreationInputTokens),M=C(l?.duration_ms,l?.durationMs),R={model:r||null,input_tokens:a,output_tokens:d,cache_read_tokens:m,cache_creation_tokens:N,tool_call_count:typeof n=="number"?n:null,duration_ms:M,captured_at:new Date().toISOString()},E=A(e,s);try{x(E,{recursive:!0})}catch{}P(A(E,"usage.json"),JSON.stringify(R,null,2),"utf-8")}function C(...l){for(let e of l){if(typeof e=="number"&&Number.isFinite(e))return e;if(typeof e=="string"&&e.trim()!==""&&!isNaN(Number(e)))return Number(e)}return null}function ie(l,e){let s=A(z(),".claude"),r=A(s,".credentials.json"),n={claudeAiOauth:{accessToken:l,...e?{subscriptionType:String(e).toLowerCase()}:{}}};try{if(q(r)&&JSON.parse(X(r,"utf-8"))?.claudeAiOauth?.accessToken===l)return;x(s,{recursive:!0}),P(r,JSON.stringify(n,null,2),"utf-8"),i.debug(`Wrote Claude Code OAuth credentials to ${r}`)}catch(o){i.warn(`Could not write ${r}: ${o.message} \u2014 Claude SDK may fail to authenticate.`)}}function ce(l){let e=A(z(),".claude"),s=A(e,".credentials.json"),r;try{r=JSON.parse(l)}catch(n){i.warn(`CLAUDE_CODE_CREDENTIALS_JSON not valid JSON: ${n.message}`);return}if(!r?.claudeAiOauth?.accessToken){i.warn("CLAUDE_CODE_CREDENTIALS_JSON missing claudeAiOauth.accessToken \u2014 skipping write.");return}try{if(q(s)){let n=JSON.parse(X(s,"utf-8"));if(n?.claudeAiOauth?.accessToken===r.claudeAiOauth.accessToken&&n?.claudeAiOauth?.expiresAt===r.claudeAiOauth.expiresAt)return}x(e,{recursive:!0}),P(s,JSON.stringify(r,null,2),"utf-8"),i.debug(`Wrote full Claude Code credentials blob to ${s}`)}catch(n){i.warn(`Could not write ${s}: ${n.message} \u2014 Claude SDK may fail to authenticate.`)}}export{Z as ClaudeAgentStrategy};
|