@zibby/core 0.1.38 → 0.1.40
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/agents/base.js +47 -40
- package/dist/framework/agents/cursor-strategy.js +29 -22
- package/dist/framework/agents/gemini-strategy.js +11 -11
- package/dist/framework/agents/index.js +45 -38
- package/dist/framework/agents/utils/cursor-output-formatter.js +2 -2
- package/dist/framework/agents/utils/structured-output-formatter.js +1 -1
- package/dist/framework/code-generator.js +33 -26
- package/dist/framework/graph-compiler.js +47 -40
- package/dist/framework/graph.js +50 -43
- package/dist/framework/index.js +23 -16
- package/dist/framework/node-registry.js +39 -32
- package/dist/framework/node.js +43 -36
- package/dist/index.js +47 -40
- package/dist/package.json +1 -1
- package/package.json +1 -1
package/dist/framework/index.js
CHANGED
|
@@ -34,7 +34,7 @@ JSON types (strict \u2014 validators reject wrong types):
|
|
|
34
34
|
- Use true/false without quotes for booleans.
|
|
35
35
|
- Use unquoted null where a field may be null.
|
|
36
36
|
|
|
37
|
-
Rules: valid JSON only, no markdown wrapping, no extra text in the file.`}static _buildExample(e){if(!e)return{};let t=e.type;if(t==="object"&&e.properties){let o={};for(let[r,n]of Object.entries(e.properties))o[r]=this._buildExample(n);return o}if(t==="array"&&e.items)return[this._buildExample(e.items)];if(t==="string")return"<string>";if(t==="number"||t==="integer")return 0;if(t==="boolean")return!1;if(e.description)return`<${e.description}>`;if(e.nullable||e.oneOf||e.anyOf){let o=e.oneOf?.find(r=>r.type!=="null")||e.anyOf?.find(r=>r.type!=="null");return o?this._buildExample(o):null}return"<value>"}}});import Jo from"axios";import{homedir as Go}from"node:os";import{join as Ko}from"node:path";import{existsSync as Yo,readFileSync as Wo}from"node:fs";import{toJSONSchema as zo}from"zod";function Zo(){if(process.env.OPENAI_PROXY_TOKEN)return u.debug("[Auth] Using OPENAI_PROXY_TOKEN (ECS execution)"),process.env.OPENAI_PROXY_TOKEN;if(process.env.ZIBBY_USER_TOKEN)return u.debug("[Auth] Using ZIBBY_USER_TOKEN (CI/CD PAT)"),process.env.ZIBBY_USER_TOKEN;try{let s=Ko(Go(),".zibby","config.json");if(Yo(s)){let e=JSON.parse(Wo(s,"utf-8"));if(e.sessionToken)return u.debug("[Auth] Using session token from zibby login"),e.sessionToken}}catch(s){u.debug(`[Auth] Could not read zibby login session: ${s.message}`)}return null}function Ho(){return process.env.OPENAI_PROXY_URL?process.env.OPENAI_PROXY_URL.replace(/\/v1\/?$/,""):"https://api-prod.zibby.app/openai-proxy"}function ce(s){if(!(typeof s!="object"||s===null)){if(Object.keys(s).length===0){s.type="object",s.additionalProperties=!0;return}if(s.type||(s.properties?s.type="object":s.items&&(s.type="array")),s.type==="object")if(s.properties){for(let[e,t]of Object.entries(s.properties))t.type==="object"&&t.additionalProperties&&t.additionalProperties!==!1&&(!t.properties||Object.keys(t.properties).length===0)&&(s.properties[e]={type:["object","null"]});s.additionalProperties=!1,s.required=Object.keys(s.properties),Object.values(s.properties).forEach(ce)}else"additionalProperties"in s||(s.additionalProperties=!0);s.type==="array"&&s.items&&ce(s.items),s.anyOf&&s.anyOf.forEach(ce),s.oneOf&&s.oneOf.forEach(ce),s.allOf&&s.allOf.forEach(ce)}}async function Ft(s,e){u.info("\u{1F527} [OpenAI Proxy] Formatting structured output...");let t=Zo();if(!t)throw new Error("Authentication required for structured output processing.\n Local development: Run `zibby login`\n CI/CD: Set ZIBBY_USER_TOKEN environment variable (Personal Access Token from UI settings)");let o=Ho();u.info(`\u{1F517} Using OpenAI proxy: ${o}`);let r=zo(e),n=r;if(r.$ref&&r.definitions){let d=r.$ref.split("/").pop();n=r.definitions[d]||r,u.debug(`Extracted schema from $ref: ${d}`)}delete n.$schema,ce(n);let i=4e5,a=s;s.length>i&&(u.warn(`\u26A0\uFE0F [OpenAI Proxy] Raw text (${s.length} chars) exceeds limit, keeping last ${i} chars`),a=`... [truncated early content] ...
|
|
37
|
+
Rules: valid JSON only, no markdown wrapping, no extra text in the file.`}static _buildExample(e){if(!e)return{};let t=e.type;if(t==="object"&&e.properties){let o={};for(let[r,n]of Object.entries(e.properties))o[r]=this._buildExample(n);return o}if(t==="array"&&e.items)return[this._buildExample(e.items)];if(e.enum&&Array.isArray(e.enum)&&e.enum.length>0)return`<${e.enum.join("|")}>`;if(t==="string")return"<string>";if(t==="number"||t==="integer")return 0;if(t==="boolean")return!1;if(e.description)return`<${e.description}>`;if(e.nullable||e.oneOf||e.anyOf){let o=e.oneOf?.find(r=>r.type!=="null")||e.anyOf?.find(r=>r.type!=="null");return o?this._buildExample(o):null}return"<value>"}}});import Jo from"axios";import{homedir as Go}from"node:os";import{join as Ko}from"node:path";import{existsSync as Yo,readFileSync as Wo}from"node:fs";import{toJSONSchema as zo}from"zod";function Zo(){if(process.env.OPENAI_PROXY_TOKEN)return u.debug("[Auth] Using OPENAI_PROXY_TOKEN (ECS execution)"),process.env.OPENAI_PROXY_TOKEN;if(process.env.ZIBBY_USER_TOKEN)return u.debug("[Auth] Using ZIBBY_USER_TOKEN (CI/CD PAT)"),process.env.ZIBBY_USER_TOKEN;try{let s=Ko(Go(),".zibby","config.json");if(Yo(s)){let e=JSON.parse(Wo(s,"utf-8"));if(e.sessionToken)return u.debug("[Auth] Using session token from zibby login"),e.sessionToken}}catch(s){u.debug(`[Auth] Could not read zibby login session: ${s.message}`)}return null}function Ho(){return process.env.OPENAI_PROXY_URL?process.env.OPENAI_PROXY_URL.replace(/\/v1\/?$/,""):"https://api-prod.zibby.app/openai-proxy"}function ce(s){if(!(typeof s!="object"||s===null)){if(Object.keys(s).length===0){s.type="object",s.additionalProperties=!0;return}if(s.type||(s.properties?s.type="object":s.items&&(s.type="array")),s.type==="object")if(s.properties){for(let[e,t]of Object.entries(s.properties))t.type==="object"&&t.additionalProperties&&t.additionalProperties!==!1&&(!t.properties||Object.keys(t.properties).length===0)&&(s.properties[e]={type:["object","null"]});s.additionalProperties=!1,s.required=Object.keys(s.properties),Object.values(s.properties).forEach(ce)}else"additionalProperties"in s||(s.additionalProperties=!0);s.type==="array"&&s.items&&ce(s.items),s.anyOf&&s.anyOf.forEach(ce),s.oneOf&&s.oneOf.forEach(ce),s.allOf&&s.allOf.forEach(ce)}}async function Ft(s,e){u.info("\u{1F527} [OpenAI Proxy] Formatting structured output...");let t=Zo();if(!t)throw new Error("Authentication required for structured output processing.\n Local development: Run `zibby login`\n CI/CD: Set ZIBBY_USER_TOKEN environment variable (Personal Access Token from UI settings)");let o=Ho();u.info(`\u{1F517} Using OpenAI proxy: ${o}`);let r=zo(e),n=r;if(r.$ref&&r.definitions){let d=r.$ref.split("/").pop();n=r.definitions[d]||r,u.debug(`Extracted schema from $ref: ${d}`)}delete n.$schema,ce(n);let i=4e5,a=s;s.length>i&&(u.warn(`\u26A0\uFE0F [OpenAI Proxy] Raw text (${s.length} chars) exceeds limit, keeping last ${i} chars`),a=`... [truncated early content] ...
|
|
38
38
|
${s.slice(-i)}`);let l=`Extract and format the following information into structured JSON matching the schema.
|
|
39
39
|
|
|
40
40
|
RAW CONTENT:
|
|
@@ -60,42 +60,49 @@ ${A}`}let h=process.env.CURSOR_API_KEY,$=h?` | key: ***${h.slice(-4)}`:" | key:
|
|
|
60
60
|
\u25C6 Model: ${f||"auto"}${$}
|
|
61
61
|
`);let x=(await import("chalk")).default;console.log(`
|
|
62
62
|
${x.bold("Prompt sent to LLM:")}`),console.log(x.dim("\u2500".repeat(60))),console.log(x.dim(e)),console.log(x.dim("\u2500".repeat(60)));let v=["--print","--force","--approve-mcps","--output-format","stream-json","--stream-partial-output","--model",f||"auto"];if(process.env.CURSOR_API_KEY&&v.push("--api-key",process.env.CURSOR_API_KEY),v.push(e),u.debug(`[Agent] Prompt: ${e.length} chars, model: ${f||"auto"}`),u.debug(`[Agent] Workspace: ${o}`),process.env.LOG_LEVEL==="debug"||process.env.ZIBBY_LOG_CURSOR_CLI==="1")try{console.log(`\u{1F527} Cursor CLI --model ${f||"auto"} (from .zibby.config.js agent.cursor.model)
|
|
63
|
-
`)}catch{}let T,E=null;try{let O=a||(process.env.ZIBBY_SESSION_PATH?String(process.env.ZIBBY_SESSION_PATH).trim():null);T=await this._spawnWithStreaming(y,v,o,c,null,O,g)}catch(O){E=O}let
|
|
63
|
+
`)}catch{}let T,E=null;try{let O=a||(process.env.ZIBBY_SESSION_PATH?String(process.env.ZIBBY_SESSION_PATH).trim():null);T=await this._spawnWithStreaming(y,v,o,c,null,O,g)}catch(O){E=O}let k=T?.stdout||"";if(n){let O=typeof n.parse=="function",S=null,A=!!(w&&$e(w));if(w&&u.info(`[Agent] Result file: ${A?"present":"missing"} at ${w}`),A)try{let m=Ht(w,"utf-8").trim();S=JSON.parse(m),u.info(`[Agent] Parsed JSON from result file OK (${m.length} chars) \u2192 object ready for validation`),E&&u.debug("[Agent] Agent exited non-zero but result file was written \u2014 recovering")}catch(m){u.warn(`\u26A0\uFE0F [Agent] Result file exists on disk but is not valid JSON: ${m.message}`)}else if(E)u.warn(`[Agent] Result file missing at ${w} (agent process error \u2014 may still recover if strictMode repairs)`);else throw u.error(`\u274C [Agent] Result file was never created at ${w}`),new Error(`Agent did not write required result file at ${w}`);if(S&&O)try{let m=n.parse(S);return u.info("\u2705 [Agent] Zod validation passed for structured result file"),p&&u.debug("[Agent] strictMode enabled but not needed \u2014 agent wrote valid file"),{raw:k,structured:m}}catch(m){let C=m.issues?m.issues.map(P=>` - ${P.path.join(".")}: ${P.message}`).join(`
|
|
64
|
+
`):m.message?.slice(0,400);if(u.error(`\u274C [Agent] Zod validation FAILED - invalid output schema:
|
|
65
|
+
${C}`),u.error(`\u{1F4C4} [Agent] Invalid JSON written to file:
|
|
66
|
+
${JSON.stringify(S,null,2).slice(0,500)}`),!p)throw new Error(`Agent output failed schema validation:
|
|
67
|
+
${C}
|
|
68
|
+
|
|
69
|
+
The agent wrote invalid data that doesn't match the required outputSchema.
|
|
70
|
+
Enable strictMode in .zibby.config.js for automatic repair.`,{cause:m})}else{if(S)return u.info("\u2705 [Agent] File-based output extracted (no Zod parse fn) \u2014 accepting as structured"),p&&u.debug("[Agent] strictMode enabled but not needed \u2014 agent wrote valid file"),{raw:k,structured:S};A&&u.error("\u274C [Agent] Result file exists but produced no in-memory JSON (parse failed earlier)")}if(p&&!E){let m=T.parsedText,C=S?JSON.stringify(S):m;u.info(`[Agent] strictMode: calling OpenAI proxy to fix structured output (${C.length} chars in)`);try{let P=await Ft(C,n);if(O){let L=n.parse(P.structured);return u.info("\u2705 [Agent] Proxy output passed Zod validation"),{raw:k,structured:L}}return{raw:k,...P}}catch(P){if(u.warn(`\u26A0\uFE0F [Agent] strictMode proxy failed: ${P.message}`),S)return u.warn("[Agent] Using agent's original result file as fallback"),{raw:k,structured:S}}}if(E)throw E;let N=A?S==null?"file existed but JSON.parse failed \u2014 see WARN log above":O?"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 u.error(`\u274C [Agent] No validated structured output: ${N}`),u.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 ${w}. Enable strictMode for proxy fallback.`)}if(E)throw E;return this._extractFinalResult(k)||T?.parsedText||k}_extractFinalResult(e){if(!e)return null;let t=e.split(`
|
|
64
71
|
`),o=null;for(let r of t){let n=r.trim();if(n)try{let i=JSON.parse(n);if(i.type==="assistant"&&i.message?.content){let a=i.message.content;if(Array.isArray(a)){let l=a.filter(c=>c.type==="text"&&c.text).map(c=>c.text).join("");l&&(o=l)}else typeof a=="string"&&a&&(o=a)}}catch{}}return o?.trim()||null}_setupMcpConfig(e,t,o,r=null,n=null){let i=o?.headless,a=W(xe(),".cursor"),l=W(a,"mcp.json"),c={};if($e(l))try{c=JSON.parse(Ht(l,"utf-8"))}catch{}let d=c.mcpServers||{},p=o?.paths?.output||ie,f=W(t||process.cwd(),p,q),_=Array.isArray(r)?r.map(y=>j(y)).filter(Boolean):[...Ue()].map(([,y])=>y),g=new Set;for(let y of _)typeof y.resolve=="function"&&(g.has(y.serverName)||(g.add(y.serverName),this._ensureSkillConfigured(d,y,e,f,n,i)));if(e){let y=j("browser");y&&typeof y.resolve=="function"&&!g.has(y.serverName)&&this._ensureSkillConfigured(d,y,e,f,"execute_live",i)}if(Object.keys(d).length===0)return u.debug("[MCP] No MCP servers configured - agent will run without tool access"),{isolatedMcpHome:null};let b=`${JSON.stringify({mcpServers:d},null,2)}
|
|
65
|
-
`;if(Kt(e)){let y=Yt(t||process.cwd()),w=W(y,".cursor","mcp.json");return Zt(w,b,"utf8"),u.debug(`[MCP] Isolated cursor-agent HOME (session-scoped mcp.json): ${y} | servers: ${Object.keys(d).join(", ")}`),{isolatedMcpHome:y}}return $e(a)||Vt(a,{recursive:!0}),Zt(l,b,"utf8"),u.debug(`[MCP] Global ~/.cursor/mcp.json | servers: ${Object.keys(d).join(", ")}`),{isolatedMcpHome:null}}_ensureSkillConfigured(e,t,o,r,n=null,i){let a=t.cursorKey||t.serverName,l=e[a]?a:e[t.serverName]?t.serverName:null;if(l&&o){let d=typeof t.resolve=="function"?t.resolve({sessionPath:o,nodeName:n,headless:i}):null;d?.args?e[l].args=d.args:e[l].args=(e[l].args||[]).map(_=>_.startsWith("--output-dir=")?`--output-dir=${o}`:_);let p=d?.env||{},f=t.sessionEnvKey?{[t.sessionEnvKey]:r}:{};e[l].env={...e[l].env||{},...p,...f},u.debug(`[MCP] Updated ${l} session \u2192 ${o}`);return}if(l)return;let c=t.resolve({sessionPath:o,nodeName:n,headless:i});c&&(e[a]={...c,...t.sessionEnvKey&&{env:{...c.env||{},[t.sessionEnvKey]:r}}},u.debug(`[MCP] Configured ${a}`))}_spawnWithStreaming(e,t,o,r,n=null,i=null,a=null){return new Promise((l,c)=>{let d=Date.now(),p="",f="",_=Date.now(),g=0,b=!1,y=null,w=!1,h=!1,$=null;if(i)try{$=W(ir(String(i)),Le)}catch{$=null}let x=!1,v=()=>{x||(x=!0,Wt(a))},T={...process.env};a&&(T.HOME=a,process.platform==="win32"&&(T.USERPROFILE=a),u.debug(`[Agent] cursor-agent HOME=${a} (isolated MCP config)`));let E=sr(e,t,{cwd:o,shell:!1,stdio:["pipe","pipe","pipe"],env:T});u.debug(`[Agent] PID: ${E.pid}`),E.stdin.on("error",m=>{m.code!=="EPIPE"&&u.warn(`[Agent] stdin error: ${m.message}`)}),E.stdout.on("error",m=>{m.code!=="EPIPE"&&u.warn(`[Agent] stdout error: ${m.message}`)}),E.stderr.on("error",m=>{m.code!=="EPIPE"&&u.warn(`[Agent] stderr error: ${m.message}`)}),n?(E.stdin.write(n,m=>{m&&m.code!=="EPIPE"&&u.warn(`[Agent] Failed to write to stdin: ${m.message}`),E.stdin.end()}),u.debug(`[Agent] Prompt also piped to stdin (${n.length} chars)`)):E.stdin.end();let
|
|
72
|
+
`;if(Kt(e)){let y=Yt(t||process.cwd()),w=W(y,".cursor","mcp.json");return Zt(w,b,"utf8"),u.debug(`[MCP] Isolated cursor-agent HOME (session-scoped mcp.json): ${y} | servers: ${Object.keys(d).join(", ")}`),{isolatedMcpHome:y}}return $e(a)||Vt(a,{recursive:!0}),Zt(l,b,"utf8"),u.debug(`[MCP] Global ~/.cursor/mcp.json | servers: ${Object.keys(d).join(", ")}`),{isolatedMcpHome:null}}_ensureSkillConfigured(e,t,o,r,n=null,i){let a=t.cursorKey||t.serverName,l=e[a]?a:e[t.serverName]?t.serverName:null;if(l&&o){let d=typeof t.resolve=="function"?t.resolve({sessionPath:o,nodeName:n,headless:i}):null;d?.args?e[l].args=d.args:e[l].args=(e[l].args||[]).map(_=>_.startsWith("--output-dir=")?`--output-dir=${o}`:_);let p=d?.env||{},f=t.sessionEnvKey?{[t.sessionEnvKey]:r}:{};e[l].env={...e[l].env||{},...p,...f},u.debug(`[MCP] Updated ${l} session \u2192 ${o}`);return}if(l)return;let c=t.resolve({sessionPath:o,nodeName:n,headless:i});c&&(e[a]={...c,...t.sessionEnvKey&&{env:{...c.env||{},[t.sessionEnvKey]:r}}},u.debug(`[MCP] Configured ${a}`))}_spawnWithStreaming(e,t,o,r,n=null,i=null,a=null){return new Promise((l,c)=>{let d=Date.now(),p="",f="",_=Date.now(),g=0,b=!1,y=null,w=!1,h=!1,$=null;if(i)try{$=W(ir(String(i)),Le)}catch{$=null}let x=!1,v=()=>{x||(x=!0,Wt(a))},T={...process.env};a&&(T.HOME=a,process.platform==="win32"&&(T.USERPROFILE=a),u.debug(`[Agent] cursor-agent HOME=${a} (isolated MCP config)`));let E=sr(e,t,{cwd:o,shell:!1,stdio:["pipe","pipe","pipe"],env:T});u.debug(`[Agent] PID: ${E.pid}`),E.stdin.on("error",m=>{m.code!=="EPIPE"&&u.warn(`[Agent] stdin error: ${m.message}`)}),E.stdout.on("error",m=>{m.code!=="EPIPE"&&u.warn(`[Agent] stdout error: ${m.message}`)}),E.stderr.on("error",m=>{m.code!=="EPIPE"&&u.warn(`[Agent] stderr error: ${m.message}`)}),n?(E.stdin.write(n,m=>{m&&m.code!=="EPIPE"&&u.warn(`[Agent] Failed to write to stdin: ${m.message}`),E.stdin.end()}),u.debug(`[Agent] Prompt also piped to stdin (${n.length} chars)`)):E.stdin.end();let k=null;$&&(k=setInterval(()=>{if(!(b||h))try{if($e($)){b=!0,y="studio-stop";try{nr($)}catch{}u.warn("\u{1F6D1} Studio stop requested \u2014 terminating Cursor agent (and MCP browser session)"),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},2e3)}}catch{}},600));let I=new Set,O=new Date(d).toISOString().replace(/\.\d+Z$/,""),S=setInterval(()=>{let m=Math.round((Date.now()-d)/1e3),C=Math.round((Date.now()-_)/1e3),P=[];try{let G=Math.ceil(m/60)+1,M=te(`find "${o}" -type f -mmin -${G} -not -path '*/node_modules/*' -not -path '*/.git/*' -not -path '*/target/*' 2>/dev/null | head -20`,{encoding:"utf-8",timeout:5e3}).trim();if(M)for(let K of M.split(`
|
|
66
73
|
`)){let B=K.replace(`${o}/`,"");I.has(B)||(I.add(B),P.push(B))}}catch{}let L="";P.length>0&&(L=` | \u{1F4C1} new: ${P.map(M=>M.split("/").pop()).join(", ")}`),I.size>0&&(L+=` | \u{1F4E6} total: ${I.size} files`),u.debug(`\u{1F493} [Agent] Running for ${m}s | ${g} lines output${L}`),g===0&&m>=30&&I.size===0&&(m<35&&u.warn(`\u26A0\uFE0F [Agent] No output after ${m}s \u2014 agent may be stuck. Check your CURSOR_API_KEY.`),m>=60&&(b=!0,y=y||"stall",u.error(`\u274C [Agent] No response after ${m}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),A=setTimeout(()=>{b=!0,y=y||"timeout";let m=Math.round((Date.now()-d)/1e3);u.error(`\u23F1\uFE0F [Agent] Timeout after ${m}s \u2014 killing process (PID: ${E.pid})`),p.trim()&&u.warn(`\u{1F4E4} [Agent] Partial output (${p.length} chars) before timeout:
|
|
67
|
-
${p.slice(-2e3)}`),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},5e3)},r),
|
|
68
|
-
`).filter(G=>G.trim());g+=L.length}),E.stderr.on("data",m=>{let
|
|
69
|
-
`).filter(L=>L.trim());for(let L of P)u.warn(`\u26A0\uFE0F [Agent stderr] ${L}`)}),E.on("close",(m,
|
|
74
|
+
${p.slice(-2e3)}`),E.kill("SIGTERM"),setTimeout(()=>{E.killed||E.kill("SIGKILL")},5e3)},r),N=new ae;N.onToolCall=(m,C)=>{let P=m,L=C;if(m==="mcpToolCall"&&C?.name)P=C.name.replace(/^mcp_+[^_]+_+/,""),P.includes("-")&&P.split("-")[0]===P.split("-")[1]&&(P=P.split("-")[0]),L=C.args??C.input??C;else{if(m==="readToolCall"||m==="editToolCall"||m==="writeToolCall")return;(m.startsWith("mcp__")||m.includes("ToolCall"))&&(P=m.replace(/^mcp_+[^_]+_+/,"").replace(/ToolCall$/,""))}if(P.includes("memory")?D.stepMemory(`Tool: ${P}`):D.stepTool(`Tool: ${P}`),L!=null&&typeof L=="object"&&Object.keys(L).length>0&&!h){let M=JSON.stringify(L),K=M.length>100?`${M.substring(0,100)}...`:M;console.log(` Input: ${K}`)}},E.stdout.on("data",m=>{let C=m.toString();p+=C,_=Date.now(),w||(w=!0);let P=N.processChunk(C);P&&!h&&process.stdout.write(P);let L=C.split(`
|
|
75
|
+
`).filter(G=>G.trim());g+=L.length}),E.stderr.on("data",m=>{let C=m.toString();f+=C,_=Date.now(),w||(w=!0);let P=C.split(`
|
|
76
|
+
`).filter(L=>L.trim());for(let L of P)u.warn(`\u26A0\uFE0F [Agent stderr] ${L}`)}),E.on("close",(m,C)=>{h=!0,v(),clearTimeout(A),clearInterval(S),k&&clearInterval(k),N.flush();let P=Math.round((Date.now()-d)/1e3);if(u.debug(`[Agent] Exited: code=${m}, signal=${C}, elapsed=${P}s, output=${p.length} chars`),b){if(y==="studio-stop"){c(new Error("Stopped from Zibby Studio"));return}c(new Error(`Cursor Agent timed out after ${P}s (limit: ${r/1e3}s). ${g} lines produced. Last output ${Math.round((Date.now()-_)/1e3)}s ago. ${p.trim()?`
|
|
70
77
|
Partial output (last 500 chars):
|
|
71
|
-
${p.slice(-500)}`:"No output captured."}`));return}if(m!==0){c(new Error(`Cursor Agent failed: exit code ${m}, signal ${
|
|
78
|
+
${p.slice(-500)}`:"No output captured."}`));return}if(m!==0){c(new Error(`Cursor Agent failed: exit code ${m}, signal ${C}. ${f.trim()?`
|
|
72
79
|
Stderr: ${f.slice(-1e3)}`:""}${p.trim()?`
|
|
73
|
-
Stdout (last 500 chars): ${p.slice(-500)}`:""}`));return}let L=
|
|
80
|
+
Stdout (last 500 chars): ${p.slice(-500)}`:""}`));return}let L=N.getResult(),G=L?JSON.stringify(L,null,2):N.getRawText()||p||"";l({stdout:p||f||"",parsedText:G})}),E.on("error",m=>{v(),clearTimeout(A),clearInterval(S),k&&clearInterval(k),c(new Error(`Cursor Agent spawn error: ${m.message}
|
|
74
81
|
Binary: ${e}
|
|
75
82
|
This usually means the binary is not in PATH. Try:
|
|
76
83
|
echo 'export PATH="$HOME/.local/bin:$PATH"' >> ~/.zshrc && source ~/.zshrc`))})})}}});import{query as ar}from"@anthropic-ai/claude-agent-sdk";import{zodToJsonSchema as lr}from"zod-to-json-schema";var Te,eo=R(()=>{Q();Y();ne();ee();H();Te=class extends F{constructor(){super("claude","Claude (Anthropic API)",50)}canHandle(e){let t=!!process.env.ANTHROPIC_API_KEY;return t||u.debug("ClaudeAgentStrategy: ANTHROPIC_API_KEY not set"),t}async invoke(e,t={}){let{model:o,workspace:r=process.cwd(),schema:n=null,images:i=[],skills:a=null,sessionPath:l=null,nodeName:c=null,timeout:d,config:p={}}=t,f=o;(!f||f==="auto")&&(u.debug(`Model is '${f||"undefined"}', using default: ${J.CLAUDE}`),f=J.CLAUDE);let _=nt[f]||f;nt[f]&&f!==_&&u.debug(`Mapped model: ${f} \u2192 ${_}`),u.debug(`Invoking Claude Agent SDK with model: ${_}, skills: ${JSON.stringify(a)}`);let g=process.env.ANTHROPIC_API_KEY,b=g?` | key: ***${g.slice(-4)}`:" | key: not set";console.log(`
|
|
77
84
|
\u25C6 Model: ${_}${b}
|
|
78
85
|
`);let y=(await import("chalk")).default;console.log(`
|
|
79
|
-
${y.bold("Prompt sent to LLM:")}`),console.log(y.dim("\u2500".repeat(60))),console.log(y.dim(e)),console.log(y.dim("\u2500".repeat(60)));let{allowedTools:w,mcpServers:h}=this._resolveSkills(a,{sessionPath:l,workspace:r,nodeName:c});try{let $={cwd:r,allowedTools:w,permissionMode:"bypassPermissions",model:_,...Object.keys(h).length>0&&{mcpServers:h}};if(n){let A=typeof n.parse=="function"?lr(n,{target:"openApi3"}):n;$.outputFormat={type:"json_schema",schema:A},u.debug("Structured output enforced via SDK outputFormat")}u.debug(`Agent SDK options: ${JSON.stringify({cwd:$.cwd,toolCount:w.length,permissionMode:$.permissionMode,model:$.model,hasOutputFormat:!!$.outputFormat})}`);let x="",v=0,T=[];u.debug("Starting Claude Agent SDK query stream");let E;try{E=ar({prompt:e,options:$})}catch(S){throw u.error(`Failed to initialize Claude Agent SDK: ${S.message}`),S}let
|
|
86
|
+
${y.bold("Prompt sent to LLM:")}`),console.log(y.dim("\u2500".repeat(60))),console.log(y.dim(e)),console.log(y.dim("\u2500".repeat(60)));let{allowedTools:w,mcpServers:h}=this._resolveSkills(a,{sessionPath:l,workspace:r,nodeName:c});try{let $={cwd:r,allowedTools:w,permissionMode:"bypassPermissions",model:_,...Object.keys(h).length>0&&{mcpServers:h}};if(n){let A=typeof n.parse=="function"?lr(n,{target:"openApi3"}):n;$.outputFormat={type:"json_schema",schema:A},u.debug("Structured output enforced via SDK outputFormat")}u.debug(`Agent SDK options: ${JSON.stringify({cwd:$.cwd,toolCount:w.length,permissionMode:$.permissionMode,model:$.model,hasOutputFormat:!!$.outputFormat})}`);let x="",v=0,T=[];u.debug("Starting Claude Agent SDK query stream");let E;try{E=ar({prompt:e,options:$})}catch(S){throw u.error(`Failed to initialize Claude Agent SDK: ${S.message}`),S}let k=null,I=0,O=3;try{for await(let S of E){if(T.push(S),S.type==="error"||S.error){let N=S.error?.message||S.error||S.message||"Unknown API error";throw new Error(typeof N=="string"?N:JSON.stringify(N))}let A=JSON.stringify(S.message?.content||S.text||"").slice(0,200);if(A===k){if(I++,I>=O){let N=(S.message?.content?.[0]?.text||S.text||"unknown").slice(0,100);throw new Error(`API stuck in loop (${I}x repeated): ${N}`)}}else k=A,I=1;if(S.type==="assistant"||S.constructor?.name==="AssistantMessage"){let N=S.message?.content||S.content||[];for(let m of N)if(m.type==="thinking"&&m.thinking)console.log(`${m.thinking.substring(0,200)}${m.thinking.length>200?"...":""}`);else if(m.type==="text"&&m.text)x+=m.text,m.text.length<500?console.log(`${m.text}`):console.log(`${m.text.substring(0,200)}... (${m.text.length} chars)`);else if(m.type==="tool_use"){v++,m.name.includes("memory")?D.stepMemory(`Tool: ${m.name}`):D.stepTool(`Tool: ${m.name}`);let P=JSON.stringify(m.input).substring(0,100);console.log(` Input: ${P}${JSON.stringify(m.input).length>100?"...":""}`)}}else if(!(S.type==="user"&&S.tool_use_result)){if(S.type==="result"||S.constructor?.name==="ResultMessage"){let N=S.result||S.text||S.content||x;if(n){if(S.structured_output){u.debug("Using SDK native structured_output");let C=typeof n.parse=="function"?n.parse(S.structured_output):S.structured_output;return{raw:N,structured:C}}if(N){let m=this._extractJson(N,n);if(m)return{raw:N,structured:m}}u.warn(`Could not extract structured output \u2014 returning raw text (${(N||"").length} chars)`)}return N||""}}}if(u.warn(`Agent SDK ended without result. Collected ${T.length} messages`),x.length>0)return u.debug("Returning accumulated text from messages"),x;throw new Error("Claude Agent SDK query ended without result")}catch(S){throw u.error(`Error during query stream: ${S.message}`),S}}catch($){throw u.error("Claude Agent SDK call failed",{error:$.message}),$}}_resolveSkills(e,t){if(e===null)return u.debug("No skills \u2014 pure LLM mode"),{allowedTools:[],mcpServers:{}};if(!Array.isArray(e)||e.length===0)return u.debug("Default IDE skills for code generation"),{allowedTools:["Read","Write","Bash","Grep","Glob"],mcpServers:{}};let o=[],r={};for(let n of e){let i=j(n);if(!i){u.warn(`Unknown skill "${n}" \u2014 skipping`);continue}if(i.allowedTools&&o.push(...i.allowedTools),typeof i.resolve=="function"){let a=i.resolve(t);a&&(r[i.serverName]=a,u.debug(`MCP: ${i.serverName} \u2192 ${a.command} ${a.args[0]}`))}}return{allowedTools:o,mcpServers:r}}_extractJson(e,t){let o=[()=>{if(e.includes("===JSON_START===")){let r=e.indexOf("===JSON_START===")+16,n=e.indexOf("===JSON_END===");return e.substring(r,n).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 r=e.indexOf("{"),n=e.lastIndexOf("}");if(r!==-1&&n>r)return e.substring(r,n+1)}];for(let r of o)try{let n=r();if(!n)continue;let i=JSON.parse(n);if(typeof i!="object"||i===null)continue;return typeof t.parse=="function"?t.parse(i):i}catch{}return null}}});import{execSync as cr}from"node:child_process";import{zodToJsonSchema as ur}from"zod-to-json-schema";var Oe,to=R(()=>{Q();Y();ne();ee();H();Oe=class extends F{constructor(){super("codex","Codex (OpenAI)",75)}canHandle(e){if(!!!(process.env.OPENAI_API_KEY||process.env.CODEX_API_KEY))return u.debug("CodexAgentStrategy: OPENAI_API_KEY or CODEX_API_KEY not set"),!1;try{return cr("codex --version",{encoding:"utf-8",timeout:5e3,stdio:"pipe"}),!0}catch{return u.warn("[Codex] codex CLI not found. Install: npm install -g @openai/codex"),!1}}async invoke(e,t={}){let{model:o,workspace:r=process.cwd(),schema:n=null,skills:i=null,sessionPath:a=null,nodeName:l=null,timeout:c,config:d={}}=t,{Codex:p}=await import("@openai/codex-sdk"),f=o;(!f||f==="auto")&&(u.debug(`Model is '${f||"undefined"}', using default: ${J.CODEX}`),f=J.CODEX);let _=it[f]||f;it[f]&&f!==_&&u.debug(`Mapped model: ${f} \u2192 ${_}`),u.debug(`Invoking Codex SDK with model: ${_}, skills: ${JSON.stringify(i)}`);let g=process.env.CODEX_API_KEY||process.env.OPENAI_API_KEY;g&&!process.env.CODEX_API_KEY&&(process.env.CODEX_API_KEY=g);let b=g?` | key: ***${g.slice(-4)}`:" | key: not set";console.log(`
|
|
80
87
|
\u25C6 Model: ${_}${b}
|
|
81
88
|
`);let y=(await import("chalk")).default;console.log(`
|
|
82
|
-
${y.bold("Prompt sent to LLM:")}`),console.log(y.dim("\u2500".repeat(60))),console.log(y.dim(e)),console.log(y.dim("\u2500".repeat(60)));let w=this._resolveSkillsToMcp(i,{sessionPath:a,workspace:r,nodeName:l}),h={};Object.keys(w).length>0&&(h.mcp_servers=w,u.debug(`[Codex] MCP servers: ${Object.keys(w).join(", ")}`));let x=new p({...Object.keys(h).length>0&&{config:h}}).startThread({workingDirectory:r,skipGitRepoCheck:!0,approvalPolicy:"never",sandboxMode:"danger-full-access",networkAccessEnabled:!0}),v=n&&typeof n.parse=="function",T={};if(n)try{let E=v?ur(n,{target:"openAi"}):n;T.outputSchema=E,u.debug("Structured output via SDK outputSchema")}catch(E){u.warn(`[Codex] Schema conversion failed, will extract from text: ${E.message}`)}try{let{events:E}=await x.runStreamed(e,T),
|
|
89
|
+
${y.bold("Prompt sent to LLM:")}`),console.log(y.dim("\u2500".repeat(60))),console.log(y.dim(e)),console.log(y.dim("\u2500".repeat(60)));let w=this._resolveSkillsToMcp(i,{sessionPath:a,workspace:r,nodeName:l}),h={};Object.keys(w).length>0&&(h.mcp_servers=w,u.debug(`[Codex] MCP servers: ${Object.keys(w).join(", ")}`));let x=new p({...Object.keys(h).length>0&&{config:h}}).startThread({workingDirectory:r,skipGitRepoCheck:!0,approvalPolicy:"never",sandboxMode:"danger-full-access",networkAccessEnabled:!0}),v=n&&typeof n.parse=="function",T={};if(n)try{let E=v?ur(n,{target:"openAi"}):n;T.outputSchema=E,u.debug("Structured output via SDK outputSchema")}catch(E){u.warn(`[Codex] Schema conversion failed, will extract from text: ${E.message}`)}try{let{events:E}=await x.runStreamed(e,T),k=0,I="";for await(let O of E){let S=O.type;if(S==="item.completed"){let A=O.item,N=A?.type;if(N==="mcp_tool_call"){k++;let m=`${A.server}/${A.tool}`;if(D.stepTool(`Tool: ${m}`),A.arguments){let C=JSON.stringify(A.arguments),P=C.length>100?`${C.substring(0,100)}...`:C;console.log(` Input: ${P}`)}}else if(N==="tool_call"||N==="function_call"||N==="command_execution"){k++;let m=A.name||A.tool||A.command||"unknown";D.stepTool(`Tool: ${m}`)}else N==="agent_message"&&(I=A.text||"",I.length<500?console.log(I):console.log(`${I.substring(0,200)}... (${I.length} chars)`))}else S==="turn.completed"?u.debug(`[Codex] Turn completed. Usage: ${JSON.stringify(O.usage||{})}`):u.debug(`[Codex] Event: ${S} ${JSON.stringify(O).slice(0,300)}`)}if(u.debug(`[Codex] Last agent message (${I.length} chars): ${I.slice(0,500)}`),n){if(!I)throw new Error("Codex agent returned no response");let O=JSON.parse(I),S=v?n.parse(O):O;return u.debug("\u2705 [Codex] Structured output validated"),{raw:I,structured:S}}return I||""}catch(E){let k=E.message||String(E);throw u.error(`\u274C [Codex] SDK call failed: ${k}`),k.includes("exited with code")&&(u.error("\u{1F4A1} [Codex] Verify: codex --version && echo $OPENAI_API_KEY"),u.error("\u{1F4A1} [Codex] If codex is missing: npm install -g @openai/codex")),E}}_resolveSkillsToMcp(e,t={}){if(!Array.isArray(e)||e.length===0)return{};let o={};for(let r of e){let n=j(r);if(!n){u.warn(`[Codex] Unknown skill "${r}" \u2014 skipping`);continue}if(typeof n.resolve!="function")continue;let i=n.resolve(t);if(!i)continue;let a=n.serverName||r,l={command:i.command};i.args?.length&&(l.args=i.args),i.env&&Object.keys(i.env).length>0&&(l.env=i.env),o[a]=l,u.debug(`[Codex] MCP: ${a} \u2192 ${i.command} ${(i.args||[]).join(" ")}`)}return o}}});import{execSync as dr,spawn as fr}from"node:child_process";import{zodToJsonSchema as pr}from"zod-to-json-schema";import{existsSync as oo,mkdirSync as ro,readFileSync as so,rmSync as mr,writeFileSync as no}from"node:fs";import{join as oe}from"node:path";function gr(s){if(!s)return null;let e=String(s),t=e.match(/```(?:json)?\s*([\s\S]*?)```/i);if(t?.[1])try{return JSON.parse(t[1].trim())}catch{}let o=e.indexOf("{");if(o<0)return null;let r=0,n=!1,i=!1,a=-1;for(let l=o;l<e.length;l++){let c=e[l];if(n){i?i=!1:c==="\\"?i=!0:c==='"'&&(n=!1);continue}if(c==='"'){n=!0;continue}if(c==="{"){r===0&&(a=l),r+=1;continue}if(c==="}"){if(r===0)continue;if(r-=1,r===0&&a>=0){let d=e.slice(a,l+1);try{return JSON.parse(d)}catch{a=-1}}}}return null}function hr(s){let e=String(s||"").trim();if(!e)return null;try{return JSON.parse(e)}catch{return gr(e)}}function yr(s){try{let e=JSON.parse(s);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 o=t.content.parts.map(r=>typeof r?.text=="string"?r.text:"").join("");if(o.trim())return o}}}catch{}return s}var Ae,io=R(()=>{Q();Y();ee();H();ct();lt();Ae=class extends F{constructor(){super("gemini","Gemini (Google)",70)}canHandle(e){if(!!!(process.env.GEMINI_API_KEY||process.env.GOOGLE_API_KEY))return u.debug("GeminiAgentStrategy: GEMINI_API_KEY or GOOGLE_API_KEY not set"),!1;try{return dr("gemini --version",{encoding:"utf-8",timeout:5e3,stdio:"pipe"}),!0}catch{return u.warn("[Gemini] gemini CLI not found. Install: npm install -g @google/gemini-cli"),!1}}async invoke(e,t={}){let{model:o,workspace:r=process.cwd(),schema:n=null,skills:i=null,sessionPath:a=null,nodeName:l=null,timeout:c=600*1e3}=t,d=o;(!d||d==="auto")&&(d=J.GEMINI);let p=Lt[d]||d,f=String(process.env.GEMINI_API_KEY||"").trim(),_=String(process.env.GOOGLE_API_KEY||"").trim(),g=this._resolveSkillsToMcp(i,{sessionPath:a,workspace:r,nodeName:l}),b=Object.keys(g).length>0,y=e,w=n&&typeof n.parse=="function",h=null;if(n){let A;try{let N=w?pr(n,{target:"openAi"}):n;A=JSON.stringify(N,null,2)}catch{A="{}"}if(b){y+=`
|
|
83
90
|
|
|
84
91
|
Write valid JSON that matches this schema:
|
|
85
|
-
${A}`;let
|
|
92
|
+
${A}`;let N=`zibby-result-${Date.now()}.json`,m=oe(r,".zibby","tmp");h=oe(m,N),ro(m,{recursive:!0}),y+=le.generateFileOutputInstructions(n,h)}else y+=`
|
|
86
93
|
|
|
87
94
|
Return ONLY valid JSON (no markdown, no commentary) that matches this schema:
|
|
88
|
-
${A}`}let $=this._createGeminiConfigDir(r,g),x=["--output-format","json"];p&&p!=="auto"&&x.push("--model",p);let v=Object.keys(g);if(v.length>0){x.push("--approval-mode","yolo");for(let A of v)x.push("--allowed-mcp-server-names",A);u.info(`[Gemini] Enabling MCP servers: ${v.join(", ")}`)}else i&&i.length>0&&u.warn(`[Gemini] Skills requested but no MCP servers configured: ${i.join(", ")}`);x.push("-p",y);let T={...process.env,GEMINI_CLI_HOME:$};f?(T.GEMINI_API_KEY=f,delete T.GOOGLE_API_KEY):_&&(T.GOOGLE_API_KEY=_,delete T.GEMINI_API_KEY),u.debug(`[Gemini] Command: gemini ${x.slice(0,8).join(" ")}... (${x.length} total args)`),u.debug(`[Gemini] Config home: ${$}`),u.debug(`[Gemini] GEMINI_CLI_HOME env: ${T.GEMINI_CLI_HOME}`);let E="",
|
|
95
|
+
${A}`}let $=this._createGeminiConfigDir(r,g),x=["--output-format","json"];p&&p!=="auto"&&x.push("--model",p);let v=Object.keys(g);if(v.length>0){x.push("--approval-mode","yolo");for(let A of v)x.push("--allowed-mcp-server-names",A);u.info(`[Gemini] Enabling MCP servers: ${v.join(", ")}`)}else i&&i.length>0&&u.warn(`[Gemini] Skills requested but no MCP servers configured: ${i.join(", ")}`);x.push("-p",y);let T={...process.env,GEMINI_CLI_HOME:$};f?(T.GEMINI_API_KEY=f,delete T.GOOGLE_API_KEY):_&&(T.GOOGLE_API_KEY=_,delete T.GEMINI_API_KEY),u.debug(`[Gemini] Command: gemini ${x.slice(0,8).join(" ")}... (${x.length} total args)`),u.debug(`[Gemini] Config home: ${$}`),u.debug(`[Gemini] GEMINI_CLI_HOME env: ${T.GEMINI_CLI_HOME}`);let E="",k=null;try{E=await new Promise((N,m)=>{let C=fr("gemini",x,{cwd:r,env:T,stdio:["ignore","pipe","pipe"]}),P="",L="",G=setTimeout(()=>{try{C.kill("SIGTERM")}catch{}},c);C.stdout.on("data",M=>{P+=M.toString()}),C.stderr.on("data",M=>{L+=M.toString()}),C.on("error",M=>{clearTimeout(G),m(M)}),C.on("close",M=>{if(clearTimeout(G),M===0)return N(P.trim());m(new Error(`gemini failed with code ${M}: ${(L||P).trim()}`))})})}catch(A){k=A}finally{try{mr($,{recursive:!0,force:!0})}catch{}}let I=yr(E).trim();if(!n){if(k)throw k;return I}if(h){let A=oo(h);if(u.info(`[Gemini] Result file: ${A?"present":"missing"} at ${h}`),A)try{let N=so(h,"utf-8").trim(),m=JSON.parse(N),C=w?n.parse(m):m;return u.info("[Gemini] Structured output recovered from result file"),{raw:I,structured:C}}catch(N){u.warn(`[Gemini] Result file parse/validation failed: ${N.message}`)}else k||u.warn("[Gemini] Result file missing; falling back to stream-parsed JSON")}let O=null;if(n){let A=new ae;A.zodSchema=n,A.processChunk(I),A.flush(),O=A.getResult()}if(u.info(`[Gemini] Raw stdout length: ${E.length} chars`),u.info(`[Gemini] Extracted text length: ${I.length} chars`),u.info(`[Gemini] StreamParser result: ${O?"extracted":"null"}`),O||(I.length<2e3?u.info(`[Gemini] Raw text preview:
|
|
89
96
|
${I}`):u.info(`[Gemini] Raw text preview (first 1000 chars):
|
|
90
|
-
${I.slice(0,1e3)}`),O=hr(I)),!O)throw
|
|
97
|
+
${I.slice(0,1e3)}`),O=hr(I)),!O)throw k||(u.error("[Gemini] Failed to extract valid JSON from output"),u.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 S=w?n.parse(O):O;return{raw:I,structured:S}}_resolveSkillsToMcp(e,t={}){if(!Array.isArray(e)||e.length===0)return{};let o={};for(let r of e){let n=j(r);if(!n||typeof n.resolve!="function")continue;let i=n.resolve(t);if(!i)continue;let a=n.cursorKey||n.serverName||r,l={command:i.command};i.args?.length&&(l.args=i.args),i.env&&Object.keys(i.env).length>0&&(l.env=i.env),i.cwd&&(l.cwd=i.cwd),o[a]=l}return o}_createGeminiConfigDir(e,t){let o=`${Date.now()}-${Math.random().toString(16).slice(2,10)}`,r=oe(e||process.cwd(),".zibby","tmp",`gemini-home-${o}`),n=oe(r,".gemini");ro(n,{recursive:!0});let i=oe(n,"settings.json"),a={},l=oe(process.env.HOME||"",".gemini","settings.json");if(oo(l))try{a=JSON.parse(so(l,"utf-8"))}catch{a={}}let c={...a,mcpServers:{...a.mcpServers&&typeof a.mcpServers=="object"?a.mcpServers:{},...t||{}}};no(i,`${JSON.stringify(c,null,2)}
|
|
91
98
|
`,"utf-8");let d=oe(e||process.cwd(),".zibby","tmp","gemini-settings-debug.json");try{no(d,`${JSON.stringify(c,null,2)}
|
|
92
99
|
`,"utf-8")}catch{}return u.debug(`[Gemini] Created isolated config with ${Object.keys(c.mcpServers||{}).length} MCP servers`),u.debug(`[Gemini] MCP servers: ${JSON.stringify(Object.keys(c.mcpServers||{}),null,2)}`),r}}});var ve,dt=R(()=>{ve=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 ue,ao=R(()=>{dt();ue=class extends ve{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(o=>({id:o.id,name:o.function.name,args:JSON.parse(o.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 Ie,ft=R(()=>{Ie=class{async fetchCompletion(e,t,o={}){throw new Error("fetchCompletion() must be implemented")}async fetchStreamingCompletion(e,t,o={}){throw new Error("fetchStreamingCompletion() must be implemented")}}});function Fe(s){return Buffer.byteLength(JSON.stringify(s),"utf8")}function Je(s,e){let t=String(s||"");if(t.length<=e)return t;let o=Math.max(0,e-28);return`${t.slice(0,o)}
|
|
93
100
|
|
|
94
101
|
[truncated for size budget]`}function pt(s,e=0){if(!s||typeof s!="object"||e>8)return s;if(Array.isArray(s))return s.map(o=>pt(o,e+1));let t={};for(let[o,r]of Object.entries(s))o==="description"||o==="title"||o==="examples"||o==="default"||(t[o]=pt(r,e+1));return t}function Sr(s=[]){return s.map(e=>({...e,function:{...e.function,description:Je(e.function?.description||"",180),parameters:pt(e.function?.parameters||{type:"object",properties:{}})}}))}function lo(s){let e=new Set;for(let r of s)if(r.role==="assistant"&&Array.isArray(r.tool_calls))for(let n of r.tool_calls)e.add(n.id);let t=s.filter(r=>r.role==="tool"?e.has(r.tool_call_id):!0),o=new Set;for(let r of t)r.role==="tool"&&o.add(r.tool_call_id);return t.map(r=>{if(r.role!=="assistant"||!Array.isArray(r.tool_calls)||r.tool_calls.every(l=>o.has(l.id)))return r;let{tool_calls:i,...a}=r;return{...a,content:a.content||""}})}function mt(s,e={}){let t=e.maxBytes||49e3,o=e.systemMaxChars||12e3,r={...s,messages:Array.isArray(s.messages)?[...s.messages]:[],tools:Array.isArray(s.tools)?Sr(s.tools):s.tools};r.messages.length>0&&r.messages[0]?.role==="system"&&(r.messages[0]={...r.messages[0],content:Je(r.messages[0].content,o)});let n=!1;for(;Fe(r)>t&&r.messages.length>2;)r.messages.splice(1,1),n=!0;if(n&&(r.messages=lo(r.messages)),Fe(r)>t&&r.messages.length>0&&(r.messages[0]={...r.messages[0],content:Je(r.messages[0].content,6e3)},n=!0),Fe(r)>t){let i=r.messages.find(l=>l.role==="system")||r.messages[0],a=r.messages.slice(-2);r.messages=lo([i,...a].filter(Boolean).map((l,c)=>({...l,content:Je(l.content,c===0?4e3:8e3)}))),n=!0}return{body:r,meta:{bytes:Fe(r),trimmed:n,maxBytes:t,messageCount:r.messages.length}}}var co=R(()=>{});var uo,de,fo=R(()=>{ft();co();uo=s=>Buffer.byteLength(JSON.stringify(s),"utf8"),de=class extends Ie{async fetchCompletion(e,t,o={}){let r=uo(e),{body:n,meta:i}=mt(e,o.payloadCompaction);o.onBudget?.({streaming:!1,beforeBytes:r,meta:i});let a=this.#e(o),l=`${t.baseUrl}${o.chatCompletionsPath||"/v1/chat/completions"}`,c=await fetch(l,{method:"POST",headers:t.headers,body:JSON.stringify(n),signal:a});if(!c.ok){let d=await c.text();throw c.status===401||c.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${c.status}: ${d}`)}return c.json()}async fetchStreamingCompletion(e,t,o={}){let r={...e,stream:!0},n=uo(r),{body:i,meta:a}=mt(r,o.payloadCompaction);o.onBudget?.({streaming:!0,beforeBytes:n,meta:a});let l=this.#e(o),c=`${t.baseUrl}${o.chatCompletionsPath||"/v1/chat/completions"}`,d=await fetch(c,{method:"POST",headers:t.headers,body:JSON.stringify(i),signal:l});if(!d.ok){let y=await d.text();throw d.status===401||d.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${d.status}: ${y}`)}let p=d.body.getReader(),f=new TextDecoder,_="",g="",b=new Map;for(;;){let{done:y,value:w}=await p.read();if(y)break;_+=f.decode(w,{stream:!0});let h=_.split(`
|
|
95
|
-
`);_=h.pop();for(let $ of h){if(!$.startsWith("data: "))continue;let x=$.slice(6).trim();if(x==="[DONE]")continue;let v;try{v=JSON.parse(x)}catch{continue}let T=v.choices?.[0]?.delta;if(T&&(T.content&&(g+=T.content,o.onToken&&o.onToken(T.content)),T.tool_calls))for(let E of T.tool_calls){let
|
|
102
|
+
`);_=h.pop();for(let $ of h){if(!$.startsWith("data: "))continue;let x=$.slice(6).trim();if(x==="[DONE]")continue;let v;try{v=JSON.parse(x)}catch{continue}let T=v.choices?.[0]?.delta;if(T&&(T.content&&(g+=T.content,o.onToken&&o.onToken(T.content)),T.tool_calls))for(let E of T.tool_calls){let k=E.index??0;b.has(k)||b.set(k,{id:"",name:"",args:""});let I=b.get(k);E.id&&(I.id=E.id),E.function?.name&&(I.name=E.function.name),E.function?.arguments!=null&&(I.args+=E.function.arguments)}}}if(b.size>0){let y=[...b.entries()].sort(([w],[h])=>w-h).map(([,w])=>({id:w.id,type:"function",function:{name:w.name,arguments:w.args}}));return{choices:[{message:{role:"assistant",content:g||null,tool_calls:y}}]}}return{choices:[{message:{role:"assistant",content:g}}]}}#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}}});var po=R(()=>{dt();ao();ft();fo()});import{Client as _r}from"@modelcontextprotocol/sdk/client/index.js";import{StdioClientTransport as wr}from"@modelcontextprotocol/sdk/client/stdio.js";var Ge,mo=R(()=>{Y();Ge=class{#e=new Map;async ensureServer(e,t){if(this.#e.has(e))return this.#e.get(e);let{command:o,args:r=[],env:n={}}=t;u.debug(`[MCP] Starting ${e}: ${o} ${r.join(" ")}`);let i=new wr({command:o,args:r,env:{...process.env,...n}}),a=new _r({name:`zibby-chat-${e}`,version:"1.0.0"},{capabilities:{}});await a.connect(i);let l={client:a,transport:i,serverConfig:t};return this.#e.set(e,l),l}async callTool(e,t,o={}){let r=this.#e.get(e);if(!r)throw new Error(`MCP server "${e}" not running`);u.debug(`[MCP] ${e}.${t}(${JSON.stringify(o).slice(0,200)})`);let n=await r.client.callTool({name:t,arguments:o});return{text:n.content?.filter(a=>a.type==="text").map(a=>a.text).join(`
|
|
96
103
|
`)||"",isError:n.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(o){u.debug(`[MCP] Error closing ${e}: ${o.message}`)}this.#e.delete(e)}}async stopAll(){let e=[...this.#e.keys()];await Promise.allSettled(e.map(t=>this.stopServer(t)))}}});import{existsSync as Er,readFileSync as $r}from"node:fs";import{join as xr}from"node:path";import{homedir as br}from"node:os";function Tr(){try{let s=xr(br(),".zibby","config.json");return Er(s)?JSON.parse($r(s,"utf-8")):{}}catch{return{}}}function Ke(s){return String(s||"").replace(/\/v1\/?$/,"")}function Or(s,e){let t=process.env.OPENAI_PROXY_URL;if(t)return Ke(t);if(s==="session")return e.proxyUrl?Ke(e.proxyUrl):`${(process.env.ZIBBY_API_URL||"https://api-prod.zibby.app").replace(/\/$/,"")}/openai-proxy`;if(s==="byok"){let o=process.env.OPENAI_BASE_URL;return Ke(o||"https://api.openai.com")}return Ke(t||"")}function gt(){let s=Tr(),e=process.env.ZIBBY_USER_TOKEN||s.sessionToken||null,t=process.env.OPENAI_API_KEY||null,o=(process.env.ASSISTANT_AUTH_MODE||"").trim().toLowerCase(),r=process.env.OPENAI_PROXY_NO_AUTH==="true",n=process.env.OPENAI_PROXY_URL,i="session";o==="byok"||o==="local"||o==="session"?i=o:e?i="session":t?i="byok":n&&r&&(i="local");let a=Or(i,s),l={"Content-Type":"application/json"};if(i==="session"){if(!e)return{ok:!1,mode:i,reason:"missing_session_token"};l.Authorization=`Bearer ${e}`}else if(i==="byok"){if(!t)return{ok:!1,mode:i,reason:"missing_openai_api_key"};l.Authorization=`Bearer ${t}`}else if(i==="local"){if(!a)return{ok:!1,mode:i,reason:"missing_openai_proxy_url"};!r&&t&&(l.Authorization=`Bearer ${t}`)}return a?{ok:!0,mode:i,baseUrl:a,headers:l,tokenPreview:l.Authorization?`***${l.Authorization.slice(-4)}`:"none"}:{ok:!1,mode:i,reason:"missing_base_url"}}var go=R(()=>{});function Ye(s,e){let t=String(s??"");return t.length<=e?t:`${t.slice(0,Math.max(0,e-30))}
|
|
97
104
|
|
|
98
|
-
[tool result truncated for size]`}function Ar(s,e){if(typeof s=="string")return Ye(s,e);try{return Ye(JSON.stringify(s),e)}catch{return Ye(String(s),e)}}function ho(s){let e=new Set;for(let r of s)if(r.role==="assistant"&&Array.isArray(r.tool_calls))for(let n of r.tool_calls)e.add(n.id);let t=s.filter(r=>r.role==="tool"?e.has(r.tool_call_id):!0),o=new Set;for(let r of t)r.role==="tool"&&o.add(r.tool_call_id);return t.map(r=>{if(r.role!=="assistant"||!Array.isArray(r.tool_calls)||r.tool_calls.every(l=>o.has(l.id)))return r;let{tool_calls:i,...a}=r;return{...a,content:a.content||""}})}function vr(s){let e=Array.isArray(s?.messages)?s.messages:[],t=e.find(n=>n.role==="system"),o=e.slice(-4).map(n=>({...n,content:Ye(n.content,n.role==="tool"?1200:2500)}));o=ho(o);let r={...s,messages:[t,...o].filter(Boolean)};return delete r.tools,r}async function Ir({body:s,streaming:e,auth:t,options:o,fetchCompletion:r,fetchStreamingCompletion:n,onFallbackLog:i}){try{return e?await n(s,t,o):await r(s,t,o)}catch(a){let l=String(a?.message||a||"");if(!/proxy error 413|payload too large/i.test(l))throw a;let c=vr(s);return typeof i=="function"&&i(s,c),{data:e?await n(c,t,o):await r(c,t,o),fallback:c}}}async function yo({body:s,auth:e,options:t,streaming:o,toolContext:r,activeSkills:n,round:i,verbose:a,dependencies:l,config:c={}}){let d=c.maxToolResultChars||3e3,{fetchCompletion:p,fetchStreamingCompletion:f,onFallbackLog:_,hasToolCalls:g,getTextContent:b,parseToolCalls:y,buildAssistantMessage:w,buildToolResultMessage:h,executeTool:$,onToolCallLog:x,injectTools:v}=l;Array.isArray(s?.messages)&&(s.messages=ho(s.messages));let T=await Ir({body:s,streaming:o,auth:e,options:t,fetchCompletion:p,fetchStreamingCompletion:f,onFallbackLog:_}),E=T?.data||T;if(!g(E))return{done:!0,text:b(E),body:T?.fallback||s};let
|
|
105
|
+
[tool result truncated for size]`}function Ar(s,e){if(typeof s=="string")return Ye(s,e);try{return Ye(JSON.stringify(s),e)}catch{return Ye(String(s),e)}}function ho(s){let e=new Set;for(let r of s)if(r.role==="assistant"&&Array.isArray(r.tool_calls))for(let n of r.tool_calls)e.add(n.id);let t=s.filter(r=>r.role==="tool"?e.has(r.tool_call_id):!0),o=new Set;for(let r of t)r.role==="tool"&&o.add(r.tool_call_id);return t.map(r=>{if(r.role!=="assistant"||!Array.isArray(r.tool_calls)||r.tool_calls.every(l=>o.has(l.id)))return r;let{tool_calls:i,...a}=r;return{...a,content:a.content||""}})}function vr(s){let e=Array.isArray(s?.messages)?s.messages:[],t=e.find(n=>n.role==="system"),o=e.slice(-4).map(n=>({...n,content:Ye(n.content,n.role==="tool"?1200:2500)}));o=ho(o);let r={...s,messages:[t,...o].filter(Boolean)};return delete r.tools,r}async function Ir({body:s,streaming:e,auth:t,options:o,fetchCompletion:r,fetchStreamingCompletion:n,onFallbackLog:i}){try{return e?await n(s,t,o):await r(s,t,o)}catch(a){let l=String(a?.message||a||"");if(!/proxy error 413|payload too large/i.test(l))throw a;let c=vr(s);return typeof i=="function"&&i(s,c),{data:e?await n(c,t,o):await r(c,t,o),fallback:c}}}async function yo({body:s,auth:e,options:t,streaming:o,toolContext:r,activeSkills:n,round:i,verbose:a,dependencies:l,config:c={}}){let d=c.maxToolResultChars||3e3,{fetchCompletion:p,fetchStreamingCompletion:f,onFallbackLog:_,hasToolCalls:g,getTextContent:b,parseToolCalls:y,buildAssistantMessage:w,buildToolResultMessage:h,executeTool:$,onToolCallLog:x,injectTools:v}=l;Array.isArray(s?.messages)&&(s.messages=ho(s.messages));let T=await Ir({body:s,streaming:o,auth:e,options:t,fetchCompletion:p,fetchStreamingCompletion:f,onFallbackLog:_}),E=T?.data||T;if(!g(E))return{done:!0,text:b(E),body:T?.fallback||s};let k=y(E),I=T?.fallback||s;I.messages.push(w(E)),a&&typeof x=="function"&&x(k);let O=await Promise.all(k.map((S,A)=>(typeof t.onToolCall=="function"&&t.onToolCall(S.name,S.args,{round:i,index:A,total:k.length}),$(S,r))));for(let S=0;S<k.length;S++){let N=k[S].name==="get_skill_context"?typeof O[S]=="string"?O[S]:JSON.stringify(O[S]):Ar(O[S],d);I.messages.push(h(k[S].id,N))}return typeof t.onToolCall=="function"&&t.onToolCall(null),v(I,n),{done:!1,body:I}}var So=R(()=>{});function pe(s){!s||typeof s!="object"||(s.type==="object"&&s.properties&&(s.required=Object.keys(s.properties),s.additionalProperties=!1,Object.values(s.properties).forEach(pe)),s.type==="array"&&s.items&&pe(s.items),s.anyOf&&s.anyOf.forEach(pe),s.oneOf&&s.oneOf.forEach(pe),s.allOf&&s.allOf.forEach(pe))}function We(s){return Array.isArray(s)?new Set(s.map(e=>String(e||"").trim()).filter(Boolean)):new Set}function Pr(s){return Array.isArray(s)?s.map(e=>String(e||"").trim()).filter(Boolean):[]}var Cr,kr,ht,_o,fe,Nr,Ce,wo=R(()=>{Q();Y();ee();po();mo();H();go();So();Cr=J.ASSISTANT,kr=15,ht="get_skill_context";_o={maxBytes:49e3,systemMaxChars:12e3},fe=()=>process.env.ZIBBY_VERBOSE==="true"||process.env.ZIBBY_DEBUG==="true",Nr=s=>Buffer.byteLength(JSON.stringify(s),"utf8");Ce=class extends F{#e;#t;#o=new Ge;constructor(e=null){super("assistant","Zibby Assistant",200),e&&typeof e=="object"&&(e.toolProvider||e.completionProvider)?(this.#e=e.toolProvider||new ue,this.#t=e.completionProvider||new de):(this.#e=e||new ue,this.#t=new de)}canHandle(e){return gt().ok}async invoke(e,t={}){let o=t.model&&t.model!=="auto"?t.model:Cr,r=gt();if(!r.ok)throw r.reason==="missing_session_token"?new Error("Login required. Run `zibby login` to authenticate."):r.reason==="missing_openai_api_key"?new Error("Missing OPENAI_API_KEY for BYOK mode."):r.reason==="missing_openai_proxy_url"?new Error("Missing OPENAI_PROXY_URL for local mode."):new Error(`Assistant auth unavailable (${r.reason}).`);let n=t.messages||[{role:"user",content:e}],i=r.baseUrl,a=fe();if(a?await this.#m(n,o,i,r.tokenPreview||"none"):u.debug(`[Assistant] ${i} | model: ${o} | messages: ${n.length}`),t.schema)return this.#c(o,n,r,t);let l=t.activeSkills||[],c=this.#d(t),d=this.#f(l,c),p=this.#p(d),f=this.#u(t),_={...t,payloadCompaction:f},g={activeSkills:d,options:_,executionRegistry:p,capabilityPolicy:c},b=!!t.stream,y={model:o,messages:[...n],stream:!1};this.#s(y,d,c);for(let w=0;w<kr;w++){if(t.signal?.aborted)throw new Error("Aborted");let h=await yo({body:y,auth:r,options:_,streaming:b,toolContext:g,activeSkills:d,round:w,verbose:a,dependencies:{fetchCompletion:this.#n.bind(this),fetchStreamingCompletion:this.#l.bind(this),onFallbackLog:($,x)=>{fe()&&console.log(`413 fallback: messages ${$.messages.length} -> ${x.messages.length}, bytes=${Nr(x)}`)},hasToolCalls:$=>this.#e.hasToolCalls($),getTextContent:$=>this.#e.getTextContent($),parseToolCalls:$=>this.#e.parseToolCalls($),buildAssistantMessage:$=>this.#e.buildAssistantMessage($),buildToolResultMessage:($,x)=>this.#e.buildToolResultMessage($,x),executeTool:($,x)=>this.#a($,x),onToolCallLog:async $=>{let x=(await import("chalk")).default;console.log(x.dim(` ${$.map(v=>`${v.name}(${JSON.stringify(v.args).slice(0,80)})`).join(", ")}`))},injectTools:($,x)=>this.#s($,x,c)},config:{maxToolResultChars:t.maxToolResultChars||3e3}});if(h.done)return h.text;y.messages=h.body.messages,h.body.tools?y.tools=h.body.tools:delete y.tools}return"I hit my tool-calling limit for this turn. Please try again."}async cleanup(){await this.#o.stopAll()}#s(e,t,o=null){let r=this.#i(t,o),n=this.#e.formatTools(r);this.#e.injectToolsIntoBody(e,n)}#i(e,t=null){let o=[],r=new Set;for(let n of e){let i=j(n);if(i?.tools?.length)for(let a of i.tools)this.#r(a.name,t)&&(r.has(a.name)||(r.add(a.name),o.push({name:a.name,description:a.description,parameters:a.parameters||a.input_schema||{type:"object",properties:{}}})))}return!t?.disableSkillContextTool&&this.#r(ht,t)&&o.push({name:ht,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"]}}),o}async#a(e,t){let{activeSkills:o,options:r,executionRegistry:n,capabilityPolicy:i}=t;if(!this.#r(e.name,i))return`Tool "${e.name}" blocked by policy`;if(e.name===ht){let c=String(e.args?.skillId||"").trim();if(!c)return JSON.stringify({error:"skillId is required"});if(!o.includes(c))return JSON.stringify({error:`Skill "${c}" is not active`,activeSkills:o});let d=j(c);if(!d)return JSON.stringify({error:`Skill "${c}" not found`});let p=typeof d.promptFragment=="function"?d.promptFragment():d.promptFragment||"",f=(d.tools||[]).map(g=>g.name),_=JSON.stringify({skillId:c,description:d.description||"",toolNames:f,promptFragment:p||""});return fe()&&(console.log(`
|
|
99
106
|
\u{1F4D6} get_skill_context("${c}") \u2192 ${_.length} chars (fragment: ${p.length} chars)`),console.log(` tools: [${f.join(", ")}]`),console.log(` fragment preview: ${p.slice(0,200).replace(/\n/g,"\\n")}\u2026
|
|
100
107
|
`)),_}let a=n?.get(e.name)||null;if(!a)return`Unknown tool: ${e.name}`;let l=j(a.skillId);if(!l)return`Skill "${a.skillId}" not found for tool "${e.name}"`;if(a.mode==="handler")try{return l.handleToolCall(e.name,e.args,t)}catch(c){return`Error in ${e.name}: ${c.message}`}if(a.mode==="mcp")try{if(!this.#o.isRunning(l.serverName)){let d=l.resolve(r);if(!d)return`Skill "${a.skillId}" is not available (cannot start server)`;await this.#o.ensureServer(l.serverName,d)}let c=await this.#o.callTool(l.serverName,e.name,e.args);return c.text||(c.isError?"Tool call failed":"Done")}catch(c){return`MCP error (${l.serverName}): ${c.message}`}return`Skill "${a.skillId}" owns tool "${e.name}" but has no execution mode`}async#l(e,t,o){return this.#t.fetchStreamingCompletion(e,t,{...o,onBudget:({beforeBytes:r,meta:n})=>{fe()&&console.log(`payload bytes (stream) before=${r} after=${n.bytes} trimmed=${n.trimmed} messages=${n.messageCount}`)}})}async#n(e,t,o){return this.#t.fetchCompletion(e,t,{...o,onBudget:({beforeBytes:r,meta:n})=>{fe()&&console.log(`payload bytes before=${r} after=${n.bytes} trimmed=${n.trimmed} messages=${n.messageCount}`)}})}async#c(e,t,o,r){let{zodToJsonSchema:n}=await import("zod-to-json-schema"),i=typeof r.schema?.parse=="function",a=i?n(r.schema):r.schema;delete a.$schema,pe(a);let l={model:e,messages:t,stream:!1,response_format:{type:"json_schema",json_schema:{name:"extract",schema:a,strict:!0}}},c=await this.#n(l,o,r),d=this.#e.getTextContent(c),p=JSON.parse(d),f=i?r.schema.parse(p):p;return{raw:d,structured:f}}#u(e={}){let t=e?.config?.agent?.assistant?.payloadCompaction||{};return{maxBytes:Number(t.maxBytes||e.maxPayloadBytes||_o.maxBytes),systemMaxChars:Number(t.systemMaxChars||_o.systemMaxChars)}}#d(e={}){let t=e?.config?.agent?.assistant?.toolPolicy||{};return{allowTools:We(t.allowTools||e.allowTools),denyTools:We(t.denyTools||e.denyTools),denyPrefixes:Pr(t.denyPrefixes||e.denyToolPrefixes),includeSkills:We(t.includeSkills||e.includeSkills),excludeSkills:We(t.excludeSkills||e.excludeSkills),disableSkillContextTool:!!(t.disableSkillContextTool||e.disableSkillContextTool)}}#f(e,t){let o=t?.includeSkills||new Set,r=t?.excludeSkills||new Set;return o.size===0&&r.size===0?e:e.filter(n=>!(o.size>0&&!o.has(n)||r.has(n)))}#r(e,t){let o=String(e||"").trim();if(!o)return!1;let r=t?.allowTools;if(r&&r.size>0&&!r.has(o))return!1;let n=t?.denyTools;return!(n&&n.has(o)||(t?.denyPrefixes||[]).some(a=>o.startsWith(a)))}#p(e){let t=new Map,o=[];for(let r of e){let n=j(r);if(!n?.tools?.length)continue;let i=typeof n.handleToolCall=="function"?"handler":n.serverName&&typeof n.resolve=="function"?"mcp":null;if(i)for(let a of n.tools){let l=String(a?.name||"").trim();if(l){if(t.has(l)){o.push({tool:l,winner:t.get(l).skillId,skipped:r});continue}t.set(l,{skillId:r,mode:i})}}}if(o.length>0&&fe()){let r=o.slice(0,5).map(n=>`${n.tool}:${n.winner}>${n.skipped}`).join(", ");console.log(`tool registry collisions: ${r}${o.length>5?" ...":""}`)}return t}async#m(e,t,o,r){console.log(`
|
|
101
108
|
\u25C6 Model: ${t} | proxy: ${o} | token: ${r||"none"}
|
|
@@ -125,7 +132,7 @@ ${n}`);let a=o(),l=a.cwd||process.cwd(),c=a.sessionPath;try{if(c){let p=_t(c,q);
|
|
|
125
132
|
`).slice(2,14).join(`
|
|
126
133
|
`);console.log(`[zibby:session] stack (${s}):
|
|
127
134
|
${c}`)}}function Fr(){return process.env.ZIBBY_RUN_SOURCE==="studio"||process.env.ZIBBY_KEEP_SESSION_ENV==="1"||process.env.ZIBBY_KEEP_SESSION_ENV==="true"}function Jr(){if(process.env.ZIBBY_RUN_SOURCE!=="studio")return;let s=process.env.ZIBBY_SESSION_PATH;if(!(s==null||String(s).trim()===""))try{return Co(String(s).trim())}catch{return String(s).trim()}}function Gr(){Fr()||(delete process.env.ZIBBY_SESSION_PATH,delete process.env.ZIBBY_SESSION_ID)}function Kr({sessionPath:s,sessionId:e}){s&&typeof s=="string"&&(process.env.ZIBBY_SESSION_PATH=s),e!=null&&String(e).trim()!==""&&(process.env.ZIBBY_SESSION_ID=String(e).trim())}function Yr(s={}){let e=Dt.map(n=>process.env[n]).find(Boolean),t=Math.random().toString(36).slice(2,6),o=e||`${Date.now()}_${t}`,r=s.paths?.sessionPrefix;return r?`${r}_${o}`:o}function Wr({cwd:s=process.cwd(),config:e={},initialState:t={},traceFrom:o="resolveWorkflowSession"}={}){let r=t.sessionPath,n=t.sessionTimestamp,i="initialState.sessionPath";if(!r&&process.env.ZIBBY_SESSION_PATH)try{let c=Co(String(process.env.ZIBBY_SESSION_PATH));c&&(r=c,i="ZIBBY_SESSION_PATH")}catch{}let a;if(r)a=String(r).split(/[/\\]/).filter(Boolean).pop(),n==null&&(n=Date.now());else{let c=process.env.ZIBBY_SESSION_ID&&String(process.env.ZIBBY_SESSION_ID).trim();if(c)a=c,i="ZIBBY_SESSION_ID";else{let p=e.sessionId!=null?String(e.sessionId).trim():"";p&&p!=="last"?(a=p,i="config.sessionId"):(a=Yr(e),i="generated")}n=n??Date.now();let d=e.paths?.output||ie;r=se(s,d,Rt,a)}let l=!Et(r);return l&&Io(r,{recursive:!0}),Ur({traceFrom:o,sessionId:a,sessionPath:r,idSource:i,mkdirFresh:l}),Kr({sessionPath:r,sessionId:a}),{sessionPath:r,sessionId:a,sessionTimestamp:n}}var Pe=class{constructor(e={}){this.nodes=new Map,this.edges=new Map,this.entryPoint=null,this.middleware=Array.isArray(e.middleware)?[...e.middleware]:[],e.nodeMiddleware&&this.middleware.push(e.nodeMiddleware),this.nodeTypeMap=new Map,this.conditionalCodeMap=new Map,this.stateSchema=e.stateSchema||null,this.nodePrompts=new Map,this.nodeOptions=new Map,this._invokeAgent=e.invokeAgent||null}setStateSchema(e){return this.stateSchema=e,this}getStateSchema(){return this.stateSchema}addNode(e,t,o={}){let r=t instanceof re?t:new re(t);return r.name=e,this.nodes.set(e,r),o.prompt&&this.nodePrompts.set(e,o.prompt),Object.keys(o).length>0&&this.nodeOptions.set(e,o),this}addConditionalNode(e,t){let o=new Ne({...t,name:e});return this.nodes.set(e,o),this}addEdge(e,t){return this.edges.set(e,t),this}setNodeType(e,t){return this.nodeTypeMap.set(e,t),this}addConditionalEdges(e,t,{labels:o}={}){return this.edges.set(e,{conditional:!0,routes:t,labels:o}),typeof t=="function"&&this.conditionalCodeMap.set(e,t.toString()),this}setEntryPoint(e){return this.entryPoint=e,this}use(e){return typeof e=="function"&&this.middleware.push(e),this}_composeMiddleware(e,t,o,r,n){let i=o;for(let a=e.length-1;a>=0;a--){let l=e[a],c=i;i=()=>l(t,c,r,n)}return i()}serialize(){let e=[],t={};for(let[n,i]of this.nodes){let a=this.nodeTypeMap.get(n)||n;e.push({id:n,type:a,data:{nodeType:a,label:n}});let l=i._isCustomCode||!1,c={};l&&typeof i.execute=="function"&&(c.customCode=i.execute.toString());let d=this.nodePrompts.get(n);if(d&&(c.prompt=d),typeof i.customExecute=="function"&&(c.executeCode=i.customExecute.toString()),i.outputSchema)try{if(typeof i.outputSchema._def<"u"){let _=vo(i.outputSchema,{target:"openApi3"}),g=this._flattenJsonSchemaToVariables(_);c.outputSchema={jsonSchema:_,variables:g}}else c.outputSchema={schema:i.outputSchema}}catch(f){console.warn(`Failed to convert schema for ${n}:`,f.message)}let p=(this.resolvedToolsMap||{})[n];p?.toolIds&&(c.tools=p.toolIds),Object.keys(c).length>0&&(t[n]=c)}let o=[];for(let[n,i]of this.edges)if(typeof i=="string")o.push({source:n,target:i});else if(i.conditional){let a=this.conditionalCodeMap.get(n)||i.routes.toString(),l=this._inferConditionalTargets(i.routes),c=i.labels||{};for(let d of l){let p={source:n,target:d,data:{conditionalCode:a}};c[d]&&(p.label=c[d]),o.push(p)}}let r=null;if(this.stateSchema)try{r=vo(this.stateSchema,{target:"openApi3"})}catch{r=this.stateSchema}return{nodes:e,edges:o,nodeConfigs:t,stateSchema:r}}_inferConditionalTargets(e){let t=e.toString(),o=new Set,r=/return\s+['"]([^'"]+)['"]/g,n;for(;(n=r.exec(t))!==null;)o.add(n[1]);return[...o]}_flattenJsonSchemaToVariables(e,t=""){let o=e;if(e.$ref&&e.definitions){let r=e.$ref.replace("#/definitions/","");o=e.definitions[r]||e}return this._flattenSchema(o,t)}_flattenSchema(e,t=""){if(!e||typeof e!="object")return[];let o=[],r=e.properties||{},n=e.required||[];for(let[i,a]of Object.entries(r)){let l=t?`${t}.${i}`:i,c=!n.includes(i);if(o.push({path:l,type:a.type||"unknown",label:a.description||this._formatLabel(i),optional:c}),a.type==="object"&&a.properties){let d=this._flattenSchema(a,l);o.push(...d)}if(a.type==="array"&&a.items?.type==="object"&&a.items.properties){let d=this._flattenSchema(a.items,`${l}[]`);o.push(...d)}}return o}_formatLabel(e){return e.replace(/([A-Z])/g," $1").replace(/^./,t=>t.toUpperCase()).trim()}_summarizeNodeOutput(e,t){if(!t||typeof t!="object")return[];let o=[];t.success!==void 0&&o.push(`Result: ${t.success?"passed":"failed"}`);for(let[r,n]of Object.entries(t))if(!(r==="success"||r==="raw"||r==="nextNode")){if(typeof n=="string"&&n.length<=80)o.push(`${r}: ${n}`);else if(Array.isArray(n)){let i=n.length,a=n.filter(c=>c?.passed===!0).length;if(n.some(c=>c?.passed!==void 0)){let c=i-a;o.push(`${r}: ${a}/${i} passed${c?`, ${c} failed`:""}`)}else o.push(`${r}: ${i} items`)}if(o.length>=4)break}return o}async run(e,t={}){if(!this.entryPoint)throw new Error("No entry point set for graph");let o=t.cwd||process.cwd();Br({path:se(o,".env")});let r=t.config||{};if(!r||Object.keys(r).length===0)try{let v=se(o,".zibby.config.js");Et(v)&&(r=(await import(v)).default||{})}catch{}process.env.EXECUTION_ID&&!r.agent?.strictMode&&(r.agent={...r.agent,strictMode:!0});let n=t.agentType;if(!n){let v=r?.agent;v?.provider?n=v.provider:v?.gemini?n="gemini":v?.claude?n="claude":v?.cursor?n="cursor":v?.codex?n="codex":n=process.env.AGENT_TYPE||"cursor"}let i=t.contextConfig||e?.config?.contextConfig||e?.config?.context||r?.context||{};if(this.stateSchema){let v=this.stateSchema.safeParse(t);if(!v.success){let T=v.error.issues.map(E=>`${E.path.join(".")}: ${E.message}`);throw console.error("\u274C Initial state validation failed:"),T.forEach(E=>console.error(` - ${E}`)),new Error(`State validation failed: ${T.join(", ")}`)}D.step("State validated against schema")}let a=Jr(),l=t.sessionPath||a;l||Gr();let{sessionPath:c,sessionTimestamp:d,sessionId:p}=Wr({cwd:o,config:r,traceFrom:"WorkflowGraph.run",initialState:{sessionPath:l,sessionTimestamp:t.sessionTimestamp}});D.step(`Session ${p}`);let f=await Ze.loadContext(t.specPath||"",o,i);Object.keys(f).length>0&&D.step(`Context loaded: ${Object.keys(f).join(", ")}`);let _=t.outputPath;!_&&t.specPath&&(e?.calculateOutputPath?_=e.calculateOutputPath(t.specPath):console.warn(`\u26A0\uFE0F outputPath not resolved (specPath=${t.specPath})`));let g=new ye({...t,config:r,agentType:n,outputPath:_,sessionPath:c,sessionTimestamp:d,context:f,resolvedTools:this.resolvedToolsMap||{}}),b=new Map;try{await import("@zibby/skills")}catch{}let{getSkill:y}=await Promise.resolve().then(()=>(H(),at)),w=new Set;for(let[,v]of this.nodes)for(let T of v.config?.skills||[])w.add(T);for(let v of w){let T=y(v);if(typeof T?.middleware=="function")try{let E=await T.middleware();typeof E=="function"&&b.set(v,E)}catch{}}let h=this.entryPoint,$=[];for(;h&&h!=="END";){let v=se(c,Le);if(Et(v)){console.warn(`
|
|
128
|
-
\u{1F6D1} Studio stop requested \u2014 ending workflow.`);try{Lr(v)}catch{}if(e&&typeof e.cleanup=="function")try{await e.cleanup()}catch{}return D.step("Workflow stopped by Studio"),{success:!0,state:g.getAll(),executionLog:$,stoppedByStudio:!0}}let T=this.nodes.get(h);if(!T)throw new Error(`Node '${h}' not found in graph`);let E=JSON.stringify({sessionPath:c,sessionTimestamp:d,currentNode:h,createdAt:new Date().toISOString(),config:g.get("config")}),
|
|
135
|
+
\u{1F6D1} Studio stop requested \u2014 ending workflow.`);try{Lr(v)}catch{}if(e&&typeof e.cleanup=="function")try{await e.cleanup()}catch{}return D.step("Workflow stopped by Studio"),{success:!0,state:g.getAll(),executionLog:$,stoppedByStudio:!0}}let T=this.nodes.get(h);if(!T)throw new Error(`Node '${h}' not found in graph`);let E=JSON.stringify({sessionPath:c,sessionTimestamp:d,currentNode:h,createdAt:new Date().toISOString(),config:g.get("config")}),k=se(c,q);Ao(k,E,"utf-8");let I=g.get("config")?.paths?.output||ie,O=se(o,I,q);Io(se(o,I),{recursive:!0});try{Ao(O,E,"utf-8")}catch{}let S=t.onPipelineProgress;if(typeof S=="function")try{S({cwd:o,sessionPath:c,sessionId:p,outputBase:g.get("config")?.paths?.output||ie,currentNode:h})}catch{}let A=(this.resolvedToolsMap||{})[h]||null;g.set("_currentNodeTools",A);let N=g.get("nodeConfigs")||{};g.set("_currentNodeConfig",N[h]||{}),D.nodeStart(h);let m=Date.now(),C=this.nodePrompts.get(h);if(!this._invokeAgent){let M=await Promise.resolve().then(()=>(me(),ke));this._invokeAgent=M.invokeAgent}let P=this._invokeAgent,G={state:g,invokeAgent:async(M={},K={})=>{let B=K.prompt||"";if(C)try{B=jr.compile(C,{noEscape:!0})(M)}catch(V){throw console.error(`\u274C Template rendering failed for node '${h}':`,V.message),new Error(`Template rendering failed: ${V.message}`,{cause:V})}else if(!B)throw new Error(`No prompt template configured for node '${h}' and no prompt provided in options`);let ge={state:g.getAll(),images:K.images||[]},qe={model:K.model||g.get("model"),workspace:g.get("workspace"),schema:K.schema,...K};return P(B,ge,qe)},_coreInvokeAgent:P,agent:e,nodeId:h,promptTemplate:C,getPromptTemplate:()=>C,...g.getAll()};try{let M=(T.config?.skills||[]).map(Re=>b.get(Re)).filter(Boolean),K=[...this.middleware,...M],B;K.length>0?B=await this._composeMiddleware(K,h,async()=>T.execute(G,g),g.getAll(),g):B=await T.execute(G,g);let ge=Date.now()-m;if($.push({node:h,success:B.success,duration:ge,timestamp:new Date().toISOString()}),!B.success){if(String(B.error||"").includes("Stopped from Zibby Studio")){if(D.step("Workflow stopped by Studio"),g.set("stoppedByStudio",!0),e&&typeof e.cleanup=="function")try{await e.cleanup()}catch{}return{success:!0,state:g.getAll(),executionLog:$,stoppedByStudio:!0}}g.append("errors",{node:h,error:B.error});let he=T.config?.retries||0,Ot=`${h}_retries`,De=g.getAll()[Ot]||0;if(De<he){D.stepInfo(`Retrying (attempt ${De+1}/${he})`),g.update({[Ot]:De+1,[`${h}_raw`]:B.raw});continue}throw D.nodeFailed(h,B.error,{duration:ge}),new Error(`Node '${h}' failed after ${De} attempts: ${B.error}`)}g.update({[h]:B.output});let qe=this._summarizeNodeOutput(h,B.output);D.nodeComplete(h,{duration:ge,details:qe});let V=this.edges.get(h);if(!V)h="END";else if(V.conditional){let Re=g.getAll(),he=V.routes(Re);D.route(h,he),h=he}else h=V}catch(M){throw D.isInsideNode&&D.nodeFailed(h,M.message,{duration:Date.now()-m}),g.set("failed",!0),g.set("failedAt",h),M}}D.graphComplete();let x={success:!0,state:g.getAll(),executionLog:$};return e&&typeof e.onComplete=="function"&&await e.onComplete(x),x}};var Me=new Map;function ko(s,e){Me.set(s,e)}function $t(s){return Me.get(s)}function He(s){return Me.has(s)}function zr(){return Array.from(Me.keys())}function xt(s){let e=Me.get(s);return e?e.factory&&typeof e.create=="function"?e.create.toString():typeof e.execute=="function"?e.execute.toString():typeof e=="function"?e.toString():null:null}ko("ai_agent",{name:"ai_agent",factory:!0,create:(s,e={})=>({name:s,execute:async t=>{let o=t?._coreInvokeAgent;o||(o=(await Promise.resolve().then(()=>(me(),ke))).invokeAgent);let r=e.extraPromptInstructions||"Execute the task based on the current state.",n=Xr(r,t),i=await o(n,{cwd:t.workspace||process.cwd(),model:t.model,tools:e.resolvedTools||null});return{success:!0,output:{raw:i,nodeId:s},raw:typeof i=="string"?i:i.raw}}})});function Zr(s){let e=/@([\w.]+)/g,t=new Set,o;for(;(o=e.exec(s))!==null;)t.add(o[1]);return Array.from(t)}function Hr(s,e){let t=e.split("."),o=s;for(let r of t){if(o==null)return;o=o[r]}return o}function Vr(s){return s==null?"[not available]":typeof s=="string"?s:typeof s=="object"?s.raw&&typeof s.raw=="string"?s.raw:JSON.stringify(s,null,2):String(s)}function Xr(s,e){let t=Zr(s);if(t.length===0)return s;let o=[],r=new Set;for(let n of t){let i=n.split(".")[0];if(r.has(i))continue;let a=Hr(e,n);if(a!==void 0){let l=Vr(a),c=n.replace(/_/g," ").replace(/\b\w/g,d=>d.toUpperCase());o.push(`## ${c}
|
|
129
136
|
${l}`),n.includes(".")||r.add(i)}}return o.length===0?s:`${s}
|
|
130
137
|
|
|
131
138
|
---
|