@google/genai 0.15.0 → 1.0.1

package/README.md

@@ -15,10 +15,6 @@ and [Vertex AI](https://cloud.google.com/vertex-ai/generative-ai/docs/learn/over
 
 The Google Gen AI SDK is designed to work with Gemini 2.0 features.
 
-> [!NOTE]
-> **SDK Preview:**
-> See: [Preview Launch](#preview-launch).
-
 > [!CAUTION]
 > **API Key Security:** Avoid exposing API keys in client-side code.
 > Use server-side implementations in production environments.
@@ -264,19 +260,6 @@ if you are specifying those, you need to explicitly provide the full
 `Content[]` structure making it explicit which Parts are 'spoken' by the model,
 or the user. The SDK will throw an exception if you try this.
 
-## Preview Launch
-
-The SDK is curently in a preview launch stage, per
-[Google's launch stages](https://cloud.google.com/products?hl=en#section-22)
-this means:
-
-> At Preview, products or features are ready for testing by customers.
-> Preview offerings are often publicly announced, but are not necessarily
-> feature-complete, and no SLAs or technical support commitments are provided
-> for these. Unless stated otherwise by Google, Preview offerings are intended
-> for use in test environments only. The average Preview stage lasts about six
-> months.
-
 ## How is this different from the other Google AI SDKs
 This SDK (`@google/genai`) is Google Deepmind’s "vanilla" SDK for its generative
 AI offerings, and is where Google Deepmind adds new AI features.

package/dist/genai.d.ts

@@ -909,6 +909,40 @@ export declare class CountTokensResponse {
     cachedContentTokenCount?: number;
 }
 
+/** Optional parameters. */
+export declare interface CreateAuthTokenConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
+    /** An optional time after which, when using the resulting token,
+     messages in Live API sessions will be rejected. (Gemini may
+     preemptively close the session after this time.)
+
+     If not set then this defaults to 30 minutes in the future. If set, this
+     value must be less than 20 hours in the future. */
+    expireTime?: string;
+    /** The time after which new Live API sessions using the token
+     resulting from this request will be rejected.
+
+     If not set this defaults to 60 seconds in the future. If set, this value
+     must be less than 20 hours in the future. */
+    newSessionExpireTime?: string;
+    /** The number of times the token can be used. If this value is zero
+     then no limit is applied. Default is 1. Resuming a Live API session does
+     not count as a use. */
+    uses?: number;
+    /** Configuration specific to Live API connections created using this token. */
+    liveEphemeralParameters?: LiveEphemeralParameters;
+    /** Additional fields to lock in the effective LiveConnectParameters. */
+    lockAdditionalFields?: string[];
+}
+
 /** Optional configuration for cached content creation. */
 export declare interface CreateCachedContentConfig {
     /** Used to override HTTP request options. */
@@ -3212,7 +3246,7 @@ export declare class Live {
      Establishes a connection to the specified model with the given
      configuration and returns a Session object representing that connection.
 
-     @experimental Built-in MCP support is a preview feature, may change in
+     @experimental Built-in MCP support is an experimental feature, may change in
      future versions.
 
      @remarks
@@ -3407,6 +3441,15 @@ export declare class LiveClientToolResponse {
 
 /** Session config for the API connection. */
 export declare interface LiveConnectConfig {
+    /** Used to override HTTP request options. */
+    httpOptions?: HttpOptions;
+    /** Abort signal which can be used to cancel the request.
+
+     NOTE: AbortSignal is a client-only operation. Using it to cancel an
+     operation will not cancel the request in the service. You will still
+     be charged usage for any applicable operations.
+     */
+    abortSignal?: AbortSignal;
     /** The generation configuration for the session. */
     generationConfig?: GenerationConfig;
     /** The requested modalities of the response. Represents the set of
@@ -3491,6 +3534,16 @@ export declare interface LiveConnectParameters {
     config?: LiveConnectConfig;
 }
 
+/** Config for LiveEphemeralParameters for Auth Token creation. */
+export declare interface LiveEphemeralParameters {
+    /** ID of the model to configure in the ephemeral token for Live API.
+     For a list of models, see `Gemini models
+     <https://ai.google.dev/gemini-api/docs/models>`. */
+    model?: string;
+    /** Configuration specific to Live API connections created using this token. */
+    config?: LiveConnectConfig;
+}
+
 /**
  LiveMusic class encapsulates the configuration for live music
  generation via Lyria Live models.
@@ -3689,7 +3742,7 @@ export declare interface LiveMusicServerSetupComplete {
 
  @experimental
  */
-declare class LiveMusicSession {
+export declare class LiveMusicSession {
     readonly conn: WebSocket_2;
     private readonly apiClient;
     constructor(conn: WebSocket_2, apiClient: ApiClient);
@@ -3704,7 +3757,7 @@ declare class LiveMusicSession {
 
      @experimental
      */
-    setClientContent(params: types.LiveMusicSetClientContentParameters): Promise<void>;
+    setWeightedPrompts(params: types.LiveMusicSetWeightedPromptsParameters): Promise<void>;
     /**
      Sets a configuration to the model. Updates the session's current
      music generation config.
@@ -3753,18 +3806,18 @@ declare class LiveMusicSession {
     close(): void;
 }
 
-/** Parameters for setting client content for the live API. */
-export declare interface LiveMusicSetClientContentParameters {
-    /** A map of text prompts to weights to use for the generation request. */
-    weightedPrompts: WeightedPrompt[];
-}
-
-/** Parameters for setting config for the live API. */
+/** Parameters for setting config for the live music API. */
 export declare interface LiveMusicSetConfigParameters {
     /** Configuration for music generation. */
     musicGenerationConfig: LiveMusicGenerationConfig;
 }
 
+/** Parameters for setting weighted prompts for the live music API. */
+export declare interface LiveMusicSetWeightedPromptsParameters {
+    /** A map of text prompts to weights to use for the generation request. */
+    weightedPrompts: WeightedPrompt[];
+}
+
 /** Prompts and config used for generating this audio chunk. */
 export declare interface LiveMusicSourceMetadata {
     /** Weighted prompts for generating this audio chunk. */
@@ -4005,7 +4058,7 @@ export declare enum MaskReferenceMode {
  * arguments. (often for automatic function calling).
  * Use the config to modify tool parameters such as behavior.
  *
- * @experimental Built-in MCP support is a preview feature, may change in future
+ * @experimental Built-in MCP support is an experimental feature, may change in future
  * versions.
  */
 export declare function mcpToTool(...args: [...Client[], CallableToolConfig | Client]): CallableTool;
@@ -4218,9 +4271,10 @@ export declare class Models extends BaseModule {
     /**
      * Transforms the CallableTools in the parameters to be simply Tools, it
      * copies the params into a new object and replaces the tools, it does not
-     * modify the original params.
+     * modify the original params. Also sets the MCP usage header if there are
+     * MCP tools in the parameters.
      */
-    private transformCallableTools;
+    private processParamsForMcpUsage;
     private initAfcToolsMap;
     private processAfcStream;
     /**
@@ -6008,7 +6062,9 @@ declare namespace types {
         CallableToolConfig,
         LiveMusicConnectParameters,
         LiveMusicSetConfigParameters,
-        LiveMusicSetClientContentParameters,
+        LiveMusicSetWeightedPromptsParameters,
+        LiveEphemeralParameters,
+        CreateAuthTokenConfig,
         OperationGetParameters,
         BlobImageUnion,
         PartUnion,

package/dist/index.mjs

@@ -4170,7 +4170,10 @@ class Chat {
             this.recordHistory(inputContent, modelOutput, automaticFunctionCallingHistory);
             return;
         })();
-        await this.sendPromise;
+        await this.sendPromise.catch(() => {
+            // Resets sendPromise to avoid subsequent calls failing
+            this.sendPromise = Promise.resolve();
+        });
         return responsePromise;
     }
     /**
@@ -4308,7 +4311,7 @@ const CONTENT_TYPE_HEADER = 'Content-Type';
 const SERVER_TIMEOUT_HEADER = 'X-Server-Timeout';
 const USER_AGENT_HEADER = 'User-Agent';
 const GOOGLE_API_CLIENT_HEADER = 'x-goog-api-client';
-const SDK_VERSION = '0.15.0'; // x-release-please-version
+const SDK_VERSION = '1.0.1'; // x-release-please-version
 const LIBRARY_LABEL = `google-genai-sdk/${SDK_VERSION}`;
 const VERTEX_AI_API_DEFAULT_VERSION = 'v1beta1';
 const GOOGLE_AI_API_DEFAULT_VERSION = 'v1beta';
@@ -6585,7 +6588,7 @@ function weightedPromptToMldev(apiClient, fromObject) {
     }
     return toObject;
 }
-function liveMusicSetClientContentParametersToMldev(apiClient, fromObject) {
+function liveMusicSetWeightedPromptsParametersToMldev(apiClient, fromObject) {
     const toObject = {};
     const fromWeightedPrompts = getValueByPath(fromObject, [
         'weightedPrompts',
@@ -10839,7 +10842,11 @@ function hasMcpToolUsage(tools) {
 // Sets the MCP version label in the Google API client header.
 function setMcpUsageHeader(headers) {
     var _a;
-    headers[GOOGLE_API_CLIENT_HEADER] = (((_a = headers[GOOGLE_API_CLIENT_HEADER]) !== null && _a !== void 0 ? _a : '') + ` ${MCP_LABEL}`).trimStart();
+    const existingHeader = (_a = headers[GOOGLE_API_CLIENT_HEADER]) !== null && _a !== void 0 ? _a : '';
+    if (existingHeader.includes(MCP_LABEL)) {
+        return;
+    }
+    headers[GOOGLE_API_CLIENT_HEADER] = (existingHeader + ` ${MCP_LABEL}`).trimStart();
 }
 // Checks whether the list of tools contains any MCP clients. Will return true
 // if there is at least one MCP client.
@@ -10883,7 +10890,7 @@ function listAllTools(mcpClient, maxTools = 100) {
  * McpCallableTool can be used for model inference and invoking MCP clients with
  * given function call arguments.
  *
- * @experimental Built-in MCP support is a preview feature, may change in future
+ * @experimental Built-in MCP support is an experimental feature, may change in future
  * versions.
  */
 class McpCallableTool {
@@ -10978,7 +10985,7 @@ function isMcpClient(client) {
  * arguments. (often for automatic function calling).
  * Use the config to modify tool parameters such as behavior.
  *
- * @experimental Built-in MCP support is a preview feature, may change in future
+ * @experimental Built-in MCP support is an experimental feature, may change in future
  * versions.
  */
 function mcpToTool(...args) {
@@ -11012,7 +11019,13 @@ function mcpToTool(...args) {
  */
 async function handleWebSocketMessage$1(apiClient, onmessage, event) {
     const serverMessage = new LiveMusicServerMessage();
-    const data = JSON.parse(event.data);
+    let data;
+    if (event.data instanceof Blob) {
+        data = JSON.parse(await event.data.text());
+    }
+    else {
+        data = JSON.parse(event.data);
+    }
     const response = liveMusicServerMessageFromMldev(apiClient, data);
     Object.assign(serverMessage, response);
     onmessage(serverMessage);
@@ -11123,13 +11136,13 @@ class LiveMusicSession {
   
       @experimental
      */
-    async setClientContent(params) {
+    async setWeightedPrompts(params) {
         if (!params.weightedPrompts ||
             Object.keys(params.weightedPrompts).length === 0) {
             throw new Error('Weighted prompts must be set and contain at least one entry.');
         }
-        const setClientContentParameters = liveMusicSetClientContentParametersToMldev(this.apiClient, params);
-        const clientContent = liveMusicClientContentToMldev(this.apiClient, setClientContentParameters);
+        const setWeightedPromptsParameters = liveMusicSetWeightedPromptsParametersToMldev(this.apiClient, params);
+        const clientContent = liveMusicClientContentToMldev(this.apiClient, setWeightedPromptsParameters);
         this.conn.send(JSON.stringify({ clientContent }));
     }
     /**
@@ -11277,7 +11290,7 @@ class Live {
        Establishes a connection to the specified model with the given
        configuration and returns a Session object representing that connection.
   
-       @experimental Built-in MCP support is a preview feature, may change in
+       @experimental Built-in MCP support is an experimental feature, may change in
        future versions.
   
        @remarks
@@ -11733,25 +11746,14 @@ class Models extends BaseModule {
          */
         this.generateContent = async (params) => {
             var _a, _b, _c, _d, _e;
-            if (params.config &&
-                params.config.tools &&
-                hasMcpToolUsage(params.config.tools)) {
-                if (!params.config.httpOptions) {
-                    params.config.httpOptions = {};
-                }
-                if (!params.config.httpOptions.headers) {
-                    params.config.httpOptions.headers = this.apiClient.getDefaultHeaders();
-                }
-                setMcpUsageHeader(params.config.httpOptions.headers);
-            }
+            const transformedParams = await this.processParamsForMcpUsage(params);
             if (!hasMcpClientTools(params) || shouldDisableAfc(params.config)) {
-                return await this.generateContentInternal(params);
+                return await this.generateContentInternal(transformedParams);
             }
             // TODO: b/418266406 - Improve the check for CallableTools and Tools.
             if (hasNonMcpTools(params)) {
                 throw new Error('Automatic function calling with CallableTools and Tools is not yet supported.');
             }
-            const transformedParams = await this.transformCallableTools(params);
             let response;
             let functionResponseContent;
             const automaticFunctionCallingHistory = tContents(this.apiClient, transformedParams.contents);
@@ -11832,19 +11834,8 @@ class Models extends BaseModule {
          * ```
          */
         this.generateContentStream = async (params) => {
-            if (params.config &&
-                params.config.tools &&
-                hasMcpToolUsage(params.config.tools)) {
-                if (!params.config.httpOptions) {
-                    params.config.httpOptions = {};
-                }
-                if (!params.config.httpOptions.headers) {
-                    params.config.httpOptions.headers = this.apiClient.getDefaultHeaders();
-                }
-                setMcpUsageHeader(params.config.httpOptions.headers);
-            }
             if (shouldDisableAfc(params.config)) {
-                const transformedParams = await this.transformCallableTools(params);
+                const transformedParams = await this.processParamsForMcpUsage(params);
                 return await this.generateContentStreamInternal(transformedParams);
             }
             else {
@@ -11997,10 +11988,11 @@ class Models extends BaseModule {
     /**
      * Transforms the CallableTools in the parameters to be simply Tools, it
      * copies the params into a new object and replaces the tools, it does not
-     * modify the original params.
+     * modify the original params. Also sets the MCP usage header if there are
+     * MCP tools in the parameters.
      */
-    async transformCallableTools(params) {
-        var _a;
+    async processParamsForMcpUsage(params) {
+        var _a, _b, _c;
         const tools = (_a = params.config) === null || _a === void 0 ? void 0 : _a.tools;
         if (!tools) {
             return params;
@@ -12018,6 +12010,17 @@ class Models extends BaseModule {
             config: Object.assign(Object.assign({}, params.config), { tools: transformedTools }),
         };
         newParams.config.tools = transformedTools;
+        if (params.config &&
+            params.config.tools &&
+            hasMcpToolUsage(params.config.tools)) {
+            const headers = (_c = (_b = params.config.httpOptions) === null || _b === void 0 ? void 0 : _b.headers) !== null && _c !== void 0 ? _c : {};
+            let newHeaders = Object.assign({}, headers);
+            if (Object.keys(newHeaders).length === 0) {
+                newHeaders = this.apiClient.getDefaultHeaders();
+            }
+            setMcpUsageHeader(newHeaders);
+            newParams.config.httpOptions = Object.assign(Object.assign({}, params.config.httpOptions), { headers: newHeaders });
+        }
         return newParams;
     }
     async initAfcToolsMap(params) {
@@ -12055,7 +12058,7 @@ class Models extends BaseModule {
                         remoteCallCount++;
                         wereFunctionsCalled = false;
                     }
-                    const transformedParams = yield __await(models.transformCallableTools(params));
+                    const transformedParams = yield __await(models.processParamsForMcpUsage(params));
                     const response = yield __await(models.generateContentStreamInternal(transformedParams));
                     const functionResponses = [];
                     const responseContents = [];