@providerprotocol/ai 0.0.34 → 0.0.36

package/dist/google/index.js

@@ -1,85 +1,41 @@
 import {
   Image
-} from "../chunk-WAKD3OO5.js";
+} from "../chunk-N5DX5JW3.js";
+import {
+  parseSSEStream
+} from "../chunk-PMK5LZ5Z.js";
 import {
   parseJsonResponse
-} from "../chunk-3C7O2RNO.js";
+} from "../chunk-A2IM7PGT.js";
 import {
-  StreamEventType,
-  objectDelta
-} from "../chunk-6S222DHN.js";
+  resolveApiKey
+} from "../chunk-ARVM24K2.js";
+import {
+  createProvider
+} from "../chunk-JA3UZALR.js";
 import {
   AssistantMessage,
-  createProvider,
   isAssistantMessage,
   isToolResultMessage,
   isUserMessage
-} from "../chunk-ILR2D5PN.js";
+} from "../chunk-BRP5XJ6Q.js";
 import {
-  parseSSEStream
-} from "../chunk-TOJCZMVU.js";
-import {
-  resolveApiKey
-} from "../chunk-3D6XGGVG.js";
-import {
-  ErrorCode,
-  ModalityType,
-  UPPError,
   doFetch,
   doStreamFetch,
   normalizeHttpError,
   toError
-} from "../chunk-4J6OFUKX.js";
+} from "../chunk-AY55T37A.js";
+import {
+  ErrorCode,
+  ModalityType,
+  UPPError
+} from "../chunk-COS4ON4G.js";
+import {
+  StreamEventType,
+  objectDelta
+} from "../chunk-RJGTRQ47.js";
 
 // src/providers/google/transform.ts
-function transformRequest(request, modelId) {
-  const params = request.params ?? {};
-  const { cachedContent, tools: builtInTools, toolConfig, ...generationParams } = params;
-  const googleRequest = {
-    contents: transformMessages(request.messages)
-  };
-  const normalizedSystem = normalizeSystem(request.system);
-  if (normalizedSystem !== void 0) {
-    if (typeof normalizedSystem === "string") {
-      googleRequest.systemInstruction = {
-        parts: [{ text: normalizedSystem }]
-      };
-    } else if (normalizedSystem.length > 0) {
-      googleRequest.systemInstruction = {
-        parts: normalizedSystem
-      };
-    }
-  }
-  const generationConfig = {
-    ...generationParams
-  };
-  if (request.structure) {
-    generationConfig.responseMimeType = "application/json";
-    generationConfig.responseSchema = request.structure;
-  }
-  if (Object.keys(generationConfig).length > 0) {
-    googleRequest.generationConfig = generationConfig;
-  }
-  const requestTools = [];
-  if (request.tools && request.tools.length > 0) {
-    requestTools.push({
-      functionDeclarations: request.tools.map(transformTool)
-    });
-  }
-  if (builtInTools && builtInTools.length > 0) {
-    requestTools.push(...builtInTools);
-  }
-  if (requestTools.length > 0) {
-    googleRequest.tools = requestTools;
-  }
-  if (toolConfig) {
-    googleRequest.toolConfig = toolConfig;
-  }
-  if (cachedContent) {
-    googleRequest.cachedContent = cachedContent;
-  }
-  return googleRequest;
-}
 function normalizeSystem(system) {
   if (system === void 0 || system === null) return void 0;
   if (typeof system === "string") return system;
@@ -133,6 +89,67 @@ function extractGoogleToolName(toolCallId) {
 function filterValidContent(content) {
   return content.filter((c) => c && typeof c.type === "string");
 }
+function uint8ArrayToBase64(bytes) {
+  return Buffer.from(bytes).toString("base64");
+}
+function transformContentBlock(block) {
+  switch (block.type) {
+    case "text":
+      return { text: block.text };
+    case "image": {
+      const imageBlock = block;
+      let data;
+      if (imageBlock.source.type === "base64") {
+        data = imageBlock.source.data;
+      } else if (imageBlock.source.type === "bytes") {
+        data = uint8ArrayToBase64(imageBlock.source.data);
+      } else {
+        throw new Error("Google API does not support URL image sources directly");
+      }
+      return {
+        inlineData: {
+          mimeType: imageBlock.mimeType,
+          data
+        }
+      };
+    }
+    case "document": {
+      const documentBlock = block;
+      if (documentBlock.source.type === "base64") {
+        return {
+          inlineData: {
+            mimeType: documentBlock.mimeType,
+            data: documentBlock.source.data
+          }
+        };
+      }
+      if (documentBlock.source.type === "text") {
+        return { text: documentBlock.source.data };
+      }
+      throw new Error("Google API does not support URL document sources directly");
+    }
+    case "audio": {
+      const audioBlock = block;
+      return {
+        inlineData: {
+          mimeType: audioBlock.mimeType,
+          data: uint8ArrayToBase64(audioBlock.data)
+        }
+      };
+    }
+    case "video": {
+      const videoBlock = block;
+      return {
+        inlineData: {
+          mimeType: videoBlock.mimeType,
+          data: uint8ArrayToBase64(videoBlock.data)
+        }
+      };
+    }
+    default:
+      throw new Error(`Unsupported content type: ${block.type}`);
+  }
+}
 function transformMessages(messages) {
   const contents = [];
   for (const msg of messages) {
@@ -205,67 +222,6 @@ function transformMessages(messages) {
   }
   return contents;
 }
-function uint8ArrayToBase64(bytes) {
-  return Buffer.from(bytes).toString("base64");
-}
-function transformContentBlock(block) {
-  switch (block.type) {
-    case "text":
-      return { text: block.text };
-    case "image": {
-      const imageBlock = block;
-      let data;
-      if (imageBlock.source.type === "base64") {
-        data = imageBlock.source.data;
-      } else if (imageBlock.source.type === "bytes") {
-        data = uint8ArrayToBase64(imageBlock.source.data);
-      } else {
-        throw new Error("Google API does not support URL image sources directly");
-      }
-      return {
-        inlineData: {
-          mimeType: imageBlock.mimeType,
-          data
-        }
-      };
-    }
-    case "document": {
-      const documentBlock = block;
-      if (documentBlock.source.type === "base64") {
-        return {
-          inlineData: {
-            mimeType: documentBlock.mimeType,
-            data: documentBlock.source.data
-          }
-        };
-      }
-      if (documentBlock.source.type === "text") {
-        return { text: documentBlock.source.data };
-      }
-      throw new Error("Google API does not support URL document sources directly");
-    }
-    case "audio": {
-      const audioBlock = block;
-      return {
-        inlineData: {
-          mimeType: audioBlock.mimeType,
-          data: uint8ArrayToBase64(audioBlock.data)
-        }
-      };
-    }
-    case "video": {
-      const videoBlock = block;
-      return {
-        inlineData: {
-          mimeType: videoBlock.mimeType,
-          data: uint8ArrayToBase64(videoBlock.data)
-        }
-      };
-    }
-    default:
-      throw new Error(`Unsupported content type: ${block.type}`);
-  }
-}
 function transformTool(tool) {
   return {
     name: tool.name,
@@ -277,6 +233,71 @@ function transformTool(tool) {
     }
   };
 }
+function transformRequest(request, modelId) {
+  const params = request.params ?? {};
+  const { cachedContent, tools: builtInTools, toolConfig, ...generationParams } = params;
+  const googleRequest = {
+    contents: transformMessages(request.messages)
+  };
+  const normalizedSystem = normalizeSystem(request.system);
+  if (normalizedSystem !== void 0) {
+    if (typeof normalizedSystem === "string") {
+      googleRequest.systemInstruction = {
+        parts: [{ text: normalizedSystem }]
+      };
+    } else if (normalizedSystem.length > 0) {
+      googleRequest.systemInstruction = {
+        parts: normalizedSystem
+      };
+    }
+  }
+  const generationConfig = {
+    ...generationParams
+  };
+  if (request.structure) {
+    generationConfig.responseMimeType = "application/json";
+    generationConfig.responseSchema = request.structure;
+  }
+  if (Object.keys(generationConfig).length > 0) {
+    googleRequest.generationConfig = generationConfig;
+  }
+  const requestTools = [];
+  if (request.tools && request.tools.length > 0) {
+    requestTools.push({
+      functionDeclarations: request.tools.map(transformTool)
+    });
+  }
+  if (builtInTools && builtInTools.length > 0) {
+    requestTools.push(...builtInTools);
+  }
+  if (requestTools.length > 0) {
+    googleRequest.tools = requestTools;
+  }
+  if (toolConfig) {
+    googleRequest.toolConfig = toolConfig;
+  }
+  if (cachedContent) {
+    googleRequest.cachedContent = cachedContent;
+  }
+  return googleRequest;
+}
+function normalizeStopReason(reason) {
+  switch (reason) {
+    case "STOP":
+      return "end_turn";
+    case "MAX_TOKENS":
+      return "max_tokens";
+    case "SAFETY":
+    case "RECITATION":
+      return "content_filter";
+    case "TOOL_USE":
+      return "tool_use";
+    case "OTHER":
+      return "end_turn";
+    default:
+      return "end_turn";
+  }
+}
 function transformResponse(data) {
   const candidate = data.candidates?.[0];
   if (!candidate) {
@@ -381,6 +402,14 @@ function createStreamState() {
     isFirstChunk: true
   };
 }
+function decodeBase64(base64) {
+  const binaryString = atob(base64);
+  const bytes = new Uint8Array(binaryString.length);
+  for (let i = 0; i < binaryString.length; i += 1) {
+    bytes[i] = binaryString.charCodeAt(i);
+  }
+  return bytes;
+}
 function transformStreamChunk(chunk, state) {
   const events = [];
   if (state.isFirstChunk) {
@@ -535,31 +564,6 @@ function buildResponseFromState(state) {
     data: structuredData
   };
 }
-function normalizeStopReason(reason) {
-  switch (reason) {
-    case "STOP":
-      return "end_turn";
-    case "MAX_TOKENS":
-      return "max_tokens";
-    case "SAFETY":
-    case "RECITATION":
-      return "content_filter";
-    case "TOOL_USE":
-      return "tool_use";
-    case "OTHER":
-      return "end_turn";
-    default:
-      return "end_turn";
-  }
-}
-function decodeBase64(base64) {
-  const binaryString = atob(base64);
-  const bytes = new Uint8Array(binaryString.length);
-  for (let i = 0; i < binaryString.length; i += 1) {
-    bytes[i] = binaryString.charCodeAt(i);
-  }
-  return bytes;
-}
 
 // src/providers/google/llm.ts
 var GOOGLE_API_BASE = "https://generativelanguage.googleapis.com/v1beta";
@@ -704,10 +708,9 @@ function createLLMHandler() {
                   }
                   const events = transformStreamChunk(chunk, state);
                   for (const event of events) {
+                    yield event;
                     if (request.structure && event.type === StreamEventType.TextDelta) {
                       yield objectDelta(event.delta.text ?? "", event.index);
-                    } else {
-                      yield event;
                     }
                   }
                 }
@@ -835,33 +838,27 @@ function getCapabilities() {
     maxImages: 4
   };
 }
-function createImageHandler() {
-  let providerRef = null;
+function buildParameters(params) {
+  return params ? { ...params } : {};
+}
+function transformResponse2(data) {
+  if (!data.predictions || data.predictions.length === 0) {
+    throw new UPPError(
+      "No images in response",
+      ErrorCode.ProviderError,
+      "google",
+      ModalityType.Image
+    );
+  }
+  const images = data.predictions.map((prediction) => {
+    const mimeType = prediction.mimeType ?? "image/png";
+    const image = Image.fromBase64(prediction.bytesBase64Encoded, mimeType);
+    return { image };
+  });
   return {
-    _setProvider(provider) {
-      providerRef = provider;
-    },
-    bind(modelId) {
-      if (!providerRef) {
-        throw new UPPError(
-          "Provider reference not set. Handler must be used with createProvider().",
-          ErrorCode.InvalidRequest,
-          "google",
-          ModalityType.Image
-        );
-      }
-      const capabilities = getCapabilities();
-      const model = {
-        modelId,
-        capabilities,
-        get provider() {
-          return providerRef;
-        },
-        async generate(request) {
-          return executeGenerate(modelId, request);
-        }
-      };
-      return model;
+    images,
+    usage: {
+      imagesGenerated: images.length
     }
   };
 }
@@ -900,27 +897,33 @@ async function executeGenerate(modelId, request) {
   const data = await parseJsonResponse(response, "google", "image");
   return transformResponse2(data);
 }
-function buildParameters(params) {
-  return params ? { ...params } : {};
-}
-function transformResponse2(data) {
-  if (!data.predictions || data.predictions.length === 0) {
-    throw new UPPError(
-      "No images in response",
-      ErrorCode.ProviderError,
-      "google",
-      ModalityType.Image
-    );
-  }
-  const images = data.predictions.map((prediction) => {
-    const mimeType = prediction.mimeType ?? "image/png";
-    const image = Image.fromBase64(prediction.bytesBase64Encoded, mimeType);
-    return { image };
-  });
+function createImageHandler() {
+  let providerRef = null;
   return {
-    images,
-    usage: {
-      imagesGenerated: images.length
+    _setProvider(provider) {
+      providerRef = provider;
+    },
+    bind(modelId) {
+      if (!providerRef) {
+        throw new UPPError(
+          "Provider reference not set. Handler must be used with createProvider().",
+          ErrorCode.InvalidRequest,
+          "google",
+          ModalityType.Image
+        );
+      }
+      const capabilities = getCapabilities();
+      const model = {
+        modelId,
+        capabilities,
+        get provider() {
+          return providerRef;
+        },
+        async generate(request) {
+          return executeGenerate(modelId, request);
+        }
+      };
+      return model;
     }
   };
 }