langchain 0.0.180 → 0.0.182-rc.0

package/dist/chat_models/openai.js

@@ -1,5 +1,5 @@
 import { OpenAI as OpenAIClient } from "openai";
-import { AIMessage, AIMessageChunk, ChatGenerationChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessage, HumanMessageChunk, SystemMessage, SystemMessageChunk, } from "../schema/index.js";
+import { AIMessage, AIMessageChunk, ChatGenerationChunk, ChatMessage, ChatMessageChunk, FunctionMessageChunk, HumanMessageChunk, SystemMessageChunk, } from "../schema/index.js";
 import { formatToOpenAIFunction } from "../tools/convert_to_openai.js";
 import { getEndpoint } from "../util/azure.js";
 import { getEnvironmentVariable } from "../util/env.js";
@@ -51,14 +51,10 @@ function messageToOpenAIMessage(message) {
 }
 function openAIResponseToChatMessage(message) {
     switch (message.role) {
-        case "user":
-            return new HumanMessage(message.content || "");
         case "assistant":
             return new AIMessage(message.content || "", {
                 function_call: message.function_call,
             });
-        case "system":
-            return new SystemMessage(message.content || "");
         default:
             return new ChatMessage(message.content || "", message.role ?? "unknown");
     }
@@ -574,7 +570,8 @@ export class ChatOpenAI extends BaseChatModel {
             let count = textCount + tokensPerMessage + roleCount + nameCount;
             // From: https://github.com/hmarr/openai-chat-tokens/blob/main/src/index.ts messageTokenEstimate
             const openAIMessage = messageToOpenAIMessage(message);
-            if (openAIMessage.role === "function") {
+            if (openAIMessage.role === "function" ||
+                openAIMessage.role === "tool") {
                 count -= 2;
             }
             if (openAIMessage.function_call) {

package/dist/chat_models/portkey.cjs

@@ -98,10 +98,15 @@ class PortkeyChat extends base_js_1.BaseChatModel {
         return "portkey";
     }
     async _generate(messages, options, _) {
-        const messagesList = messages.map((message) => ({
-            role: message._getType(),
-            content: message.content,
-        }));
+        const messagesList = messages.map((message) => {
+            if (typeof message.content !== "string") {
+                throw new Error("PortkeyChat does not support non-string message content.");
+            }
+            return {
+                role: message._getType(),
+                content: message.content,
+            };
+        });
         const response = await this.session.portkey.chatCompletions.create({
             messages: messagesList,
             ...options,
@@ -124,10 +129,15 @@ class PortkeyChat extends base_js_1.BaseChatModel {
         };
     }
     async *_streamResponseChunks(messages, options, runManager) {
-        const messagesList = messages.map((message) => ({
-            role: message._getType(),
-            content: message.content,
-        }));
+        const messagesList = messages.map((message) => {
+            if (typeof message.content !== "string") {
+                throw new Error("PortkeyChat does not support non-string message content.");
+            }
+            return {
+                role: message._getType(),
+                content: message.content,
+            };
+        });
         const response = await this.session.portkey.chatCompletions.create({
             messages: messagesList,
             ...options,

package/dist/chat_models/portkey.js

@@ -95,10 +95,15 @@ export class PortkeyChat extends BaseChatModel {
         return "portkey";
     }
     async _generate(messages, options, _) {
-        const messagesList = messages.map((message) => ({
-            role: message._getType(),
-            content: message.content,
-        }));
+        const messagesList = messages.map((message) => {
+            if (typeof message.content !== "string") {
+                throw new Error("PortkeyChat does not support non-string message content.");
+            }
+            return {
+                role: message._getType(),
+                content: message.content,
+            };
+        });
         const response = await this.session.portkey.chatCompletions.create({
             messages: messagesList,
             ...options,
@@ -121,10 +126,15 @@ export class PortkeyChat extends BaseChatModel {
         };
     }
     async *_streamResponseChunks(messages, options, runManager) {
-        const messagesList = messages.map((message) => ({
-            role: message._getType(),
-            content: message.content,
-        }));
+        const messagesList = messages.map((message) => {
+            if (typeof message.content !== "string") {
+                throw new Error("PortkeyChat does not support non-string message content.");
+            }
+            return {
+                role: message._getType(),
+                content: message.content,
+            };
+        });
         const response = await this.session.portkey.chatCompletions.create({
             messages: messagesList,
             ...options,

package/dist/chat_models/yandex.cjs

@@ -9,6 +9,9 @@ function _parseChatHistory(history) {
     const chatHistory = [];
     let instruction = "";
     for (const message of history) {
+        if (typeof message.content !== "string") {
+            throw new Error("ChatYandexGPT does not support non-string message content.");
+        }
         if ("content" in message) {
             if (message._getType() === "human") {
                 chatHistory.push({ role: "user", text: message.content });

package/dist/chat_models/yandex.js

@@ -6,6 +6,9 @@ function _parseChatHistory(history) {
     const chatHistory = [];
     let instruction = "";
     for (const message of history) {
+        if (typeof message.content !== "string") {
+            throw new Error("ChatYandexGPT does not support non-string message content.");
+        }
         if ("content" in message) {
             if (message._getType() === "human") {
                 chatHistory.push({ role: "user", text: message.content });

package/dist/embeddings/cache_backed.cjs

@@ -1,10 +1,7 @@
 "use strict";
-var __importDefault = (this && this.__importDefault) || function (mod) {
-    return (mod && mod.__esModule) ? mod : { "default": mod };
-};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.CacheBackedEmbeddings = void 0;
-const object_hash_1 = __importDefault(require("object-hash"));
+const hash_js_1 = require("../util/js-sha1/hash.cjs");
 const encoder_backed_js_1 = require("../storage/encoder_backed.cjs");
 const base_js_1 = require("./base.cjs");
 /**
@@ -96,7 +93,7 @@ class CacheBackedEmbeddings extends base_js_1.Embeddings {
         const decoder = new TextDecoder();
         const encoderBackedStore = new encoder_backed_js_1.EncoderBackedStore({
             store: documentEmbeddingStore,
-            keyEncoder: (key) => (options?.namespace ?? "") + (0, object_hash_1.default)(key),
+            keyEncoder: (key) => (options?.namespace ?? "") + (0, hash_js_1.insecureHash)(key),
             valueSerializer: (value) => encoder.encode(JSON.stringify(value)),
             valueDeserializer: (serializedValue) => JSON.parse(decoder.decode(serializedValue)),
         });

package/dist/embeddings/cache_backed.js

@@ -1,4 +1,4 @@
-import hash from "object-hash";
+import { insecureHash } from "../util/js-sha1/hash.js";
 import { EncoderBackedStore } from "../storage/encoder_backed.js";
 import { Embeddings } from "./base.js";
 /**
@@ -90,7 +90,7 @@ export class CacheBackedEmbeddings extends Embeddings {
         const decoder = new TextDecoder();
         const encoderBackedStore = new EncoderBackedStore({
             store: documentEmbeddingStore,
-            keyEncoder: (key) => (options?.namespace ?? "") + hash(key),
+            keyEncoder: (key) => (options?.namespace ?? "") + insecureHash(key),
             valueSerializer: (value) => encoder.encode(JSON.stringify(value)),
             valueDeserializer: (serializedValue) => JSON.parse(decoder.decode(serializedValue)),
         });

package/dist/embeddings/voyage.cjs

@@ -0,0 +1,120 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.VoyageEmbeddings = void 0;
+const chunk_js_1 = require("../util/chunk.cjs");
+const env_js_1 = require("../util/env.cjs");
+const base_js_1 = require("./base.cjs");
+/**
+ * A class for generating embeddings using the Voyage AI API.
+ */
+class VoyageEmbeddings extends base_js_1.Embeddings {
+    /**
+     * Constructor for the VoyageEmbeddings class.
+     * @param fields - An optional object with properties to configure the instance.
+     */
+    constructor(fields) {
+        const fieldsWithDefaults = { ...fields };
+        super(fieldsWithDefaults);
+        Object.defineProperty(this, "modelName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "voyage-01"
+        });
+        Object.defineProperty(this, "batchSize", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 8
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "basePath", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "https://api.voyageai.com/v1"
+        });
+        Object.defineProperty(this, "apiUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "headers", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const apiKey = fieldsWithDefaults?.apiKey || (0, env_js_1.getEnvironmentVariable)("VOYAGEAI_API_KEY");
+        if (!apiKey) {
+            throw new Error("Voyage AI API key not found");
+        }
+        this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
+        this.batchSize = fieldsWithDefaults?.batchSize ?? this.batchSize;
+        this.apiKey = apiKey;
+        this.apiUrl = `${this.basePath}/embeddings`;
+    }
+    /**
+     * Generates embeddings for an array of texts.
+     * @param texts - An array of strings to generate embeddings for.
+     * @returns A Promise that resolves to an array of embeddings.
+     */
+    async embedDocuments(texts) {
+        const batches = (0, chunk_js_1.chunkArray)(texts, this.batchSize);
+        const batchRequests = batches.map((batch) => this.embeddingWithRetry({
+            model: this.modelName,
+            input: batch,
+        }));
+        const batchResponses = await Promise.all(batchRequests);
+        const embeddings = [];
+        for (let i = 0; i < batchResponses.length; i += 1) {
+            const batch = batches[i];
+            const { data: batchResponse } = batchResponses[i];
+            for (let j = 0; j < batch.length; j += 1) {
+                embeddings.push(batchResponse[j].embedding);
+            }
+        }
+        return embeddings;
+    }
+    /**
+     * Generates an embedding for a single text.
+     * @param text - A string to generate an embedding for.
+     * @returns A Promise that resolves to an array of numbers representing the embedding.
+     */
+    async embedQuery(text) {
+        const { data } = await this.embeddingWithRetry({
+            model: this.modelName,
+            input: text,
+        });
+        return data[0].embedding;
+    }
+    /**
+     * Makes a request to the Voyage AI API to generate embeddings for an array of texts.
+     * @param request - An object with properties to configure the request.
+     * @returns A Promise that resolves to the response from the Voyage AI API.
+     */
+    async embeddingWithRetry(request) {
+        const makeCompletionRequest = async () => {
+            const url = `${this.apiUrl}`;
+            const response = await fetch(url, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    Authorization: `Bearer ${this.apiKey}`,
+                    ...this.headers,
+                },
+                body: JSON.stringify(request),
+            });
+            const json = await response.json();
+            return json;
+        };
+        return this.caller.call(makeCompletionRequest);
+    }
+}
+exports.VoyageEmbeddings = VoyageEmbeddings;

package/dist/embeddings/voyage.d.ts

@@ -0,0 +1,66 @@
+import { Embeddings, EmbeddingsParams } from "./base.js";
+/**
+ * Interface that extends EmbeddingsParams and defines additional
+ * parameters specific to the VoyageEmbeddings class.
+ */
+export interface VoyageEmbeddingsParams extends EmbeddingsParams {
+    modelName: string;
+    /**
+     * The maximum number of documents to embed in a single request. This is
+     * limited by the Voyage AI API to a maximum of 8.
+     */
+    batchSize?: number;
+}
+/**
+ * Interface for the request body to generate embeddings.
+ */
+export interface CreateVoyageEmbeddingRequest {
+    /**
+     * @type {string}
+     * @memberof CreateVoyageEmbeddingRequest
+     */
+    model: string;
+    /**
+     *  Text to generate vector expectation
+     * @type {CreateEmbeddingRequestInput}
+     * @memberof CreateVoyageEmbeddingRequest
+     */
+    input: string | string[];
+}
+/**
+ * A class for generating embeddings using the Voyage AI API.
+ */
+export declare class VoyageEmbeddings extends Embeddings implements VoyageEmbeddingsParams {
+    modelName: string;
+    batchSize: number;
+    private apiKey;
+    basePath?: string;
+    apiUrl: string;
+    headers?: Record<string, string>;
+    /**
+     * Constructor for the VoyageEmbeddings class.
+     * @param fields - An optional object with properties to configure the instance.
+     */
+    constructor(fields?: Partial<VoyageEmbeddingsParams> & {
+        verbose?: boolean;
+        apiKey?: string;
+    });
+    /**
+     * Generates embeddings for an array of texts.
+     * @param texts - An array of strings to generate embeddings for.
+     * @returns A Promise that resolves to an array of embeddings.
+     */
+    embedDocuments(texts: string[]): Promise<number[][]>;
+    /**
+     * Generates an embedding for a single text.
+     * @param text - A string to generate an embedding for.
+     * @returns A Promise that resolves to an array of numbers representing the embedding.
+     */
+    embedQuery(text: string): Promise<number[]>;
+    /**
+     * Makes a request to the Voyage AI API to generate embeddings for an array of texts.
+     * @param request - An object with properties to configure the request.
+     * @returns A Promise that resolves to the response from the Voyage AI API.
+     */
+    private embeddingWithRetry;
+}

package/dist/embeddings/voyage.js

@@ -0,0 +1,116 @@
+import { chunkArray } from "../util/chunk.js";
+import { getEnvironmentVariable } from "../util/env.js";
+import { Embeddings } from "./base.js";
+/**
+ * A class for generating embeddings using the Voyage AI API.
+ */
+export class VoyageEmbeddings extends Embeddings {
+    /**
+     * Constructor for the VoyageEmbeddings class.
+     * @param fields - An optional object with properties to configure the instance.
+     */
+    constructor(fields) {
+        const fieldsWithDefaults = { ...fields };
+        super(fieldsWithDefaults);
+        Object.defineProperty(this, "modelName", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "voyage-01"
+        });
+        Object.defineProperty(this, "batchSize", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: 8
+        });
+        Object.defineProperty(this, "apiKey", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "basePath", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: "https://api.voyageai.com/v1"
+        });
+        Object.defineProperty(this, "apiUrl", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        Object.defineProperty(this, "headers", {
+            enumerable: true,
+            configurable: true,
+            writable: true,
+            value: void 0
+        });
+        const apiKey = fieldsWithDefaults?.apiKey || getEnvironmentVariable("VOYAGEAI_API_KEY");
+        if (!apiKey) {
+            throw new Error("Voyage AI API key not found");
+        }
+        this.modelName = fieldsWithDefaults?.modelName ?? this.modelName;
+        this.batchSize = fieldsWithDefaults?.batchSize ?? this.batchSize;
+        this.apiKey = apiKey;
+        this.apiUrl = `${this.basePath}/embeddings`;
+    }
+    /**
+     * Generates embeddings for an array of texts.
+     * @param texts - An array of strings to generate embeddings for.
+     * @returns A Promise that resolves to an array of embeddings.
+     */
+    async embedDocuments(texts) {
+        const batches = chunkArray(texts, this.batchSize);
+        const batchRequests = batches.map((batch) => this.embeddingWithRetry({
+            model: this.modelName,
+            input: batch,
+        }));
+        const batchResponses = await Promise.all(batchRequests);
+        const embeddings = [];
+        for (let i = 0; i < batchResponses.length; i += 1) {
+            const batch = batches[i];
+            const { data: batchResponse } = batchResponses[i];
+            for (let j = 0; j < batch.length; j += 1) {
+                embeddings.push(batchResponse[j].embedding);
+            }
+        }
+        return embeddings;
+    }
+    /**
+     * Generates an embedding for a single text.
+     * @param text - A string to generate an embedding for.
+     * @returns A Promise that resolves to an array of numbers representing the embedding.
+     */
+    async embedQuery(text) {
+        const { data } = await this.embeddingWithRetry({
+            model: this.modelName,
+            input: text,
+        });
+        return data[0].embedding;
+    }
+    /**
+     * Makes a request to the Voyage AI API to generate embeddings for an array of texts.
+     * @param request - An object with properties to configure the request.
+     * @returns A Promise that resolves to the response from the Voyage AI API.
+     */
+    async embeddingWithRetry(request) {
+        const makeCompletionRequest = async () => {
+            const url = `${this.apiUrl}`;
+            const response = await fetch(url, {
+                method: "POST",
+                headers: {
+                    "Content-Type": "application/json",
+                    Authorization: `Bearer ${this.apiKey}`,
+                    ...this.headers,
+                },
+                body: JSON.stringify(request),
+            });
+            const json = await response.json();
+            return json;
+        };
+        return this.caller.call(makeCompletionRequest);
+    }
+}

package/dist/experimental/autogpt/prompt.cjs

@@ -81,6 +81,10 @@ class AutoGPTPrompt extends chat_js_1.BaseChatPromptTemplate {
     async formatMessages({ goals, memory, messages: previousMessages, user_input, }) {
         const basePrompt = new index_js_1.SystemMessage(this.constructFullPrompt(goals));
         const timePrompt = new index_js_1.SystemMessage(`The current time and date is ${new Date().toLocaleString()}`);
+        if (typeof basePrompt.content !== "string" ||
+            typeof timePrompt.content !== "string") {
+            throw new Error("Non-string message content is not supported.");
+        }
         const usedTokens = (await this.tokenCounter(basePrompt.content)) +
             (await this.tokenCounter(timePrompt.content));
         const relevantDocs = await memory.getRelevantDocuments(JSON.stringify(previousMessages.slice(-10)));
@@ -92,9 +96,15 @@ class AutoGPTPrompt extends chat_js_1.BaseChatPromptTemplate {
         }
         const contentFormat = `This reminds you of these events from your past:\n${relevantMemory.join("\n")}\n\n`;
         const memoryMessage = new index_js_1.SystemMessage(contentFormat);
+        if (typeof memoryMessage.content !== "string") {
+            throw new Error("Non-string message content is not supported.");
+        }
         const usedTokensWithMemory = (await usedTokens) + (await this.tokenCounter(memoryMessage.content));
         const historicalMessages = [];
         for (const message of previousMessages.slice(-10).reverse()) {
+            if (typeof message.content !== "string") {
+                throw new Error("Non-string message content is not supported.");
+            }
             const messageTokens = await this.tokenCounter(message.content);
             if (usedTokensWithMemory + messageTokens > this.sendTokenLimit - 1000) {
                 break;

package/dist/experimental/autogpt/prompt.js

@@ -78,6 +78,10 @@ export class AutoGPTPrompt extends BaseChatPromptTemplate {
     async formatMessages({ goals, memory, messages: previousMessages, user_input, }) {
         const basePrompt = new SystemMessage(this.constructFullPrompt(goals));
         const timePrompt = new SystemMessage(`The current time and date is ${new Date().toLocaleString()}`);
+        if (typeof basePrompt.content !== "string" ||
+            typeof timePrompt.content !== "string") {
+            throw new Error("Non-string message content is not supported.");
+        }
         const usedTokens = (await this.tokenCounter(basePrompt.content)) +
             (await this.tokenCounter(timePrompt.content));
         const relevantDocs = await memory.getRelevantDocuments(JSON.stringify(previousMessages.slice(-10)));
@@ -89,9 +93,15 @@ export class AutoGPTPrompt extends BaseChatPromptTemplate {
         }
         const contentFormat = `This reminds you of these events from your past:\n${relevantMemory.join("\n")}\n\n`;
         const memoryMessage = new SystemMessage(contentFormat);
+        if (typeof memoryMessage.content !== "string") {
+            throw new Error("Non-string message content is not supported.");
+        }
         const usedTokensWithMemory = (await usedTokens) + (await this.tokenCounter(memoryMessage.content));
         const historicalMessages = [];
         for (const message of previousMessages.slice(-10).reverse()) {
+            if (typeof message.content !== "string") {
+                throw new Error("Non-string message content is not supported.");
+            }
             const messageTokens = await this.tokenCounter(message.content);
             if (usedTokensWithMemory + messageTokens > this.sendTokenLimit - 1000) {
                 break;

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.cjs

@@ -247,6 +247,9 @@ class ViolationOfExpectationsChain extends base_js_1.BaseChain {
             function_call: { name: types_js_1.PREDICTION_VIOLATIONS_FUNCTION.name },
         });
         const chain = violation_of_expectations_prompt_js_1.PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(this.jsonOutputParser);
+        if (typeof userResponse?.content !== "string") {
+            throw new Error("This chain does not support non-string model output.");
+        }
         const res = (await chain.invoke({
             predicted_output: userPredictions.predictedUserMessage,
             actual_output: userResponse?.content ?? "",
@@ -299,6 +302,9 @@ class ViolationOfExpectationsChain extends base_js_1.BaseChain {
      */
     async generateFacts({ userResponse, predictions, runManager, }) {
         const chain = violation_of_expectations_prompt_js_1.GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(this.stringOutputParser);
+        if (typeof userResponse?.content !== "string") {
+            throw new Error("This chain does not support non-string model output.");
+        }
         const res = await chain.invoke({
             prediction_violations: predictions.explainedPredictionErrors.join("\n"),
             prediction: predictions.revisedPrediction,

package/dist/experimental/chains/violation_of_expectations/violation_of_expectations_chain.js

@@ -244,6 +244,9 @@ export class ViolationOfExpectationsChain extends BaseChain {
             function_call: { name: PREDICTION_VIOLATIONS_FUNCTION.name },
         });
         const chain = PREDICTION_VIOLATIONS_PROMPT.pipe(llmWithFunctions).pipe(this.jsonOutputParser);
+        if (typeof userResponse?.content !== "string") {
+            throw new Error("This chain does not support non-string model output.");
+        }
         const res = (await chain.invoke({
             predicted_output: userPredictions.predictedUserMessage,
             actual_output: userResponse?.content ?? "",
@@ -296,6 +299,9 @@ export class ViolationOfExpectationsChain extends BaseChain {
      */
     async generateFacts({ userResponse, predictions, runManager, }) {
         const chain = GENERATE_FACTS_PROMPT.pipe(this.llm).pipe(this.stringOutputParser);
+        if (typeof userResponse?.content !== "string") {
+            throw new Error("This chain does not support non-string model output.");
+        }
         const res = await chain.invoke({
             prediction_violations: predictions.explainedPredictionErrors.join("\n"),
             prediction: predictions.revisedPrediction,

package/dist/experimental/chat_models/anthropic_functions.cjs

@@ -112,6 +112,9 @@ class AnthropicFunctions extends base_js_1.BaseChatModel {
         }
         const chatResult = await this.llm._generate(promptMessages, options, runManager);
         const chatGenerationContent = chatResult.generations[0].message.content;
+        if (typeof chatGenerationContent !== "string") {
+            throw new Error("AnthropicFunctions does not support non-string output.");
+        }
         if (forced) {
             const parser = new fast_xml_parser_1.XMLParser();
             const result = parser.parse(`${chatGenerationContent}</tool_input>`);

package/dist/experimental/chat_models/anthropic_functions.js

@@ -109,6 +109,9 @@ export class AnthropicFunctions extends BaseChatModel {
         }
         const chatResult = await this.llm._generate(promptMessages, options, runManager);
         const chatGenerationContent = chatResult.generations[0].message.content;
+        if (typeof chatGenerationContent !== "string") {
+            throw new Error("AnthropicFunctions does not support non-string output.");
+        }
         if (forced) {
             const parser = new XMLParser();
             const result = parser.parse(`${chatGenerationContent}</tool_input>`);

package/dist/experimental/chat_models/bittensor.cjs

@@ -60,10 +60,15 @@ class NIBittensorChatModel extends base_js_1.BaseChatModel {
       const res = await chat.call([message]);
      */
     async _generate(messages) {
-        const processed_messages = messages.map((message) => ({
-            role: this.messageToOpenAIRole(message),
-            content: message.content,
-        }));
+        const processed_messages = messages.map((message) => {
+            if (typeof message.content !== "string") {
+                throw new Error("NIBittensorChat does not support non-string output.");
+            }
+            return {
+                role: this.messageToOpenAIRole(message),
+                content: message.content,
+            };
+        });
         const generations = [];
         try {
             // Retrieve API KEY

package/dist/experimental/chat_models/bittensor.js

@@ -57,10 +57,15 @@ export class NIBittensorChatModel extends BaseChatModel {
       const res = await chat.call([message]);
      */
     async _generate(messages) {
-        const processed_messages = messages.map((message) => ({
-            role: this.messageToOpenAIRole(message),
-            content: message.content,
-        }));
+        const processed_messages = messages.map((message) => {
+            if (typeof message.content !== "string") {
+                throw new Error("NIBittensorChat does not support non-string output.");
+            }
+            return {
+                role: this.messageToOpenAIRole(message),
+                content: message.content,
+            };
+        });
         const generations = [];
         try {
             // Retrieve API KEY