From a76b35652614798261ffe75351694c5f502b41b6 Mon Sep 17 00:00:00 2001
From: Gabriel Luiz Freitas Almeida <gabriel@logspace.ai>
Date: Wed, 7 Feb 2024 18:57:19 -0300
Subject: [PATCH] Refactor CharacterTextSplitterComponent to use typing and
 update return value

---
 .../components/textsplitters/CharacterTextSplitter.py    | 9 ++++++---
 1 file changed, 6 insertions(+), 3 deletions(-)

diff --git a/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py b/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py
index d65c28dab..1bfd2d2e3 100644
--- a/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py
+++ b/src/backend/langflow/components/textsplitters/CharacterTextSplitter.py
@@ -1,7 +1,8 @@
-from langflow import CustomComponent
+from typing import List
+
 from langchain.text_splitter import CharacterTextSplitter
 from langchain_core.documents.base import Document
-from typing import List
+from langflow import CustomComponent
 
 
 class CharacterTextSplitterComponent(CustomComponent):
@@ -23,8 +24,10 @@ class CharacterTextSplitterComponent(CustomComponent):
         chunk_size: int = 1000,
         separator: str = "\n",
     ) -> List[Document]:
-        return CharacterTextSplitter(
+        docs = CharacterTextSplitter(
             chunk_overlap=chunk_overlap,
             chunk_size=chunk_size,
             separator=separator,
         ).split_documents(documents)
+        self.status = docs
+        return docs