From 0a21f2b453a3e69ef753af0de4e1f7bd25fe222d Mon Sep 17 00:00:00 2001 From: anovazzi1 Date: Tue, 16 Jan 2024 16:49:50 -0300 Subject: [PATCH] Refactor build method in JsonToolkitComponent --- src/backend/langflow/components/toolkits/JsonToolkit.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/components/toolkits/JsonToolkit.py b/src/backend/langflow/components/toolkits/JsonToolkit.py index d6d1710bb..70e3355e4 100644 --- a/src/backend/langflow/components/toolkits/JsonToolkit.py +++ b/src/backend/langflow/components/toolkits/JsonToolkit.py @@ -1,5 +1,4 @@ from langflow import CustomComponent -from langflow.field_typing import Tool from langchain_community.tools.json.tool import JsonSpec from langchain_community.agent_toolkits.json.toolkit import JsonToolkit @@ -13,7 +12,5 @@ class JsonToolkitComponent(CustomComponent): "spec": {"display_name": "Spec", "type": JsonSpec}, } - def build(self, spec: JsonSpec) -> Tool: - # Assuming JsonToolkit is the class that should be instantiated with the spec - # The actual class name should be used in place of JsonToolkit if it is different - return JsonToolkit(spec=spec) # Replace JsonToolkit with the actual class name if necessary + def build(self, spec: JsonSpec) -> JsonToolkit: + return JsonToolkit(spec=spec) \ No newline at end of file