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

langflow/api/utils.py

@@ -8,11 +8,11 @@ from typing import TYPE_CHECKING, Annotated, Any
 
 from fastapi import Depends, HTTPException, Query
 from fastapi_pagination import Params
+from lfx.graph.graph.base import Graph
+from lfx.log.logger import logger
 from sqlalchemy import delete
 from sqlmodel.ext.asyncio.session import AsyncSession
 
-from langflow.graph.graph.base import Graph
-from langflow.logging.logger import logger
 from langflow.services.auth.utils import get_current_active_user, get_current_active_user_mcp
 from langflow.services.database.models.flow.model import Flow
 from langflow.services.database.models.message.model import MessageTable

langflow/api/v1/base.py

@@ -1,7 +1,6 @@
+from lfx.template.frontend_node.base import FrontendNode
 from pydantic import BaseModel, field_validator, model_serializer
 
-from langflow.template.frontend_node.base import FrontendNode
-
 
 class CacheResponse(BaseModel):
     data: dict

langflow/api/v1/callback.py

@@ -1,19 +1,16 @@
 from __future__ import annotations
 
-from typing import TYPE_CHECKING, Any
+from typing import Any
 from uuid import UUID
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.callbacks.base import AsyncCallbackHandler
+from lfx.log.logger import logger
+from lfx.utils.util import remove_ansi_escape_codes
 from typing_extensions import override
 
 from langflow.api.v1.schemas import ChatResponse, PromptResponse
-from langflow.logging.logger import logger
-from langflow.services.deps import get_chat_service, get_socket_service
-from langflow.utils.util import remove_ansi_escape_codes
-
-if TYPE_CHECKING:
-    from langflow.services.socket.service import SocketIOService
+from langflow.services.deps import get_chat_service
 
 
 # https://github.com/hwchase17/chat-langchain/blob/master/callback.py
@@ -28,9 +25,7 @@ class AsyncStreamingLLMCallbackHandleSIO(AsyncCallbackHandler):
     def __init__(self, session_id: str):
         self.chat_service = get_chat_service()
         self.client_id = session_id
-        self.socketio_service: SocketIOService = get_socket_service()
         self.sid = session_id
-        # self.socketio_service = self.chat_service.active_connections[self.client_id]
 
     @override
     async def on_llm_new_token(self, token: str, **kwargs: Any) -> None:  # type: ignore[misc]

langflow/api/v1/chat.py

@@ -8,6 +8,11 @@ from typing import TYPE_CHECKING, Annotated
 
 from fastapi import APIRouter, BackgroundTasks, Body, Depends, HTTPException, Request, status
 from fastapi.responses import StreamingResponse
+from lfx.graph.graph.base import Graph
+from lfx.graph.utils import log_vertex_build
+from lfx.log.logger import logger
+from lfx.schema.schema import InputValueRequest, OutputValue
+from lfx.services.cache.utils import CacheMiss
 
 from langflow.api.build import cancel_flow_build, get_flow_events_response, start_flow_build
 from langflow.api.limited_background_tasks import LimitVertexBuildBackgroundTasks
@@ -26,18 +31,12 @@ from langflow.api.utils import (
 from langflow.api.v1.schemas import (
     CancelFlowResponse,
     FlowDataRequest,
-    InputValueRequest,
     ResultDataResponse,
     StreamData,
     VertexBuildResponse,
     VerticesOrderResponse,
 )
 from langflow.exceptions.component import ComponentBuildError
-from langflow.graph.graph.base import Graph
-from langflow.graph.utils import log_vertex_build
-from langflow.logging.logger import logger
-from langflow.schema.schema import OutputValue
-from langflow.services.cache.utils import CacheMiss
 from langflow.services.chat.service import ChatService
 from langflow.services.database.models.flow.model import Flow
 from langflow.services.deps import (
@@ -51,7 +50,7 @@ from langflow.services.job_queue.service import JobQueueNotFoundError, JobQueueS
 from langflow.services.telemetry.schema import ComponentPayload, PlaygroundPayload
 
 if TYPE_CHECKING:
-    from langflow.graph.vertex.vertex_types import InterfaceVertex
+    from lfx.graph.vertex.vertex_types import InterfaceVertex
 
 router = APIRouter(tags=["Chat"])
 

langflow/api/v1/endpoints.py

@@ -11,6 +11,18 @@ import sqlalchemy as sa
 from fastapi import APIRouter, BackgroundTasks, Body, Depends, HTTPException, Request, UploadFile, status
 from fastapi.encoders import jsonable_encoder
 from fastapi.responses import StreamingResponse
+from lfx.custom.custom_component.component import Component
+from lfx.custom.utils import (
+    add_code_field_to_build_config,
+    build_custom_component_template,
+    get_instance_name,
+    update_component_build_config,
+)
+from lfx.graph.graph.base import Graph
+from lfx.graph.schema import RunOutputs
+from lfx.log.logger import logger
+from lfx.schema.schema import InputValueRequest
+from lfx.services.settings.service import SettingsService
 from sqlmodel import select
 
 from langflow.api.utils import CurrentActiveUser, DbSession, parse_value
@@ -18,28 +30,17 @@ from langflow.api.v1.schemas import (
     ConfigResponse,
     CustomComponentRequest,
     CustomComponentResponse,
-    InputValueRequest,
     RunResponse,
     SimplifiedAPIRequest,
     TaskStatusResponse,
     UpdateCustomComponentRequest,
     UploadFileResponse,
 )
-from langflow.custom.custom_component.component import Component
-from langflow.custom.utils import (
-    add_code_field_to_build_config,
-    build_custom_component_template,
-    get_instance_name,
-    update_component_build_config,
-)
 from langflow.events.event_manager import create_stream_tokens_event_manager
 from langflow.exceptions.api import APIException, InvalidChatInputError
 from langflow.exceptions.serialization import SerializationError
-from langflow.graph.graph.base import Graph
-from langflow.graph.schema import RunOutputs
 from langflow.helpers.flow import get_flow_by_id_or_endpoint_name
 from langflow.interface.initialize.loading import update_params_with_load_from_db_fields
-from langflow.logging.logger import logger
 from langflow.processing.process import process_tweaks, run_graph_internal
 from langflow.schema.graph import Tweaks
 from langflow.services.auth.utils import api_key_security, get_current_active_user, get_webhook_user
@@ -54,7 +55,6 @@ from langflow.utils.version import get_version_info
 
 if TYPE_CHECKING:
     from langflow.events.event_manager import EventManager
-    from langflow.services.settings.service import SettingsService
 
 router = APIRouter(tags=["Base"])
 
@@ -723,9 +723,9 @@ async def custom_component_update(
                 for field_name, field_dict in template.items()
                 if isinstance(field_dict, dict) and field_dict.get("load_from_db") and field_dict.get("value")
             ]
-
-            params = await update_params_with_load_from_db_fields(cc_instance, params, load_from_db_fields)
-            cc_instance.set_attributes(params)
+            if isinstance(cc_instance, Component):
+                params = await update_params_with_load_from_db_fields(cc_instance, params, load_from_db_fields)
+                cc_instance.set_attributes(params)
         updated_build_config = code_request.get_template()
         await update_component_build_config(
             cc_instance,

langflow/api/v1/files.py

@@ -8,12 +8,12 @@ from uuid import UUID
 
 from fastapi import APIRouter, Depends, HTTPException, UploadFile
 from fastapi.responses import StreamingResponse
+from lfx.services.settings.service import SettingsService
 
 from langflow.api.utils import CurrentActiveUser, DbSession
 from langflow.api.v1.schemas import UploadFileResponse
 from langflow.services.database.models.flow.model import Flow
 from langflow.services.deps import get_settings_service, get_storage_service
-from langflow.services.settings.service import SettingsService
 from langflow.services.storage.service import StorageService
 from langflow.services.storage.utils import build_content_type_from_extension
 

langflow/api/v1/flows.py

@@ -16,6 +16,7 @@ from fastapi.encoders import jsonable_encoder
 from fastapi.responses import StreamingResponse
 from fastapi_pagination import Page, Params
 from fastapi_pagination.ext.sqlmodel import apaginate
+from lfx.log import logger
 from sqlmodel import and_, col, select
 from sqlmodel.ext.asyncio.session import AsyncSession
 
@@ -23,7 +24,6 @@ from langflow.api.utils import CurrentActiveUser, DbSession, cascade_delete_flow
 from langflow.api.v1.schemas import FlowListCreate
 from langflow.helpers.user import get_user_by_flow_id_or_endpoint_name
 from langflow.initial_setup.constants import STARTER_FOLDER_NAME
-from langflow.logging import logger
 from langflow.services.database.models.flow.model import (
     AccessTypeEnum,
     Flow,

langflow/api/v1/knowledge_bases.py

@@ -6,10 +6,10 @@ from pathlib import Path
 import pandas as pd
 from fastapi import APIRouter, HTTPException
 from langchain_chroma import Chroma
+from lfx.log import logger
 from pydantic import BaseModel
 
 from langflow.api.utils import CurrentActiveUser
-from langflow.logging import logger
 from langflow.services.deps import get_settings_service
 
 router = APIRouter(tags=["Knowledge Bases"], prefix="/knowledge_bases")

langflow/api/v1/mcp.py

@@ -4,6 +4,7 @@ import pydantic
 from anyio import BrokenResourceError
 from fastapi import APIRouter, HTTPException, Request, Response
 from fastapi.responses import HTMLResponse, StreamingResponse
+from lfx.log.logger import logger
 from mcp import types
 from mcp.server import NotificationOptions, Server
 from mcp.server.sse import SseServerTransport
@@ -17,7 +18,6 @@ from langflow.api.v1.mcp_utils import (
     handle_mcp_errors,
     handle_read_resource,
 )
-from langflow.logging.logger import logger
 from langflow.services.deps import get_settings_service
 
 router = APIRouter(prefix="/mcp", tags=["mcp"])

langflow/api/v1/mcp_projects.py

@@ -14,6 +14,11 @@ from uuid import UUID
 from anyio import BrokenResourceError
 from fastapi import APIRouter, Depends, HTTPException, Request, Response
 from fastapi.responses import HTMLResponse
+from lfx.base.mcp.constants import MAX_MCP_SERVER_NAME_LENGTH
+from lfx.base.mcp.util import sanitize_mcp_name
+from lfx.log import logger
+from lfx.services.deps import get_settings_service, session_scope
+from lfx.services.settings.feature_flags import FEATURE_FLAGS
 from mcp import types
 from mcp.server import NotificationOptions, Server
 from mcp.server.sse import SseServerTransport
@@ -36,16 +41,11 @@ from langflow.api.v1.schemas import (
     MCPProjectUpdateRequest,
     MCPSettings,
 )
-from langflow.base.mcp.constants import MAX_MCP_SERVER_NAME_LENGTH
-from langflow.base.mcp.util import sanitize_mcp_name
-from langflow.logging import logger
 from langflow.services.auth.mcp_encryption import decrypt_auth_settings, encrypt_auth_settings
 from langflow.services.database.models import Flow, Folder
 from langflow.services.database.models.api_key.crud import check_key, create_api_key
 from langflow.services.database.models.api_key.model import ApiKeyCreate
 from langflow.services.database.models.user.model import User
-from langflow.services.deps import get_settings_service, session_scope
-from langflow.services.settings.feature_flags import FEATURE_FLAGS
 
 router = APIRouter(prefix="/mcp/project", tags=["mcp_projects"])
 
@@ -249,6 +249,15 @@ async def handle_project_sse(
     current_user: Annotated[User, Depends(verify_project_auth_conditional)],
 ):
     """Handle SSE connections for a specific project."""
+    # Verify project exists and user has access
+    async with session_scope() as session:
+        project = (
+            await session.exec(select(Folder).where(Folder.id == project_id, Folder.user_id == current_user.id))
+        ).first()
+
+    if not project:
+        raise HTTPException(status_code=404, detail="Project not found")
+
     # Get project-specific SSE transport and MCP server
     sse = get_project_sse(project_id)
     project_server = get_project_mcp_server(project_id)

langflow/api/v1/mcp_utils.py

@@ -12,15 +12,15 @@ from typing import Any, ParamSpec, TypeVar
 from urllib.parse import quote, unquote, urlparse
 from uuid import uuid4
 
+from lfx.base.mcp.constants import MAX_MCP_TOOL_NAME_LENGTH
+from lfx.base.mcp.util import get_flow_snake_case, get_unique_name, sanitize_mcp_name
+from lfx.log.logger import logger
 from mcp import types
 from sqlmodel import select
 
 from langflow.api.v1.endpoints import simple_run_flow
 from langflow.api.v1.schemas import SimplifiedAPIRequest
-from langflow.base.mcp.constants import MAX_MCP_TOOL_NAME_LENGTH
-from langflow.base.mcp.util import get_flow_snake_case, get_unique_name, sanitize_mcp_name
 from langflow.helpers.flow import json_schema_from_flow
-from langflow.logging.logger import logger
 from langflow.schema.message import Message
 from langflow.services.database.models import Flow
 from langflow.services.database.models.user.model import User

langflow/api/v1/openai_responses.py

@@ -7,20 +7,20 @@ from typing import Annotated, Any
 
 from fastapi import APIRouter, BackgroundTasks, Depends, HTTPException, Request
 from fastapi.responses import StreamingResponse
-from loguru import logger
+from lfx.log.logger import logger
+from lfx.schema.openai_responses_schemas import create_openai_error
 
 from langflow.api.v1.endpoints import consume_and_yield, run_flow_generator, simple_run_flow
 from langflow.api.v1.schemas import SimplifiedAPIRequest
 from langflow.events.event_manager import create_stream_tokens_event_manager
 from langflow.helpers.flow import get_flow_by_id_or_endpoint_name
-from langflow.schema.content_types import ToolContent
-from langflow.schema.openai_responses_schemas import (
+from langflow.schema import (
     OpenAIErrorResponse,
     OpenAIResponsesRequest,
     OpenAIResponsesResponse,
     OpenAIResponsesStreamChunk,
-    create_openai_error,
 )
+from langflow.schema.content_types import ToolContent
 from langflow.services.auth.utils import api_key_security
 from langflow.services.database.models.flow.model import FlowRead
 from langflow.services.database.models.user.model import UserRead

langflow/api/v1/schemas.py

@@ -4,6 +4,9 @@ from pathlib import Path
 from typing import Any, Literal
 from uuid import UUID
 
+from lfx.graph.schema import RunOutputs
+from lfx.services.settings.base import Settings
+from lfx.services.settings.feature_flags import FEATURE_FLAGS, FeatureFlags
 from pydantic import (
     BaseModel,
     ConfigDict,
@@ -14,7 +17,6 @@ from pydantic import (
     model_serializer,
 )
 
-from langflow.graph.schema import RunOutputs
 from langflow.schema.dotdict import dotdict
 from langflow.schema.graph import Tweaks
 from langflow.schema.schema import InputType, OutputType, OutputValue
@@ -23,8 +25,6 @@ from langflow.services.database.models.api_key.model import ApiKeyRead
 from langflow.services.database.models.base import orjson_dumps
 from langflow.services.database.models.flow.model import FlowCreate, FlowRead
 from langflow.services.database.models.user.model import UserRead
-from langflow.services.settings.base import Settings
-from langflow.services.settings.feature_flags import FEATURE_FLAGS, FeatureFlags
 from langflow.services.tracing.schema import Log
 
 
@@ -335,41 +335,6 @@ class VerticesBuiltResponse(BaseModel):
     vertices: list[VertexBuildResponse]
 
 
-class InputValueRequest(BaseModel):
-    components: list[str] | None = []
-    input_value: str | None = None
-    session: str | None = None
-    type: InputType | None = Field(
-        "any",
-        description="Defines on which components the input value should be applied. "
-        "'any' applies to all input components.",
-    )
-
-    # add an example
-    model_config = ConfigDict(
-        json_schema_extra={
-            "examples": [
-                {
-                    "components": ["components_id", "Component Name"],
-                    "input_value": "input_value",
-                    "session": "session_id",
-                },
-                {"components": ["Component Name"], "input_value": "input_value"},
-                {"input_value": "input_value"},
-                {
-                    "components": ["Component Name"],
-                    "input_value": "input_value",
-                    "session": "session_id",
-                },
-                {"input_value": "input_value", "session": "session_id"},
-                {"type": "chat", "input_value": "input_value"},
-                {"type": "json", "input_value": '{"key": "value"}'},
-            ]
-        },
-        extra="forbid",
-    )
-
-
 class SimplifiedAPIRequest(BaseModel):
     input_value: str | None = Field(default=None, description="The input value")
     input_type: InputType | None = Field(default="chat", description="The input type")

langflow/api/v1/starter_projects.py

@@ -1,17 +1,75 @@
+from typing import Any
+
 from fastapi import APIRouter, Depends, HTTPException
+from pydantic import BaseModel
 
-from langflow.graph.graph.schema import GraphDump
 from langflow.services.auth.utils import get_current_active_user
 
 router = APIRouter(prefix="/starter-projects", tags=["Flows"])
 
 
+# Pydantic models for API schema compatibility
+class ViewPort(BaseModel):
+    x: float
+    y: float
+    zoom: float
+
+
+class NodeData(BaseModel):
+    # This is a simplified version - the actual NodeData has many more fields
+    # but we only need the basic structure for the API schema
+    model_config = {"extra": "allow"}  # Allow extra fields
+
+
+class EdgeData(BaseModel):
+    # This is a simplified version - the actual EdgeData has many more fields
+    # but we only need the basic structure for the API schema
+    model_config = {"extra": "allow"}  # Allow extra fields
+
+
+class GraphData(BaseModel):
+    nodes: list[dict[str, Any]]  # Use dict to be flexible with the complex NodeData structure
+    edges: list[dict[str, Any]]  # Use dict to be flexible with the complex EdgeData structure
+    viewport: ViewPort | None = None
+
+
+class GraphDumpResponse(BaseModel):
+    data: GraphData
+    is_component: bool | None = None
+    name: str | None = None
+    description: str | None = None
+    endpoint_name: str | None = None
+
+
 @router.get("/", dependencies=[Depends(get_current_active_user)], status_code=200)
-async def get_starter_projects() -> list[GraphDump]:
+async def get_starter_projects() -> list[GraphDumpResponse]:
     """Get a list of starter projects."""
     from langflow.initial_setup.load import get_starter_projects_dump
 
     try:
-        return get_starter_projects_dump()
+        # Get the raw data from lfx GraphDump
+        raw_data = get_starter_projects_dump()
+
+        # Convert TypedDict GraphDump to Pydantic GraphDumpResponse
+        results = []
+        for item in raw_data:
+            # Create GraphData
+            graph_data = GraphData(
+                nodes=item.get("data", {}).get("nodes", []),
+                edges=item.get("data", {}).get("edges", []),
+                viewport=item.get("data", {}).get("viewport"),
+            )
+
+            # Create GraphDumpResponse
+            graph_dump = GraphDumpResponse(
+                data=graph_data,
+                is_component=item.get("is_component"),
+                name=item.get("name"),
+                description=item.get("description"),
+                endpoint_name=item.get("endpoint_name"),
+            )
+            results.append(graph_dump)
+
     except Exception as exc:
         raise HTTPException(status_code=500, detail=str(exc)) from exc
+    return results

langflow/api/v1/store.py

@@ -2,9 +2,9 @@ from typing import Annotated
 from uuid import UUID
 
 from fastapi import APIRouter, Depends, HTTPException, Query
+from lfx.log.logger import logger
 
 from langflow.api.utils import CurrentActiveUser, check_langflow_version
-from langflow.logging.logger import logger
 from langflow.services.auth import utils as auth_utils
 from langflow.services.deps import get_settings_service, get_store_service
 from langflow.services.store.exceptions import CustomError

langflow/api/v1/validate.py

@@ -1,10 +1,10 @@
 from fastapi import APIRouter, HTTPException
+from lfx.base.prompts.api_utils import process_prompt_template
+from lfx.custom.validate import validate_code
+from lfx.log.logger import logger
 
 from langflow.api.utils import CurrentActiveUser
 from langflow.api.v1.base import Code, CodeValidationResponse, PromptValidationResponse, ValidatePromptRequest
-from langflow.base.prompts.api_utils import process_prompt_template
-from langflow.logging.logger import logger
-from langflow.utils.validate import validate_code
 
 # build router
 router = APIRouter(prefix="/validate", tags=["Validate"])

langflow/api/v1/voice_mode.py

@@ -18,14 +18,14 @@ import websockets
 from cryptography.fernet import InvalidToken
 from elevenlabs import ElevenLabs
 from fastapi import APIRouter, BackgroundTasks
+from lfx.log import logger
+from lfx.schema.schema import InputValueRequest
 from openai import OpenAI
 from sqlalchemy import select
 from starlette.websockets import WebSocket, WebSocketDisconnect
 
 from langflow.api.utils import CurrentActiveUser, DbSession
 from langflow.api.v1.chat import build_flow_and_stream
-from langflow.api.v1.schemas import InputValueRequest
-from langflow.logging import logger
 from langflow.memory import aadd_messagetables
 from langflow.schema.properties import Properties
 from langflow.services.auth.utils import get_current_user_for_websocket

langflow/api/v2/files.py

@@ -11,11 +11,11 @@ from zoneinfo import ZoneInfo
 
 from fastapi import APIRouter, Depends, File, HTTPException, UploadFile
 from fastapi.responses import StreamingResponse
+from lfx.log.logger import logger
 from sqlmodel import col, select
 
 from langflow.api.schemas import UploadFileResponse
 from langflow.api.utils import CurrentActiveUser, DbSession
-from langflow.logging.logger import logger
 from langflow.services.database.models.file.model import File as UserFile
 from langflow.services.deps import get_settings_service, get_storage_service
 from langflow.services.storage.service import StorageService

langflow/api/v2/mcp.py

@@ -3,11 +3,11 @@ import json
 from io import BytesIO
 
 from fastapi import APIRouter, Depends, HTTPException, UploadFile
+from lfx.base.mcp.util import update_tools
+from lfx.log import logger
 
 from langflow.api.utils import CurrentActiveUser, DbSession
 from langflow.api.v2.files import MCP_SERVERS_FILE, delete_file, download_file, get_file_by_name, upload_user_file
-from langflow.base.mcp.util import update_tools
-from langflow.logging import logger
 from langflow.services.deps import get_settings_service, get_storage_service
 
 router = APIRouter(tags=["MCP"], prefix="/mcp")

langflow/base/__init__.py

@@ -0,0 +1,11 @@
+"""Backwards compatibility module for langflow.base.
+
+This module imports from lfx.base to maintain compatibility with existing code
+that expects to import from langflow.base.
+"""
+
+# Import all base modules from lfx for backwards compatibility
+from lfx.base import *  # noqa: F403
+
+# Import langflow-specific modules that aren't in lfx.base
+from . import knowledge_bases  # noqa: F401

langflow/base/agents/__init__.py

@@ -0,0 +1,3 @@
+"""Backwards compatibility module for langflow.base.agents."""
+
+from lfx.base.agents import *  # noqa: F403

langflow/base/data/__init__.py

@@ -1,5 +1,3 @@
-from .base_file import BaseFileComponent
+"""Backwards compatibility module for langflow.base.data."""
 
-__all__ = [
-    "BaseFileComponent",
-]
+from lfx.base.data import *  # noqa: F403