File tree Expand file tree Collapse file tree 1 file changed +2
-11
lines changed Expand file tree Collapse file tree 1 file changed +2
-11
lines changed Original file line number Diff line number Diff line change @@ -62,14 +62,7 @@ def _create_graph(self) -> BaseGraph:
6262 "loader_kwargs" : self .config .get ("loader_kwargs" , {}),
6363 }
6464 )
65- parse_node = ParseNode (
66- input = "doc" ,
67- output = ["parsed_doc" ],
68- node_config = {
69- "chunk_size" : self .model_token ,
70- "llm_model" : self .llm_model
71- }
72- )
65+
7366 search_link_node = SearchLinkNode (
7467 input = "doc" ,
7568 output = ["parsed_doc" ],
@@ -84,12 +77,10 @@ def _create_graph(self) -> BaseGraph:
8477 return BaseGraph (
8578 nodes = [
8679 fetch_node ,
87- parse_node ,
8880 search_link_node
8981 ],
9082 edges = [
91- (fetch_node , parse_node ),
92- (parse_node , search_link_node )
83+ (fetch_node , search_link_node )
9384 ],
9485 entry_point = fetch_node ,
9586 graph_name = self .__class__ .__name__
You can’t perform that action at this time.
0 commit comments