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 1111from ..nodes import (
1212 FetchNode ,
1313 ParseNode ,
14- RAGNode ,
1514 GenerateAnswerNode ,
1615 TextToSpeechNode ,
1716)
@@ -72,13 +71,6 @@ def _create_graph(self) -> BaseGraph:
7271 "chunk_size" : self .model_token
7372 }
7473 )
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- )
8274 generate_answer_node = GenerateAnswerNode (
8375 input = "user_prompt & (relevant_chunks | parsed_doc | doc)" ,
8476 output = ["answer" ],
@@ -100,14 +92,12 @@ def _create_graph(self) -> BaseGraph:
10092 nodes = [
10193 fetch_node ,
10294 parse_node ,
103- rag_node ,
10495 generate_answer_node ,
10596 text_to_speech_node
10697 ],
10798 edges = [
10899 (fetch_node , parse_node ),
109- (parse_node , rag_node ),
110- (rag_node , generate_answer_node ),
100+ (parse_node , generate_answer_node ),
111101 (generate_answer_node , text_to_speech_node )
112102 ],
113103 entry_point = fetch_node ,
You can’t perform that action at this time.
0 commit comments