From 66e4b33b7c966d9a073699f5d37b6c007dd34ce9 Mon Sep 17 00:00:00 2001 From: Ibis Prevedello Date: Thu, 9 Mar 2023 21:23:42 -0300 Subject: [PATCH] refac: remove iosession as required --- .../backend/langflow_backend/interface/signature.py | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/langflow/backend/langflow_backend/interface/signature.py b/langflow/backend/langflow_backend/interface/signature.py index ad1ffbb79..c5d24d560 100644 --- a/langflow/backend/langflow_backend/interface/signature.py +++ b/langflow/backend/langflow_backend/interface/signature.py @@ -86,7 +86,7 @@ def get_tool_signature(name: str): type_dict = { "str": { "type": "str", - "required": False, + "required": True, "list": False, "show": True, "placeholder": "", @@ -111,9 +111,15 @@ def get_tool_signature(name: str): params = [] template = { - param: (type_dict[param] if param == "llm" else type_dict["str"]) + param: (type_dict[param].copy() if param == "llm" else type_dict["str"].copy()) for param in params - } # type: Dict[str, Any] + } + + # Remove required from aiosession + if "aiosession" in template.keys(): + template["aiosession"]["required"] = False + template["aiosession"]["show"] = False + template["_type"] = tool_type return {