From 0cc0d8b55635042ebaeacf13ac068e5fe0e4b18d Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 22 Nov 2023 21:20:38 -0300 Subject: [PATCH] Fix return type in CallableCodeDetails constructor --- src/backend/langflow/interface/custom/code_parser.py | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/src/backend/langflow/interface/custom/code_parser.py b/src/backend/langflow/interface/custom/code_parser.py index 2e7deb7da..d1b33a4e2 100644 --- a/src/backend/langflow/interface/custom/code_parser.py +++ b/src/backend/langflow/interface/custom/code_parser.py @@ -11,6 +11,12 @@ class CodeSyntaxError(HTTPException): pass +def get_data_type(): + from langflow.field_typing import Data + + return Data + + class CodeParser: """ A parser for Python source code, extracting code details. @@ -132,11 +138,7 @@ class CodeParser: pass func = CallableCodeDetails( - name=node.name, - doc=ast.get_docstring(node), - args=[], - body=[], - return_type=return_type, + name=node.name, doc=ast.get_docstring(node), args=[], body=[], return_type=return_type or get_data_type() ) func.args = self.parse_function_args(node)