11package com.embabel.hub.integrations
22
3+ import com.embabel.agent.api.models.AnthropicModels
4+ import com.embabel.agent.api.models.DeepSeekModels
5+ import com.embabel.agent.api.models.OpenAiModels
6+
37/* *
48 * Supported LLM providers for BYOK (Bring Your Own Key).
59 * Each provider defines default models for each role in the Guide app.
@@ -12,18 +16,18 @@ enum class LlmProvider(
1216 val validationModel : String = chatModel,
1317) {
1418 OPENAI (
15- chatModel = " gpt-4.1 " ,
16- classifierModel = " gpt-4.1-mini " ,
17- narratorModel = " gpt-4.1-mini " ,
18- summarizerModel = " gpt-4.1-nano " ,
19- 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 ,
2024 ),
2125 ANTHROPIC (
22- chatModel = " claude-sonnet-4-6 " ,
23- classifierModel = " claude-haiku-4-5 " ,
24- narratorModel = " claude-haiku-4-5 " ,
25- summarizerModel = " claude-haiku-4-5 " ,
26- 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 ,
2731 ),
2832 MISTRAL (
2933 chatModel = " mistral-large-latest" ,
@@ -33,10 +37,10 @@ enum class LlmProvider(
3337 validationModel = " mistral-small-latest" ,
3438 ),
3539 DEEPSEEK (
36- chatModel = " deepseek-chat " ,
37- classifierModel = " deepseek-chat " ,
38- narratorModel = " deepseek-chat " ,
39- summarizerModel = " deepseek-chat " ,
40+ chatModel = DeepSeekModels . DEEPSEEK_V4_PRO ,
41+ classifierModel = DeepSeekModels . DEEPSEEK_V4_FLASH ,
42+ narratorModel = DeepSeekModels . DEEPSEEK_V4_FLASH ,
43+ summarizerModel = DeepSeekModels . DEEPSEEK_V4_FLASH ,
4044 ),
4145
4246}
0 commit comments