Add OpenAI support, default to gpt-4o-mini

This commit is contained in:
2026-01-27 22:45:06 +00:00
parent e4f900c111
commit 00a07dc95a
4 changed files with 17 additions and 3 deletions

View File

@@ -1,11 +1,19 @@
import { ChatAnthropic } from '@langchain/anthropic';
import { ChatOpenAI } from '@langchain/openai';
import { ChatPromptTemplate } from '@langchain/core/prompts';
import { StringOutputParser } from '@langchain/core/output_parsers';
export type AIProvider = 'anthropic' | 'openai';
// Get model based on provider
function getModel(provider: AIProvider = 'anthropic') {
function getModel(provider: AIProvider = 'openai') {
if (provider === 'openai') {
return new ChatOpenAI({
modelName: 'gpt-4o-mini',
openAIApiKey: process.env.OPENAI_API_KEY,
});
}
if (provider === 'anthropic') {
return new ChatAnthropic({
modelName: 'claude-sonnet-4-20250514',
@@ -13,8 +21,7 @@ function getModel(provider: AIProvider = 'anthropic') {
});
}
// Add OpenAI support later
throw new Error(`Provider ${provider} not yet supported`);
throw new Error(`Provider ${provider} not supported`);
}
// Email generation prompt