feat(groq): switch to @langchain/groq for better handling
This commit is contained in:
parent
37cd6d3ab5
commit
eadbedb713
3 changed files with 25 additions and 9 deletions
|
|
@ -1,4 +1,4 @@
|
|||
import { ChatOpenAI } from '@langchain/openai';
|
||||
import { ChatGroq } from '@langchain/groq';
|
||||
import { getGroqApiKey } from '../config';
|
||||
import { ChatModel } from '.';
|
||||
|
||||
|
|
@ -28,16 +28,10 @@ export const loadGroqChatModels = async () => {
|
|||
groqChatModels.forEach((model: any) => {
|
||||
chatModels[model.id] = {
|
||||
displayName: model.id,
|
||||
model: new ChatOpenAI({
|
||||
model: new ChatGroq({
|
||||
apiKey: groqApiKey,
|
||||
modelName: model.id,
|
||||
model: model.id,
|
||||
temperature: 0.7,
|
||||
configuration: {
|
||||
baseURL: 'https://api.groq.com/openai/v1',
|
||||
},
|
||||
metadata: {
|
||||
'model-type': 'groq',
|
||||
},
|
||||
}) as unknown as BaseChatModel,
|
||||
};
|
||||
});
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue