catui-agent 1.0.0-beta.0

package/dist/core/lib/ai/src/models.js

@@ -0,0 +1 @@
+var s=Object.defineProperty;var r=(o,t)=>s(o,"name",{value:t,configurable:!0});import{MODELS as p}from"./models.generated.js";const e=new Map;for(const[o,t]of Object.entries(p)){const c=new Map;for(const[i,n]of Object.entries(t))c.set(i,n);e.set(o,c)}function f(o,t){return e.get(o)?.get(t)}r(f,"getModel");function l(){return Array.from(e.keys())}r(l,"getProviders");function a(o){const t=e.get(o);return t?Array.from(t.values()):[]}r(a,"getModels");function h(o,t){return t.cost.input=o.cost.input/1e6*t.input,t.cost.output=o.cost.output/1e6*t.output,t.cost.cacheRead=o.cost.cacheRead/1e6*t.cacheRead,t.cost.cacheWrite=o.cost.cacheWrite/1e6*t.cacheWrite,t.cost.total=t.cost.input+t.cost.output+t.cost.cacheRead+t.cost.cacheWrite,t.cost}r(h,"calculateCost");function v(o){return o.id.includes("gpt-5.2")||o.id.includes("gpt-5.3")?!0:o.api==="anthropic-messages"?o.id.includes("opus-4-6")||o.id.includes("opus-4.6"):!1}r(v,"supportsXhigh");function M(o,t){return!o||!t?!1:o.id===t.id&&o.provider===t.provider}r(M,"modelsAreEqual");export{h as calculateCost,f as getModel,a as getModels,l as getProviders,M as modelsAreEqual,v as supportsXhigh};

package/dist/core/lib/ai/src/overflow.d.ts

@@ -0,0 +1,7 @@
+/**
+ * [WHO]: isContextOverflow, getOverflowPatterns
+ * [FROM]: Depends on overflow utility for provider context-window detection
+ * [TO]: Consumed by @catui/ai/overflow subpath consumers
+ * [HERE]: core/lib/ai/src/overflow.ts - additive overflow helper subpath entry for AI package
+ */
+export * from "./utils/overflow.js";

package/dist/core/lib/ai/src/overflow.js

@@ -0,0 +1 @@
+export*from"./utils/overflow.js";

package/dist/core/lib/ai/src/providers/amazon-bedrock.d.ts

@@ -0,0 +1,20 @@
+/**
+ * [WHO]: BedrockOptions, streamBedrock, streamSimpleBedrock
+ * [FROM]: Depends on ../models.js, ../utils/event-stream.js, ../utils/json-parse.js, ../utils/sanitize-unicode.js, ./simple-options.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/amazon-bedrock.ts -
+ */
+import type { SimpleStreamOptions, StreamFunction, StreamOptions, ThinkingBudgets, ThinkingLevel } from "../types.js";
+export interface BedrockOptions extends StreamOptions {
+    region?: string;
+    profile?: string;
+    toolChoice?: "auto" | "any" | "none" | {
+        type: "tool";
+        name: string;
+    };
+    reasoning?: ThinkingLevel;
+    thinkingBudgets?: ThinkingBudgets;
+    interleavedThinking?: boolean;
+}
+export declare const streamBedrock: StreamFunction<"bedrock-converse-stream", BedrockOptions>;
+export declare const streamSimpleBedrock: StreamFunction<"bedrock-converse-stream", SimpleStreamOptions>;

package/dist/core/lib/ai/src/providers/amazon-bedrock.js

@@ -0,0 +1 @@
+var b=Object.defineProperty;var g=(e,n)=>b(e,"name",{value:n,configurable:!0});import{BedrockRuntimeClient as _,StopReason as h,CachePointType as S,CacheTTL as w,ConversationRole as f,ConverseStreamCommand as I,ImageFormat as m,ToolResultStatus as k}from"@aws-sdk/client-bedrock-runtime";import{calculateCost as B}from"../models.js";import{AssistantMessageEventStream as v}from"../utils/event-stream.js";import{parseStreamingJson as T}from"../utils/json-parse.js";import{sanitizeSurrogates as p}from"../utils/sanitize-unicode.js";import{adjustMaxTokensForThinking as U,buildBaseOptions as A,clampReasoning as O}from"./simple-options.js";import{transformMessages as N}from"./transform-messages.js";const x=g((e,n,t={})=>{const o=new v;return(async()=>{const s={role:"assistant",content:[],api:"bedrock-converse-stream",provider:e.provider,model:e.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()},r=s.content,c={profile:t.profile};if(typeof process<"u"&&(process.versions?.node||process.versions?.bun)){const i=t.region||process.env.AWS_REGION||process.env.AWS_DEFAULT_REGION;if(i?c.region=i:process.env.AWS_PROFILE||(c.region="us-east-1"),process.env.AWS_BEDROCK_SKIP_AUTH==="1"&&(c.credentials={accessKeyId:"dummy-access-key",secretAccessKey:"dummy-secret-key"}),process.env.HTTP_PROXY||process.env.HTTPS_PROXY||process.env.NO_PROXY||process.env.http_proxy||process.env.https_proxy||process.env.no_proxy){const a=await import("@smithy/node-http-handler"),u=await import("proxy-agent"),d=new u.ProxyAgent;c.requestHandler=new a.NodeHttpHandler({httpAgent:d,httpsAgent:d})}else if(process.env.AWS_BEDROCK_FORCE_HTTP1==="1"){const a=await import("@smithy/node-http-handler");c.requestHandler=new a.NodeHttpHandler}}else c.region=t.region||"us-east-1";try{const i=new _(c),a=F(t.cacheRetention),u={modelId:e.id,messages:K(n,e,a),system:J(n.systemPrompt,e,a),inferenceConfig:{maxTokens:t.maxTokens,temperature:t.temperature},toolConfig:X(n.tools,t.toolChoice),additionalModelRequestFields:q(e,t)};t?.onPayload?.(u);const d=new I(u),R=await i.send(d,{abortSignal:t.signal});for await(const l of R.stream)if(l.messageStart){if(l.messageStart.role!==f.ASSISTANT)throw new Error("Unexpected assistant message start but got user message start instead");o.push({type:"start",partial:s})}else if(l.contentBlockStart)P(l.contentBlockStart,r,s,o);else if(l.contentBlockDelta)H(l.contentBlockDelta,r,s,o);else if(l.contentBlockStop)M(l.contentBlockStop,r,s,o);else if(l.messageStop)s.stopReason=$(l.messageStop.stopReason);else if(l.metadata)D(l.metadata,e,s);else{if(l.internalServerException)throw new Error(`Internal server error: ${l.internalServerException.message}`);if(l.modelStreamErrorException)throw new Error(`Model stream error: ${l.modelStreamErrorException.message}`);if(l.validationException)throw new Error(`Validation error: ${l.validationException.message}`);if(l.throttlingException)throw new Error(`Throttling error: ${l.throttlingException.message}`);if(l.serviceUnavailableException)throw new Error(`Service unavailable: ${l.serviceUnavailableException.message}`)}if(t.signal?.aborted)throw new Error("Request was aborted");if(s.stopReason==="error"||s.stopReason==="aborted")throw new Error("An unknown error occurred");o.push({type:"done",reason:s.stopReason,message:s}),o.end()}catch(i){for(const a of s.content)delete a.index,delete a.partialJson;s.stopReason=t.signal?.aborted?"aborted":"error",s.errorMessage=i instanceof Error?i.message:JSON.stringify(i),o.push({type:"error",reason:s.stopReason,error:s}),o.end()}})(),o},"streamBedrock"),ne=g((e,n,t)=>{const o=A(e,t,void 0),s=t?.toolChoice;if(!t?.reasoning)return x(e,n,{...o,toolChoice:s,reasoning:void 0});if(e.id.includes("anthropic.claude")||e.id.includes("anthropic/claude")){if(E(e.id))return x(e,n,{...o,toolChoice:s,reasoning:t.reasoning,thinkingBudgets:t.thinkingBudgets});const r=U(o.maxTokens||0,e.maxTokens,t.reasoning,t.thinkingBudgets);return x(e,n,{...o,toolChoice:s,maxTokens:r.maxTokens,reasoning:t.reasoning,thinkingBudgets:{...t.thinkingBudgets||{},[O(t.reasoning)]:r.thinkingBudget}})}return x(e,n,{...o,toolChoice:s,reasoning:t.reasoning,thinkingBudgets:t.thinkingBudgets})},"streamSimpleBedrock");function P(e,n,t,o){const s=e.contentBlockIndex,r=e.start;if(r?.toolUse){const c={type:"toolCall",id:r.toolUse.toolUseId||"",name:r.toolUse.name||"",arguments:{},partialJson:"",index:s};t.content.push(c),o.push({type:"toolcall_start",contentIndex:n.length-1,partial:t})}}g(P,"handleContentBlockStart");function H(e,n,t,o){const s=e.contentBlockIndex,r=e.delta;let c=n.findIndex(a=>a.index===s),i=n[c];if(r?.text!==void 0){if(!i){const a={type:"text",text:"",index:s};t.content.push(a),c=n.length-1,i=n[c],o.push({type:"text_start",contentIndex:c,partial:t})}i.type==="text"&&(i.text+=r.text,o.push({type:"text_delta",contentIndex:c,delta:r.text,partial:t}))}else if(r?.toolUse&&i?.type==="toolCall")i.partialJson=(i.partialJson||"")+(r.toolUse.input||""),i.arguments=T(i.partialJson),o.push({type:"toolcall_delta",contentIndex:c,delta:r.toolUse.input||"",partial:t});else if(r?.reasoningContent){let a=i,u=c;if(!a){const d={type:"thinking",thinking:"",thinkingSignature:"",index:s};t.content.push(d),u=n.length-1,a=n[u],o.push({type:"thinking_start",contentIndex:u,partial:t})}a?.type==="thinking"&&(r.reasoningContent.text&&(a.thinking+=r.reasoningContent.text,o.push({type:"thinking_delta",contentIndex:u,delta:r.reasoningContent.text,partial:t})),r.reasoningContent.signature&&(a.thinkingSignature=(a.thinkingSignature||"")+r.reasoningContent.signature))}}g(H,"handleContentBlockDelta");function D(e,n,t){e.usage&&(t.usage.input=e.usage.inputTokens||0,t.usage.output=e.usage.outputTokens||0,t.usage.cacheRead=e.usage.cacheReadInputTokens||0,t.usage.cacheWrite=e.usage.cacheWriteInputTokens||0,t.usage.totalTokens=e.usage.totalTokens||t.usage.input+t.usage.output,B(n,t.usage))}g(D,"handleMetadata");function M(e,n,t,o){const s=n.findIndex(c=>c.index===e.contentBlockIndex),r=n[s];if(r)switch(delete r.index,r.type){case"text":o.push({type:"text_end",contentIndex:s,content:r.text,partial:t});break;case"thinking":o.push({type:"thinking_end",contentIndex:s,content:r.thinking,partial:t});break;case"toolCall":r.arguments=T(r.partialJson),delete r.partialJson,o.push({type:"toolcall_end",contentIndex:s,toolCall:r,partial:t});break}}g(M,"handleContentBlockStop");function E(e){return e.includes("opus-4-6")||e.includes("opus-4.6")||e.includes("sonnet-4-6")||e.includes("sonnet-4.6")}g(E,"supportsAdaptiveThinking");function W(e,n){switch(e){case"minimal":case"low":return"low";case"medium":return"medium";case"high":return"high";case"xhigh":return n.includes("opus-4-6")||n.includes("opus-4.6")?"max":"high";default:return"high"}}g(W,"mapThinkingLevelToEffort");function F(e){return e||(typeof process<"u"&&(process.env.CATUI_CACHE_RETENTION??process.env.NANOPENCIL_CACHE_RETENTION)==="long"?"long":"short")}g(F,"resolveCacheRetention");function C(e){if(e.cost.cacheRead||e.cost.cacheWrite)return!0;const n=e.id.toLowerCase();return!!(n.includes("claude")&&(n.includes("-4-")||n.includes("-4."))||n.includes("claude-3-7-sonnet")||n.includes("claude-3-5-haiku"))}g(C,"supportsPromptCaching");function L(e){const n=e.id.toLowerCase();return n.includes("anthropic.claude")||n.includes("anthropic/claude")}g(L,"supportsThinkingSignature");function J(e,n,t){if(!e)return;const o=[{text:p(e)}];return t!=="none"&&C(n)&&o.push({cachePoint:{type:S.DEFAULT,...t==="long"?{ttl:w.ONE_HOUR}:{}}}),o}g(J,"buildSystemPrompt");function j(e){const n=e.replace(/[^a-zA-Z0-9_-]/g,"_");return n.length>64?n.slice(0,64):n}g(j,"normalizeToolCallId");function K(e,n,t){const o=[],s=N(e.messages,n,j);for(let r=0;r<s.length;r++){const c=s[r];switch(c.role){case"user":o.push({role:f.USER,content:typeof c.content=="string"?[{text:p(c.content)}]:c.content.map(i=>{switch(i.type){case"text":return{text:p(i.text)};case"image":return{image:y(i.mimeType,i.data)};default:throw new Error("Unknown user content type")}})});break;case"assistant":{if(c.content.length===0)continue;const i=[];for(const a of c.content)switch(a.type){case"text":if(a.text.trim().length===0)continue;i.push({text:p(a.text)});break;case"toolCall":i.push({toolUse:{toolUseId:a.id,name:a.name,input:a.arguments}});break;case"thinking":if(a.thinking.trim().length===0)continue;L(n)?i.push({reasoningContent:{reasoningText:{text:p(a.thinking),signature:a.thinkingSignature}}}):i.push({reasoningContent:{reasoningText:{text:p(a.thinking)}}});break;default:throw new Error("Unknown assistant content type")}if(i.length===0)continue;o.push({role:f.ASSISTANT,content:i});break}case"toolResult":{const i=[];i.push({toolResult:{toolUseId:c.toolCallId,content:c.content.map(u=>u.type==="image"?{image:y(u.mimeType,u.data)}:u.type==="document"?{text:"[PDF document - content not available in Bedrock format]"}:{text:p(u.text)}),status:c.isError?k.ERROR:k.SUCCESS}});let a=r+1;for(;a<s.length&&s[a].role==="toolResult";){const u=s[a];i.push({toolResult:{toolUseId:u.toolCallId,content:u.content.map(d=>d.type==="image"?{image:y(d.mimeType,d.data)}:d.type==="document"?{text:"[PDF document - content not available in Bedrock format]"}:{text:p(d.text)}),status:u.isError?k.ERROR:k.SUCCESS}}),a++}r=a-1,o.push({role:f.USER,content:i});break}default:throw new Error("Unknown message role")}}if(t!=="none"&&C(n)&&o.length>0){const r=o[o.length-1];r.role===f.USER&&r.content&&r.content.push({cachePoint:{type:S.DEFAULT,...t==="long"?{ttl:w.ONE_HOUR}:{}}})}return o}g(K,"convertMessages");function X(e,n){if(!e?.length||n==="none")return;const t=e.map(s=>({toolSpec:{name:s.name,description:s.description,inputSchema:{json:s.parameters}}}));let o;switch(n){case"auto":o={auto:{}};break;case"any":o={any:{}};break;default:n?.type==="tool"&&(o={tool:{name:n.name}})}return{tools:t,toolChoice:o}}g(X,"convertToolConfig");function $(e){switch(e){case h.END_TURN:case h.STOP_SEQUENCE:return"stop";case h.MAX_TOKENS:case h.MODEL_CONTEXT_WINDOW_EXCEEDED:return"length";case h.TOOL_USE:return"toolUse";default:return"error"}}g($,"mapStopReason");function q(e,n){if(!(!n.reasoning||!e.reasoning)&&(e.id.includes("anthropic.claude")||e.id.includes("anthropic/claude"))){const t=E(e.id)?{thinking:{type:"adaptive"},output_config:{effort:W(n.reasoning,e.id)}}:(()=>{const o={minimal:1024,low:2048,medium:8192,high:16384,xhigh:16384},s=n.reasoning==="xhigh"?"high":n.reasoning;return{thinking:{type:"enabled",budget_tokens:n.thinkingBudgets?.[s]??o[n.reasoning]}}})();return!E(e.id)&&(n.interleavedThinking??!0)&&(t.anthropic_beta=["interleaved-thinking-2025-05-14"]),t}}g(q,"buildAdditionalModelRequestFields");function y(e,n){let t;switch(e){case"image/jpeg":case"image/jpg":t=m.JPEG;break;case"image/png":t=m.PNG;break;case"image/gif":t=m.GIF;break;case"image/webp":t=m.WEBP;break;default:throw new Error(`Unknown image type: ${e}`)}const o=atob(n),s=new Uint8Array(o.length);for(let r=0;r<o.length;r++)s[r]=o.charCodeAt(r);return{source:{bytes:s},format:t}}g(y,"createImageBlock");export{x as streamBedrock,ne as streamSimpleBedrock};

package/dist/core/lib/ai/src/providers/anthropic.d.ts

@@ -0,0 +1,38 @@
+/**
+ * [WHO]: AnthropicOptions, streamAnthropic, streamSimpleAnthropic
+ * [FROM]: Depends on @anthropic-ai/sdk, ../env-api-keys.js, ../models.js, ../utils/event-stream.js, ../utils/json-parse.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/anthropic.ts -
+ */
+import type { SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+export type AnthropicEffort = "low" | "medium" | "high" | "max";
+export interface AnthropicOptions extends StreamOptions {
+    /**
+     * Enable extended thinking.
+     * For Opus 4.6 and Sonnet 4.6: uses adaptive thinking (model decides when/how much to think).
+     * For older models: uses budget-based thinking with thinkingBudgetTokens.
+     */
+    thinkingEnabled?: boolean;
+    /**
+     * Token budget for extended thinking (older models only).
+     * Ignored for Opus 4.6 and Sonnet 4.6, which use adaptive thinking.
+     */
+    thinkingBudgetTokens?: number;
+    /**
+     * Effort level for adaptive thinking (Opus 4.6 and Sonnet 4.6).
+     * Controls how much thinking Claude allocates:
+     * - "max": Always thinks with no constraints (Opus 4.6 only)
+     * - "high": Always thinks, deep reasoning (default)
+     * - "medium": Moderate thinking, may skip for simple queries
+     * - "low": Minimal thinking, skips for simple tasks
+     * Ignored for older models.
+     */
+    effort?: AnthropicEffort;
+    interleavedThinking?: boolean;
+    toolChoice?: "auto" | "any" | "none" | {
+        type: "tool";
+        name: string;
+    };
+}
+export declare const streamAnthropic: StreamFunction<"anthropic-messages", AnthropicOptions>;
+export declare const streamSimpleAnthropic: StreamFunction<"anthropic-messages", SimpleStreamOptions>;

package/dist/core/lib/ai/src/providers/anthropic.js

@@ -0,0 +1,2 @@
+var E=Object.defineProperty;var h=(t,o)=>E(t,"name",{value:o,configurable:!0});import m from"@anthropic-ai/sdk";import{getEnvApiKey as C}from"../env-api-keys.js";import{calculateCost as T}from"../models.js";import{AssistantMessageEventStream as I}from"../utils/event-stream.js";import{parseStreamingJson as w}from"../utils/json-parse.js";import{sanitizeSurrogates as y}from"../utils/sanitize-unicode.js";import{buildCopilotDynamicHeaders as A,hasCopilotVisionInput as S}from"./github-copilot-headers.js";import{adjustMaxTokensForThinking as j,buildBaseOptions as B}from"./simple-options.js";import{transformMessages as L}from"./transform-messages.js";function M(t){return t||(typeof process<"u"&&process.env.CATUI_CACHE_RETENTION==="long"?"long":"short")}h(M,"resolveCacheRetention");function U(t,o){const s=M(o);if(s==="none")return{retention:s};const n=s==="long"&&t.includes("api.anthropic.com")?"1h":void 0;return{retention:s,cacheControl:{type:"ephemeral",...n&&{ttl:n}}}}h(U,"getCacheControl");const N="2.1.62",O=["Read","Write","Edit","Bash","Grep","Glob","AskUserQuestion","EnterPlanMode","ExitPlanMode","KillShell","NotebookEdit","Skill","Task","TaskOutput","TodoWrite","WebFetch","WebSearch"],W=new Map(O.map(t=>[t.toLowerCase(),t])),R=h(t=>W.get(t.toLowerCase())??t,"toClaudeCodeName"),P=h((t,o)=>{if(o&&o.length>0){const s=t.toLowerCase(),n=o.find(e=>e.name.toLowerCase()===s);if(n)return n.name}return t},"fromClaudeCodeName");function v(t){if(!t.some(e=>e.type!=="text"))return y(t.map(e=>e.text).join(`
+`));const s=t.map(e=>e.type==="text"?{type:"text",text:y(e.text)}:e.type==="document"?{type:"document",source:{type:"base64",media_type:"application/pdf",data:e.source.data}}:{type:"image",source:{type:"base64",media_type:e.mimeType,data:e.data}});return s.some(e=>e.type==="text")||s.unshift({type:"text",text:"(see attached document)"}),s}h(v,"convertContentBlocks");function _(...t){const o={};for(const s of t)s&&Object.assign(o,s);return o}h(_,"mergeHeaders");const x=h((t,o,s)=>{const n=new I;return(async()=>{const e={role:"assistant",content:[],api:t.api,provider:t.provider,model:t.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const r=s?.apiKey??C(t.provider)??"";let i;if(t.provider==="github-copilot"){const a=S(o.messages);i=A({messages:o.messages,hasImages:a})}const{client:l,isOAuthToken:p}=K(t,r,s?.interleavedThinking??!0,s?.headers,i),c=q(t,o,p,s);s?.onPayload?.(c);const g=l.messages.stream({...c,stream:!0},{signal:s?.signal});n.push({type:"start",partial:e});const f=e.content;for await(const a of g)if(a.type==="message_start")e.usage.input=a.message.usage.input_tokens||0,e.usage.output=a.message.usage.output_tokens||0,e.usage.cacheRead=a.message.usage.cache_read_input_tokens||0,e.usage.cacheWrite=a.message.usage.cache_creation_input_tokens||0,e.usage.totalTokens=e.usage.input+e.usage.output+e.usage.cacheRead+e.usage.cacheWrite,T(t,e.usage);else if(a.type==="content_block_start"){if(a.content_block.type==="text"){const d={type:"text",text:"",index:a.index};e.content.push(d),n.push({type:"text_start",contentIndex:e.content.length-1,partial:e})}else if(a.content_block.type==="thinking"){const d={type:"thinking",thinking:"",thinkingSignature:"",index:a.index};e.content.push(d),n.push({type:"thinking_start",contentIndex:e.content.length-1,partial:e})}else if(a.content_block.type==="redacted_thinking"){const d={type:"thinking",thinking:"[Reasoning redacted]",thinkingSignature:a.content_block.data,redacted:!0,index:a.index};e.content.push(d),n.push({type:"thinking_start",contentIndex:e.content.length-1,partial:e})}else if(a.content_block.type==="tool_use"){const d={type:"toolCall",id:a.content_block.id,name:p?P(a.content_block.name,o.tools):a.content_block.name,arguments:a.content_block.input??{},partialJson:"",index:a.index};e.content.push(d),n.push({type:"toolcall_start",contentIndex:e.content.length-1,partial:e})}}else if(a.type==="content_block_delta"){if(a.delta.type==="text_delta"){const d=f.findIndex(k=>k.index===a.index),u=f[d];u&&u.type==="text"&&(u.text+=a.delta.text,n.push({type:"text_delta",contentIndex:d,delta:a.delta.text,partial:e}))}else if(a.delta.type==="thinking_delta"){const d=f.findIndex(k=>k.index===a.index),u=f[d];u&&u.type==="thinking"&&(u.thinking+=a.delta.thinking,n.push({type:"thinking_delta",contentIndex:d,delta:a.delta.thinking,partial:e}))}else if(a.delta.type==="input_json_delta"){const d=f.findIndex(k=>k.index===a.index),u=f[d];u&&u.type==="toolCall"&&(u.partialJson+=a.delta.partial_json,u.arguments=w(u.partialJson),n.push({type:"toolcall_delta",contentIndex:d,delta:a.delta.partial_json,partial:e}))}else if(a.delta.type==="signature_delta"){const d=f.findIndex(k=>k.index===a.index),u=f[d];u&&u.type==="thinking"&&(u.thinkingSignature=u.thinkingSignature||"",u.thinkingSignature+=a.delta.signature)}}else if(a.type==="content_block_stop"){const d=f.findIndex(k=>k.index===a.index),u=f[d];u&&(delete u.index,u.type==="text"?n.push({type:"text_end",contentIndex:d,content:u.text,partial:e}):u.type==="thinking"?n.push({type:"thinking_end",contentIndex:d,content:u.thinking,partial:e}):u.type==="toolCall"&&(u.arguments=w(u.partialJson),delete u.partialJson,n.push({type:"toolcall_end",contentIndex:d,toolCall:u,partial:e})))}else a.type==="message_delta"&&(a.delta.stop_reason&&(e.stopReason=D(a.delta.stop_reason)),a.usage.input_tokens!=null&&(e.usage.input=a.usage.input_tokens),a.usage.output_tokens!=null&&(e.usage.output=a.usage.output_tokens),a.usage.cache_read_input_tokens!=null&&(e.usage.cacheRead=a.usage.cache_read_input_tokens),a.usage.cache_creation_input_tokens!=null&&(e.usage.cacheWrite=a.usage.cache_creation_input_tokens),e.usage.totalTokens=e.usage.input+e.usage.output+e.usage.cacheRead+e.usage.cacheWrite,T(t,e.usage));if(s?.signal?.aborted)throw new Error("Request was aborted");if(e.stopReason==="aborted"||e.stopReason==="error")throw new Error("An unknown error occurred");n.push({type:"done",reason:e.stopReason,message:e}),n.end()}catch(r){for(const i of e.content)delete i.index;if(e.stopReason=s?.signal?.aborted?"aborted":"error",e.errorMessage=r instanceof Error?r.message:JSON.stringify(r),r&&typeof r=="object"&&"headers"in r){const i=r.headers;i&&typeof i=="object"&&(e.errorHeaders=Object.fromEntries(Object.entries(i).map(([l,p])=>[l.toLowerCase(),String(p)])))}n.push({type:"error",reason:e.stopReason,error:e}),n.end()}})(),n},"streamAnthropic");function b(t){return t.includes("opus-4-6")||t.includes("opus-4.6")||t.includes("sonnet-4-6")||t.includes("sonnet-4.6")}h(b,"supportsAdaptiveThinking");function J(t,o){switch(t){case"minimal":return"low";case"low":return"low";case"medium":return"medium";case"high":return"high";case"xhigh":return o.includes("opus-4-6")||o.includes("opus-4.6")?"max":"high";default:return"high"}}h(J,"mapThinkingLevelToEffort");const se=h((t,o,s)=>{const n=s?.apiKey||C(t.provider);if(!n)throw new Error(`No API key for provider: ${t.provider}`);const e=B(t,s,n),r=s?.toolChoice;if(!s?.reasoning)return x(t,o,{...e,toolChoice:r,thinkingEnabled:!1});if(b(t.id)){const l=J(s.reasoning,t.id);return x(t,o,{...e,toolChoice:r,thinkingEnabled:!0,effort:l})}const i=j(e.maxTokens||0,t.maxTokens,s.reasoning,s.thinkingBudgets);return x(t,o,{...e,toolChoice:r,maxTokens:i.maxTokens,thinkingEnabled:!0,thinkingBudgetTokens:i.thinkingBudget})},"streamSimpleAnthropic");function H(t){return t.includes("sk-ant-oat")}h(H,"isOAuthToken");function K(t,o,s,n,e){const r=s&&!b(t.id);if(t.provider==="github-copilot"){const p=[];return r&&p.push("interleaved-thinking-2025-05-14"),{client:new m({apiKey:null,authToken:o,baseURL:t.baseUrl,dangerouslyAllowBrowser:!0,defaultHeaders:_({accept:"application/json","anthropic-dangerous-direct-browser-access":"true",...p.length>0?{"anthropic-beta":p.join(",")}:{}},t.headers,e,n)}),isOAuthToken:!1}}const i=["fine-grained-tool-streaming-2025-05-14"];return r&&i.push("interleaved-thinking-2025-05-14"),H(o)?{client:new m({apiKey:null,authToken:o,baseURL:t.baseUrl,dangerouslyAllowBrowser:!0,defaultHeaders:_({accept:"application/json","anthropic-dangerous-direct-browser-access":"true","anthropic-beta":`claude-code-20250219,oauth-2025-04-20,${i.join(",")}`,"user-agent":`claude-cli/${N}`,"x-app":"cli"},t.headers,n)}),isOAuthToken:!0}:{client:new m({apiKey:o,baseURL:t.baseUrl,dangerouslyAllowBrowser:!0,defaultHeaders:_({accept:"application/json","anthropic-dangerous-direct-browser-access":"true","anthropic-beta":i.join(",")},t.headers,n)}),isOAuthToken:!1}}h(K,"createClient");function q(t,o,s,n){const{cacheControl:e}=U(t.baseUrl,n?.cacheRetention),r={model:t.id,messages:$(o.messages,t,s,e),max_tokens:n?.maxTokens||t.maxTokens/3|0,stream:!0};if(s?(r.system=[{type:"text",text:"You are Claude Code, Anthropic's official CLI for Claude.",...e?{cache_control:e}:{}}],o.systemPrompt&&r.system.push({type:"text",text:y(o.systemPrompt),...e?{cache_control:e}:{}})):o.systemPrompt&&(r.system=[{type:"text",text:y(o.systemPrompt),...e?{cache_control:e}:{}}]),n?.temperature!==void 0&&!n?.thinkingEnabled&&(r.temperature=n.temperature),o.tools&&(r.tools=z(o.tools,s)),n?.thinkingEnabled&&t.reasoning&&(b(t.id)?(r.thinking={type:"adaptive"},n.effort&&(r.output_config={effort:n.effort})):r.thinking={type:"enabled",budget_tokens:n.thinkingBudgetTokens||1024}),n?.metadata){const i=n.metadata.user_id;typeof i=="string"&&(r.metadata={user_id:i})}return n?.toolChoice&&(typeof n.toolChoice=="string"?r.tool_choice={type:n.toolChoice}:r.tool_choice=n.toolChoice),r}h(q,"buildParams");function F(t){return t.replace(/[^a-zA-Z0-9_-]/g,"_").slice(0,64)}h(F,"normalizeToolCallId");function $(t,o,s,n){const e=[],r=L(t,o,F);for(let i=0;i<r.length;i++){const l=r[i];if(l.role==="user")if(typeof l.content=="string")l.content.trim().length>0&&e.push({role:"user",content:y(l.content)});else{const p=l.content.map(g=>g.type==="text"?{type:"text",text:y(g.text)}:g.type==="document"?{type:"document",source:{type:"base64",media_type:"application/pdf",data:g.source.data}}:{type:"image",source:{type:"base64",media_type:g.mimeType,data:g.data}});let c=o?.input.includes("image")?p:p.filter(g=>g.type!=="image");if(c=c.filter(g=>g.type==="text"?g.text.trim().length>0:!0),c.length===0)continue;e.push({role:"user",content:c})}else if(l.role==="assistant"){const p=[];for(const c of l.content)if(c.type==="text"){if(c.text.trim().length===0)continue;p.push({type:"text",text:y(c.text)})}else if(c.type==="thinking"){if(c.redacted){p.push({type:"redacted_thinking",data:c.thinkingSignature});continue}if(c.thinking.trim().length===0)continue;!c.thinkingSignature||c.thinkingSignature.trim().length===0?p.push({type:"text",text:y(c.thinking)}):p.push({type:"thinking",thinking:y(c.thinking),signature:c.thinkingSignature})}else c.type==="toolCall"&&p.push({type:"tool_use",id:c.id,name:s?R(c.name):c.name,input:c.arguments??{}});if(p.length===0)continue;e.push({role:"assistant",content:p})}else if(l.role==="toolResult"){const p=[];p.push({type:"tool_result",tool_use_id:l.toolCallId,content:v(l.content),is_error:l.isError});let c=i+1;for(;c<r.length&&r[c].role==="toolResult";){const g=r[c];p.push({type:"tool_result",tool_use_id:g.toolCallId,content:v(g.content),is_error:g.isError}),c++}i=c-1,e.push({role:"user",content:p})}}if(n&&e.length>0){const i=e[e.length-1];if(i.role==="user")if(Array.isArray(i.content)){const l=i.content[i.content.length-1];l&&(l.type==="text"||l.type==="image"||l.type==="tool_result")&&(l.cache_control=n)}else typeof i.content=="string"&&(i.content=[{type:"text",text:i.content,cache_control:n}])}return e}h($,"convertMessages");function z(t,o){return t?t.map(s=>{const n=s.parameters;return{name:o?R(s.name):s.name,description:s.description,input_schema:{type:"object",properties:n.properties||{},required:n.required||[]}}}):[]}h(z,"convertTools");function D(t){switch(t){case"end_turn":return"stop";case"max_tokens":return"length";case"tool_use":return"toolUse";case"refusal":return"error";case"pause_turn":return"stop";case"stop_sequence":return"stop";case"sensitive":return"error";default:throw new Error(`Unhandled stop reason: ${t}`)}}h(D,"mapStopReason");export{x as streamAnthropic,se as streamSimpleAnthropic};

package/dist/core/lib/ai/src/providers/azure-openai-responses.d.ts

@@ -0,0 +1,21 @@
+/**
+ * [WHO]: AzureOpenAIResponsesOptions, streamAzureOpenAIResponses, streamSimpleAzureOpenAIResponses
+ * [FROM]: Depends on openai, ../env-api-keys.js, ../models.js, ../utils/event-stream.js, ./openai-responses-shared.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/azure-openai-responses.ts -
+ */
+import type { SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+export interface AzureOpenAIResponsesOptions extends StreamOptions {
+    reasoningEffort?: "minimal" | "low" | "medium" | "high" | "xhigh";
+    reasoningSummary?: "auto" | "detailed" | "concise" | null;
+    toolChoice?: unknown;
+    azureApiVersion?: string;
+    azureResourceName?: string;
+    azureBaseUrl?: string;
+    azureDeploymentName?: string;
+}
+/**
+ * Generate function for Azure OpenAI Responses API
+ */
+export declare const streamAzureOpenAIResponses: StreamFunction<"azure-openai-responses", AzureOpenAIResponsesOptions>;
+export declare const streamSimpleAzureOpenAIResponses: StreamFunction<"azure-openai-responses", SimpleStreamOptions>;

package/dist/core/lib/ai/src/providers/azure-openai-responses.js

@@ -0,0 +1 @@
+var f=Object.defineProperty;var o=(r,t)=>f(r,"name",{value:t,configurable:!0});import{AzureOpenAI as l}from"openai";import{getEnvApiKey as u}from"../env-api-keys.js";import{supportsXhigh as A}from"../models.js";import{AssistantMessageEventStream as E}from"../utils/event-stream.js";import{convertResponsesMessages as d,convertResponsesTools as _,processResponsesStream as R}from"./openai-responses-shared.js";import{buildBaseOptions as g,clampReasoning as U}from"./simple-options.js";const h="v1",y=new Set(["openai","openai-codex","opencode","azure-openai-responses"]);function I(r){const t=new Map;if(!r)return t;for(const e of r.split(",")){const n=e.trim();if(!n)continue;const[a,s]=n.split("=",2);!a||!s||t.set(a.trim(),s.trim())}return t}o(I,"parseDeploymentNameMap");function N(r,t){return t?.azureDeploymentName?t.azureDeploymentName:I(process.env.AZURE_OPENAI_DEPLOYMENT_NAME_MAP).get(r.id)||r.id}o(N,"resolveDeploymentName");const O=o((r,t,e)=>{const n=new E;return(async()=>{const a=N(r,e),s={role:"assistant",content:[],api:"azure-openai-responses",provider:r.provider,model:r.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const i=e?.apiKey||u(r.provider)||"",c=b(r,i,e),p=z(r,t,e,a);e?.onPayload?.(p);const m=await c.responses.create(p,e?.signal?{signal:e.signal}:void 0);if(n.push({type:"start",partial:s}),await R(m,s,n,r),e?.signal?.aborted)throw new Error("Request was aborted");if(s.stopReason==="aborted"||s.stopReason==="error")throw new Error("An unknown error occurred");n.push({type:"done",reason:s.stopReason,message:s}),n.end()}catch(i){for(const c of s.content)delete c.index;s.stopReason=e?.signal?.aborted?"aborted":"error",s.errorMessage=i instanceof Error?i.message:JSON.stringify(i),n.push({type:"error",reason:s.stopReason,error:s}),n.end()}})(),n},"streamAzureOpenAIResponses"),B=o((r,t,e)=>{const n=e?.apiKey||u(r.provider);if(!n)throw new Error(`No API key for provider: ${r.provider}`);const a=g(r,e,n),s=A(r)?e?.reasoning:U(e?.reasoning);return O(r,t,{...a,reasoningEffort:s,toolChoice:e?.toolChoice})},"streamSimpleAzureOpenAIResponses");function v(r){return r.replace(/\/+$/,"")}o(v,"normalizeAzureBaseUrl");function w(r){return`https://${r}.openai.azure.com/openai/v1`}o(w,"buildDefaultBaseUrl");function P(r,t){const e=t?.azureApiVersion||process.env.AZURE_OPENAI_API_VERSION||h,n=t?.azureBaseUrl?.trim()||process.env.AZURE_OPENAI_BASE_URL?.trim()||void 0,a=t?.azureResourceName||process.env.AZURE_OPENAI_RESOURCE_NAME;let s=n;if(!s&&a&&(s=w(a)),!s&&r.baseUrl&&(s=r.baseUrl),!s)throw new Error("Azure OpenAI base URL is required. Set AZURE_OPENAI_BASE_URL or AZURE_OPENAI_RESOURCE_NAME, or pass azureBaseUrl, azureResourceName, or model.baseUrl.");return{baseUrl:v(s),apiVersion:e}}o(P,"resolveAzureConfig");function b(r,t,e){if(!t){if(!process.env.AZURE_OPENAI_API_KEY)throw new Error("Azure OpenAI API key is required. Set AZURE_OPENAI_API_KEY environment variable or pass it as an argument.");t=process.env.AZURE_OPENAI_API_KEY}const n={...r.headers};e?.headers&&Object.assign(n,e.headers);const{baseUrl:a,apiVersion:s}=P(r,e);return new l({apiKey:t,apiVersion:s,dangerouslyAllowBrowser:!0,defaultHeaders:n,baseURL:a})}o(b,"createClient");function z(r,t,e,n){const a=d(r,t,y),s={model:n,input:a,stream:!0,prompt_cache_key:e?.sessionId};return e?.maxTokens&&(s.max_output_tokens=e?.maxTokens),e?.temperature!==void 0&&(s.temperature=e?.temperature),t.tools&&(s.tools=_(t.tools)),e?.toolChoice!==void 0&&(s.tool_choice=e.toolChoice),r.reasoning&&(e?.reasoningEffort||e?.reasoningSummary?(s.reasoning={effort:e?.reasoningEffort||"medium",summary:e?.reasoningSummary||"auto"},s.include=["reasoning.encrypted_content"]):r.name.toLowerCase().startsWith("gpt-5")&&a.push({role:"developer",content:[{type:"input_text",text:"# Juice: 0 !important"}]})),s}o(z,"buildParams");export{O as streamAzureOpenAIResponses,B as streamSimpleAzureOpenAIResponses};

package/dist/core/lib/ai/src/providers/github-copilot-headers.d.ts

@@ -0,0 +1,13 @@
+/**
+ * [WHO]: inferCopilotInitiator, hasCopilotVisionInput, buildCopilotDynamicHeaders
+ * [FROM]: No external dependencies
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/github-copilot-headers.ts -
+ */
+import type { Message } from "../types.js";
+export declare function inferCopilotInitiator(messages: Message[]): "user" | "agent";
+export declare function hasCopilotVisionInput(messages: Message[]): boolean;
+export declare function buildCopilotDynamicHeaders(params: {
+    messages: Message[];
+    hasImages: boolean;
+}): Record<string, string>;

package/dist/core/lib/ai/src/providers/github-copilot-headers.js

@@ -0,0 +1 @@
+var o=Object.defineProperty;var n=(e,t)=>o(e,"name",{value:t,configurable:!0});function i(e){const t=e[e.length-1];return t&&t.role!=="user"?"agent":"user"}n(i,"inferCopilotInitiator");function s(e){return e.some(t=>t.role==="user"&&Array.isArray(t.content)?t.content.some(r=>r.type==="image"):t.role==="toolResult"&&Array.isArray(t.content)?t.content.some(r=>r.type==="image"):!1)}n(s,"hasCopilotVisionInput");function u(e){const t={"X-Initiator":i(e.messages),"Openai-Intent":"conversation-edits"};return e.hasImages&&(t["Copilot-Vision-Request"]="true"),t}n(u,"buildCopilotDynamicHeaders");export{u as buildCopilotDynamicHeaders,s as hasCopilotVisionInput,i as inferCopilotInitiator};

package/dist/core/lib/ai/src/providers/google-gemini-cli.d.ts

@@ -0,0 +1,79 @@
+/**
+ * Google Gemini CLI / Antigravity provider.
+ * Shared implementation for both google-gemini-cli and google-antigravity providers.
+ * Uses the Cloud Code Assist API endpoint to access Gemini and Claude models.
+ */
+/**
+ * [WHO]: GoogleGeminiCliOptions, extractRetryDelay, streamGoogleGeminiCli, streamSimpleGoogleGeminiCli, buildRequest
+ * [FROM]: Depends on ../models.js, ../utils/event-stream.js, ../utils/sanitize-unicode.js, ./simple-options.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/google-gemini-cli.ts -
+ */
+import type { Content, ThinkingConfig } from "@google/genai";
+import type { Context, Model, SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+import { convertTools, mapToolChoice } from "./google-shared.js";
+/**
+ * Thinking level for Gemini 3 models.
+ * Mirrors Google's ThinkingLevel enum values.
+ */
+export type GoogleThinkingLevel = "THINKING_LEVEL_UNSPECIFIED" | "MINIMAL" | "LOW" | "MEDIUM" | "HIGH";
+export interface GoogleGeminiCliOptions extends StreamOptions {
+    toolChoice?: "auto" | "none" | "any";
+    /**
+     * Thinking/reasoning configuration.
+     * - Gemini 2.x models: use `budgetTokens` to set the thinking budget
+     * - Gemini 3 models (gemini-3-pro-*, gemini-3-flash-*): use `level` instead
+     *
+     * When using `streamSimple`, this is handled automatically based on the model.
+     */
+    thinking?: {
+        enabled: boolean;
+        /** Thinking budget in tokens. Use for Gemini 2.x models. */
+        budgetTokens?: number;
+        /** Thinking level. Use for Gemini 3 models (LOW/HIGH for Pro, MINIMAL/LOW/MEDIUM/HIGH for Flash). */
+        level?: GoogleThinkingLevel;
+    };
+    projectId?: string;
+}
+/**
+ * Extract retry delay from Gemini error response (in milliseconds).
+ * Checks headers first (Retry-After, x-ratelimit-reset, x-ratelimit-reset-after),
+ * then parses body patterns like:
+ * - "Your quota will reset after 39s"
+ * - "Your quota will reset after 18h31m10s"
+ * - "Please retry in Xs" or "Please retry in Xms"
+ * - "retryDelay": "34.074824224s" (JSON field)
+ */
+export declare function extractRetryDelay(errorText: string, response?: Response | Headers): number | undefined;
+interface CloudCodeAssistRequest {
+    project: string;
+    model: string;
+    request: {
+        contents: Content[];
+        sessionId?: string;
+        systemInstruction?: {
+            role?: string;
+            parts: {
+                text: string;
+            }[];
+        };
+        generationConfig?: {
+            maxOutputTokens?: number;
+            temperature?: number;
+            thinkingConfig?: ThinkingConfig;
+        };
+        tools?: ReturnType<typeof convertTools>;
+        toolConfig?: {
+            functionCallingConfig: {
+                mode: ReturnType<typeof mapToolChoice>;
+            };
+        };
+    };
+    requestType?: string;
+    userAgent?: string;
+    requestId?: string;
+}
+export declare const streamGoogleGeminiCli: StreamFunction<"google-gemini-cli", GoogleGeminiCliOptions>;
+export declare const streamSimpleGoogleGeminiCli: StreamFunction<"google-gemini-cli", SimpleStreamOptions>;
+export declare function buildRequest(model: Model<"google-gemini-cli">, context: Context, projectId: string, options?: GoogleGeminiCliOptions, isAntigravity?: boolean): CloudCodeAssistRequest;
+export {};

package/dist/core/lib/ai/src/providers/google-gemini-cli.js

@@ -0,0 +1,2 @@
+var ne=Object.defineProperty;var c=(t,s)=>ne(t,"name",{value:s,configurable:!0});import{calculateCost as re}from"../models.js";import{AssistantMessageEventStream as oe}from"../utils/event-stream.js";import{sanitizeSurrogates as ae}from"../utils/sanitize-unicode.js";import{convertMessages as se,convertTools as ie,isThinkingPart as ce,mapStopReasonString as le,mapToolChoice as ue,retainThoughtSignature as W}from"./google-shared.js";import{buildBaseOptions as de,clampReasoning as ge}from"./simple-options.js";const J="https://cloudcode-pa.googleapis.com",he="https://daily-cloudcode-pa.sandbox.googleapis.com",fe=[he,J],pe={"User-Agent":"google-cloud-sdk vscode_cloudshelleditor/0.1","X-Goog-Api-Client":"gl-node/22.17.0","Client-Metadata":JSON.stringify({ideType:"IDE_UNSPECIFIED",platform:"PLATFORM_UNSPECIFIED",pluginType:"GEMINI"})},me="1.18.3";function ye(){return{"User-Agent":`antigravity/${process.env.CATUI_AI_ANTIGRAVITY_VERSION||me} darwin/arm64`,"X-Goog-Api-Client":"google-cloud-sdk vscode_cloudshelleditor/0.1","Client-Metadata":JSON.stringify({ideType:"IDE_UNSPECIFIED",platform:"PLATFORM_UNSPECIFIED",pluginType:"GEMINI"})}}c(ye,"getAntigravityHeaders");const V="You are Antigravity, a powerful agentic AI coding assistant designed by the Google Deepmind team working on Advanced Agentic Coding.You are pair programming with a USER to solve their coding task. The task may require creating a new codebase, modifying or debugging an existing codebase, or simply answering a question.**Absolute paths only****Proactiveness**";let we=0;const G=3,j=1e3,K=2,Ie=500,ke="interleaved-thinking-2025-05-14";function Ee(t,s){const r=c(a=>a>0?Math.ceil(a+1e3):void 0,"normalizeDelay"),n=s instanceof Headers?s:s?.headers;if(n){const a=n.get("retry-after");if(a){const g=Number(a);if(Number.isFinite(g)){const E=r(g*1e3);if(E!==void 0)return E}const x=new Date(a).getTime();if(!Number.isNaN(x)){const E=r(x-Date.now());if(E!==void 0)return E}}const d=n.get("x-ratelimit-reset");if(d){const g=Number.parseInt(d,10);if(!Number.isNaN(g)){const p=r(g*1e3-Date.now());if(p!==void 0)return p}}const f=n.get("x-ratelimit-reset-after");if(f){const g=Number(f);if(Number.isFinite(g)){const p=r(g*1e3);if(p!==void 0)return p}}}const e=t.match(/reset after (?:(\d+)h)?(?:(\d+)m)?(\d+(?:\.\d+)?)s/i);if(e){const a=e[1]?parseInt(e[1],10):0,d=e[2]?parseInt(e[2],10):0,f=parseFloat(e[3]);if(!Number.isNaN(f)){const g=((a*60+d)*60+f)*1e3,p=r(g);if(p!==void 0)return p}}const h=t.match(/Please retry in ([0-9.]+)(ms|s)/i);if(h?.[1]){const a=parseFloat(h[1]);if(!Number.isNaN(a)&&a>0){const d=h[2].toLowerCase()==="ms"?a:a*1e3,f=r(d);if(f!==void 0)return f}}const i=t.match(/"retryDelay":\s*"([0-9.]+)(ms|s)"/i);if(i?.[1]){const a=parseFloat(i[1]);if(!Number.isNaN(a)&&a>0){const d=i[2].toLowerCase()==="ms"?a:a*1e3,f=r(d);if(f!==void 0)return f}}}c(Ee,"extractRetryDelay");function Ce(t){const s=t.toLowerCase();return s.includes("claude")&&s.includes("thinking")}c(Ce,"isClaudeThinkingModel");function z(t){return/gemini-3(?:\.1)?-pro/.test(t.toLowerCase())}c(z,"isGemini3ProModel");function Te(t){return/gemini-3(?:\.1)?-flash/.test(t.toLowerCase())}c(Te,"isGemini3FlashModel");function be(t){return z(t)||Te(t)}c(be,"isGemini3Model");function Ae(t,s){return t===429||t===500||t===502||t===503||t===504?!0:/resource.?exhausted|rate.?limit|overloaded|service.?unavailable|other.?side.?closed/i.test(s)}c(Ae,"isRetryableError");function X(t){try{const s=JSON.parse(t);if(s.error?.message)return s.error.message}catch{}return t}c(X,"extractErrorMessage");function L(t,s){return new Promise((r,n)=>{if(s?.aborted){n(new Error("Request was aborted"));return}const e=setTimeout(r,t);s?.addEventListener("abort",()=>{clearTimeout(e),n(new Error("Request was aborted"))})})}c(L,"sleep");const O=c((t,s,r)=>{const n=new oe;return(async()=>{const e={role:"assistant",content:[],api:"google-gemini-cli",provider:t.provider,model:t.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const h=r?.apiKey;if(!h)throw new Error("Google Cloud Code Assist requires OAuth authentication. Use /login to authenticate.");let i,a;try{const l=JSON.parse(h);i=l.token,a=l.projectId}catch{throw new Error("Invalid Google Cloud Code Assist credentials. Use /login to re-authenticate.")}if(!i||!a)throw new Error("Missing token or projectId in Google Cloud credentials. Use /login to re-authenticate.");const d=t.provider==="google-antigravity",f=t.baseUrl?.trim(),g=f?[f]:d?fe:[J],p=xe(t,s,a,r,d);r?.onPayload?.(p);const x=d?ye():pe,E={Authorization:`Bearer ${i}`,"Content-Type":"application/json",Accept:"text/event-stream",...x,...Ce(t.id)?{"anthropic-beta":ke}:{},...r?.headers},P=JSON.stringify(p);let I,T,N;for(let l=0;l<=G;l++){if(r?.signal?.aborted)throw new Error("Request was aborted");try{if(N=`${g[Math.min(l,g.length-1)]}/v1internal:streamGenerateContent?alt=sse`,I=await fetch(N,{method:"POST",headers:E,body:P,signal:r?.signal}),I.ok)break;const o=await I.text();if(l<G&&Ae(I.status,o)){const k=Ee(o,I),m=k??j*2**l,b=r?.maxRetryDelayMs??6e4;if(b>0&&k&&k>b){const _=Math.ceil(k/1e3);throw new Error(`Server requested ${_}s retry delay (max: ${Math.ceil(b/1e3)}s). ${X(o)}`)}await L(m,r?.signal);continue}throw new Error(`Cloud Code Assist API error (${I.status}): ${X(o)}`)}catch(y){if(y instanceof Error&&(y.name==="AbortError"||y.message==="Request was aborted"))throw new Error("Request was aborted");if(T=y instanceof Error?y:new Error(String(y)),T.message==="fetch failed"&&T.cause instanceof Error&&(T=new Error(`Network error: ${T.cause.message}`)),l<G){const o=j*2**l;await L(o,r?.signal);continue}throw T}}if(!I||!I.ok)throw T??new Error("Failed to get response after retries");let S=!1;const M=c(()=>{S||(n.push({type:"start",partial:e}),S=!0)},"ensureStarted"),Q=c(()=>{e.content=[],e.usage={input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},e.stopReason="stop",e.errorMessage=void 0,e.timestamp=Date.now(),S=!1},"resetOutput"),Z=c(async l=>{if(!l.body)throw new Error("No response body");let y=!1,o=null;const k=e.content,m=c(()=>k.length-1,"blockIndex"),b=l.body.getReader(),_=new TextDecoder;let v="";const U=c(()=>{b.cancel().catch(()=>{})},"abortHandler");r?.signal?.addEventListener("abort",U);try{for(;;){if(r?.signal?.aborted)throw new Error("Request was aborted");const{done:ee,value:te}=await b.read();if(ee)break;v+=_.decode(te,{stream:!0});const $=v.split(`
+`);v=$.pop()||"";for(const F of $){if(!F.startsWith("data:"))continue;const B=F.slice(5).trim();if(!B)continue;let Y;try{Y=JSON.parse(B)}catch{continue}const C=Y.response;if(!C)continue;const R=C.candidates?.[0];if(R?.content?.parts)for(const u of R.content.parts){if(u.text!==void 0){y=!0;const w=ce(u);(!o||w&&o.type!=="thinking"||!w&&o.type!=="text")&&(o&&(o.type==="text"?n.push({type:"text_end",contentIndex:k.length-1,content:o.text,partial:e}):n.push({type:"thinking_end",contentIndex:m(),content:o.thinking,partial:e})),w?(o={type:"thinking",thinking:"",thinkingSignature:void 0},e.content.push(o),M(),n.push({type:"thinking_start",contentIndex:m(),partial:e})):(o={type:"text",text:""},e.content.push(o),M(),n.push({type:"text_start",contentIndex:m(),partial:e}))),o.type==="thinking"?(o.thinking+=u.text,o.thinkingSignature=W(o.thinkingSignature,u.thoughtSignature),n.push({type:"thinking_delta",contentIndex:m(),delta:u.text,partial:e})):(o.text+=u.text,o.textSignature=W(o.textSignature,u.thoughtSignature),n.push({type:"text_delta",contentIndex:m(),delta:u.text,partial:e}))}if(u.functionCall){y=!0,o&&(o.type==="text"?n.push({type:"text_end",contentIndex:m(),content:o.text,partial:e}):n.push({type:"thinking_end",contentIndex:m(),content:o.thinking,partial:e}),o=null);const w=u.functionCall.id,D={type:"toolCall",id:!w||e.content.some(H=>H.type==="toolCall"&&H.id===w)?`${u.functionCall.name}_${Date.now()}_${++we}`:w,name:u.functionCall.name||"",arguments:u.functionCall.args??{},...u.thoughtSignature&&{thoughtSignature:u.thoughtSignature}};e.content.push(D),M(),n.push({type:"toolcall_start",contentIndex:m(),partial:e}),n.push({type:"toolcall_delta",contentIndex:m(),delta:JSON.stringify(D.arguments),partial:e}),n.push({type:"toolcall_end",contentIndex:m(),toolCall:D,partial:e})}}if(R?.finishReason&&(e.stopReason=le(R.finishReason),e.content.some(u=>u.type==="toolCall")&&(e.stopReason="toolUse")),C.usageMetadata){const u=C.usageMetadata.promptTokenCount||0,w=C.usageMetadata.cachedContentTokenCount||0;e.usage={input:u-w,output:(C.usageMetadata.candidatesTokenCount||0)+(C.usageMetadata.thoughtsTokenCount||0),cacheRead:w,cacheWrite:0,totalTokens:C.usageMetadata.totalTokenCount||0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},re(t,e.usage)}}}}finally{r?.signal?.removeEventListener("abort",U)}return o&&(o.type==="text"?n.push({type:"text_end",contentIndex:m(),content:o.text,partial:e}):n.push({type:"thinking_end",contentIndex:m(),content:o.thinking,partial:e})),y},"streamResponse");let q=!1,A=I;for(let l=0;l<=K;l++){if(r?.signal?.aborted)throw new Error("Request was aborted");if(l>0){const o=Ie*2**(l-1);if(await L(o,r?.signal),!N)throw new Error("Missing request URL");if(A=await fetch(N,{method:"POST",headers:E,body:P,signal:r?.signal}),!A.ok){const k=await A.text();throw new Error(`Cloud Code Assist API error (${A.status}): ${k}`)}}if(await Z(A)){q=!0;break}l<K&&Q()}if(!q)throw new Error("Cloud Code Assist API returned an empty response");if(r?.signal?.aborted)throw new Error("Request was aborted");if(e.stopReason==="aborted"||e.stopReason==="error")throw new Error("An unknown error occurred");n.push({type:"done",reason:e.stopReason,message:e}),n.end()}catch(h){for(const i of e.content)"index"in i&&delete i.index;e.stopReason=r?.signal?.aborted?"aborted":"error",e.errorMessage=h instanceof Error?h.message:JSON.stringify(h),n.push({type:"error",reason:e.stopReason,error:e}),n.end()}})(),n},"streamGoogleGeminiCli"),Oe=c((t,s,r)=>{const n=r?.apiKey;if(!n)throw new Error("Google Cloud Code Assist requires OAuth authentication. Use /login to authenticate.");const e=de(t,r,n),h=r?.toolChoice;if(!r?.reasoning)return O(t,s,{...e,toolChoice:h,thinking:{enabled:!1}});const i=ge(r.reasoning);if(be(t.id))return O(t,s,{...e,toolChoice:h,thinking:{enabled:!0,level:Ne(i,t.id)}});const d={...{minimal:1024,low:2048,medium:8192,high:16384},...r.thinkingBudgets},f=1024;let g=d[i];const p=Math.min((e.maxTokens||0)+g,t.maxTokens);return p<=g&&(g=Math.max(0,p-f)),O(t,s,{...e,toolChoice:h,maxTokens:p,thinking:{enabled:!0,budgetTokens:g}})},"streamSimpleGoogleGeminiCli");function xe(t,s,r,n={},e=!1){const h=se(t,s),i={};n.temperature!==void 0&&(i.temperature=n.temperature),n.maxTokens!==void 0&&(i.maxOutputTokens=n.maxTokens),n.thinking?.enabled&&t.reasoning&&(i.thinkingConfig={includeThoughts:!0},n.thinking.level!==void 0?i.thinkingConfig.thinkingLevel=n.thinking.level:n.thinking.budgetTokens!==void 0&&(i.thinkingConfig.thinkingBudget=n.thinking.budgetTokens));const a={contents:h};if(a.sessionId=n.sessionId,s.systemPrompt&&(a.systemInstruction={parts:[{text:ae(s.systemPrompt)}]}),Object.keys(i).length>0&&(a.generationConfig=i),s.tools&&s.tools.length>0){const d=t.id.startsWith("claude-");a.tools=ie(s.tools,d),n.toolChoice&&(a.toolConfig={functionCallingConfig:{mode:ue(n.toolChoice)}})}if(e){const d=a.systemInstruction?.parts??[];a.systemInstruction={role:"user",parts:[{text:V},{text:`Please ignore following [ignore]${V}[/ignore]`},...d]}}return{project:r,model:t.id,request:a,...e?{requestType:"agent"}:{},userAgent:e?"antigravity":"catui-coding-agent",requestId:`${e?"agent":"catui"}-${Date.now()}-${Math.random().toString(36).slice(2,11)}`}}c(xe,"buildRequest");function Ne(t,s){if(z(s))switch(t){case"minimal":case"low":return"LOW";case"medium":case"high":return"HIGH"}switch(t){case"minimal":return"MINIMAL";case"low":return"LOW";case"medium":return"MEDIUM";case"high":return"HIGH"}}c(Ne,"getGeminiCliThinkingLevel");export{xe as buildRequest,Ee as extractRetryDelay,O as streamGoogleGeminiCli,Oe as streamSimpleGoogleGeminiCli};

package/dist/core/lib/ai/src/providers/google-shared.d.ts

@@ -0,0 +1,70 @@
+/**
+ * Shared utilities for Google Generative AI and Google Cloud Code Assist providers.
+ */
+/**
+ * [WHO]: isThinkingPart, retainThoughtSignature, requiresToolCallId, convertMessages, convertTools
+ * [FROM]: Depends on @google/genai, ../utils/sanitize-unicode.js, ./transform-messages.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/google-shared.ts -
+ */
+import { type Content, FinishReason, FunctionCallingConfigMode, type Part } from "@google/genai";
+import type { Context, Model, StopReason, Tool } from "../types.js";
+type GoogleApiType = "google-generative-ai" | "google-gemini-cli" | "google-vertex";
+/**
+ * Determines whether a streamed Gemini `Part` should be treated as "thinking".
+ *
+ * Protocol note (Gemini / Vertex AI thought signatures):
+ * - `thought: true` is the definitive marker for thinking content (thought summaries).
+ * - `thoughtSignature` is an encrypted representation of the model's internal thought process
+ *   used to preserve reasoning context across multi-turn interactions.
+ * - `thoughtSignature` can appear on ANY part type (text, functionCall, etc.) - it does NOT
+ *   indicate the part itself is thinking content.
+ * - For non-functionCall responses, the signature appears on the last part for context replay.
+ * - When persisting/replaying model outputs, signature-bearing parts must be preserved as-is;
+ *   do not merge/move signatures across parts.
+ *
+ * See: https://ai.google.dev/gemini-api/docs/thought-signatures
+ */
+export declare function isThinkingPart(part: Pick<Part, "thought" | "thoughtSignature">): boolean;
+/**
+ * Retain thought signatures during streaming.
+ *
+ * Some backends only send `thoughtSignature` on the first delta for a given part/block; later deltas may omit it.
+ * This helper preserves the last non-empty signature for the current block.
+ *
+ * Note: this does NOT merge or move signatures across distinct response parts. It only prevents
+ * a signature from being overwritten with `undefined` within the same streamed block.
+ */
+export declare function retainThoughtSignature(existing: string | undefined, incoming: string | undefined): string | undefined;
+/**
+ * Models via Google APIs that require explicit tool call IDs in function calls/responses.
+ */
+export declare function requiresToolCallId(modelId: string): boolean;
+/**
+ * Convert internal messages to Gemini Content[] format.
+ */
+export declare function convertMessages<T extends GoogleApiType>(model: Model<T>, context: Context): Content[];
+/**
+ * Convert tools to Gemini function declarations format.
+ *
+ * By default uses `parametersJsonSchema` which supports full JSON Schema (including
+ * anyOf, oneOf, const, etc.). Set `useParameters` to true to use the legacy `parameters`
+ * field instead (OpenAPI 3.03 Schema). This is needed for Cloud Code Assist with Claude
+ * models, where the API translates `parameters` into Anthropic's `input_schema`.
+ */
+export declare function convertTools(tools: Tool[], useParameters?: boolean): {
+    functionDeclarations: Record<string, unknown>[];
+}[] | undefined;
+/**
+ * Map tool choice string to Gemini FunctionCallingConfigMode.
+ */
+export declare function mapToolChoice(choice: string): FunctionCallingConfigMode;
+/**
+ * Map Gemini FinishReason to our StopReason.
+ */
+export declare function mapStopReason(reason: FinishReason): StopReason;
+/**
+ * Map string finish reason to our StopReason (for raw API responses).
+ */
+export declare function mapStopReasonString(reason: string): StopReason;
+export {};

package/dist/core/lib/ai/src/providers/google-shared.js

@@ -0,0 +1,2 @@
+var O=Object.defineProperty;var i=(t,s)=>O(t,"name",{value:s,configurable:!0});import{FinishReason as r,FunctionCallingConfigMode as g}from"@google/genai";import{sanitizeSurrogates as f}from"../utils/sanitize-unicode.js";import{transformMessages as N}from"./transform-messages.js";function F(t){return t.thought===!0}i(F,"isThinkingPart");function G(t,s){return typeof s=="string"&&s.length>0?s:t}i(G,"retainThoughtSignature");const _=/^[A-Za-z0-9+/]+={0,2}$/;function R(t){return!t||t.length%4!==0?!1:_.test(t)}i(R,"isValidThoughtSignature");function d(t,s){return t&&R(s)?s:void 0}i(d,"resolveThoughtSignature");function T(t){return t.startsWith("claude-")||t.startsWith("gpt-oss-")}i(T,"requiresToolCallId");function L(t,s){const o=[],I=i(n=>T(t.id)?n.replace(/[^a-zA-Z0-9_-]/g,"_").slice(0,64):n,"normalizeToolCallId"),A=N(s.messages,t,I);for(const n of A)if(n.role==="user")if(typeof n.content=="string")o.push({role:"user",parts:[{text:f(n.content)}]});else{const a=n.content.map(e=>e.type==="text"?{text:f(e.text)}:e.type==="document"?{text:"[PDF document - content not available in Google format]"}:{inlineData:{mimeType:e.mimeType,data:e.data}}),c=t.input.includes("image")?a:a.filter(e=>e.text!==void 0);if(c.length===0)continue;o.push({role:"user",parts:c})}else if(n.role==="assistant"){const a=[],c=n.provider===t.provider&&n.model===t.id;for(const e of n.content)if(e.type==="text"){if(!e.text||e.text.trim()==="")continue;const u=d(c,e.textSignature);a.push({text:f(e.text),...u&&{thoughtSignature:u}})}else if(e.type==="thinking"){if(!e.thinking||e.thinking.trim()==="")continue;if(c){const u=d(c,e.thinkingSignature);a.push({thought:!0,text:f(e.thinking),...u&&{thoughtSignature:u}})}else a.push({text:f(e.thinking)})}else if(e.type==="toolCall"){const u=d(c,e.thoughtSignature);if(t.id.toLowerCase().includes("gemini-3")&&!u){const p=JSON.stringify(e.arguments??{},null,2);a.push({text:`[Historical context: a different model called tool "${e.name}" with arguments: ${p}. Do not mimic this format - use proper function calling.]`})}else{const p={functionCall:{name:e.name,args:e.arguments??{},...T(t.id)?{id:e.id}:{}}};u&&(p.thoughtSignature=u),a.push(p)}}if(a.length===0)continue;o.push({role:"model",parts:a})}else if(n.role==="toolResult"){const c=n.content.filter(l=>l.type==="text").map(l=>l.text).join(`
+`),e=t.input.includes("image")?n.content.filter(l=>l.type==="image"):[],u=c.length>0,h=e.length>0,p=t.id.includes("gemini-3"),x=u?f(c):h?"(see attached image)":"",S=e.map(l=>({inlineData:{mimeType:l.mimeType,data:l.data}})),C=T(t.id),E={functionResponse:{name:n.toolName,response:n.isError?{error:x}:{output:x},...h&&p&&{parts:S},...C?{id:n.toolCallId}:{}}},m=o[o.length-1];m?.role==="user"&&m.parts?.some(l=>l.functionResponse)?m.parts.push(E):o.push({role:"user",parts:[E]}),h&&!p&&o.push({role:"user",parts:[{text:"Tool result image:"},...S]})}return o}i(L,"convertMessages");function v(t,s=!1){if(t.length!==0)return[{functionDeclarations:t.map(o=>({name:o.name,description:o.description,...s?{parameters:o.parameters}:{parametersJsonSchema:o.parameters}}))}]}i(v,"convertTools");function k(t){switch(t){case"auto":return g.AUTO;case"none":return g.NONE;case"any":return g.ANY;default:return g.AUTO}}i(k,"mapToolChoice");function w(t){switch(t){case r.STOP:return"stop";case r.MAX_TOKENS:return"length";case r.BLOCKLIST:case r.PROHIBITED_CONTENT:case r.SPII:case r.SAFETY:case r.IMAGE_SAFETY:case r.IMAGE_PROHIBITED_CONTENT:case r.IMAGE_RECITATION:case r.IMAGE_OTHER:case r.RECITATION:case r.FINISH_REASON_UNSPECIFIED:case r.OTHER:case r.LANGUAGE:case r.MALFORMED_FUNCTION_CALL:case r.UNEXPECTED_TOOL_CALL:case r.NO_IMAGE:return"error";default:{const s=t;throw new Error(`Unhandled stop reason: ${s}`)}}}i(w,"mapStopReason");function U(t){switch(t){case"STOP":return"stop";case"MAX_TOKENS":return"length";default:return"error"}}i(U,"mapStopReasonString");export{L as convertMessages,v as convertTools,F as isThinkingPart,w as mapStopReason,U as mapStopReasonString,k as mapToolChoice,T as requiresToolCallId,G as retainThoughtSignature};

package/dist/core/lib/ai/src/providers/google-vertex.d.ts

@@ -0,0 +1,20 @@
+/**
+ * [WHO]: GoogleVertexOptions, streamGoogleVertex, streamSimpleGoogleVertex
+ * [FROM]: Depends on ../models.js, ../utils/event-stream.js, ../utils/sanitize-unicode.js, ./simple-options.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/google-vertex.ts -
+ */
+import type { SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+import type { GoogleThinkingLevel } from "./google-gemini-cli.js";
+export interface GoogleVertexOptions extends StreamOptions {
+    toolChoice?: "auto" | "none" | "any";
+    thinking?: {
+        enabled: boolean;
+        budgetTokens?: number;
+        level?: GoogleThinkingLevel;
+    };
+    project?: string;
+    location?: string;
+}
+export declare const streamGoogleVertex: StreamFunction<"google-vertex", GoogleVertexOptions>;
+export declare const streamSimpleGoogleVertex: StreamFunction<"google-vertex", SimpleStreamOptions>;

package/dist/core/lib/ai/src/providers/google-vertex.js

@@ -0,0 +1 @@
+var T=Object.defineProperty;var c=(e,o)=>T(e,"name",{value:o,configurable:!0});import{GoogleGenAI as E,ThinkingLevel as p}from"@google/genai";import{calculateCost as L}from"../models.js";import{AssistantMessageEventStream as b}from"../utils/event-stream.js";import{sanitizeSurrogates as w}from"../utils/sanitize-unicode.js";import{convertMessages as G,convertTools as M,isThinkingPart as S,mapStopReason as R,mapToolChoice as v,retainThoughtSignature as y}from"./google-shared.js";import{buildBaseOptions as N,clampReasoning as P}from"./simple-options.js";const D="v1",A={THINKING_LEVEL_UNSPECIFIED:p.THINKING_LEVEL_UNSPECIFIED,MINIMAL:p.MINIMAL,LOW:p.LOW,MEDIUM:p.MEDIUM,HIGH:p.HIGH};let U=0;const k=c((e,o,n)=>{const i=new b;return(async()=>{const t={role:"assistant",content:[],api:"google-vertex",provider:e.provider,model:e.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const s=V(n),u=j(n),h=H(e,s,u,n?.headers),I=W(e,o,n);n?.onPayload?.(I);const _=await h.models.generateContentStream(I);i.push({type:"start",partial:t});let a=null;const C=t.content,l=c(()=>C.length-1,"blockIndex");for await(const d of _){const f=d.candidates?.[0];if(f?.content?.parts)for(const r of f.content.parts){if(r.text!==void 0){const g=S(r);(!a||g&&a.type!=="thinking"||!g&&a.type!=="text")&&(a&&(a.type==="text"?i.push({type:"text_end",contentIndex:C.length-1,content:a.text,partial:t}):i.push({type:"thinking_end",contentIndex:l(),content:a.thinking,partial:t})),g?(a={type:"thinking",thinking:"",thinkingSignature:void 0},t.content.push(a),i.push({type:"thinking_start",contentIndex:l(),partial:t})):(a={type:"text",text:""},t.content.push(a),i.push({type:"text_start",contentIndex:l(),partial:t}))),a.type==="thinking"?(a.thinking+=r.text,a.thinkingSignature=y(a.thinkingSignature,r.thoughtSignature),i.push({type:"thinking_delta",contentIndex:l(),delta:r.text,partial:t})):(a.text+=r.text,a.textSignature=y(a.textSignature,r.thoughtSignature),i.push({type:"text_delta",contentIndex:l(),delta:r.text,partial:t}))}if(r.functionCall){a&&(a.type==="text"?i.push({type:"text_end",contentIndex:l(),content:a.text,partial:t}):i.push({type:"thinking_end",contentIndex:l(),content:a.thinking,partial:t}),a=null);const g=r.functionCall.id,m={type:"toolCall",id:!g||t.content.some(x=>x.type==="toolCall"&&x.id===g)?`${r.functionCall.name}_${Date.now()}_${++U}`:g,name:r.functionCall.name||"",arguments:r.functionCall.args??{},...r.thoughtSignature&&{thoughtSignature:r.thoughtSignature}};t.content.push(m),i.push({type:"toolcall_start",contentIndex:l(),partial:t}),i.push({type:"toolcall_delta",contentIndex:l(),delta:JSON.stringify(m.arguments),partial:t}),i.push({type:"toolcall_end",contentIndex:l(),toolCall:m,partial:t})}}f?.finishReason&&(t.stopReason=R(f.finishReason),t.content.some(r=>r.type==="toolCall")&&(t.stopReason="toolUse")),d.usageMetadata&&(t.usage={input:d.usageMetadata.promptTokenCount||0,output:(d.usageMetadata.candidatesTokenCount||0)+(d.usageMetadata.thoughtsTokenCount||0),cacheRead:d.usageMetadata.cachedContentTokenCount||0,cacheWrite:0,totalTokens:d.usageMetadata.totalTokenCount||0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},L(e,t.usage))}if(a&&(a.type==="text"?i.push({type:"text_end",contentIndex:l(),content:a.text,partial:t}):i.push({type:"thinking_end",contentIndex:l(),content:a.thinking,partial:t})),n?.signal?.aborted)throw new Error("Request was aborted");if(t.stopReason==="aborted"||t.stopReason==="error")throw new Error("An unknown error occurred");i.push({type:"done",reason:t.stopReason,message:t}),i.end()}catch(s){for(const u of t.content)"index"in u&&delete u.index;t.stopReason=n?.signal?.aborted?"aborted":"error",t.errorMessage=s instanceof Error?s.message:JSON.stringify(s),i.push({type:"error",reason:t.stopReason,error:t}),i.end()}})(),i},"streamGoogleVertex"),et=c((e,o,n)=>{const i=N(e,n,void 0),t=n?.toolChoice;if(!n?.reasoning)return k(e,o,{...i,toolChoice:t,thinking:{enabled:!1}});const s=P(n.reasoning),u=e;return O(u)||J(u)?k(e,o,{...i,toolChoice:t,thinking:{enabled:!0,level:B(s,u)}}):k(e,o,{...i,toolChoice:t,thinking:{enabled:!0,budgetTokens:q(u,s,n.thinkingBudgets)}})},"streamSimpleGoogleVertex");function H(e,o,n,i){const t={};(e.headers||i)&&(t.headers={...e.headers,...i});const s=Object.values(t).some(Boolean);return new E({vertexai:!0,project:o,location:n,apiVersion:D,httpOptions:s?t:void 0})}c(H,"createClient");function V(e){const o=e?.project||process.env.GOOGLE_CLOUD_PROJECT||process.env.GCLOUD_PROJECT;if(!o)throw new Error("Vertex AI requires a project ID. Set GOOGLE_CLOUD_PROJECT/GCLOUD_PROJECT or pass project in options.");return o}c(V,"resolveProject");function j(e){const o=e?.location||process.env.GOOGLE_CLOUD_LOCATION;if(!o)throw new Error("Vertex AI requires a location. Set GOOGLE_CLOUD_LOCATION or pass location in options.");return o}c(j,"resolveLocation");function W(e,o,n={}){const i=G(e,o),t={};n.temperature!==void 0&&(t.temperature=n.temperature),n.maxTokens!==void 0&&(t.maxOutputTokens=n.maxTokens);const s={...Object.keys(t).length>0&&t,...o.systemPrompt&&{systemInstruction:w(o.systemPrompt)},...o.tools&&o.tools.length>0&&{tools:M(o.tools)}};if(o.tools&&o.tools.length>0&&n.toolChoice?s.toolConfig={functionCallingConfig:{mode:v(n.toolChoice)}}:s.toolConfig=void 0,n.thinking?.enabled&&e.reasoning){const h={includeThoughts:!0};n.thinking.level!==void 0?h.thinkingLevel=A[n.thinking.level]:n.thinking.budgetTokens!==void 0&&(h.thinkingBudget=n.thinking.budgetTokens),s.thinkingConfig=h}if(n.signal){if(n.signal.aborted)throw new Error("Request aborted");s.abortSignal=n.signal}return{model:e.id,contents:i,config:s}}c(W,"buildParams");function O(e){return e.id.includes("3-pro")}c(O,"isGemini3ProModel");function J(e){return e.id.includes("3-flash")}c(J,"isGemini3FlashModel");function B(e,o){if(O(o))switch(e){case"minimal":case"low":return"LOW";case"medium":case"high":return"HIGH"}switch(e){case"minimal":return"MINIMAL";case"low":return"LOW";case"medium":return"MEDIUM";case"high":return"HIGH"}}c(B,"getGemini3ThinkingLevel");function q(e,o,n){return n?.[o]!==void 0?n[o]:e.id.includes("2.5-pro")?{minimal:128,low:2048,medium:8192,high:32768}[o]:e.id.includes("2.5-flash")?{minimal:128,low:2048,medium:8192,high:24576}[o]:-1}c(q,"getGoogleBudget");export{k as streamGoogleVertex,et as streamSimpleGoogleVertex};

package/dist/core/lib/ai/src/providers/google.d.ts

@@ -0,0 +1,18 @@
+/**
+ * [WHO]: GoogleOptions, streamGoogle, streamSimpleGoogle
+ * [FROM]: Depends on ../env-api-keys.js, ../models.js, ../utils/event-stream.js, ../utils/sanitize-unicode.js, ./simple-options.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/google.ts -
+ */
+import type { SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+import type { GoogleThinkingLevel } from "./google-gemini-cli.js";
+export interface GoogleOptions extends StreamOptions {
+    toolChoice?: "auto" | "none" | "any";
+    thinking?: {
+        enabled: boolean;
+        budgetTokens?: number;
+        level?: GoogleThinkingLevel;
+    };
+}
+export declare const streamGoogle: StreamFunction<"google-generative-ai", GoogleOptions>;
+export declare const streamSimpleGoogle: StreamFunction<"google-generative-ai", SimpleStreamOptions>;

package/dist/core/lib/ai/src/providers/google.js

@@ -0,0 +1 @@
+var w=Object.defineProperty;var c=(e,o)=>w(e,"name",{value:o,configurable:!0});import{GoogleGenAI as S}from"@google/genai";import{getEnvApiKey as x}from"../env-api-keys.js";import{calculateCost as T}from"../models.js";import{AssistantMessageEventStream as R}from"../utils/event-stream.js";import{sanitizeSurrogates as v}from"../utils/sanitize-unicode.js";import{convertMessages as M,convertTools as _,isThinkingPart as G,mapStopReason as O,mapToolChoice as E,retainThoughtSignature as C}from"./google-shared.js";import{buildBaseOptions as P,clampReasoning as A}from"./simple-options.js";let N=0;const m=c((e,o,n)=>{const i=new R;return(async()=>{const t={role:"assistant",content:[],api:"google-generative-ai",provider:e.provider,model:e.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const s=n?.apiKey||x(e.provider)||"",g=W(e,s,n?.headers),u=K(e,o,n);n?.onPayload?.(u);const I=await g.models.generateContentStream(u);i.push({type:"start",partial:t});let a=null;const k=t.content,l=c(()=>k.length-1,"blockIndex");for await(const d of I){const p=d.candidates?.[0];if(p?.content?.parts)for(const r of p.content.parts){if(r.text!==void 0){const h=G(r);(!a||h&&a.type!=="thinking"||!h&&a.type!=="text")&&(a&&(a.type==="text"?i.push({type:"text_end",contentIndex:k.length-1,content:a.text,partial:t}):i.push({type:"thinking_end",contentIndex:l(),content:a.thinking,partial:t})),h?(a={type:"thinking",thinking:"",thinkingSignature:void 0},t.content.push(a),i.push({type:"thinking_start",contentIndex:l(),partial:t})):(a={type:"text",text:""},t.content.push(a),i.push({type:"text_start",contentIndex:l(),partial:t}))),a.type==="thinking"?(a.thinking+=r.text,a.thinkingSignature=C(a.thinkingSignature,r.thoughtSignature),i.push({type:"thinking_delta",contentIndex:l(),delta:r.text,partial:t})):(a.text+=r.text,a.textSignature=C(a.textSignature,r.thoughtSignature),i.push({type:"text_delta",contentIndex:l(),delta:r.text,partial:t}))}if(r.functionCall){a&&(a.type==="text"?i.push({type:"text_end",contentIndex:l(),content:a.text,partial:t}):i.push({type:"thinking_end",contentIndex:l(),content:a.thinking,partial:t}),a=null);const h=r.functionCall.id,f={type:"toolCall",id:!h||t.content.some(y=>y.type==="toolCall"&&y.id===h)?`${r.functionCall.name}_${Date.now()}_${++N}`:h,name:r.functionCall.name||"",arguments:r.functionCall.args??{},...r.thoughtSignature&&{thoughtSignature:r.thoughtSignature}};t.content.push(f),i.push({type:"toolcall_start",contentIndex:l(),partial:t}),i.push({type:"toolcall_delta",contentIndex:l(),delta:JSON.stringify(f.arguments),partial:t}),i.push({type:"toolcall_end",contentIndex:l(),toolCall:f,partial:t})}}p?.finishReason&&(t.stopReason=O(p.finishReason),t.content.some(r=>r.type==="toolCall")&&(t.stopReason="toolUse")),d.usageMetadata&&(t.usage={input:d.usageMetadata.promptTokenCount||0,output:(d.usageMetadata.candidatesTokenCount||0)+(d.usageMetadata.thoughtsTokenCount||0),cacheRead:d.usageMetadata.cachedContentTokenCount||0,cacheWrite:0,totalTokens:d.usageMetadata.totalTokenCount||0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},T(e,t.usage))}if(a&&(a.type==="text"?i.push({type:"text_end",contentIndex:l(),content:a.text,partial:t}):i.push({type:"thinking_end",contentIndex:l(),content:a.thinking,partial:t})),n?.signal?.aborted)throw new Error("Request was aborted");if(t.stopReason==="aborted"||t.stopReason==="error")throw new Error("An unknown error occurred");i.push({type:"done",reason:t.stopReason,message:t}),i.end()}catch(s){for(const g of t.content)"index"in g&&delete g.index;t.stopReason=n?.signal?.aborted?"aborted":"error",t.errorMessage=s instanceof Error?s.message:JSON.stringify(s),i.push({type:"error",reason:t.stopReason,error:t}),i.end()}})(),i},"streamGoogle"),X=c((e,o,n)=>{const i=n?.apiKey||x(e.provider);if(!i)throw new Error(`No API key for provider: ${e.provider}`);const t=P(e,n,i),s=n?.toolChoice;if(!n?.reasoning)return m(e,o,{...t,toolChoice:s,thinking:{enabled:!1}});const g=A(n.reasoning),u=e;return b(u)||L(u)?m(e,o,{...t,toolChoice:s,thinking:{enabled:!0,level:U(g,u)}}):m(e,o,{...t,toolChoice:s,thinking:{enabled:!0,budgetTokens:B(u,g,n.thinkingBudgets)}})},"streamSimpleGoogle");function W(e,o,n){const i={};return e.baseUrl&&(i.baseUrl=e.baseUrl,i.apiVersion=""),(e.headers||n)&&(i.headers={...e.headers,...n}),new S({apiKey:o,httpOptions:Object.keys(i).length>0?i:void 0})}c(W,"createClient");function K(e,o,n={}){const i=M(e,o),t={};n.temperature!==void 0&&(t.temperature=n.temperature),n.maxTokens!==void 0&&(t.maxOutputTokens=n.maxTokens);const s={...Object.keys(t).length>0&&t,...o.systemPrompt&&{systemInstruction:v(o.systemPrompt)},...o.tools&&o.tools.length>0&&{tools:_(o.tools)}};if(o.tools&&o.tools.length>0&&n.toolChoice?s.toolConfig={functionCallingConfig:{mode:E(n.toolChoice)}}:s.toolConfig=void 0,n.thinking?.enabled&&e.reasoning){const u={includeThoughts:!0};n.thinking.level!==void 0?u.thinkingLevel=n.thinking.level:n.thinking.budgetTokens!==void 0&&(u.thinkingBudget=n.thinking.budgetTokens),s.thinkingConfig=u}if(n.signal){if(n.signal.aborted)throw new Error("Request aborted");s.abortSignal=n.signal}return{model:e.id,contents:i,config:s}}c(K,"buildParams");function b(e){return e.id.includes("3-pro")}c(b,"isGemini3ProModel");function L(e){return e.id.includes("3-flash")}c(L,"isGemini3FlashModel");function U(e,o){if(b(o))switch(e){case"minimal":case"low":return"LOW";case"medium":case"high":return"HIGH"}switch(e){case"minimal":return"MINIMAL";case"low":return"LOW";case"medium":return"MEDIUM";case"high":return"HIGH"}}c(U,"getGemini3ThinkingLevel");function B(e,o,n){return n?.[o]!==void 0?n[o]:e.id.includes("2.5-pro")?{minimal:128,low:2048,medium:8192,high:32768}[o]:e.id.includes("2.5-flash")?{minimal:128,low:2048,medium:8192,high:24576}[o]:-1}c(B,"getGoogleBudget");export{m as streamGoogle,X as streamSimpleGoogle};

package/dist/core/lib/ai/src/providers/openai-codex-responses.d.ts

@@ -0,0 +1,8 @@
+import type { SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+export interface OpenAICodexResponsesOptions extends StreamOptions {
+    reasoningEffort?: "none" | "minimal" | "low" | "medium" | "high" | "xhigh";
+    reasoningSummary?: "auto" | "concise" | "detailed" | "off" | "on" | null;
+    textVerbosity?: "low" | "medium" | "high";
+}
+export declare const streamOpenAICodexResponses: StreamFunction<"openai-codex-responses", OpenAICodexResponsesOptions>;
+export declare const streamSimpleOpenAICodexResponses: StreamFunction<"openai-codex-responses", SimpleStreamOptions>;

package/dist/core/lib/ai/src/providers/openai-codex-responses.js

@@ -0,0 +1,7 @@
+var N=Object.defineProperty;var a=(e,t)=>N(e,"name",{value:t,configurable:!0});let g=null;typeof process<"u"&&(process.versions?.node||process.versions?.bun)&&import("node:os").then(e=>{g=e});import{getEnvApiKey as R}from"../env-api-keys.js";import{supportsXhigh as q}from"../models.js";import{AssistantMessageEventStream as B}from"../utils/event-stream.js";import{convertResponsesMessages as D,convertResponsesTools as M,processResponsesStream as T}from"./openai-responses-shared.js";import{buildBaseOptions as P,clampReasoning as U}from"./simple-options.js";const I="https://chatgpt.com/backend-api",J="https://api.openai.com/auth",S=3,_=1e3,K=new Set(["openai","openai-codex","opencode"]),H=new Set(["completed","incomplete","failed","cancelled","queued","in_progress"]);function X(e,t){return e===429||e===500||e===502||e===503||e===504?!0:/rate.?limit|overloaded|service.?unavailable|upstream.?connect|connection.?refused/i.test(t)}a(X,"isRetryableError");function x(e,t){return new Promise((r,n)=>{if(t?.aborted){n(new Error("Request was aborted"));return}const o=setTimeout(r,e);t?.addEventListener("abort",()=>{clearTimeout(o),n(new Error("Request was aborted"))})})}a(x,"sleep");const z=a((e,t,r)=>{const n=new B;return(async()=>{const o={role:"assistant",content:[],api:"openai-codex-responses",provider:e.provider,model:e.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const s=r?.apiKey||R(e.provider)||"";if(!s)throw new Error(`No API key for provider: ${e.provider}`);const c=ue(s),i=j(e,t,r);r?.onPayload?.(i);const u=de(e.headers,r?.headers,c,s,r?.sessionId),l=JSON.stringify(i),y=r?.transport||"sse";if(y!=="sse"){let w=!1;try{if(await ie(V(e.baseUrl),i,u,o,n,e,()=>{w=!0},r),r?.signal?.aborted)throw new Error("Request was aborted");n.push({type:"done",reason:o.stopReason,message:o}),n.end();return}catch(d){if(y==="websocket"||w)throw d}}let p,f;for(let w=0;w<=S;w++){if(r?.signal?.aborted)throw new Error("Request was aborted");try{if(p=await fetch(A(e.baseUrl),{method:"POST",headers:u,body:l,signal:r?.signal}),p.ok)break;const d=await p.text();if(w<S&&X(p.status,d)){const $=_*2**w;await x($,r?.signal);continue}const b=new Response(d,{status:p.status,statusText:p.statusText}),v=await ce(b);throw new Error(v.friendlyMessage||v.message)}catch(d){if(d instanceof Error&&(d.name==="AbortError"||d.message==="Request was aborted"))throw new Error("Request was aborted");if(f=d instanceof Error?d:new Error(String(d)),w<S&&!f.message.includes("usage limit")){const b=_*2**w;await x(b,r?.signal);continue}throw f}}if(!p?.ok)throw f??new Error("Failed after retries");if(!p.body)throw new Error("No response body");if(n.push({type:"start",partial:o}),await Y(p,o,n,e),r?.signal?.aborted)throw new Error("Request was aborted");n.push({type:"done",reason:o.stopReason,message:o}),n.end()}catch(s){o.stopReason=r?.signal?.aborted?"aborted":"error",o.errorMessage=s instanceof Error?s.message:String(s),n.push({type:"error",reason:o.stopReason,error:o}),n.end()}})(),n},"streamOpenAICodexResponses"),he=a((e,t,r)=>{const n=r?.apiKey||R(e.provider);if(!n)throw new Error(`No API key for provider: ${e.provider}`);const o=P(e,r,n),s=q(e)?r?.reasoning:U(r?.reasoning);return z(e,t,{...o,reasoningEffort:s})},"streamSimpleOpenAICodexResponses");function j(e,t,r){const n=D(e,t,K,{includeSystemPrompt:!1}),o={model:e.id,store:!1,stream:!0,instructions:t.systemPrompt,input:n,text:{verbosity:r?.textVerbosity||"medium"},include:["reasoning.encrypted_content"],prompt_cache_key:r?.sessionId,tool_choice:"auto",parallel_tool_calls:!0};return r?.temperature!==void 0&&(o.temperature=r.temperature),t.tools&&(o.tools=M(t.tools,{strict:null})),r?.reasoningEffort!==void 0&&(o.reasoning={effort:F(e.id,r.reasoningEffort),summary:r.reasoningSummary??"auto"}),o}a(j,"buildRequestBody");function F(e,t){const r=e.includes("/")?e.split("/").pop():e;return(r.startsWith("gpt-5.2")||r.startsWith("gpt-5.3"))&&t==="minimal"?"low":r==="gpt-5.1"&&t==="xhigh"?"high":r==="gpt-5.1-codex-mini"?t==="high"||t==="xhigh"?"high":"medium":t}a(F,"clampReasoningEffort");function A(e){const r=(e&&e.trim().length>0?e:I).replace(/\/+$/,"");return r.endsWith("/codex/responses")?r:r.endsWith("/codex")?`${r}/responses`:`${r}/codex/responses`}a(A,"resolveCodexUrl");function V(e){const t=new URL(A(e));return t.protocol==="https:"&&(t.protocol="wss:"),t.protocol==="http:"&&(t.protocol="ws:"),t.toString()}a(V,"resolveCodexWebSocketUrl");async function Y(e,t,r,n){await T(L(Q(e)),t,r,n)}a(Y,"processStream");async function*L(e){for await(const t of e){const r=typeof t.type=="string"?t.type:void 0;if(r){if(r==="error"){const n=t.code||"",o=t.message||"";throw new Error(`Codex error: ${o||n||JSON.stringify(t)}`)}if(r==="response.failed"){const n=t.response?.error?.message;throw new Error(n||"Codex response failed")}if(r==="response.done"||r==="response.completed"){const n=t.response,o=n&&{...n,status:G(n.status)};yield{...t,type:"response.completed",response:o};continue}yield t}}}a(L,"mapCodexEvents");function G(e){if(typeof e=="string")return H.has(e)?e:void 0}a(G,"normalizeCodexStatus");async function*Q(e){if(!e.body)return;const t=e.body.getReader(),r=new TextDecoder;let n="";for(;;){const{done:o,value:s}=await t.read();if(o)break;n+=r.decode(s,{stream:!0});let c=n.indexOf(`
+
+`);for(;c!==-1;){const i=n.slice(0,c);n=n.slice(c+2);const u=i.split(`
+`).filter(l=>l.startsWith("data:")).map(l=>l.slice(5).trim());if(u.length>0){const l=u.join(`
+`).trim();if(l&&l!=="[DONE]")try{yield JSON.parse(l)}catch{}}c=n.indexOf(`
+
+`)}}}a(Q,"parseSSE");const Z="responses_websockets=2026-02-06",ee=300*1e3,m=new Map;function te(){const e=globalThis.WebSocket;return typeof e!="function"?null:e}a(te,"getWebSocketConstructor");function re(e){const t={};for(const[r,n]of e.entries())t[r]=n;return t}a(re,"headersToRecord");function ne(e){const t=e.readyState;return typeof t=="number"?t:void 0}a(ne,"getWebSocketReadyState");function E(e){const t=ne(e);return t===void 0||t===1}a(E,"isWebSocketReusable");function h(e,t=1e3,r="done"){try{e.close(t,r)}catch{}}a(h,"closeWebSocketSilently");function O(e,t){t.idleTimer&&clearTimeout(t.idleTimer),t.idleTimer=setTimeout(()=>{t.busy||(h(t.socket,1e3,"idle_timeout"),m.delete(e))},ee)}a(O,"scheduleSessionWebSocketExpiry");async function k(e,t,r){const n=te();if(!n)throw new Error("WebSocket transport is not available in this runtime");const o=re(t);return o["OpenAI-Beta"]=Z,new Promise((s,c)=>{let i=!1,u;try{u=new n(e,{headers:o})}catch(d){c(d instanceof Error?d:new Error(String(d)));return}const l=a(()=>{i||(i=!0,w(),s(u))},"onOpen"),y=a(d=>{i||(i=!0,w(),c(W(d)))},"onError"),p=a(d=>{i||(i=!0,w(),c(C(d)))},"onClose"),f=a(()=>{i||(i=!0,w(),u.close(1e3,"aborted"),c(new Error("Request was aborted")))},"onAbort"),w=a(()=>{u.removeEventListener("open",l),u.removeEventListener("error",y),u.removeEventListener("close",p),r?.removeEventListener("abort",f)},"cleanup");u.addEventListener("open",l),u.addEventListener("error",y),u.addEventListener("close",p),r?.addEventListener("abort",f)})}a(k,"connectWebSocket");async function oe(e,t,r,n){if(!r){const i=await k(e,t,n);return{socket:i,release:a(({keep:u}={})=>{if(u===!1){h(i);return}h(i)},"release")}}const o=m.get(r);if(o){if(o.idleTimer&&(clearTimeout(o.idleTimer),o.idleTimer=void 0),!o.busy&&E(o.socket))return o.busy=!0,{socket:o.socket,release:a(({keep:i}={})=>{if(!i||!E(o.socket)){h(o.socket),m.delete(r);return}o.busy=!1,O(r,o)},"release")};if(o.busy){const i=await k(e,t,n);return{socket:i,release:a(()=>{h(i)},"release")}}E(o.socket)||(h(o.socket),m.delete(r))}const s=await k(e,t,n),c={socket:s,busy:!0};return m.set(r,c),{socket:s,release:a(({keep:i}={})=>{if(!i||!E(c.socket)){h(c.socket),c.idleTimer&&clearTimeout(c.idleTimer),m.get(r)===c&&m.delete(r);return}c.busy=!1,O(r,c)},"release")}}a(oe,"acquireWebSocket");function W(e){if(e&&typeof e=="object"&&"message"in e){const t=e.message;if(typeof t=="string"&&t.length>0)return new Error(t)}return new Error("WebSocket error")}a(W,"extractWebSocketError");function C(e){if(e&&typeof e=="object"){const t="code"in e?e.code:void 0,r="reason"in e?e.reason:void 0,n=typeof t=="number"?` ${t}`:"",o=typeof r=="string"&&r.length>0?` ${r}`:"";return new Error(`WebSocket closed${n}${o}`.trim())}return new Error("WebSocket closed")}a(C,"extractWebSocketCloseError");async function se(e){if(typeof e=="string")return e;if(e instanceof ArrayBuffer)return new TextDecoder().decode(new Uint8Array(e));if(ArrayBuffer.isView(e)){const t=e;return new TextDecoder().decode(new Uint8Array(t.buffer,t.byteOffset,t.byteLength))}if(e&&typeof e=="object"&&"arrayBuffer"in e){const r=await e.arrayBuffer();return new TextDecoder().decode(new Uint8Array(r))}return null}a(se,"decodeWebSocketData");async function*ae(e,t){const r=[];let n=null,o=!1,s=null,c=!1;const i=a(()=>{if(!n)return;const f=n;n=null,f()},"wake"),u=a(f=>{(async()=>{if(!f||typeof f!="object"||!("data"in f))return;const w=await se(f.data);if(w)try{const d=JSON.parse(w),b=typeof d.type=="string"?d.type:"";(b==="response.completed"||b==="response.done")&&(c=!0,o=!0),r.push(d),i()}catch{}})()},"onMessage"),l=a(f=>{s=W(f),o=!0,i()},"onError"),y=a(f=>{if(c){o=!0,i();return}s||(s=C(f)),o=!0,i()},"onClose"),p=a(()=>{s=new Error("Request was aborted"),o=!0,i()},"onAbort");e.addEventListener("message",u),e.addEventListener("error",l),e.addEventListener("close",y),t?.addEventListener("abort",p);try{for(;;){if(t?.aborted)throw new Error("Request was aborted");if(r.length>0){yield r.shift();continue}if(o)break;await new Promise(f=>{n=f})}if(s)throw s;if(!c)throw new Error("WebSocket stream closed before response.completed")}finally{e.removeEventListener("message",u),e.removeEventListener("error",l),e.removeEventListener("close",y),t?.removeEventListener("abort",p)}}a(ae,"parseWebSocket");async function ie(e,t,r,n,o,s,c,i){const{socket:u,release:l}=await oe(e,r,i?.sessionId,i?.signal);let y=!0;try{u.send(JSON.stringify({type:"response.create",...t})),c(),o.push({type:"start",partial:n}),await T(L(ae(u,i?.signal)),n,o,s),i?.signal?.aborted&&(y=!1)}catch(p){throw y=!1,p}finally{l({keep:y})}}a(ie,"processWebSocketStream");async function ce(e){const t=await e.text();let r=t||e.statusText||"Request failed",n;try{const s=JSON.parse(t)?.error;if(s){const c=s.code||s.type||"";if(/usage_limit_reached|usage_not_included|rate_limit_exceeded/i.test(c)||e.status===429){const i=s.plan_type?` (${s.plan_type.toLowerCase()} plan)`:"",u=s.resets_at?Math.max(0,Math.round((s.resets_at*1e3-Date.now())/6e4)):void 0,l=u!==void 0?` Try again in ~${u} min.`:"";n=`You have hit your ChatGPT usage limit${i}.${l}`.trim()}r=s.message||n||r}}catch{}return{message:r,friendlyMessage:n}}a(ce,"parseErrorResponse");function ue(e){try{const t=e.split(".");if(t.length!==3)throw new Error("Invalid token");const n=JSON.parse(atob(t[1]))?.[J]?.chatgpt_account_id;if(!n)throw new Error("No account ID in token");return n}catch{throw new Error("Failed to extract accountId from token")}}a(ue,"extractAccountId");function de(e,t,r,n,o){const s=new Headers(e);s.set("Authorization",`Bearer ${n}`),s.set("chatgpt-account-id",r),s.set("OpenAI-Beta","responses=experimental"),s.set("originator","catui");const c=g?`catui (${g.platform()} ${g.release()}; ${g.arch()})`:"catui (browser)";s.set("User-Agent",c),s.set("accept","text/event-stream"),s.set("content-type","application/json");for(const[i,u]of Object.entries(t||{}))s.set(i,u);return o&&s.set("session_id",o),s}a(de,"buildHeaders");export{z as streamOpenAICodexResponses,he as streamSimpleOpenAICodexResponses};

package/dist/core/lib/ai/src/providers/openai-completions.d.ts

@@ -0,0 +1,20 @@
+/**
+ * [WHO]: OpenAICompletionsOptions, streamOpenAICompletions, streamSimpleOpenAICompletions, convertMessages
+ * [FROM]: Depends on openai, ../env-api-keys.js, ../models.js, ../utils/event-stream.js, ../utils/json-parse.js
+ * [TO]: Consumed by core/lib/ai/src/index.ts
+ * [HERE]: core/lib/ai/src/providers/openai-completions.ts -
+ */
+import type { ChatCompletionMessageParam } from "openai/resources/chat/completions.js";
+import type { Context, Model, OpenAICompletionsCompat, SimpleStreamOptions, StreamFunction, StreamOptions } from "../types.js";
+export interface OpenAICompletionsOptions extends StreamOptions {
+    toolChoice?: "auto" | "none" | "required" | {
+        type: "function";
+        function: {
+            name: string;
+        };
+    };
+    reasoningEffort?: "minimal" | "low" | "medium" | "high" | "xhigh";
+}
+export declare const streamOpenAICompletions: StreamFunction<"openai-completions", OpenAICompletionsOptions>;
+export declare const streamSimpleOpenAICompletions: StreamFunction<"openai-completions", SimpleStreamOptions>;
+export declare function convertMessages(model: Model<"openai-completions">, context: Context, compat: Required<OpenAICompletionsCompat>): ChatCompletionMessageParam[];

package/dist/core/lib/ai/src/providers/openai-completions.js

@@ -0,0 +1,6 @@
+var M=Object.defineProperty;var _=(t,e)=>M(t,"name",{value:e,configurable:!0});import O from"openai";import{getEnvApiKey as q}from"../env-api-keys.js";import{calculateCost as N,supportsXhigh as P}from"../models.js";import{AssistantMessageEventStream as B}from"../utils/event-stream.js";import{parseStreamingJson as b}from"../utils/json-parse.js";import{sanitizeSurrogates as S}from"../utils/sanitize-unicode.js";import{buildCopilotDynamicHeaders as F,hasCopilotVisionInput as j}from"./github-copilot-headers.js";import{buildBaseOptions as D,clampReasoning as U}from"./simple-options.js";import{transformMessages as z}from"./transform-messages.js";import{getDebugLogger as G}from"../debug-logger.js";function $(t){let e=t.replace(/[^a-zA-Z0-9]/g,"");return e.length<9?e=e+"ABCDEFGHI".slice(0,9-e.length):e.length>9&&(e=e.slice(0,9)),e}_($,"normalizeMistralToolId");function K(t){for(const e of t)if(e.role==="toolResult"||e.role==="assistant"&&e.content.some(o=>o.type==="toolCall"))return!0;return!1}_(K,"hasToolHistory");const L=_((t,e,o)=>{const i=new B;return(async()=>{const s={role:"assistant",content:[],api:t.api,provider:t.provider,model:t.id,usage:{input:0,output:0,cacheRead:0,cacheWrite:0,totalTokens:0,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},stopReason:"stop",timestamp:Date.now()};try{const r=o?.apiKey||q(t.provider)||"",u=H(t,e,r,o?.headers),a=J(t,e,o);o?.onPayload?.(a);const g=G();g.logProviderRequest(t.provider||"unknown",t.id||"unknown",a);const f=await u.chat.completions.create(a,{signal:o?.signal});i.push({type:"start",partial:s});let n=null;const d=s.content,y=_(()=>d.length-1,"blockIndex"),R=_(c=>{c&&(c.type==="text"?i.push({type:"text_end",contentIndex:y(),content:c.text,partial:s}):c.type==="thinking"?i.push({type:"thinking_end",contentIndex:y(),content:c.thinking,partial:s}):c.type==="toolCall"&&(c.arguments=b(c.partialArgs),delete c.partialArgs,i.push({type:"toolcall_end",contentIndex:y(),toolCall:c,partial:s})))},"finishCurrentBlock");let m="",T=!1;const A=t.provider||"unknown",p=t.id||"unknown";for await(const c of f){if(g.logProviderChunk(A,p,c),c.usage){const E=c.usage.prompt_tokens_details?.cached_tokens||0,v=c.usage.completion_tokens_details?.reasoning_tokens||0,C=(c.usage.prompt_tokens||0)-E,l=(c.usage.completion_tokens||0)+v;s.usage={input:C,output:l,cacheRead:E,cacheWrite:0,totalTokens:C+l+E,cost:{input:0,output:0,cacheRead:0,cacheWrite:0,total:0}},N(t,s.usage)}const h=c.choices?.[0];if(h&&(h.finish_reason&&(s.stopReason=X(h.finish_reason)),h.delta)){if((h.delta.content!==void 0||h.delta.tool_calls!==void 0)&&g.logContentParse("choice.delta",{content:h.delta.content,tool_calls:h.delta.tool_calls,reasoning_content:h.delta.reasoning_content,reasoning:h.delta.reasoning}),h.delta.content!==null&&h.delta.content!==void 0&&h.delta.content.length>0)for(m+=h.delta.content;m.length>0;)if(T){const l=["</thinking>","</think>"];let k=-1,I=0;for(const x of l){const w=m.indexOf(x);w!==-1&&(k===-1||w<k)&&(k=w,I=x.length)}if(k!==-1){const x=m.slice(0,k);x&&((!n||n.type!=="thinking")&&(R(n),n={type:"thinking",thinking:"",thinkingSignature:"minimax-thinking-tag"},s.content.push(n),i.push({type:"thinking_start",contentIndex:y(),partial:s})),n.type==="thinking"&&(n.thinking+=x,i.push({type:"thinking_delta",contentIndex:y(),delta:x,partial:s}))),m=m.slice(k+I),T=!1,n&&n.type==="thinking"&&(R(n),n=null)}else{const x=m;m="",(!n||n.type!=="thinking")&&(R(n),n={type:"thinking",thinking:"",thinkingSignature:"minimax-thinking-tag"},s.content.push(n),i.push({type:"thinking_start",contentIndex:y(),partial:s})),n.type==="thinking"&&(n.thinking+=x,i.push({type:"thinking_delta",contentIndex:y(),delta:x,partial:s}))}}else{const l=["<thinking>","<think>"];let k=-1,I=0;for(const x of l){const w=m.indexOf(x);w!==-1&&(k===-1||w<k)&&(k=w,I=x.length)}if(k!==-1){const x=m.slice(0,k);m=m.slice(k+I),T=!0,x&&((!n||n.type!=="text")&&(R(n),n={type:"text",text:""},s.content.push(n),i.push({type:"text_start",contentIndex:y(),partial:s})),n.type==="text"&&(n.text+=x,i.push({type:"text_delta",contentIndex:y(),delta:x,partial:s})))}else{const x=m;m="",(!n||n.type!=="text")&&(R(n),n={type:"text",text:""},s.content.push(n),i.push({type:"text_start",contentIndex:y(),partial:s})),n.type==="text"&&(n.text+=x,i.push({type:"text_delta",contentIndex:y(),delta:x,partial:s}))}}const E=["reasoning_content","reasoning","reasoning_text"];let v=null;for(const l of E)if(h.delta[l]!==null&&h.delta[l]!==void 0&&h.delta[l].length>0&&!v){v=l;break}if(v&&((!n||n.type!=="thinking")&&(R(n),n={type:"thinking",thinking:"",thinkingSignature:v},s.content.push(n),i.push({type:"thinking_start",contentIndex:y(),partial:s})),n.type==="thinking")){const l=h.delta[v];n.thinking+=l,i.push({type:"thinking_delta",contentIndex:y(),delta:l,partial:s})}if(h?.delta?.tool_calls){for(const l of h.delta.tool_calls)if((!n||n.type!=="toolCall"||l.id&&n.id!==l.id)&&(R(n),n={type:"toolCall",id:l.id||"",name:l.function?.name||"",arguments:{},partialArgs:""},s.content.push(n),i.push({type:"toolcall_start",contentIndex:y(),partial:s})),n.type==="toolCall"){l.id&&(n.id=l.id),l.function?.name&&(n.name=l.function.name);let k="";l.function?.arguments&&(k=l.function.arguments,n.partialArgs+=l.function.arguments,n.arguments=b(n.partialArgs)),i.push({type:"toolcall_delta",contentIndex:y(),delta:k,partial:s})}}const C=h.delta.reasoning_details;if(C&&Array.isArray(C)){for(const l of C)if(l.type==="reasoning.encrypted"&&l.id&&l.data){const k=s.content.find(I=>I.type==="toolCall"&&I.id===l.id);k&&(k.thoughtSignature=JSON.stringify(l))}}}}if(R(n),o?.signal?.aborted)throw new Error("Request was aborted");if(s.stopReason==="aborted"||s.stopReason==="error")throw new Error("An unknown error occurred");i.push({type:"done",reason:s.stopReason,message:s}),i.end()}catch(r){for(const a of s.content)delete a.index;s.stopReason=o?.signal?.aborted?"aborted":"error",s.errorMessage=r instanceof Error?r.message:JSON.stringify(r);const u=r?.error?.metadata?.raw;if(u&&(s.errorMessage+=`
+${u}`),r&&typeof r=="object"&&"headers"in r){const a=r.headers;a&&typeof a=="object"&&(s.errorHeaders=Object.fromEntries(Object.entries(a).map(([g,f])=>[g.toLowerCase(),String(f)])))}i.push({type:"error",reason:s.stopReason,error:s}),i.end()}})(),i},"streamOpenAICompletions"),gt=_((t,e,o)=>{const i=o?.apiKey||q(t.provider);if(!i)throw new Error(`No API key for provider: ${t.provider}`);const s=D(t,o,i),r=P(t)?o?.reasoning:U(o?.reasoning),u=o?.toolChoice;return L(t,e,{...s,reasoningEffort:r,toolChoice:u})},"streamSimpleOpenAICompletions");function H(t,e,o,i){if(!o){if(!process.env.OPENAI_API_KEY)throw new Error("OpenAI API key is required. Set OPENAI_API_KEY environment variable or pass it as an argument.");o=process.env.OPENAI_API_KEY}const s={...t.headers};if(t.provider==="github-copilot"){const r=j(e.messages),u=F({messages:e.messages,hasImages:r});Object.assign(s,u)}return i&&Object.assign(s,i),new O({apiKey:o,baseURL:t.baseUrl,dangerouslyAllowBrowser:!0,defaultHeaders:s})}_(H,"createClient");function J(t,e,o){const i=tt(t),s=Z(t,e,i);Y(t,s);const r={model:t.id,messages:s,stream:!0};if(i.supportsUsageInStreaming!==!1&&(r.stream_options={include_usage:!0}),i.supportsStore&&(r.store=!1),o?.maxTokens&&(i.maxTokensField==="max_tokens"?r.max_tokens=o.maxTokens:r.max_completion_tokens=o.maxTokens),o?.temperature!==void 0&&(r.temperature=o.temperature),e.tools?r.tools=V(e.tools,i):K(e.messages)&&(r.tools=[]),o?.toolChoice&&(r.tool_choice=o.toolChoice),(i.thinkingFormat==="zai"||i.thinkingFormat==="qwen")&&t.reasoning?r.enable_thinking=!!o?.reasoningEffort:o?.reasoningEffort&&t.reasoning&&i.supportsReasoningEffort&&(r.reasoning_effort=W(o.reasoningEffort,i.reasoningEffortMap)),t.baseUrl.includes("openrouter.ai")&&t.compat?.openRouterRouting&&(r.provider=t.compat.openRouterRouting),t.baseUrl.includes("ai-gateway.vercel.sh")&&t.compat?.vercelGatewayRouting){const u=t.compat.vercelGatewayRouting;if(u.only||u.order){const a={};u.only&&(a.only=u.only),u.order&&(a.order=u.order),r.providerOptions={gateway:a}}}return r}_(J,"buildParams");function W(t,e){return e[t]??t}_(W,"mapReasoningEffort");function Y(t,e){if(!(t.provider!=="openrouter"||!t.id.startsWith("anthropic/")))for(let o=e.length-1;o>=0;o--){const i=e[o];if(i.role!=="user"&&i.role!=="assistant")continue;const s=i.content;if(typeof s=="string"){i.content=[Object.assign({type:"text",text:s},{cache_control:{type:"ephemeral"}})];return}if(Array.isArray(s))for(let r=s.length-1;r>=0;r--){const u=s[r];if(u?.type==="text"){Object.assign(u,{cache_control:{type:"ephemeral"}});return}}}}_(Y,"maybeAddOpenRouterAnthropicCacheControl");function Z(t,e,o){const i=[],s=_(a=>{if(o.requiresMistralToolIds)return $(a);if(a.includes("|")){const[g]=a.split("|");return g.replace(/[^a-zA-Z0-9_-]/g,"_").slice(0,40)}return t.provider==="openai"&&a.length>40?a.slice(0,40):a},"normalizeToolCallId"),r=z(e.messages,t,a=>s(a));if(e.systemPrompt){const g=t.reasoning&&o.supportsDeveloperRole?"developer":"system";i.push({role:g,content:S(e.systemPrompt)})}let u=null;for(let a=0;a<r.length;a++){const g=r[a];if(o.requiresAssistantAfterToolResult&&u==="toolResult"&&g.role==="user"&&i.push({role:"assistant",content:"I have processed the tool results."}),g.role==="user")if(typeof g.content=="string")i.push({role:"user",content:S(g.content)});else{const f=g.content.map(d=>d.type==="text"?{type:"text",text:S(d.text)}:d.type==="document"?{type:"text",text:"[PDF document - content not available in OpenAI format]"}:{type:"image_url",image_url:{url:`data:${d.mimeType};base64,${d.data}`}}),n=t.input.includes("image")?f:f.filter(d=>d.type!=="image_url");if(n.length===0)continue;i.push({role:"user",content:n})}else if(g.role==="assistant"){const f={role:"assistant",content:o.requiresAssistantAfterToolResult?"":null},d=g.content.filter(p=>p.type==="text").filter(p=>p.text&&p.text.trim().length>0);d.length>0&&(t.provider==="github-copilot"?f.content=d.map(p=>S(p.text)).join(""):f.content=d.map(p=>({type:"text",text:S(p.text)})));const R=g.content.filter(p=>p.type==="thinking").filter(p=>p.thinking&&p.thinking.trim().length>0);if(R.length>0)if(o.requiresThinkingAsText){const p=R.map(h=>h.thinking).join(`
+
+`),c=f.content;c?c.unshift({type:"text",text:p}):f.content=[{type:"text",text:p}]}else{const p=R[0].thinkingSignature;p&&p.length>0&&(f[p]=R.map(c=>c.thinking).join(`
+`))}const m=g.content.filter(p=>p.type==="toolCall");if(m.length>0){f.tool_calls=m.map(c=>({id:c.id,type:"function",function:{name:c.name,arguments:JSON.stringify(c.arguments)}}));const p=m.filter(c=>c.thoughtSignature).map(c=>{try{return JSON.parse(c.thoughtSignature)}catch{return null}}).filter(Boolean);p.length>0&&(f.reasoning_details=p)}const T=f.content;if(!(T!=null&&T.length>0)&&!f.tool_calls)continue;i.push(f)}else if(g.role==="toolResult"){const f=[];let n=a;for(;n<r.length&&r[n].role==="toolResult";n++){const d=r[n],y=d.content.filter(A=>A.type==="text").map(A=>A.text).join(`
+`),R=d.content.some(A=>A.type==="image"),m=y.length>0,T={role:"tool",content:S(m?y:"(see attached image)"),tool_call_id:d.toolCallId};if(o.requiresToolResultName&&d.toolName&&(T.name=d.toolName),i.push(T),R&&t.input.includes("image"))for(const A of d.content)A.type==="image"&&f.push({type:"image_url",image_url:{url:`data:${A.mimeType};base64,${A.data}`}})}a=n-1,f.length>0?(o.requiresAssistantAfterToolResult&&i.push({role:"assistant",content:"I have processed the tool results."}),i.push({role:"user",content:[{type:"text",text:"Attached image(s) from tool result:"},...f]}),u="user"):u="toolResult";continue}u=g.role}return i}_(Z,"convertMessages");function V(t,e){return t.map(o=>({type:"function",function:{name:o.name,description:o.description,parameters:o.parameters,...e.supportsStrictMode!==!1&&{strict:!1}}}))}_(V,"convertTools");function X(t){if(t===null)return"stop";switch(t){case"stop":return"stop";case"length":return"length";case"function_call":case"tool_calls":return"toolUse";case"content_filter":return"error";default:{const e=t;throw new Error(`Unhandled stop reason: ${e}`)}}}_(X,"mapStopReason");function Q(t){const e=t.provider,o=t.baseUrl,i=e==="zai"||o.includes("api.z.ai"),s=e==="cerebras"||o.includes("cerebras.ai")||e==="xai"||o.includes("api.x.ai")||e==="mistral"||o.includes("mistral.ai")||o.includes("chutes.ai")||o.includes("deepseek.com")||i||e==="opencode"||o.includes("opencode.ai"),r=e==="mistral"||o.includes("mistral.ai")||o.includes("chutes.ai"),u=e==="xai"||o.includes("api.x.ai"),a=e==="groq"||o.includes("groq.com"),g=e==="mistral"||o.includes("mistral.ai"),f=a&&t.id==="qwen/qwen3-32b"?{minimal:"default",low:"default",medium:"default",high:"default",xhigh:"default"}:{};return{supportsStore:!s,supportsDeveloperRole:!s,supportsReasoningEffort:!u&&!i,reasoningEffortMap:f,supportsUsageInStreaming:!0,maxTokensField:r?"max_tokens":"max_completion_tokens",requiresToolResultName:g,requiresAssistantAfterToolResult:!1,requiresThinkingAsText:g,requiresMistralToolIds:g,thinkingFormat:i?"zai":"openai",openRouterRouting:{},vercelGatewayRouting:{},supportsStrictMode:!0}}_(Q,"detectCompat");function tt(t){const e=Q(t);return t.compat?{supportsStore:t.compat.supportsStore??e.supportsStore,supportsDeveloperRole:t.compat.supportsDeveloperRole??e.supportsDeveloperRole,supportsReasoningEffort:t.compat.supportsReasoningEffort??e.supportsReasoningEffort,reasoningEffortMap:t.compat.reasoningEffortMap??e.reasoningEffortMap,supportsUsageInStreaming:t.compat.supportsUsageInStreaming??e.supportsUsageInStreaming,maxTokensField:t.compat.maxTokensField??e.maxTokensField,requiresToolResultName:t.compat.requiresToolResultName??e.requiresToolResultName,requiresAssistantAfterToolResult:t.compat.requiresAssistantAfterToolResult??e.requiresAssistantAfterToolResult,requiresThinkingAsText:t.compat.requiresThinkingAsText??e.requiresThinkingAsText,requiresMistralToolIds:t.compat.requiresMistralToolIds??e.requiresMistralToolIds,thinkingFormat:t.compat.thinkingFormat??e.thinkingFormat,openRouterRouting:t.compat.openRouterRouting??{},vercelGatewayRouting:t.compat.vercelGatewayRouting??e.vercelGatewayRouting,supportsStrictMode:t.compat.supportsStrictMode??e.supportsStrictMode}:e}_(tt,"getCompat");export{Z as convertMessages,L as streamOpenAICompletions,gt as streamSimpleOpenAICompletions};