@langchain/google-common 0.2.5 → 0.2.6

package/dist/connection.cjs

@@ -114,17 +114,17 @@ class GoogleHostConnection extends GoogleConnection {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "apiVersion", {
+        Object.defineProperty(this, "_apiVersion", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: "v1"
+            value: void 0
         });
         this.caller = caller;
         this.platformType = fields?.platformType;
         this._endpoint = fields?.endpoint;
         this._location = fields?.location;
-        this.apiVersion = fields?.apiVersion ?? this.apiVersion;
+        this._apiVersion = fields?.apiVersion;
         this.client = client;
     }
     get platform() {
@@ -133,6 +133,12 @@ class GoogleHostConnection extends GoogleConnection {
     get computedPlatformType() {
         return "gcp";
     }
+    get computedApiVersion() {
+        return "v1";
+    }
+    get apiVersion() {
+        return this._apiVersion ?? this.computedApiVersion;
+    }
     get location() {
         return this._location ?? this.computedLocation;
     }
@@ -238,6 +244,14 @@ class GoogleAIConnection extends GoogleHostConnection {
             return "gcp";
         }
     }
+    get computedApiVersion() {
+        switch (this.platform) {
+            case "gai":
+                return "v1beta";
+            default:
+                return "v1";
+        }
+    }
     get computedLocation() {
         switch (this.apiName) {
             case "google":

package/dist/connection.d.ts

@@ -25,10 +25,12 @@ export declare abstract class GoogleHostConnection<CallOptions extends AsyncCall
     platformType: GooglePlatformType | undefined;
     _endpoint: string | undefined;
     _location: string | undefined;
-    apiVersion: string;
+    _apiVersion: string | undefined;
     constructor(fields: GoogleConnectionParams<AuthOptions> | undefined, caller: AsyncCaller, client: GoogleAbstractedClient, streaming?: boolean);
     get platform(): GooglePlatformType;
     get computedPlatformType(): GooglePlatformType;
+    get computedApiVersion(): string;
+    get apiVersion(): string;
     get location(): string;
     get computedLocation(): string;
     get endpoint(): string;
@@ -52,6 +54,7 @@ export declare abstract class GoogleAIConnection<CallOptions extends AsyncCaller
     get api(): GoogleAIAPI;
     get isApiKey(): boolean;
     get computedPlatformType(): GooglePlatformType;
+    get computedApiVersion(): string;
     get computedLocation(): string;
     abstract buildUrlMethod(): Promise<string>;
     buildUrlGenerativeLanguage(): Promise<string>;

package/dist/connection.js

@@ -110,17 +110,17 @@ export class GoogleHostConnection extends GoogleConnection {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "apiVersion", {
+        Object.defineProperty(this, "_apiVersion", {
             enumerable: true,
             configurable: true,
             writable: true,
-            value: "v1"
+            value: void 0
         });
         this.caller = caller;
         this.platformType = fields?.platformType;
         this._endpoint = fields?.endpoint;
         this._location = fields?.location;
-        this.apiVersion = fields?.apiVersion ?? this.apiVersion;
+        this._apiVersion = fields?.apiVersion;
         this.client = client;
     }
     get platform() {
@@ -129,6 +129,12 @@ export class GoogleHostConnection extends GoogleConnection {
     get computedPlatformType() {
         return "gcp";
     }
+    get computedApiVersion() {
+        return "v1";
+    }
+    get apiVersion() {
+        return this._apiVersion ?? this.computedApiVersion;
+    }
     get location() {
         return this._location ?? this.computedLocation;
     }
@@ -232,6 +238,14 @@ export class GoogleAIConnection extends GoogleHostConnection {
             return "gcp";
         }
     }
+    get computedApiVersion() {
+        switch (this.platform) {
+            case "gai":
+                return "v1beta";
+            default:
+                return "v1";
+        }
+    }
     get computedLocation() {
         switch (this.apiName) {
             case "google":

package/dist/experimental/media.cjs

@@ -328,14 +328,8 @@ class AIStudioMediaBlob extends media_core_js_1.MediaBlob {
 }
 exports.AIStudioMediaBlob = AIStudioMediaBlob;
 class AIStudioFileUploadConnection extends GoogleMultipartUploadConnection {
-    constructor() {
-        super(...arguments);
-        Object.defineProperty(this, "apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "v1beta"
-        });
+    get computedApiVersion() {
+        return "v1beta";
     }
     async buildUrl() {
         return `https://generativelanguage.googleapis.com/upload/${this.apiVersion}/files`;
@@ -357,15 +351,12 @@ class AIStudioFileDownloadConnection extends GoogleDownloadConnection {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "v1beta"
-        });
         this.method = fields.method;
         this.name = fields.name;
     }
+    get computedApiVersion() {
+        return "v1beta";
+    }
     buildMethod() {
         return this.method;
     }

package/dist/experimental/media.d.ts

@@ -166,7 +166,7 @@ export interface AIStudioFileConnectionParams {
 export interface AIStudioFileUploadConnectionParams<AuthOptions> extends GoogleUploadConnectionParams<AuthOptions>, AIStudioFileConnectionParams {
 }
 export declare class AIStudioFileUploadConnection<AuthOptions> extends GoogleMultipartUploadConnection<AsyncCallerCallOptions, AIStudioFileSaveResponse, AuthOptions> {
-    apiVersion: string;
+    get computedApiVersion(): string;
     buildUrl(): Promise<string>;
 }
 export interface AIStudioFileDownloadConnectionParams<AuthOptions> extends AIStudioFileConnectionParams, GoogleConnectionParams<AuthOptions> {
@@ -176,8 +176,8 @@ export interface AIStudioFileDownloadConnectionParams<AuthOptions> extends AIStu
 export declare class AIStudioFileDownloadConnection<ResponseType extends GoogleResponse, AuthOptions> extends GoogleDownloadConnection<AsyncCallerCallOptions, ResponseType, AuthOptions> {
     method: GoogleAbstractedClientOpsMethod;
     name: string;
-    apiVersion: string;
     constructor(fields: AIStudioFileDownloadConnectionParams<AuthOptions>, caller: AsyncCaller, client: GoogleAbstractedClient);
+    get computedApiVersion(): string;
     buildMethod(): GoogleAbstractedClientOpsMethod;
     buildUrl(): Promise<string>;
 }

package/dist/experimental/media.js

@@ -315,14 +315,8 @@ export class AIStudioMediaBlob extends MediaBlob {
     }
 }
 export class AIStudioFileUploadConnection extends GoogleMultipartUploadConnection {
-    constructor() {
-        super(...arguments);
-        Object.defineProperty(this, "apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "v1beta"
-        });
+    get computedApiVersion() {
+        return "v1beta";
     }
     async buildUrl() {
         return `https://generativelanguage.googleapis.com/upload/${this.apiVersion}/files`;
@@ -343,15 +337,12 @@ export class AIStudioFileDownloadConnection extends GoogleDownloadConnection {
             writable: true,
             value: void 0
         });
-        Object.defineProperty(this, "apiVersion", {
-            enumerable: true,
-            configurable: true,
-            writable: true,
-            value: "v1beta"
-        });
         this.method = fields.method;
         this.name = fields.name;
     }
+    get computedApiVersion() {
+        return "v1beta";
+    }
     buildMethod() {
         return this.method;
     }

package/dist/utils/gemini.cjs

@@ -651,12 +651,9 @@ function getGeminiAPI(config) {
         if (!data) {
             return {};
         }
-        return {
-            usage_metadata: {
-                prompt_token_count: data.usageMetadata?.promptTokenCount,
-                candidates_token_count: data.usageMetadata?.candidatesTokenCount,
-                total_token_count: data.usageMetadata?.totalTokenCount,
-            },
+        const finish_reason = data.candidates[0]?.finishReason;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const ret = {
             safety_ratings: data.candidates[0]?.safetyRatings?.map((rating) => ({
                 category: rating.category,
                 probability: rating.probability,
@@ -666,11 +663,21 @@ function getGeminiAPI(config) {
             })),
             citation_metadata: data.candidates[0]?.citationMetadata,
             grounding_metadata: data.candidates[0]?.groundingMetadata,
-            finish_reason: data.candidates[0]?.finishReason,
+            finish_reason,
             finish_message: data.candidates[0]?.finishMessage,
             avgLogprobs: data.candidates[0]?.avgLogprobs,
             logprobs: candidateToLogprobs(data.candidates[0]),
         };
+        // Only add the usage_metadata on the last chunk
+        // sent while streaming (see issue 8102).
+        if (typeof finish_reason === "string") {
+            ret.usage_metadata = {
+                prompt_token_count: data.usageMetadata?.promptTokenCount,
+                candidates_token_count: data.usageMetadata?.candidatesTokenCount,
+                total_token_count: data.usageMetadata?.totalTokenCount,
+            };
+        }
+        return ret;
     }
     function responseToChatGeneration(response) {
         return new outputs_1.ChatGenerationChunk({

package/dist/utils/gemini.js

@@ -646,12 +646,9 @@ export function getGeminiAPI(config) {
         if (!data) {
             return {};
         }
-        return {
-            usage_metadata: {
-                prompt_token_count: data.usageMetadata?.promptTokenCount,
-                candidates_token_count: data.usageMetadata?.candidatesTokenCount,
-                total_token_count: data.usageMetadata?.totalTokenCount,
-            },
+        const finish_reason = data.candidates[0]?.finishReason;
+        // eslint-disable-next-line @typescript-eslint/no-explicit-any
+        const ret = {
             safety_ratings: data.candidates[0]?.safetyRatings?.map((rating) => ({
                 category: rating.category,
                 probability: rating.probability,
@@ -661,11 +658,21 @@ export function getGeminiAPI(config) {
             })),
             citation_metadata: data.candidates[0]?.citationMetadata,
             grounding_metadata: data.candidates[0]?.groundingMetadata,
-            finish_reason: data.candidates[0]?.finishReason,
+            finish_reason,
             finish_message: data.candidates[0]?.finishMessage,
             avgLogprobs: data.candidates[0]?.avgLogprobs,
             logprobs: candidateToLogprobs(data.candidates[0]),
         };
+        // Only add the usage_metadata on the last chunk
+        // sent while streaming (see issue 8102).
+        if (typeof finish_reason === "string") {
+            ret.usage_metadata = {
+                prompt_token_count: data.usageMetadata?.promptTokenCount,
+                candidates_token_count: data.usageMetadata?.candidatesTokenCount,
+                total_token_count: data.usageMetadata?.totalTokenCount,
+            };
+        }
+        return ret;
     }
     function responseToChatGeneration(response) {
         return new ChatGenerationChunk({

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/google-common",
-  "version": "0.2.5",
+  "version": "0.2.6",
   "description": "Core types and classes for Google services.",
   "type": "module",
   "engines": {
@@ -40,7 +40,7 @@
   },
   "devDependencies": {
     "@jest/globals": "^29.5.0",
-    "@langchain/core": "0.3.48",
+    "@langchain/core": "0.3.53",
     "@langchain/scripts": ">=0.1.0 <0.2.0",
     "@swc/core": "^1.3.90",
     "@swc/jest": "^0.2.29",