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

langflow/components/data/url.py

@@ -3,12 +3,12 @@ import re
 import requests
 from bs4 import BeautifulSoup
 from langchain_community.document_loaders import RecursiveUrlLoader
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.field_typing.range_spec import RangeSpec
 from langflow.helpers.data import safe_convert
 from langflow.io import BoolInput, DropdownInput, IntInput, MessageTextInput, Output, SliderInput, TableInput
+from langflow.logging.logger import logger
 from langflow.schema.dataframe import DataFrame
 from langflow.schema.message import Message
 from langflow.services.deps import get_settings_service

langflow/components/datastax/astra_assistant_manager.py

@@ -4,7 +4,6 @@ from typing import TYPE_CHECKING, Any, cast
 
 from astra_assistants.astra_assistants_manager import AssistantManager
 from langchain_core.agents import AgentFinish
-from loguru import logger
 
 from langflow.base.agents.events import ExceptionWithMessageError, process_agent_events
 from langflow.base.astra_assistants.util import (
@@ -15,6 +14,7 @@ from langflow.base.astra_assistants.util import (
 )
 from langflow.custom.custom_component.component_with_cache import ComponentWithCache
 from langflow.inputs.inputs import DropdownInput, FileInput, HandleInput, MultilineInput
+from langflow.logging.logger import logger
 from langflow.memory import delete_message
 from langflow.schema.content_block import ContentBlock
 from langflow.schema.message import Message
@@ -186,8 +186,8 @@ class AstraAssistantManager(ComponentWithCache):
                 self.initialized = True
 
     async def process_inputs(self) -> None:
-        logger.info(f"env_set is {self.env_set}")
-        logger.info(self.input_tools)
+        await logger.ainfo(f"env_set is {self.env_set}")
+        await logger.ainfo(self.input_tools)
         tools = []
         tool_obj = None
         if self.input_tools is None:

langflow/components/datastax/create_assistant.py

@@ -1,8 +1,7 @@
-from loguru import logger
-
 from langflow.base.astra_assistants.util import get_patched_openai_client
 from langflow.custom.custom_component.component_with_cache import ComponentWithCache
 from langflow.inputs.inputs import MultilineInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.message import Message
 from langflow.template.field.base import Output
 

langflow/components/deactivated/merge_data.py

@@ -1,7 +1,6 @@
-from loguru import logger
-
 from langflow.custom.custom_component.component import Component
 from langflow.io import DataInput, Output
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 

langflow/components/deactivated/sub_flow.py

@@ -1,12 +1,11 @@
 from typing import TYPE_CHECKING, Any
 
-from loguru import logger
-
 from langflow.base.flow_processing.utils import build_data_from_result_data
 from langflow.custom.custom_component.custom_component import CustomComponent
 from langflow.graph.graph.base import Graph
 from langflow.graph.vertex.base import Vertex
 from langflow.helpers.flow import get_flow_inputs
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.schema.dotdict import dotdict
 from langflow.template.field.base import Input
@@ -36,7 +35,7 @@ class SubFlowComponent(CustomComponent):
         return None
 
     async def update_build_config(self, build_config: dotdict, field_value: Any, field_name: str | None = None):
-        logger.debug(f"Updating build config with field value {field_value} and field name {field_name}")
+        await logger.adebug(f"Updating build config with field value {field_value} and field name {field_name}")
         if field_name == "flow_name":
             build_config["flow_name"]["options"] = await self.get_flow_names()
         # Clean up the build config
@@ -47,11 +46,11 @@ class SubFlowComponent(CustomComponent):
             try:
                 flow_data = await self.get_flow(field_value)
             except Exception:  # noqa: BLE001
-                logger.exception(f"Error getting flow {field_value}")
+                await logger.aexception(f"Error getting flow {field_value}")
             else:
                 if not flow_data:
                     msg = f"Flow {field_value} not found."
-                    logger.error(msg)
+                    await logger.aerror(msg)
                 else:
                     try:
                         graph = Graph.from_payload(flow_data.data["data"])
@@ -60,7 +59,7 @@ class SubFlowComponent(CustomComponent):
                         # Add inputs to the build config
                         build_config = self.add_inputs_to_build_config(inputs, build_config)
                     except Exception:  # noqa: BLE001
-                        logger.exception(f"Error building graph for flow {field_value}")
+                        await logger.aexception(f"Error building graph for flow {field_value}")
 
         return build_config
 
@@ -121,5 +120,5 @@ class SubFlowComponent(CustomComponent):
                     data.extend(build_data_from_result_data(output))
 
         self.status = data
-        logger.debug(data)
+        await logger.adebug(data)
         return data

langflow/components/deactivated/vectara_self_query.py

@@ -68,9 +68,9 @@ class VectaraSelfQueryRetriverComponent(CustomComponent):
             metadata_field_obj.append(attribute_info)
 
         return SelfQueryRetriever.from_llm(
-            self.llm,  # noqa: ignore[attr-defined]
-            self.vectorstore,  # noqa: ignore[attr-defined]
-            self.document_content_description,  # noqa: ignore[attr-defined]
+            self.llm,  # type: ignore[attr-defined]
+            self.vectorstore,  # type: ignore[attr-defined]
+            self.document_content_description,  # type: ignore[attr-defined]
             metadata_field_obj,
             verbose=True,
         )

langflow/components/docling/__init__.py

@@ -1,13 +1,7 @@
 from __future__ import annotations
 
-import signal
-import sys
-import traceback
-from contextlib import suppress
 from typing import TYPE_CHECKING, Any
 
-from loguru import logger
-
 from langflow.components._importing import import_mod
 
 if TYPE_CHECKING:
@@ -47,195 +41,3 @@ def __getattr__(attr_name: str) -> Any:
 
 def __dir__() -> list[str]:
     return list(__all__)
-
-
-def docling_worker(file_paths: list[str], queue, pipeline: str, ocr_engine: str):
-    """Worker function for processing files with Docling in a separate process."""
-    # Signal handling for graceful shutdown
-    shutdown_requested = False
-
-    def signal_handler(signum: int, frame) -> None:  # noqa: ARG001
-        """Handle shutdown signals gracefully."""
-        nonlocal shutdown_requested
-        signal_names: dict[int, str] = {signal.SIGTERM: "SIGTERM", signal.SIGINT: "SIGINT"}
-        signal_name = signal_names.get(signum, f"signal {signum}")
-
-        logger.debug(f"Docling worker received {signal_name}, initiating graceful shutdown...")
-        shutdown_requested = True
-
-        # Send shutdown notification to parent process
-        with suppress(Exception):
-            queue.put({"error": f"Worker interrupted by {signal_name}", "shutdown": True})
-
-        # Exit gracefully
-        sys.exit(0)
-
-    def check_shutdown() -> None:
-        """Check if shutdown was requested and exit if so."""
-        if shutdown_requested:
-            logger.info("Shutdown requested, exiting worker...")
-
-            with suppress(Exception):
-                queue.put({"error": "Worker shutdown requested", "shutdown": True})
-
-            sys.exit(0)
-
-    # Register signal handlers early
-    try:
-        signal.signal(signal.SIGTERM, signal_handler)
-        signal.signal(signal.SIGINT, signal_handler)
-        logger.debug("Signal handlers registered for graceful shutdown")
-    except (OSError, ValueError) as e:
-        # Some signals might not be available on all platforms
-        logger.warning(f"Warning: Could not register signal handlers: {e}")
-
-    # Check for shutdown before heavy imports
-    check_shutdown()
-
-    try:
-        from docling.datamodel.base_models import ConversionStatus, InputFormat
-        from docling.datamodel.pipeline_options import (
-            OcrOptions,
-            PdfPipelineOptions,
-            VlmPipelineOptions,
-        )
-        from docling.document_converter import DocumentConverter, FormatOption, PdfFormatOption
-        from docling.models.factories import get_ocr_factory
-        from docling.pipeline.vlm_pipeline import VlmPipeline
-
-        # Check for shutdown after imports
-        check_shutdown()
-        logger.debug("Docling dependencies loaded successfully")
-
-    except ModuleNotFoundError:
-        msg = (
-            "Docling is an optional dependency of Langflow. "
-            "Install with `uv pip install 'langflow[docling]'` "
-            "or refer to the documentation"
-        )
-        queue.put({"error": msg})
-        return
-    except ImportError as e:
-        # A different import failed (e.g., a transitive dependency); preserve details.
-        queue.put({"error": f"Failed to import a Docling dependency: {e}"})
-        return
-    except KeyboardInterrupt:
-        logger.warning("KeyboardInterrupt during imports, exiting...")
-        queue.put({"error": "Worker interrupted during imports", "shutdown": True})
-        return
-
-    # Configure the standard PDF pipeline
-    def _get_standard_opts() -> PdfPipelineOptions:
-        check_shutdown()  # Check before heavy operations
-
-        pipeline_options = PdfPipelineOptions()
-        pipeline_options.do_ocr = ocr_engine != ""
-        if pipeline_options.do_ocr:
-            ocr_factory = get_ocr_factory(
-                allow_external_plugins=False,
-            )
-
-            ocr_options: OcrOptions = ocr_factory.create_options(
-                kind=ocr_engine,
-            )
-            pipeline_options.ocr_options = ocr_options
-        return pipeline_options
-
-    # Configure the VLM pipeline
-    def _get_vlm_opts() -> VlmPipelineOptions:
-        check_shutdown()  # Check before heavy operations
-        return VlmPipelineOptions()
-
-    # Configure the main format options and create the DocumentConverter()
-    def _get_converter() -> DocumentConverter:
-        check_shutdown()  # Check before heavy operations
-
-        if pipeline == "standard":
-            pdf_format_option = PdfFormatOption(
-                pipeline_options=_get_standard_opts(),
-            )
-        elif pipeline == "vlm":
-            pdf_format_option = PdfFormatOption(pipeline_cls=VlmPipeline, pipeline_options=_get_vlm_opts())
-        else:
-            msg = f"Unknown pipeline: {pipeline!r}"
-            raise ValueError(msg)
-
-        format_options: dict[InputFormat, FormatOption] = {
-            InputFormat.PDF: pdf_format_option,
-            InputFormat.IMAGE: pdf_format_option,
-        }
-
-        return DocumentConverter(format_options=format_options)
-
-    try:
-        # Check for shutdown before creating converter (can be slow)
-        check_shutdown()
-        logger.info(f"Initializing {pipeline} pipeline with OCR: {ocr_engine or 'disabled'}")
-
-        converter = _get_converter()
-
-        # Check for shutdown before processing files
-        check_shutdown()
-        logger.info(f"Starting to process {len(file_paths)} files...")
-
-        # Process files with periodic shutdown checks
-        results = []
-        for i, file_path in enumerate(file_paths):
-            # Check for shutdown before processing each file
-            check_shutdown()
-
-            logger.debug(f"Processing file {i + 1}/{len(file_paths)}: {file_path}")
-
-            try:
-                # Process single file (we can't easily interrupt convert_all)
-                single_result = converter.convert_all([file_path])
-                results.extend(single_result)
-
-                # Check for shutdown after each file
-                check_shutdown()
-
-            except (OSError, ValueError, RuntimeError, ImportError) as file_error:
-                # Handle specific file processing errors
-                logger.error(f"Error processing file {file_path}: {file_error}")
-                # Continue with other files, but check for shutdown
-                check_shutdown()
-            except Exception as file_error:  # noqa: BLE001
-                # Catch any other unexpected errors to prevent worker crash
-                logger.error(f"Unexpected error processing file {file_path}: {file_error}")
-                # Continue with other files, but check for shutdown
-                check_shutdown()
-
-        # Final shutdown check before sending results
-        check_shutdown()
-
-        # Process the results while maintaining the original structure
-        processed_data = [
-            {"document": res.document, "file_path": str(res.input.file), "status": res.status.name}
-            if res.status == ConversionStatus.SUCCESS
-            else None
-            for res in results
-        ]
-
-        logger.info(f"Successfully processed {len([d for d in processed_data if d])} files")
-        queue.put(processed_data)
-
-    except KeyboardInterrupt:
-        logger.warning("KeyboardInterrupt during processing, exiting gracefully...")
-        queue.put({"error": "Worker interrupted during processing", "shutdown": True})
-        return
-    except Exception as e:  # noqa: BLE001
-        if shutdown_requested:
-            logger.exception("Exception occurred during shutdown, exiting...")
-            return
-
-        # Send any processing error to the main process with traceback
-        error_info = {"error": str(e), "traceback": traceback.format_exc()}
-        logger.error(f"Error in worker: {error_info}")
-        queue.put(error_info)
-    finally:
-        logger.info("Docling worker finishing...")
-        # Ensure we don't leave any hanging processes
-        if shutdown_requested:
-            logger.debug("Worker shutdown completed")
-        else:
-            logger.debug("Worker completed normally")

langflow/components/docling/docling_inline.py

@@ -3,7 +3,7 @@ from multiprocessing import Queue, get_context
 from queue import Empty
 
 from langflow.base.data import BaseFileComponent
-from langflow.components.docling import docling_worker
+from langflow.base.data.docling_utils import docling_worker
 from langflow.inputs import DropdownInput
 from langflow.schema import Data
 

langflow/components/embeddings/text_embedder.py

@@ -1,8 +1,8 @@
-import logging
 from typing import TYPE_CHECKING
 
 from langflow.custom.custom_component.component import Component
 from langflow.io import HandleInput, MessageInput, Output
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 if TYPE_CHECKING:
@@ -57,8 +57,8 @@ class TextEmbedderComponent(Component):
             embedding_vector = embeddings[0]
             self.status = {"text": text_content, "embeddings": embedding_vector}
             return Data(data={"text": text_content, "embeddings": embedding_vector})
-        except Exception as e:
-            logging.exception("Error generating embeddings")
+        except Exception as e:  # noqa: BLE001
+            logger.exception("Error generating embeddings")
             error_data = Data(data={"text": "", "embeddings": [], "error": str(e)})
             self.status = {"error": str(e)}
             return error_data

langflow/components/firecrawl/firecrawl_extract_api.py

@@ -1,13 +1,6 @@
-from loguru import logger
-
 from langflow.custom.custom_component.component import Component
-from langflow.io import (
-    BoolInput,
-    DataInput,
-    MultilineInput,
-    Output,
-    SecretStrInput,
-)
+from langflow.io import BoolInput, DataInput, MultilineInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 

langflow/components/google/gmail.py

@@ -11,11 +11,11 @@ from googleapiclient.discovery import build
 from langchain_core.chat_sessions import ChatSession
 from langchain_core.messages import HumanMessage
 from langchain_google_community.gmail.loader import GMailLoader
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.inputs.inputs import MessageTextInput
 from langflow.io import SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.template.field.base import Output
 

langflow/components/google/google_generative_ai.py

@@ -1,21 +1,14 @@
 from typing import Any
 
 import requests
-from loguru import logger
 from pydantic.v1 import SecretStr
 
 from langflow.base.models.google_generative_ai_constants import GOOGLE_GENERATIVE_AI_MODELS
 from langflow.base.models.model import LCModelComponent
 from langflow.field_typing import LanguageModel
 from langflow.field_typing.range_spec import RangeSpec
-from langflow.inputs.inputs import (
-    BoolInput,
-    DropdownInput,
-    FloatInput,
-    IntInput,
-    SecretStrInput,
-    SliderInput,
-)
+from langflow.inputs.inputs import BoolInput, DropdownInput, FloatInput, IntInput, SecretStrInput, SliderInput
+from langflow.logging.logger import logger
 from langflow.schema.dotdict import dotdict
 
 
@@ -105,7 +98,7 @@ class GoogleGenerativeAIComponent(LCModelComponent):
             google_api_key=SecretStr(google_api_key).get_secret_value(),
         )
 
-    def get_models(self, tool_model_enabled: bool | None = None) -> list[str]:
+    def get_models(self, *, tool_model_enabled: bool | None = None) -> list[str]:
         try:
             import google.generativeai as genai
 
@@ -145,8 +138,9 @@ class GoogleGenerativeAIComponent(LCModelComponent):
                     except (ImportError, ValueError, requests.exceptions.RequestException) as e:
                         logger.exception(f"Error getting model names: {e}")
                         ids = GOOGLE_GENERATIVE_AI_MODELS
+                build_config.setdefault("model_name", {})
                 build_config["model_name"]["options"] = ids
-                build_config["model_name"]["value"] = ids[0]
+                build_config["model_name"].setdefault("value", ids[0])
             except Exception as e:
                 msg = f"Error getting model names: {e}"
                 raise ValueError(msg) from e

langflow/components/groq/groq.py

@@ -1,5 +1,4 @@
 import requests
-from loguru import logger
 from pydantic.v1 import SecretStr
 
 from langflow.base.models.groq_constants import (
@@ -11,6 +10,7 @@ from langflow.base.models.model import LCModelComponent
 from langflow.field_typing import LanguageModel
 from langflow.field_typing.range_spec import RangeSpec
 from langflow.io import BoolInput, DropdownInput, IntInput, MessageTextInput, SecretStrInput, SliderInput
+from langflow.logging.logger import logger
 
 
 class GroqModel(LCModelComponent):
@@ -74,7 +74,7 @@ class GroqModel(LCModelComponent):
         ),
     ]
 
-    def get_models(self, tool_model_enabled: bool | None = None) -> list[str]:
+    def get_models(self, *, tool_model_enabled: bool | None = None) -> list[str]:
         try:
             url = f"{self.base_url}/openai/v1/models"
             headers = {"Authorization": f"Bearer {self.api_key}", "Content-Type": "application/json"}
@@ -114,8 +114,9 @@ class GroqModel(LCModelComponent):
                     except (ImportError, ValueError, requests.exceptions.RequestException) as e:
                         logger.exception(f"Error getting model names: {e}")
                         ids = GROQ_MODELS
+                    build_config.setdefault("model_name", {})
                     build_config["model_name"]["options"] = ids
-                    build_config["model_name"]["value"] = ids[0]
+                    build_config["model_name"].setdefault("value", ids[0])
             except Exception as e:
                 msg = f"Error getting model names: {e}"
                 raise ValueError(msg) from e

langflow/components/helpers/current_date.py

@@ -1,10 +1,9 @@
 from datetime import datetime
 from zoneinfo import ZoneInfo, available_timezones
 
-from loguru import logger
-
 from langflow.custom.custom_component.component import Component
 from langflow.io import DropdownInput, Output
+from langflow.logging.logger import logger
 from langflow.schema.message import Message
 
 
@@ -37,7 +36,7 @@ class CurrentDateComponent(Component):
             self.status = result
             return Message(text=result)
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error getting current date")
+            logger.debug("Error getting current date", exc_info=True)
             error_message = f"Error: {e}"
             self.status = error_message
             return Message(text=error_message)

langflow/components/helpers/memory.py

@@ -220,7 +220,7 @@ class MemoryComponent(Component):
                 stored = stored[-n_messages:] if order == "ASC" else stored[:n_messages]
 
         # self.status = stored
-        return cast(Data, stored)
+        return cast("Data", stored)
 
     async def retrieve_messages_as_text(self) -> Message:
         stored_text = data_to_text(self.template, await self.retrieve_messages())

langflow/components/ibm/watsonx.py

@@ -3,13 +3,13 @@ from typing import Any
 
 import requests
 from langchain_ibm import ChatWatsonx
-from loguru import logger
 from pydantic.v1 import SecretStr
 
 from langflow.base.models.model import LCModelComponent
 from langflow.field_typing import LanguageModel
 from langflow.field_typing.range_spec import RangeSpec
 from langflow.inputs.inputs import BoolInput, DropdownInput, IntInput, SecretStrInput, SliderInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.dotdict import dotdict
 
 

langflow/components/ibm/watsonx_embeddings.py

@@ -4,12 +4,12 @@ import requests
 from ibm_watsonx_ai import APIClient, Credentials
 from ibm_watsonx_ai.metanames import EmbedTextParamsMetaNames
 from langchain_ibm import WatsonxEmbeddings
-from loguru import logger
 from pydantic.v1 import SecretStr
 
 from langflow.base.embeddings.model import LCEmbeddingsModel
 from langflow.field_typing import Embeddings
 from langflow.io import BoolInput, DropdownInput, IntInput, SecretStrInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.dotdict import dotdict
 
 

langflow/components/langwatch/langwatch.py

@@ -3,7 +3,6 @@ import os
 from typing import Any
 
 import httpx
-from loguru import logger
 
 from langflow.base.langwatch.utils import get_cached_evaluators
 from langflow.custom.custom_component.component import Component
@@ -18,6 +17,7 @@ from langflow.io import (
     Output,
     SecretStrInput,
 )
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.schema.dotdict import dotdict
 
@@ -226,7 +226,7 @@ class LangWatchComponent(Component):
         if not evaluator_name:
             if self.evaluators:
                 evaluator_name = next(iter(self.evaluators))
-                logger.info(f"No evaluator was selected. Using default: {evaluator_name}")
+                await logger.ainfo(f"No evaluator was selected. Using default: {evaluator_name}")
             else:
                 return Data(
                     data={"error": "No evaluator selected and no evaluators available. Please choose an evaluator."}
@@ -237,7 +237,7 @@ class LangWatchComponent(Component):
             if not evaluator:
                 return Data(data={"error": f"Selected evaluator '{evaluator_name}' not found."})
 
-            logger.info(f"Evaluating with evaluator: {evaluator_name}")
+            await logger.ainfo(f"Evaluating with evaluator: {evaluator_name}")
 
             endpoint = f"/api/evaluations/{evaluator_name}/evaluate"
             url = f"{os.getenv('LANGWATCH_ENDPOINT', 'https://app.langwatch.ai')}{endpoint}"

langflow/components/logic/flow_tool.py

@@ -1,6 +1,5 @@
 from typing import Any
 
-from loguru import logger
 from typing_extensions import override
 
 from langflow.base.langchain_utilities.model import LCToolComponent
@@ -9,6 +8,7 @@ from langflow.field_typing import Tool
 from langflow.graph.graph.base import Graph
 from langflow.helpers.flow import get_flow_inputs
 from langflow.io import BoolInput, DropdownInput, Output, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.schema.dotdict import dotdict
 
@@ -91,7 +91,7 @@ class FlowToolComponent(LCToolComponent):
         try:
             graph.set_run_id(self.graph.run_id)
         except Exception:  # noqa: BLE001
-            logger.opt(exception=True).warning("Failed to set run_id")
+            logger.warning("Failed to set run_id", exc_info=True)
         inputs = get_flow_inputs(graph)
         tool_description = self.tool_description.strip() or flow_data.description
         tool = FlowTool(

langflow/components/logic/notify.py

@@ -85,4 +85,4 @@ class NotifyComponent(Component):
             self.status = "No record provided."
         self._vertex.is_state = True
         self.graph.activate_state_vertices(name=self.context_key, caller=self._id)
-        return cast(Data, input_value)
+        return cast("Data", input_value)

langflow/components/logic/run_flow.py

@@ -1,9 +1,8 @@
 from typing import Any
 
-from loguru import logger
-
 from langflow.base.tools.run_flow import RunFlowBaseComponent
 from langflow.helpers.flow import run_flow
+from langflow.logging.logger import logger
 from langflow.schema.dotdict import dotdict
 
 
@@ -34,7 +33,7 @@ class RunFlowComponent(RunFlowBaseComponent):
                     build_config = self.update_build_config_from_graph(build_config, graph)
                 except Exception as e:
                     msg = f"Error building graph for flow {field_value}"
-                    logger.exception(msg)
+                    await logger.aexception(msg)
                     raise RuntimeError(msg) from e
         return build_config
 

langflow/components/logic/sub_flow.py

@@ -1,13 +1,12 @@
 from typing import Any
 
-from loguru import logger
-
 from langflow.base.flow_processing.utils import build_data_from_result_data
 from langflow.custom.custom_component.component import Component
 from langflow.graph.graph.base import Graph
 from langflow.graph.vertex.base import Vertex
 from langflow.helpers.flow import get_flow_inputs
 from langflow.io import DropdownInput, Output
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.schema.dotdict import dotdict
 
@@ -41,11 +40,11 @@ class SubFlowComponent(Component):
             try:
                 flow_data = await self.get_flow(field_value)
             except Exception:  # noqa: BLE001
-                logger.exception(f"Error getting flow {field_value}")
+                await logger.aexception(f"Error getting flow {field_value}")
             else:
                 if not flow_data:
                     msg = f"Flow {field_value} not found."
-                    logger.error(msg)
+                    await logger.aerror(msg)
                 else:
                     try:
                         graph = Graph.from_payload(flow_data.data["data"])
@@ -54,7 +53,7 @@ class SubFlowComponent(Component):
                         # Add inputs to the build config
                         build_config = self.add_inputs_to_build_config(inputs, build_config)
                     except Exception:  # noqa: BLE001
-                        logger.exception(f"Error building graph for flow {field_value}")
+                        await logger.aexception(f"Error building graph for flow {field_value}")
 
         return build_config