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],)