fix: Update BaseChatModel import checks for MistralAI compatibility (#8527)

Signed-off-by: Oleg Ivaniv <me@olegivaniv.com>
This commit is contained in:
oleg
2024-02-05 16:09:23 +01:00
committed by GitHub
parent b62c1d7c41
commit c8b8379015
5 changed files with 18 additions and 11 deletions

View File

@@ -19,9 +19,10 @@ import {
import type { BaseOutputParser } from 'langchain/schema/output_parser';
import { CombiningOutputParser } from 'langchain/output_parsers';
import { LLMChain } from 'langchain/chains';
import { BaseChatModel } from 'langchain/chat_models/base';
import type { BaseChatModel } from 'langchain/chat_models/base';
import { HumanMessage } from 'langchain/schema';
import { getTemplateNoticeField } from '../../../utils/sharedFields';
import { isChatInstance } from '../../../utils/helpers';
interface MessagesTemplate {
type: string;
@@ -94,7 +95,7 @@ async function getChainPromptTemplate(
partialVariables: formatInstructions ? { formatInstructions } : undefined,
});
if (llm instanceof BaseChatModel) {
if (isChatInstance(llm)) {
const parsedMessages = await Promise.all(
(messages ?? []).map(async (message) => {
const messageClass = [