From 270778e624b23df7414f90d596c3f81b061a43e7 Mon Sep 17 00:00:00 2001 From: Ravi Theja <ravi03071991@gmail.com> Date: Thu, 14 Mar 2024 02:56:02 +0530 Subject: [PATCH] Add Anthropic haiku model (#11916) --- .../llama_index/llms/anthropic/utils.py | 1 + .../llms/llama-index-llms-anthropic/pyproject.toml | 4 ++-- .../llama_index/multi_modal_llms/anthropic/utils.py | 1 + .../llama-index-multi-modal-llms-anthropic/pyproject.toml | 4 ++-- 4 files changed, 6 insertions(+), 4 deletions(-) diff --git a/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py b/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py index 19c1ae5dc7..c7056d1084 100644 --- a/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py +++ b/llama-index-integrations/llms/llama-index-llms-anthropic/llama_index/llms/anthropic/utils.py @@ -13,6 +13,7 @@ CLAUDE_MODELS: Dict[str, int] = { "claude-2.1": 200000, "claude-3-opus-20240229": 180000, "claude-3-sonnet-20240229": 180000, + "claude-3-haiku-20240307": 180000, } diff --git a/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml b/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml index 543b8789f7..0122456f04 100644 --- a/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml +++ b/llama-index-integrations/llms/llama-index-llms-anthropic/pyproject.toml @@ -27,12 +27,12 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-llms-anthropic" readme = "README.md" -version = "0.1.5" +version = "0.1.6" [tool.poetry.dependencies] python = ">=3.8.1,<4.0" llama-index-core = "^0.10.1" -anthropic = "^0.17.0" +anthropic = "^0.20.0" [tool.poetry.group.dev.dependencies] ipython = "8.10.0" diff --git a/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/llama_index/multi_modal_llms/anthropic/utils.py b/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/llama_index/multi_modal_llms/anthropic/utils.py index 8ee491b079..0ab1930266 100644 --- a/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/llama_index/multi_modal_llms/anthropic/utils.py +++ b/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/llama_index/multi_modal_llms/anthropic/utils.py @@ -15,6 +15,7 @@ DEFAULT_ANTHROPIC_API_VERSION = "" ANTHROPIC_MULTI_MODAL_MODELS = { "claude-3-opus-20240229": 180000, "claude-3-sonnet-20240229": 180000, + "claude-3-haiku-20240307": 180000, } diff --git a/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/pyproject.toml b/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/pyproject.toml index 50bee985a7..d367dada58 100644 --- a/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/pyproject.toml +++ b/llama-index-integrations/multi_modal_llms/llama-index-multi-modal-llms-anthropic/pyproject.toml @@ -27,12 +27,12 @@ exclude = ["**/BUILD"] license = "MIT" name = "llama-index-multi-modal-llms-anthropic" readme = "README.md" -version = "0.1.2" +version = "0.1.3" [tool.poetry.dependencies] python = ">=3.8.1,<4.0" llama-index-core = "^0.10.1" -anthropic = "0.17.0" +anthropic = "0.20.0" [tool.poetry.group.dev.dependencies] ipython = "8.10.0" -- GitLab