11package com.embabel.hub.integrations
22
3+ import com.embabel.agent.api.models.AnthropicModels
34import com.embabel.agent.api.models.DeepSeekModels
5+ import com.embabel.agent.api.models.OpenAiModels
46
57/* *
68 * Supported LLM providers for BYOK (Bring Your Own Key).
@@ -14,18 +16,18 @@ enum class LlmProvider(
1416 val validationModel : String = chatModel,
1517) {
1618 OPENAI (
17- chatModel = " gpt-4.1 " ,
18- classifierModel = " gpt-4.1-mini " ,
19- narratorModel = " gpt-4.1-mini " ,
20- summarizerModel = " gpt-4.1-nano " ,
21- validationModel = " gpt-4.1-nano " ,
19+ chatModel = OpenAiModels . GPT_41 ,
20+ classifierModel = OpenAiModels . GPT_41_MINI ,
21+ narratorModel = OpenAiModels . GPT_41_MINI ,
22+ summarizerModel = OpenAiModels . GPT_41_NANO ,
23+ validationModel = OpenAiModels . GPT_41_NANO ,
2224 ),
2325 ANTHROPIC (
24- chatModel = " claude-sonnet-4-6 " ,
25- classifierModel = " claude-haiku-4-5 " ,
26- narratorModel = " claude-haiku-4-5 " ,
27- summarizerModel = " claude-haiku-4-5 " ,
28- validationModel = " claude-haiku-4-5 " ,
26+ chatModel = AnthropicModels . CLAUDE_SONNET_4_6 ,
27+ classifierModel = AnthropicModels . CLAUDE_HAIKU_4_5 ,
28+ narratorModel = AnthropicModels . CLAUDE_HAIKU_4_5 ,
29+ summarizerModel = AnthropicModels . CLAUDE_HAIKU_4_5 ,
30+ validationModel = AnthropicModels . CLAUDE_HAIKU_4_5 ,
2931 ),
3032 MISTRAL (
3133 chatModel = " mistral-large-latest" ,
0 commit comments