langflow-base-nightly 0.5.1.dev3__py3-none-any.whl → 0.5.1.dev4__py3-none-any.whl

langflow/components/zep/zep.py

@@ -1,44 +0,0 @@
-from langflow.base.memory.model import LCChatMemoryComponent
-from langflow.field_typing.constants import Memory
-from langflow.inputs.inputs import DropdownInput, MessageTextInput, SecretStrInput
-
-
-class ZepChatMemory(LCChatMemoryComponent):
-    display_name = "Zep Chat Memory"
-    description = "Retrieves and store chat messages from Zep."
-    name = "ZepChatMemory"
-    icon = "ZepMemory"
-    legacy = True
-
-    inputs = [
-        MessageTextInput(name="url", display_name="Zep URL", info="URL of the Zep instance."),
-        SecretStrInput(name="api_key", display_name="API Key", info="API Key for the Zep instance."),
-        DropdownInput(
-            name="api_base_path",
-            display_name="API Base Path",
-            options=["api/v1", "api/v2"],
-            value="api/v1",
-            advanced=True,
-        ),
-        MessageTextInput(
-            name="session_id", display_name="Session ID", info="Session ID for the message.", advanced=True
-        ),
-    ]
-
-    def build_message_history(self) -> Memory:
-        try:
-            # Monkeypatch API_BASE_PATH to
-            # avoid 404
-            # This is a workaround for the local Zep instance
-            # cloud Zep works with v2
-            import zep_python.zep_client
-            from zep_python import ZepClient
-            from zep_python.langchain import ZepChatMessageHistory
-
-            zep_python.zep_client.API_BASE_PATH = self.api_base_path
-        except ImportError as e:
-            msg = "Could not import zep-python package. Please install it with `pip install zep-python`."
-            raise ImportError(msg) from e
-
-        zep_client = ZepClient(api_url=self.url, api_key=self.api_key)
-        return ZepChatMessageHistory(session_id=self.session_id, zep_client=zep_client)

langflow/custom/attributes.py

@@ -1,86 +0,0 @@
-from collections.abc import Callable
-
-import emoji
-
-from langflow.logging.logger import logger
-
-
-def validate_icon(value: str):
-    # we are going to use the emoji library to validate the emoji
-    # emojis can be defined using the :emoji_name: syntax
-
-    if not value.startswith(":") and not value.endswith(":"):
-        return value
-    if not value.startswith(":") or not value.endswith(":"):
-        # emoji should have both starting and ending colons
-        # so if one of them is missing, we will raise
-        msg = f"Invalid emoji. {value} is not a valid emoji."
-        raise ValueError(msg)
-
-    emoji_value = emoji.emojize(value, variant="emoji_type")
-    if value == emoji_value:
-        logger.warning(f"Invalid emoji. {value} is not a valid emoji.")
-        return value
-    return emoji_value
-
-
-def getattr_return_str(value):
-    return str(value) if value else ""
-
-
-def getattr_return_bool(value):
-    if isinstance(value, bool):
-        return value
-    return None
-
-
-def getattr_return_int(value):
-    if isinstance(value, int):
-        return value
-    return None
-
-
-def getattr_return_list_of_str(value):
-    if isinstance(value, list):
-        return [str(val) for val in value]
-    return []
-
-
-def getattr_return_list_of_object(value):
-    if isinstance(value, list):
-        return value
-    return []
-
-
-def getattr_return_list_of_values_from_dict(value):
-    if isinstance(value, dict):
-        return list(value.values())
-    return []
-
-
-def getattr_return_dict(value):
-    if isinstance(value, dict):
-        return value
-    return {}
-
-
-ATTR_FUNC_MAPPING: dict[str, Callable] = {
-    "display_name": getattr_return_str,
-    "description": getattr_return_str,
-    "beta": getattr_return_bool,
-    "legacy": getattr_return_bool,
-    "documentation": getattr_return_str,
-    "priority": getattr_return_int,
-    "icon": validate_icon,
-    "minimized": getattr_return_bool,
-    "frozen": getattr_return_bool,
-    "is_input": getattr_return_bool,
-    "is_output": getattr_return_bool,
-    "conditional_paths": getattr_return_list_of_str,
-    "_outputs_map": getattr_return_list_of_values_from_dict,
-    "_inputs": getattr_return_list_of_values_from_dict,
-    "outputs": getattr_return_list_of_object,
-    "inputs": getattr_return_list_of_object,
-    "metadata": getattr_return_dict,
-    "tool_mode": getattr_return_bool,
-}

langflow/custom/code_parser/__init__.py

@@ -1,3 +0,0 @@
-from .code_parser import CodeParser
-
-__all__ = ["CodeParser"]

langflow/custom/code_parser/code_parser.py

@@ -1,361 +0,0 @@
-import ast
-import contextlib
-import inspect
-import traceback
-from itertools import starmap
-from pathlib import Path
-from typing import Any
-
-from cachetools import TTLCache, keys
-from fastapi import HTTPException
-
-from langflow.custom.eval import eval_custom_component_code
-from langflow.custom.schema import CallableCodeDetails, ClassCodeDetails, MissingDefault
-from langflow.logging.logger import logger
-
-
-class CodeSyntaxError(HTTPException):
-    pass
-
-
-def get_data_type():
-    from langflow.field_typing import Data
-
-    return Data
-
-
-def find_class_ast_node(class_obj):
-    """Finds the AST node corresponding to the given class object."""
-    # Get the source file where the class is defined
-    source_file = inspect.getsourcefile(class_obj)
-    if not source_file:
-        return None, []
-
-    # Read the source code from the file
-    source_code = Path(source_file).read_text(encoding="utf-8")
-
-    # Parse the source code into an AST
-    tree = ast.parse(source_code)
-
-    # Search for the class definition node in the AST
-    class_node = None
-    import_nodes = []
-    for node in ast.walk(tree):
-        if isinstance(node, ast.ClassDef) and node.name == class_obj.__name__:
-            class_node = node
-        elif isinstance(node, ast.Import | ast.ImportFrom):
-            import_nodes.append(node)
-
-    return class_node, import_nodes
-
-
-def imports_key(*args, **kwargs):
-    imports = kwargs.pop("imports")
-    key = keys.methodkey(*args, **kwargs)
-    key += tuple(imports)
-    return key
-
-
-class CodeParser:
-    """A parser for Python source code, extracting code details."""
-
-    def __init__(self, code: str | type) -> None:
-        """Initializes the parser with the provided code."""
-        self.cache: TTLCache = TTLCache(maxsize=1024, ttl=60)
-        if isinstance(code, type):
-            if not inspect.isclass(code):
-                msg = "The provided code must be a class."
-                raise ValueError(msg)
-            # If the code is a class, get its source code
-            code = inspect.getsource(code)
-        self.code = code
-        self.data: dict[str, Any] = {
-            "imports": [],
-            "functions": [],
-            "classes": [],
-            "global_vars": [],
-        }
-        self.handlers = {
-            ast.Import: self.parse_imports,
-            ast.ImportFrom: self.parse_imports,
-            ast.FunctionDef: self.parse_functions,
-            ast.ClassDef: self.parse_classes,
-            ast.Assign: self.parse_global_vars,
-        }
-
-    def get_tree(self):
-        """Parses the provided code to validate its syntax.
-
-        It tries to parse the code into an abstract syntax tree (AST).
-        """
-        try:
-            tree = ast.parse(self.code)
-        except SyntaxError as err:
-            raise CodeSyntaxError(
-                status_code=400,
-                detail={"error": err.msg, "traceback": traceback.format_exc()},
-            ) from err
-
-        return tree
-
-    def parse_node(self, node: ast.stmt | ast.AST) -> None:
-        """Parses an AST node and updates the data dictionary with the relevant information."""
-        if handler := self.handlers.get(type(node)):
-            handler(node)  # type: ignore[operator]
-
-    def parse_imports(self, node: ast.Import | ast.ImportFrom) -> None:
-        """Extracts "imports" from the code, including aliases."""
-        if isinstance(node, ast.Import):
-            for alias in node.names:
-                if alias.asname:
-                    self.data["imports"].append(f"{alias.name} as {alias.asname}")
-                else:
-                    self.data["imports"].append(alias.name)
-        elif isinstance(node, ast.ImportFrom):
-            for alias in node.names:
-                if alias.asname:
-                    self.data["imports"].append((node.module, f"{alias.name} as {alias.asname}"))
-                else:
-                    self.data["imports"].append((node.module, alias.name))
-
-    def parse_functions(self, node: ast.FunctionDef) -> None:
-        """Extracts "functions" from the code."""
-        self.data["functions"].append(self.parse_callable_details(node))
-
-    def parse_arg(self, arg, default):
-        """Parses an argument and its default value."""
-        arg_dict = {"name": arg.arg, "default": default}
-        if arg.annotation:
-            arg_dict["type"] = ast.unparse(arg.annotation)
-        return arg_dict
-
-    # @cachedmethod(operator.attrgetter("cache"))
-    def construct_eval_env(self, return_type_str: str, imports) -> dict:
-        """Constructs an evaluation environment.
-
-        Constructs an evaluation environment with the necessary imports for the return type,
-        taking into account module aliases.
-        """
-        eval_env: dict = {}
-        for import_entry in imports:
-            if isinstance(import_entry, tuple):  # from module import name
-                module, name = import_entry
-                if name in return_type_str:
-                    exec(f"import {module}", eval_env)
-                    exec(f"from {module} import {name}", eval_env)
-            else:  # import module
-                module = import_entry
-                alias = None
-                if " as " in module:
-                    module, alias = module.split(" as ")
-                if module in return_type_str or (alias and alias in return_type_str):
-                    exec(f"import {module} as {alias or module}", eval_env)
-        return eval_env
-
-    def parse_callable_details(self, node: ast.FunctionDef) -> dict[str, Any]:
-        """Extracts details from a single function or method node."""
-        return_type = None
-        if node.returns:
-            return_type_str = ast.unparse(node.returns)
-            eval_env = self.construct_eval_env(return_type_str, tuple(self.data["imports"]))
-
-            # Handle cases where the type is not found in the constructed environment
-            with contextlib.suppress(NameError):
-                return_type = eval(return_type_str, eval_env)  # noqa: S307
-
-        func = CallableCodeDetails(
-            name=node.name,
-            doc=ast.get_docstring(node),
-            args=self.parse_function_args(node),
-            body=self.parse_function_body(node),
-            return_type=return_type,
-            has_return=self.parse_return_statement(node),
-        )
-
-        return func.model_dump()
-
-    def parse_function_args(self, node: ast.FunctionDef) -> list[dict[str, Any]]:
-        """Parses the arguments of a function or method node."""
-        args = []
-
-        args += self.parse_positional_args(node)
-        args += self.parse_varargs(node)
-        args += self.parse_keyword_args(node)
-        # Commented out because we don't want kwargs
-        # showing up as fields in the frontend
-        args += self.parse_kwargs(node)
-
-        return args
-
-    def parse_positional_args(self, node: ast.FunctionDef) -> list[dict[str, Any]]:
-        """Parses the positional arguments of a function or method node."""
-        num_args = len(node.args.args)
-        num_defaults = len(node.args.defaults)
-        num_missing_defaults = num_args - num_defaults
-        missing_defaults = [MissingDefault()] * num_missing_defaults
-        default_values = [ast.unparse(default).strip("'") if default else None for default in node.args.defaults]
-        # Now check all default values to see if there
-        # are any "None" values in the middle
-        default_values = [None if value == "None" else value for value in default_values]
-
-        defaults = missing_defaults + default_values
-
-        return list(starmap(self.parse_arg, zip(node.args.args, defaults, strict=True)))
-
-    def parse_varargs(self, node: ast.FunctionDef) -> list[dict[str, Any]]:
-        """Parses the *args argument of a function or method node."""
-        args = []
-
-        if node.args.vararg:
-            args.append(self.parse_arg(node.args.vararg, None))
-
-        return args
-
-    def parse_keyword_args(self, node: ast.FunctionDef) -> list[dict[str, Any]]:
-        """Parses the keyword-only arguments of a function or method node."""
-        kw_defaults = [None] * (len(node.args.kwonlyargs) - len(node.args.kw_defaults)) + [
-            ast.unparse(default) if default else None for default in node.args.kw_defaults
-        ]
-
-        return list(starmap(self.parse_arg, zip(node.args.kwonlyargs, kw_defaults, strict=True)))
-
-    def parse_kwargs(self, node: ast.FunctionDef) -> list[dict[str, Any]]:
-        """Parses the **kwargs argument of a function or method node."""
-        args = []
-
-        if node.args.kwarg:
-            args.append(self.parse_arg(node.args.kwarg, None))
-
-        return args
-
-    def parse_function_body(self, node: ast.FunctionDef) -> list[str]:
-        """Parses the body of a function or method node."""
-        return [ast.unparse(line) for line in node.body]
-
-    def parse_return_statement(self, node: ast.FunctionDef) -> bool:
-        """Parses the return statement of a function or method node, including nested returns."""
-
-        def has_return(node):
-            if isinstance(node, ast.Return):
-                return True
-            if isinstance(node, ast.If):
-                return any(has_return(child) for child in node.body) or any(has_return(child) for child in node.orelse)
-            if isinstance(node, ast.Try):
-                return (
-                    any(has_return(child) for child in node.body)
-                    or any(has_return(child) for child in node.handlers)
-                    or any(has_return(child) for child in node.finalbody)
-                )
-            if isinstance(node, ast.For | ast.While):
-                return any(has_return(child) for child in node.body) or any(has_return(child) for child in node.orelse)
-            if isinstance(node, ast.With):
-                return any(has_return(child) for child in node.body)
-            return False
-
-        return any(has_return(child) for child in node.body)
-
-    def parse_assign(self, stmt):
-        """Parses an Assign statement and returns a dictionary with the target's name and value."""
-        for target in stmt.targets:
-            if isinstance(target, ast.Name):
-                return {"name": target.id, "value": ast.unparse(stmt.value)}
-        return None
-
-    def parse_ann_assign(self, stmt):
-        """Parses an AnnAssign statement and returns a dictionary with the target's name, value, and annotation."""
-        if isinstance(stmt.target, ast.Name):
-            return {
-                "name": stmt.target.id,
-                "value": ast.unparse(stmt.value) if stmt.value else None,
-                "annotation": ast.unparse(stmt.annotation),
-            }
-        return None
-
-    def parse_function_def(self, stmt):
-        """Parse a FunctionDef statement.
-
-        Parse a FunctionDef statement and return the parsed method and a boolean indicating if it's an __init__ method.
-        """
-        method = self.parse_callable_details(stmt)
-        return (method, True) if stmt.name == "__init__" else (method, False)
-
-    def get_base_classes(self):
-        """Returns the base classes of the custom component class."""
-        try:
-            bases = self.execute_and_inspect_classes(self.code)
-        except Exception:
-            # If the code cannot be executed, return an empty list
-            bases = []
-            raise
-        return bases
-
-    def parse_classes(self, node: ast.ClassDef) -> None:
-        """Extracts "classes" from the code, including inheritance and init methods."""
-        bases = self.get_base_classes()
-        nodes = []
-        for base in bases:
-            if base.__name__ == node.name or base.__name__ in {"CustomComponent", "Component", "BaseComponent"}:
-                continue
-            try:
-                class_node, import_nodes = find_class_ast_node(base)
-                if class_node is None:
-                    continue
-                for import_node in import_nodes:
-                    self.parse_imports(import_node)
-                nodes.append(class_node)
-            except Exception:  # noqa: BLE001
-                logger.exception("Error finding base class node")
-        nodes.insert(0, node)
-        class_details = ClassCodeDetails(
-            name=node.name,
-            doc=ast.get_docstring(node),
-            bases=[b.__name__ for b in bases],
-            attributes=[],
-            methods=[],
-            init=None,
-        )
-        for _node in nodes:
-            self.process_class_node(_node, class_details)
-        self.data["classes"].append(class_details.model_dump())
-
-    def process_class_node(self, node, class_details) -> None:
-        for stmt in node.body:
-            if isinstance(stmt, ast.Assign):
-                if attr := self.parse_assign(stmt):
-                    class_details.attributes.append(attr)
-            elif isinstance(stmt, ast.AnnAssign):
-                if attr := self.parse_ann_assign(stmt):
-                    class_details.attributes.append(attr)
-            elif isinstance(stmt, ast.FunctionDef | ast.AsyncFunctionDef):
-                method, is_init = self.parse_function_def(stmt)
-                if is_init:
-                    class_details.init = method
-                else:
-                    class_details.methods.append(method)
-
-    def parse_global_vars(self, node: ast.Assign) -> None:
-        """Extracts global variables from the code."""
-        global_var = {
-            "targets": [t.id if hasattr(t, "id") else ast.dump(t) for t in node.targets],
-            "value": ast.unparse(node.value),
-        }
-        self.data["global_vars"].append(global_var)
-
-    def execute_and_inspect_classes(self, code: str):
-        custom_component_class = eval_custom_component_code(code)
-        custom_component = custom_component_class(_code=code)
-        dunder_class = custom_component.__class__
-        # Get the base classes at two levels of inheritance
-        bases = []
-        for base in dunder_class.__bases__:
-            bases.append(base)
-            bases.extend(base.__bases__)
-        return bases
-
-    def parse_code(self) -> dict[str, Any]:
-        """Runs all parsing operations and returns the resulting data."""
-        tree = self.get_tree()
-
-        for node in ast.walk(tree):
-            self.parse_node(node)
-        return self.data

langflow/custom/custom_component/base_component.py

@@ -1,118 +0,0 @@
-import copy
-import operator
-import re
-from typing import TYPE_CHECKING, Any, ClassVar
-
-from cachetools import TTLCache, cachedmethod
-from fastapi import HTTPException
-
-from langflow.custom.attributes import ATTR_FUNC_MAPPING
-from langflow.custom.code_parser.code_parser import CodeParser
-from langflow.custom.eval import eval_custom_component_code
-from langflow.logging.logger import logger
-from langflow.utils import validate
-
-if TYPE_CHECKING:
-    from uuid import UUID
-
-
-class ComponentCodeNullError(HTTPException):
-    pass
-
-
-class ComponentFunctionEntrypointNameNullError(HTTPException):
-    pass
-
-
-class BaseComponent:
-    ERROR_CODE_NULL: ClassVar[str] = "Python code must be provided."
-    ERROR_FUNCTION_ENTRYPOINT_NAME_NULL: ClassVar[str] = "The name of the entrypoint function must be provided."
-
-    def __init__(self, **data) -> None:
-        self._code: str | None = None
-        self._function_entrypoint_name: str = "build"
-        self.field_config: dict = {}
-        self._user_id: str | UUID | None = None
-        self._template_config: dict = {}
-
-        self.cache: TTLCache = TTLCache(maxsize=1024, ttl=60)
-
-        for key, value in data.items():
-            if key == "user_id":
-                self._user_id = value
-            else:
-                setattr(self, key, value)
-
-    def __setattr__(self, key, value) -> None:
-        if key == "_user_id":
-            try:
-                if self._user_id is not None:
-                    logger.warning("user_id is immutable and cannot be changed.")
-            except (KeyError, AttributeError):
-                pass
-        super().__setattr__(key, value)
-
-    @cachedmethod(cache=operator.attrgetter("cache"))
-    def get_code_tree(self, code: str):
-        parser = CodeParser(code)
-        return parser.parse_code()
-
-    def get_function(self):
-        if not self._code:
-            raise ComponentCodeNullError(
-                status_code=400,
-                detail={"error": self.ERROR_CODE_NULL, "traceback": ""},
-            )
-
-        if not self._function_entrypoint_name:
-            raise ComponentFunctionEntrypointNameNullError(
-                status_code=400,
-                detail={
-                    "error": self.ERROR_FUNCTION_ENTRYPOINT_NAME_NULL,
-                    "traceback": "",
-                },
-            )
-
-        return validate.create_function(self._code, self._function_entrypoint_name)
-
-    @staticmethod
-    def get_template_config(component):
-        """Gets the template configuration for the custom component itself."""
-        template_config = {}
-
-        for attribute, func in ATTR_FUNC_MAPPING.items():
-            if hasattr(component, attribute):
-                value = getattr(component, attribute)
-                if value is not None:
-                    value_copy = copy.deepcopy(value)
-                    template_config[attribute] = func(value=value_copy)
-
-        for key in template_config.copy():
-            if key not in ATTR_FUNC_MAPPING:
-                template_config.pop(key, None)
-
-        return template_config
-
-    def build_template_config(self) -> dict:
-        """Builds the template configuration for the custom component.
-
-        Returns:
-            A dictionary representing the template configuration.
-        """
-        if not self._code:
-            return {}
-
-        try:
-            cc_class = eval_custom_component_code(self._code)
-
-        except AttributeError as e:
-            pattern = r"module '.*?' has no attribute '.*?'"
-            if re.search(pattern, str(e)):
-                raise ImportError(e) from e
-            raise
-
-        component_instance = cc_class(_code=self._code)
-        return self.get_template_config(component_instance)
-
-    def build(self, *args: Any, **kwargs: Any) -> Any:
-        raise NotImplementedError