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

langflow/services/socket/utils.py

@@ -2,6 +2,7 @@ import time
 from collections.abc import Callable
 
 import socketio
+from lfx.log.logger import logger
 from sqlmodel import select
 
 from langflow.api.utils import format_elapsed_time
@@ -10,18 +11,10 @@ from langflow.graph.graph.base import Graph
 from langflow.graph.graph.utils import layered_topological_sort
 from langflow.graph.utils import log_vertex_build
 from langflow.graph.vertex.base import Vertex
-from langflow.logging.logger import logger
 from langflow.services.database.models.flow.model import Flow
 from langflow.services.deps import get_session
 
 
-def set_socketio_server(socketio_server) -> None:
-    from langflow.services.deps import get_socket_service
-
-    socket_service = get_socket_service()
-    socket_service.init(socketio_server)
-
-
 async def get_vertices(sio, sid, flow_id, chat_service) -> None:
     try:
         session = await anext(get_session())

langflow/services/state/factory.py

@@ -1,7 +1,7 @@
+from lfx.services.settings.service import SettingsService
 from typing_extensions import override
 
 from langflow.services.factory import ServiceFactory
-from langflow.services.settings.service import SettingsService
 from langflow.services.state.service import InMemoryStateService
 
 

langflow/services/state/service.py

@@ -2,9 +2,10 @@ from collections import defaultdict
 from collections.abc import Callable
 from threading import Lock
 
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
+from lfx.services.settings.service import SettingsService
+
 from langflow.services.base import Service
-from langflow.services.settings.service import SettingsService
 
 
 class StateService(Service):

langflow/services/storage/factory.py

@@ -1,9 +1,9 @@
+from lfx.log.logger import logger
+from lfx.services.settings.service import SettingsService
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.services.factory import ServiceFactory
 from langflow.services.session.service import SessionService
-from langflow.services.settings.service import SettingsService
 from langflow.services.storage.service import StorageService
 
 

langflow/services/storage/local.py

@@ -1,7 +1,6 @@
 import anyio
 from aiofile import async_open
-
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
 
 from .service import StorageService
 

langflow/services/storage/s3.py

@@ -1,7 +1,6 @@
 import boto3
 from botocore.exceptions import ClientError, NoCredentialsError
-
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
 
 from .service import StorageService
 

langflow/services/storage/service.py

@@ -8,8 +8,9 @@ import anyio
 from langflow.services.base import Service
 
 if TYPE_CHECKING:
+    from lfx.services.settings.service import SettingsService
+
     from langflow.services.session.service import SessionService
-    from langflow.services.settings.service import SettingsService
 
 
 class StorageService(Service):

langflow/services/store/factory.py

@@ -8,7 +8,7 @@ from langflow.services.factory import ServiceFactory
 from langflow.services.store.service import StoreService
 
 if TYPE_CHECKING:
-    from langflow.services.settings.service import SettingsService
+    from lfx.services.settings.service import SettingsService
 
 
 class StoreServiceFactory(ServiceFactory):

langflow/services/store/service.py

@@ -6,8 +6,8 @@ from uuid import UUID
 
 import httpx
 from httpx import HTTPError, HTTPStatusError
+from lfx.log.logger import logger
 
-from langflow.logging.logger import logger
 from langflow.services.base import Service
 from langflow.services.store.exceptions import APIKeyError, FilterError, ForbiddenError
 from langflow.services.store.schema import (
@@ -24,7 +24,7 @@ from langflow.services.store.utils import (
 )
 
 if TYPE_CHECKING:
-    from langflow.services.settings.service import SettingsService
+    from lfx.services.settings.service import SettingsService
 
 from contextlib import asynccontextmanager
 from contextvars import ContextVar

langflow/services/store/utils.py

@@ -1,8 +1,7 @@
 from typing import TYPE_CHECKING
 
 import httpx
-
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
 
 if TYPE_CHECKING:
     from langflow.services.store.schema import ListComponentResponse

langflow/services/task/service.py

@@ -7,7 +7,8 @@ from langflow.services.base import Service
 from langflow.services.task.backends.anyio import AnyIOBackend
 
 if TYPE_CHECKING:
-    from langflow.services.settings.service import SettingsService
+    from lfx.services.settings.service import SettingsService
+
     from langflow.services.task.backends.base import TaskBackend
 
 

langflow/services/task/temp_flow_cleanup.py

@@ -4,9 +4,9 @@ import asyncio
 import contextlib
 from typing import TYPE_CHECKING
 
+from lfx.log.logger import logger
 from sqlmodel import col, delete, select
 
-from langflow.logging.logger import logger
 from langflow.services.database.models.message.model import MessageTable
 from langflow.services.database.models.transactions.model import TransactionTable
 from langflow.services.database.models.vertex_builds.model import VertexBuildTable

langflow/services/telemetry/factory.py

@@ -8,7 +8,7 @@ from langflow.services.factory import ServiceFactory
 from langflow.services.telemetry.service import TelemetryService
 
 if TYPE_CHECKING:
-    from langflow.services.settings.service import SettingsService
+    from lfx.services.settings.service import SettingsService
 
 
 class TelemetryServiceFactory(ServiceFactory):

langflow/services/telemetry/service.py

@@ -9,8 +9,8 @@ from datetime import datetime, timezone
 from typing import TYPE_CHECKING
 
 import httpx
+from lfx.log.logger import logger
 
-from langflow.logging.logger import logger
 from langflow.services.base import Service
 from langflow.services.telemetry.opentelemetry import OpenTelemetry
 from langflow.services.telemetry.schema import (
@@ -24,10 +24,9 @@ from langflow.services.telemetry.schema import (
 from langflow.utils.version import get_version_info
 
 if TYPE_CHECKING:
+    from lfx.services.settings.service import SettingsService
     from pydantic import BaseModel
 
-    from langflow.services.settings.service import SettingsService
-
 
 class TelemetryService(Service):
     name = "telemetry_service"

langflow/services/tracing/arize_phoenix.py

@@ -10,14 +10,14 @@ from typing import TYPE_CHECKING, Any
 
 from langchain_core.documents import Document
 from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
+from lfx.log.logger import logger
+from lfx.schema.data import Data
 from openinference.semconv.trace import OpenInferenceMimeTypeValues, SpanAttributes
 from opentelemetry.semconv.trace import SpanAttributes as OTELSpanAttributes
 from opentelemetry.trace import Span, Status, StatusCode, use_span
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
 from typing_extensions import override
 
-from langflow.logging.logger import logger
-from langflow.schema.data import Data
 from langflow.schema.message import Message
 from langflow.services.tracing.base import BaseTracer
 
@@ -26,10 +26,10 @@ if TYPE_CHECKING:
     from uuid import UUID
 
     from langchain.callbacks.base import BaseCallbackHandler
+    from lfx.graph.vertex.base import Vertex
     from opentelemetry.propagators.textmap import CarrierT
     from opentelemetry.util.types import AttributeValue
 
-    from langflow.graph.vertex.base import Vertex
     from langflow.services.tracing.schema import Log
 
 

langflow/services/tracing/base.py

@@ -8,8 +8,8 @@ if TYPE_CHECKING:
     from uuid import UUID
 
     from langchain.callbacks.base import BaseCallbackHandler
+    from lfx.graph.vertex.base import Vertex
 
-    from langflow.graph.vertex.base import Vertex
     from langflow.services.tracing.schema import Log
 
 

langflow/services/tracing/factory.py

@@ -8,7 +8,7 @@ from langflow.services.factory import ServiceFactory
 from langflow.services.tracing.service import TracingService
 
 if TYPE_CHECKING:
-    from langflow.services.settings.service import SettingsService
+    from lfx.services.settings.service import SettingsService
 
 
 class TracingServiceFactory(ServiceFactory):

langflow/services/tracing/langfuse.py

@@ -5,9 +5,9 @@ from collections import OrderedDict
 from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any
 
+from lfx.log.logger import logger
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.serialization.serialization import serialize
 from langflow.services.tracing.base import BaseTracer
 
@@ -16,8 +16,8 @@ if TYPE_CHECKING:
     from uuid import UUID
 
     from langchain.callbacks.base import BaseCallbackHandler
+    from lfx.graph.vertex.base import Vertex
 
-    from langflow.graph.vertex.base import Vertex
     from langflow.services.tracing.schema import Log
 
 

langflow/services/tracing/langsmith.py

@@ -6,9 +6,9 @@ import types
 from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any
 
+from lfx.log.logger import logger
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.serialization.serialization import serialize
 from langflow.services.tracing.base import BaseTracer
@@ -19,8 +19,8 @@ if TYPE_CHECKING:
 
     from langchain.callbacks.base import BaseCallbackHandler
     from langsmith.run_trees import RunTree
+    from lfx.graph.vertex.base import Vertex
 
-    from langflow.graph.vertex.base import Vertex
     from langflow.services.tracing.schema import Log
 
 

langflow/services/tracing/langwatch.py

@@ -4,9 +4,9 @@ import os
 from typing import TYPE_CHECKING, Any, cast
 
 import nanoid
+from lfx.log.logger import logger
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.services.tracing.base import BaseTracer
 
@@ -16,8 +16,8 @@ if TYPE_CHECKING:
 
     from langchain.callbacks.base import BaseCallbackHandler
     from langwatch.tracer import ContextSpan
+    from lfx.graph.vertex.base import Vertex
 
-    from langflow.graph.vertex.base import Vertex
     from langflow.services.tracing.schema import Log
 
 
@@ -155,9 +155,9 @@ class LangWatchTracer(BaseTracer):
         return autoconvert_typed_values(converted)
 
     def _convert_to_langwatch_type(self, value):
+        from langchain_core.messages import BaseMessage
         from langwatch.langchain import langchain_message_to_chat_message, langchain_messages_to_chat_messages
-
-        from langflow.schema.message import BaseMessage, Message
+        from lfx.schema.message import Message
 
         if isinstance(value, dict):
             value = {key: self._convert_to_langwatch_type(val) for key, val in value.items()}

langflow/services/tracing/opik.py

@@ -6,9 +6,9 @@ from typing import TYPE_CHECKING, Any
 
 from langchain_core.documents import Document
 from langchain_core.messages import BaseMessage, HumanMessage, SystemMessage
+from lfx.log.logger import logger
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.schema.message import Message
 from langflow.services.tracing.base import BaseTracer
@@ -18,8 +18,8 @@ if TYPE_CHECKING:
     from uuid import UUID
 
     from langchain.callbacks.base import BaseCallbackHandler
+    from lfx.graph.vertex.base import Vertex
 
-    from langflow.graph.vertex.base import Vertex
     from langflow.services.tracing.schema import Log
 
 

langflow/services/tracing/service.py

@@ -7,17 +7,18 @@ from contextlib import asynccontextmanager
 from contextvars import ContextVar
 from typing import TYPE_CHECKING, Any
 
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
+
 from langflow.services.base import Service
 
 if TYPE_CHECKING:
     from uuid import UUID
 
     from langchain.callbacks.base import BaseCallbackHandler
+    from lfx.custom.custom_component.component import Component
+    from lfx.graph.vertex.base import Vertex
+    from lfx.services.settings.service import SettingsService
 
-    from langflow.custom.custom_component.component import Component
-    from langflow.graph.vertex.base import Vertex
-    from langflow.services.settings.service import SettingsService
     from langflow.services.tracing.base import BaseTracer
     from langflow.services.tracing.schema import Log
 
@@ -286,8 +287,7 @@ class TracingService(Service):
             return
         trace_context = trace_context_var.get()
         if trace_context is None:
-            msg = "called end_tracers but no trace context found"
-            raise RuntimeError(msg)
+            return
         await self._stop(trace_context)
         self._end_all_tracers(trace_context, outputs, error)
 
@@ -380,7 +380,9 @@ class TracingService(Service):
         trace_context = trace_context_var.get()
         if trace_context is None:
             msg = "called trace_component but no trace context found"
-            raise RuntimeError(msg)
+            logger.warning(msg)
+            yield self
+            return
         trace_context.all_inputs[trace_name] |= inputs or {}
         await trace_context.traces_queue.put((self._start_component_traces, (component_trace_context, trace_context)))
         try:
@@ -402,7 +404,8 @@ class TracingService(Service):
         trace_context = trace_context_var.get()
         if trace_context is None:
             msg = "called project_name but no trace context found"
-            raise RuntimeError(msg)
+            logger.warning(msg)
+            return None
         return trace_context.project_name
 
     def add_log(self, trace_name: str, log: Log) -> None:
@@ -433,14 +436,16 @@ class TracingService(Service):
         trace_context = trace_context_var.get()
         if trace_context is None:
             msg = "called set_outputs but no trace context found"
-            raise RuntimeError(msg)
+            logger.warning(msg)
+            return
         trace_context.all_outputs[trace_name] |= outputs or {}
 
     def get_tracer(self, tracer_name: str) -> BaseTracer | None:
         trace_context = trace_context_var.get()
         if trace_context is None:
             msg = "called get_tracer but no trace context found"
-            raise RuntimeError(msg)
+            logger.warning(msg)
+            return None
         return trace_context.tracers.get(tracer_name)
 
     def get_langchain_callbacks(self) -> list[BaseCallbackHandler]:
@@ -450,7 +455,8 @@ class TracingService(Service):
         trace_context = trace_context_var.get()
         if trace_context is None:
             msg = "called get_langchain_callbacks but no trace context found"
-            raise RuntimeError(msg)
+            logger.warning(msg)
+            return []
         for tracer in trace_context.tracers.values():
             if not tracer.ready:  # type: ignore[truthy-function]
                 continue

langflow/services/tracing/traceloop.py

@@ -8,7 +8,7 @@ from datetime import datetime, timezone
 from typing import TYPE_CHECKING, Any
 from urllib.parse import urlparse
 
-from loguru import logger
+from lfx.log.logger import logger
 from opentelemetry import trace
 from opentelemetry.trace import Span, use_span
 from opentelemetry.trace.propagation.tracecontext import TraceContextTextMapPropagator
@@ -77,7 +77,7 @@ class TraceloopTracer(BaseTracer):
                 self.propagator.inject(carrier=self.carrier)
 
         except Exception:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error setting up Traceloop tracer")
+            logger.debug("Error setting up Traceloop tracer", exc_info=True)
             self._ready = False
 
     @property

langflow/services/tracing/utils.py

@@ -1,6 +1,6 @@
 from typing import Any
 
-from langflow.schema.data import Data
+from lfx.schema.data import Data
 
 
 def convert_to_langchain_type(value):

langflow/services/utils.py

@@ -3,11 +3,12 @@ from __future__ import annotations
 import asyncio
 from typing import TYPE_CHECKING
 
+from lfx.log.logger import logger
+from lfx.services.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD
 from sqlalchemy import delete
 from sqlalchemy import exc as sqlalchemy_exc
 from sqlmodel import col, select
 
-from langflow.logging.logger import logger
 from langflow.services.auth.utils import create_super_user, verify_password
 from langflow.services.cache.base import ExternalAsyncBaseCacheService
 from langflow.services.cache.factory import CacheServiceFactory
@@ -15,15 +16,13 @@ from langflow.services.database.models.transactions.model import TransactionTabl
 from langflow.services.database.models.vertex_builds.model import VertexBuildTable
 from langflow.services.database.utils import initialize_database
 from langflow.services.schema import ServiceType
-from langflow.services.settings.constants import DEFAULT_SUPERUSER, DEFAULT_SUPERUSER_PASSWORD
 
-from .deps import get_db_service, get_service, get_settings_service
+from .deps import get_db_service, get_service, get_settings_service, session_scope
 
 if TYPE_CHECKING:
+    from lfx.services.settings.manager import SettingsService
     from sqlmodel.ext.asyncio.session import AsyncSession
 
-    from langflow.services.settings.manager import SettingsService
-
 
 async def get_or_create_super_user(session: AsyncSession, username, password, is_default):
     from langflow.services.database.models.user.model import User
@@ -129,17 +128,18 @@ async def teardown_superuser(settings_service, session: AsyncSession) -> None:
 
 async def teardown_services() -> None:
     """Teardown all the services."""
-    async with get_db_service().with_session() as session:
+    async with session_scope() as session:
         await teardown_superuser(get_settings_service(), session)
 
-    from langflow.services.manager import service_manager
+    from lfx.services.manager import get_service_manager
 
+    service_manager = get_service_manager()
     await service_manager.teardown()
 
 
 def initialize_settings_service() -> None:
     """Initialize the settings manager."""
-    from langflow.services.settings import factory as settings_factory
+    from lfx.services.settings import factory as settings_factory
 
     get_service(ServiceType.SETTINGS_SERVICE, settings_factory.SettingsServiceFactory())
 
@@ -220,8 +220,53 @@ async def clean_vertex_builds(settings_service: SettingsService, session: AsyncS
         # Don't re-raise since this is a cleanup task
 
 
+def register_all_service_factories() -> None:
+    """Register all available service factories with the service manager."""
+    # Import all service factories
+    from lfx.services.manager import get_service_manager
+
+    service_manager = get_service_manager()
+    from lfx.services.settings import factory as settings_factory
+
+    from langflow.services.auth import factory as auth_factory
+    from langflow.services.cache import factory as cache_factory
+    from langflow.services.chat import factory as chat_factory
+    from langflow.services.database import factory as database_factory
+    from langflow.services.job_queue import factory as job_queue_factory
+    from langflow.services.session import factory as session_factory
+    from langflow.services.shared_component_cache import factory as shared_component_cache_factory
+    from langflow.services.state import factory as state_factory
+    from langflow.services.storage import factory as storage_factory
+    from langflow.services.store import factory as store_factory
+    from langflow.services.task import factory as task_factory
+    from langflow.services.telemetry import factory as telemetry_factory
+    from langflow.services.tracing import factory as tracing_factory
+    from langflow.services.variable import factory as variable_factory
+
+    # Register all factories
+    service_manager.register_factory(settings_factory.SettingsServiceFactory())
+    service_manager.register_factory(cache_factory.CacheServiceFactory())
+    service_manager.register_factory(chat_factory.ChatServiceFactory())
+    service_manager.register_factory(database_factory.DatabaseServiceFactory())
+    service_manager.register_factory(session_factory.SessionServiceFactory())
+    service_manager.register_factory(storage_factory.StorageServiceFactory())
+    service_manager.register_factory(variable_factory.VariableServiceFactory())
+    service_manager.register_factory(telemetry_factory.TelemetryServiceFactory())
+    service_manager.register_factory(tracing_factory.TracingServiceFactory())
+    service_manager.register_factory(state_factory.StateServiceFactory())
+    service_manager.register_factory(job_queue_factory.JobQueueServiceFactory())
+    service_manager.register_factory(task_factory.TaskServiceFactory())
+    service_manager.register_factory(store_factory.StoreServiceFactory())
+    service_manager.register_factory(shared_component_cache_factory.SharedComponentCacheServiceFactory())
+    service_manager.register_factory(auth_factory.AuthServiceFactory())
+    service_manager.set_factory_registered()
+
+
 async def initialize_services(*, fix_migration: bool = False) -> None:
     """Initialize all the services needed."""
+    # Register all service factories first
+    register_all_service_factories()
+
     cache_service = get_service(ServiceType.CACHE_SERVICE, default=CacheServiceFactory())
     # Test external cache connection
     if isinstance(cache_service, ExternalAsyncBaseCacheService) and not (await cache_service.is_connected()):
@@ -232,7 +277,7 @@ async def initialize_services(*, fix_migration: bool = False) -> None:
     await initialize_database(fix_migration=fix_migration)
     db_service = get_db_service()
     await db_service.initialize_alembic_log_file()
-    async with db_service.with_session() as session:
+    async with session_scope() as session:
         settings_service = get_service(ServiceType.SETTINGS_SERVICE)
         await setup_superuser(settings_service, session)
     try:

langflow/services/variable/factory.py

@@ -8,7 +8,7 @@ from langflow.services.factory import ServiceFactory
 from langflow.services.variable.service import DatabaseVariableService, VariableService
 
 if TYPE_CHECKING:
-    from langflow.services.settings.service import SettingsService
+    from lfx.services.settings.service import SettingsService
 
 
 class VariableServiceFactory(ServiceFactory):

langflow/services/variable/kubernetes.py

@@ -4,9 +4,9 @@ import asyncio
 import os
 from typing import TYPE_CHECKING
 
+from lfx.log.logger import logger
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.services.auth import utils as auth_utils
 from langflow.services.base import Service
 from langflow.services.database.models.variable.model import Variable, VariableCreate, VariableRead
@@ -17,11 +17,10 @@ from langflow.services.variable.kubernetes_secrets import KubernetesSecretManage
 if TYPE_CHECKING:
     from uuid import UUID
 
+    from lfx.services.settings.service import SettingsService
     from sqlmodel import Session
     from sqlmodel.ext.asyncio.session import AsyncSession
 
-    from langflow.services.settings.service import SettingsService
-
 
 class KubernetesSecretService(VariableService, Service):
     def __init__(self, settings_service: SettingsService):

langflow/services/variable/kubernetes_secrets.py

@@ -4,8 +4,7 @@ from uuid import UUID
 
 from kubernetes import client, config
 from kubernetes.client.rest import ApiException
-
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
 
 
 class KubernetesSecretManager:

langflow/services/variable/service.py

@@ -4,10 +4,10 @@ import os
 from datetime import datetime, timezone
 from typing import TYPE_CHECKING
 
+from lfx.log.logger import logger
 from sqlmodel import select
 from typing_extensions import override
 
-from langflow.logging.logger import logger
 from langflow.services.auth import utils as auth_utils
 from langflow.services.base import Service
 from langflow.services.database.models.variable.model import Variable, VariableCreate, VariableRead, VariableUpdate
@@ -18,10 +18,9 @@ if TYPE_CHECKING:
     from collections.abc import Sequence
     from uuid import UUID
 
+    from lfx.services.settings.service import SettingsService
     from sqlmodel.ext.asyncio.session import AsyncSession
 
-    from langflow.services.settings.service import SettingsService
-
 
 class DatabaseVariableService(VariableService, Service):
     def __init__(self, settings_service: SettingsService):

langflow/template/__init__.py

@@ -1,10 +1,3 @@
-from langflow.template.field.base import Input, Output
-from langflow.template.frontend_node.base import FrontendNode
-from langflow.template.template.base import Template
+"""Backwards compatibility module for langflow.template."""
 
-__all__ = [
-    "FrontendNode",
-    "Input",
-    "Output",
-    "Template",
-]
+from lfx.template import *  # noqa: F403

langflow/template/field/__init__.py

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