@omnicross/core 0.4.2 → 0.4.4

package/dist/{chunk-WX2E2G7H.cjs → chunk-4NAMH4UE.cjs}

@@ -246,6 +246,13 @@ async function transformResponseIn(response, logger) {
 }
 
 // src/transformer/transformers/utils/gemini.stream.ts
+function inlineDataToDataUrl(part) {
+  const inline = _nullishCoalesce(part.inlineData, () => ( part.inline_data));
+  const data = _optionalChain([inline, 'optionalAccess', _25 => _25.data]);
+  if (typeof data !== "string" || !data) return null;
+  const mime = _nullishCoalesce(_nullishCoalesce(_optionalChain([inline, 'optionalAccess', _26 => _26.mimeType]), () => ( _optionalChain([inline, 'optionalAccess', _27 => _27.mime_type]))), () => ( "image/png"));
+  return `data:${mime};base64,${data}`;
+}
 async function transformResponseOut(response, providerName, logger) {
   const contentType = _nullishCoalesce(response.headers.get("Content-Type"), () => ( ""));
   if (contentType.includes("application/json")) {
@@ -257,8 +264,8 @@ async function transformResponseOut(response, providerName, logger) {
 }
 async function handleJsonResponse(response, providerName, logger) {
   const jsonResponse = await response.json();
-  _optionalChain([logger, 'optionalAccess', _25 => _25.debug, 'call', _26 => _26(`${providerName} JSON response received`)]);
-  const parts = _optionalChain([jsonResponse, 'access', _27 => _27.candidates, 'optionalAccess', _28 => _28[0], 'optionalAccess', _29 => _29.content, 'optionalAccess', _30 => _30.parts]) || [];
+  _optionalChain([logger, 'optionalAccess', _28 => _28.debug, 'call', _29 => _29(`${providerName} JSON response received`)]);
+  const parts = _optionalChain([jsonResponse, 'access', _30 => _30.candidates, 'optionalAccess', _31 => _31[0], 'optionalAccess', _32 => _32.content, 'optionalAccess', _33 => _33.parts]) || [];
   let thinkingContent = "";
   let thinkingSignature = "";
   const nonThinkingParts = [];
@@ -269,25 +276,27 @@ async function handleJsonResponse(response, providerName, logger) {
       nonThinkingParts.push(part);
     }
   }
-  thinkingSignature = _nullishCoalesce(_optionalChain([parts, 'access', _31 => _31.find, 'call', _32 => _32((part) => part.thoughtSignature), 'optionalAccess', _33 => _33.thoughtSignature]), () => ( ""));
+  thinkingSignature = _nullishCoalesce(_optionalChain([parts, 'access', _34 => _34.find, 'call', _35 => _35((part) => part.thoughtSignature), 'optionalAccess', _36 => _36.thoughtSignature]), () => ( ""));
   const toolCalls = nonThinkingParts.filter((part) => part.functionCall).map((part) => ({
-    id: _optionalChain([part, 'access', _34 => _34.functionCall, 'optionalAccess', _35 => _35.id]) || `tool_${Math.random().toString(36).substring(2, 15)}`,
+    id: _optionalChain([part, 'access', _37 => _37.functionCall, 'optionalAccess', _38 => _38.id]) || `tool_${Math.random().toString(36).substring(2, 15)}`,
     type: "function",
     function: {
-      name: _nullishCoalesce(_optionalChain([part, 'access', _36 => _36.functionCall, 'optionalAccess', _37 => _37.name]), () => ( "")),
-      arguments: JSON.stringify(_optionalChain([part, 'access', _38 => _38.functionCall, 'optionalAccess', _39 => _39.args]) || {})
+      name: _nullishCoalesce(_optionalChain([part, 'access', _39 => _39.functionCall, 'optionalAccess', _40 => _40.name]), () => ( "")),
+      arguments: JSON.stringify(_optionalChain([part, 'access', _41 => _41.functionCall, 'optionalAccess', _42 => _42.args]) || {})
     }
   }));
   const textContent = nonThinkingParts.filter((part) => part.text).map((part) => part.text).join("\n");
+  const images = nonThinkingParts.map((part) => inlineDataToDataUrl(part)).filter((url) => url !== null);
   const openAIResponse = {
     id: _nullishCoalesce(jsonResponse.responseId, () => ( "")),
     choices: [
       {
-        finish_reason: (_nullishCoalesce(_optionalChain([jsonResponse, 'access', _40 => _40.candidates, 'optionalAccess', _41 => _41[0], 'optionalAccess', _42 => _42.finishReason]), () => ( ""))).toLowerCase() || null,
+        finish_reason: (_nullishCoalesce(_optionalChain([jsonResponse, 'access', _43 => _43.candidates, 'optionalAccess', _44 => _44[0], 'optionalAccess', _45 => _45.finishReason]), () => ( ""))).toLowerCase() || null,
         index: 0,
         message: {
           content: textContent,
           role: "assistant",
+          images: images.length > 0 ? images : void 0,
           tool_calls: toolCalls.length > 0 ? toolCalls : void 0,
           ...thinkingSignature && {
             thinking: {
@@ -302,14 +311,14 @@ async function handleJsonResponse(response, providerName, logger) {
     model: _nullishCoalesce(jsonResponse.modelVersion, () => ( "")),
     object: "chat.completion",
     usage: {
-      completion_tokens: _optionalChain([jsonResponse, 'access', _43 => _43.usageMetadata, 'optionalAccess', _44 => _44.candidatesTokenCount]) || 0,
-      prompt_tokens: _optionalChain([jsonResponse, 'access', _45 => _45.usageMetadata, 'optionalAccess', _46 => _46.promptTokenCount]) || 0,
+      completion_tokens: _optionalChain([jsonResponse, 'access', _46 => _46.usageMetadata, 'optionalAccess', _47 => _47.candidatesTokenCount]) || 0,
+      prompt_tokens: _optionalChain([jsonResponse, 'access', _48 => _48.usageMetadata, 'optionalAccess', _49 => _49.promptTokenCount]) || 0,
       prompt_tokens_details: {
-        cached_tokens: _optionalChain([jsonResponse, 'access', _47 => _47.usageMetadata, 'optionalAccess', _48 => _48.cachedContentTokenCount]) || 0
+        cached_tokens: _optionalChain([jsonResponse, 'access', _50 => _50.usageMetadata, 'optionalAccess', _51 => _51.cachedContentTokenCount]) || 0
       },
-      total_tokens: _optionalChain([jsonResponse, 'access', _49 => _49.usageMetadata, 'optionalAccess', _50 => _50.totalTokenCount]) || 0,
+      total_tokens: _optionalChain([jsonResponse, 'access', _52 => _52.usageMetadata, 'optionalAccess', _53 => _53.totalTokenCount]) || 0,
       output_tokens_details: {
-        reasoning_tokens: _optionalChain([jsonResponse, 'access', _51 => _51.usageMetadata, 'optionalAccess', _52 => _52.thoughtsTokenCount]) || 0
+        reasoning_tokens: _optionalChain([jsonResponse, 'access', _54 => _54.usageMetadata, 'optionalAccess', _55 => _55.thoughtsTokenCount]) || 0
       }
     }
   };
@@ -340,15 +349,15 @@ function handleStreamResponse(response, providerName, logger) {
         if (!line.startsWith("data: ")) return;
         const chunkStr = line.slice(6).trim();
         if (!chunkStr) return;
-        _optionalChain([logger, 'optionalAccess', _53 => _53.debug, 'call', _54 => _54(`${providerName} chunk: ${chunkStr.substring(0, 100)}...`)]);
+        _optionalChain([logger, 'optionalAccess', _56 => _56.debug, 'call', _57 => _57(`${providerName} chunk: ${chunkStr.substring(0, 100)}...`)]);
         try {
           const chunk = JSON.parse(chunkStr);
-          if (!_optionalChain([chunk, 'access', _55 => _55.candidates, 'optionalAccess', _56 => _56[0]])) {
-            _optionalChain([logger, 'optionalAccess', _57 => _57.debug, 'call', _58 => _58("Invalid chunk structure")]);
+          if (!_optionalChain([chunk, 'access', _58 => _58.candidates, 'optionalAccess', _59 => _59[0]])) {
+            _optionalChain([logger, 'optionalAccess', _60 => _60.debug, 'call', _61 => _61("Invalid chunk structure")]);
             return;
           }
           const candidate = chunk.candidates[0];
-          const parts = _optionalChain([candidate, 'access', _59 => _59.content, 'optionalAccess', _60 => _60.parts]) || [];
+          const parts = _optionalChain([candidate, 'access', _62 => _62.content, 'optionalAccess', _63 => _63.parts]) || [];
           textBuffered = false;
           parts.filter((part) => part.text && part.thought === true).forEach((part) => {
             hasThinkingContent = true;
@@ -366,7 +375,7 @@ function handleStreamResponse(response, providerName, logger) {
 
 `));
           });
-          const signature = _optionalChain([parts, 'access', _61 => _61.find, 'call', _62 => _62((part) => part.thoughtSignature), 'optionalAccess', _63 => _63.thoughtSignature]);
+          const signature = _optionalChain([parts, 'access', _64 => _64.find, 'call', _65 => _65((part) => part.thoughtSignature), 'optionalAccess', _66 => _66.thoughtSignature]);
           if (signature && !signatureSent) {
             if (!hasThinkingContent) {
               const thinkingChunk = createChunk({
@@ -413,11 +422,11 @@ function handleStreamResponse(response, providerName, logger) {
             }
           }
           const toolCalls = parts.filter((part) => part.functionCall).map((part) => ({
-            id: _optionalChain([part, 'access', _64 => _64.functionCall, 'optionalAccess', _65 => _65.id]) || `ccr_tool_${Math.random().toString(36).substring(2, 15)}`,
+            id: _optionalChain([part, 'access', _67 => _67.functionCall, 'optionalAccess', _68 => _68.id]) || `ccr_tool_${Math.random().toString(36).substring(2, 15)}`,
             type: "function",
             function: {
-              name: _nullishCoalesce(_optionalChain([part, 'access', _66 => _66.functionCall, 'optionalAccess', _67 => _67.name]), () => ( "")),
-              arguments: JSON.stringify(_optionalChain([part, 'access', _68 => _68.functionCall, 'optionalAccess', _69 => _69.args]) || {})
+              name: _nullishCoalesce(_optionalChain([part, 'access', _69 => _69.functionCall, 'optionalAccess', _70 => _70.name]), () => ( "")),
+              arguments: JSON.stringify(_optionalChain([part, 'access', _71 => _71.functionCall, 'optionalAccess', _72 => _72.args]) || {})
             }
           }));
           const textContent = parts.filter((part) => part.text && part.thought !== true).map((part) => part.text).join("\n");
@@ -425,7 +434,7 @@ function handleStreamResponse(response, providerName, logger) {
             contentSent = true;
           }
           if (hasThinkingContent && textContent && !signatureSent) {
-            if (_optionalChain([chunk, 'access', _70 => _70.modelVersion, 'optionalAccess', _71 => _71.includes, 'call', _72 => _72("3")])) {
+            if (_optionalChain([chunk, 'access', _73 => _73.modelVersion, 'optionalAccess', _74 => _74.includes, 'call', _75 => _75("3")])) {
               pendingContent += textContent;
               textBuffered = true;
             } else {
@@ -445,6 +454,7 @@ function handleStreamResponse(response, providerName, logger) {
               signatureSent = true;
             }
           }
+          const images = parts.map((part) => inlineDataToDataUrl(part)).filter((url) => url !== null);
           if (textContent && !textBuffered) {
             if (!pendingContent) contentIndex++;
             const contentChunk = createChunk({
@@ -458,6 +468,21 @@ function handleStreamResponse(response, providerName, logger) {
             });
             controller.enqueue(encoder.encode(`data: ${JSON.stringify(contentChunk)}
 
+`));
+            contentSent = true;
+          }
+          if (images.length > 0) {
+            const imageChunk = createChunk({
+              responseId: chunk.responseId,
+              modelVersion: chunk.modelVersion,
+              contentIndex,
+              delta: { role: "assistant", images },
+              finishReason: candidate.finishReason,
+              usageMetadata: chunk.usageMetadata,
+              groundingMetadata: candidate.groundingMetadata
+            });
+            controller.enqueue(encoder.encode(`data: ${JSON.stringify(imageChunk)}
+
 `));
             contentSent = true;
           }
@@ -484,7 +509,7 @@ function handleStreamResponse(response, providerName, logger) {
             contentSent = true;
           }
         } catch (_error) {
-          _optionalChain([logger, 'optionalAccess', _73 => _73.error, 'call', _74 => _74(`Error parsing ${providerName} stream chunk: ${chunkStr}`)]);
+          _optionalChain([logger, 'optionalAccess', _76 => _76.error, 'call', _77 => _77(`Error parsing ${providerName} stream chunk: ${chunkStr}`)]);
         }
       };
       try {
@@ -528,7 +553,7 @@ function createChunk(options) {
     choices: [
       {
         delta,
-        finish_reason: _optionalChain([finishReason, 'optionalAccess', _75 => _75.toLowerCase, 'call', _76 => _76()]) || null,
+        finish_reason: _optionalChain([finishReason, 'optionalAccess', _78 => _78.toLowerCase, 'call', _79 => _79()]) || null,
         index: contentIndex,
         logprobs: null
       }
@@ -552,19 +577,19 @@ function createChunk(options) {
       }
     };
   }
-  if (_optionalChain([groundingMetadata, 'optionalAccess', _77 => _77.groundingChunks, 'optionalAccess', _78 => _78.length])) {
+  if (_optionalChain([groundingMetadata, 'optionalAccess', _80 => _80.groundingChunks, 'optionalAccess', _81 => _81.length])) {
     const annotations = groundingMetadata.groundingChunks.map((groundingChunk, index) => {
-      const support = _optionalChain([groundingMetadata, 'access', _79 => _79.groundingSupports, 'optionalAccess', _80 => _80.find, 'call', _81 => _81(
-        (s) => _optionalChain([s, 'access', _82 => _82.groundingChunkIndices, 'optionalAccess', _83 => _83.includes, 'call', _84 => _84(index)])
+      const support = _optionalChain([groundingMetadata, 'access', _82 => _82.groundingSupports, 'optionalAccess', _83 => _83.find, 'call', _84 => _84(
+        (s) => _optionalChain([s, 'access', _85 => _85.groundingChunkIndices, 'optionalAccess', _86 => _86.includes, 'call', _87 => _87(index)])
       )]);
       return {
         type: "url_citation",
         url_citation: {
-          url: _optionalChain([groundingChunk, 'access', _85 => _85.web, 'optionalAccess', _86 => _86.uri]) || "",
-          title: _optionalChain([groundingChunk, 'access', _87 => _87.web, 'optionalAccess', _88 => _88.title]) || "",
-          content: _optionalChain([support, 'optionalAccess', _89 => _89.segment, 'optionalAccess', _90 => _90.text]) || "",
-          start_index: _optionalChain([support, 'optionalAccess', _91 => _91.segment, 'optionalAccess', _92 => _92.startIndex]) || 0,
-          end_index: _optionalChain([support, 'optionalAccess', _93 => _93.segment, 'optionalAccess', _94 => _94.endIndex]) || 0
+          url: _optionalChain([groundingChunk, 'access', _88 => _88.web, 'optionalAccess', _89 => _89.uri]) || "",
+          title: _optionalChain([groundingChunk, 'access', _90 => _90.web, 'optionalAccess', _91 => _91.title]) || "",
+          content: _optionalChain([support, 'optionalAccess', _92 => _92.segment, 'optionalAccess', _93 => _93.text]) || "",
+          start_index: _optionalChain([support, 'optionalAccess', _94 => _94.segment, 'optionalAccess', _95 => _95.startIndex]) || 0,
+          end_index: _optionalChain([support, 'optionalAccess', _96 => _96.segment, 'optionalAccess', _97 => _97.endIndex]) || 0
         }
       };
     });
@@ -612,10 +637,10 @@ function processJsonSchema(jsonSchema) {
   }
   const incomingAnyOf = workingSchema.anyOf;
   if (incomingAnyOf && Array.isArray(incomingAnyOf) && incomingAnyOf.length === 2) {
-    if (_optionalChain([incomingAnyOf, 'access', _95 => _95[0], 'optionalAccess', _96 => _96.type]) === "null") {
+    if (_optionalChain([incomingAnyOf, 'access', _98 => _98[0], 'optionalAccess', _99 => _99.type]) === "null") {
       genAISchema.nullable = true;
       workingSchema = incomingAnyOf[1];
-    } else if (_optionalChain([incomingAnyOf, 'access', _97 => _97[1], 'optionalAccess', _98 => _98.type]) === "null") {
+    } else if (_optionalChain([incomingAnyOf, 'access', _100 => _100[1], 'optionalAccess', _101 => _101.type]) === "null") {
       genAISchema.nullable = true;
       workingSchema = incomingAnyOf[0];
     }
@@ -697,19 +722,19 @@ function transformTool(tool) {
 // src/transformer/transformers/utils/gemini.util.ts
 function buildRequestBody(request, provider, preserveNativeEffort = false) {
   const tools = [];
-  const functionDeclarations = _optionalChain([request, 'access', _99 => _99.tools, 'optionalAccess', _100 => _100.filter, 'call', _101 => _101((tool) => tool.function.name !== "web_search"), 'optionalAccess', _102 => _102.map, 'call', _103 => _103((tool) => ({
+  const functionDeclarations = _optionalChain([request, 'access', _102 => _102.tools, 'optionalAccess', _103 => _103.filter, 'call', _104 => _104((tool) => tool.function.name !== "web_search"), 'optionalAccess', _105 => _105.map, 'call', _106 => _106((tool) => ({
     name: tool.function.name,
     description: tool.function.description,
     parametersJsonSchema: tool.function.parameters
   }))]);
-  if (_optionalChain([functionDeclarations, 'optionalAccess', _104 => _104.length])) {
+  if (_optionalChain([functionDeclarations, 'optionalAccess', _107 => _107.length])) {
     tools.push(
       transformTool({
         functionDeclarations
       })
     );
   }
-  const webSearch = _optionalChain([request, 'access', _105 => _105.tools, 'optionalAccess', _106 => _106.find, 'call', _107 => _107((tool) => tool.function.name === "web_search")]);
+  const webSearch = _optionalChain([request, 'access', _108 => _108.tools, 'optionalAccess', _109 => _109.find, 'call', _110 => _110((tool) => tool.function.name === "web_search")]);
   if (webSearch) {
     tools.push({ googleSearch: {} });
   }
@@ -727,7 +752,7 @@ function buildRequestBody(request, provider, preserveNativeEffort = false) {
     const parts = [];
     if (typeof message.content === "string") {
       const part = { text: message.content };
-      if (_optionalChain([message, 'access', _108 => _108.thinking, 'optionalAccess', _109 => _109.signature])) {
+      if (_optionalChain([message, 'access', _111 => _111.thinking, 'optionalAccess', _112 => _112.signature])) {
         part.thoughtSignature = message.thinking.signature;
       }
       parts.push(part);
@@ -736,7 +761,7 @@ function buildRequestBody(request, provider, preserveNativeEffort = false) {
         if (content.type === "text") {
           parts.push({ text: content.text || "" });
         } else if (content.type === "image_url") {
-          const imageUrl = _nullishCoalesce(_optionalChain([content, 'access', _110 => _110.image_url, 'optionalAccess', _111 => _111.url]), () => ( ""));
+          const imageUrl = _nullishCoalesce(_optionalChain([content, 'access', _113 => _113.image_url, 'optionalAccess', _114 => _114.url]), () => ( ""));
           if (imageUrl.startsWith("http")) {
             parts.push({
               file_data: {
@@ -773,7 +798,7 @@ function buildRequestBody(request, provider, preserveNativeEffort = false) {
             args: JSON.parse(toolCall.function.arguments || "{}")
           }
         };
-        if (index === 0 && _optionalChain([message, 'access', _112 => _112.thinking, 'optionalAccess', _113 => _113.signature])) {
+        if (index === 0 && _optionalChain([message, 'access', _115 => _115.thinking, 'optionalAccess', _116 => _116.signature])) {
           functionCallPart.thoughtSignature = message.thinking.signature;
         }
         parts.push(functionCallPart);
@@ -789,8 +814,8 @@ function buildRequestBody(request, provider, preserveNativeEffort = false) {
           const response = toolResponses.find((item) => item.tool_call_id === tool.id);
           return {
             functionResponse: {
-              name: _nullishCoalesce(_optionalChain([tool, 'access', _114 => _114.function, 'optionalAccess', _115 => _115.name]), () => ( "")),
-              response: { result: _optionalChain([response, 'optionalAccess', _116 => _116.content]) }
+              name: _nullishCoalesce(_optionalChain([tool, 'access', _117 => _117.function, 'optionalAccess', _118 => _118.name]), () => ( "")),
+              response: { result: _optionalChain([response, 'optionalAccess', _119 => _119.content]) }
             }
           };
         }
@@ -821,17 +846,20 @@ function buildRequestBody(request, provider, preserveNativeEffort = false) {
     requestMaxTokens: request.max_tokens,
     preserveNativeEffort
   });
-  if (_optionalChain([reasoningPlan, 'optionalAccess', _117 => _117.kind]) === "level") {
+  if (_optionalChain([reasoningPlan, 'optionalAccess', _120 => _120.kind]) === "level") {
     generationConfig.thinkingConfig = {
       includeThoughts: reasoningPlan.enabled,
       thinkingLevel: reasoningPlan.effort
     };
-  } else if (_optionalChain([reasoningPlan, 'optionalAccess', _118 => _118.kind]) === "budget") {
+  } else if (_optionalChain([reasoningPlan, 'optionalAccess', _121 => _121.kind]) === "budget") {
     generationConfig.thinkingConfig = {
       includeThoughts: reasoningPlan.enabled,
       thinkingBudget: reasoningPlan.budgetTokens
     };
   }
+  if (!generationConfig.responseModalities && typeof request.model === "string" && request.model.startsWith("gemini") && request.model.includes("-image")) {
+    generationConfig.responseModalities = ["TEXT", "IMAGE"];
+  }
   const body = {
     contents,
     tools: tools.length > 0 ? tools : void 0,
@@ -847,7 +875,7 @@ function buildRequestBody(request, provider, preserveNativeEffort = false) {
       toolConfig.functionCallingConfig.mode = "none";
     } else if (request.tool_choice === "required") {
       toolConfig.functionCallingConfig.mode = "any";
-    } else if (typeof request.tool_choice === "object" && _optionalChain([request, 'access', _119 => _119.tool_choice, 'access', _120 => _120.function, 'optionalAccess', _121 => _121.name])) {
+    } else if (typeof request.tool_choice === "object" && _optionalChain([request, 'access', _122 => _122.tool_choice, 'access', _123 => _123.function, 'optionalAccess', _124 => _124.name])) {
       toolConfig.functionCallingConfig.mode = "any";
       toolConfig.functionCallingConfig.allowedFunctionNames = [
         request.tool_choice.function.name
@@ -862,7 +890,7 @@ function transformRequestOut(request) {
   const tools = request.tools;
   const model = request.model;
   const generationConfig = request.generationConfig;
-  const maxTokens = _nullishCoalesce(_optionalChain([generationConfig, 'optionalAccess', _122 => _122.maxOutputTokens]), () => ( request.max_tokens));
+  const maxTokens = _nullishCoalesce(_optionalChain([generationConfig, 'optionalAccess', _125 => _125.maxOutputTokens]), () => ( request.max_tokens));
   const temperature = request.temperature;
   const stream = request.stream;
   const toolChoice = request.tool_choice;
@@ -874,14 +902,14 @@ function transformRequestOut(request) {
     stream,
     tool_choice: toolChoice
   };
-  const thinkingConfig = _optionalChain([generationConfig, 'optionalAccess', _123 => _123.thinkingConfig]);
-  const discreteEffort = _chunkIUL4VWNUcjs.normalizeThinkLevel.call(void 0, _optionalChain([thinkingConfig, 'optionalAccess', _124 => _124.thinkingLevel]));
+  const thinkingConfig = _optionalChain([generationConfig, 'optionalAccess', _126 => _126.thinkingConfig]);
+  const discreteEffort = _chunkIUL4VWNUcjs.normalizeThinkLevel.call(void 0, _optionalChain([thinkingConfig, 'optionalAccess', _127 => _127.thinkingLevel]));
   if (discreteEffort) {
     unifiedRequest.reasoning = {
       effort: discreteEffort,
       enabled: discreteEffort !== "none"
     };
-  } else if (typeof _optionalChain([thinkingConfig, 'optionalAccess', _125 => _125.thinkingBudget]) === "number" && Number.isFinite(thinkingConfig.thinkingBudget) && thinkingConfig.thinkingBudget >= 0) {
+  } else if (typeof _optionalChain([thinkingConfig, 'optionalAccess', _128 => _128.thinkingBudget]) === "number" && Number.isFinite(thinkingConfig.thinkingBudget) && thinkingConfig.thinkingBudget >= 0) {
     const budget = thinkingConfig.thinkingBudget;
     unifiedRequest.reasoning = budget === 0 ? { effort: "none", enabled: false, max_tokens: 0 } : { effort: _chunk77USGNRWcjs.getThinkLevel.call(void 0, budget), enabled: true, max_tokens: budget };
   }
@@ -923,14 +951,14 @@ function transformRequestOut(request) {
         } else if ("inlineData" in part || "file_data" in part) {
           const inline = part.inlineData;
           const fileData = part.file_data;
-          if (_optionalChain([inline, 'optionalAccess', _126 => _126.data])) {
+          if (_optionalChain([inline, 'optionalAccess', _129 => _129.data])) {
             const mime = inline.mime_type || "image/png";
             contentParts.push({
               type: "image_url",
               image_url: { url: `data:${mime};base64,${inline.data}` },
               media_type: mime
             });
-          } else if (_optionalChain([fileData, 'optionalAccess', _127 => _127.file_uri])) {
+          } else if (_optionalChain([fileData, 'optionalAccess', _130 => _130.file_uri])) {
             contentParts.push({
               type: "image_url",
               image_url: { url: fileData.file_uri },
@@ -950,7 +978,7 @@ function transformRequestOut(request) {
         const queue = toolCallIdsByName.get(fnName);
         const id = queue && queue.length > 0 ? queue.shift() : `tool_${Math.random().toString(36).substring(2, 15)}`;
         const response = fr.functionResponse.response;
-        const result = _optionalChain([response, 'optionalAccess', _128 => _128.result]);
+        const result = _optionalChain([response, 'optionalAccess', _131 => _131.result]);
         const toolContent = typeof result === "string" ? result : result !== void 0 ? JSON.stringify(result) : response !== void 0 ? JSON.stringify(response) : "";
         unifiedRequest.messages.push({
           role: "tool",

package/dist/{chunk-LQYS2N47.js → chunk-4RM22ETG.js}

@@ -1811,6 +1811,11 @@ function assertFiniteImageApiLimits(limits) {
     throw new TypeError("Image API aggregate output limit must cover one output.");
   }
 }
+function resolveImageRequestProvider(runtime, model) {
+  const providerId = runtime.modelRoutes.get(model);
+  if (!providerId) throw new ImageGenerationError("unsupported_model", { param: "model" });
+  return providerId;
+}
 
 // src/image-generation/openai-images/generateHandler.ts
 function validateRuntime(runtime) {
@@ -1863,7 +1868,10 @@ function createImageGenerateHandler(deps) {
           ...runtime.preferredAccountGroup ? { preferredAccountGroup: runtime.preferredAccountGroup } : {},
           ...runtime.boundAccountFallbackPolicy ? { boundAccountFallbackPolicy: runtime.boundAccountFallbackPolicy } : {}
         },
-        { providerId: runtime.providerId, ...runtime.retention ? { retention: runtime.retention } : {} }
+        {
+          providerId: resolveImageRequestProvider(runtime, request2.model),
+          ...runtime.retention ? { retention: runtime.retention } : {}
+        }
       );
       if (request2.stream) {
         await writeImageApiSse({
@@ -2536,7 +2544,10 @@ function createImageEditHandler(deps) {
           ...runtime.preferredAccountGroup ? { preferredAccountGroup: runtime.preferredAccountGroup } : {},
           ...runtime.boundAccountFallbackPolicy ? { boundAccountFallbackPolicy: runtime.boundAccountFallbackPolicy } : {}
         },
-        { providerId: runtime.providerId, ...runtime.retention ? { retention: runtime.retention } : {} }
+        {
+          providerId: resolveImageRequestProvider(runtime, request2.model),
+          ...runtime.retention ? { retention: runtime.retention } : {}
+        }
       );
       if (request2.stream) {
         await writeImageApiSse({
@@ -2883,6 +2894,7 @@ var NORMALIZATION_RUNTIME = {
   providerId: "responses-image-inspection",
   defaultModel: "responses-image-model",
   modelAliases: /* @__PURE__ */ new Map(),
+  modelRoutes: /* @__PURE__ */ new Map(),
   limits: DEFAULT_IMAGE_API_LIMITS
 };
 function invalid5(param) {

package/dist/{chunk-YEVWA4C7.cjs → chunk-57Y2EA2G.cjs}

@@ -3,7 +3,7 @@
 var _chunkQBZJ7P2Tcjs = require('./chunk-QBZJ7P2T.cjs');
 
 
-var _chunk4IH4EL7Mcjs = require('./chunk-4IH4EL7M.cjs');
+var _chunkLL3MZ7O5cjs = require('./chunk-LL3MZ7O5.cjs');
 
 // src/outbound-api/kindDetection.ts
 var _canonicalmodels = require('@omnicross/contracts/canonical-models');
@@ -66,15 +66,15 @@ function isNonBlankRef(ref) {
   return typeof ref === "string" && ref.trim() !== "";
 }
 function isKindMappedEndpoint(endpoint) {
-  return Object.prototype.hasOwnProperty.call(_chunk4IH4EL7Mcjs.ENDPOINT_MODEL_KINDS, endpoint);
+  return Object.prototype.hasOwnProperty.call(_chunkLL3MZ7O5cjs.ENDPOINT_MODEL_KINDS, endpoint);
 }
 function modelKindsForEndpoint(endpoint) {
-  return _chunk4IH4EL7Mcjs.ENDPOINT_MODEL_KINDS[endpoint];
+  return _chunkLL3MZ7O5cjs.ENDPOINT_MODEL_KINDS[endpoint];
 }
 function detectModelKind(endpoint, requestedModelId) {
   if (!requestedModelId || requestedModelId.trim() === "") return void 0;
   if (endpoint === "messages") {
-    const declared = _chunk4IH4EL7Mcjs.ENDPOINT_MODEL_KINDS.messages;
+    const declared = _chunkLL3MZ7O5cjs.ENDPOINT_MODEL_KINDS.messages;
     for (const tok of modelTokens(_canonicalmodels.normalizeModelId.call(void 0, requestedModelId))) {
       if (declared.includes(tok)) return tok;
     }
@@ -122,7 +122,8 @@ var SUBSCRIPTION_PROVIDER_IDS = [
   "opencodego",
   "kimi",
   "grok",
-  "copilot"
+  "copilot",
+  "antigravity"
 ];
 var SUBSCRIPTION_ID_SET = new Set(SUBSCRIPTION_PROVIDER_IDS);
 function isSubscriptionProviderId(providerId) {

package/dist/chunk-6I4SAWCN.js

@@ -0,0 +1,24 @@
+// src/outbound-api/types.ts
+var IMAGE_PROVIDER_IDS = [
+  "codex-subscription",
+  "antigravity-subscription"
+];
+var DEFAULT_IMAGE_MODEL_ROUTES = Object.freeze({
+  "gpt-image-2": "codex-subscription",
+  "gpt-image-2-5": "codex-subscription",
+  "gemini-2.5-flash-image": "antigravity-subscription",
+  "gemini-2.5-flash-image-preview": "antigravity-subscription",
+  "gemini-3.1-flash-image": "antigravity-subscription",
+  "gemini-3.1-flash-image-preview": "antigravity-subscription",
+  "gemini-3-pro-image-preview": "antigravity-subscription"
+});
+var ENDPOINT_MODEL_KINDS = {
+  messages: ["fable", "opus", "sonnet", "haiku"],
+  responses: ["codex", "mini"]
+};
+
+export {
+  IMAGE_PROVIDER_IDS,
+  DEFAULT_IMAGE_MODEL_ROUTES,
+  ENDPOINT_MODEL_KINDS
+};

package/dist/{chunk-KIQ2HA2A.cjs → chunk-BTSADEX4.cjs}

@@ -3,7 +3,7 @@
 
 
 
-var _chunkWX2E2G7Hcjs = require('./chunk-WX2E2G7H.cjs');
+var _chunk4NAMH4UEcjs = require('./chunk-4NAMH4UE.cjs');
 
 // src/transformer/transformers/GeminiTransformer.ts
 var GeminiTransformer = (_class = class {
@@ -44,7 +44,7 @@ var GeminiTransformer = (_class = class {
    * Also builds the correct URL for the Gemini API
    */
   async transformRequestIn(request, provider, context) {
-    const body = _chunkWX2E2G7Hcjs.buildRequestBody.call(void 0, 
+    const body = _chunk4NAMH4UEcjs.buildRequestBody.call(void 0, 
       request,
       provider,
       context.reasoningSourceFormat === this.name
@@ -69,7 +69,7 @@ var GeminiTransformer = (_class = class {
    * (For requests coming into the Gemini endpoint)
    */
   async transformRequestOut(request, context) {
-    const unifiedRequest = _chunkWX2E2G7Hcjs.transformRequestOut.call(void 0, request);
+    const unifiedRequest = _chunk4NAMH4UEcjs.transformRequestOut.call(void 0, request);
     if (_optionalChain([unifiedRequest, 'access', _2 => _2.reasoning, 'optionalAccess', _3 => _3.effort])) {
       context.reasoningSourceFormat = this.name;
     }
@@ -79,14 +79,14 @@ var GeminiTransformer = (_class = class {
    * Transform Gemini response to OpenAI-compatible format
    */
   async transformResponseOut(response, _context) {
-    return _chunkWX2E2G7Hcjs.transformResponseOut.call(void 0, response, this.name, this.logger);
+    return _chunk4NAMH4UEcjs.transformResponseOut.call(void 0, response, this.name, this.logger);
   }
   /**
    * Transform OpenAI-compatible response back to Gemini format
    * (For endpoint mode — returning Gemini-format responses to the client)
    */
   async transformResponseIn(response, _context) {
-    return _chunkWX2E2G7Hcjs.transformResponseIn.call(void 0, response, this.logger);
+    return _chunk4NAMH4UEcjs.transformResponseIn.call(void 0, response, this.logger);
   }
 }, _class.__initStatic(), _class);
 

package/dist/chunk-CGEZQC2G.cjs

@@ -0,0 +1,67 @@
+"use strict";Object.defineProperty(exports, "__esModule", {value: true}); function _nullishCoalesce(lhs, rhsFn) { if (lhs != null) { return lhs; } else { return rhsFn(); } } function _optionalChain(ops) { let lastAccessLHS = undefined; let value = ops[0]; let i = 1; while (i < ops.length) { const op = ops[i]; const fn = ops[i + 1]; i += 2; if ((op === 'optionalAccess' || op === 'optionalCall') && value == null) { return undefined; } if (op === 'access' || op === 'optionalAccess') { lastAccessLHS = value; value = fn(value); } else if (op === 'call' || op === 'optionalCall') { value = fn((...args) => value.call(lastAccessLHS, ...args)); lastAccessLHS = undefined; } } return value; }
+
+var _chunkX6GHTNDLcjs = require('./chunk-X6GHTNDL.cjs');
+
+// src/transformer/transformers/antigravityFailover.ts
+var ANTIGRAVITY_SANDBOX_ENDPOINT = "https://daily-cloudcode-pa.sandbox.googleapis.com";
+var sandboxFailoverEnabled = false;
+function setAntigravitySandboxFailover(enabled) {
+  sandboxFailoverEnabled = enabled;
+}
+function isAntigravitySandboxFailoverEnabled() {
+  return sandboxFailoverEnabled;
+}
+function __resetAntigravityFailoverState() {
+  sandboxFailoverEnabled = false;
+}
+function isAntigravityRetryableFailure(status, bodyText) {
+  if (status === null) return true;
+  if (status === 429 || status >= 500) return true;
+  if (status === 400 || status === 404) {
+    const message = (_nullishCoalesce(bodyText, () => ( ""))).toLowerCase();
+    return message.includes("requested model is currently unavailable") || message.includes("requested entity was not found") || message.includes("not found") || message.includes("tool_use") || message.includes("tool_result");
+  }
+  return false;
+}
+function maybeAntigravityFailoverUrl(failedUrl, status, bodyText, methodPath) {
+  if (!sandboxFailoverEnabled) return null;
+  if (!isAntigravityRetryableFailure(status, bodyText)) return null;
+  let parsed;
+  try {
+    parsed = new URL(failedUrl);
+  } catch (e) {
+    return null;
+  }
+  if (parsed.origin !== _chunkX6GHTNDLcjs.ANTIGRAVITY_ENDPOINT) return null;
+  const tail = _nullishCoalesce(methodPath, () => ( `${parsed.pathname}${parsed.search}`));
+  return `${ANTIGRAVITY_SANDBOX_ENDPOINT}${tail.startsWith("/") ? tail : `/${tail}`}`;
+}
+async function fetchWithAntigravityFailover(url, fetcher, signal) {
+  _optionalChain([signal, 'optionalAccess', _ => _.throwIfAborted, 'call', _2 => _2()]);
+  let response;
+  try {
+    response = await fetcher(url);
+  } catch (error) {
+    _optionalChain([signal, 'optionalAccess', _3 => _3.throwIfAborted, 'call', _4 => _4()]);
+    const fallback2 = maybeAntigravityFailoverUrl(url, null);
+    if (!fallback2) throw error;
+    return fetcher(fallback2);
+  }
+  if (!sandboxFailoverEnabled || response.ok) return response;
+  const body = response.status === 400 || response.status === 404 ? await response.clone().text().catch(() => "") : void 0;
+  const fallback = maybeAntigravityFailoverUrl(url, response.status, body);
+  if (!fallback) return response;
+  await _optionalChain([response, 'access', _5 => _5.body, 'optionalAccess', _6 => _6.cancel, 'call', _7 => _7(), 'access', _8 => _8.catch, 'call', _9 => _9(() => void 0)]);
+  _optionalChain([signal, 'optionalAccess', _10 => _10.throwIfAborted, 'call', _11 => _11()]);
+  return fetcher(fallback);
+}
+
+
+
+
+
+
+
+
+
+exports.ANTIGRAVITY_SANDBOX_ENDPOINT = ANTIGRAVITY_SANDBOX_ENDPOINT; exports.setAntigravitySandboxFailover = setAntigravitySandboxFailover; exports.isAntigravitySandboxFailoverEnabled = isAntigravitySandboxFailoverEnabled; exports.__resetAntigravityFailoverState = __resetAntigravityFailoverState; exports.isAntigravityRetryableFailure = isAntigravityRetryableFailure; exports.maybeAntigravityFailoverUrl = maybeAntigravityFailoverUrl; exports.fetchWithAntigravityFailover = fetchWithAntigravityFailover;