Added TextChunks Module

This commit is contained in:
Jason Schleifer
2024-09-01 05:56:33 +12:00
parent 19ebd9d2bf
commit 7bf5d3a1a4
2 changed files with 18 additions and 0 deletions
+2
View File
@@ -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,
}
@@ -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],)