diff --git a/langflow/backend/signature.py b/langflow/backend/signature.py index 9d32a837a..768afa20d 100644 --- a/langflow/backend/signature.py +++ b/langflow/backend/signature.py @@ -1,3 +1,4 @@ +from typing import Any, Dict from fastapi import APIRouter, HTTPException from langchain import agents, chains, llms, prompts from langchain.agents.load_tools import ( @@ -131,11 +132,13 @@ def get_tool(name: str): elif tool_type in _EXTRA_OPTIONAL_TOOLS: _, extra_keys = _EXTRA_OPTIONAL_TOOLS[tool_type] params = extra_keys + else: + params = [] template = { param: (type_dict[param] if param == "llm" else type_dict["str"]) for param in params - } + } # type: Dict[str, Any] template["_type"] = tool_type return { diff --git a/langflow/backend/util.py b/langflow/backend/util.py index 2474bf322..9861f5bc2 100644 --- a/langflow/backend/util.py +++ b/langflow/backend/util.py @@ -113,8 +113,8 @@ def get_default_factory(module: str, function: str): pattern = r"" if match := re.search(pattern, function): - module = importlib.import_module(module) - return getattr(module, match[1])() + imported_module = importlib.import_module(module) + return getattr(imported_module, match[1])() return None