diff --git a/src/backend/langflow/components/llms/ChatLiteLLM.py b/src/backend/langflow/components/llms/ChatLiteLLM.py index 01b831bae..788b552ce 100644 --- a/src/backend/langflow/components/llms/ChatLiteLLM.py +++ b/src/backend/langflow/components/llms/ChatLiteLLM.py @@ -6,7 +6,7 @@ from langchain_community.chat_models.litellm import ChatLiteLLM, ChatLiteLLMExce import os -class LiteLLMComponent(CustomComponent): +class ChatLiteLLMComponent(CustomComponent): display_name = "ChatLiteLLM" description = "`LiteLLM` collection of large language models." documentation = "https://python.langchain.com/docs/integrations/chat/litellm"