@@ -721,6 +721,15 @@ class LargeLanguageModels(Enum):
721
721
)
722
722
723
723
# 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
+ )
724
733
claude_4_sonnet = LLMSpec (
725
734
label = "Claude 4 Sonnet • Anthropic" ,
726
735
model_id = "claude-4-sonnet-20250514" ,
@@ -1546,6 +1555,7 @@ def run_openai_chat(
1546
1555
elif model in [
1547
1556
LargeLanguageModels .claude_4_sonnet ,
1548
1557
LargeLanguageModels .claude_4_opus ,
1558
+ LargeLanguageModels .claude_4_1_opus ,
1549
1559
LargeLanguageModels .gemini_2_5_pro ,
1550
1560
LargeLanguageModels .gemini_2_5_flash ,
1551
1561
]:
@@ -1825,7 +1835,7 @@ def get_openai_client(model: str):
1825
1835
max_retries = 0 ,
1826
1836
base_url = "https://api.sarvam.ai/v1" ,
1827
1837
)
1828
- elif model .startswith ("claude-4- " ):
1838
+ elif model .startswith ("claude-" ):
1829
1839
client = openai .OpenAI (
1830
1840
api_key = settings .ANTHROPIC_API_KEY ,
1831
1841
max_retries = 0 ,
0 commit comments