@zibby/core 0.5.6 → 0.5.7
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 +53 -53
- package/dist/models.js +1 -1
- package/dist/package.json +1 -1
- package/dist/register-built-in-strategies.js +37 -37
- package/dist/state-utils.js +1 -1
- package/dist/strategies/assistant-strategy.js +9 -9
- package/dist/strategies/claude-strategy.js +2 -2
- package/dist/strategies/codex-strategy.js +3 -3
- package/dist/strategies/cursor-strategy.js +29 -29
- package/dist/strategies/gemini-strategy.js +16 -16
- package/dist/strategies/index.js +45 -45
- package/dist/strategies/utils/cursor-output-formatter.js +4 -4
- package/dist/strategies/utils/openai-proxy-formatter.js +3 -3
- package/dist/strategies/utils/prompt-builder.js +6 -6
- package/dist/strategies/utils/structured-output-formatter.js +4 -4
- package/dist/strategies/utils/zod-json-schema.js +1 -0
- package/package.json +1 -1
|
@@ -1,11 +1,11 @@
|
|
|
1
|
-
import{AgentStrategy as
|
|
1
|
+
var ce=Object.defineProperty;var ue=(n,e)=>()=>(n&&(e=n(n=0)),e);var de=(n,e)=>{for(var o in e)ce(n,o,{get:e[o],enumerable:!0})};var se={};de(se,{zodToJson:()=>Oe});import{z as ne}from"zod";import{zodToJsonSchema as we}from"zod-to-json-schema";function Oe(n,e={target:"openApi3"}){if(!n)return null;if(typeof ne?.toJSONSchema=="function")try{return ne.toJSONSchema(n)}catch{}try{return we(n,e)}catch{return null}}var ie=ue(()=>{});import{AgentStrategy as Ae,getSkill as F}from"@zibby/agent-workflow";import M from"chalk";var v={debug:0,info:1,warn:2,error:3,silent:4},J=class{constructor(){this._level=this._getLogLevel()}_getLogLevel(){if(process.env.ZIBBY_DEBUG==="true")return v.debug;if(process.env.ZIBBY_VERBOSE==="true")return v.info;let e=process.env.LOG_LEVEL?.toLowerCase();return e&&e in v?v[e]:v.info}_shouldLog(e){return v[e]>=this._level}_formatMessage(e,o,s={}){let t=new Date().toISOString(),a=`${this._getPrefix(e)} ${o}`;return Object.keys(s).length>0&&(a+=M.dim(` ${JSON.stringify(s)}`)),a}_getPrefix(e){return{debug:M.gray("[DEBUG]"),info:M.cyan("[INFO]"),warn:M.yellow("[WARN]"),error:M.red("\u274C [ERROR]")}[e]||""}debug(e,o){this._shouldLog("debug")&&console.log(this._formatMessage("debug",e,o))}info(e,o){this._shouldLog("info")&&console.log(this._formatMessage("info",e,o))}warn(e,o){this._shouldLog("warn")&&console.warn(this._formatMessage("warn",e,o))}error(e,o){this._shouldLog("error")&&console.error(this._formatMessage("error",e,o))}setLevel(e){e in v&&(this._level=v[e])}getLevel(){return Object.keys(v).find(e=>v[e]===this._level)}},C=new J;var fe=[{alias:"auto",canonicalId:"claude-sonnet-4-6",vendor:"claude",label:"Claude \xB7 Sonnet 4.6",family:"claude-sonnet-4",pickerVisible:!0,isDefault:!0},{alias:"opus-4.8",canonicalId:"claude-opus-4-8",vendor:"claude",label:"Claude \xB7 Opus 4.8",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"sonnet-4.8",canonicalId:"claude-sonnet-4-8",vendor:"claude",label:"Claude \xB7 Sonnet 4.8",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"opus-4.7",canonicalId:"claude-opus-4-7",vendor:"claude",label:"Claude \xB7 Opus 4.7",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"sonnet-4.7",canonicalId:"claude-sonnet-4-7",vendor:"claude",label:"Claude \xB7 Sonnet 4.7",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"sonnet-4.6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"opus-4.6",canonicalId:"claude-opus-4-6",vendor:"claude",label:"Claude \xB7 Opus 4.6",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"sonnet-4.5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",label:"Claude \xB7 Sonnet 4.5",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"opus-4.5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",label:"Claude \xB7 Opus 4.5",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"haiku-4.5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",label:"Claude \xB7 Haiku 4.5",family:"claude-haiku-4"},{alias:"haiku-4-5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"claude-sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"claude-opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"claude-sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"claude-sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"claude-opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"claude-sonnet-4-5-20250929",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"claude-opus-4-5-20251101",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"claude-haiku-4-5-20251001",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-20250514",canonicalId:"claude-opus-4-20250514",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"auto",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",isDefault:!0},{alias:"o4-mini",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",pickerVisible:!0},{alias:"o3",canonicalId:"o3",vendor:"codex",family:"o3"},{alias:"o3-mini",canonicalId:"o3-mini",vendor:"codex",family:"o3-mini"},{alias:"codex-mini",canonicalId:"codex-mini-latest",vendor:"codex",family:"codex-mini"},{alias:"gpt-4o",canonicalId:"gpt-4o",vendor:"codex",label:"Codex \xB7 GPT-4o",family:"gpt-4o",pickerVisible:!0},{alias:"gpt-4o-mini",canonicalId:"gpt-4o-mini",vendor:"codex",label:"Codex \xB7 GPT-4o mini",family:"gpt-4o-mini",pickerVisible:!0},{alias:"gpt-5.2-codex",canonicalId:"gpt-5.2-codex",vendor:"codex",label:"Codex \xB7 GPT-5.2",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.2",canonicalId:"gpt-5.2",vendor:"codex",label:"Codex \xB7 GPT-5.2 base",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3",canonicalId:"gpt-5.3",vendor:"codex",label:"Codex \xB7 GPT-5.3",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3-codex",canonicalId:"gpt-5.3-codex",vendor:"codex",family:"gpt-5"},{alias:"gpt-5.4",canonicalId:"gpt-5.4",vendor:"codex",label:"Codex \xB7 GPT-5.4",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.5",canonicalId:"gpt-5.5",vendor:"codex",label:"Codex \xB7 GPT-5.5",family:"gpt-5",pickerVisible:!0},{alias:"auto",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",isDefault:!0},{alias:"gemini-3-pro",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",pickerVisible:!0},{alias:"gemini-3-flash",canonicalId:"gemini-3-flash",vendor:"gemini",label:"Gemini \xB7 3 Flash",family:"gemini-3",pickerVisible:!0},{alias:"gemini-2.5-pro",canonicalId:"gemini-2.5-pro",vendor:"gemini",label:"Gemini \xB7 2.5 Pro",family:"gemini-2.5",pickerVisible:!0},{alias:"gemini-2.5-flash",canonicalId:"gemini-2.5-flash",vendor:"gemini",label:"Gemini \xB7 2.5 Flash",family:"gemini-2.5",pickerVisible:!0}];function N(n){let e={};for(let o of fe)o.vendor===n&&(e[o.alias]=o.canonicalId);return e}var q={ASSISTANT:"gpt-5.4-nano-2026-03-17",CLAUDE:"claude-sonnet-4-6",CURSOR:"auto",CODEX:"o4-mini",GEMINI:"gemini-3-pro",OPENAI_POSTPROCESSING:"gpt-4o-mini"};var Ne=N("claude"),De=N("codex"),Ue=N("gemini"),je={CURSOR_AGENT_DEFAULT:1200*1e3,OPENAI_REQUEST:18e4};var L=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,o){throw new Error("buildToolResultMessage() must be implemented")}injectToolsIntoBody(e,o){throw new Error("injectToolsIntoBody() must be implemented")}};var I=class extends L{formatTools(e){return e.map(o=>({type:"function",function:{name:o.name,description:o.description,parameters:o.parameters||o.input_schema||{type:"object",properties:{}}}}))}hasToolCalls(e){let o=e.choices?.[0]?.message;return!!(o?.tool_calls&&o.tool_calls.length>0)}parseToolCalls(e){return(e.choices?.[0]?.message?.tool_calls||[]).map(s=>({id:s.id,name:s.function.name,args:JSON.parse(s.function.arguments||"{}")}))}getTextContent(e){return e.choices?.[0]?.message?.content||""}buildAssistantMessage(e){return e.choices?.[0]?.message}buildToolResultMessage(e,o){return{role:"tool",tool_call_id:e,content:typeof o=="string"?o:JSON.stringify(o)}}injectToolsIntoBody(e,o){return o.length>0&&(e.tools=o),e}};var B=class{async fetchCompletion(e,o,s={}){throw new Error("fetchCompletion() must be implemented")}async fetchStreamingCompletion(e,o,s={}){throw new Error("fetchStreamingCompletion() must be implemented")}};function D(n){return Buffer.byteLength(JSON.stringify(n),"utf8")}function U(n,e){let o=String(n||"");if(o.length<=e)return o;let s=Math.max(0,e-28);return`${o.slice(0,s)}
|
|
2
2
|
|
|
3
|
-
[truncated for size budget]`}function
|
|
4
|
-
`);
|
|
5
|
-
`)||"",isError:i.isError||!1}}isRunning(e){return this.#e.has(e)}async stopServer(e){let o=this.#e.get(e);if(o){try{await o.client.close()}catch(
|
|
3
|
+
[truncated for size budget]`}function Y(n,e=0){if(!n||typeof n!="object"||e>8)return n;if(Array.isArray(n))return n.map(s=>Y(s,e+1));let o={};for(let[s,t]of Object.entries(n))s==="description"||s==="title"||s==="examples"||s==="default"||(o[s]=Y(t,e+1));return o}function me(n=[]){return n.map(e=>({...e,function:{...e.function,description:U(e.function?.description||"",180),parameters:Y(e.function?.parameters||{type:"object",properties:{}})}}))}function W(n){let e=new Set;for(let t of n)if(t.role==="assistant"&&Array.isArray(t.tool_calls))for(let i of t.tool_calls)e.add(i.id);let o=n.filter(t=>t.role==="tool"?e.has(t.tool_call_id):!0),s=new Set;for(let t of o)t.role==="tool"&&s.add(t.tool_call_id);return o.map(t=>{if(t.role!=="assistant"||!Array.isArray(t.tool_calls)||t.tool_calls.every(r=>s.has(r.id)))return t;let{tool_calls:a,...l}=t;return{...l,content:l.content||""}})}function X(n,e={}){let o=e.maxBytes||49e3,s=e.systemMaxChars||12e3,t={...n,messages:Array.isArray(n.messages)?[...n.messages]:[],tools:Array.isArray(n.tools)?me(n.tools):n.tools};t.messages.length>0&&t.messages[0]?.role==="system"&&(t.messages[0]={...t.messages[0],content:U(t.messages[0].content,s)});let i=!1;for(;D(t)>o&&t.messages.length>2;)t.messages.splice(1,1),i=!0;if(i&&(t.messages=W(t.messages)),D(t)>o&&t.messages.length>0&&(t.messages[0]={...t.messages[0],content:U(t.messages[0].content,6e3)},i=!0),D(t)>o){let a=t.messages.find(r=>r.role==="system")||t.messages[0],l=t.messages.slice(-2);t.messages=W([a,...l].filter(Boolean).map((r,c)=>({...r,content:U(r.content,c===0?4e3:8e3)}))),i=!0}return{body:t,meta:{bytes:D(t),trimmed:i,maxBytes:o,messageCount:t.messages.length}}}var Q=n=>Buffer.byteLength(JSON.stringify(n),"utf8"),E=class extends B{async fetchCompletion(e,o,s={}){let t=Q(e),{body:i,meta:a}=X(e,s.payloadCompaction);s.onBudget?.({streaming:!1,beforeBytes:t,meta:a});let l=this.#e(s),r=`${o.baseUrl}${s.chatCompletionsPath||"/v1/chat/completions"}`,c=await fetch(r,{method:"POST",headers:o.headers,body:JSON.stringify(i),signal:l});if(!c.ok){let u=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}: ${u}`)}return c.json()}async fetchStreamingCompletion(e,o,s={}){let t={...e,stream:!0},i=Q(t),{body:a,meta:l}=X(t,s.payloadCompaction);s.onBudget?.({streaming:!0,beforeBytes:i,meta:l});let r=this.#e(s),c=`${o.baseUrl}${s.chatCompletionsPath||"/v1/chat/completions"}`,u=await fetch(c,{method:"POST",headers:o.headers,body:JSON.stringify(a),signal:r});if(!u.ok){let O=await u.text();throw u.status===401||u.status===403?new Error("Session expired. Run `zibby login` to re-authenticate."):new Error(`Proxy error ${u.status}: ${O}`)}let f=u.body.getReader(),b=new TextDecoder,_="",x="",w=new Map;for(;;){let{done:O,value:m}=await f.read();if(O)break;_+=b.decode(m,{stream:!0});let S=_.split(`
|
|
4
|
+
`);_=S.pop();for(let T of S){if(!T.startsWith("data: "))continue;let d=T.slice(6).trim();if(d==="[DONE]")continue;let p;try{p=JSON.parse(d)}catch{continue}let g=p.choices?.[0]?.delta;if(g&&(g.content&&(x+=g.content,s.onToken&&s.onToken(g.content)),g.tool_calls))for(let y of g.tool_calls){let k=y.index??0;w.has(k)||w.set(k,{id:"",name:"",args:""});let A=w.get(k);y.id&&(A.id=y.id),y.function?.name&&(A.name=y.function.name),y.function?.arguments!=null&&(A.args+=y.function.arguments)}}}if(w.size>0){let O=[...w.entries()].sort(([m],[S])=>m-S).map(([,m])=>({id:m.id,type:"function",function:{name:m.name,arguments:m.args}}));return{choices:[{message:{role:"assistant",content:x||null,tool_calls:O}}]}}return{choices:[{message:{role:"assistant",content:x}}]}}#e(e={}){let o=[e.signal,e.timeout?AbortSignal.timeout(e.timeout):null].filter(Boolean);return o.length>1?AbortSignal.any(o):o[0]||void 0}};import{Client as pe}from"@modelcontextprotocol/sdk/client/index.js";import{StdioClientTransport as ge}from"@modelcontextprotocol/sdk/client/stdio.js";var j=class{#e=new Map;async ensureServer(e,o){if(this.#e.has(e))return this.#e.get(e);let{command:s,args:t=[],env:i={}}=o;C.debug(`[MCP] Starting ${e}: ${s} ${t.join(" ")}`);let a=new ge({command:s,args:t,env:{...process.env,...i}}),l=new pe({name:`zibby-chat-${e}`,version:"1.0.0"},{capabilities:{}});await l.connect(a);let r={client:l,transport:a,serverConfig:o};return this.#e.set(e,r),r}async callTool(e,o,s={}){let t=this.#e.get(e);if(!t)throw new Error(`MCP server "${e}" not running`);C.debug(`[MCP] ${e}.${o}(${JSON.stringify(s).slice(0,200)})`);let i=await t.client.callTool({name:o,arguments:s});return{text:i.content?.filter(l=>l.type==="text").map(l=>l.text).join(`
|
|
5
|
+
`)||"",isError:i.isError||!1}}isRunning(e){return this.#e.has(e)}async stopServer(e){let o=this.#e.get(e);if(o){try{await o.client.close()}catch(s){C.debug(`[MCP] Error closing ${e}: ${s.message}`)}this.#e.delete(e)}}async stopAll(){let e=[...this.#e.keys()];await Promise.allSettled(e.map(o=>this.stopServer(o)))}};import{existsSync as ye,readFileSync as he}from"node:fs";import{join as be}from"node:path";import{homedir as _e}from"node:os";function xe(){try{let n=be(_e(),".zibby","config.json");return ye(n)?JSON.parse(he(n,"utf-8")):{}}catch{return{}}}function V(n){return String(n||"").replace(/\/v1\/?$/,"")}function Se(n,e){let o=process.env.OPENAI_PROXY_URL;if(o)return V(o);if(n==="session")return e.proxyUrl?V(e.proxyUrl):`${(process.env.ZIBBY_API_URL||"https://api-prod.zibby.app").replace(/\/$/,"")}/openai-proxy`;if(n==="byok"){let s=process.env.OPENAI_BASE_URL;return V(s||"https://api.openai.com")}return V(o||"")}function H(){let n=xe(),e=process.env.ZIBBY_USER_TOKEN||n.sessionToken||null,o=process.env.OPENAI_API_KEY||null,s=(process.env.ASSISTANT_AUTH_MODE||"").trim().toLowerCase(),t=process.env.OPENAI_PROXY_NO_AUTH==="true",i=process.env.OPENAI_PROXY_URL,a="session";s==="byok"||s==="local"||s==="session"?a=s:e?a="session":o?a="byok":i&&t&&(a="local");let l=Se(a,n),r={"Content-Type":"application/json"};if(a==="session"){if(!e)return{ok:!1,mode:a,reason:"missing_session_token"};r.Authorization=`Bearer ${e}`}else if(a==="byok"){if(!o)return{ok:!1,mode:a,reason:"missing_openai_api_key"};r.Authorization=`Bearer ${o}`}else if(a==="local"){if(!l)return{ok:!1,mode:a,reason:"missing_openai_proxy_url"};!t&&o&&(r.Authorization=`Bearer ${o}`)}return l?{ok:!0,mode:a,baseUrl:l,headers:r,tokenPreview:r.Authorization?`***${r.Authorization.slice(-4)}`:"none"}:{ok:!1,mode:a,reason:"missing_base_url"}}function z(n,e){let o=String(n??"");return o.length<=e?o:`${o.slice(0,Math.max(0,e-30))}
|
|
6
6
|
|
|
7
|
-
[tool result truncated for size]`}function
|
|
8
|
-
\u{1F4D6} get_skill_context("${c}") \u2192 ${
|
|
9
|
-
`)),
|
|
10
|
-
\u25C6 Model: ${o} | proxy: ${
|
|
11
|
-
`);let i=(await import("chalk")).default;console.log(i.bold("Prompt sent to LLM:")),console.log(i.dim("\u2500".repeat(60)));let
|
|
7
|
+
[tool result truncated for size]`}function Te(n,e){if(typeof n=="string")return z(n,e);try{return z(JSON.stringify(n),e)}catch{return z(String(n),e)}}function ee(n){let e=new Set;for(let t of n)if(t.role==="assistant"&&Array.isArray(t.tool_calls))for(let i of t.tool_calls)e.add(i.id);let o=n.filter(t=>t.role==="tool"?e.has(t.tool_call_id):!0),s=new Set;for(let t of o)t.role==="tool"&&s.add(t.tool_call_id);return o.map(t=>{if(t.role!=="assistant"||!Array.isArray(t.tool_calls)||t.tool_calls.every(r=>s.has(r.id)))return t;let{tool_calls:a,...l}=t;return{...l,content:l.content||""}})}function ke(n){let e=Array.isArray(n?.messages)?n.messages:[],o=e.find(i=>i.role==="system"),s=e.slice(-4).map(i=>({...i,content:z(i.content,i.role==="tool"?1200:2500)}));s=ee(s);let t={...n,messages:[o,...s].filter(Boolean)};return delete t.tools,t}async function ve({body:n,streaming:e,auth:o,options:s,fetchCompletion:t,fetchStreamingCompletion:i,onFallbackLog:a}){try{return e?await i(n,o,s):await t(n,o,s)}catch(l){let r=String(l?.message||l||"");if(!/proxy error 413|payload too large/i.test(r))throw l;let c=ke(n);return typeof a=="function"&&a(n,c),{data:e?await i(c,o,s):await t(c,o,s),fallback:c}}}async function oe({body:n,auth:e,options:o,streaming:s,toolContext:t,activeSkills:i,round:a,verbose:l,dependencies:r,config:c={}}){let u=c.maxToolResultChars||3e3,{fetchCompletion:f,fetchStreamingCompletion:b,onFallbackLog:_,hasToolCalls:x,getTextContent:w,parseToolCalls:O,buildAssistantMessage:m,buildToolResultMessage:S,executeTool:T,onToolCallLog:d,injectTools:p}=r;Array.isArray(n?.messages)&&(n.messages=ee(n.messages));let g=await ve({body:n,streaming:s,auth:e,options:o,fetchCompletion:f,fetchStreamingCompletion:b,onFallbackLog:_}),y=g?.data||g;if(!x(y))return{done:!0,text:w(y),body:g?.fallback||n};let k=O(y),A=g?.fallback||n;A.messages.push(m(y)),l&&typeof d=="function"&&d(k);let R=await Promise.all(k.map((h,K)=>(typeof o.onToolCall=="function"&&o.onToolCall(h.name,h.args,{round:a,index:K,total:k.length}),T(h,t))));for(let h=0;h<k.length;h++){let re=k[h].name==="get_skill_context"?typeof R[h]=="string"?R[h]:JSON.stringify(R[h]):Te(R[h],u);A.messages.push(S(k[h].id,re))}return typeof o.onToolCall=="function"&&o.onToolCall(null),p(A,i),{done:!1,body:A}}function te(n){let e=`${n}_POOL`,o=process.env[e];if(!o||typeof o!="string")return{picked:null,count:0,dispose:()=>{}};let s=o.split(/[,\n]+/).map(a=>a.trim()).filter(Boolean);if(s.length===0)return{picked:null,count:0,dispose:()=>{}};let t=s[Math.floor(Math.random()*s.length)],i=process.env[n];return process.env[n]=t,C.debug(`[auth-pool] ${n}: picked 1 of ${s.length} from pool (***${t.slice(-4)})`),{picked:t,count:s.length,dispose(){i===void 0?delete process.env[n]:process.env[n]=i}}}var Ce=q.ASSISTANT,Ie=15,Z="get_skill_context";function $(n){!n||typeof n!="object"||(n.type==="object"&&n.properties&&(n.required=Object.keys(n.properties),n.additionalProperties=!1,Object.values(n.properties).forEach($)),n.type==="array"&&n.items&&$(n.items),n.anyOf&&n.anyOf.forEach($),n.oneOf&&n.oneOf.forEach($),n.allOf&&n.allOf.forEach($))}var le={maxBytes:49e3,systemMaxChars:12e3},P=()=>process.env.ZIBBY_VERBOSE==="true"||process.env.ZIBBY_DEBUG==="true",Ee=n=>Buffer.byteLength(JSON.stringify(n),"utf8");function G(n){return Array.isArray(n)?new Set(n.map(e=>String(e||"").trim()).filter(Boolean)):new Set}function Pe(n){return Array.isArray(n)?n.map(e=>String(e||"").trim()).filter(Boolean):[]}var ae=class extends Ae{#e;#o;#t=new j;constructor(e=null){super("assistant","Zibby Assistant",200),e&&typeof e=="object"&&(e.toolProvider||e.completionProvider)?(this.#e=e.toolProvider||new I,this.#o=e.completionProvider||new E):(this.#e=e||new I,this.#o=new E)}canHandle(e){return H().ok}async invoke(e,o={}){let s=te("OPENAI_API_KEY");try{let t=o.model&&o.model!=="auto"?o.model:Ce,i=H();if(!i.ok)throw i.reason==="missing_session_token"?new Error("Login required. Run `zibby login` to authenticate."):i.reason==="missing_openai_api_key"?new Error("Missing OPENAI_API_KEY for BYOK mode."):i.reason==="missing_openai_proxy_url"?new Error("Missing OPENAI_PROXY_URL for local mode."):new Error(`Assistant auth unavailable (${i.reason}).`);let a=o.messages||[{role:"user",content:e}],l=i.baseUrl,r=P();if(r?await this.#p(a,t,l,i.tokenPreview||"none"):C.debug(`[Assistant] ${l} | model: ${t} | messages: ${a.length}`),o.schema)return this.#c(t,a,i,o);let c=o.activeSkills||[],u=this.#d(o),f=this.#f(c,u),b=this.#m(f),_=this.#u(o),x={...o,payloadCompaction:_},w={activeSkills:f,options:x,executionRegistry:b,capabilityPolicy:u},O=!!o.stream,m={model:t,messages:[...a],stream:!1};this.#s(m,f,u);for(let S=0;S<Ie;S++){if(o.signal?.aborted){let d=new Error("Aborted via signal");throw d.name="AbortError",d}let T=await oe({body:m,auth:i,options:x,streaming:O,toolContext:w,activeSkills:f,round:S,verbose:r,dependencies:{fetchCompletion:this.#i.bind(this),fetchStreamingCompletion:this.#r.bind(this),onFallbackLog:(d,p)=>{P()&&console.log(`413 fallback: messages ${d.messages.length} -> ${p.messages.length}, bytes=${Ee(p)}`)},hasToolCalls:d=>this.#e.hasToolCalls(d),getTextContent:d=>this.#e.getTextContent(d),parseToolCalls:d=>this.#e.parseToolCalls(d),buildAssistantMessage:d=>this.#e.buildAssistantMessage(d),buildToolResultMessage:(d,p)=>this.#e.buildToolResultMessage(d,p),executeTool:(d,p)=>this.#a(d,p),onToolCallLog:async d=>{let p=(await import("chalk")).default;console.log(p.dim(` ${d.map(g=>`${g.name}(${JSON.stringify(g.args).slice(0,80)})`).join(", ")}`))},injectTools:(d,p)=>this.#s(d,p,u)},config:{maxToolResultChars:o.maxToolResultChars||3e3}});if(T.done)return T.text;m.messages=T.body.messages,T.body.tools?m.tools=T.body.tools:delete m.tools}return"I hit my tool-calling limit for this turn. Please try again."}finally{s.dispose()}}async cleanup(){await this.#t.stopAll()}#s(e,o,s=null){let t=this.#l(o,s),i=this.#e.formatTools(t);this.#e.injectToolsIntoBody(e,i)}#l(e,o=null){let s=[],t=new Set;for(let i of e){let a=F(i);if(a?.tools?.length)for(let l of a.tools)this.#n(l.name,o)&&(t.has(l.name)||(t.add(l.name),s.push({name:l.name,description:l.description,parameters:l.parameters||l.input_schema||{type:"object",properties:{}}})))}return!o?.disableSkillContextTool&&this.#n(Z,o)&&s.push({name:Z,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"]}}),s}async#a(e,o){let{activeSkills:s,options:t,executionRegistry:i,capabilityPolicy:a}=o;if(!this.#n(e.name,a))return`Tool "${e.name}" blocked by policy`;if(e.name===Z){let c=String(e.args?.skillId||"").trim();if(!c)return JSON.stringify({error:"skillId is required"});if(!s.includes(c))return JSON.stringify({error:`Skill "${c}" is not active`,activeSkills:s});let u=F(c);if(!u)return JSON.stringify({error:`Skill "${c}" not found`});let f=typeof u.promptFragment=="function"?u.promptFragment():u.promptFragment||"",b=(u.tools||[]).map(x=>x.name),_=JSON.stringify({skillId:c,description:u.description||"",toolNames:b,promptFragment:f||""});return P()&&(console.log(`
|
|
8
|
+
\u{1F4D6} get_skill_context("${c}") \u2192 ${_.length} chars (fragment: ${f.length} chars)`),console.log(` tools: [${b.join(", ")}]`),console.log(` fragment preview: ${f.slice(0,200).replace(/\n/g,"\\n")}\u2026
|
|
9
|
+
`)),_}let l=i?.get(e.name)||null;if(!l)return`Unknown tool: ${e.name}`;let r=F(l.skillId);if(!r)return`Skill "${l.skillId}" not found for tool "${e.name}"`;if(l.mode==="handler")try{return r.handleToolCall(e.name,e.args,o)}catch(c){return`Error in ${e.name}: ${c.message}`}if(l.mode==="mcp")try{if(!this.#t.isRunning(r.serverName)){let u=r.resolve(t);if(!u)return`Skill "${l.skillId}" is not available (cannot start server)`;await this.#t.ensureServer(r.serverName,u)}let c=await this.#t.callTool(r.serverName,e.name,e.args);return c.text||(c.isError?"Tool call failed":"Done")}catch(c){return`MCP error (${r.serverName}): ${c.message}`}return`Skill "${l.skillId}" owns tool "${e.name}" but has no execution mode`}async#r(e,o,s){return this.#o.fetchStreamingCompletion(e,o,{...s,onBudget:({beforeBytes:t,meta:i})=>{P()&&console.log(`payload bytes (stream) before=${t} after=${i.bytes} trimmed=${i.trimmed} messages=${i.messageCount}`)}})}async#i(e,o,s){return this.#o.fetchCompletion(e,o,{...s,onBudget:({beforeBytes:t,meta:i})=>{P()&&console.log(`payload bytes before=${t} after=${i.bytes} trimmed=${i.trimmed} messages=${i.messageCount}`)}})}async#c(e,o,s,t){let{zodToJson:i}=await Promise.resolve().then(()=>(ie(),se)),a=typeof t.schema?.parse=="function",l=a?i(t.schema,{}):t.schema;delete l.$schema,$(l);let r={model:e,messages:o,stream:!1,response_format:{type:"json_schema",json_schema:{name:"extract",schema:l,strict:!0}}},c=await this.#i(r,s,t),u=this.#e.getTextContent(c),f=JSON.parse(u),b=a?t.schema.parse(f):f;return{raw:u,structured:b}}#u(e={}){let o=e?.config?.agent?.assistant?.payloadCompaction||{};return{maxBytes:Number(o.maxBytes||e.maxPayloadBytes||le.maxBytes),systemMaxChars:Number(o.systemMaxChars||le.systemMaxChars)}}#d(e={}){let o=e?.config?.agent?.assistant?.toolPolicy||{};return{allowTools:G(o.allowTools||e.allowTools),denyTools:G(o.denyTools||e.denyTools),denyPrefixes:Pe(o.denyPrefixes||e.denyToolPrefixes),includeSkills:G(o.includeSkills||e.includeSkills),excludeSkills:G(o.excludeSkills||e.excludeSkills),disableSkillContextTool:!!(o.disableSkillContextTool||e.disableSkillContextTool)}}#f(e,o){let s=o?.includeSkills||new Set,t=o?.excludeSkills||new Set;return s.size===0&&t.size===0?e:e.filter(i=>!(s.size>0&&!s.has(i)||t.has(i)))}#n(e,o){let s=String(e||"").trim();if(!s)return!1;let t=o?.allowTools;if(t&&t.size>0&&!t.has(s))return!1;let i=o?.denyTools;return!(i&&i.has(s)||(o?.denyPrefixes||[]).some(l=>s.startsWith(l)))}#m(e){let o=new Map,s=[];for(let t of e){let i=F(t);if(!i?.tools?.length)continue;let a=typeof i.handleToolCall=="function"?"handler":i.serverName&&typeof i.resolve=="function"?"mcp":null;if(a)for(let l of i.tools){let r=String(l?.name||"").trim();if(r){if(o.has(r)){s.push({tool:r,winner:o.get(r).skillId,skipped:t});continue}o.set(r,{skillId:t,mode:a})}}}if(s.length>0&&P()){let t=s.slice(0,5).map(i=>`${i.tool}:${i.winner}>${i.skipped}`).join(", ");console.log(`tool registry collisions: ${t}${s.length>5?" ...":""}`)}return o}async#p(e,o,s,t){console.log(`
|
|
10
|
+
\u25C6 Model: ${o} | proxy: ${s} | token: ${t||"none"}
|
|
11
|
+
`);let i=(await import("chalk")).default;console.log(i.bold("Prompt sent to LLM:")),console.log(i.dim("\u2500".repeat(60)));let a=!1;for(let l of e)if(l.role==="system")console.log(i.dim(`[System] ${l.content||""}`));else{a||(console.log(i.dim("\u2500\u2500\u2500 chat history \u2500\u2500\u2500")),a=!0);let r=l.role==="user"?"Human":"AI",c=l.content?.length>200?`${l.content.slice(0,200)}...`:l.content||"";console.log(i.dim(`[${r}] ${c}`))}console.log(i.dim("\u2500".repeat(60)))}};export{ae as AssistantStrategy};
|
|
@@ -1,3 +1,3 @@
|
|
|
1
|
-
import{AgentStrategy as le,getSkill as re}from"@zibby/agent-workflow";import{query as ce}from"@anthropic-ai/claude-agent-sdk";import{zodToJsonSchema as ue}from"zod-to-json-schema";import{writeFileSync as R,mkdirSync as U,existsSync as Q,readFileSync as ee}from"fs";import{homedir as oe}from"os";import{join as v}from"path";import I from"chalk";var g={debug:0,info:1,warn:2,error:3,silent:4},N=class{constructor(){this._level=this._getLogLevel()}_getLogLevel(){if(process.env.ZIBBY_DEBUG==="true")return g.debug;if(process.env.ZIBBY_VERBOSE==="true")return g.info;let e=process.env.LOG_LEVEL?.toLowerCase();return e&&e in g?g[e]:g.info}_shouldLog(e){return g[e]>=this._level}_formatMessage(e,n,i={}){let s=new Date().toISOString(),d=`${this._getPrefix(e)} ${n}`;return Object.keys(i).length>0&&(d+=I.dim(` ${JSON.stringify(i)}`)),d}_getPrefix(e){return{debug:I.gray("[DEBUG]"),info:I.cyan("[INFO]"),warn:I.yellow("[WARN]"),error:I.red("\u274C [ERROR]")}[e]||""}debug(e,n){this._shouldLog("debug")&&console.log(this._formatMessage("debug",e,n))}info(e,n){this._shouldLog("info")&&console.log(this._formatMessage("info",e,n))}warn(e,n){this._shouldLog("warn")&&console.warn(this._formatMessage("warn",e,n))}error(e,n){this._shouldLog("error")&&console.error(this._formatMessage("error",e,n))}setLevel(e){e in g&&(this._level=g[e])}getLevel(){return Object.keys(g).find(e=>g[e]===this._level)}},a=new N;import{timeline as $,Timeline as he,WORKFLOW_GRAPH_LOG_MARKER_PREFIX as ye}from"@zibby/agent-workflow";var ie=[{alias:"auto",canonicalId:"claude-sonnet-4-6",vendor:"claude",label:"Claude \xB7 Sonnet 4.6",family:"claude-sonnet-4",pickerVisible:!0,isDefault:!0},{alias:"opus-4.8",canonicalId:"claude-opus-4-8",vendor:"claude",label:"Claude \xB7 Opus 4.8",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"sonnet-4.8",canonicalId:"claude-sonnet-4-8",vendor:"claude",label:"Claude \xB7 Sonnet 4.8",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"opus-4.7",canonicalId:"claude-opus-4-7",vendor:"claude",label:"Claude \xB7 Opus 4.7",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"sonnet-4.7",canonicalId:"claude-sonnet-4-7",vendor:"claude",label:"Claude \xB7 Sonnet 4.7",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"sonnet-4.6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"opus-4.6",canonicalId:"claude-opus-4-6",vendor:"claude",label:"Claude \xB7 Opus 4.6",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"sonnet-4.5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",label:"Claude \xB7 Sonnet 4.5",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"opus-4.5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",label:"Claude \xB7 Opus 4.5",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"haiku-4.5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",label:"Claude \xB7 Haiku 4.5",family:"claude-haiku-4"},{alias:"haiku-4-5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"claude-sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"claude-opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"claude-sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"claude-sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"claude-opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"claude-sonnet-4-5-20250929",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"claude-opus-4-5-20251101",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"claude-haiku-4-5-20251001",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-20250514",canonicalId:"claude-opus-4-20250514",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"auto",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",isDefault:!0},{alias:"o4-mini",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",pickerVisible:!0},{alias:"o3",canonicalId:"o3",vendor:"codex",family:"o3"},{alias:"o3-mini",canonicalId:"o3-mini",vendor:"codex",family:"o3-mini"},{alias:"codex-mini",canonicalId:"codex-mini-latest",vendor:"codex",family:"codex-mini"},{alias:"gpt-4o",canonicalId:"gpt-4o",vendor:"codex",label:"Codex \xB7 GPT-4o",family:"gpt-4o",pickerVisible:!0},{alias:"gpt-4o-mini",canonicalId:"gpt-4o-mini",vendor:"codex",label:"Codex \xB7 GPT-4o mini",family:"gpt-4o-mini",pickerVisible:!0},{alias:"gpt-5.2-codex",canonicalId:"gpt-5.2-codex",vendor:"codex",label:"Codex \xB7 GPT-5.2",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.2",canonicalId:"gpt-5.2",vendor:"codex",label:"Codex \xB7 GPT-5.2 base",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3",canonicalId:"gpt-5.3",vendor:"codex",label:"Codex \xB7 GPT-5.3",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3-codex",canonicalId:"gpt-5.3-codex",vendor:"codex",family:"gpt-5"},{alias:"gpt-5.4",canonicalId:"gpt-5.4",vendor:"codex",label:"Codex \xB7 GPT-5.4",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.5",canonicalId:"gpt-5.5",vendor:"codex",label:"Codex \xB7 GPT-5.5",family:"gpt-5",pickerVisible:!0},{alias:"auto",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",isDefault:!0},{alias:"gemini-3-pro",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",pickerVisible:!0},{alias:"gemini-3-flash",canonicalId:"gemini-3-flash",vendor:"gemini",label:"Gemini \xB7 3 Flash",family:"gemini-3",pickerVisible:!0},{alias:"gemini-2.5-pro",canonicalId:"gemini-2.5-pro",vendor:"gemini",label:"Gemini \xB7 2.5 Pro",family:"gemini-2.5",pickerVisible:!0},{alias:"gemini-2.5-flash",canonicalId:"gemini-2.5-flash",vendor:"gemini",label:"Gemini \xB7 2.5 Flash",family:"gemini-2.5",pickerVisible:!0}];function w(r){let e={};for(let n of ie)n.vendor===r&&(e[n.alias]=n.canonicalId);return e}var P={ASSISTANT:"gpt-5.4-nano-2026-03-17",CLAUDE:"claude-sonnet-4-6",CURSOR:"auto",CODEX:"o4-mini",GEMINI:"gemini-3-pro",OPENAI_POSTPROCESSING:"gpt-4o-mini"};var M=w("claude"),Ee=w("codex"),be=w("gemini"),Se={CURSOR_AGENT_DEFAULT:1200*1e3,OPENAI_REQUEST:18e4};function D(r){let e=`${r}_POOL`,n=process.env[e];if(!n||typeof n!="string")return{picked:null,count:0,dispose:()=>{}};let i=n.split(/[,\n]+/).map(d=>d.trim()).filter(Boolean);if(i.length===0)return{picked:null,count:0,dispose:()=>{}};let s=i[Math.floor(Math.random()*i.length)],t=process.env[r];return process.env[r]=s,a.debug(`[auth-pool] ${r}: picked 1 of ${i.length} from pool (***${s.slice(-4)})`),{picked:s,count:i.length,dispose(){t===void 0?delete process.env[r]:process.env[r]=t}}}var z=class extends le{constructor(){super("claude","Claude (Anthropic API)",50)}canHandle(e){let n=!!process.env.ANTHROPIC_API_KEY||!!process.env.ANTHROPIC_API_KEY_POOL,i=!!process.env.CLAUDE_CODE_OAUTH_TOKEN||!!process.env.CLAUDE_CODE_OAUTH_TOKEN_POOL,s=!!process.env.ANTHROPIC_AUTH_TOKEN||!!process.env.ANTHROPIC_AUTH_TOKEN_POOL,t=n||i||s;return t||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)."),t}async invoke(e,n={}){let{model:i,workspace:s=process.cwd(),schema:t=null,images:d=[],skills:p=null,sessionPath:O=null,nodeName:C=null,timeout:K,config:G={},signal:A=null,sessionId:J,resume:F,forkSession:ne,sessionStore:H}=n,m=i;(!m||m==="auto")&&(a.debug(`Model is '${m||"undefined"}', using default: ${P.CLAUDE}`),m=P.CLAUDE);let E=M[m]||m;M[m]&&m!==E&&a.debug(`Mapped model: ${m} \u2192 ${E}`);let te=[D("ANTHROPIC_API_KEY"),D("CLAUDE_CODE_OAUTH_TOKEN"),D("ANTHROPIC_AUTH_TOKEN")];try{a.debug(`Invoking Claude Agent SDK with model: ${E}, skills: ${JSON.stringify(p)}`);let h=process.env.CLAUDE_CODE_OAUTH_TOKEN,V=process.env.ANTHROPIC_API_KEY,j=process.env.ANTHROPIC_AUTH_TOKEN;process.env.CLAUDE_CODE_CREDENTIALS_JSON?fe(process.env.CLAUDE_CODE_CREDENTIALS_JSON):h&&pe(h,process.env.__ZIBBY_CLAUDE_PLAN);let L=process.env.__ZIBBY_CLAUDE_PLAN,B=L?L.charAt(0).toUpperCase()+L.slice(1):null,b;h?b=B?`Claude ${B} plan \xB7 ***${h.slice(-4)}`:`Subscription \xB7 ***${h.slice(-4)}`:V?b=`API key \xB7 ***${V.slice(-4)}`:j?b=`Bearer \xB7 ***${j.slice(-4)}`:b="NONE \u2014 request will fail";let _=(await import("chalk")).default;console.log(""),console.log(` ${_.cyan("\u25C6")} ${_.bold(E)}`),console.log(` ${_.dim("Auth: ")}${b}`),console.log(` ${_.dim("Workdir: ")}${s}`),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:W,mcpServers:S}=this._resolveSkills(p,{sessionPath:O,workspace:s,nodeName:C});try{let f={cwd:s,allowedTools:W,settings:{permissions:{allow:["Bash(*)","Read(*)","Write(*)","Edit(*)","MultiEdit(*)","NotebookEdit(*)","Grep(*)","Glob(*)","WebFetch(*)","WebSearch(*)","Task(*)","TodoWrite(*)","Skill(*)"]}},model:E,...Object.keys(S).length>0&&{mcpServers:S},...J&&{sessionId:J},...F&&{resume:F},...ne&&{forkSession:!0},...H&&{sessionStore:H}};if(t){let y=typeof t.parse=="function"?ue(t,{target:"openApi3"}):t;f.outputFormat={type:"json_schema",schema:y},a.debug("Structured output enforced via SDK outputFormat")}a.debug(`Agent SDK options: ${JSON.stringify({cwd:f.cwd,toolCount:W.length,permissionMode:f.permissionMode,model:f.model,hasOutputFormat:!!f.outputFormat})}`);let T="",Y=0,Z=[];a.debug("Starting Claude Agent SDK query stream");let q;try{q=ce({prompt:e,options:f})}catch(o){throw a.error(`Failed to initialize Claude Agent SDK: ${o.message}`),o}let X=null,x=0,se=3;if(A?.aborted){let o=new Error("Aborted via signal");throw o.name="AbortError",o}try{for await(let o of q){if(A?.aborted){let c=new Error("Aborted via signal");throw c.name="AbortError",c}if(Z.push(o),o.type==="system"&&o.subtype==="init"){let c=o.mcp_servers||[];if(c.length>0){let l=c.map(u=>`${u.name}=${u.status}${u.error?` (${String(u.error).slice(0,120)})`:""}`).join(", ");a.info(`[mcp] SDK init \u2014 ${c.length} server(s): ${l}`);for(let u of c)u.status!=="connected"&&a.warn(`[mcp] server '${u.name}' did not connect: status=${u.status}`,{name:u.name,status:u.status,error:u.error})}else Object.keys(S||{}).length>0&&a.warn(`[mcp] SDK init returned no mcp_servers despite passing ${Object.keys(S).length} server(s) \u2014 SDK version may not expose this field, or config was dropped`);if(Array.isArray(o.tools)){let l=o.tools.filter(u=>u.startsWith("mcp__"));l.length>0?a.info(`[mcp] tools visible to model: ${l.join(", ")}`):Object.keys(S||{}).length>0&&a.warn("[mcp] no mcp__* tools in model's tool list \u2014 MCP servers configured but tools didn't reach the prompt")}}if(o.type==="error"||o.error){let c=o.error?.message||o.error||o.message||"Unknown API error";try{a.error("SDK error event raw payload",{fullMessage:JSON.stringify(o,null,2)})}catch{}throw new Error(typeof c=="string"?c:JSON.stringify(c))}let y=JSON.stringify(o.message?.content||o.text||"").slice(0,200);if(y===X){if(x++,x>=se){let c=(o.message?.content?.[0]?.text||o.text||"unknown").slice(0,100);throw new Error(`API stuck in loop (${x}x repeated): ${c}`)}}else X=y,x=1;if(o.type==="assistant"||o.constructor?.name==="AssistantMessage"){let c=o.message?.content||o.content||[];for(let l of c)if(l.type==="thinking"&&l.thinking)console.log(`${l.thinking.substring(0,200)}${l.thinking.length>200?"...":""}`);else if(l.type==="text"&&l.text)T+=l.text,l.text.length<500?console.log(`${l.text}`):console.log(`${l.text.substring(0,200)}... (${l.text.length} chars)`);else if(l.type==="tool_use"){Y++,l.name.includes("memory")?$.stepMemory(`Tool: ${l.name}`):$.stepTool(`Tool: ${l.name}`);let ae=JSON.stringify(l.input).substring(0,100);console.log(` Input: ${ae}${JSON.stringify(l.input).length>100?"...":""}`)}}else if(!(o.type==="user"&&o.tool_use_result)){if(o.type==="result"||o.constructor?.name==="ResultMessage"){let c=o.result||o.text||o.content||T;try{de({message:o,sessionPath:O,nodeName:C,model:m,toolCallCount:Y})}catch(l){a.debug(`usage.json write failed: ${l.message}`)}if(t){if(o.structured_output){a.debug("Using SDK native structured_output");let u=typeof t.parse=="function"?t.parse(o.structured_output):o.structured_output;return{raw:c,structured:u}}if(c){let l=this._extractJson(c,t);if(l)return{raw:c,structured:l}}a.warn(`Could not extract structured output \u2014 returning raw text (${(c||"").length} chars)`)}return c||""}}}if(a.warn(`Agent SDK ended without result. Collected ${Z.length} messages`),T.length>0)return a.debug("Returning accumulated text from messages"),T;throw new Error("Claude Agent SDK query ended without result")}catch(o){let y={message:o?.message,name:o?.name,code:o?.code,cause:o?.cause?.message||o?.cause,stack:o?.stack?.split(`
|
|
1
|
+
import{AgentStrategy as ue,getSkill as de}from"@zibby/agent-workflow";import{query as pe}from"@anthropic-ai/claude-agent-sdk";import{z as X}from"zod";import{zodToJsonSchema as le}from"zod-to-json-schema";function Q(l,e={target:"openApi3"}){if(!l)return null;if(typeof X?.toJSONSchema=="function")try{return X.toJSONSchema(l)}catch{}try{return le(l,e)}catch{return null}}import{writeFileSync as R,mkdirSync as U,existsSync as oe,readFileSync as ne}from"fs";import{homedir as te}from"os";import{join as v}from"path";import I from"chalk";var g={debug:0,info:1,warn:2,error:3,silent:4},D=class{constructor(){this._level=this._getLogLevel()}_getLogLevel(){if(process.env.ZIBBY_DEBUG==="true")return g.debug;if(process.env.ZIBBY_VERBOSE==="true")return g.info;let e=process.env.LOG_LEVEL?.toLowerCase();return e&&e in g?g[e]:g.info}_shouldLog(e){return g[e]>=this._level}_formatMessage(e,n,i={}){let s=new Date().toISOString(),d=`${this._getPrefix(e)} ${n}`;return Object.keys(i).length>0&&(d+=I.dim(` ${JSON.stringify(i)}`)),d}_getPrefix(e){return{debug:I.gray("[DEBUG]"),info:I.cyan("[INFO]"),warn:I.yellow("[WARN]"),error:I.red("\u274C [ERROR]")}[e]||""}debug(e,n){this._shouldLog("debug")&&console.log(this._formatMessage("debug",e,n))}info(e,n){this._shouldLog("info")&&console.log(this._formatMessage("info",e,n))}warn(e,n){this._shouldLog("warn")&&console.warn(this._formatMessage("warn",e,n))}error(e,n){this._shouldLog("error")&&console.error(this._formatMessage("error",e,n))}setLevel(e){e in g&&(this._level=g[e])}getLevel(){return Object.keys(g).find(e=>g[e]===this._level)}},a=new D;import{timeline as $,Timeline as Ee,WORKFLOW_GRAPH_LOG_MARKER_PREFIX as Ae}from"@zibby/agent-workflow";var ce=[{alias:"auto",canonicalId:"claude-sonnet-4-6",vendor:"claude",label:"Claude \xB7 Sonnet 4.6",family:"claude-sonnet-4",pickerVisible:!0,isDefault:!0},{alias:"opus-4.8",canonicalId:"claude-opus-4-8",vendor:"claude",label:"Claude \xB7 Opus 4.8",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"sonnet-4.8",canonicalId:"claude-sonnet-4-8",vendor:"claude",label:"Claude \xB7 Sonnet 4.8",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"opus-4.7",canonicalId:"claude-opus-4-7",vendor:"claude",label:"Claude \xB7 Opus 4.7",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"sonnet-4.7",canonicalId:"claude-sonnet-4-7",vendor:"claude",label:"Claude \xB7 Sonnet 4.7",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"sonnet-4.6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"opus-4.6",canonicalId:"claude-opus-4-6",vendor:"claude",label:"Claude \xB7 Opus 4.6",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"sonnet-4.5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",label:"Claude \xB7 Sonnet 4.5",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"opus-4.5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",label:"Claude \xB7 Opus 4.5",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"haiku-4.5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",label:"Claude \xB7 Haiku 4.5",family:"claude-haiku-4"},{alias:"haiku-4-5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"claude-sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"claude-opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"claude-sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"claude-sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"claude-opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"claude-sonnet-4-5-20250929",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"claude-opus-4-5-20251101",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"claude-haiku-4-5-20251001",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-20250514",canonicalId:"claude-opus-4-20250514",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"auto",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",isDefault:!0},{alias:"o4-mini",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",pickerVisible:!0},{alias:"o3",canonicalId:"o3",vendor:"codex",family:"o3"},{alias:"o3-mini",canonicalId:"o3-mini",vendor:"codex",family:"o3-mini"},{alias:"codex-mini",canonicalId:"codex-mini-latest",vendor:"codex",family:"codex-mini"},{alias:"gpt-4o",canonicalId:"gpt-4o",vendor:"codex",label:"Codex \xB7 GPT-4o",family:"gpt-4o",pickerVisible:!0},{alias:"gpt-4o-mini",canonicalId:"gpt-4o-mini",vendor:"codex",label:"Codex \xB7 GPT-4o mini",family:"gpt-4o-mini",pickerVisible:!0},{alias:"gpt-5.2-codex",canonicalId:"gpt-5.2-codex",vendor:"codex",label:"Codex \xB7 GPT-5.2",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.2",canonicalId:"gpt-5.2",vendor:"codex",label:"Codex \xB7 GPT-5.2 base",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3",canonicalId:"gpt-5.3",vendor:"codex",label:"Codex \xB7 GPT-5.3",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3-codex",canonicalId:"gpt-5.3-codex",vendor:"codex",family:"gpt-5"},{alias:"gpt-5.4",canonicalId:"gpt-5.4",vendor:"codex",label:"Codex \xB7 GPT-5.4",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.5",canonicalId:"gpt-5.5",vendor:"codex",label:"Codex \xB7 GPT-5.5",family:"gpt-5",pickerVisible:!0},{alias:"auto",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",isDefault:!0},{alias:"gemini-3-pro",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",pickerVisible:!0},{alias:"gemini-3-flash",canonicalId:"gemini-3-flash",vendor:"gemini",label:"Gemini \xB7 3 Flash",family:"gemini-3",pickerVisible:!0},{alias:"gemini-2.5-pro",canonicalId:"gemini-2.5-pro",vendor:"gemini",label:"Gemini \xB7 2.5 Pro",family:"gemini-2.5",pickerVisible:!0},{alias:"gemini-2.5-flash",canonicalId:"gemini-2.5-flash",vendor:"gemini",label:"Gemini \xB7 2.5 Flash",family:"gemini-2.5",pickerVisible:!0}];function w(l){let e={};for(let n of ce)n.vendor===l&&(e[n.alias]=n.canonicalId);return e}var P={ASSISTANT:"gpt-5.4-nano-2026-03-17",CLAUDE:"claude-sonnet-4-6",CURSOR:"auto",CODEX:"o4-mini",GEMINI:"gemini-3-pro",OPENAI_POSTPROCESSING:"gpt-4o-mini"};var M=w("claude"),Ce=w("codex"),Te=w("gemini"),xe={CURSOR_AGENT_DEFAULT:1200*1e3,OPENAI_REQUEST:18e4};function N(l){let e=`${l}_POOL`,n=process.env[e];if(!n||typeof n!="string")return{picked:null,count:0,dispose:()=>{}};let i=n.split(/[,\n]+/).map(d=>d.trim()).filter(Boolean);if(i.length===0)return{picked:null,count:0,dispose:()=>{}};let s=i[Math.floor(Math.random()*i.length)],t=process.env[l];return process.env[l]=s,a.debug(`[auth-pool] ${l}: picked 1 of ${i.length} from pool (***${s.slice(-4)})`),{picked:s,count:i.length,dispose(){t===void 0?delete process.env[l]:process.env[l]=t}}}var ee=class extends ue{constructor(){super("claude","Claude (Anthropic API)",50)}canHandle(e){let n=!!process.env.ANTHROPIC_API_KEY||!!process.env.ANTHROPIC_API_KEY_POOL,i=!!process.env.CLAUDE_CODE_OAUTH_TOKEN||!!process.env.CLAUDE_CODE_OAUTH_TOKEN_POOL,s=!!process.env.ANTHROPIC_AUTH_TOKEN||!!process.env.ANTHROPIC_AUTH_TOKEN_POOL,t=n||i||s;return t||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)."),t}async invoke(e,n={}){let{model:i,workspace:s=process.cwd(),schema:t=null,images:d=[],skills:p=null,sessionPath:h=null,nodeName:C=null,timeout:K,config:J={},signal:b=null,sessionId:G,resume:F,forkSession:se,sessionStore:V}=n,m=i;(!m||m==="auto")&&(a.debug(`Model is '${m||"undefined"}', using default: ${P.CLAUDE}`),m=P.CLAUDE);let S=M[m]||m;M[m]&&m!==S&&a.debug(`Mapped model: ${m} \u2192 ${S}`);let ae=[N("ANTHROPIC_API_KEY"),N("CLAUDE_CODE_OAUTH_TOKEN"),N("ANTHROPIC_AUTH_TOKEN")];try{a.debug(`Invoking Claude Agent SDK with model: ${S}, skills: ${JSON.stringify(p)}`);let _=process.env.CLAUDE_CODE_OAUTH_TOKEN,H=process.env.ANTHROPIC_API_KEY,j=process.env.ANTHROPIC_AUTH_TOKEN;process.env.CLAUDE_CODE_CREDENTIALS_JSON?ge(process.env.CLAUDE_CODE_CREDENTIALS_JSON):_&&me(_,process.env.__ZIBBY_CLAUDE_PLAN);let L=process.env.__ZIBBY_CLAUDE_PLAN,B=L?L.charAt(0).toUpperCase()+L.slice(1):null,E;_?E=B?`Claude ${B} plan \xB7 ***${_.slice(-4)}`:`Subscription \xB7 ***${_.slice(-4)}`:H?E=`API key \xB7 ***${H.slice(-4)}`:j?E=`Bearer \xB7 ***${j.slice(-4)}`:E="NONE \u2014 request will fail";let O=(await import("chalk")).default;console.log(""),console.log(` ${O.cyan("\u25C6")} ${O.bold(S)}`),console.log(` ${O.dim("Auth: ")}${E}`),console.log(` ${O.dim("Workdir: ")}${s}`),console.log(""),console.log(O.bold("Prompt sent to LLM:")),console.log(O.dim("\u2500".repeat(60))),console.log(O.dim(e)),console.log(O.dim("\u2500".repeat(60)));let{allowedTools:W,mcpServers:A}=this._resolveSkills(p,{sessionPath:h,workspace:s,nodeName:C});try{let f={cwd:s,allowedTools:W,settings:{permissions:{allow:["Bash(*)","Read(*)","Write(*)","Edit(*)","MultiEdit(*)","NotebookEdit(*)","Grep(*)","Glob(*)","WebFetch(*)","WebSearch(*)","Task(*)","TodoWrite(*)","Skill(*)"]}},model:S,...Object.keys(A).length>0&&{mcpServers:A},...G&&{sessionId:G},...F&&{resume:F},...se&&{forkSession:!0},...V&&{sessionStore:V}};if(t){let y=typeof t.parse=="function"?Q(t):t;f.outputFormat={type:"json_schema",schema:y},a.debug("Structured output enforced via SDK outputFormat")}a.debug(`Agent SDK options: ${JSON.stringify({cwd:f.cwd,toolCount:W.length,permissionMode:f.permissionMode,model:f.model,hasOutputFormat:!!f.outputFormat})}`);let T="",Y=0,Z=[];a.debug("Starting Claude Agent SDK query stream");let q;try{q=pe({prompt:e,options:f})}catch(o){throw a.error(`Failed to initialize Claude Agent SDK: ${o.message}`),o}let z=null,x=0,ie=3;if(b?.aborted){let o=new Error("Aborted via signal");throw o.name="AbortError",o}try{for await(let o of q){if(b?.aborted){let c=new Error("Aborted via signal");throw c.name="AbortError",c}if(Z.push(o),o.type==="system"&&o.subtype==="init"){let c=o.mcp_servers||[];if(c.length>0){let r=c.map(u=>`${u.name}=${u.status}${u.error?` (${String(u.error).slice(0,120)})`:""}`).join(", ");a.info(`[mcp] SDK init \u2014 ${c.length} server(s): ${r}`);for(let u of c)u.status!=="connected"&&a.warn(`[mcp] server '${u.name}' did not connect: status=${u.status}`,{name:u.name,status:u.status,error:u.error})}else Object.keys(A||{}).length>0&&a.warn(`[mcp] SDK init returned no mcp_servers despite passing ${Object.keys(A).length} server(s) \u2014 SDK version may not expose this field, or config was dropped`);if(Array.isArray(o.tools)){let r=o.tools.filter(u=>u.startsWith("mcp__"));r.length>0?a.info(`[mcp] tools visible to model: ${r.join(", ")}`):Object.keys(A||{}).length>0&&a.warn("[mcp] no mcp__* tools in model's tool list \u2014 MCP servers configured but tools didn't reach the prompt")}}if(o.type==="error"||o.error){let c=o.error?.message||o.error||o.message||"Unknown API error";try{a.error("SDK error event raw payload",{fullMessage:JSON.stringify(o,null,2)})}catch{}throw new Error(typeof c=="string"?c:JSON.stringify(c))}let y=JSON.stringify(o.message?.content||o.text||"").slice(0,200);if(y===z){if(x++,x>=ie){let c=(o.message?.content?.[0]?.text||o.text||"unknown").slice(0,100);throw new Error(`API stuck in loop (${x}x repeated): ${c}`)}}else z=y,x=1;if(o.type==="assistant"||o.constructor?.name==="AssistantMessage"){let c=o.message?.content||o.content||[];for(let r of c)if(r.type==="thinking"&&r.thinking)console.log(`${r.thinking.substring(0,200)}${r.thinking.length>200?"...":""}`);else if(r.type==="text"&&r.text)T+=r.text,r.text.length<500?console.log(`${r.text}`):console.log(`${r.text.substring(0,200)}... (${r.text.length} chars)`);else if(r.type==="tool_use"){Y++,r.name.includes("memory")?$.stepMemory(`Tool: ${r.name}`):$.stepTool(`Tool: ${r.name}`);let re=JSON.stringify(r.input).substring(0,100);console.log(` Input: ${re}${JSON.stringify(r.input).length>100?"...":""}`)}}else if(!(o.type==="user"&&o.tool_use_result)){if(o.type==="result"||o.constructor?.name==="ResultMessage"){let c=o.result||o.text||o.content||T;try{fe({message:o,sessionPath:h,nodeName:C,model:m,toolCallCount:Y})}catch(r){a.debug(`usage.json write failed: ${r.message}`)}if(t){if(o.structured_output){a.debug("Using SDK native structured_output");let u=typeof t.parse=="function"?t.parse(o.structured_output):o.structured_output;return{raw:c,structured:u}}if(c){let r=this._extractJson(c,t);if(r)return{raw:c,structured:r}}a.warn(`Could not extract structured output \u2014 returning raw text (${(c||"").length} chars)`)}return c||""}}}if(a.warn(`Agent SDK ended without result. Collected ${Z.length} messages`),T.length>0)return a.debug("Returning accumulated text from messages"),T;throw new Error("Claude Agent SDK query ended without result")}catch(o){let y={message:o?.message,name:o?.name,code:o?.code,cause:o?.cause?.message||o?.cause,stack:o?.stack?.split(`
|
|
2
2
|
`).slice(0,5).join(`
|
|
3
|
-
`),stringified:(()=>{try{return JSON.stringify(o,Object.getOwnPropertyNames(Object(o)))}catch{return String(o)}})()};throw a.error(`Error during query stream: ${o?.message||o}`,y),o}}catch(f){throw a.error("Claude Agent SDK call failed",{message:f?.message,name:f?.name,code:f?.code,cause:f?.cause?.message||f?.cause}),f}}finally{for(let
|
|
3
|
+
`),stringified:(()=>{try{return JSON.stringify(o,Object.getOwnPropertyNames(Object(o)))}catch{return String(o)}})()};throw a.error(`Error during query stream: ${o?.message||o}`,y),o}}catch(f){throw a.error("Claude Agent SDK call failed",{message:f?.message,name:f?.name,code:f?.code,cause:f?.cause?.message||f?.cause}),f}}finally{for(let _ of ae)_.dispose()}}_resolveSkills(e,n){let i=["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:[...i],mcpServers:{}};let s=[...i],t={};for(let d of e){let p=de(d);if(!p){a.warn(`Unknown skill "${d}" \u2014 skipping`);continue}if(p.allowedTools&&s.push(...p.allowedTools),typeof p.resolve=="function"){let h=p.resolve(n);h&&(t[p.serverName]=h,a.debug(`MCP: ${p.serverName} \u2192 ${h.command} ${h.args[0]}`))}}return{allowedTools:s,mcpServers:t}}_extractJson(e,n){let i=[()=>{if(e.includes("===JSON_START===")){let s=e.indexOf("===JSON_START===")+16,t=e.indexOf("===JSON_END===");return e.substring(s,t).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 s=e.indexOf("{"),t=e.lastIndexOf("}");if(s!==-1&&t>s)return e.substring(s,t+1)}];for(let s of i)try{let t=s();if(!t)continue;let d=JSON.parse(t);if(typeof d!="object"||d===null)continue;return typeof n.parse=="function"?n.parse(d):d}catch{}return null}};function fe({message:l,sessionPath:e,nodeName:n,model:i,toolCallCount:s}){if(!e||!n)return;let t=l?.usage||l?.message?.usage||{},d=k(t.input_tokens,t.inputTokens),p=k(t.output_tokens,t.outputTokens),h=k(t.cache_read_input_tokens,t.cacheReadInputTokens),C=k(t.cache_creation_input_tokens,t.cacheCreationInputTokens),K=k(l?.duration_ms,l?.durationMs),J={model:i||null,input_tokens:d,output_tokens:p,cache_read_tokens:h,cache_creation_tokens:C,tool_call_count:typeof s=="number"?s:null,duration_ms:K,captured_at:new Date().toISOString()},b=v(e,n);try{U(b,{recursive:!0})}catch{}R(v(b,"usage.json"),JSON.stringify(J,null,2),"utf-8")}function k(...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 me(l,e){let n=v(te(),".claude"),i=v(n,".credentials.json"),s={claudeAiOauth:{accessToken:l,...e?{subscriptionType:String(e).toLowerCase()}:{}}};try{if(oe(i)&&JSON.parse(ne(i,"utf-8"))?.claudeAiOauth?.accessToken===l)return;U(n,{recursive:!0}),R(i,JSON.stringify(s,null,2),"utf-8"),a.debug(`Wrote Claude Code OAuth credentials to ${i}`)}catch(t){a.warn(`Could not write ${i}: ${t.message} \u2014 Claude SDK may fail to authenticate.`)}}function ge(l){let e=v(te(),".claude"),n=v(e,".credentials.json"),i;try{i=JSON.parse(l)}catch(s){a.warn(`CLAUDE_CODE_CREDENTIALS_JSON not valid JSON: ${s.message}`);return}if(!i?.claudeAiOauth?.accessToken){a.warn("CLAUDE_CODE_CREDENTIALS_JSON missing claudeAiOauth.accessToken \u2014 skipping write.");return}try{if(oe(n)){let s=JSON.parse(ne(n,"utf-8"));if(s?.claudeAiOauth?.accessToken===i.claudeAiOauth.accessToken&&s?.claudeAiOauth?.expiresAt===i.claudeAiOauth.expiresAt)return}U(e,{recursive:!0}),R(n,JSON.stringify(i,null,2),"utf-8"),a.debug(`Wrote full Claude Code credentials blob to ${n}`)}catch(s){a.warn(`Could not write ${n}: ${s.message} \u2014 Claude SDK may fail to authenticate.`)}}export{ee as ClaudeAgentStrategy};
|
|
@@ -1,4 +1,4 @@
|
|
|
1
|
-
import{AgentStrategy as
|
|
2
|
-
\u25C6 Model: ${
|
|
1
|
+
import{AgentStrategy as j,getSkill as B}from"@zibby/agent-workflow";import{execSync as z}from"node:child_process";import{z as G}from"zod";import{zodToJsonSchema as F}from"zod-to-json-schema";function V(l,e={target:"openApi3"}){if(!l)return null;if(typeof G?.toJSONSchema=="function")try{return G.toJSONSchema(l)}catch{}try{return F(l,e)}catch{return null}}import y from"chalk";var u={debug:0,info:1,warn:2,error:3,silent:4},A=class{constructor(){this._level=this._getLogLevel()}_getLogLevel(){if(process.env.ZIBBY_DEBUG==="true")return u.debug;if(process.env.ZIBBY_VERBOSE==="true")return u.info;let e=process.env.LOG_LEVEL?.toLowerCase();return e&&e in u?u[e]:u.info}_shouldLog(e){return u[e]>=this._level}_formatMessage(e,o,s={}){let c=new Date().toISOString(),a=`${this._getPrefix(e)} ${o}`;return Object.keys(s).length>0&&(a+=y.dim(` ${JSON.stringify(s)}`)),a}_getPrefix(e){return{debug:y.gray("[DEBUG]"),info:y.cyan("[INFO]"),warn:y.yellow("[WARN]"),error:y.red("\u274C [ERROR]")}[e]||""}debug(e,o){this._shouldLog("debug")&&console.log(this._formatMessage("debug",e,o))}info(e,o){this._shouldLog("info")&&console.log(this._formatMessage("info",e,o))}warn(e,o){this._shouldLog("warn")&&console.warn(this._formatMessage("warn",e,o))}error(e,o){this._shouldLog("error")&&console.error(this._formatMessage("error",e,o))}setLevel(e){e in u&&(this._level=u[e])}getLevel(){return Object.keys(u).find(e=>u[e]===this._level)}},n=new A;import{timeline as L,Timeline as le,WORKFLOW_GRAPH_LOG_MARKER_PREFIX as se}from"@zibby/agent-workflow";var J=[{alias:"auto",canonicalId:"claude-sonnet-4-6",vendor:"claude",label:"Claude \xB7 Sonnet 4.6",family:"claude-sonnet-4",pickerVisible:!0,isDefault:!0},{alias:"opus-4.8",canonicalId:"claude-opus-4-8",vendor:"claude",label:"Claude \xB7 Opus 4.8",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"sonnet-4.8",canonicalId:"claude-sonnet-4-8",vendor:"claude",label:"Claude \xB7 Sonnet 4.8",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"opus-4.7",canonicalId:"claude-opus-4-7",vendor:"claude",label:"Claude \xB7 Opus 4.7",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"sonnet-4.7",canonicalId:"claude-sonnet-4-7",vendor:"claude",label:"Claude \xB7 Sonnet 4.7",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"sonnet-4.6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"opus-4.6",canonicalId:"claude-opus-4-6",vendor:"claude",label:"Claude \xB7 Opus 4.6",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"sonnet-4.5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",label:"Claude \xB7 Sonnet 4.5",family:"claude-sonnet-4",pickerVisible:!0},{alias:"sonnet-4-5",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"opus-4.5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",label:"Claude \xB7 Opus 4.5",family:"claude-opus-4",pickerVisible:!0},{alias:"opus-4-5",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"haiku-4.5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",label:"Claude \xB7 Haiku 4.5",family:"claude-haiku-4"},{alias:"haiku-4-5",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-8",canonicalId:"claude-opus-4-8",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.8"},{alias:"claude-sonnet-4-8",canonicalId:"claude-sonnet-4-8",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.8"},{alias:"claude-opus-4-7",canonicalId:"claude-opus-4-7",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.7"},{alias:"claude-sonnet-4-7",canonicalId:"claude-sonnet-4-7",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.7"},{alias:"claude-sonnet-4-6",canonicalId:"claude-sonnet-4-6",vendor:"claude",family:"claude-sonnet-4",aliasOf:"auto"},{alias:"claude-opus-4-6",canonicalId:"claude-opus-4-6",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.6"},{alias:"claude-sonnet-4-5-20250929",canonicalId:"claude-sonnet-4-5-20250929",vendor:"claude",family:"claude-sonnet-4",aliasOf:"sonnet-4.5"},{alias:"claude-opus-4-5-20251101",canonicalId:"claude-opus-4-5-20251101",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"claude-haiku-4-5-20251001",canonicalId:"claude-haiku-4-5-20251001",vendor:"claude",family:"claude-haiku-4",aliasOf:"haiku-4.5"},{alias:"claude-opus-4-20250514",canonicalId:"claude-opus-4-20250514",vendor:"claude",family:"claude-opus-4",aliasOf:"opus-4.5"},{alias:"auto",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",isDefault:!0},{alias:"o4-mini",canonicalId:"o4-mini",vendor:"codex",label:"Codex \xB7 o4-mini",family:"o4-mini",pickerVisible:!0},{alias:"o3",canonicalId:"o3",vendor:"codex",family:"o3"},{alias:"o3-mini",canonicalId:"o3-mini",vendor:"codex",family:"o3-mini"},{alias:"codex-mini",canonicalId:"codex-mini-latest",vendor:"codex",family:"codex-mini"},{alias:"gpt-4o",canonicalId:"gpt-4o",vendor:"codex",label:"Codex \xB7 GPT-4o",family:"gpt-4o",pickerVisible:!0},{alias:"gpt-4o-mini",canonicalId:"gpt-4o-mini",vendor:"codex",label:"Codex \xB7 GPT-4o mini",family:"gpt-4o-mini",pickerVisible:!0},{alias:"gpt-5.2-codex",canonicalId:"gpt-5.2-codex",vendor:"codex",label:"Codex \xB7 GPT-5.2",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.2",canonicalId:"gpt-5.2",vendor:"codex",label:"Codex \xB7 GPT-5.2 base",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3",canonicalId:"gpt-5.3",vendor:"codex",label:"Codex \xB7 GPT-5.3",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.3-codex",canonicalId:"gpt-5.3-codex",vendor:"codex",family:"gpt-5"},{alias:"gpt-5.4",canonicalId:"gpt-5.4",vendor:"codex",label:"Codex \xB7 GPT-5.4",family:"gpt-5",pickerVisible:!0},{alias:"gpt-5.5",canonicalId:"gpt-5.5",vendor:"codex",label:"Codex \xB7 GPT-5.5",family:"gpt-5",pickerVisible:!0},{alias:"auto",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",isDefault:!0},{alias:"gemini-3-pro",canonicalId:"gemini-3-pro",vendor:"gemini",label:"Gemini \xB7 3 Pro",family:"gemini-3",pickerVisible:!0},{alias:"gemini-3-flash",canonicalId:"gemini-3-flash",vendor:"gemini",label:"Gemini \xB7 3 Flash",family:"gemini-3",pickerVisible:!0},{alias:"gemini-2.5-pro",canonicalId:"gemini-2.5-pro",vendor:"gemini",label:"Gemini \xB7 2.5 Pro",family:"gemini-2.5",pickerVisible:!0},{alias:"gemini-2.5-flash",canonicalId:"gemini-2.5-flash",vendor:"gemini",label:"Gemini \xB7 2.5 Flash",family:"gemini-2.5",pickerVisible:!0}];function _(l){let e={};for(let o of J)o.vendor===l&&(e[o.alias]=o.canonicalId);return e}var w={ASSISTANT:"gpt-5.4-nano-2026-03-17",CLAUDE:"claude-sonnet-4-6",CURSOR:"auto",CODEX:"o4-mini",GEMINI:"gemini-3-pro",OPENAI_POSTPROCESSING:"gpt-4o-mini"};var ce=_("claude"),M=_("codex"),de=_("gemini"),ue={CURSOR_AGENT_DEFAULT:1200*1e3,OPENAI_REQUEST:18e4};function $(l){let e=`${l}_POOL`,o=process.env[e];if(!o||typeof o!="string")return{picked:null,count:0,dispose:()=>{}};let s=o.split(/[,\n]+/).map(a=>a.trim()).filter(Boolean);if(s.length===0)return{picked:null,count:0,dispose:()=>{}};let c=s[Math.floor(Math.random()*s.length)],i=process.env[l];return process.env[l]=c,n.debug(`[auth-pool] ${l}: picked 1 of ${s.length} from pool (***${c.slice(-4)})`),{picked:c,count:s.length,dispose(){i===void 0?delete process.env[l]:process.env[l]=i}}}var R=class extends j{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 n.debug("CodexAgentStrategy: OPENAI_API_KEY or CODEX_API_KEY not set"),!1;try{return z("codex --version",{encoding:"utf-8",timeout:5e3,stdio:"pipe"}),!0}catch{return n.warn("[Codex] codex CLI not found. Install: npm install -g @openai/codex"),!1}}async invoke(e,o={}){let{model:s,workspace:c=process.cwd(),schema:i=null,skills:a=null,sessionPath:O=null,nodeName:m=null,timeout:W,config:H={},signal:D=null}=o,K=[$("OPENAI_API_KEY"),$("CODEX_API_KEY")];try{let{Codex:E}=await import("@openai/codex-sdk"),p=s;(!p||p==="auto")&&(n.debug(`Model is '${p||"undefined"}', using default: ${w.CODEX}`),p=w.CODEX);let b=M[p]||p;M[p]&&p!==b&&n.debug(`Mapped model: ${p} \u2192 ${b}`),n.debug(`Invoking Codex SDK with model: ${b}, skills: ${JSON.stringify(a)}`);let x=process.env.CODEX_API_KEY||process.env.OPENAI_API_KEY;x&&!process.env.CODEX_API_KEY&&(process.env.CODEX_API_KEY=x);let Y=x?` | key: ***${x.slice(-4)}`:" | key: not set";console.log(`
|
|
2
|
+
\u25C6 Model: ${b}${Y}
|
|
3
3
|
`);let h=(await import("chalk")).default;console.log(`
|
|
4
|
-
${h.bold("Prompt sent to LLM:")}`),console.log(h.dim("\u2500".repeat(60))),console.log(h.dim(e)),console.log(h.dim("\u2500".repeat(60)));let k=this._resolveSkillsToMcp(a,{sessionPath:O,workspace:c,nodeName:m}),C={};Object.keys(k).length>0&&(C.mcp_servers=k,n.debug(`[Codex] MCP servers: ${Object.keys(k).join(", ")}`));let
|
|
4
|
+
${h.bold("Prompt sent to LLM:")}`),console.log(h.dim("\u2500".repeat(60))),console.log(h.dim(e)),console.log(h.dim("\u2500".repeat(60)));let k=this._resolveSkillsToMcp(a,{sessionPath:O,workspace:c,nodeName:m}),C={};Object.keys(k).length>0&&(C.mcp_servers=k,n.debug(`[Codex] MCP servers: ${Object.keys(k).join(", ")}`));let U=new E({...Object.keys(C).length>0&&{config:C}}).startThread({workingDirectory:c,skipGitRepoCheck:!0,approvalPolicy:"never",sandboxMode:"danger-full-access",networkAccessEnabled:!0}),T=i&&typeof i.parse=="function",N={};if(i)try{let t=T?V(i,{target:"openAi"}):i;N.outputSchema=t,n.debug("Structured output via SDK outputSchema")}catch(t){n.warn(`[Codex] Schema conversion failed, will extract from text: ${t.message}`)}if(D?.aborted){let t=new Error("Aborted via signal");throw t.name="AbortError",t}try{let{events:t}=await U.runStreamed(e,N),g=0,d="";for await(let f of t){if(D?.aborted){let r=new Error("Aborted via signal");throw r.name="AbortError",r}let v=f.type;if(v==="item.completed"){let r=f.item,I=r?.type;if(I==="mcp_tool_call"){g++;let S=`${r.server}/${r.tool}`;if(L.stepTool(`Tool: ${S}`),r.arguments){let P=JSON.stringify(r.arguments),X=P.length>100?`${P.substring(0,100)}...`:P;console.log(` Input: ${X}`)}}else if(I==="tool_call"||I==="function_call"||I==="command_execution"){g++;let S=r.name||r.tool||r.command||"unknown";L.stepTool(`Tool: ${S}`)}else I==="agent_message"&&(d=r.text||"",d.length<500?console.log(d):console.log(`${d.substring(0,200)}... (${d.length} chars)`))}else v==="turn.completed"?n.debug(`[Codex] Turn completed. Usage: ${JSON.stringify(f.usage||{})}`):n.debug(`[Codex] Event: ${v} ${JSON.stringify(f).slice(0,300)}`)}if(n.debug(`[Codex] Last agent message (${d.length} chars): ${d.slice(0,500)}`),i){if(!d)throw new Error("Codex agent returned no response");let f=JSON.parse(d),v=T?i.parse(f):f;return n.debug("\u2705 [Codex] Structured output validated"),{raw:d,structured:v}}return d||""}catch(t){let g=t.message||String(t);throw n.error(`\u274C [Codex] SDK call failed: ${g}`),g.includes("exited with code")&&(n.error("\u{1F4A1} [Codex] Verify: codex --version && echo $OPENAI_API_KEY"),n.error("\u{1F4A1} [Codex] If codex is missing: npm install -g @openai/codex")),t}}finally{for(let E of K)E.dispose()}}_resolveSkillsToMcp(e,o={}){if(!Array.isArray(e)||e.length===0)return{};let s={};for(let c of e){let i=B(c);if(!i){n.warn(`[Codex] Unknown skill "${c}" \u2014 skipping`);continue}if(typeof i.resolve!="function")continue;let a=i.resolve(o);if(!a)continue;let O=i.serverName||c,m={command:a.command};a.args?.length&&(m.args=a.args),a.env&&Object.keys(a.env).length>0&&(m.env=a.env),s[O]=m,n.debug(`[Codex] MCP: ${O} \u2192 ${a.command} ${(a.args||[]).join(" ")}`)}return s}};export{R as CodexAgentStrategy};
|