From 9e95699277fe0db3bfee1654276a43bfba9ecc64 Mon Sep 17 00:00:00 2001 From: Jeremi Joslin Date: Wed, 24 Jan 2024 12:42:29 +0700 Subject: [PATCH] community[patch]: Fix error message when litellm is not installed (#16316) The error message was mentioning the wrong package. I updated it to the correct one. --- libs/community/langchain_community/chat_models/litellm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/libs/community/langchain_community/chat_models/litellm.py b/libs/community/langchain_community/chat_models/litellm.py index cb6a2107d3..39f7284c3b 100644 --- a/libs/community/langchain_community/chat_models/litellm.py +++ b/libs/community/langchain_community/chat_models/litellm.py @@ -246,8 +246,8 @@ class ChatLiteLLM(BaseChatModel): import litellm except ImportError: raise ChatLiteLLMException( - "Could not import google.generativeai python package. " - "Please install it with `pip install google-generativeai`" + "Could not import litellm python package. " + "Please install it with `pip install litellm`" ) values["openai_api_key"] = get_from_dict_or_env(