jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Import litellm package and update ChatLiteLLMComponent class

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-02-19 13:29:32 -03:00
commit 451f176fb4
1 changed files with 6 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

11
src/backend/langflow/components/llms/ChatLiteLLM.py
View file

@ -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 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): class ChatLiteLLMComponent(CustomComponent):
@ -107,7 +108,7 @@ class ChatLiteLLMComponent(CustomComponent):
verbose: bool = False, verbose: bool = False,
) -> Union[BaseLanguageModel, Callable]: ) -> Union[BaseLanguageModel, Callable]:
try: try:
import litellm import litellm # type: ignore
litellm.drop_params = True litellm.drop_params = True
litellm.set_verbose = verbose litellm.set_verbose = verbose