@providerprotocol/ai 0.0.21 → 0.0.23

package/dist/google/index.js

@@ -1,25 +1,34 @@
 import {
   Image
 } from "../chunk-WAKD3OO5.js";
+import {
+  parseJsonResponse
+} from "../chunk-Z6DKC37J.js";
+import {
+  StreamEventType
+} from "../chunk-73IIE3QT.js";
 import {
   AssistantMessage,
   createProvider,
   isAssistantMessage,
   isToolResultMessage,
   isUserMessage
-} from "../chunk-M4BMM5IB.js";
+} from "../chunk-MF5ETY5O.js";
 import {
   parseSSEStream
-} from "../chunk-Z7RBRCRN.js";
+} from "../chunk-NWS5IKNR.js";
 import {
   resolveApiKey
-} from "../chunk-Y3GBJNA2.js";
+} from "../chunk-55X3W2MN.js";
 import {
+  ErrorCode,
+  ModalityType,
   UPPError,
   doFetch,
   doStreamFetch,
-  normalizeHttpError
-} from "../chunk-EDENPF3E.js";
+  normalizeHttpError,
+  toError
+} from "../chunk-QNJO7DSD.js";
 
 // src/providers/google/transform.ts
 function transformRequest(request, modelId) {
@@ -28,14 +37,15 @@ function transformRequest(request, modelId) {
   const googleRequest = {
     contents: transformMessages(request.messages)
   };
-  if (request.system) {
-    if (typeof request.system === "string") {
+  const normalizedSystem = normalizeSystem(request.system);
+  if (normalizedSystem !== void 0) {
+    if (typeof normalizedSystem === "string") {
       googleRequest.systemInstruction = {
-        parts: [{ text: request.system }]
+        parts: [{ text: normalizedSystem }]
       };
-    } else {
+    } else if (normalizedSystem.length > 0) {
       googleRequest.systemInstruction = {
-        parts: request.system
+        parts: normalizedSystem
       };
     }
   }
@@ -69,6 +79,56 @@ function transformRequest(request, modelId) {
   }
   return googleRequest;
 }
+function normalizeSystem(system) {
+  if (system === void 0 || system === null) return void 0;
+  if (typeof system === "string") return system;
+  if (!Array.isArray(system)) {
+    throw new UPPError(
+      "System prompt must be a string or an array of text parts",
+      ErrorCode.InvalidRequest,
+      "google",
+      ModalityType.LLM
+    );
+  }
+  const parts = [];
+  for (const part of system) {
+    if (!part || typeof part !== "object" || !("text" in part)) {
+      throw new UPPError(
+        "Google system prompt array must contain text parts",
+        ErrorCode.InvalidRequest,
+        "google",
+        ModalityType.LLM
+      );
+    }
+    const textValue = part.text;
+    if (typeof textValue !== "string") {
+      throw new UPPError(
+        "Google system prompt text must be a string",
+        ErrorCode.InvalidRequest,
+        "google",
+        ModalityType.LLM
+      );
+    }
+    parts.push(part);
+  }
+  return parts.length > 0 ? parts : void 0;
+}
+var GOOGLE_TOOLCALL_PREFIX = "google_toolcall";
+function createGoogleToolCallId(name, index) {
+  return `${GOOGLE_TOOLCALL_PREFIX}:${index}:${name}`;
+}
+function extractGoogleToolName(toolCallId) {
+  const prefix = `${GOOGLE_TOOLCALL_PREFIX}:`;
+  if (!toolCallId.startsWith(prefix)) {
+    return toolCallId;
+  }
+  const rest = toolCallId.slice(prefix.length);
+  const separatorIndex = rest.indexOf(":");
+  if (separatorIndex === -1) {
+    return toolCallId;
+  }
+  return rest.slice(separatorIndex + 1);
+}
 function filterValidContent(content) {
   return content.filter((c) => c && typeof c.type === "string");
 }
@@ -124,7 +184,7 @@ function transformMessages(messages) {
         role: "user",
         parts: msg.results.map((result) => ({
           functionResponse: {
-            name: result.toolCallId,
+            name: extractGoogleToolName(result.toolCallId),
             response: typeof result.result === "object" ? result.result : { result: result.result }
           }
         }))
@@ -191,8 +251,9 @@ function transformResponse(data) {
       }
     } else if ("functionCall" in part) {
       const fc = part;
+      const toolCallId = createGoogleToolCallId(fc.functionCall.name, toolCalls.length);
       toolCalls.push({
-        toolCallId: fc.functionCall.name,
+        toolCallId,
         toolName: fc.functionCall.name,
         arguments: fc.functionCall.args
       });
@@ -234,7 +295,7 @@ ${codeResult.codeExecutionResult.output}\`\`\`
   return {
     message,
     usage,
-    stopReason: candidate.finishReason ?? "STOP",
+    stopReason: normalizeStopReason(candidate.finishReason),
     data: structuredData
   };
 }
@@ -252,7 +313,7 @@ function createStreamState() {
 function transformStreamChunk(chunk, state) {
   const events = [];
   if (state.isFirstChunk) {
-    events.push({ type: "message_start", index: 0, delta: {} });
+    events.push({ type: StreamEventType.MessageStart, index: 0, delta: {} });
     state.isFirstChunk = false;
   }
   if (chunk.usageMetadata) {
@@ -268,22 +329,24 @@ function transformStreamChunk(chunk, state) {
     if ("text" in part) {
       state.content += part.text;
       events.push({
-        type: "text_delta",
+        type: StreamEventType.TextDelta,
         index: 0,
         delta: { text: part.text }
       });
     } else if ("functionCall" in part) {
       const fc = part;
+      const toolCallId = createGoogleToolCallId(fc.functionCall.name, state.toolCalls.length);
       state.toolCalls.push({
+        id: toolCallId,
         name: fc.functionCall.name,
         args: fc.functionCall.args,
         thoughtSignature: fc.thoughtSignature
       });
       events.push({
-        type: "tool_call_delta",
+        type: StreamEventType.ToolCallDelta,
         index: state.toolCalls.length - 1,
         delta: {
-          toolCallId: fc.functionCall.name,
+          toolCallId,
           toolName: fc.functionCall.name,
           argumentsJson: JSON.stringify(fc.functionCall.args)
         }
@@ -297,7 +360,7 @@ ${codeResult.codeExecutionResult.output}\`\`\`
 `;
         state.content += outputText;
         events.push({
-          type: "text_delta",
+          type: StreamEventType.TextDelta,
           index: 0,
           delta: { text: outputText }
         });
@@ -306,7 +369,7 @@ ${codeResult.codeExecutionResult.output}\`\`\`
   }
   if (candidate.finishReason) {
     state.finishReason = candidate.finishReason;
-    events.push({ type: "message_stop", index: 0, delta: {} });
+    events.push({ type: StreamEventType.MessageStop, index: 0, delta: {} });
   }
   return events;
 }
@@ -323,8 +386,9 @@ function buildResponseFromState(state) {
     }
   }
   for (const tc of state.toolCalls) {
+    const toolCallId = tc.id || createGoogleToolCallId(tc.name, toolCalls.length);
     toolCalls.push({
-      toolCallId: tc.name,
+      toolCallId,
       toolName: tc.name,
       arguments: tc.args
     });
@@ -356,10 +420,25 @@ function buildResponseFromState(state) {
   return {
     message,
     usage,
-    stopReason: state.finishReason ?? "STOP",
+    stopReason: normalizeStopReason(state.finishReason),
     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 "OTHER":
+      return "end_turn";
+    default:
+      return "end_turn";
+  }
+}
 
 // src/providers/google/llm.ts
 var GOOGLE_API_BASE = "https://generativelanguage.googleapis.com/v1beta";
@@ -371,9 +450,8 @@ var GOOGLE_CAPABILITIES = {
   videoInput: true,
   audioInput: true
 };
-function buildUrl(modelId, action, apiKey) {
-  const base = `${GOOGLE_API_BASE}/models/${modelId}:${action}`;
-  return `${base}?key=${apiKey}`;
+function buildUrl(modelId, action) {
+  return `${GOOGLE_API_BASE}/models/${modelId}:${action}`;
 }
 function createLLMHandler() {
   let providerRef = null;
@@ -385,9 +463,9 @@ function createLLMHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider().",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "google",
-          "llm"
+          ModalityType.LLM
         );
       }
       const model = {
@@ -403,10 +481,11 @@ function createLLMHandler() {
             "google",
             "llm"
           );
-          const url = request.config.baseUrl ? `${request.config.baseUrl}/models/${modelId}:generateContent?key=${apiKey}` : buildUrl(modelId, "generateContent", apiKey);
+          const url = request.config.baseUrl ? `${request.config.baseUrl}/models/${modelId}:generateContent` : buildUrl(modelId, "generateContent");
           const body = transformRequest(request, modelId);
           const headers = {
-            "Content-Type": "application/json"
+            "Content-Type": "application/json",
+            "x-goog-api-key": apiKey
           };
           if (request.config.headers) {
             for (const [key, value] of Object.entries(request.config.headers)) {
@@ -427,7 +506,7 @@ function createLLMHandler() {
             "google",
             "llm"
           );
-          const data = await response.json();
+          const data = await parseJsonResponse(response, "google", "llm");
           return transformResponse(data);
         },
         stream(request) {
@@ -446,10 +525,12 @@ function createLLMHandler() {
                 "google",
                 "llm"
               );
-              const url = request.config.baseUrl ? `${request.config.baseUrl}/models/${modelId}:streamGenerateContent?alt=sse&key=${apiKey}` : `${buildUrl(modelId, "streamGenerateContent", apiKey)}&alt=sse`;
+              const url = request.config.baseUrl ? `${request.config.baseUrl}/models/${modelId}:streamGenerateContent?alt=sse` : `${buildUrl(modelId, "streamGenerateContent")}?alt=sse`;
               const body = transformRequest(request, modelId);
               const headers = {
-                "Content-Type": "application/json"
+                "Content-Type": "application/json",
+                Accept: "text/event-stream",
+                "x-goog-api-key": apiKey
               };
               if (request.config.headers) {
                 for (const [key, value] of Object.entries(request.config.headers)) {
@@ -478,9 +559,9 @@ function createLLMHandler() {
               if (!response.body) {
                 const error = new UPPError(
                   "No response body for streaming request",
-                  "PROVIDER_ERROR",
+                  ErrorCode.ProviderError,
                   "google",
-                  "llm"
+                  ModalityType.LLM
                 );
                 responseReject(error);
                 throw error;
@@ -491,9 +572,9 @@ function createLLMHandler() {
                   if (chunk.error) {
                     const error = new UPPError(
                       chunk.error.message,
-                      "PROVIDER_ERROR",
+                      ErrorCode.ProviderError,
                       "google",
-                      "llm"
+                      ModalityType.LLM
                     );
                     responseReject(error);
                     throw error;
@@ -506,8 +587,9 @@ function createLLMHandler() {
               }
               responseResolve(buildResponseFromState(state));
             } catch (error) {
-              responseReject(error);
-              throw error;
+              const err = toError(error);
+              responseReject(err);
+              throw err;
             }
           }
           return {
@@ -536,9 +618,9 @@ function createEmbeddingHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider().",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "google",
-          "embedding"
+          ModalityType.Embedding
         );
       }
       const model = {
@@ -562,9 +644,9 @@ function createEmbeddingHandler() {
             if (!text) {
               throw new UPPError(
                 "Google embeddings only support text input",
-                "INVALID_REQUEST",
+                ErrorCode.InvalidRequest,
                 "google",
-                "embedding"
+                ModalityType.Embedding
               );
             }
             const embedRequest = {
@@ -574,9 +656,10 @@ function createEmbeddingHandler() {
             };
             return embedRequest;
           });
-          const url = `${baseUrl}/models/${modelId}:batchEmbedContents?key=${apiKey}`;
+          const url = `${baseUrl}/models/${modelId}:batchEmbedContents`;
           const headers = {
-            "Content-Type": "application/json"
+            "Content-Type": "application/json",
+            "x-goog-api-key": apiKey
           };
           if (request.config.headers) {
             for (const [key, value] of Object.entries(request.config.headers)) {
@@ -591,7 +674,7 @@ function createEmbeddingHandler() {
             body: JSON.stringify({ requests }),
             signal: request.signal
           }, request.config, "google", "embedding");
-          const data = await response.json();
+          const data = await parseJsonResponse(response, "google", "embedding");
           let totalTokens = 0;
           for (const emb of data.embeddings) {
             totalTokens += emb.statistics?.tokenCount ?? 0;
@@ -635,9 +718,9 @@ function createImageHandler() {
       if (!providerRef) {
         throw new UPPError(
           "Provider reference not set. Handler must be used with createProvider().",
-          "INVALID_REQUEST",
+          ErrorCode.InvalidRequest,
           "google",
-          "image"
+          ModalityType.Image
         );
       }
       const capabilities = getCapabilities();
@@ -687,7 +770,7 @@ async function executeGenerate(modelId, request) {
     body: JSON.stringify(body),
     signal: request.signal
   }, request.config, "google", "image");
-  const data = await response.json();
+  const data = await parseJsonResponse(response, "google", "image");
   return transformResponse2(data);
 }
 function buildParameters(params) {
@@ -697,9 +780,9 @@ function transformResponse2(data) {
   if (!data.predictions || data.predictions.length === 0) {
     throw new UPPError(
       "No images in response",
-      "PROVIDER_ERROR",
+      ErrorCode.ProviderError,
       "google",
-      "image"
+      ModalityType.Image
     );
   }
   const images = data.predictions.map((prediction) => {
@@ -716,10 +799,19 @@ function transformResponse2(data) {
 }
 
 // src/providers/google/cache.ts
-var CACHE_API_BASE = "https://generativelanguage.googleapis.com/v1beta/cachedContents";
+var DEFAULT_BASE_URL = "https://generativelanguage.googleapis.com/v1beta";
+function resolveBaseUrl(config) {
+  if (config?.baseUrl) {
+    const trimmed = config.baseUrl.replace(/\/$/, "");
+    return trimmed.endsWith("/v1beta") ? trimmed : `${trimmed}/v1beta`;
+  }
+  return DEFAULT_BASE_URL;
+}
 async function create(options) {
   const {
     apiKey,
+    config,
+    signal,
     model,
     displayName,
     contents,
@@ -728,6 +820,8 @@ async function create(options) {
     ttl,
     expireTime
   } = options;
+  const baseUrl = resolveBaseUrl(config);
+  const requestConfig = { ...config, apiKey };
   const requestBody = {
     model: model.startsWith("models/") ? model : `models/${model}`
   };
@@ -750,56 +844,163 @@ async function create(options) {
   } else if (expireTime) {
     requestBody.expireTime = expireTime;
   }
-  const response = await fetch(`${CACHE_API_BASE}?key=${apiKey}`, {
-    method: "POST",
-    headers: { "Content-Type": "application/json" },
-    body: JSON.stringify(requestBody)
-  });
-  if (!response.ok) {
-    throw await normalizeHttpError(response, "google", "llm");
+  const headers = {
+    "Content-Type": "application/json",
+    "x-goog-api-key": apiKey
+  };
+  if (config?.headers) {
+    for (const [key, value] of Object.entries(config.headers)) {
+      if (value !== void 0) {
+        headers[key] = value;
+      }
+    }
   }
-  return response.json();
+  const response = await doFetch(
+    `${baseUrl}/cachedContents`,
+    {
+      method: "POST",
+      headers,
+      body: JSON.stringify(requestBody),
+      signal
+    },
+    requestConfig,
+    "google",
+    "llm"
+  );
+  return parseJsonResponse(response, "google", "llm");
 }
-async function get(name, apiKey) {
+async function get(name, apiKey, config, signal) {
   const cacheName = name.startsWith("cachedContents/") ? name : `cachedContents/${name}`;
-  const url = `https://generativelanguage.googleapis.com/v1beta/${cacheName}?key=${apiKey}`;
-  const response = await fetch(url, { method: "GET" });
-  if (!response.ok) {
-    throw await normalizeHttpError(response, "google", "llm");
+  const baseUrl = resolveBaseUrl(config);
+  const url = `${baseUrl}/${cacheName}`;
+  const requestConfig = { ...config, apiKey };
+  const headers = { "x-goog-api-key": apiKey };
+  if (config?.headers) {
+    for (const [key, value] of Object.entries(config.headers)) {
+      if (value !== void 0) {
+        headers[key] = value;
+      }
+    }
   }
-  return response.json();
+  const response = await doFetch(
+    url,
+    {
+      method: "GET",
+      headers,
+      signal
+    },
+    requestConfig,
+    "google",
+    "llm"
+  );
+  return parseJsonResponse(response, "google", "llm");
 }
 async function list(options) {
-  const { apiKey, pageSize, pageToken } = options;
-  const params = new URLSearchParams({ key: apiKey });
+  const { apiKey, config, signal, pageSize, pageToken } = options;
+  const baseUrl = resolveBaseUrl(config);
+  const requestConfig = { ...config, apiKey };
+  const params = new URLSearchParams();
   if (pageSize) params.set("pageSize", String(pageSize));
   if (pageToken) params.set("pageToken", pageToken);
-  const response = await fetch(`${CACHE_API_BASE}?${params}`, { method: "GET" });
-  if (!response.ok) {
-    throw await normalizeHttpError(response, "google", "llm");
+  const headers = { "x-goog-api-key": apiKey };
+  if (config?.headers) {
+    for (const [key, value] of Object.entries(config.headers)) {
+      if (value !== void 0) {
+        headers[key] = value;
+      }
+    }
   }
-  return response.json();
+  const response = await doFetch(
+    `${baseUrl}/cachedContents?${params}`,
+    {
+      method: "GET",
+      headers,
+      signal
+    },
+    requestConfig,
+    "google",
+    "llm"
+  );
+  return parseJsonResponse(response, "google", "llm");
 }
-async function update(name, updateRequest, apiKey) {
+async function update(name, updateRequest, apiKey, config, signal) {
+  if (updateRequest.expireTime && updateRequest.ttl) {
+    throw new UPPError(
+      "Provide either expireTime or ttl (not both)",
+      ErrorCode.InvalidRequest,
+      "google",
+      ModalityType.LLM
+    );
+  }
+  const updateMaskParts = [];
+  if (updateRequest.expireTime) {
+    updateMaskParts.push("expireTime");
+  }
+  if (updateRequest.ttl) {
+    updateMaskParts.push("ttl");
+  }
+  if (updateMaskParts.length === 0) {
+    throw new UPPError(
+      "Update request must include expireTime or ttl",
+      ErrorCode.InvalidRequest,
+      "google",
+      ModalityType.LLM
+    );
+  }
   const cacheName = name.startsWith("cachedContents/") ? name : `cachedContents/${name}`;
-  const url = `https://generativelanguage.googleapis.com/v1beta/${cacheName}?key=${apiKey}`;
-  const response = await fetch(url, {
-    method: "PATCH",
-    headers: { "Content-Type": "application/json" },
-    body: JSON.stringify(updateRequest)
-  });
-  if (!response.ok) {
-    throw await normalizeHttpError(response, "google", "llm");
+  const baseUrl = resolveBaseUrl(config);
+  const params = new URLSearchParams({ updateMask: updateMaskParts.join(",") });
+  const url = `${baseUrl}/${cacheName}?${params.toString()}`;
+  const requestConfig = { ...config, apiKey };
+  const headers = {
+    "Content-Type": "application/json",
+    "x-goog-api-key": apiKey
+  };
+  if (config?.headers) {
+    for (const [key, value] of Object.entries(config.headers)) {
+      if (value !== void 0) {
+        headers[key] = value;
+      }
+    }
   }
-  return response.json();
+  const response = await doFetch(
+    url,
+    {
+      method: "PATCH",
+      headers,
+      body: JSON.stringify(updateRequest),
+      signal
+    },
+    requestConfig,
+    "google",
+    "llm"
+  );
+  return parseJsonResponse(response, "google", "llm");
 }
-async function deleteCache(name, apiKey) {
+async function deleteCache(name, apiKey, config, signal) {
   const cacheName = name.startsWith("cachedContents/") ? name : `cachedContents/${name}`;
-  const url = `https://generativelanguage.googleapis.com/v1beta/${cacheName}?key=${apiKey}`;
-  const response = await fetch(url, { method: "DELETE" });
-  if (!response.ok) {
-    throw await normalizeHttpError(response, "google", "llm");
+  const baseUrl = resolveBaseUrl(config);
+  const url = `${baseUrl}/${cacheName}`;
+  const requestConfig = { ...config, apiKey };
+  const headers = { "x-goog-api-key": apiKey };
+  if (config?.headers) {
+    for (const [key, value] of Object.entries(config.headers)) {
+      if (value !== void 0) {
+        headers[key] = value;
+      }
+    }
   }
+  const response = await doFetch(
+    url,
+    {
+      method: "DELETE",
+      headers,
+      signal
+    },
+    requestConfig,
+    "google",
+    "llm"
+  );
 }
 var cache = {
   create,