@langchain/google-common 0.0.5 → 0.0.7

package/dist/utils/common.cjs

@@ -11,6 +11,7 @@ function copyAIModelParamsInto(params, options, target) {
     const model = options?.model ?? params?.model ?? target.model;
     ret.modelName =
         model ?? options?.modelName ?? params?.modelName ?? target.modelName;
+    ret.model = model;
     ret.temperature =
         options?.temperature ?? params?.temperature ?? target.temperature;
     ret.maxOutputTokens =

package/dist/utils/common.js

@@ -7,6 +7,7 @@ export function copyAIModelParamsInto(params, options, target) {
     const model = options?.model ?? params?.model ?? target.model;
     ret.modelName =
         model ?? options?.modelName ?? params?.modelName ?? target.modelName;
+    ret.model = model;
     ret.temperature =
         options?.temperature ?? params?.temperature ?? target.temperature;
     ret.maxOutputTokens =

package/dist/utils/gemini.cjs

@@ -5,6 +5,15 @@ const uuid_1 = require("uuid");
 const messages_1 = require("@langchain/core/messages");
 const outputs_1 = require("@langchain/core/outputs");
 const safety_js_1 = require("./safety.cjs");
+const extractMimeType = (str) => {
+    if (str.startsWith("data:")) {
+        return {
+            mimeType: str.split(":")[1].split(";")[0],
+            data: str.split(",")[1],
+        };
+    }
+    return null;
+};
 function messageContentText(content) {
     if (content?.text && content?.text.length > 0) {
         return {
@@ -22,12 +31,10 @@ function messageContentImageUrl(content) {
     if (!url) {
         throw new Error("Missing Image URL");
     }
-    if (url.startsWith("data:")) {
+    const mineTypeAndData = extractMimeType(url);
+    if (mineTypeAndData) {
         return {
-            inlineData: {
-                mimeType: url.split(":")[1].split(";")[0],
-                data: url.split(",")[1],
-            },
+            inlineData: mineTypeAndData,
         };
     }
     else {
@@ -40,6 +47,27 @@ function messageContentImageUrl(content) {
         };
     }
 }
+function messageContentMedia(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+content) {
+    if ("mimeType" in content && "data" in content) {
+        return {
+            inlineData: {
+                mimeType: content.mimeType,
+                data: content.data,
+            },
+        };
+    }
+    else if ("mimeType" in content && "fileUri" in content) {
+        return {
+            fileData: {
+                mimeType: content.mimeType,
+                fileUri: content.fileUri,
+            },
+        };
+    }
+    throw new Error("Invalid media content");
+}
 function messageContentToParts(content) {
     // Convert a string to a text type MessageContent if needed
     const messageContent = typeof content === "string"
@@ -65,6 +93,8 @@ function messageContentToParts(content) {
                     return messageContentImageUrl(content);
                 }
                 break;
+            case "media":
+                return messageContentMedia(content);
             default:
                 throw new Error(`Unsupported type received while converting message to message parts`);
         }
@@ -149,7 +179,6 @@ function toolMessageToContent(message, prevMessage) {
     const responseName = ((0, messages_1.isAIMessage)(prevMessage) && !!prevMessage.tool_calls?.length
         ? prevMessage.tool_calls[0].name
         : prevMessage.name) ?? message.tool_call_id;
-    console.log(contentStr);
     try {
         const content = JSON.parse(contentStr);
         return [

package/dist/utils/gemini.js

@@ -2,6 +2,15 @@ import { v4 as uuidv4 } from "uuid";
 import { AIMessage, AIMessageChunk, isAIMessage, } from "@langchain/core/messages";
 import { ChatGenerationChunk, } from "@langchain/core/outputs";
 import { GoogleAISafetyError } from "./safety.js";
+const extractMimeType = (str) => {
+    if (str.startsWith("data:")) {
+        return {
+            mimeType: str.split(":")[1].split(";")[0],
+            data: str.split(",")[1],
+        };
+    }
+    return null;
+};
 function messageContentText(content) {
     if (content?.text && content?.text.length > 0) {
         return {
@@ -19,12 +28,10 @@ function messageContentImageUrl(content) {
     if (!url) {
         throw new Error("Missing Image URL");
     }
-    if (url.startsWith("data:")) {
+    const mineTypeAndData = extractMimeType(url);
+    if (mineTypeAndData) {
         return {
-            inlineData: {
-                mimeType: url.split(":")[1].split(";")[0],
-                data: url.split(",")[1],
-            },
+            inlineData: mineTypeAndData,
         };
     }
     else {
@@ -37,6 +44,27 @@ function messageContentImageUrl(content) {
         };
     }
 }
+function messageContentMedia(
+// eslint-disable-next-line @typescript-eslint/no-explicit-any
+content) {
+    if ("mimeType" in content && "data" in content) {
+        return {
+            inlineData: {
+                mimeType: content.mimeType,
+                data: content.data,
+            },
+        };
+    }
+    else if ("mimeType" in content && "fileUri" in content) {
+        return {
+            fileData: {
+                mimeType: content.mimeType,
+                fileUri: content.fileUri,
+            },
+        };
+    }
+    throw new Error("Invalid media content");
+}
 export function messageContentToParts(content) {
     // Convert a string to a text type MessageContent if needed
     const messageContent = typeof content === "string"
@@ -62,6 +90,8 @@ export function messageContentToParts(content) {
                     return messageContentImageUrl(content);
                 }
                 break;
+            case "media":
+                return messageContentMedia(content);
             default:
                 throw new Error(`Unsupported type received while converting message to message parts`);
         }
@@ -145,7 +175,6 @@ function toolMessageToContent(message, prevMessage) {
     const responseName = (isAIMessage(prevMessage) && !!prevMessage.tool_calls?.length
         ? prevMessage.tool_calls[0].name
         : prevMessage.name) ?? message.tool_call_id;
-    console.log(contentStr);
     try {
         const content = JSON.parse(contentStr);
         return [

package/package.json

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