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

langflow/custom/validate.py

@@ -0,0 +1 @@
+from lfx.custom.validate import *  # noqa: F403

langflow/events/event_manager.py

@@ -1,108 +1,18 @@
-from __future__ import annotations
-
-import inspect
-import json
-import time
-import uuid
-from functools import partial
-from typing import TYPE_CHECKING
-
-from fastapi.encoders import jsonable_encoder
-from typing_extensions import Protocol
-
-from langflow.logging.logger import logger
-from langflow.schema.playground_events import create_event_by_type
-
-if TYPE_CHECKING:
-    import asyncio
-
-    from langflow.schema.log import LoggableType
-
-
-class EventCallback(Protocol):
-    def __call__(self, *, manager: EventManager, event_type: str, data: LoggableType): ...
-
-
-class PartialEventCallback(Protocol):
-    def __call__(self, *, data: LoggableType): ...
-
-
-class EventManager:
-    def __init__(self, queue: asyncio.Queue):
-        self.queue = queue
-        self.events: dict[str, PartialEventCallback] = {}
-
-    @staticmethod
-    def _validate_callback(callback: EventCallback) -> None:
-        if not callable(callback):
-            msg = "Callback must be callable"
-            raise TypeError(msg)
-        # Check if it has `self, event_type and data`
-        sig = inspect.signature(callback)
-        parameters = ["manager", "event_type", "data"]
-        if len(sig.parameters) != len(parameters):
-            msg = "Callback must have exactly 3 parameters"
-            raise ValueError(msg)
-        if not all(param.name in parameters for param in sig.parameters.values()):
-            msg = "Callback must have exactly 3 parameters: manager, event_type, and data"
-            raise ValueError(msg)
-
-    def register_event(
-        self,
-        name: str,
-        event_type: str,
-        callback: EventCallback | None = None,
-    ) -> None:
-        if not name:
-            msg = "Event name cannot be empty"
-            raise ValueError(msg)
-        if not name.startswith("on_"):
-            msg = "Event name must start with 'on_'"
-            raise ValueError(msg)
-        if callback is None:
-            callback_ = partial(self.send_event, event_type=event_type)
-        else:
-            callback_ = partial(callback, manager=self, event_type=event_type)
-        self.events[name] = callback_
-
-    def send_event(self, *, event_type: str, data: LoggableType):
-        try:
-            if isinstance(data, dict) and event_type in {"message", "error", "warning", "info", "token"}:
-                data = create_event_by_type(event_type, **data)
-        except TypeError as e:
-            logger.debug(f"Error creating playground event: {e}")
-        except Exception:
-            raise
-        jsonable_data = jsonable_encoder(data)
-        json_data = {"event": event_type, "data": jsonable_data}
-        event_id = f"{event_type}-{uuid.uuid4()}"
-        str_data = json.dumps(json_data) + "\n\n"
-        self.queue.put_nowait((event_id, str_data.encode("utf-8"), time.time()))
-
-    def noop(self, *, data: LoggableType) -> None:
-        pass
-
-    def __getattr__(self, name: str) -> PartialEventCallback:
-        return self.events.get(name, self.noop)
-
-
-def create_default_event_manager(queue):
-    manager = EventManager(queue)
-    manager.register_event("on_token", "token")
-    manager.register_event("on_vertices_sorted", "vertices_sorted")
-    manager.register_event("on_error", "error")
-    manager.register_event("on_end", "end")
-    manager.register_event("on_message", "add_message")
-    manager.register_event("on_remove_message", "remove_message")
-    manager.register_event("on_end_vertex", "end_vertex")
-    manager.register_event("on_build_start", "build_start")
-    manager.register_event("on_build_end", "build_end")
-    return manager
-
-
-def create_stream_tokens_event_manager(queue):
-    manager = EventManager(queue)
-    manager.register_event("on_message", "add_message")
-    manager.register_event("on_token", "token")
-    manager.register_event("on_end", "end")
-    return manager
+# Backwards compatibility module for langflow.events.event_manager
+# This module redirects imports to the new lfx.events.event_manager module
+
+from lfx.events.event_manager import (
+    EventCallback,
+    EventManager,
+    PartialEventCallback,
+    create_default_event_manager,
+    create_stream_tokens_event_manager,
+)
+
+__all__ = [
+    "EventCallback",
+    "EventManager",
+    "PartialEventCallback",
+    "create_default_event_manager",
+    "create_stream_tokens_event_manager",
+]

langflow/field_typing/__init__.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from .constants import (
+from lfx.field_typing.constants import (
     AgentExecutor,
     BaseChatMemory,
     BaseChatModel,
@@ -29,17 +29,17 @@ from .constants import (
     Tool,
     VectorStore,
 )
-from .range_spec import RangeSpec
+from lfx.field_typing.range_spec import RangeSpec
 
 
 def _import_input_class():
-    from langflow.template.field.base import Input
+    from lfx.template.field.base import Input
 
     return Input
 
 
 def _import_output_class():
-    from langflow.template.field.base import Output
+    from lfx.template.field.base import Output
 
     return Output
 
@@ -48,9 +48,10 @@ def __getattr__(name: str) -> Any:
     # This is to avoid circular imports
     if name == "Input":
         return _import_input_class()
-        return RangeSpec
     if name == "Output":
         return _import_output_class()
+    if name == "RangeSpec":
+        return RangeSpec
     # The other names should work as if they were imported from constants
     # Import the constants module langflow.field_typing.constants
     from . import constants
@@ -77,7 +78,6 @@ __all__ = [
     "Data",
     "Document",
     "Embeddings",
-    "Input",
     "LanguageModel",
     "NestedDict",
     "Object",

langflow/field_typing/constants.py

@@ -1,135 +1,100 @@
+# Re-export everything from lfx.field_typing.constants for backward compatibility
+# Import additional types
 from collections.abc import Callable
-from typing import Text, TypeAlias, TypeVar
+from typing import Text
 
-from langchain.agents.agent import AgentExecutor
-from langchain.chains.base import Chain
-from langchain.memory.chat_memory import BaseChatMemory
-from langchain_core.chat_history import BaseChatMessageHistory
-from langchain_core.document_loaders import BaseLoader
-from langchain_core.documents import Document
-from langchain_core.documents.compressor import BaseDocumentCompressor
-from langchain_core.embeddings import Embeddings
-from langchain_core.language_models import BaseLanguageModel, BaseLLM
-from langchain_core.language_models.chat_models import BaseChatModel
-from langchain_core.memory import BaseMemory
-from langchain_core.output_parsers import BaseLLMOutputParser, BaseOutputParser
-from langchain_core.prompts import BasePromptTemplate, ChatPromptTemplate, PromptTemplate
-from langchain_core.retrievers import BaseRetriever
-from langchain_core.tools import BaseTool, Tool
-from langchain_core.vectorstores import VectorStore, VectorStoreRetriever
-from langchain_text_splitters import TextSplitter
-
-from langflow.schema.data import Data
-from langflow.schema.dataframe import DataFrame
-from langflow.schema.message import Message
-
-NestedDict: TypeAlias = dict[str, str | dict]
-LanguageModel = TypeVar("LanguageModel", BaseLanguageModel, BaseLLM, BaseChatModel)
-ToolEnabledLanguageModel = TypeVar("ToolEnabledLanguageModel", BaseLanguageModel, BaseLLM, BaseChatModel)
-Memory = TypeVar("Memory", bound=BaseChatMessageHistory)
-
-Retriever = TypeVar(
-    "Retriever",
+from lfx.field_typing.constants import (
+    CUSTOM_COMPONENT_SUPPORTED_TYPES,
+    DEFAULT_IMPORT_STRING,
+    LANGCHAIN_BASE_TYPES,
+    # Import all the langchain types that may be needed
+    AgentExecutor,
+    BaseChatMemory,
+    BaseChatMessageHistory,
+    BaseChatModel,
+    BaseDocumentCompressor,
+    BaseLanguageModel,
+    BaseLLM,
+    BaseLLMOutputParser,
+    BaseLoader,
+    BaseMemory,
+    BaseOutputParser,
+    BasePromptTemplate,
     BaseRetriever,
+    BaseTool,
+    Chain,
+    ChatPromptTemplate,
+    Code,
+    Document,
+    Embeddings,
+    LanguageModel,
+    Memory,
+    NestedDict,
+    Object,
+    OutputParser,
+    PromptTemplate,
+    Retriever,
+    TextSplitter,
+    Tool,
+    ToolEnabledLanguageModel,
+    VectorStore,
     VectorStoreRetriever,
 )
-OutputParser = TypeVar(
-    "OutputParser",
-    BaseOutputParser,
-    BaseLLMOutputParser,
-)
 
+# Import lfx schema types
+from lfx.schema.data import Data
+from lfx.schema.dataframe import DataFrame
 
-class Object:
-    pass
-
-
-class Code:
-    pass
-
+# Import Message from langflow.schema for backward compatibility
+from langflow.schema.message import Message
 
-LANGCHAIN_BASE_TYPES = {
-    "Chain": Chain,
-    "AgentExecutor": AgentExecutor,
-    "BaseTool": BaseTool,
-    "Tool": Tool,
-    "BaseLLM": BaseLLM,
-    "BaseLanguageModel": BaseLanguageModel,
-    "PromptTemplate": PromptTemplate,
-    "ChatPromptTemplate": ChatPromptTemplate,
-    "BasePromptTemplate": BasePromptTemplate,
-    "BaseLoader": BaseLoader,
-    "Document": Document,
-    "TextSplitter": TextSplitter,
-    "VectorStore": VectorStore,
-    "Embeddings": Embeddings,
-    "BaseRetriever": BaseRetriever,
-    "BaseOutputParser": BaseOutputParser,
-    "BaseMemory": BaseMemory,
-    "BaseChatMemory": BaseChatMemory,
-    "BaseChatModel": BaseChatModel,
-    "Memory": Memory,
-    "BaseDocumentCompressor": BaseDocumentCompressor,
-}
-# Langchain base types plus Python base types
+# Add Message and DataFrame to CUSTOM_COMPONENT_SUPPORTED_TYPES
 CUSTOM_COMPONENT_SUPPORTED_TYPES = {
-    **LANGCHAIN_BASE_TYPES,
-    "NestedDict": NestedDict,
-    "Data": Data,
+    **CUSTOM_COMPONENT_SUPPORTED_TYPES,
     "Message": Message,
-    "Text": Text,  # noqa: UP019
-    "Object": Object,
-    "Callable": Callable,
-    "LanguageModel": LanguageModel,
-    "Retriever": Retriever,
     "DataFrame": DataFrame,
 }
 
-DEFAULT_IMPORT_STRING = """from langchain.agents.agent import AgentExecutor
-from langchain.chains.base import Chain
-from langchain.memory.chat_memory import BaseChatMemory
-from langchain_core.chat_history import BaseChatMessageHistory
-from langchain_core.document_loaders import BaseLoader
-from langchain_core.documents import Document
-from langchain_core.embeddings import Embeddings
-from langchain_core.language_models import BaseLanguageModel, BaseLLM
-from langchain_core.language_models.chat_models import BaseChatModel
-from langchain_core.memory import BaseMemory
-from langchain_core.output_parsers import BaseLLMOutputParser, BaseOutputParser
-from langchain_core.prompts import BasePromptTemplate, ChatPromptTemplate, PromptTemplate
-from langchain_core.retrievers import BaseRetriever
-from langchain_core.documents.compressor import BaseDocumentCompressor
-from langchain_core.tools import BaseTool, Tool
-from langchain_core.vectorstores import VectorStore, VectorStoreRetriever
-from langchain_text_splitters import TextSplitter
-
-from langflow.io import (
-    BoolInput,
-    CodeInput,
-    DataFrameInput,
-    DataInput,
-    DefaultPromptField,
-    DictInput,
-    DropdownInput,
-    FileInput,
-    FloatInput,
-    HandleInput,
-    IntInput,
-    LinkInput,
-    MessageInput,
-    MessageTextInput,
-    MultilineInput,
-    MultilineSecretInput,
-    MultiselectInput,
-    NestedDictInput,
-    Output,
-    PromptInput,
-    SecretStrInput,
-    SliderInput,
-    StrInput,
-    TableInput,
-)
-from langflow.schema.data import Data
-from langflow.schema.dataframe import DataFrame
-from langflow.schema.message import Message
-"""
+__all__ = [
+    "CUSTOM_COMPONENT_SUPPORTED_TYPES",
+    "DEFAULT_IMPORT_STRING",
+    "LANGCHAIN_BASE_TYPES",
+    # Langchain types
+    "AgentExecutor",
+    "BaseChatMemory",
+    "BaseChatMessageHistory",
+    "BaseChatModel",
+    "BaseDocumentCompressor",
+    "BaseLLM",
+    "BaseLLMOutputParser",
+    "BaseLanguageModel",
+    "BaseLoader",
+    "BaseMemory",
+    "BaseOutputParser",
+    "BasePromptTemplate",
+    "BaseRetriever",
+    "BaseTool",
+    # Additional types
+    "Callable",
+    "Chain",
+    "ChatPromptTemplate",
+    "Code",
+    "Data",
+    "DataFrame",
+    "Document",
+    "Embeddings",
+    "LanguageModel",
+    "Memory",
+    "Message",
+    "NestedDict",
+    "Object",
+    "OutputParser",
+    "PromptTemplate",
+    "Retriever",
+    "Text",
+    "TextSplitter",
+    "Tool",
+    "ToolEnabledLanguageModel",
+    "VectorStore",
+    "VectorStoreRetriever",
+]

langflow/field_typing/range_spec.py

@@ -1,33 +1,3 @@
-from typing import Literal
+from lfx.field_typing.range_spec import RangeSpec
 
-from pydantic import BaseModel, field_validator
-
-
-class RangeSpec(BaseModel):
-    step_type: Literal["int", "float"] = "float"
-    min: float = -1.0
-    max: float = 1.0
-    step: float = 0.1
-
-    @field_validator("max")
-    @classmethod
-    def max_must_be_greater_than_min(cls, v, values):
-        if "min" in values.data and v <= values.data["min"]:
-            msg = "Max must be greater than min"
-            raise ValueError(msg)
-        return v
-
-    @field_validator("step")
-    @classmethod
-    def step_must_be_positive(cls, v, values):
-        if v <= 0:
-            msg = "Step must be positive"
-            raise ValueError(msg)
-        if values.data["step_type"] == "int" and isinstance(v, float) and not v.is_integer():
-            msg = "When step_type is int, step must be an integer"
-            raise ValueError(msg)
-        return v
-
-    @classmethod
-    def set_step_type(cls, step_type: Literal["int", "float"], range_spec: "RangeSpec") -> "RangeSpec":
-        return cls(min=range_spec.min, max=range_spec.max, step=range_spec.step, step_type=step_type)
+__all__ = ["RangeSpec"]

langflow/frontend/assets/{SlackIcon-Cc7Qnzki.js → SlackIcon-v88osOTA.js}

@@ -1 +1 @@
-import{j as a}from"./index-D234yKNJ.js";const s=l=>a.jsx("svg",{xmlns:"http://www.w3.org/2000/svg",viewBox:"0 0 54 54",width:"1em",height:"1em",...l,children:a.jsxs("g",{fill:"none",fillRule:"evenodd",children:[a.jsx("path",{fill:"#36C5F0",d:"M19.712.133a5.381 5.381 0 0 0-5.376 5.387 5.381 5.381 0 0 0 5.376 5.386h5.376V5.52A5.381 5.381 0 0 0 19.712.133m0 14.365H5.376A5.381 5.381 0 0 0 0 19.884a5.381 5.381 0 0 0 5.376 5.387h14.336a5.381 5.381 0 0 0 5.376-5.387 5.381 5.381 0 0 0-5.376-5.386"}),a.jsx("path",{fill:"#2EB67D",d:"M53.76 19.884a5.381 5.381 0 0 0-5.376-5.386 5.381 5.381 0 0 0-5.376 5.386v5.387h5.376a5.381 5.381 0 0 0 5.376-5.387m-14.336 0V5.52A5.381 5.381 0 0 0 34.048.133a5.381 5.381 0 0 0-5.376 5.387v14.364a5.381 5.381 0 0 0 5.376 5.387 5.381 5.381 0 0 0 5.376-5.387"}),a.jsx("path",{fill:"#ECB22E",d:"M34.048 54a5.381 5.381 0 0 0 5.376-5.387 5.381 5.381 0 0 0-5.376-5.386h-5.376v5.386A5.381 5.381 0 0 0 34.048 54m0-14.365h14.336a5.381 5.381 0 0 0 5.376-5.386 5.381 5.381 0 0 0-5.376-5.387H34.048a5.381 5.381 0 0 0-5.376 5.387 5.381 5.381 0 0 0 5.376 5.386"}),a.jsx("path",{fill:"#E01E5A",d:"M0 34.249a5.381 5.381 0 0 0 5.376 5.386 5.381 5.381 0 0 0 5.376-5.386v-5.387H5.376A5.381 5.381 0 0 0 0 34.25m14.336-.001v14.364A5.381 5.381 0 0 0 19.712 54a5.381 5.381 0 0 0 5.376-5.387V34.25a5.381 5.381 0 0 0-5.376-5.387 5.381 5.381 0 0 0-5.376 5.387"})]})});export{s as default};
+import{j as a}from"./index-pBO0SZLD.js";const s=l=>a.jsx("svg",{xmlns:"http://www.w3.org/2000/svg",viewBox:"0 0 54 54",width:"1em",height:"1em",...l,children:a.jsxs("g",{fill:"none",fillRule:"evenodd",children:[a.jsx("path",{fill:"#36C5F0",d:"M19.712.133a5.381 5.381 0 0 0-5.376 5.387 5.381 5.381 0 0 0 5.376 5.386h5.376V5.52A5.381 5.381 0 0 0 19.712.133m0 14.365H5.376A5.381 5.381 0 0 0 0 19.884a5.381 5.381 0 0 0 5.376 5.387h14.336a5.381 5.381 0 0 0 5.376-5.387 5.381 5.381 0 0 0-5.376-5.386"}),a.jsx("path",{fill:"#2EB67D",d:"M53.76 19.884a5.381 5.381 0 0 0-5.376-5.386 5.381 5.381 0 0 0-5.376 5.386v5.387h5.376a5.381 5.381 0 0 0 5.376-5.387m-14.336 0V5.52A5.381 5.381 0 0 0 34.048.133a5.381 5.381 0 0 0-5.376 5.387v14.364a5.381 5.381 0 0 0 5.376 5.387 5.381 5.381 0 0 0 5.376-5.387"}),a.jsx("path",{fill:"#ECB22E",d:"M34.048 54a5.381 5.381 0 0 0 5.376-5.387 5.381 5.381 0 0 0-5.376-5.386h-5.376v5.386A5.381 5.381 0 0 0 34.048 54m0-14.365h14.336a5.381 5.381 0 0 0 5.376-5.386 5.381 5.381 0 0 0-5.376-5.387H34.048a5.381 5.381 0 0 0-5.376 5.387 5.381 5.381 0 0 0 5.376 5.386"}),a.jsx("path",{fill:"#E01E5A",d:"M0 34.249a5.381 5.381 0 0 0 5.376 5.386 5.381 5.381 0 0 0 5.376-5.386v-5.387H5.376A5.381 5.381 0 0 0 0 34.25m14.336-.001v14.364A5.381 5.381 0 0 0 19.712 54a5.381 5.381 0 0 0 5.376-5.387V34.25a5.381 5.381 0 0 0-5.376-5.387 5.381 5.381 0 0 0-5.376 5.387"})]})});export{s as default};