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

langflow/graph/vertex/vertex_types.py

@@ -1,489 +0,0 @@
-from __future__ import annotations
-
-import contextlib
-import json
-from collections.abc import AsyncIterator, Generator, Iterator
-from typing import TYPE_CHECKING, Any, cast
-
-import yaml
-from langchain_core.messages import AIMessage, AIMessageChunk
-
-from langflow.graph.schema import CHAT_COMPONENTS, RECORDS_COMPONENTS, InterfaceComponentTypes, ResultData
-from langflow.graph.utils import UnbuiltObject, log_vertex_build, rewrite_file_path
-from langflow.graph.vertex.base import Vertex
-from langflow.graph.vertex.exceptions import NoComponentInstanceError
-from langflow.logging.logger import logger
-from langflow.schema.artifact import ArtifactType
-from langflow.schema.data import Data
-from langflow.schema.message import Message
-from langflow.schema.schema import INPUT_FIELD_NAME
-from langflow.serialization.serialization import serialize
-from langflow.template.field.base import UNDEFINED, Output
-from langflow.utils.schemas import ChatOutputResponse, DataOutputResponse
-from langflow.utils.util import unescape_string
-
-if TYPE_CHECKING:
-    from langflow.graph.edge.base import CycleEdge
-    from langflow.graph.vertex.schema import NodeData
-    from langflow.inputs.inputs import InputTypes
-
-
-class CustomComponentVertex(Vertex):
-    def __init__(self, data: NodeData, graph):
-        super().__init__(data, graph=graph, base_type="custom_components")
-
-    def built_object_repr(self):
-        if self.artifacts and "repr" in self.artifacts:
-            return self.artifacts["repr"] or super().built_object_repr()
-        return None
-
-
-class ComponentVertex(Vertex):
-    def __init__(self, data: NodeData, graph):
-        super().__init__(data, graph=graph, base_type="component")
-
-    def get_input(self, name: str) -> InputTypes:
-        if self.custom_component is None:
-            msg = f"Vertex {self.id} does not have a component instance."
-            raise ValueError(msg)
-        return self.custom_component.get_input(name)
-
-    def get_output(self, name: str) -> Output:
-        if self.custom_component is None:
-            raise NoComponentInstanceError(self.id)
-        return self.custom_component.get_output(name)
-
-    def built_object_repr(self):
-        if self.artifacts and "repr" in self.artifacts:
-            return self.artifacts["repr"] or super().built_object_repr()
-        return None
-
-    def _update_built_object_and_artifacts(self, result) -> None:
-        """Updates the built object and its artifacts."""
-        if isinstance(result, tuple):
-            if len(result) == 2:  # noqa: PLR2004
-                self.built_object, self.artifacts = result
-            elif len(result) == 3:  # noqa: PLR2004
-                self.custom_component, self.built_object, self.artifacts = result
-                self.logs = self.custom_component._output_logs
-                for key in self.artifacts:
-                    if self.artifacts_raw is None:
-                        self.artifacts_raw = {}
-                    self.artifacts_raw[key] = self.artifacts[key].get("raw", None)
-                    self.artifacts_type[key] = self.artifacts[key].get("type", None) or ArtifactType.UNKNOWN.value
-        else:
-            self.built_object = result
-
-        for key, value in self.built_object.items():
-            self.add_result(key, value)
-
-    def get_edge_with_target(self, target_id: str) -> Generator[CycleEdge, None, None]:
-        """Get the edge with the target id.
-
-        Args:
-            target_id: The target id of the edge.
-
-        Returns:
-            The edge with the target id.
-        """
-        for edge in self.edges:
-            if edge.target_id == target_id:
-                yield edge
-
-    async def _get_result(self, requester: Vertex, target_handle_name: str | None = None) -> Any:
-        """Retrieves the result of the built component.
-
-        If the component has not been built yet, a ValueError is raised.
-
-        Returns:
-            The built result if use_result is True, else the built object.
-        """
-        flow_id = self.graph.flow_id
-        if not self.built:
-            default_value: Any = UNDEFINED
-            for edge in self.get_edge_with_target(requester.id):
-                # We need to check if the edge is a normal edge
-                if edge.is_cycle and edge.target_param:
-                    if edge.target_param in requester.output_names:
-                        default_value = None
-                    else:
-                        default_value = requester.get_value_from_template_dict(edge.target_param)
-
-            if flow_id:
-                await self._log_transaction_async(source=self, target=requester, flow_id=str(flow_id), status="error")
-            if default_value is not UNDEFINED:
-                return default_value
-            msg = f"Component {self.display_name} has not been built yet"
-            raise ValueError(msg)
-
-        if requester is None:
-            msg = "Requester Vertex is None"
-            raise ValueError(msg)
-
-        edges = self.get_edge_with_target(requester.id)
-        result = UNDEFINED
-        for edge in edges:
-            if (
-                edge is not None
-                and edge.source_handle.name in self.results
-                and edge.target_handle.field_name == target_handle_name
-            ):
-                # Get the result from the output instead of the results dict
-                try:
-                    output = self.get_output(edge.source_handle.name)
-
-                    if output.value is UNDEFINED:
-                        result = self.results[edge.source_handle.name]
-                    else:
-                        result = cast("Any", output.value)
-                except NoComponentInstanceError:
-                    result = self.results[edge.source_handle.name]
-                break
-        if result is UNDEFINED:
-            if edge is None:
-                msg = f"Edge not found between {self.display_name} and {requester.display_name}"
-                raise ValueError(msg)
-            if edge.source_handle.name not in self.results:
-                msg = f"Result not found for {edge.source_handle.name}. Results: {self.results}"
-                raise ValueError(msg)
-            msg = f"Result not found for {edge.source_handle.name} in {edge}"
-            raise ValueError(msg)
-        if flow_id:
-            await self._log_transaction_async(source=self, target=requester, flow_id=str(flow_id), status="success")
-        return result
-
-    def extract_messages_from_artifacts(self, artifacts: dict[str, Any]) -> list[dict]:
-        """Extracts messages from the artifacts.
-
-        Args:
-            artifacts (Dict[str, Any]): The artifacts to extract messages from.
-
-        Returns:
-            List[str]: The extracted messages.
-        """
-        messages = []
-        for key, artifact in artifacts.items():
-            if any(
-                k not in artifact for k in ["text", "sender", "sender_name", "session_id", "stream_url"]
-            ) and not isinstance(artifact, Message):
-                continue
-            message_dict = artifact if isinstance(artifact, dict) else artifact.model_dump()
-            if not message_dict.get("text"):
-                continue
-            with contextlib.suppress(KeyError):
-                messages.append(
-                    ChatOutputResponse(
-                        message=message_dict["text"],
-                        sender=message_dict.get("sender"),
-                        sender_name=message_dict.get("sender_name"),
-                        session_id=message_dict.get("session_id"),
-                        stream_url=message_dict.get("stream_url"),
-                        files=[
-                            {"path": file} if isinstance(file, str) else file for file in message_dict.get("files", [])
-                        ],
-                        component_id=self.id,
-                        type=self.artifacts_type[key],
-                    ).model_dump(exclude_none=True)
-                )
-        return messages
-
-    def finalize_build(self) -> None:
-        result_dict = self.get_built_result()
-        # We need to set the artifacts to pass information
-        # to the frontend
-        messages = self.extract_messages_from_artifacts(result_dict)
-        result_dict = ResultData(
-            results=result_dict,
-            artifacts=self.artifacts,
-            outputs=self.outputs_logs,
-            logs=self.logs,
-            messages=messages,
-            component_display_name=self.display_name,
-            component_id=self.id,
-        )
-        self.set_result(result_dict)
-
-
-class InterfaceVertex(ComponentVertex):
-    def __init__(self, data: NodeData, graph):
-        super().__init__(data, graph=graph)
-        self.added_message = None
-        self.steps = [self._build, self._run]
-        self.is_interface_component = True
-
-    def build_stream_url(self) -> str:
-        return f"/api/v1/build/{self.graph.flow_id}/{self.id}/stream"
-
-    def built_object_repr(self):
-        if self.task_id and self.is_task:
-            if task := self.get_task():
-                return str(task.info)
-            return f"Task {self.task_id} is not running"
-        if self.artifacts:
-            # dump as a yaml string
-            if isinstance(self.artifacts, dict):
-                artifacts_ = [self.artifacts]
-            elif hasattr(self.artifacts, "data"):
-                artifacts_ = self.artifacts.data
-            else:
-                artifacts_ = self.artifacts
-            artifacts = []
-            for artifact in artifacts_:
-                # artifacts = {k.title().replace("_", " "): v for k, v in self.artifacts.items() if v is not None}
-                artifact_ = {k.title().replace("_", " "): v for k, v in artifact.items() if v is not None}
-                artifacts.append(artifact_)
-            return yaml.dump(artifacts, default_flow_style=False, allow_unicode=True)
-        return super().built_object_repr()
-
-    def _process_chat_component(self):
-        """Process the chat component and return the message.
-
-        This method processes the chat component by extracting the necessary parameters
-        such as sender, sender_name, and message from the `params` dictionary. It then
-        performs additional operations based on the type of the `built_object` attribute.
-        If `built_object` is an instance of `AIMessage`, it creates a `ChatOutputResponse`
-        object using the `from_message` method. If `built_object` is not an instance of
-        `UnbuiltObject`, it checks the type of `built_object` and performs specific
-        operations accordingly. If `built_object` is a dictionary, it converts it into a
-        code block. If `built_object` is an instance of `Data`, it assigns the `text`
-        attribute to the `message` variable. If `message` is an instance of `AsyncIterator`
-        or `Iterator`, it builds a stream URL and sets `message` to an empty string. If
-        `built_object` is not a string, it converts it to a string. If `message` is a
-        generator or iterator, it assigns it to the `message` variable. Finally, it creates
-        a `ChatOutputResponse` object using the extracted parameters and assigns it to the
-        `artifacts` attribute. If `artifacts` is not None, it calls the `model_dump` method
-        on it and assigns the result to the `artifacts` attribute. It then returns the
-        `message` variable.
-
-        Returns:
-            str: The processed message.
-        """
-        artifacts = None
-        sender = self.params.get("sender", None)
-        sender_name = self.params.get("sender_name", None)
-        message = self.params.get(INPUT_FIELD_NAME, None)
-        files = self.params.get("files", [])
-        treat_file_path = files is not None and not isinstance(files, list) and isinstance(files, str)
-        if treat_file_path:
-            self.params["files"] = rewrite_file_path(files)
-        files = [{"path": file} if isinstance(file, str) else file for file in self.params.get("files", [])]
-        if isinstance(message, str):
-            message = unescape_string(message)
-        stream_url = None
-        if "text" in self.results:
-            text_output = self.results["text"]
-        elif "message" in self.results:
-            text_output = self.results["message"].text
-        else:
-            text_output = message
-        if isinstance(text_output, AIMessage | AIMessageChunk):
-            artifacts = ChatOutputResponse.from_message(
-                text_output,
-                sender=sender,
-                sender_name=sender_name,
-            )
-        elif not isinstance(text_output, UnbuiltObject):
-            if isinstance(text_output, dict):
-                # Turn the dict into a pleasing to
-                # read JSON inside a code block
-                message = dict_to_codeblock(text_output)
-            elif isinstance(text_output, Data):
-                message = text_output.text
-            elif isinstance(message, AsyncIterator | Iterator):
-                stream_url = self.build_stream_url()
-                message = ""
-                self.results["text"] = message
-                self.results["message"].text = message
-                self.built_object = self.results
-            elif not isinstance(text_output, str):
-                message = str(text_output)
-            # if the message is a generator or iterator
-            # it means that it is a stream of messages
-
-            else:
-                message = text_output
-
-            if hasattr(sender_name, "get_text"):
-                sender_name = sender_name.get_text()
-
-            artifact_type = ArtifactType.STREAM if stream_url is not None else ArtifactType.OBJECT
-            artifacts = ChatOutputResponse(
-                message=message,
-                sender=sender,
-                sender_name=sender_name,
-                stream_url=stream_url,
-                files=files,
-                type=artifact_type,
-            )
-
-            self.will_stream = stream_url is not None
-        if artifacts:
-            self.artifacts = artifacts.model_dump(exclude_none=True)
-
-        return message
-
-    def _process_data_component(self):
-        """Process the record component of the vertex.
-
-        If the built object is an instance of `Data`, it calls the `model_dump` method
-        and assigns the result to the `artifacts` attribute.
-
-        If the built object is a list, it iterates over each element and checks if it is
-        an instance of `Data`. If it is, it calls the `model_dump` method and appends
-        the result to the `artifacts` list. If it is not, it raises a `ValueError` if the
-        `ignore_errors` parameter is set to `False`, or logs an error message if it is set
-        to `True`.
-
-        Returns:
-            The built object.
-
-        Raises:
-            ValueError: If an element in the list is not an instance of `Data` and
-                `ignore_errors` is set to `False`.
-        """
-        if isinstance(self.built_object, Data):
-            artifacts = [self.built_object.data]
-        elif isinstance(self.built_object, list):
-            artifacts = []
-            ignore_errors = self.params.get("ignore_errors", False)
-            for value in self.built_object:
-                if isinstance(value, Data):
-                    artifacts.append(value.data)
-                elif ignore_errors:
-                    logger.error(f"Data expected, but got {value} of type {type(value)}")
-                else:
-                    msg = f"Data expected, but got {value} of type {type(value)}"
-                    raise ValueError(msg)
-        self.artifacts = DataOutputResponse(data=artifacts)
-        return self.built_object
-
-    async def _run(self, *args, **kwargs) -> None:  # noqa: ARG002
-        if self.vertex_type in CHAT_COMPONENTS:
-            message = self._process_chat_component()
-        elif self.vertex_type in RECORDS_COMPONENTS:
-            message = self._process_data_component()
-        if isinstance(self.built_object, AsyncIterator | Iterator):
-            if self.params.get("return_data", False):
-                self.built_object = Data(text=message, data=self.artifacts)
-            else:
-                self.built_object = message
-        self.built_result = self.built_object
-
-    async def stream(self):
-        iterator = self.params.get(INPUT_FIELD_NAME, None)
-        if not isinstance(iterator, AsyncIterator | Iterator):
-            msg = "The message must be an iterator or an async iterator."
-            raise TypeError(msg)
-        is_async = isinstance(iterator, AsyncIterator)
-        complete_message = ""
-        if is_async:
-            async for message in iterator:
-                message_ = message.content if hasattr(message, "content") else message
-                message_ = message_.text if hasattr(message_, "text") else message_
-                yield message_
-                complete_message += message_
-        else:
-            for message in iterator:
-                message_ = message.content if hasattr(message, "content") else message
-                message_ = message_.text if hasattr(message_, "text") else message_
-                yield message_
-                complete_message += message_
-
-        files = self.params.get("files", [])
-
-        treat_file_path = files is not None and not isinstance(files, list) and isinstance(files, str)
-        if treat_file_path:
-            self.params["files"] = rewrite_file_path(files)
-
-        if hasattr(self.params.get("sender_name"), "get_text"):
-            sender_name = self.params.get("sender_name").get_text()
-        else:
-            sender_name = self.params.get("sender_name")
-        self.artifacts = ChatOutputResponse(
-            message=complete_message,
-            sender=self.params.get("sender", ""),
-            sender_name=sender_name,
-            files=[{"path": file} if isinstance(file, str) else file for file in self.params.get("files", [])],
-            type=ArtifactType.OBJECT.value,
-        ).model_dump()
-
-        message = await Message.create(
-            text=complete_message,
-            sender=self.params.get("sender", ""),
-            sender_name=self.params.get("sender_name", ""),
-            files=self.params.get("files", []),
-            flow_id=self.graph.flow_id,
-            session_id=self.params.get("session_id", ""),
-        )
-        self.params[INPUT_FIELD_NAME] = complete_message
-        if isinstance(self.built_object, dict):
-            for key, value in self.built_object.items():
-                if hasattr(value, "text") and (isinstance(value.text, AsyncIterator | Iterator) or value.text == ""):
-                    self.built_object[key] = message
-        else:
-            self.built_object = message
-            self.artifacts_type = ArtifactType.MESSAGE
-
-        # Update artifacts with the message
-        # and remove the stream_url
-        self.finalize_build()
-        await logger.adebug(f"Streamed message: {complete_message}")
-        # Set the result in the vertex of origin
-        edges = self.get_edge_with_target(self.id)
-        for edge in edges:
-            origin_vertex = self.graph.get_vertex(edge.source_id)
-            for key, value in origin_vertex.results.items():
-                if isinstance(value, AsyncIterator | Iterator):
-                    origin_vertex.results[key] = complete_message
-        if (
-            self.custom_component
-            and hasattr(self.custom_component, "should_store_message")
-            and hasattr(self.custom_component, "store_message")
-        ):
-            await self.custom_component.store_message(message)
-        await log_vertex_build(
-            flow_id=self.graph.flow_id,
-            vertex_id=self.id,
-            valid=True,
-            params=self.built_object_repr(),
-            data=self.result,
-            artifacts=self.artifacts,
-        )
-
-        self._validate_built_object()
-        self.built = True
-
-    async def consume_async_generator(self) -> None:
-        async for _ in self.stream():
-            pass
-
-    def _is_chat_input(self):
-        return self.vertex_type == InterfaceComponentTypes.ChatInput and self.is_input
-
-
-class StateVertex(ComponentVertex):
-    def __init__(self, data: NodeData, graph):
-        """Initializes a StateVertex with the provided node data and graph.
-
-        Sets up the build steps and marks the vertex as a state vertex.
-        """
-        super().__init__(data, graph=graph)
-        self.steps = [self._build]
-        self.is_state = True
-
-    def built_object_repr(self):
-        """Returns a string representation of the built object from the artifacts if available.
-
-        If the artifacts dictionary contains a non-empty "repr" key, its value is returned.
-        If the "repr" value is falsy, falls back to the superclass representation.
-        Returns None if no representation is available.
-        """
-        if self.artifacts and "repr" in self.artifacts:
-            return self.artifacts["repr"] or super().built_object_repr()
-        return None
-
-
-def dict_to_codeblock(d: dict) -> str:
-    serialized = {key: serialize(val) for key, val in d.items()}
-    json_str = json.dumps(serialized, indent=4)
-    return f"```json\n{json_str}\n```"

langflow/legacy_custom/customs.py

@@ -1,16 +0,0 @@
-from langflow.template import frontend_node
-
-# These should always be instantiated
-CUSTOM_NODES: dict[str, dict[str, frontend_node.base.FrontendNode]] = {
-    "custom_components": {
-        "CustomComponent": frontend_node.custom_components.CustomComponentFrontendNode(),
-    },
-    "component": {
-        "Component": frontend_node.custom_components.ComponentFrontendNode(),
-    },
-}
-
-
-def get_custom_nodes(node_type: str):
-    """Get custom nodes."""
-    return CUSTOM_NODES.get(node_type, {})