From dfc733be7d587c6ce78cc076b439ea13ccfa1653 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 27 Mar 2024 16:43:30 -0300 Subject: [PATCH] Add llm_creator and prompt_creator to VertexTypesDict --- src/backend/base/langflow/graph/graph/constants.py | 6 ++++-- src/backend/base/langflow/interface/initialize/loading.py | 2 ++ src/backend/base/langflow/interface/llms/base.py | 4 +--- 3 files changed, 7 insertions(+), 5 deletions(-) diff --git a/src/backend/base/langflow/graph/graph/constants.py b/src/backend/base/langflow/graph/graph/constants.py index 10c10ab07..658c3c68a 100644 --- a/src/backend/base/langflow/graph/graph/constants.py +++ b/src/backend/base/langflow/graph/graph/constants.py @@ -3,8 +3,10 @@ from langflow.interface.agents.base import agent_creator from langflow.interface.custom.base import custom_component_creator from langflow.interface.document_loaders.base import documentloader_creator from langflow.interface.embeddings.base import embedding_creator +from langflow.interface.llms.base import llm_creator from langflow.interface.memories.base import memory_creator from langflow.interface.output_parsers.base import output_parser_creator +from langflow.interface.prompts.base import prompt_creator from langflow.interface.retrievers.base import retriever_creator from langflow.interface.text_splitters.base import textsplitter_creator from langflow.interface.toolkits.base import toolkits_creator @@ -33,13 +35,13 @@ class VertexTypesDict(LazyLoadDictBase): def get_type_dict(self): return { - # **{t: types.PromptVertex for t in prompt_creator.to_list()}, + **{t: types.PromptVertex for t in prompt_creator.to_list()}, **{t: types.AgentVertex for t in agent_creator.to_list()}, # **{t: types.ChainVertex for t in chain_creator.to_list()}, **{t: types.ToolVertex for t in tool_creator.to_list()}, **{t: types.ToolkitVertex for t in toolkits_creator.to_list()}, **{t: types.WrapperVertex for t in wrapper_creator.to_list()}, - # **{t: types.LLMVertex for t in llm_creator.to_list()}, + **{t: types.LLMVertex for t in llm_creator.to_list()}, **{t: types.MemoryVertex for t in memory_creator.to_list()}, **{t: types.EmbeddingVertex for t in embedding_creator.to_list()}, # **{t: types.VectorStoreVertex for t in vectorstore_creator.to_list()}, diff --git a/src/backend/base/langflow/interface/initialize/loading.py b/src/backend/base/langflow/interface/initialize/loading.py index 08720db74..79fc60a80 100644 --- a/src/backend/base/langflow/interface/initialize/loading.py +++ b/src/backend/base/langflow/interface/initialize/loading.py @@ -168,6 +168,8 @@ async def instantiate_custom_component(params, user_id, vertex): custom_repr = custom_component.custom_repr() if not custom_repr and isinstance(build_result, (dict, Record, str)): custom_repr = build_result + if not isinstance(custom_repr, str): + custom_repr = str(custom_repr) return custom_component, build_result, {"repr": custom_repr} diff --git a/src/backend/base/langflow/interface/llms/base.py b/src/backend/base/langflow/interface/llms/base.py index ba611da8d..1eacade6b 100644 --- a/src/backend/base/langflow/interface/llms/base.py +++ b/src/backend/base/langflow/interface/llms/base.py @@ -4,7 +4,6 @@ from loguru import logger from langflow.interface.base import LangChainTypeCreator from langflow.interface.custom_lists import llm_type_to_cls_dict -from langflow.services.deps import get_settings_service from langflow.template.frontend_node.llms import LLMFrontendNode from langflow.utils.util import build_template_from_class @@ -34,11 +33,10 @@ class LLMCreator(LangChainTypeCreator): return None def to_list(self) -> List[str]: - settings_service = get_settings_service() return [ llm.__name__ for llm in self.type_to_loader_dict.values() - if llm.__name__ in settings_service.settings.LLMS or settings_service.settings.DEV + # if llm.__name__ in settings_service.settings.LLMS or settings_service.settings.DEV ]