Skip to content

Commit e7e04f9

Browse files
authored
Merge pull request llmware-ai#1181 from RS-labhub/modelCatalog
feat && fix: added new models to modelCatalog, updated .gitignore to ignore .env files
2 parents 18d3b04 + 4dfb925 commit e7e04f9

File tree

3 files changed

+12
-1
lines changed

3 files changed

+12
-1
lines changed

‎.gitignore‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -14,3 +14,6 @@ docs/_site/
1414

1515
# Mac
1616
.DS_Store/
17+
18+
# Testing
19+
.env

‎llmware/model_configs.py‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -427,6 +427,14 @@
427427
{"model_name": 'claude-2.0', "display_name": "Anthropic Claude-Claude2-.0",
428428
"model_family": "ClaudeModel", "model_category": "generative-api", "model_location": "api", "context_window": 8192},
429429

430+
{"model_name": 'claude-3-haiku-20240307', "display_name": "Anthropic Claude 3 Haiku", "model_family": "ClaudeModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
431+
432+
{"model_name": 'claude-3-5-haiku-20241022', "display_name": "Anthropic Claude 3.5 Haiku", "model_family": "ClaudeModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
433+
434+
{"model_name": 'claude-3-5-sonnet-20240620', "display_name": "Anthropic Claude 3.5 Sonnet", "model_family": "ClaudeModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
435+
436+
{"model_name": 'claude-3-7-sonnet-20250219', "display_name": "Anthropic Claude 3.7 Sonnet", "model_family": "ClaudeModel", "model_category": "generative-api", "model_location": "api", "context_window": 200000},
437+
430438
{"model_name": 'command-medium-nightly', "display_name": "Cohere Command Medium", "model_family": "CohereGenModel",
431439
"model_category": "generative-api","model_location": "api", "context_window": 2048},
432440

‎llmware/models.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6033,7 +6033,7 @@ def inference(self, prompt, add_context=None, add_prompt_engineering=None, infer
60336033
# new Claude 3 models use the 'messages' API
60346034
# please check that you have pip installed the latest anthropic python sdk
60356035

6036-
if self.model_name in ["claude-3-opus-20240229", "claude-3-sonnet-20240229"]:
6036+
if self.model_name in ["claude-3-opus-20240229", "claude-3-sonnet-20240229","claude-3-haiku-20240307", "claude-3-5-haiku-20241022", "claude-3-5-sonnet-20240620", "claude-3-7-sonnet-20250219"]:
60376037

60386038
# use messages API
60396039
message = client.messages.create(model=self.model_name, max_tokens=self.target_requested_output_tokens,

0 commit comments

Comments
 (0)