langflow-base-nightly 0.5.0.dev37__py3-none-any.whl → 0.5.0.dev38__py3-none-any.whl

langflow/__main__.py

@@ -162,7 +162,7 @@ def wait_for_server_ready(host, port, protocol) -> None:
         except HTTPError:
             time.sleep(1)
         except Exception:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error while waiting for the server to become ready.")
+            logger.debug("Error while waiting for the server to become ready.", exc_info=True)
             time.sleep(1)
 
 

langflow/alembic/versions/4e5980a44eaa_fix_date_times_again.py

@@ -6,19 +6,19 @@ Create Date: 2024-04-12 18:11:06.454037
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from loguru import logger
 from sqlalchemy.dialects import postgresql
-from sqlalchemy.engine.reflection import Inspector
+
+from langflow.logging.logger import logger
 
 # revision identifiers, used by Alembic.
 revision: str = "4e5980a44eaa"
-down_revision: Union[str, None] = "79e675cb6752"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "79e675cb6752"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
@@ -37,11 +37,10 @@ def upgrade() -> None:
                     type_=sa.DateTime(timezone=True),
                     existing_nullable=False,
                 )
+        elif created_at_column is None:
+            logger.warning("Column 'created_at' not found in table 'apikey'")
         else:
-            if created_at_column is None:
-                logger.warning("Column 'created_at' not found in table 'apikey'")
-            else:
-                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
+            logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
     if "variable" in table_names:
         columns = inspector.get_columns("variable")
         created_at_column = next((column for column in columns if column["name"] == "created_at"), None)
@@ -54,11 +53,10 @@ def upgrade() -> None:
                     type_=sa.DateTime(timezone=True),
                     existing_nullable=True,
                 )
+            elif created_at_column is None:
+                logger.warning("Column 'created_at' not found in table 'variable'")
             else:
-                if created_at_column is None:
-                    logger.warning("Column 'created_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
             if updated_at_column is not None and isinstance(updated_at_column["type"], postgresql.TIMESTAMP):
                 batch_op.alter_column(
                     "updated_at",
@@ -66,11 +64,10 @@ def upgrade() -> None:
                     type_=sa.DateTime(timezone=True),
                     existing_nullable=True,
                 )
+            elif updated_at_column is None:
+                logger.warning("Column 'updated_at' not found in table 'variable'")
             else:
-                if updated_at_column is None:
-                    logger.warning("Column 'updated_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
 
     # ### end Alembic commands ###
 
@@ -92,11 +89,10 @@ def downgrade() -> None:
                     type_=postgresql.TIMESTAMP(),
                     existing_nullable=True,
                 )
+            elif updated_at_column is None:
+                logger.warning("Column 'updated_at' not found in table 'variable'")
             else:
-                if updated_at_column is None:
-                    logger.warning("Column 'updated_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
             if created_at_column is not None and isinstance(created_at_column["type"], sa.DateTime):
                 batch_op.alter_column(
                     "created_at",
@@ -104,11 +100,10 @@ def downgrade() -> None:
                     type_=postgresql.TIMESTAMP(),
                     existing_nullable=True,
                 )
+            elif created_at_column is None:
+                logger.warning("Column 'created_at' not found in table 'variable'")
             else:
-                if created_at_column is None:
-                    logger.warning("Column 'created_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
 
     if "apikey" in table_names:
         columns = inspector.get_columns("apikey")
@@ -121,10 +116,9 @@ def downgrade() -> None:
                     type_=postgresql.TIMESTAMP(),
                     existing_nullable=False,
                 )
+        elif created_at_column is None:
+            logger.warning("Column 'created_at' not found in table 'apikey'")
         else:
-            if created_at_column is None:
-                logger.warning("Column 'created_at' not found in table 'apikey'")
-            else:
-                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
+            logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
 
     # ### end Alembic commands ###

langflow/alembic/versions/58b28437a398_modify_nullable.py

@@ -6,16 +6,16 @@ Create Date: 2024-04-13 10:57:23.061709
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from loguru import logger
-from sqlalchemy.engine.reflection import Inspector
 
-down_revision: Union[str, None] = "4e5980a44eaa"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+from langflow.logging.logger import logger
+
+down_revision: str | None = "4e5980a44eaa"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 # Revision identifiers, used by Alembic.
 revision = "58b28437a398"

langflow/alembic/versions/79e675cb6752_change_datetime_type.py

@@ -6,19 +6,19 @@ Create Date: 2024-04-11 19:23:10.697335
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 from alembic import op
-from loguru import logger
 from sqlalchemy.dialects import postgresql
-from sqlalchemy.engine.reflection import Inspector
+
+from langflow.logging.logger import logger
 
 # revision identifiers, used by Alembic.
 revision: str = "79e675cb6752"
-down_revision: Union[str, None] = "e3bc869fa272"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "e3bc869fa272"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
@@ -37,11 +37,10 @@ def upgrade() -> None:
                     type_=sa.DateTime(timezone=True),
                     existing_nullable=False,
                 )
+        elif created_at_column is None:
+            logger.warning("Column 'created_at' not found in table 'apikey'")
         else:
-            if created_at_column is None:
-                logger.warning("Column 'created_at' not found in table 'apikey'")
-            else:
-                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
+            logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
     if "variable" in table_names:
         columns = inspector.get_columns("variable")
         created_at_column = next((column for column in columns if column["name"] == "created_at"), None)
@@ -54,11 +53,10 @@ def upgrade() -> None:
                     type_=sa.DateTime(timezone=True),
                     existing_nullable=True,
                 )
+            elif created_at_column is None:
+                logger.warning("Column 'created_at' not found in table 'variable'")
             else:
-                if created_at_column is None:
-                    logger.warning("Column 'created_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
             if updated_at_column is not None and isinstance(updated_at_column["type"], postgresql.TIMESTAMP):
                 batch_op.alter_column(
                     "updated_at",
@@ -66,11 +64,10 @@ def upgrade() -> None:
                     type_=sa.DateTime(timezone=True),
                     existing_nullable=True,
                 )
+            elif updated_at_column is None:
+                logger.warning("Column 'updated_at' not found in table 'variable'")
             else:
-                if updated_at_column is None:
-                    logger.warning("Column 'updated_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
 
     # ### end Alembic commands ###
 
@@ -92,11 +89,10 @@ def downgrade() -> None:
                     type_=postgresql.TIMESTAMP(),
                     existing_nullable=True,
                 )
+            elif updated_at_column is None:
+                logger.warning("Column 'updated_at' not found in table 'variable'")
             else:
-                if updated_at_column is None:
-                    logger.warning("Column 'updated_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'updated_at' has type {updated_at_column['type']} in table 'variable'")
             if created_at_column is not None and isinstance(created_at_column["type"], sa.DateTime):
                 batch_op.alter_column(
                     "created_at",
@@ -104,11 +100,10 @@ def downgrade() -> None:
                     type_=postgresql.TIMESTAMP(),
                     existing_nullable=True,
                 )
+            elif created_at_column is None:
+                logger.warning("Column 'created_at' not found in table 'variable'")
             else:
-                if created_at_column is None:
-                    logger.warning("Column 'created_at' not found in table 'variable'")
-                else:
-                    logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
+                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'variable'")
 
     if "apikey" in table_names:
         columns = inspector.get_columns("apikey")
@@ -121,10 +116,9 @@ def downgrade() -> None:
                     type_=postgresql.TIMESTAMP(),
                     existing_nullable=False,
                 )
+        elif created_at_column is None:
+            logger.warning("Column 'created_at' not found in table 'apikey'")
         else:
-            if created_at_column is None:
-                logger.warning("Column 'created_at' not found in table 'apikey'")
-            else:
-                logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
+            logger.warning(f"Column 'created_at' has type {created_at_column['type']} in table 'apikey'")
 
     # ### end Alembic commands ###

langflow/alembic/versions/b2fa308044b5_add_unique_constraints.py

@@ -1,4 +1,4 @@
-"""Add unique constraints
+"""Add unique constraints.
 
 Revision ID: b2fa308044b5
 Revises: 0b8757876a7c
@@ -6,25 +6,25 @@ Create Date: 2024-01-26 13:31:14.797548
 
 """
 
-from typing import Sequence, Union
+from collections.abc import Sequence
 
 import sqlalchemy as sa
 import sqlmodel
 from alembic import op
-from loguru import logger  # noqa
-from sqlalchemy.engine.reflection import Inspector
+
+from langflow.logging.logger import logger
 
 # revision identifiers, used by Alembic.
 revision: str = "b2fa308044b5"
-down_revision: Union[str, None] = "0b8757876a7c"
-branch_labels: Union[str, Sequence[str], None] = None
-depends_on: Union[str, Sequence[str], None] = None
+down_revision: str | None = "0b8757876a7c"
+branch_labels: str | Sequence[str] | None = None
+depends_on: str | Sequence[str] | None = None
 
 
 def upgrade() -> None:
     # ### commands auto generated by Alembic - please adjust! ###
     conn = op.get_bind()
-    inspector = sa.inspect(conn)  # type: ignore
+    inspector = sa.inspect(conn)
     tables = inspector.get_table_names()
     # ### commands auto generated by Alembic - please adjust! ###
     try:
@@ -53,14 +53,13 @@ def upgrade() -> None:
             if "fk_flow_user_id_user" not in constraint_names:
                 batch_op.create_foreign_key("fk_flow_user_id_user", "user", ["user_id"], ["id"])
 
-    except Exception as e:
+    except Exception as e:  # noqa: BLE001
         logger.exception(f"Error during upgrade: {e}")
-        pass
 
 
 def downgrade() -> None:
     conn = op.get_bind()
-    inspector = sa.inspect(conn)  # type: ignore
+    inspector = sa.inspect(conn)
     try:
         # Re-create the dropped table 'flowstyle' if it was previously dropped in upgrade
         if "flowstyle" not in inspector.get_table_names():
@@ -97,6 +96,6 @@ def downgrade() -> None:
             if "fk_flow_user_id_user" in constraint_names:
                 batch_op.drop_constraint("fk_flow_user_id_user", type_="foreignkey")
 
-    except Exception as e:
+    except Exception as e:  # noqa: BLE001
         # It's generally a good idea to log the exception or handle it in a way other than a bare pass
-        print(f"Error during downgrade: {e}")
+        logger.exception(f"Error during downgrade: {e}")

langflow/api/build.py

@@ -6,7 +6,6 @@ import uuid
 from collections.abc import AsyncIterator
 
 from fastapi import BackgroundTasks, HTTPException, Response
-from loguru import logger
 from sqlmodel import select
 
 from langflow.api.disconnect import DisconnectHandlerStreamingResponse
@@ -20,16 +19,12 @@ 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, InputValueRequest, 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
@@ -75,7 +70,7 @@ async def start_flow_build(
         )
         queue_service.start_job(job_id, task_coro)
     except Exception as e:
-        logger.exception("Failed to create queue and start task")
+        await logger.aexception("Failed to create queue and start task")
         raise HTTPException(status_code=500, detail=str(e)) from e
     return job_id
 
@@ -91,7 +86,7 @@ async def get_flow_events_response(
         main_queue, event_manager, event_task, _ = queue_service.get_queue_data(job_id)
         if event_delivery in (EventDeliveryType.STREAMING, EventDeliveryType.DIRECT):
             if event_task is None:
-                logger.error(f"No event task found for job {job_id}")
+                await logger.aerror(f"No event task found for job {job_id}")
                 raise HTTPException(status_code=404, detail="No event task found for job")
             return await create_flow_response(
                 queue=main_queue,
@@ -130,19 +125,19 @@ async def get_flow_events_response(
             content = "\n".join([event for event in events if event is not None])
             return Response(content=content, media_type="application/x-ndjson")
         except asyncio.CancelledError as exc:
-            logger.info(f"Event polling was cancelled for job {job_id}")
+            await logger.ainfo(f"Event polling was cancelled for job {job_id}")
             raise HTTPException(status_code=499, detail="Event polling was cancelled") from exc
         except asyncio.TimeoutError:
-            logger.warning(f"Timeout while waiting for events for job {job_id}")
+            await logger.awarning(f"Timeout while waiting for events for job {job_id}")
             return Response(content="", media_type="application/x-ndjson")  # Return empty response instead of error
 
     except JobQueueNotFoundError as exc:
-        logger.error(f"Job not found: {job_id}. Error: {exc!s}")
+        await logger.aerror(f"Job not found: {job_id}. Error: {exc!s}")
         raise HTTPException(status_code=404, detail=f"Job not found: {exc!s}") from exc
     except Exception as exc:
         if isinstance(exc, HTTPException):
             raise
-        logger.exception(f"Unexpected error processing flow events for job {job_id}")
+        await logger.aexception(f"Unexpected error processing flow events for job {job_id}")
         raise HTTPException(status_code=500, detail=f"Unexpected error: {exc!s}") from exc
 
 
@@ -161,9 +156,9 @@ async def create_flow_response(
                     break
                 get_time = time.time()
                 yield value.decode("utf-8")
-                logger.debug(f"Event {event_id} consumed in {get_time - put_time:.4f}s")
+                await logger.adebug(f"Event {event_id} consumed in {get_time - put_time:.4f}s")
             except Exception as exc:  # noqa: BLE001
-                logger.exception(f"Error consuming event: {exc}")
+                await logger.aexception(f"Error consuming event: {exc}")
                 break
 
     def on_disconnect() -> None:
@@ -233,7 +228,7 @@ async def generate_flow_events(
 
             if "stream or streaming set to True" in str(exc):
                 raise HTTPException(status_code=400, detail=str(exc)) from exc
-            logger.exception("Error checking build status")
+            await logger.aexception("Error checking build status")
             raise HTTPException(status_code=500, detail=str(exc)) from exc
         return first_layer, vertices_to_run, graph
 
@@ -317,7 +312,7 @@ async def generate_flow_events(
                     tb = exc.formatted_traceback
                 else:
                     tb = traceback.format_exc()
-                    logger.exception("Error building Component")
+                    await logger.aexception("Error building Component")
                     params = format_exception_message(exc)
                 message = {"errorMessage": params, "stackTrace": tb}
                 valid = False
@@ -390,7 +385,7 @@ async def generate_flow_events(
                     component_error_message=str(exc),
                 ),
             )
-            logger.exception("Error building Component")
+            await logger.aexception("Error building Component")
             message = parse_exception(exc)
             raise HTTPException(status_code=500, detail=message) from exc
 
@@ -411,7 +406,7 @@ async def generate_flow_events(
         try:
             vertex_build_response: VertexBuildResponse = await _build_vertex(vertex_id, graph, event_manager)
         except asyncio.CancelledError as exc:
-            logger.error(f"Build cancelled: {exc}")
+            await logger.aerror(f"Build cancelled: {exc}")
             raise
 
         # send built event or error event
@@ -459,7 +454,7 @@ async def generate_flow_events(
         background_tasks.add_task(graph.end_all_traces_in_context())
         raise
     except Exception as e:
-        logger.error(f"Error building vertices: {e}")
+        await logger.aerror(f"Error building vertices: {e}")
         custom_component = graph.get_vertex(vertex_id).custom_component
         trace_name = getattr(custom_component, "trace_name", None)
         error_message = ErrorMessage(
@@ -499,11 +494,11 @@ async def cancel_flow_build(
     _, _, event_task, _ = queue_service.get_queue_data(job_id)
 
     if event_task is None:
-        logger.warning(f"No event task found for job_id {job_id}")
+        await logger.awarning(f"No event task found for job_id {job_id}")
         return True  # Nothing to cancel is still a success
 
     if event_task.done():
-        logger.info(f"Task for job_id {job_id} is already completed")
+        await logger.ainfo(f"Task for job_id {job_id} is already completed")
         return True  # Nothing to cancel is still a success
 
     # Store the task reference to check status after cleanup
@@ -515,18 +510,18 @@ async def cancel_flow_build(
     except asyncio.CancelledError:
         # Check if the task was actually cancelled
         if task_before_cleanup.cancelled():
-            logger.info(f"Successfully cancelled flow build for job_id {job_id} (CancelledError caught)")
+            await logger.ainfo(f"Successfully cancelled flow build for job_id {job_id} (CancelledError caught)")
             return True
         # If the task wasn't cancelled, re-raise the exception
-        logger.error(f"CancelledError caught but task for job_id {job_id} was not cancelled")
+        await logger.aerror(f"CancelledError caught but task for job_id {job_id} was not cancelled")
         raise
 
     # If no exception was raised, verify that the task was actually cancelled
     # The task should be done (cancelled) after cleanup
     if task_before_cleanup.cancelled():
-        logger.info(f"Successfully cancelled flow build for job_id {job_id}")
+        await logger.ainfo(f"Successfully cancelled flow build for job_id {job_id}")
         return True
 
     # If we get here, the task wasn't cancelled properly
-    logger.error(f"Failed to cancel flow build for job_id {job_id}, task is still running")
+    await logger.aerror(f"Failed to cancel flow build for job_id {job_id}, task is still running")
     return False

langflow/api/health_check_router.py

@@ -1,11 +1,11 @@
 import uuid
 
 from fastapi import APIRouter, HTTPException, status
-from loguru 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
 
@@ -49,7 +49,7 @@ async def health_check(
         (await session.exec(stmt)).first()
         response.db = "ok"
     except Exception:  # noqa: BLE001
-        logger.exception("Error checking database")
+        await logger.aexception("Error checking database")
 
     try:
         chat = get_chat_service()
@@ -57,7 +57,7 @@ async def health_check(
         await chat.get_cache("health_check")
         response.chat = "ok"
     except Exception:  # noqa: BLE001
-        logger.exception("Error checking chat service")
+        await logger.aexception("Error checking chat service")
 
     if response.has_error():
         raise HTTPException(status_code=status.HTTP_500_INTERNAL_SERVER_ERROR, detail=response.model_dump())

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 loguru 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
@@ -119,7 +119,7 @@ async def check_langflow_version(component: StoreComponentCreate) -> None:
     if langflow_version is None:
         raise HTTPException(status_code=500, detail="Unable to verify the latest version of Langflow")
     if langflow_version != component.last_tested_version:
-        logger.warning(
+        await logger.awarning(
             f"Your version of Langflow ({component.last_tested_version}) is outdated. "
             f"Please update to the latest version ({langflow_version}) and try again."
         )
@@ -371,7 +371,7 @@ async def verify_public_flow_and_get_user(flow_id: uuid.UUID, client_id: str | N
         user = await get_user_by_flow_id_or_endpoint_name(str(flow_id))
 
     except Exception as exc:
-        logger.exception(f"Error getting user for public flow {flow_id}")
+        await logger.aexception(f"Error getting user for public flow {flow_id}")
         raise HTTPException(status_code=403, detail="Flow is not accessible") from exc
 
     if not user:

langflow/api/v1/callback.py

@@ -5,10 +5,10 @@ from uuid import UUID
 
 from langchain_core.agents import AgentAction, AgentFinish
 from langchain_core.callbacks.base import AsyncCallbackHandler
-from loguru import logger
 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
 
@@ -78,7 +78,7 @@ class AsyncStreamingLLMCallbackHandleSIO(AsyncCallbackHandler):
             for resp in resps:
                 await self.socketio_service.emit_token(to=self.sid, data=resp.model_dump())
         except Exception:  # noqa: BLE001
-            logger.exception("Error sending response")
+            await logger.aexception("Error sending response")
 
     async def on_tool_error(
         self,