diff --git a/langflow/backend/langflow_backend/interface/loading.py b/langflow/backend/langflow_backend/interface/loading.py index 7291135d3..0a1f62012 100644 --- a/langflow/backend/langflow_backend/interface/loading.py +++ b/langflow/backend/langflow_backend/interface/loading.py @@ -92,7 +92,7 @@ def load_agent_executor_from_config( ) -def load_tools_from_config(tool_list: list[dict]) -> list[Tool]: +def load_tools_from_config(tool_list: list[dict]) -> list: """Load tools based on a config list. Args: diff --git a/langflow/backend/langflow_backend/interface/signature.py b/langflow/backend/langflow_backend/interface/signature.py index 247c79789..70bafc3d1 100644 --- a/langflow/backend/langflow_backend/interface/signature.py +++ b/langflow/backend/langflow_backend/interface/signature.py @@ -7,7 +7,6 @@ from langchain.agents.load_tools import ( _LLM_TOOLS, get_all_tool_names, ) -from langchain.chains.conversation import memory as memories from langflow_backend.utils import util from langflow_backend.custom import customs @@ -21,7 +20,6 @@ def get_signature(name: str, object_type: str): "prompts": get_prompt_signature, "llms": get_llm_signature, "tools": get_tool_signature, - "memories": get_memory_signature, }.get(object_type, lambda name: f"Invalid type: {name}")(name) @@ -63,14 +61,6 @@ def get_llm_signature(name: str): raise ValueError("LLM not found") from exc -def get_memory_signature(name: str): - """Get the signature of a memory.""" - try: - return util.build_template_from_class(name, memories.type_to_cls_dict) - except ValueError as exc: - raise ValueError("Memory not found") from exc - - def get_tool_signature(name: str): """Get the signature of a tool."""