jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Fix caching bug in CodeParser

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 11:49:46 -03:00
commit 843eeac7ee
1 changed files with 12 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

17
src/backend/langflow/interface/custom/code_parser.py
View file

@ -4,8 +4,9 @@ import operator
import traceback import traceback
from typing import Any, Dict, List, Type, Union from typing import Any, Dict, List, Type, Union
from cachetools import TTLCache, cachedmethod from cachetools import TTLCache, cachedmethod, keys
from fastapi import HTTPException from fastapi import HTTPException
from langflow.interface.custom.schema import CallableCodeDetails, ClassCodeDetails from langflow.interface.custom.schema import CallableCodeDetails, ClassCodeDetails
@ -19,6 +20,13 @@ def get_data_type():
return Data return Data
def imports_key(*args, **kwargs):
imports = kwargs.pop("imports")
key = keys.methodkey(*args, **kwargs)
key += tuple(imports)
return key
class CodeParser: class CodeParser:
""" """
A parser for Python source code, extracting code details. A parser for Python source code, extracting code details.
@ -105,8 +113,7 @@ class CodeParser:
return arg_dict return arg_dict
@cachedmethod(operator.attrgetter("cache")) @cachedmethod(operator.attrgetter("cache"))
@staticmethod def construct_eval_env(self, return_type_str: str, imports) -> dict:
def construct_eval_env(return_type_str: str, imports) -> dict:
""" """
Constructs an evaluation environment with the necessary imports for the return type, Constructs an evaluation environment with the necessary imports for the return type,
taking into account module aliases. taking into account module aliases.
@ -127,7 +134,7 @@ class CodeParser:
exec(f"import {module} as {alias if alias else module}", eval_env) exec(f"import {module} as {alias if alias else module}", eval_env)
return eval_env return eval_env
@cachedmethod(operator.attrgetter("cache")) @cachedmethod(cache=operator.attrgetter("cache"))
def parse_callable_details(self, node: ast.FunctionDef) -> Dict[str, Any]: def parse_callable_details(self, node: ast.FunctionDef) -> Dict[str, Any]:
""" """
Extracts details from a single function or method node. Extracts details from a single function or method node.
@ -135,7 +142,7 @@ class CodeParser:
return_type = None return_type = None
if node.returns: if node.returns:
return_type_str = ast.unparse(node.returns) return_type_str = ast.unparse(node.returns)
eval_env = self.construct_eval_env(return_type_str, self.data["imports"]) eval_env = self.construct_eval_env(return_type_str, tuple(self.data["imports"]))
try: try:
return_type = eval(return_type_str, eval_env) return_type = eval(return_type_str, eval_env)