From b43a22fbee40cac1bf73ebeeeab8e56a96c673da Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sat, 2 Dec 2023 14:00:56 -0300 Subject: [PATCH] Refactor langchain imports and update llm_type_to_cls_dict --- .../langflow/interface/custom_lists.py | 29 ++++++++----------- 1 file changed, 12 insertions(+), 17 deletions(-) diff --git a/src/backend/langflow/interface/custom_lists.py b/src/backend/langflow/interface/custom_lists.py index 9ad34edc2..f77bacbbd 100644 --- a/src/backend/langflow/interface/custom_lists.py +++ b/src/backend/langflow/interface/custom_lists.py @@ -1,28 +1,23 @@ import inspect from typing import Any -from langchain import ( - document_loaders, - embeddings, - llms, - memory, - requests, - text_splitter, -) +from langchain import (document_loaders, embeddings, llms, memory, requests, + text_splitter) from langchain.agents import agent_toolkits -from langchain.chat_models import ( - AzureChatOpenAI, - ChatOpenAI, - ChatVertexAI, - ChatAnthropic, -) - -from langflow.interface.importing.utils import import_class +from langchain.chat_models import (AzureChatOpenAI, ChatAnthropic, ChatOpenAI, + ChatVertexAI) from langflow.interface.agents.custom import CUSTOM_AGENTS from langflow.interface.chains.custom import CUSTOM_CHAINS +from langflow.interface.importing.utils import import_class # LLMs -llm_type_to_cls_dict = llms.type_to_cls_dict +llm_type_to_cls_dict = {} + +for k, v in llms.get_type_to_cls_dict().items(): + try: + llm_type_to_cls_dict[k] = v() + except Exception: + pass llm_type_to_cls_dict["anthropic-chat"] = ChatAnthropic # type: ignore llm_type_to_cls_dict["azure-chat"] = AzureChatOpenAI # type: ignore llm_type_to_cls_dict["openai-chat"] = ChatOpenAI # type: ignore