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

langflow/utils/constants.py

@@ -1,205 +1,29 @@
-from typing import Any
-
-OPENAI_MODELS = [
-    "text-davinci-003",
-    "text-davinci-002",
-    "text-curie-001",
-    "text-babbage-001",
-    "text-ada-001",
-]
-CHAT_OPENAI_MODELS = [
-    "gpt-4o",
-    "gpt-4o-mini",
-    "gpt-4-turbo-preview",
-    "gpt-4-0125-preview",
-    "gpt-4-1106-preview",
-    "gpt-4-vision-preview",
-    "gpt-3.5-turbo-0125",
-    "gpt-3.5-turbo-1106",
-]
-
-REASONING_OPENAI_MODELS = [
-    "o1",
-    "o1-mini",
-    "o1-pro",
-    "o3-mini",
-    "o3",
-    "o3-pro",
-    "o4-mini",
-    "o4-mini-high",
+from lfx.utils.constants import (
+    ANTHROPIC_MODELS,
+    CHAT_OPENAI_MODELS,
+    DEFAULT_PYTHON_FUNCTION,
+    DIRECT_TYPES,
+    LOADERS_INFO,
+    MESSAGE_SENDER_AI,
+    MESSAGE_SENDER_NAME_AI,
+    MESSAGE_SENDER_NAME_USER,
+    MESSAGE_SENDER_USER,
+    OPENAI_MODELS,
+    PYTHON_BASIC_TYPES,
+    REASONING_OPENAI_MODELS,
+)
+
+__all__ = [
+    "ANTHROPIC_MODELS",
+    "CHAT_OPENAI_MODELS",
+    "DEFAULT_PYTHON_FUNCTION",
+    "DIRECT_TYPES",
+    "LOADERS_INFO",
+    "MESSAGE_SENDER_AI",
+    "MESSAGE_SENDER_NAME_AI",
+    "MESSAGE_SENDER_NAME_USER",
+    "MESSAGE_SENDER_USER",
+    "OPENAI_MODELS",
+    "PYTHON_BASIC_TYPES",
+    "REASONING_OPENAI_MODELS",
 ]
-
-ANTHROPIC_MODELS = [
-    # largest model, ideal for a wide range of more complex tasks.
-    "claude-v1",
-    # An enhanced version of claude-v1 with a 100,000 token (roughly 75,000 word) context window.
-    "claude-v1-100k",
-    # A smaller model with far lower latency, sampling at roughly 40 words/sec!
-    "claude-instant-v1",
-    # Like claude-instant-v1 with a 100,000 token context window but retains its performance.
-    "claude-instant-v1-100k",
-    # Specific sub-versions of the above models:
-    # Vs claude-v1.2: better instruction-following, code, and non-English dialogue and writing.
-    "claude-v1.3",
-    # An enhanced version of claude-v1.3 with a 100,000 token (roughly 75,000 word) context window.
-    "claude-v1.3-100k",
-    # Vs claude-v1.1: small adv in general helpfulness, instruction following, coding, and other tasks.
-    "claude-v1.2",
-    # An earlier version of claude-v1.
-    "claude-v1.0",
-    # Latest version of claude-instant-v1. Better than claude-instant-v1.0 at most tasks.
-    "claude-instant-v1.1",
-    # Version of claude-instant-v1.1 with a 100K token context window.
-    "claude-instant-v1.1-100k",
-    # An earlier version of claude-instant-v1.
-    "claude-instant-v1.0",
-]
-
-DEFAULT_PYTHON_FUNCTION = """
-def python_function(text: str) -> str:
-    \"\"\"This is a default python function that returns the input text\"\"\"
-    return text
-"""
-
-
-PYTHON_BASIC_TYPES = [str, bool, int, float, tuple, list, dict, set]
-DIRECT_TYPES = [
-    "str",
-    "bool",
-    "dict",
-    "int",
-    "float",
-    "Any",
-    "prompt",
-    "code",
-    "NestedDict",
-    "table",
-    "slider",
-    "tab",
-    "sortableList",
-    "auth",
-    "connect",
-    "query",
-    "tools",
-    "mcp",
-]
-
-
-LOADERS_INFO: list[dict[str, Any]] = [
-    {
-        "loader": "AirbyteJSONLoader",
-        "name": "Airbyte JSON (.jsonl)",
-        "import": "langchain_community.document_loaders.AirbyteJSONLoader",
-        "defaultFor": ["jsonl"],
-        "allowdTypes": ["jsonl"],
-    },
-    {
-        "loader": "JSONLoader",
-        "name": "JSON (.json)",
-        "import": "langchain_community.document_loaders.JSONLoader",
-        "defaultFor": ["json"],
-        "allowdTypes": ["json"],
-    },
-    {
-        "loader": "BSHTMLLoader",
-        "name": "BeautifulSoup4 HTML (.html, .htm)",
-        "import": "langchain_community.document_loaders.BSHTMLLoader",
-        "allowdTypes": ["html", "htm"],
-    },
-    {
-        "loader": "CSVLoader",
-        "name": "CSV (.csv)",
-        "import": "langchain_community.document_loaders.CSVLoader",
-        "defaultFor": ["csv"],
-        "allowdTypes": ["csv"],
-    },
-    {
-        "loader": "CoNLLULoader",
-        "name": "CoNLL-U (.conllu)",
-        "import": "langchain_community.document_loaders.CoNLLULoader",
-        "defaultFor": ["conllu"],
-        "allowdTypes": ["conllu"],
-    },
-    {
-        "loader": "EverNoteLoader",
-        "name": "EverNote (.enex)",
-        "import": "langchain_community.document_loaders.EverNoteLoader",
-        "defaultFor": ["enex"],
-        "allowdTypes": ["enex"],
-    },
-    {
-        "loader": "FacebookChatLoader",
-        "name": "Facebook Chat (.json)",
-        "import": "langchain_community.document_loaders.FacebookChatLoader",
-        "allowdTypes": ["json"],
-    },
-    {
-        "loader": "OutlookMessageLoader",
-        "name": "Outlook Message (.msg)",
-        "import": "langchain_community.document_loaders.OutlookMessageLoader",
-        "defaultFor": ["msg"],
-        "allowdTypes": ["msg"],
-    },
-    {
-        "loader": "PyPDFLoader",
-        "name": "PyPDF (.pdf)",
-        "import": "langchain_community.document_loaders.PyPDFLoader",
-        "defaultFor": ["pdf"],
-        "allowdTypes": ["pdf"],
-    },
-    {
-        "loader": "STRLoader",
-        "name": "Subtitle (.str)",
-        "import": "langchain_community.document_loaders.STRLoader",
-        "defaultFor": ["str"],
-        "allowdTypes": ["str"],
-    },
-    {
-        "loader": "TextLoader",
-        "name": "Text (.txt)",
-        "import": "langchain_community.document_loaders.TextLoader",
-        "defaultFor": ["txt"],
-        "allowdTypes": ["txt"],
-    },
-    {
-        "loader": "UnstructuredEmailLoader",
-        "name": "Unstructured Email (.eml)",
-        "import": "langchain_community.document_loaders.UnstructuredEmailLoader",
-        "defaultFor": ["eml"],
-        "allowdTypes": ["eml"],
-    },
-    {
-        "loader": "UnstructuredHTMLLoader",
-        "name": "Unstructured HTML (.html, .htm)",
-        "import": "langchain_community.document_loaders.UnstructuredHTMLLoader",
-        "defaultFor": ["html", "htm"],
-        "allowdTypes": ["html", "htm"],
-    },
-    {
-        "loader": "UnstructuredMarkdownLoader",
-        "name": "Unstructured Markdown (.md)",
-        "import": "langchain_community.document_loaders.UnstructuredMarkdownLoader",
-        "defaultFor": ["md", "mdx"],
-        "allowdTypes": ["md", "mdx"],
-    },
-    {
-        "loader": "UnstructuredPowerPointLoader",
-        "name": "Unstructured PowerPoint (.pptx)",
-        "import": "langchain_community.document_loaders.UnstructuredPowerPointLoader",
-        "defaultFor": ["pptx"],
-        "allowdTypes": ["pptx"],
-    },
-    {
-        "loader": "UnstructuredWordLoader",
-        "name": "Unstructured Word (.docx)",
-        "import": "langchain_community.document_loaders.UnstructuredWordLoader",
-        "defaultFor": ["docx"],
-        "allowdTypes": ["docx"],
-    },
-]
-
-
-MESSAGE_SENDER_AI = "Machine"
-MESSAGE_SENDER_USER = "User"
-MESSAGE_SENDER_NAME_AI = "AI"
-MESSAGE_SENDER_NAME_USER = "User"

langflow/utils/lazy_load.py

@@ -1,15 +1,4 @@
-class LazyLoadDictBase:
-    def __init__(self) -> None:
-        self._all_types_dict = None
+"""Lazy load utilities for langflow - imports from lfx."""
 
-    @property
-    def all_types_dict(self):
-        if self._all_types_dict is None:
-            self._all_types_dict = self._build_dict()
-        return self._all_types_dict
-
-    def _build_dict(self):
-        raise NotImplementedError
-
-    def get_type_dict(self):
-        raise NotImplementedError
+# Import everything from lfx.utils.lazy_load
+from lfx.utils.lazy_load import *  # noqa: F403

langflow/utils/schemas.py

@@ -1,12 +1,11 @@
 import enum
 
 from langchain_core.messages import BaseMessage
+from lfx.base.data.utils import IMG_FILE_TYPES, TEXT_FILE_TYPES
+from lfx.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI
 from pydantic import BaseModel, field_validator, model_validator
 from typing_extensions import TypedDict
 
-from langflow.base.data.utils import IMG_FILE_TYPES, TEXT_FILE_TYPES
-from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI
-
 
 class File(TypedDict):
     """File schema."""

langflow/utils/template_validation.py

@@ -9,8 +9,8 @@ import json
 import uuid
 from typing import Any
 
-from langflow.graph.graph.base import Graph
-from langflow.utils.validate import validate_code
+from lfx.custom.validate import validate_code
+from lfx.graph.graph.base import Graph
 
 
 def validate_template_structure(template_data: dict[str, Any], filename: str) -> list[str]: