Skip to content

Commit 28d5009

Browse files
authored
Added new models (#908)
Added the latest llama-3.1 models available on Amazon Bedrock, and gpt-4o mini. All these models were recently released and have become popular.
1 parent 8ae2491 commit 28d5009

File tree

2 files changed

+3
-0
lines changed

2 files changed

+3
-0
lines changed

packages/jupyter-ai-magics/jupyter_ai_magics/partner_providers/openai.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -46,6 +46,7 @@ class ChatOpenAIProvider(BaseProvider, ChatOpenAI):
4646
"gpt-4-0125-preview",
4747
"gpt-4-1106-preview",
4848
"gpt-4o",
49+
"gpt-4o-mini",
4950
]
5051
model_id_key = "model_name"
5152
pypi_package_deps = ["langchain_openai"]

packages/jupyter-ai-magics/jupyter_ai_magics/providers.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -798,6 +798,8 @@ class BedrockProvider(BaseProvider, Bedrock):
798798
"meta.llama2-70b-chat-v1",
799799
"meta.llama3-8b-instruct-v1:0",
800800
"meta.llama3-70b-instruct-v1:0",
801+
"meta.llama3-1-8b-instruct-v1:0",
802+
"meta.llama3-1-70b-instruct-v1:0",
801803
"mistral.mistral-7b-instruct-v0:2",
802804
"mistral.mixtral-8x7b-instruct-v0:1",
803805
"mistral.mistral-large-2402-v1:0",

0 commit comments

Comments
 (0)