feat(agents): replace LLMs with chat LLMs

This commit is contained in:
ItzCrazyKns 2024-04-18 18:15:17 +05:30
parent f9ab543bcf
commit dd1ce4e324
No known key found for this signature in database
GPG key ID: 8162927C7CCE3065
7 changed files with 19 additions and 44 deletions

View file

@ -9,7 +9,7 @@ import {
RunnableMap,
RunnableLambda,
} from '@langchain/core/runnables';
import { ChatOpenAI, OpenAI, OpenAIEmbeddings } from '@langchain/openai';
import { ChatOpenAI, OpenAIEmbeddings } from '@langchain/openai';
import { StringOutputParser } from '@langchain/core/output_parsers';
import { Document } from '@langchain/core/documents';
import { searchSearxng } from '../core/searxng';
@ -18,16 +18,11 @@ import formatChatHistoryAsString from '../utils/formatHistory';
import eventEmitter from 'events';
import computeSimilarity from '../utils/computeSimilarity';
const chatLLM = new ChatOpenAI({
const llm = new ChatOpenAI({
modelName: process.env.MODEL_NAME,
temperature: 0.7,
});
const llm = new OpenAI({
temperature: 0,
modelName: process.env.MODEL_NAME,
});
const embeddings = new OpenAIEmbeddings({
modelName: 'text-embedding-3-large',
});
@ -211,7 +206,7 @@ const basicRedditSearchAnsweringChain = RunnableSequence.from([
new MessagesPlaceholder('chat_history'),
['user', '{query}'],
]),
chatLLM,
llm,
strParser,
]).withConfig({
runName: 'FinalResponseGenerator',