From 0e9aa23a1c7c210064b7cc196521440992d0f076 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Tue, 27 Feb 2024 16:57:51 -0300
Subject: [PATCH] Refactor ChatOutput class to inherit from ChatComponent

---
 src/backend/langflow/components/io/ChatOutput.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/backend/langflow/components/io/ChatOutput.py b/src/backend/langflow/components/io/ChatOutput.py
index e4c5d03ee..842a083fb 100644
--- a/src/backend/langflow/components/io/ChatOutput.py
+++ b/src/backend/langflow/components/io/ChatOutput.py
@@ -1,11 +1,11 @@
 from typing import Optional, Union
 
-from langflow import CustomComponent
 from langflow.field_typing import Text
+from langflow.io.schema import ChatComponent
 from langflow.schema import Record
 
 
-class ChatOutput(CustomComponent):
+class ChatOutput(ChatComponent):
     display_name = "Chat Output"
     description = "Used to send a message to the chat."