@langchain/google-common 0.2.11 → 0.2.12

package/dist/chat_models.cjs

@@ -409,10 +409,12 @@ class ChatGoogleBase extends chat_models_1.BaseChatModel {
                 functionName = schema.name;
             }
             else {
+                // We are providing the schema for *just* the parameters, probably
+                const parameters = (0, zod_to_gemini_parameters_js_1.removeAdditionalProperties)(schema);
                 geminiFunctionDefinition = {
                     name: functionName,
                     description: schema.description ?? "",
-                    parameters: schema,
+                    parameters,
                 };
             }
             tools = [

package/dist/chat_models.js

@@ -11,7 +11,7 @@ import { AbstractGoogleLLMConnection } from "./connection.js";
 import { DefaultGeminiSafetyHandler, getGeminiAPI } from "./utils/gemini.js";
 import { ApiKeyGoogleAuth } from "./auth.js";
 import { ensureParams } from "./utils/failed_handler.js";
-import { schemaToGeminiParameters } from "./utils/zod_to_gemini_parameters.js";
+import { removeAdditionalProperties, schemaToGeminiParameters, } from "./utils/zod_to_gemini_parameters.js";
 export class ChatConnection extends AbstractGoogleLLMConnection {
     constructor(fields, caller, client, streaming) {
         super(fields, caller, client, streaming);
@@ -405,10 +405,12 @@ export class ChatGoogleBase extends BaseChatModel {
                 functionName = schema.name;
             }
             else {
+                // We are providing the schema for *just* the parameters, probably
+                const parameters = removeAdditionalProperties(schema);
                 geminiFunctionDefinition = {
                     name: functionName,
                     description: schema.description ?? "",
-                    parameters: schema,
+                    parameters,
                 };
             }
             tools = [

package/dist/types-anthropic.d.ts

@@ -154,6 +154,7 @@ export interface AnthropicUsage {
     output_tokens: number;
     cache_creation_input_tokens: number | null;
     cache_creation_output_tokens: number | null;
+    cache_read_input_tokens: number | null;
 }
 export type AnthropicResponseData = AnthropicResponseMessage | AnthropicStreamBaseEvent;
 export interface AnthropicResponseMessage {

package/dist/utils/anthropic.cjs

@@ -116,13 +116,23 @@ function getAnthropicAPI(config) {
         };
         return newAIMessageChunk(ret);
     }
-    function messageToGenerationInfo(message) {
+    function messageToUsageMetadata(message) {
         const usage = message?.usage;
+        const inputTokens = usage?.input_tokens ?? 0;
+        const outputTokens = usage?.output_tokens ?? 0;
         const usageMetadata = {
-            input_tokens: usage?.input_tokens ?? 0,
-            output_tokens: usage?.output_tokens ?? 0,
-            total_tokens: (usage?.input_tokens ?? 0) + (usage?.output_tokens ?? 0),
+            input_tokens: inputTokens,
+            output_tokens: outputTokens,
+            total_tokens: inputTokens + outputTokens,
+            input_token_details: {
+                cache_read: usage?.cache_read_input_tokens ?? 0,
+                cache_creation: usage?.cache_creation_input_tokens ?? 0,
+            },
         };
+        return usageMetadata;
+    }
+    function messageToGenerationInfo(message) {
+        const usageMetadata = messageToUsageMetadata(message);
         return {
             usage_metadata: usageMetadata,
             finish_reason: message.stop_reason,

package/dist/utils/anthropic.js

@@ -113,13 +113,23 @@ export function getAnthropicAPI(config) {
         };
         return newAIMessageChunk(ret);
     }
-    function messageToGenerationInfo(message) {
+    function messageToUsageMetadata(message) {
         const usage = message?.usage;
+        const inputTokens = usage?.input_tokens ?? 0;
+        const outputTokens = usage?.output_tokens ?? 0;
         const usageMetadata = {
-            input_tokens: usage?.input_tokens ?? 0,
-            output_tokens: usage?.output_tokens ?? 0,
-            total_tokens: (usage?.input_tokens ?? 0) + (usage?.output_tokens ?? 0),
+            input_tokens: inputTokens,
+            output_tokens: outputTokens,
+            total_tokens: inputTokens + outputTokens,
+            input_token_details: {
+                cache_read: usage?.cache_read_input_tokens ?? 0,
+                cache_creation: usage?.cache_creation_input_tokens ?? 0,
+            },
         };
+        return usageMetadata;
+    }
+    function messageToGenerationInfo(message) {
+        const usageMetadata = messageToUsageMetadata(message);
         return {
             usage_metadata: usageMetadata,
             finish_reason: message.stop_reason,

package/dist/utils/zod_to_gemini_parameters.cjs

@@ -1,9 +1,45 @@
 "use strict";
 /* eslint-disable @typescript-eslint/no-unused-vars */
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.jsonSchemaToGeminiParameters = exports.schemaToGeminiParameters = exports.removeAdditionalProperties = void 0;
+exports.jsonSchemaToGeminiParameters = exports.schemaToGeminiParameters = exports.removeAdditionalProperties = exports.adjustObjectType = void 0;
 const types_1 = require("@langchain/core/utils/types");
 const json_schema_1 = require("@langchain/core/utils/json_schema");
+/* eslint-disable no-param-reassign */
+function adjustObjectType(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+obj
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+) {
+    if (!Array.isArray(obj.type)) {
+        return obj;
+    }
+    const len = obj.type.length;
+    const nullIndex = obj.type.indexOf("null");
+    if (len === 2 && nullIndex >= 0) {
+        // There are only two values set for the type, and one of them is "null".
+        // Set the type to the other one and set nullable to true.
+        const typeIndex = nullIndex === 0 ? 1 : 0;
+        obj.type = obj.type[typeIndex];
+        obj.nullable = true;
+    }
+    else if (len === 1 && nullIndex === 0) {
+        // This is nullable only without a type, which doesn't
+        // make sense for Gemini
+        throw new Error("zod_to_gemini_parameters: Gemini cannot handle null type");
+    }
+    else if (len === 1) {
+        // Although an array, it has only one value.
+        // So set it to the string to match what Gemini expects.
+        obj.type = obj?.type[0];
+    }
+    else {
+        // Anything else could be a union type, so reject it.
+        throw new Error("zod_to_gemini_parameters: Gemini cannot handle union types");
+    }
+    return obj;
+}
+exports.adjustObjectType = adjustObjectType;
+/* eslint-enable no-param-reassign */
 function removeAdditionalProperties(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 obj) {
@@ -12,31 +48,7 @@ obj) {
         if ("additionalProperties" in newObj) {
             delete newObj.additionalProperties;
         }
-        if (Array.isArray(obj.type)) {
-            const len = obj.type.length;
-            const nullIndex = obj.type.indexOf("null");
-            if (len === 2 && nullIndex >= 0) {
-                // There are only two values set for the type, and one of them is "null".
-                // Set the type to the other one and set nullable to true.
-                const typeIndex = nullIndex === 0 ? 1 : 0;
-                newObj.type = obj.type[typeIndex];
-                newObj.nullable = true;
-            }
-            else if (len === 1 && nullIndex === 0) {
-                // This is nullable only without a type, which doesn't
-                // make sense for Gemini
-                throw new Error("zod_to_gemini_parameters: Gemini cannot handle null type");
-            }
-            else if (len === 1) {
-                // Although an array, it has only one value.
-                // So set it to the string to match what Gemini expects.
-                newObj.type = obj?.type[0];
-            }
-            else {
-                // Anything else could be a union type, so reject it.
-                throw new Error("zod_to_gemini_parameters: Gemini cannot handle union types");
-            }
-        }
+        adjustObjectType(newObj);
         for (const key in newObj) {
             if (key in newObj) {
                 if (Array.isArray(newObj[key])) {

package/dist/utils/zod_to_gemini_parameters.d.ts

@@ -1,6 +1,7 @@
 import { InteropZodType } from "@langchain/core/utils/types";
 import { type JsonSchema7Type } from "@langchain/core/utils/json_schema";
 import { GeminiFunctionSchema, GeminiJsonSchema } from "../types.js";
+export declare function adjustObjectType(obj: Record<string, any>): Record<string, any>;
 export declare function removeAdditionalProperties(obj: Record<string, any>): GeminiJsonSchema;
 export declare function schemaToGeminiParameters<RunOutput extends Record<string, any> = Record<string, any>>(schema: InteropZodType<RunOutput> | JsonSchema7Type): GeminiFunctionSchema;
 export declare function jsonSchemaToGeminiParameters(schema: Record<string, any>): GeminiFunctionSchema;

package/dist/utils/zod_to_gemini_parameters.js

@@ -1,6 +1,41 @@
 /* eslint-disable @typescript-eslint/no-unused-vars */
 import { isInteropZodSchema, } from "@langchain/core/utils/types";
 import { toJsonSchema, } from "@langchain/core/utils/json_schema";
+/* eslint-disable no-param-reassign */
+export function adjustObjectType(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+obj
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+) {
+    if (!Array.isArray(obj.type)) {
+        return obj;
+    }
+    const len = obj.type.length;
+    const nullIndex = obj.type.indexOf("null");
+    if (len === 2 && nullIndex >= 0) {
+        // There are only two values set for the type, and one of them is "null".
+        // Set the type to the other one and set nullable to true.
+        const typeIndex = nullIndex === 0 ? 1 : 0;
+        obj.type = obj.type[typeIndex];
+        obj.nullable = true;
+    }
+    else if (len === 1 && nullIndex === 0) {
+        // This is nullable only without a type, which doesn't
+        // make sense for Gemini
+        throw new Error("zod_to_gemini_parameters: Gemini cannot handle null type");
+    }
+    else if (len === 1) {
+        // Although an array, it has only one value.
+        // So set it to the string to match what Gemini expects.
+        obj.type = obj?.type[0];
+    }
+    else {
+        // Anything else could be a union type, so reject it.
+        throw new Error("zod_to_gemini_parameters: Gemini cannot handle union types");
+    }
+    return obj;
+}
+/* eslint-enable no-param-reassign */
 export function removeAdditionalProperties(
 // eslint-disable-next-line @typescript-eslint/no-explicit-any
 obj) {
@@ -9,31 +44,7 @@ obj) {
         if ("additionalProperties" in newObj) {
             delete newObj.additionalProperties;
         }
-        if (Array.isArray(obj.type)) {
-            const len = obj.type.length;
-            const nullIndex = obj.type.indexOf("null");
-            if (len === 2 && nullIndex >= 0) {
-                // There are only two values set for the type, and one of them is "null".
-                // Set the type to the other one and set nullable to true.
-                const typeIndex = nullIndex === 0 ? 1 : 0;
-                newObj.type = obj.type[typeIndex];
-                newObj.nullable = true;
-            }
-            else if (len === 1 && nullIndex === 0) {
-                // This is nullable only without a type, which doesn't
-                // make sense for Gemini
-                throw new Error("zod_to_gemini_parameters: Gemini cannot handle null type");
-            }
-            else if (len === 1) {
-                // Although an array, it has only one value.
-                // So set it to the string to match what Gemini expects.
-                newObj.type = obj?.type[0];
-            }
-            else {
-                // Anything else could be a union type, so reject it.
-                throw new Error("zod_to_gemini_parameters: Gemini cannot handle union types");
-            }
-        }
+        adjustObjectType(newObj);
         for (const key in newObj) {
             if (key in newObj) {
                 if (Array.isArray(newObj[key])) {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@langchain/google-common",
-  "version": "0.2.11",
+  "version": "0.2.12",
   "description": "Core types and classes for Google services.",
   "type": "module",
   "engines": {