Merge pull request #157 from Kenza-AI/feat/claude-3

updated docs and commands with claude 3
Kenza-AI · Mar 7, 2024 · 4694093 · 4694093
2 parents 272188a + 5ddf4cf
commit 4694093
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 7 deletions.
diff --git a/docs/index.md b/docs/index.md
@@ -135,9 +135,11 @@ The following models are offered for chat completions:
 
 | Model Name | URL |
 |:------------:|:-----:|
-|claude-2.1|https://docs.anthropic.com/claude/reference/models|
-|claude-2.0|https://docs.anthropic.com/claude/reference/models|
-|claude-instant-1.2|https://docs.anthropic.com/claude/reference/models|
+|claude-2.1|https://docs.anthropic.com/claude/docs/models-overview|
+|claude-2.0|https://docs.anthropic.com/claude/docs/models-overview|
+|claude-instant-1.2|https://docs.anthropic.com/claude/docs/models-overview|
+|claude-3-opus|https://docs.anthropic.com/claude/docs/models-overview|
+|claude-3-sonnet|https://docs.anthropic.com/claude/docs/models-overview|
 
 
 #### Open-Source
@@ -193,7 +195,7 @@ You need to define the following env variables before you start the LLM Gateway
 You need to define the following env variables before you start the LLM Gateway server:
 
 - `ANTHROPIC_API_KEY`: Your OpenAI API key. Example: `export ANTHROPIC_API_KEY=...`.
-- `ANTHROPIC_CHAT_COMPLETIONS_MODEL`: It should have one of values [here](https://docs.anthropic.com/claude/reference/models). Example `export ANTHROPIC_CHAT_COMPLETIONS_MODEL=claude-2.1`
+- `ANTHROPIC_CHAT_COMPLETIONS_MODEL`: It should have one of values [here](https://docs.anthropic.com/claude/docs/models-overview). Example `export ANTHROPIC_CHAT_COMPLETIONS_MODEL=claude-2.1`
 
 ### Set up open-source LLMs
 

diff --git a/sagify/commands/llm.py b/sagify/commands/llm.py
@@ -29,8 +29,8 @@
 OPENAI_URL = f'{OPENAI_BASE_URL}/{OPENAI_DOCS}/{OPENAI_MODELS}'
 
 ANTHROPIC_BASE_URL = 'https://docs.anthropic.com'
-ANTHROPIC_DOCS = 'claude/reference'
-ANTHROPIC_MODELS = 'models'
+ANTHROPIC_DOCS = 'claude/docs'
+ANTHROPIC_MODELS = 'models-overview'
 ANTHROPIC_URL = f'{ANTHROPIC_BASE_URL}/{ANTHROPIC_DOCS}/{ANTHROPIC_MODELS}'
 
 HF_BASE_URL = 'https://huggingface.co'
@@ -59,7 +59,9 @@
     'anthropic': {
         'claude-2.1': ('claude-2.1', ANTHROPIC_URL),
         'claude-2.0': ('claude-2.0', ANTHROPIC_URL),
-        'claude-instant-1.2': ('claude-instant-1.2', ANTHROPIC_URL)
+        'claude-instant-1.2': ('claude-instant-1.2', ANTHROPIC_URL),
+        'claude-3-opus': ('claude-3-opus-20240229', ANTHROPIC_URL),
+        'claude-3-sonnet': ('claude-3-sonnet-20240229', ANTHROPIC_URL)
     }
 }