@langchain/google-common 0.0.3 → 0.0.4

package/dist/utils/gemini.cjs

@@ -54,12 +54,20 @@ function messageContentToParts(content) {
         .map((content) => {
         switch (content.type) {
             case "text":
-                return messageContentText(content);
+                if ("text" in content) {
+                    return messageContentText(content);
+                }
+                break;
             case "image_url":
-                return messageContentImageUrl(content);
+                if ("image_url" in content) {
+                    // Type guard for MessageContentImageUrl
+                    return messageContentImageUrl(content);
+                }
+                break;
             default:
                 throw new Error(`Unsupported type received while converting message to message parts`);
         }
+        throw new Error(`Cannot coerce "${content.type}" message part into a string.`);
     })
         .reduce((acc, val) => {
         if (val) {
@@ -393,7 +401,21 @@ function partToChatGeneration(part) {
 exports.partToChatGeneration = partToChatGeneration;
 function responseToChatGenerations(response) {
     const parts = responseToParts(response);
-    const ret = parts.map((part) => partToChatGeneration(part));
+    let ret = parts.map((part) => partToChatGeneration(part));
+    if (ret.every((item) => typeof item.message.content === "string")) {
+        const combinedContent = ret.map((item) => item.message.content).join("");
+        const combinedText = ret.map((item) => item.text).join("");
+        ret = [
+            new outputs_1.ChatGenerationChunk({
+                message: new messages_1.AIMessageChunk({
+                    content: combinedContent,
+                    additional_kwargs: ret[ret.length - 1].message.additional_kwargs,
+                }),
+                text: combinedText,
+                generationInfo: ret[ret.length - 1].generationInfo,
+            }),
+        ];
+    }
     return ret;
 }
 exports.responseToChatGenerations = responseToChatGenerations;

package/dist/utils/gemini.js

@@ -51,12 +51,20 @@ export function messageContentToParts(content) {
         .map((content) => {
         switch (content.type) {
             case "text":
-                return messageContentText(content);
+                if ("text" in content) {
+                    return messageContentText(content);
+                }
+                break;
             case "image_url":
-                return messageContentImageUrl(content);
+                if ("image_url" in content) {
+                    // Type guard for MessageContentImageUrl
+                    return messageContentImageUrl(content);
+                }
+                break;
             default:
                 throw new Error(`Unsupported type received while converting message to message parts`);
         }
+        throw new Error(`Cannot coerce "${content.type}" message part into a string.`);
     })
         .reduce((acc, val) => {
         if (val) {
@@ -373,7 +381,21 @@ export function partToChatGeneration(part) {
 }
 export function responseToChatGenerations(response) {
     const parts = responseToParts(response);
-    const ret = parts.map((part) => partToChatGeneration(part));
+    let ret = parts.map((part) => partToChatGeneration(part));
+    if (ret.every((item) => typeof item.message.content === "string")) {
+        const combinedContent = ret.map((item) => item.message.content).join("");
+        const combinedText = ret.map((item) => item.text).join("");
+        ret = [
+            new ChatGenerationChunk({
+                message: new AIMessageChunk({
+                    content: combinedContent,
+                    additional_kwargs: ret[ret.length - 1].message.additional_kwargs,
+                }),
+                text: combinedText,
+                generationInfo: ret[ret.length - 1].generationInfo,
+            }),
+        ];
+    }
     return ret;
 }
 export function responseToBaseMessageFields(response) {

package/package.json

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