diff --git a/litellm/__pycache__/utils.cpython-311.pyc b/litellm/__pycache__/utils.cpython-311.pyc index d8e7e219c..6e7724d5f 100644 Binary files a/litellm/__pycache__/utils.cpython-311.pyc and b/litellm/__pycache__/utils.cpython-311.pyc differ diff --git a/litellm/utils.py b/litellm/utils.py index 83c0e05f9..1b8b1b4f5 100644 --- a/litellm/utils.py +++ b/litellm/utils.py @@ -947,6 +947,9 @@ def get_llm_provider(model: str, custom_llm_provider: Optional[str] = None): ## openai - chatcompletion + text completion if model in litellm.open_ai_chat_completion_models or model in litellm.open_ai_text_completion_models: custom_llm_provider = "openai" + ## anthropic + elif model in litellm.anthropic_models: + custom_llm_provider = "anthropic" ## cohere elif model in litellm.cohere_models: custom_llm_provider = "cohere" diff --git a/pyproject.toml b/pyproject.toml index 876d9a605..716b62615 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -1,6 +1,6 @@ [tool.poetry] name = "litellm" -version = "0.1.603" +version = "0.1.604" description = "Library to easily interface with LLM API providers" authors = ["BerriAI"] license = "MIT License"