6
6
from ..nodes import (
7
7
FetchNode ,
8
8
ParseNode ,
9
- RAGNode ,
10
9
GenerateScraperNode
11
10
)
12
11
from .abstract_graph import AbstractGraph
@@ -70,20 +69,12 @@ def _create_graph(self) -> BaseGraph:
70
69
input = "doc" ,
71
70
output = ["parsed_doc" ],
72
71
node_config = {"chunk_size" : self .model_token ,
73
- "verbose" : self .verbose
72
+ "verbose" : self .verbose ,
73
+ "parse_html" : False
74
74
}
75
75
)
76
- rag_node = RAGNode (
77
- input = "user_prompt & (parsed_doc | doc)" ,
78
- output = ["relevant_chunks" ],
79
- node_config = {
80
- "llm" : self .llm_model ,
81
- "embedder_model" : self .embedder_model ,
82
- "verbose" : self .verbose
83
- }
84
- )
85
76
generate_scraper_node = GenerateScraperNode (
86
- input = "user_prompt & (relevant_chunks | parsed_doc | doc)" ,
77
+ input = "user_prompt & (doc)" ,
87
78
output = ["answer" ],
88
79
node_config = {"llm" : self .llm_model ,
89
80
"verbose" : self .verbose },
@@ -95,13 +86,11 @@ def _create_graph(self) -> BaseGraph:
95
86
nodes = [
96
87
fetch_node ,
97
88
parse_node ,
98
- rag_node ,
99
89
generate_scraper_node ,
100
90
],
101
91
edges = [
102
92
(fetch_node , parse_node ),
103
- (parse_node , rag_node ),
104
- (rag_node , generate_scraper_node )
93
+ (parse_node , generate_scraper_node ),
105
94
],
106
95
entry_point = fetch_node
107
96
)
0 commit comments