From 7bf5d3a1a43b04050b78c87c9af5ca775291c19f Mon Sep 17 00:00:00 2001
From: Jason Schleifer <jason.schleifer@gmail.com>
Date: Sun, 1 Sep 2024 05:56:33 +1200
Subject: [PATCH] Added TextChunks Module

---
 __init__.py                                  |  2 ++
 nodes/rag/gtUITextChunksResponseRagModule.py | 16 ++++++++++++++++
 2 files changed, 18 insertions(+)
 create mode 100644 nodes/rag/gtUITextChunksResponseRagModule.py

diff --git a/__init__.py b/__init__.py
index b99584d..900d634 100644
--- a/__init__.py
+++ b/__init__.py
@@ -192,6 +192,7 @@ from .nodes.rag.gtUIPromptResponseRagModule import gtUIPromptResponseRagModule
 # RAG
 from .nodes.rag.gtUIRagEngine import gtUIRagEngine
 from .nodes.rag.gtUITextChunksRerankRagModule import gtUITextChunksRerankRagModule
+from .nodes.rag.gtUITextChunksResponseRagModule import gtUITextChunksResponseRagModule
 from .nodes.rag.gtUITextLoaderRetrievalRagModule import gtUITextLoaderRetrievalRagModule
 from .nodes.rag.gtUITranslateQueryRagModule import gtUITranslateQueryRagModule
 from .nodes.rag.gtUIVectorStoreRetrievalRagModule import (
@@ -407,6 +408,7 @@ NODE_CLASS_MAPPINGS = {
     "Griptape RAG Rerank: Text Chunks Module": gtUITextChunksRerankRagModule,
     "Griptape RAG Response: Prompt Module": gtUIPromptResponseRagModule,
     "Griptape RAG Response: Footnote Module": gtUIFootnotePromptResponseRagModule,
+    "Griptape RAG Response: Text Chunks Module": gtUITextChunksResponseRagModule,
     # "Griptape Display: Artifact": gtUIOutputArtifactNode,
     # "Griptape Config: Environment Variables": gtUIEnv,
 }
diff --git a/nodes/rag/gtUITextChunksResponseRagModule.py b/nodes/rag/gtUITextChunksResponseRagModule.py
new file mode 100644
index 0000000..e50e980
--- /dev/null
+++ b/nodes/rag/gtUITextChunksResponseRagModule.py
@@ -0,0 +1,16 @@
+from griptape.engines.rag.modules import TextChunksResponseRagModule
+
+from .gtUIBaseReponseRagModule import gtUIBaseResponseRagModule
+
+
+class gtUITextChunksResponseRagModule(gtUIBaseResponseRagModule):
+    """
+    Griptape TextChunks Response Rag Module.
+    """
+
+    def __init__(self):
+        pass
+
+    def create(self, **kwargs):
+        module = TextChunksResponseRagModule()
+        return ([module],)