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

langflow/alembic/versions/a72f5cf9c2f9_add_endpoint_name_col.py

@@ -6,18 +6,17 @@ Create Date: 2024-05-29 21:44:04.240816
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "a72f5cf9c2f9"
-down_revision: Union[str, None] = "29fe8f1f806b"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "29fe8f1f806b"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/b2fa308044b5_add_unique_constraints.py

@@ -11,8 +11,7 @@ from collections.abc import Sequence
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
-
-from langflow.logging.logger import logger
+from lfx.log.logger import logger
 
 # revision identifiers, used by Alembic.
 revision: str = "b2fa308044b5"

langflow/alembic/versions/bc2f01c40e4a_new_fixes.py

@@ -6,19 +6,18 @@ Create Date: 2024-01-26 13:34:14.496769
 
 """
 
-from typing import Sequence, Union
 import warnings
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "bc2f01c40e4a"
-down_revision: Union[str, None] = "b2fa308044b5"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "b2fa308044b5"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/c153816fd85f_set_name_and_value_to_not_nullable.py

@@ -6,17 +6,16 @@ Create Date: 2024-04-30 14:31:23.898995
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "c153816fd85f"
-down_revision: Union[str, None] = "1f4d6df60295"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "1f4d6df60295"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/d066bfd22890_add_message_table.py

@@ -6,7 +6,7 @@ Create Date: 2024-06-23 21:29:28.220100
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 import sqlmodel
@@ -16,9 +16,9 @@ from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
 revision: str = "325180f0c4e1"
-down_revision: Union[str, None] = "631faacf5da2"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "631faacf5da2"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/d2d475a1f7c0_add_tags_column_to_flow.py

@@ -5,28 +5,27 @@ Revises: d3dbf656a499
 Create Date: 2024-10-03 13:33:59.517261
 
 """
-from typing import Sequence, Union
+
+from collections.abc import Sequence
 
 import sqlalchemy as sa
-import sqlmodel
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
-revision: str = 'd2d475a1f7c0'
-down_revision: Union[str, None] = 'd3dbf656a499'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "d2d475a1f7c0"
+down_revision: str | None = "d3dbf656a499"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
     conn = op.get_bind()
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('flow', schema=None) as batch_op:
-        if not migration.column_exists(table_name='flow', column_name='tags', conn=conn):
-            batch_op.add_column(sa.Column('tags', sa.JSON(), nullable=True))
+    with op.batch_alter_table("flow", schema=None) as batch_op:
+        if not migration.column_exists(table_name="flow", column_name="tags", conn=conn):
+            batch_op.add_column(sa.Column("tags", sa.JSON(), nullable=True))
 
     # ### end Alembic commands ###
 
@@ -34,8 +33,8 @@ def upgrade() -> None:
 def downgrade() -> None:
     conn = op.get_bind()
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('flow', schema=None) as batch_op:
-        if migration.column_exists(table_name='flow', column_name='tags', conn=conn):
-            batch_op.drop_column('tags')
+    with op.batch_alter_table("flow", schema=None) as batch_op:
+        if migration.column_exists(table_name="flow", column_name="tags", conn=conn):
+            batch_op.drop_column("tags")
 
     # ### end Alembic commands ###

langflow/alembic/versions/d3dbf656a499_add_gradient_column_in_flow.py

@@ -5,28 +5,28 @@ Revises: e5a65ecff2cd
 Create Date: 2024-09-27 09:35:19.424089
 
 """
-from typing import Sequence, Union
+
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
-revision: str = 'd3dbf656a499'
-down_revision: Union[str, None] = 'e5a65ecff2cd'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "d3dbf656a499"
+down_revision: str | None = "e5a65ecff2cd"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
     conn = op.get_bind()
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('flow', schema=None) as batch_op:
-        if not migration.column_exists(table_name='flow', column_name='gradient', conn=conn):
-            batch_op.add_column(sa.Column('gradient', sqlmodel.sql.sqltypes.AutoString(), nullable=True))
+    with op.batch_alter_table("flow", schema=None) as batch_op:
+        if not migration.column_exists(table_name="flow", column_name="gradient", conn=conn):
+            batch_op.add_column(sa.Column("gradient", sqlmodel.sql.sqltypes.AutoString(), nullable=True))
 
     # ### end Alembic commands ###
 
@@ -34,8 +34,8 @@ def upgrade() -> None:
 def downgrade() -> None:
     conn = op.get_bind()
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('flow', schema=None) as batch_op:
-        if migration.column_exists(table_name='flow', column_name='gradient', conn=conn):
-            batch_op.drop_column('gradient')
+    with op.batch_alter_table("flow", schema=None) as batch_op:
+        if migration.column_exists(table_name="flow", column_name="gradient", conn=conn):
+            batch_op.drop_column("gradient")
 
     # ### end Alembic commands ###

langflow/alembic/versions/d9a6ea21edcd_rename_default_folder.py

@@ -5,20 +5,17 @@ Revises: 66f72f04a1de
 Create Date: 2025-07-02 09:42:46.891585
 
 """
-from typing import Sequence, Union
 
-from alembic import op
-import sqlalchemy as sa
-import sqlmodel
-from sqlalchemy.engine.reflection import Inspector
-from langflow.utils import migration
+from collections.abc import Sequence
 
+import sqlalchemy as sa
+from alembic import op
 
 # revision identifiers, used by Alembic.
-revision: str = 'd9a6ea21edcd'
-down_revision: Union[str, None] = '66f72f04a1de'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "d9a6ea21edcd"
+down_revision: str | None = "66f72f04a1de"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/dd9e0804ebd1_add_v2_file_table.py

@@ -5,19 +5,20 @@ Revises: e3162c1804e6
 Create Date: 2025-02-03 11:47:16.101523
 
 """
-from typing import Sequence, Union
 
-from alembic import op
+from collections.abc import Sequence
+
 import sqlalchemy as sa
 import sqlmodel
-from langflow.utils import migration
+from alembic import op
 
+from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
-revision: str = 'dd9e0804ebd1'
-down_revision: Union[str, None] = 'e3162c1804e6'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "dd9e0804ebd1"
+down_revision: str | None = "e3162c1804e6"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/e3162c1804e6_add_persistent_locked_state.py

@@ -5,18 +5,18 @@ Revises: 1eab2c3eb45e
 Create Date: 2024-11-07 14:50:35.201760
 
 """
-from typing import Sequence, Union
+
+from collections.abc import Sequence
 
 import sqlalchemy as sa
-import sqlmodel
 from alembic import op
 from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
-revision: str = 'e3162c1804e6'
-down_revision: Union[str, None] = '1eab2c3eb45e'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "e3162c1804e6"
+down_revision: str | None = "1eab2c3eb45e"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
@@ -25,9 +25,9 @@ def upgrade() -> None:
     table_names = inspector.get_table_names()  # noqa
     column_names = [column["name"] for column in inspector.get_columns("flow")]
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('flow', schema=None) as batch_op:
+    with op.batch_alter_table("flow", schema=None) as batch_op:
         if "locked" not in column_names:
-            batch_op.add_column(sa.Column('locked', sa.Boolean(), nullable=True))
+            batch_op.add_column(sa.Column("locked", sa.Boolean(), nullable=True))
     # ### end Alembic commands ###
 
 
@@ -37,7 +37,7 @@ def downgrade() -> None:
     table_names = inspector.get_table_names()  # noqa
     column_names = [column["name"] for column in inspector.get_columns("flow")]
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('flow', schema=None) as batch_op:
+    with op.batch_alter_table("flow", schema=None) as batch_op:
         if "locked" in column_names:
-            batch_op.drop_column('locked')
+            batch_op.drop_column("locked")
     # ### end Alembic commands ###

langflow/alembic/versions/e3bc869fa272_fix_nullable.py

@@ -6,17 +6,16 @@ Create Date: 2024-04-10 19:17:22.820455
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "e3bc869fa272"
-down_revision: Union[str, None] = "1a110b568907"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "1a110b568907"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/e56d87f8994a_add_optins_column_to_user.py

@@ -5,34 +5,33 @@ Revises: 1b8b740a6fa3
 Create Date: 2025-04-09 15:57:46.904977
 
 """
-from typing import Sequence, Union
 
-from alembic import op
+from collections.abc import Sequence
+
 import sqlalchemy as sa
-import sqlmodel
-from sqlalchemy.engine.reflection import Inspector
-from langflow.utils import migration
+from alembic import op
 
+from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
-revision: str = 'e56d87f8994a'
-down_revision: Union[str, None] = '1b8b740a6fa3'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "e56d87f8994a"
+down_revision: str | None = "1b8b740a6fa3"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
     conn = op.get_bind()
     # ### commands auto generated by Alembic - please adjust! ###
-    if not migration.column_exists(table_name='user', column_name='optins', conn=conn):
-        with op.batch_alter_table('user', schema=None) as batch_op:
-            batch_op.add_column(sa.Column('optins', sa.JSON(), nullable=True))
+    if not migration.column_exists(table_name="user", column_name="optins", conn=conn):
+        with op.batch_alter_table("user", schema=None) as batch_op:
+            batch_op.add_column(sa.Column("optins", sa.JSON(), nullable=True))
     # ### end Alembic commands ###
 
 
 def downgrade() -> None:
     conn = op.get_bind()
     # ### commands auto generated by Alembic - please adjust! ###
-    with op.batch_alter_table('user', schema=None) as batch_op:
-        batch_op.drop_column('optins')
+    with op.batch_alter_table("user", schema=None) as batch_op:
+        batch_op.drop_column("optins")
     # ### end Alembic commands ###

langflow/alembic/versions/e5a65ecff2cd_nullable_in_vertex_build.py

@@ -6,19 +6,18 @@ Create Date: 2024-09-02 14:55:19.707355
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
 revision: str = "e5a65ecff2cd"
-down_revision: Union[str, None] = "4522eb831f5c"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "4522eb831f5c"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/eb5866d51fd2_change_columns_to_be_nullable.py

@@ -6,22 +6,21 @@ Create Date: 2023-10-04 10:18:25.640458
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 from alembic import op
 
 # revision identifiers, used by Alembic.
 revision: str = "eb5866d51fd2"
-down_revision: Union[str, None] = "67cc006d50bf"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "67cc006d50bf"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
     connection = op.get_bind()  # noqa
 
-    pass
     # ### end Alembic commands ###
 
 

langflow/alembic/versions/eb5e72293a8e_add_error_and_edit_flags_to_message.py

@@ -6,17 +6,16 @@ Create Date: 2024-09-19 16:18:50.828648
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "eb5e72293a8e"
-down_revision: Union[str, None] = "5ace73a7f223"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "5ace73a7f223"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:

langflow/alembic/versions/f3b2d1f1002d_add_column_access_type_to_flow.py

@@ -5,33 +5,37 @@ Revises: 93e2705fa8d6
 Create Date: 2025-02-05 14:35:29.658101
 
 """
-from typing import Sequence, Union
 
-from alembic import op
+from collections.abc import Sequence
+
 import sqlalchemy as sa
-from langflow.utils import migration
+from alembic import op
 
+from langflow.utils import migration
 
 # revision identifiers, used by Alembic.
-revision: str = 'f3b2d1f1002d'
-down_revision: Union[str, None] = '93e2705fa8d6'
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+revision: str = "f3b2d1f1002d"
+down_revision: str | None = "93e2705fa8d6"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
     conn = op.get_bind()
-    access_type_enum = sa.Enum('PRIVATE', 'PUBLIC', name='access_type_enum')
+    access_type_enum = sa.Enum("PRIVATE", "PUBLIC", name="access_type_enum")
     access_type_enum.create(conn, checkfirst=True)
-    with op.batch_alter_table('flow', schema=None) as batch_op:
-        if not migration.column_exists(table_name='flow', column_name='access_type', conn=conn):
-            batch_op.add_column(sa.Column('access_type', access_type_enum, server_default=sa.text("'PRIVATE'"), nullable=False))
+    with op.batch_alter_table("flow", schema=None) as batch_op:
+        if not migration.column_exists(table_name="flow", column_name="access_type", conn=conn):
+            batch_op.add_column(
+                sa.Column("access_type", access_type_enum, server_default=sa.text("'PRIVATE'"), nullable=False)
+            )
+
 
 def downgrade() -> None:
     conn = op.get_bind()
-    with op.batch_alter_table('flow', schema=None) as batch_op:
-        if migration.column_exists(table_name='flow', column_name='access_type', conn=conn):
-            batch_op.drop_column('access_type')
+    with op.batch_alter_table("flow", schema=None) as batch_op:
+        if migration.column_exists(table_name="flow", column_name="access_type", conn=conn):
+            batch_op.drop_column("access_type")
 
-    access_type_enum = sa.Enum('PRIVATE', 'PUBLIC', name='access_type_enum')
+    access_type_enum = sa.Enum("PRIVATE", "PUBLIC", name="access_type_enum")
     access_type_enum.drop(conn, checkfirst=True)

langflow/alembic/versions/f5ee9749d1a6_user_id_can_be_null_in_flow.py

@@ -6,16 +6,16 @@ Create Date: 2023-10-18 23:12:27.297016
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
 
 # revision identifiers, used by Alembic.
 revision: str = "f5ee9749d1a6"
-down_revision: Union[str, None] = "7843803a87b5"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "7843803a87b5"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
@@ -25,7 +25,6 @@ def upgrade() -> None:
             batch_op.alter_column("user_id", existing_type=sa.CHAR(length=32), nullable=True)
     except Exception as e:
         print(e)
-        pass
 
     # ### end Alembic commands ###
 
@@ -37,6 +36,5 @@ def downgrade() -> None:
             batch_op.alter_column("user_id", existing_type=sa.CHAR(length=32), nullable=False)
     except Exception as e:
         print(e)
-        pass
 
     # ### end Alembic commands ###

langflow/alembic/versions/fd531f8868b1_fix_credential_table.py

@@ -6,17 +6,16 @@ Create Date: 2023-11-24 15:07:37.566516
 
 """
 
-from typing import Optional, Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from sqlalchemy.engine.reflection import Inspector
 
 # revision identifiers, used by Alembic.
 revision: str = "fd531f8868b1"
-down_revision: Union[str, None] = "2ac71eb9c3ae"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "2ac71eb9c3ae"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
@@ -35,7 +34,6 @@ def upgrade() -> None:
                 batch_op.create_foreign_key("fk_credential_user_id", "user", ["user_id"], ["id"])
     except Exception as e:
         print(e)
-        pass
 
     # ### end Alembic commands ###
 
@@ -45,7 +43,7 @@ def downgrade() -> None:
     conn = op.get_bind()
     inspector = sa.inspect(conn)  # type: ignore
     tables = inspector.get_table_names()
-    foreign_keys_names: list[Optional[str]] = []
+    foreign_keys_names: list[str | None] = []
     if "credential" in tables:
         foreign_keys = inspector.get_foreign_keys("credential")
         foreign_keys_names = [fk["name"] for fk in foreign_keys]
@@ -55,6 +53,5 @@ def downgrade() -> None:
                 batch_op.drop_constraint("fk_credential_user_id", type_="foreignkey")
     except Exception as e:
         print(e)
-        pass
 
     # ### end Alembic commands ###

langflow/api/build.py

@@ -6,6 +6,10 @@ import uuid
 from collections.abc import AsyncIterator
 
 from fastapi import BackgroundTasks, HTTPException, Response
+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
 from sqlmodel import select
 
 from langflow.api.disconnect import DisconnectHandlerStreamingResponse
@@ -19,12 +23,9 @@ from langflow.api.utils import (
     get_top_level_vertices,
     parse_exception,
 )
-from langflow.api.v1.schemas import FlowDataRequest, InputValueRequest, ResultDataResponse, VertexBuildResponse
+from langflow.api.v1.schemas import FlowDataRequest, ResultDataResponse, VertexBuildResponse
 from langflow.events.event_manager import EventManager
 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.message import ErrorMessage
 from langflow.schema.schema import OutputValue
 from langflow.services.database.models.flow.model import Flow

langflow/api/health_check_router.py

@@ -1,11 +1,11 @@
 import uuid
 
 from fastapi import APIRouter, HTTPException, status
+from lfx.log.logger import logger
 from pydantic import BaseModel
 from sqlmodel import select
 
 from langflow.api.utils import DbSession
-from langflow.logging.logger import logger
 from langflow.services.database.models.flow.model import Flow
 from langflow.services.deps import get_chat_service
 

langflow/api/limited_background_tasks.py

@@ -1,6 +1,6 @@
 from fastapi import BackgroundTasks
+from lfx.graph.utils import log_vertex_build
 
-from langflow.graph.utils import log_vertex_build
 from langflow.services.deps import get_settings_service
 
 

langflow/api/log_router.py

@@ -5,8 +5,7 @@ from typing import Annotated, Any
 
 from fastapi import APIRouter, HTTPException, Query, Request
 from fastapi.responses import JSONResponse, StreamingResponse
-
-from langflow.logging.logger import log_buffer
+from lfx.log.logger import log_buffer
 
 log_router = APIRouter(tags=["Log"])