diff --git a/src/backend/langflow/components/llms/ChatLiteLLM.py b/src/backend/langflow/components/llms/ChatLiteLLM.py index 2d2113163..f2b7f7dc3 100644 --- a/src/backend/langflow/components/llms/ChatLiteLLM.py +++ b/src/backend/langflow/components/llms/ChatLiteLLM.py @@ -1,8 +1,9 @@ -from langflow import CustomComponent -from typing import Optional, Union, Callable, Any, Dict -from langflow.field_typing import BaseLanguageModel -from langchain_community.chat_models.litellm import ChatLiteLLM, ChatLiteLLMException import os +from typing import Any, Callable, Dict, Optional, Union + +from langchain_community.chat_models.litellm import ChatLiteLLM, ChatLiteLLMException +from langflow import CustomComponent +from langflow.field_typing import BaseLanguageModel class ChatLiteLLMComponent(CustomComponent): @@ -107,7 +108,7 @@ class ChatLiteLLMComponent(CustomComponent): verbose: bool = False, ) -> Union[BaseLanguageModel, Callable]: try: - import litellm + import litellm # type: ignore litellm.drop_params = True litellm.set_verbose = verbose