feat: updated ChatOutput component that can accept Data, Dataframe and Message (#6643)

* update chatoutput

* [autofix.ci] apply automated fixes

* update

* tests

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

* Update chat.py

* update template

* fix lint errors

* [autofix.ci] apply automated fixes

* [autofix.ci] apply automated fixes (attempt 2/3)

* fix: rename variable for clarity in chat output component test

* [autofix.ci] apply automated fixes

* fix: enable loading from database for API key in starter project configurations

* update templates

* [autofix.ci] apply automated fixes

---------

Co-authored-by: autofix-ci[bot] <114827586+autofix-ci[bot]@users.noreply.github.com>
Co-authored-by: Gabriel Luiz Freitas Almeida <gabriel@langflow.org>
Co-authored-by: cristhianzl <cristhian.lousa@gmail.com>

src/backend/tests/unit/components/outputs/test_chat_output_component.py

@@ -0,0 +1,98 @@
+import pytest
+from langflow.components.outputs import ChatOutput
+from langflow.schema.data import Data
+from langflow.schema.dataframe import DataFrame
+from langflow.schema.message import Message
+from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI
+
+from tests.base import ComponentTestBaseWithClient
+
+
+@pytest.mark.usefixtures("client")
+class TestChatOutput(ComponentTestBaseWithClient):
+    @pytest.fixture
+    def component_class(self):
+        return ChatOutput
+
+    @pytest.fixture
+    def default_kwargs(self):
+        return {
+            "input_value": "Hello, how are you?",
+            "should_store_message": True,
+            "sender": MESSAGE_SENDER_AI,
+            "sender_name": MESSAGE_SENDER_NAME_AI,
+            "session_id": "test_session_123",
+            "data_template": "{text}",
+            "background_color": "#f0f0f0",
+            "chat_icon": "🤖",
+            "text_color": "#000000",
+            "clean_data": True,
+        }
+
+    @pytest.fixture
+    def file_names_mapping(self):
+        return [
+            {"version": "1.0.19", "module": "outputs", "file_name": "ChatOutput"},
+            {"version": "1.1.0", "module": "outputs", "file_name": "chat"},
+            {"version": "1.1.1", "module": "outputs", "file_name": "chat"},
+        ]
+
+    async def test_process_string_input(self, component_class, default_kwargs):
+        """Test processing a simple string input."""
+        component = component_class(**default_kwargs)
+        input_text = "Hello, this is a test message"
+        component.input_value = input_text
+        result = await component.message_response()
+        assert result.text == input_text
+        assert result.sender == MESSAGE_SENDER_AI
+        assert result.sender_name == MESSAGE_SENDER_NAME_AI
+
+    async def test_process_data_input(self, component_class, default_kwargs):
+        """Test processing a Data object input."""
+        component = component_class(**default_kwargs)
+        data = Data(text="Test data message")
+        component.input_value = data
+        result = await component.message_response()
+        assert result.text == "Test data message"
+        assert result.sender == MESSAGE_SENDER_AI
+
+    async def test_process_dataframe_input(self, component_class, default_kwargs):
+        """Test processing a DataFrame input."""
+        component = component_class(**default_kwargs)
+        sample_df = DataFrame(data={"col1": ["A", "B"], "col2": [1, 2]})
+        component.input_value = sample_df
+        result = await component.message_response()
+        assert "col1" in result.text
+        assert "col2" in result.text
+        assert "A" in result.text
+        assert "B" in result.text
+
+    async def test_process_message_input(self, component_class, default_kwargs):
+        """Test processing a Message object input."""
+        component = component_class(**default_kwargs)
+        message = Message(text="Test message content")
+        component.input_value = message
+        result = await component.message_response()
+        assert result.text == "Test message content"
+        assert result.sender == MESSAGE_SENDER_AI
+
+    async def test_process_list_input(self, component_class, default_kwargs):
+        """Test processing a list of inputs."""
+        component = component_class(**default_kwargs)
+        input_list = ["First message", Data(text="Second message"), Message(text="Third message")]
+        component.input_value = input_list
+        result = await component.message_response()
+        assert "First message" in result.text
+        assert "Second message" in result.text
+        assert "Third message" in result.text
+
+    async def test_invalid_input(self, component_class, default_kwargs):
+        """Test handling of invalid input."""
+        component = component_class(**default_kwargs)
+        component.input_value = None
+        with pytest.raises(ValueError, match="Input data cannot be None"):
+            await component.message_response()
+
+        component.input_value = 123  # Invalid type
+        with pytest.raises(TypeError, match="Expected Data or DataFrame or Message or str"):
+            await component.message_response()

src/backend/tests/unit/components/outputs/test_output_components.py

@@ -1,37 +1,7 @@
 import pytest
-from langflow.components.outputs import ChatOutput, TextOutputComponent
-from langflow.utils.constants import MESSAGE_SENDER_AI, MESSAGE_SENDER_NAME_AI
+from langflow.components.outputs import TextOutputComponent
 
-from tests.base import ComponentTestBaseWithClient, ComponentTestBaseWithoutClient
-
-
-@pytest.mark.usefixtures("client")
-class TestChatOutput(ComponentTestBaseWithClient):
-    @pytest.fixture
-    def component_class(self):
-        return ChatOutput
-
-    @pytest.fixture
-    def default_kwargs(self):
-        return {
-            "input_value": "Hello, how are you?",
-            "should_store_message": True,
-            "sender": MESSAGE_SENDER_AI,
-            "sender_name": MESSAGE_SENDER_NAME_AI,
-            "session_id": "test_session_123",
-            "data_template": "{text}",
-            "background_color": "#f0f0f0",
-            "chat_icon": "🤖",
-            "text_color": "#000000",
-        }
-
-    @pytest.fixture
-    def file_names_mapping(self):
-        return [
-            {"version": "1.0.19", "module": "outputs", "file_name": "ChatOutput"},
-            {"version": "1.1.0", "module": "outputs", "file_name": "chat"},
-            {"version": "1.1.1", "module": "outputs", "file_name": "chat"},
-        ]
+from tests.base import ComponentTestBaseWithoutClient
 
 
 class TestTextOutputComponent(ComponentTestBaseWithoutClient):