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

langflow/base/prompts/api_utils.py

@@ -3,10 +3,10 @@ from typing import Any
 
 from fastapi import HTTPException
 from langchain_core.prompts import PromptTemplate
-from loguru import logger
 
 from langflow.inputs.inputs import DefaultPromptField
 from langflow.interface.utils import extract_input_variables_from_prompt
+from langflow.logging.logger import logger
 
 _INVALID_CHARACTERS = {
     " ",

langflow/base/tools/flow_tool.py

@@ -3,13 +3,13 @@ from __future__ import annotations
 from typing import TYPE_CHECKING, Any
 
 from langchain_core.tools import BaseTool, ToolException
-from loguru import logger
 from typing_extensions import override
 
 from langflow.base.flow_processing.utils import build_data_from_result_data, format_flow_output_data
 from langflow.graph.graph.base import Graph  # cannot be a part of TYPE_CHECKING   # noqa: TC001
 from langflow.graph.vertex.base import Vertex  # cannot be a part of TYPE_CHECKING  # noqa: TC001
 from langflow.helpers.flow import build_schema_from_inputs, get_arg_names, get_flow_inputs, run_flow
+from langflow.logging.logger import logger
 from langflow.utils.async_helpers import run_until_complete
 
 if TYPE_CHECKING:
@@ -109,7 +109,7 @@ class FlowTool(BaseTool):
         try:
             run_id = self.graph.run_id if hasattr(self, "graph") and self.graph else None
         except Exception:  # noqa: BLE001
-            logger.opt(exception=True).warning("Failed to set run_id")
+            logger.warning("Failed to set run_id", exc_info=True)
             run_id = None
         run_outputs = await run_flow(
             tweaks={key: {"input_value": value} for key, value in tweaks.items()},

langflow/base/tools/run_flow.py

@@ -1,7 +1,6 @@
 from abc import abstractmethod
 from typing import TYPE_CHECKING
 
-from loguru import logger
 from typing_extensions import override
 
 from langflow.custom.custom_component.component import Component, _get_component_toolkit
@@ -9,11 +8,8 @@ from langflow.field_typing import Tool
 from langflow.graph.graph.base import Graph
 from langflow.graph.vertex.base import Vertex
 from langflow.helpers.flow import get_flow_inputs
-from langflow.inputs.inputs import (
-    DropdownInput,
-    InputTypes,
-    MessageInput,
-)
+from langflow.inputs.inputs import DropdownInput, InputTypes, MessageInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 from langflow.schema.dataframe import DataFrame
 from langflow.schema.dotdict import dotdict

langflow/components/Notion/add_content_to_page.py

@@ -4,13 +4,13 @@ from typing import Any
 import requests
 from bs4 import BeautifulSoup
 from langchain.tools import StructuredTool
-from loguru import logger
 from markdown import markdown
 from pydantic import BaseModel, Field
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.field_typing import Tool
 from langflow.inputs.inputs import MultilineInput, SecretStrInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 MIN_ROWS_IN_TABLE = 3
@@ -84,7 +84,7 @@ class AddContentToPage(LCToolComponent):
                 error_message += f" Status code: {e.response.status_code}, Response: {e.response.text}"
             return error_message
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error adding content to Notion page")
+            logger.debug("Error adding content to Notion page", exc_info=True)
             return f"Error: An unexpected error occurred while adding content to Notion page. {e}"
 
     def process_node(self, node):

langflow/components/Notion/list_database_properties.py

@@ -1,11 +1,11 @@
 import requests
 from langchain.tools import StructuredTool
-from loguru import logger
 from pydantic import BaseModel, Field
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.field_typing import Tool
 from langflow.inputs.inputs import SecretStrInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -64,5 +64,5 @@ class NotionDatabaseProperties(LCToolComponent):
         except ValueError as e:
             return f"Error parsing Notion API response: {e}"
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error fetching Notion database properties")
+            logger.debug("Error fetching Notion database properties", exc_info=True)
             return f"An unexpected error occurred: {e}"

langflow/components/Notion/list_pages.py

@@ -3,12 +3,12 @@ from typing import Any
 
 import requests
 from langchain.tools import StructuredTool
-from loguru import logger
 from pydantic import BaseModel, Field
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.field_typing import Tool
 from langflow.inputs.inputs import MultilineInput, SecretStrInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -118,5 +118,5 @@ class NotionListPages(LCToolComponent):
         except KeyError:
             return "Unexpected response format from Notion API"
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error querying Notion database")
+            logger.debug("Error querying Notion database", exc_info=True)
             return f"An unexpected error occurred: {e}"

langflow/components/Notion/page_content_viewer.py

@@ -1,11 +1,11 @@
 import requests
 from langchain.tools import StructuredTool
-from loguru import logger
 from pydantic import BaseModel, Field
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.field_typing import Tool
 from langflow.inputs.inputs import SecretStrInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -65,7 +65,7 @@ class NotionPageContent(LCToolComponent):
                 error_message += f" Status code: {e.response.status_code}, Response: {e.response.text}"
             return error_message
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error retrieving Notion page content")
+            logger.debug("Error retrieving Notion page content", exc_info=True)
             return f"Error: An unexpected error occurred while retrieving Notion page content. {e}"
 
     def parse_blocks(self, blocks: list) -> str:

langflow/components/Notion/update_page_property.py

@@ -3,12 +3,12 @@ from typing import Any
 
 import requests
 from langchain.tools import StructuredTool
-from loguru import logger
 from pydantic import BaseModel, Field
 
 from langflow.base.langchain_utilities.model import LCToolComponent
 from langflow.field_typing import Tool
 from langflow.inputs.inputs import MultilineInput, SecretStrInput, StrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 

langflow/components/agentql/agentql_api.py

@@ -1,17 +1,9 @@
 import httpx
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.field_typing.range_spec import RangeSpec
-from langflow.io import (
-    BoolInput,
-    DropdownInput,
-    IntInput,
-    MessageTextInput,
-    MultilineInput,
-    Output,
-    SecretStrInput,
-)
+from langflow.io import BoolInput, DropdownInput, IntInput, MessageTextInput, MultilineInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 

langflow/components/agents/agent.py

@@ -135,13 +135,13 @@ class AgentComponent(ToolCallingAgentComponent):
             # return result
 
         except (ValueError, TypeError, KeyError) as e:
-            logger.error(f"{type(e).__name__}: {e!s}")
+            await logger.aerror(f"{type(e).__name__}: {e!s}")
             raise
         except ExceptionWithMessageError as e:
-            logger.error(f"ExceptionWithMessageError occurred: {e}")
+            await logger.aerror(f"ExceptionWithMessageError occurred: {e}")
             raise
         except Exception as e:
-            logger.error(f"Unexpected error: {e!s}")
+            await logger.aerror(f"Unexpected error: {e!s}")
             raise
         else:
             return result

langflow/components/agents/mcp_component.py

@@ -16,16 +16,15 @@ from langflow.base.mcp.util import (
 )
 from langflow.custom.custom_component.component_with_cache import ComponentWithCache
 from langflow.inputs.inputs import InputTypes  # noqa: TC001
-from langflow.io import DropdownInput, McpInput, MessageTextInput, Output, SecretStrInput
+from langflow.io import DropdownInput, McpInput, MessageTextInput, Output
 from langflow.io.schema import flatten_schema, schema_to_langflow_inputs
 from langflow.logging import logger
 from langflow.schema.dataframe import DataFrame
 from langflow.schema.message import Message
 
 # Import get_server from the backend API
-from langflow.services.auth.utils import create_user_longterm_token, get_current_user
 from langflow.services.database.models.user.crud import get_user_by_id
-from langflow.services.deps import get_session, get_settings_service, get_storage_service
+from langflow.services.deps import get_settings_service, get_storage_service, session_scope
 
 
 class MCPToolsComponent(ComponentWithCache):
@@ -96,13 +95,6 @@ class MCPToolsComponent(ComponentWithCache):
             show=False,
             tool_mode=False,
         ),
-        SecretStrInput(
-            name="api_key",
-            display_name="Langflow API Key",
-            info="Langflow API key for authentication when fetching MCP servers and tools.",
-            required=False,
-            advanced=True,
-        ),
     ]
 
     outputs = [
@@ -125,12 +117,12 @@ class MCPToolsComponent(ComponentWithCache):
             schema_inputs = schema_to_langflow_inputs(input_schema)
             if not schema_inputs:
                 msg = f"No input parameters defined for tool '{tool_obj.name}'"
-                logger.warning(msg)
+                await logger.awarning(msg)
                 return []
 
         except Exception as e:
             msg = f"Error validating schema inputs: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             raise ValueError(msg) from e
         else:
             return schema_inputs
@@ -161,19 +153,11 @@ class MCPToolsComponent(ComponentWithCache):
             return self.tools, {"name": server_name, "config": server_config_from_value}
 
         try:
-            async for db in get_session():
-                # TODO: In 1.6, this may need to be removed or adjusted
-                # Try to get the super user token, if possible
-                if self.api_key:
-                    current_user = await get_current_user(
-                        token=None,
-                        query_param=self.api_key,
-                        header_param=None,
-                        db=db,
-                    )
-                else:
-                    user_id, _ = await create_user_longterm_token(db)
-                    current_user = await get_user_by_id(db, user_id)
+            async with session_scope() as db:
+                if not self.user_id:
+                    msg = "User ID is required for fetching MCP tools."
+                    raise ValueError(msg)
+                current_user = await get_user_by_id(db, self.user_id)
 
                 # Try to get server config from DB/API
                 server_config = await get_server(
@@ -184,47 +168,48 @@ class MCPToolsComponent(ComponentWithCache):
                     settings_service=get_settings_service(),
                 )
 
-                # If get_server returns empty but we have a config, use it
-                if not server_config and server_config_from_value:
-                    server_config = server_config_from_value
-
-                if not server_config:
-                    self.tools = []
-                    return [], {"name": server_name, "config": server_config}
+            # If get_server returns empty but we have a config, use it
+            if not server_config and server_config_from_value:
+                server_config = server_config_from_value
+
+            if not server_config:
+                self.tools = []
+                return [], {"name": server_name, "config": server_config}
+
+            _, tool_list, tool_cache = await update_tools(
+                server_name=server_name,
+                server_config=server_config,
+                mcp_stdio_client=self.stdio_client,
+                mcp_sse_client=self.sse_client,
+            )
+
+            self.tool_names = [tool.name for tool in tool_list if hasattr(tool, "name")]
+            self._tool_cache = tool_cache
+            self.tools = tool_list
+            # Cache the result using shared cache
+            cache_data = {
+                "tools": tool_list,
+                "tool_names": self.tool_names,
+                "tool_cache": tool_cache,
+                "config": server_config,
+            }
+
+            # Safely update the servers cache
+            current_servers_cache = safe_cache_get(self._shared_component_cache, "servers", {})
+            if isinstance(current_servers_cache, dict):
+                current_servers_cache[server_name] = cache_data
+                safe_cache_set(self._shared_component_cache, "servers", current_servers_cache)
 
-                _, tool_list, tool_cache = await update_tools(
-                    server_name=server_name,
-                    server_config=server_config,
-                    mcp_stdio_client=self.stdio_client,
-                    mcp_sse_client=self.sse_client,
-                )
-
-                self.tool_names = [tool.name for tool in tool_list if hasattr(tool, "name")]
-                self._tool_cache = tool_cache
-                self.tools = tool_list
-                # Cache the result using shared cache
-                cache_data = {
-                    "tools": tool_list,
-                    "tool_names": self.tool_names,
-                    "tool_cache": tool_cache,
-                    "config": server_config,
-                }
-
-                # Safely update the servers cache
-                current_servers_cache = safe_cache_get(self._shared_component_cache, "servers", {})
-                if isinstance(current_servers_cache, dict):
-                    current_servers_cache[server_name] = cache_data
-                    safe_cache_set(self._shared_component_cache, "servers", current_servers_cache)
-
-                return tool_list, {"name": server_name, "config": server_config}
         except (TimeoutError, asyncio.TimeoutError) as e:
             msg = f"Timeout updating tool list: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             raise TimeoutError(msg) from e
         except Exception as e:
             msg = f"Error updating tool list: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             raise ValueError(msg) from e
+        else:
+            return tool_list, {"name": server_name, "config": server_config}
 
     async def update_build_config(self, build_config: dict, field_value: str, field_name: str | None = None) -> dict:
         """Toggle the visibility of connection-specific fields based on the selected mode."""
@@ -238,7 +223,7 @@ class MCPToolsComponent(ComponentWithCache):
                             build_config["tool"]["placeholder"] = "Select a tool"
                         except (TimeoutError, asyncio.TimeoutError) as e:
                             msg = f"Timeout updating tool list: {e!s}"
-                            logger.exception(msg)
+                            await logger.aexception(msg)
                             if not build_config["tools_metadata"]["show"]:
                                 build_config["tool"]["show"] = True
                                 build_config["tool"]["options"] = []
@@ -264,7 +249,7 @@ class MCPToolsComponent(ComponentWithCache):
                             break
                     if tool_obj is None:
                         msg = f"Tool {field_value} not found in available tools: {self.tools}"
-                        logger.warning(msg)
+                        await logger.awarning(msg)
                         return build_config
                     await self._update_tool_config(build_config, field_value)
                 except Exception as e:
@@ -348,7 +333,7 @@ class MCPToolsComponent(ComponentWithCache):
 
         except Exception as e:
             msg = f"Error in update_build_config: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             raise ValueError(msg) from e
         else:
             return build_config
@@ -401,7 +386,7 @@ class MCPToolsComponent(ComponentWithCache):
             msg = f"Tool {tool_name} not found in available tools: {self.tools}"
             self.remove_non_default_keys(build_config)
             build_config["tool"]["value"] = ""
-            logger.warning(msg)
+            await logger.awarning(msg)
             return
 
         try:
@@ -419,14 +404,14 @@ class MCPToolsComponent(ComponentWithCache):
             self.schema_inputs = await self._validate_schema_inputs(tool_obj)
             if not self.schema_inputs:
                 msg = f"No input parameters to configure for tool '{tool_name}'"
-                logger.info(msg)
+                await logger.ainfo(msg)
                 return
 
             # Add new inputs to build config
             for schema_input in self.schema_inputs:
                 if not schema_input or not hasattr(schema_input, "name"):
                     msg = "Invalid schema input detected, skipping"
-                    logger.warning(msg)
+                    await logger.awarning(msg)
                     continue
 
                 try:
@@ -443,16 +428,16 @@ class MCPToolsComponent(ComponentWithCache):
 
                 except (AttributeError, KeyError, TypeError) as e:
                     msg = f"Error processing schema input {schema_input}: {e!s}"
-                    logger.exception(msg)
+                    await logger.aexception(msg)
                     continue
         except ValueError as e:
             msg = f"Schema validation error for tool {tool_name}: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             self.schema_inputs = []
             return
         except (AttributeError, KeyError, TypeError) as e:
             msg = f"Error updating tool config: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             raise ValueError(msg) from e
 
     async def build_output(self) -> DataFrame:
@@ -489,7 +474,7 @@ class MCPToolsComponent(ComponentWithCache):
             return DataFrame(data=[{"error": "You must select a tool"}])
         except Exception as e:
             msg = f"Error in build_output: {e!s}"
-            logger.exception(msg)
+            await logger.aexception(msg)
             raise ValueError(msg) from e
 
     def _get_session_context(self) -> str | None:

langflow/components/anthropic/anthropic.py

@@ -1,7 +1,6 @@
 from typing import Any, cast
 
 import requests
-from loguru import logger
 from pydantic import ValidationError
 
 from langflow.base.models.anthropic_constants import (
@@ -14,6 +13,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
 from langflow.schema.dotdict import dotdict
 
 
@@ -101,7 +101,7 @@ class AnthropicModelComponent(LCModelComponent):
 
         return output
 
-    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 anthropic
 
@@ -129,7 +129,7 @@ class AnthropicModelComponent(LCModelComponent):
                 model_with_tool = ChatAnthropic(
                     model=model,  # Use the current model being checked
                     anthropic_api_key=self.api_key,
-                    anthropic_api_url=cast(str, self.base_url) or DEFAULT_ANTHROPIC_API_URL,
+                    anthropic_api_url=cast("str", self.base_url) or DEFAULT_ANTHROPIC_API_URL,
                 )
 
                 if (
@@ -177,8 +177,9 @@ class AnthropicModelComponent(LCModelComponent):
                     except (ImportError, ValueError, requests.exceptions.RequestException) as e:
                         logger.exception(f"Error getting model names: {e}")
                         ids = ANTHROPIC_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])
                 build_config["model_name"]["combobox"] = True
             except Exception as e:
                 msg = f"Error getting model names: {e}"

langflow/components/assemblyai/assemblyai_get_subtitles.py

@@ -1,8 +1,8 @@
 import assemblyai as aai
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.io import DataInput, DropdownInput, IntInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -58,7 +58,7 @@ class AssemblyAIGetSubtitles(Component):
             transcript = aai.Transcript.get_by_id(transcript_id)
         except Exception as e:  # noqa: BLE001
             error = f"Getting transcription failed: {e}"
-            logger.opt(exception=True).debug(error)
+            logger.debug(error, exc_info=True)
             self.status = error
             return Data(data={"error": error})
 

langflow/components/assemblyai/assemblyai_lemur.py

@@ -1,8 +1,8 @@
 import assemblyai as aai
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.io import DataInput, DropdownInput, FloatInput, IntInput, MultilineInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -131,7 +131,7 @@ class AssemblyAILeMUR(Component):
         try:
             response = self.perform_lemur_action(transcript_group, self.endpoint)
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error running LeMUR")
+            logger.debug("Error running LeMUR", exc_info=True)
             error = f"An Error happened: {e}"
             self.status = error
             return Data(data={"error": error})

langflow/components/assemblyai/assemblyai_list_transcripts.py

@@ -1,8 +1,8 @@
 import assemblyai as aai
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.io import BoolInput, DropdownInput, IntInput, MessageTextInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -86,7 +86,7 @@ class AssemblyAIListTranscripts(Component):
                 transcripts = convert_page_to_data_list(page)
 
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error listing transcripts")
+            logger.debug("Error listing transcripts", exc_info=True)
             error_data = Data(data={"error": f"An error occurred: {e}"})
             self.status = [error_data]
             return [error_data]

langflow/components/assemblyai/assemblyai_poll_transcript.py

@@ -1,9 +1,9 @@
 import assemblyai as aai
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.field_typing.range_spec import RangeSpec
 from langflow.io import DataInput, FloatInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -54,7 +54,7 @@ class AssemblyAITranscriptionJobPoller(Component):
             transcript = aai.Transcript.get_by_id(self.transcript_id.data["transcript_id"])
         except Exception as e:  # noqa: BLE001
             error = f"Getting transcription failed: {e}"
-            logger.opt(exception=True).debug(error)
+            logger.debug(error, exc_info=True)
             self.status = error
             return Data(data={"error": error})
 

langflow/components/assemblyai/assemblyai_start_transcript.py

@@ -1,10 +1,10 @@
 from pathlib import Path
 
 import assemblyai as aai
-from loguru import logger
 
 from langflow.custom.custom_component.component import Component
 from langflow.io import BoolInput, DropdownInput, FileInput, MessageTextInput, Output, SecretStrInput
+from langflow.logging.logger import logger
 from langflow.schema.data import Data
 
 
@@ -176,7 +176,7 @@ class AssemblyAITranscriptionJobCreator(Component):
         try:
             transcript = aai.Transcriber().submit(audio, config=config)
         except Exception as e:  # noqa: BLE001
-            logger.opt(exception=True).debug("Error submitting transcription job")
+            logger.debug("Error submitting transcription job", exc_info=True)
             self.status = f"An error occurred: {e}"
             return Data(data={"error": f"An error occurred: {e}"})