cloudassist-ai-provider 0.0.1 → 0.0.2

package/dist/index.mjs

@@ -7,7 +7,7 @@ import {
 } from "@ai-sdk/provider-utils";
 
 // src/version.ts
-var VERSION = true ? "2.0.52" : "0.0.0-test";
+var VERSION = true ? "0.0.1" : "0.0.0-test";
 
 // src/google-generative-ai-embedding-model.ts
 import {
@@ -87,6 +87,19 @@ var googleGenerativeAIEmbeddingProviderOptions = lazySchema2(
   )
 );
 
+// src/provider-key.ts
+var ALLOWED_PROVIDER_KEYS = [
+  "antigravity",
+  "gemini-cli",
+  "cloudassist",
+  "google"
+];
+function getProviderKey(providerOptions) {
+  return ALLOWED_PROVIDER_KEYS.find(
+    (key) => (providerOptions == null ? void 0 : providerOptions[key]) !== void 0
+  );
+}
+
 // src/google-generative-ai-embedding-model.ts
 var GoogleGenerativeAIEmbeddingModel = class {
   constructor(modelId, config) {
@@ -105,8 +118,9 @@ var GoogleGenerativeAIEmbeddingModel = class {
     abortSignal,
     providerOptions
   }) {
+    const providerKey = getProviderKey(providerOptions);
     const googleOptions = await parseProviderOptions({
-      provider: "google",
+      provider: providerKey != null ? providerKey : "google",
       providerOptions,
       schema: googleGenerativeAIEmbeddingProviderOptions
     });
@@ -546,13 +560,14 @@ function prepareTools({
 
 // src/convert-to-google-generative-ai-messages.ts
 function convertToGoogleGenerativeAIMessages(prompt, options) {
-  var _a, _b;
+  var _a, _b, _c;
   const systemInstructionParts = [];
   const contents = [];
   let systemMessagesAllowed = true;
   const isGemmaModel = (_a = options == null ? void 0 : options.isGemmaModel) != null ? _a : false;
   const modelId = (_b = options == null ? void 0 : options.modelId) != null ? _b : "";
   const includeToolCallId = isClaudeModel(modelId);
+  const providerKey = (_c = options == null ? void 0 : options.providerKey) != null ? _c : "google";
   for (const { role, content } of prompt) {
     switch (role) {
       case "system": {
@@ -600,8 +615,8 @@ function convertToGoogleGenerativeAIMessages(prompt, options) {
         contents.push({
           role: "model",
           parts: content.map((part) => {
-            var _a2, _b2, _c;
-            const thoughtSignature = ((_b2 = (_a2 = part.providerOptions) == null ? void 0 : _a2.google) == null ? void 0 : _b2.thoughtSignature) != null ? String((_c = part.providerOptions.google) == null ? void 0 : _c.thoughtSignature) : void 0;
+            var _a2, _b2, _c2;
+            const thoughtSignature = ((_b2 = (_a2 = part.providerOptions) == null ? void 0 : _a2[providerKey]) == null ? void 0 : _b2.thoughtSignature) != null ? String((_c2 = part.providerOptions[providerKey]) == null ? void 0 : _c2.thoughtSignature) : void 0;
             switch (part.type) {
               case "text": {
                 return part.text.length === 0 ? void 0 : {
@@ -920,8 +935,9 @@ var GoogleGenerativeAILanguageModel = class {
   }) {
     var _a;
     const warnings = [];
+    const providerKey = getProviderKey(providerOptions);
     const googleOptions = await parseProviderOptions2({
-      provider: "google",
+      provider: providerKey != null ? providerKey : "google",
       providerOptions,
       schema: googleGenerativeAIProviderOptions
     });
@@ -936,7 +952,7 @@ var GoogleGenerativeAILanguageModel = class {
     const isGemmaModel = this.modelId.toLowerCase().startsWith("gemma-");
     const { contents, systemInstruction } = convertToGoogleGenerativeAIMessages(
       prompt,
-      { isGemmaModel, modelId: this.modelId }
+      { isGemmaModel, modelId: this.modelId, providerKey }
     );
     const {
       tools: googleTools2,
@@ -998,12 +1014,13 @@ var GoogleGenerativeAILanguageModel = class {
         requestType: googleOptions == null ? void 0 : googleOptions.requestType,
         userAgent: googleOptions == null ? void 0 : googleOptions.userAgent,
         requestId: googleOptions == null ? void 0 : googleOptions.requestId
-      }
+      },
+      providerKey
     };
   }
   async doGenerate(options) {
     var _a, _b, _c, _d, _e, _f, _g, _h, _i, _j, _k, _l, _m;
-    const { args, warnings, cloudAssistOptions } = await this.getArgs(options);
+    const { args, warnings, cloudAssistOptions, providerKey } = await this.getArgs(options);
     const wrappedRequest = {
       project: cloudAssistOptions.projectId,
       model: this.modelId,
@@ -1074,7 +1091,11 @@ var GoogleGenerativeAILanguageModel = class {
               content.push({
                 type: "text",
                 text: currentTextContent,
-                providerMetadata: currentTextThoughtSignature ? { google: { thoughtSignature: currentTextThoughtSignature } } : void 0
+                providerMetadata: currentTextThoughtSignature ? {
+                  [providerKey != null ? providerKey : "google"]: {
+                    thoughtSignature: currentTextThoughtSignature
+                  }
+                } : void 0
               });
               currentTextContent = "";
               currentTextThoughtSignature = void 0;
@@ -1084,7 +1105,7 @@ var GoogleGenerativeAILanguageModel = class {
                 type: "reasoning",
                 text: currentReasoningContent,
                 providerMetadata: currentReasoningThoughtSignature ? {
-                  google: {
+                  [providerKey != null ? providerKey : "google"]: {
                     thoughtSignature: currentReasoningThoughtSignature
                   }
                 } : void 0
@@ -1121,7 +1142,7 @@ var GoogleGenerativeAILanguageModel = class {
                   type: "text",
                   text: currentTextContent,
                   providerMetadata: currentTextThoughtSignature ? {
-                    google: {
+                    [providerKey != null ? providerKey : "google"]: {
                       thoughtSignature: currentTextThoughtSignature
                     }
                   } : void 0
@@ -1140,7 +1161,7 @@ var GoogleGenerativeAILanguageModel = class {
                   type: "reasoning",
                   text: currentReasoningContent,
                   providerMetadata: currentReasoningThoughtSignature ? {
-                    google: {
+                    [providerKey != null ? providerKey : "google"]: {
                       thoughtSignature: currentReasoningThoughtSignature
                     }
                   } : void 0
@@ -1159,7 +1180,11 @@ var GoogleGenerativeAILanguageModel = class {
               content.push({
                 type: "text",
                 text: currentTextContent,
-                providerMetadata: currentTextThoughtSignature ? { google: { thoughtSignature: currentTextThoughtSignature } } : void 0
+                providerMetadata: currentTextThoughtSignature ? {
+                  [providerKey != null ? providerKey : "google"]: {
+                    thoughtSignature: currentTextThoughtSignature
+                  }
+                } : void 0
               });
               currentTextContent = "";
               currentTextThoughtSignature = void 0;
@@ -1169,7 +1194,7 @@ var GoogleGenerativeAILanguageModel = class {
                 type: "reasoning",
                 text: currentReasoningContent,
                 providerMetadata: currentReasoningThoughtSignature ? {
-                  google: {
+                  [providerKey != null ? providerKey : "google"]: {
                     thoughtSignature: currentReasoningThoughtSignature
                   }
                 } : void 0
@@ -1183,14 +1208,22 @@ var GoogleGenerativeAILanguageModel = class {
               toolCallId,
               toolName: part.functionCall.name,
               input: JSON.stringify(part.functionCall.args),
-              providerMetadata: part.thoughtSignature ? { google: { thoughtSignature: part.thoughtSignature } } : void 0
+              providerMetadata: part.thoughtSignature ? {
+                [providerKey != null ? providerKey : "google"]: {
+                  thoughtSignature: part.thoughtSignature
+                }
+              } : void 0
             });
           } else if ("inlineData" in part) {
             if (currentTextContent) {
               content.push({
                 type: "text",
                 text: currentTextContent,
-                providerMetadata: currentTextThoughtSignature ? { google: { thoughtSignature: currentTextThoughtSignature } } : void 0
+                providerMetadata: currentTextThoughtSignature ? {
+                  [providerKey != null ? providerKey : "google"]: {
+                    thoughtSignature: currentTextThoughtSignature
+                  }
+                } : void 0
               });
               currentTextContent = "";
               currentTextThoughtSignature = void 0;
@@ -1200,7 +1233,7 @@ var GoogleGenerativeAILanguageModel = class {
                 type: "reasoning",
                 text: currentReasoningContent,
                 providerMetadata: currentReasoningThoughtSignature ? {
-                  google: {
+                  [providerKey != null ? providerKey : "google"]: {
                     thoughtSignature: currentReasoningThoughtSignature
                   }
                 } : void 0
@@ -1229,14 +1262,22 @@ var GoogleGenerativeAILanguageModel = class {
       content.push({
         type: "text",
         text: currentTextContent,
-        providerMetadata: currentTextThoughtSignature ? { google: { thoughtSignature: currentTextThoughtSignature } } : void 0
+        providerMetadata: currentTextThoughtSignature ? {
+          [providerKey != null ? providerKey : "google"]: {
+            thoughtSignature: currentTextThoughtSignature
+          }
+        } : void 0
       });
     }
     if (currentReasoningContent) {
       content.push({
         type: "reasoning",
         text: currentReasoningContent,
-        providerMetadata: currentReasoningThoughtSignature ? { google: { thoughtSignature: currentReasoningThoughtSignature } } : void 0
+        providerMetadata: currentReasoningThoughtSignature ? {
+          [providerKey != null ? providerKey : "google"]: {
+            thoughtSignature: currentReasoningThoughtSignature
+          }
+        } : void 0
       });
     }
     const sources = (_h = extractSources({
@@ -1258,7 +1299,7 @@ var GoogleGenerativeAILanguageModel = class {
       },
       warnings,
       providerMetadata: {
-        google: {
+        [providerKey != null ? providerKey : "google"]: {
           promptFeedback: promptFeedback != null ? promptFeedback : null,
           groundingMetadata: groundingMetadata != null ? groundingMetadata : null,
           urlContextMetadata: urlContextMetadata != null ? urlContextMetadata : null,
@@ -1274,7 +1315,7 @@ var GoogleGenerativeAILanguageModel = class {
   }
   async doStream(options) {
     var _a, _b;
-    const { args, warnings, cloudAssistOptions } = await this.getArgs(options);
+    const { args, warnings, cloudAssistOptions, providerKey } = await this.getArgs(options);
     const wrappedRequest = {
       project: cloudAssistOptions.projectId,
       model: this.modelId,
@@ -1400,7 +1441,7 @@ var GoogleGenerativeAILanguageModel = class {
                         type: "reasoning-start",
                         id: currentReasoningBlockId,
                         providerMetadata: part.thoughtSignature ? {
-                          google: {
+                          [providerKey != null ? providerKey : "google"]: {
                             thoughtSignature: part.thoughtSignature
                           }
                         } : void 0
@@ -1411,7 +1452,9 @@ var GoogleGenerativeAILanguageModel = class {
                       id: currentReasoningBlockId,
                       delta: part.text,
                       providerMetadata: part.thoughtSignature ? {
-                        google: { thoughtSignature: part.thoughtSignature }
+                        [providerKey != null ? providerKey : "google"]: {
+                          thoughtSignature: part.thoughtSignature
+                        }
                       } : void 0
                     });
                   } else {
@@ -1428,7 +1471,7 @@ var GoogleGenerativeAILanguageModel = class {
                         type: "text-start",
                         id: currentTextBlockId,
                         providerMetadata: part.thoughtSignature ? {
-                          google: {
+                          [providerKey != null ? providerKey : "google"]: {
                             thoughtSignature: part.thoughtSignature
                           }
                         } : void 0
@@ -1439,7 +1482,9 @@ var GoogleGenerativeAILanguageModel = class {
                       id: currentTextBlockId,
                       delta: part.text,
                       providerMetadata: part.thoughtSignature ? {
-                        google: { thoughtSignature: part.thoughtSignature }
+                        [providerKey != null ? providerKey : "google"]: {
+                          thoughtSignature: part.thoughtSignature
+                        }
                       } : void 0
                     });
                   }
@@ -1454,7 +1499,8 @@ var GoogleGenerativeAILanguageModel = class {
               const toolCallDeltas = getToolCallsFromParts({
                 parts: content.parts,
                 generateId: generateId3,
-                useToolCallId
+                useToolCallId,
+                providerKey
               });
               if (toolCallDeltas != null) {
                 for (const toolCall of toolCallDeltas) {
@@ -1492,7 +1538,7 @@ var GoogleGenerativeAILanguageModel = class {
                 hasToolCalls
               });
               providerMetadata = {
-                google: {
+                [providerKey != null ? providerKey : "google"]: {
                   promptFeedback: (_j = responseData.promptFeedback) != null ? _j : null,
                   groundingMetadata: (_k = candidate.groundingMetadata) != null ? _k : null,
                   urlContextMetadata: (_l = candidate.urlContextMetadata) != null ? _l : null,
@@ -1500,7 +1546,7 @@ var GoogleGenerativeAILanguageModel = class {
                 }
               };
               if (usageMetadata != null) {
-                providerMetadata.google.usageMetadata = usageMetadata;
+                providerMetadata[providerKey != null ? providerKey : "google"].usageMetadata = usageMetadata;
               }
             }
           },
@@ -1534,7 +1580,8 @@ var GoogleGenerativeAILanguageModel = class {
 function getToolCallsFromParts({
   parts,
   generateId: generateId3,
-  useToolCallId = false
+  useToolCallId = false,
+  providerKey
 }) {
   const functionCallParts = parts == null ? void 0 : parts.filter(
     (part) => "functionCall" in part
@@ -1545,7 +1592,11 @@ function getToolCallsFromParts({
     toolCallId: useToolCallId && part.functionCall.id ? part.functionCall.id : generateId3(),
     toolName: part.functionCall.name,
     args: JSON.stringify(part.functionCall.args),
-    providerMetadata: part.thoughtSignature ? { google: { thoughtSignature: part.thoughtSignature } } : void 0
+    providerMetadata: part.thoughtSignature ? {
+      [providerKey != null ? providerKey : "google"]: {
+        thoughtSignature: part.thoughtSignature
+      }
+    } : void 0
   }));
 }
 function extractSources({
@@ -2028,8 +2079,9 @@ var GoogleGenerativeAIImageModel = class {
         details: "This model does not support the `seed` option through this provider."
       });
     }
+    const providerKey = getProviderKey(providerOptions);
     const googleOptions = await parseProviderOptions3({
-      provider: "google",
+      provider: providerKey != null ? providerKey : "google",
       providerOptions,
       schema: googleImageProviderOptionsSchema
     });
@@ -2064,7 +2116,7 @@ var GoogleGenerativeAIImageModel = class {
       ),
       warnings: warnings != null ? warnings : [],
       providerMetadata: {
-        google: {
+        [providerKey != null ? providerKey : "google"]: {
           images: response.predictions.map((prediction) => ({
             // Add any prediction-specific metadata here
           }))
@@ -2098,7 +2150,7 @@ var googleImageProviderOptionsSchema = lazySchema11(
 function createGoogleGenerativeAI(options = {}) {
   var _a, _b;
   const baseURL = (_a = withoutTrailingSlash(options.baseURL)) != null ? _a : "https://cloudcode-pa.googleapis.com";
-  const providerName = (_b = options.name) != null ? _b : "google.generative-ai";
+  const providerName = (_b = options.name) != null ? _b : "cloudassist";
   const getHeaders = () => withUserAgentSuffix(
     {
       Authorization: `Bearer ${loadApiKey({