Skip to content

Commit

Permalink
apply formatting
Browse files Browse the repository at this point in the history
  • Loading branch information
sihyeong671 committed Aug 15, 2024
1 parent aadd96c commit 2dadca1
Showing 1 changed file with 3 additions and 4 deletions.
7 changes: 3 additions & 4 deletions backend/src/langchain/langchain.module.ts
Original file line number Diff line number Diff line change
Expand Up @@ -7,17 +7,16 @@ const chatModelFactory = {
provide: "ChatModel",
useFactory: () => {
const modelType = process.env.YORKIE_INTELLIGENCE;
if (modelType === "gemma2:2b"){
if (modelType === "gemma2:2b") {
return new ChatOllama({
model: modelType,
checkOrPullModel: true,
streaming: true,
});
}
else if (modelType === "openai"){
} else if (modelType === "openai") {
return new ChatOpenAI({ modelName: "gpt-4o-mini" }) as BaseChatModel;
}
}
},
};

@Module({
Expand Down

0 comments on commit 2dadca1

Please sign in to comment.