From 49bdbef548ec084f8f17a4df6748b4177ef6e879 Mon Sep 17 00:00:00 2001
From: Rodrigo <rodrigosilvanader@gmail.com>
Date: Mon, 10 Jun 2024 21:27:47 -0300
Subject: [PATCH] refactor: Improve advanced options in ChatComponent,
 URLComponent, and MessageHistoryComponent

---
 src/backend/base/langflow/base/io/chat.py              |  2 +-
 src/backend/base/langflow/components/data/URL.py       |  2 +-
 .../base/langflow/components/helpers/MessageHistory.py | 10 ++++++----
 .../base/langflow/components/helpers/RecordsToText.py  |  1 +
 .../components/SelectionMenuComponent/index.tsx        |  2 +-
 5 files changed, 10 insertions(+), 7 deletions(-)

diff --git a/src/backend/base/langflow/base/io/chat.py b/src/backend/base/langflow/base/io/chat.py
index 7e136a842..b722c16ab 100644
--- a/src/backend/base/langflow/base/io/chat.py
+++ b/src/backend/base/langflow/base/io/chat.py
@@ -23,7 +23,7 @@ class ChatComponent(CustomComponent):
                 "display_name": "Sender Type",
                 "advanced": True,
             },
-            "sender_name": {"display_name": "Sender Name"},
+            "sender_name": {"display_name": "Sender Name", "advanced": True},
             "session_id": {
                 "display_name": "Session ID",
                 "info": "If provided, the message will be stored in the memory.",
diff --git a/src/backend/base/langflow/components/data/URL.py b/src/backend/base/langflow/components/data/URL.py
index f9e515205..32ebc91ee 100644
--- a/src/backend/base/langflow/components/data/URL.py
+++ b/src/backend/base/langflow/components/data/URL.py
@@ -20,7 +20,7 @@ class URLComponent(CustomComponent):
         self,
         urls: list[str],
     ) -> list[Record]:
-        loader = WebBaseLoader(web_paths=urls)
+        loader = WebBaseLoader(web_paths=[url for url in urls if url])
         docs = loader.load()
         records = self.to_records(docs)
         self.status = records
diff --git a/src/backend/base/langflow/components/helpers/MessageHistory.py b/src/backend/base/langflow/components/helpers/MessageHistory.py
index 221d90c4e..90191e7d0 100644
--- a/src/backend/base/langflow/components/helpers/MessageHistory.py
+++ b/src/backend/base/langflow/components/helpers/MessageHistory.py
@@ -6,25 +6,27 @@ from langflow.schema import Record
 
 
 class MessageHistoryComponent(CustomComponent):
-    display_name = "Message History"
-    description = "Retrieves stored chat messages given a specific Session ID."
-    beta: bool = True
+    display_name = "Memory"
+    description = "Retrieves stored chat messages."
 
     def build_config(self):
         return {
             "sender": {
                 "options": ["Machine", "User", "Machine and User"],
                 "display_name": "Sender Type",
+                "advanced": True,
             },
             "sender_name": {"display_name": "Sender Name", "advanced": True},
             "n_messages": {
                 "display_name": "Number of Messages",
                 "info": "Number of messages to retrieve.",
+                "advanced": True,
             },
             "session_id": {
                 "display_name": "Session ID",
                 "info": "Session ID of the chat history.",
                 "input_types": ["Text"],
+                "advanced": True,
             },
             "order": {
                 "options": ["Ascending", "Descending"],
@@ -39,7 +41,7 @@ class MessageHistoryComponent(CustomComponent):
         sender: Optional[str] = "Machine and User",
         sender_name: Optional[str] = None,
         session_id: Optional[str] = None,
-        n_messages: int = 5,
+        n_messages: int = 100,
         order: Optional[str] = "Descending",
     ) -> List[Record]:
         order = "DESC" if order == "Descending" else "ASC"
diff --git a/src/backend/base/langflow/components/helpers/RecordsToText.py b/src/backend/base/langflow/components/helpers/RecordsToText.py
index d3e418792..049c99243 100644
--- a/src/backend/base/langflow/components/helpers/RecordsToText.py
+++ b/src/backend/base/langflow/components/helpers/RecordsToText.py
@@ -17,6 +17,7 @@ class RecordsToTextComponent(CustomComponent):
             "template": {
                 "display_name": "Template",
                 "info": "The template to use for formatting the records. It can contain the keys {text}, {data} or any other key in the Record.",
+                "multiline": True,
             },
         }
 
diff --git a/src/frontend/src/pages/FlowPage/components/SelectionMenuComponent/index.tsx b/src/frontend/src/pages/FlowPage/components/SelectionMenuComponent/index.tsx
index a6d83153c..7b8193caa 100644
--- a/src/frontend/src/pages/FlowPage/components/SelectionMenuComponent/index.tsx
+++ b/src/frontend/src/pages/FlowPage/components/SelectionMenuComponent/index.tsx
@@ -65,7 +65,7 @@ export default function SelectionMenu({
             className={`${
               disable
                 ? "flex h-full w-full cursor-not-allowed items-center justify-between text-sm text-muted-foreground"
-                : "flex h-full w-full items-center justify-between text-sm hover:text-indigo-500"
+                : "flex h-full w-full items-center justify-between text-sm"
             }`}
             onClick={onClick}
             disabled={disable}