From b767f274f8563298b607229e4598b4ed92056394 Mon Sep 17 00:00:00 2001
From: drebah <131481923+drebah@users.noreply.github.com>
Date: Mon, 22 Apr 2024 21:01:04 -0700
Subject: [PATCH] Adding claude 3 opus to BedRock integration (#13033)

* Adding claude 3 opus to BedRock integration

* Increment version
---
 .../llama-index-llms-bedrock/llama_index/llms/bedrock/utils.py  | 2 ++
 .../llms/llama-index-llms-bedrock/pyproject.toml                | 2 +-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/llama-index-integrations/llms/llama-index-llms-bedrock/llama_index/llms/bedrock/utils.py b/llama-index-integrations/llms/llama-index-llms-bedrock/llama_index/llms/bedrock/utils.py
index d154284c37..c28da129ef 100644
--- a/llama-index-integrations/llms/llama-index-llms-bedrock/llama_index/llms/bedrock/utils.py
+++ b/llama-index-integrations/llms/llama-index-llms-bedrock/llama_index/llms/bedrock/utils.py
@@ -46,6 +46,7 @@ CHAT_ONLY_MODELS = {
     "anthropic.claude-v2:1": 200000,
     "anthropic.claude-3-sonnet-20240229-v1:0": 200000,
     "anthropic.claude-3-haiku-20240307-v1:0": 200000,
+    "anthropic.claude-3-opus-20240229-v1:0": 200000,
     "meta.llama2-13b-chat-v1": 2048,
     "meta.llama2-70b-chat-v1": 4096,
     "mistral.mistral-7b-instruct-v0:2": 32000,
@@ -66,6 +67,7 @@ STREAMING_MODELS = {
     "anthropic.claude-v2:1",
     "anthropic.claude-3-sonnet-20240229-v1:0",
     "anthropic.claude-3-haiku-20240307-v1:0",
+    "anthropic.claude-3-opus-20240229-v1:0",
     "meta.llama2-13b-chat-v1",
     "mistral.mistral-7b-instruct-v0:2",
     "mistral.mixtral-8x7b-instruct-v0:1",
diff --git a/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml
index a02383f08c..6d980b0c48 100644
--- a/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml
+++ b/llama-index-integrations/llms/llama-index-llms-bedrock/pyproject.toml
@@ -27,7 +27,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-llms-bedrock"
 readme = "README.md"
-version = "0.1.6"
+version = "0.1.7"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"
-- 
GitLab