Added alias for bedrock titan model (#368)

* Added alias for bedrock titan model * [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
jupyterlab · Aug 29, 2023 · 47ae1c9 · 47ae1c9
1 parent 245c1cd
commit 47ae1c9
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py b/packages/jupyter-ai-magics/jupyter_ai_magics/magics.py
@@ -32,6 +32,7 @@
     "gpt3": "openai:text-davinci-003",
     "chatgpt": "openai-chat:gpt-3.5-turbo",
     "gpt4": "openai-chat:gpt-4",
+    "titan": "bedrock:amazon.titan-tg1-large",
 }