Skip to content

Commit 67f5e9d

Browse files
committed
Fix hard-coded model names.
1 parent b5e6774 commit 67f5e9d

1 file changed

Lines changed: 12 additions & 10 deletions

File tree

src/main/kotlin/com/embabel/hub/integrations/LlmProvider.kt

Lines changed: 12 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
package com.embabel.hub.integrations
22

3+
import com.embabel.agent.api.models.AnthropicModels
34
import 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

Comments
 (0)