lfx-nightly 0.1.12.dev27__py3-none-any.whl → 0.1.12.dev28__py3-none-any.whl

lfx/base/agents/agent.py

@@ -10,7 +10,7 @@ from langchain_core.runnables import Runnable
 
 from lfx.base.agents.callback import AgentAsyncHandler
 from lfx.base.agents.events import ExceptionWithMessageError, process_agent_events
-from lfx.base.agents.utils import data_to_messages
+from lfx.base.agents.utils import data_to_messages, get_chat_output_sender_name
 from lfx.custom.custom_component.component import Component, _get_component_toolkit
 from lfx.field_typing import Tool
 from lfx.inputs.inputs import InputTypes, MultilineInput
@@ -137,11 +137,15 @@ class LCAgentComponent(Component):
                 verbose=verbose,
                 max_iterations=max_iterations,
             )
-        input_dict: dict[str, str | list[BaseMessage]] = {
-            "input": self.input_value.to_lc_message()
-            if hasattr(self.input_value, "to_lc_message")
-            else self.input_value
-        }
+        # Convert input_value to proper format for agent
+        if hasattr(self.input_value, "to_lc_message") and callable(self.input_value.to_lc_message):
+            lc_message = self.input_value.to_lc_message()
+            input_text = lc_message.content if hasattr(lc_message, "content") else str(lc_message)
+        else:
+            lc_message = None
+            input_text = self.input_value
+
+        input_dict: dict[str, str | list[BaseMessage]] = {}
         if hasattr(self, "system_prompt"):
             input_dict["system_prompt"] = self.system_prompt
         if hasattr(self, "chat_history") and self.chat_history:
@@ -154,11 +158,13 @@ class LCAgentComponent(Component):
             # Handle both lfx.schema.message.Message and langflow.schema.message.Message types
             if all(hasattr(m, "to_data") and callable(m.to_data) and "text" in m.data for m in self.chat_history):
                 input_dict["chat_history"] = data_to_messages(self.chat_history)
-        if hasattr(input_dict["input"], "content") and isinstance(input_dict["input"].content, list):
+            if all(isinstance(m, Message) for m in self.chat_history):
+                input_dict["chat_history"] = data_to_messages([m.to_data() for m in self.chat_history])
+        if hasattr(lc_message, "content") and isinstance(lc_message.content, list):
             # ! Because the input has to be a string, we must pass the images in the chat_history
 
-            image_dicts = [item for item in input_dict["input"].content if item.get("type") == "image"]
-            input_dict["input"].content = [item for item in input_dict["input"].content if item.get("type") != "image"]
+            image_dicts = [item for item in lc_message.content if item.get("type") == "image"]
+            lc_message.content = [item for item in lc_message.content if item.get("type") != "image"]
 
             if "chat_history" not in input_dict:
                 input_dict["chat_history"] = []
@@ -166,7 +172,7 @@ class LCAgentComponent(Component):
                 input_dict["chat_history"].extend(HumanMessage(content=[image_dict]) for image_dict in image_dicts)
             else:
                 input_dict["chat_history"] = [HumanMessage(content=[image_dict]) for image_dict in image_dicts]
-
+        input_dict["input"] = input_text
         if hasattr(self, "graph"):
             session_id = self.graph.session_id
         elif hasattr(self, "_session_id"):
@@ -174,9 +180,11 @@ class LCAgentComponent(Component):
         else:
             session_id = None
 
+        sender_name = get_chat_output_sender_name(self) or self.display_name or "AI"
+
         agent_message = Message(
             sender=MESSAGE_SENDER_AI,
-            sender_name=self.display_name or "Agent",
+            sender_name=sender_name,
             properties={"icon": "Bot", "state": "partial"},
             content_blocks=[ContentBlock(title="Agent Steps", contents=[])],
             session_id=session_id or uuid.uuid4(),

lfx/base/agents/utils.py

@@ -204,3 +204,21 @@ def maybe_unflatten_dict(flat: dict[str, Any]) -> dict[str, Any]:
                 cur = cur.setdefault(part, {})
 
     return nested
+
+
+def get_chat_output_sender_name(self) -> str | None:
+    """Get sender_name from ChatOutput component."""
+    if not hasattr(self, "graph") or not self.graph:
+        return None
+
+    for vertex in self.graph.vertices:
+        # Safely check if vertex has data attribute, correct type, and raw_params
+        if (
+            hasattr(vertex, "data")
+            and vertex.data.get("type") == "ChatOutput"
+            and hasattr(vertex, "raw_params")
+            and vertex.raw_params
+        ):
+            return vertex.raw_params.get("sender_name")
+
+    return None

lfx/base/data/base_file.py

@@ -1,14 +1,13 @@
 import ast
-import json
 import shutil
 import tarfile
 from abc import ABC, abstractmethod
 from pathlib import Path
 from tempfile import TemporaryDirectory
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Any
 from zipfile import ZipFile, is_zipfile
 
-import anyio
+import orjson
 import pandas as pd
 
 from lfx.custom.custom_component.component import Component
@@ -224,7 +223,6 @@ class BaseFileComponent(Component, ABC):
             # Delete temporary directories
             for temp_dir in self._temp_dirs:
                 temp_dir.cleanup()
-
             # Delete files marked for deletion
             for file in final_files:
                 if file.delete_after_processing and file.path.exists():
@@ -244,15 +242,15 @@ class BaseFileComponent(Component, ABC):
             return [Data()]
         return data_list
 
-    async def _extract_file_metadata(self, data_item) -> dict:
+    def _extract_file_metadata(self, data_item) -> dict:
         """Extract metadata from a data item with file_path."""
-        metadata = {}
+        metadata: dict[str, Any] = {}
         if not hasattr(data_item, "file_path"):
             return metadata
 
         file_path = data_item.file_path
-        file_path_obj = anyio.Path(file_path)
-        file_size_stat = await file_path_obj.stat()
+        file_path_obj = Path(file_path)
+        file_size_stat = file_path_obj.stat()
         filename = file_path_obj.name
 
         # Basic file metadata
@@ -280,7 +278,7 @@ class BaseFileComponent(Component, ABC):
             return text if text is not None else str(data_item)
         return str(data_item)
 
-    async def load_files_message(self) -> Message:
+    def load_files_message(self) -> Message:
         """Load files and return as Message.
 
         Returns:
@@ -290,16 +288,27 @@ class BaseFileComponent(Component, ABC):
         if not data_list:
             return Message()
 
+        # Extract metadata from the first data item
+        metadata = self._extract_file_metadata(data_list[0])
+
         sep: str = getattr(self, "separator", "\n\n") or "\n\n"
         parts: list[str] = []
-        metadata = {}
-
-        for data_item in data_list:
-            parts.append(self._extract_text(data_item))
-
-            # Set metadata from first file only
-            if not metadata:
-                metadata = await self._extract_file_metadata(data_item)
+        for d in data_list:
+            try:
+                data_text = self._extract_text(d)
+                if data_text and isinstance(data_text, str):
+                    parts.append(data_text)
+                elif data_text:
+                    # get_text() returned non-string, convert it
+                    parts.append(str(data_text))
+                elif isinstance(d.data, dict):
+                    # convert the data dict to a readable string
+                    parts.append(orjson.dumps(d.data, default=str).decode())
+                else:
+                    parts.append(str(d))
+            except (AttributeError, TypeError, ValueError):
+                # Final fallback - just try to convert to string
+                parts.append(str(d))
 
         return Message(text=sep.join(parts), **metadata)
 
@@ -366,10 +375,10 @@ class BaseFileComponent(Component, ABC):
     def parse_string_to_dict(self, s: str) -> dict:
         # Try JSON first (handles true/false/null)
         try:
-            result = json.loads(s)
+            result = orjson.loads(s)
             if isinstance(result, dict):
                 return result
-        except json.JSONDecodeError:
+        except orjson.JSONDecodeError:
             pass
 
         # Fall back to Python literal evaluation

lfx/components/data/__init__.py

@@ -10,8 +10,6 @@ if TYPE_CHECKING:
     from lfx.components.data.directory import DirectoryComponent
     from lfx.components.data.file import FileComponent
     from lfx.components.data.json_to_data import JSONToDataComponent
-    from lfx.components.data.news_search import NewsSearchComponent
-    from lfx.components.data.rss import RSSReaderComponent
     from lfx.components.data.sql_executor import SQLComponent
     from lfx.components.data.url import URLComponent
     from lfx.components.data.web_search import WebSearchComponent
@@ -23,8 +21,6 @@ _dynamic_imports = {
     "DirectoryComponent": "directory",
     "FileComponent": "file",
     "JSONToDataComponent": "json_to_data",
-    "NewsSearchComponent": "news_search",
-    "RSSReaderComponent": "rss",
     "SQLComponent": "sql_executor",
     "URLComponent": "url",
     "WebSearchComponent": "web_search",
@@ -37,8 +33,6 @@ __all__ = [
     "DirectoryComponent",
     "FileComponent",
     "JSONToDataComponent",
-    "NewsSearchComponent",
-    "RSSReaderComponent",
     "SQLComponent",
     "URLComponent",
     "WebSearchComponent",

lfx/components/data/file.py

@@ -29,7 +29,7 @@ from lfx.schema.message import Message
 class FileComponent(BaseFileComponent):
     """File component with optional Docling processing (isolated in a subprocess)."""
 
-    display_name = "File"
+    display_name = "Read File"
     description = "Loads content from one or more files."
     documentation: str = "https://docs.langflow.org/components-data#file"
     icon = "file-text"

lfx/components/data/mock_data.py

@@ -20,20 +20,17 @@ class MockDataGeneratorComponent(Component):
     - Development and debugging of Langflow components
     """
 
-    display_name = "Mock Data Generator"
-    description = (
-        "Generate sample data for testing and development. "
-        "Choose from text messages, JSON data, or tabular data formats."
-    )
+    display_name = "Mock Data"
+    description = "Generate mock data for testing and development."
     icon = "database"
     name = "MockDataGenerator"
 
     inputs = []
 
     outputs = [
-        Output(display_name="DataFrame Output", name="dataframe_output", method="generate_dataframe_output"),
-        Output(display_name="Message Output", name="message_output", method="generate_message_output"),
-        Output(display_name="Data Output", name="data_output", method="generate_data_output"),
+        Output(display_name="Result", name="dataframe_output", method="generate_dataframe_output"),
+        Output(display_name="Result", name="message_output", method="generate_message_output"),
+        Output(display_name="Result", name="data_output", method="generate_data_output"),
     ]
 
     def build(self) -> DataFrame: