Template
20 lines
604 B
TypeScript
20 lines
604 B
TypeScript
import { createOpenRouter } from '@openrouter/ai-sdk-provider';
|
|
import {
|
|
DEFAULT_OPENROUTER_MODEL_ID,
|
|
isOpenRouterModelId,
|
|
type OpenRouterModelId,
|
|
} from '@/lib/openrouter';
|
|
|
|
export function getDefaultChatModelId(): OpenRouterModelId {
|
|
const configuredModel = process.env.OPENROUTER_MODEL;
|
|
return isOpenRouterModelId(configuredModel) ? configuredModel : DEFAULT_OPENROUTER_MODEL_ID;
|
|
}
|
|
|
|
export function getChatModel(modelId: OpenRouterModelId = getDefaultChatModelId()) {
|
|
const openrouter = createOpenRouter({
|
|
apiKey: process.env.OPENROUTER_API_KEY,
|
|
});
|
|
|
|
return openrouter(modelId);
|
|
}
|