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

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

@@ -117,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
@@ -202,11 +202,11 @@ class MCPToolsComponent(ComponentWithCache):
 
         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}
@@ -223,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"] = []
@@ -249,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:
@@ -333,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
@@ -386,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:
@@ -404,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:
@@ -428,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:
@@ -474,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}"})