File tree Expand file tree Collapse file tree 1 file changed +1
-11
lines changed Expand file tree Collapse file tree 1 file changed +1
-11
lines changed Original file line number Diff line number Diff line change 11
11
from ..nodes import (
12
12
FetchNode ,
13
13
ParseNode ,
14
- RAGNode ,
15
14
GenerateAnswerNode ,
16
15
TextToSpeechNode ,
17
16
)
@@ -72,13 +71,6 @@ def _create_graph(self) -> BaseGraph:
72
71
"chunk_size" : self .model_token
73
72
}
74
73
)
75
- rag_node = RAGNode (
76
- input = "user_prompt & (parsed_doc | doc)" ,
77
- output = ["relevant_chunks" ],
78
- node_config = {
79
- "llm_model" : self .llm_model ,
80
- "embedder_model" : self .embedder_model }
81
- )
82
74
generate_answer_node = GenerateAnswerNode (
83
75
input = "user_prompt & (relevant_chunks | parsed_doc | doc)" ,
84
76
output = ["answer" ],
@@ -100,14 +92,12 @@ def _create_graph(self) -> BaseGraph:
100
92
nodes = [
101
93
fetch_node ,
102
94
parse_node ,
103
- rag_node ,
104
95
generate_answer_node ,
105
96
text_to_speech_node
106
97
],
107
98
edges = [
108
99
(fetch_node , parse_node ),
109
- (parse_node , rag_node ),
110
- (rag_node , generate_answer_node ),
100
+ (parse_node , generate_answer_node ),
111
101
(generate_answer_node , text_to_speech_node )
112
102
],
113
103
entry_point = fetch_node ,
You can’t perform that action at this time.
0 commit comments