Skip to content

Commit 6cc85fa

Browse files
committed
feat: add claude 4.1 opus support
1 parent 5e669d2 commit 6cc85fa

File tree

1 file changed

+11
-1
lines changed

1 file changed

+11
-1
lines changed

daras_ai_v2/language_model.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -721,6 +721,15 @@ class LargeLanguageModels(Enum):
721721
)
722722

723723
# https://docs.anthropic.com/claude/docs/models-overview#model-comparison
724+
claude_4_1_opus = LLMSpec(
725+
label="Claude 4.1 Opus • Anthropic",
726+
model_id="claude-opus-4-1",
727+
llm_api=LLMApis.openai,
728+
context_window=200_000,
729+
max_output_tokens=32_000,
730+
is_vision_model=True,
731+
supports_json=True,
732+
)
724733
claude_4_sonnet = LLMSpec(
725734
label="Claude 4 Sonnet • Anthropic",
726735
model_id="claude-4-sonnet-20250514",
@@ -1546,6 +1555,7 @@ def run_openai_chat(
15461555
elif model in [
15471556
LargeLanguageModels.claude_4_sonnet,
15481557
LargeLanguageModels.claude_4_opus,
1558+
LargeLanguageModels.claude_4_1_opus,
15491559
LargeLanguageModels.gemini_2_5_pro,
15501560
LargeLanguageModels.gemini_2_5_flash,
15511561
]:
@@ -1825,7 +1835,7 @@ def get_openai_client(model: str):
18251835
max_retries=0,
18261836
base_url="https://api.sarvam.ai/v1",
18271837
)
1828-
elif model.startswith("claude-4-"):
1838+
elif model.startswith("claude-"):
18291839
client = openai.OpenAI(
18301840
api_key=settings.ANTHROPIC_API_KEY,
18311841
max_retries=0,

0 commit comments

Comments
 (0)