File tree Expand file tree Collapse file tree 3 files changed +9
-3
lines changed Expand file tree Collapse file tree 3 files changed +9
-3
lines changed Original file line number Diff line number Diff line change @@ -83,7 +83,8 @@ def _create_graph(self) -> BaseGraph:
83
83
output = ["urls" ],
84
84
node_config = {
85
85
"llm_model" : self .llm_model ,
86
- "max_results" : self .max_results
86
+ "max_results" : self .max_results ,
87
+ "search_engine" : self .copy_config .get ("search_engine" )
87
88
}
88
89
)
89
90
graph_iterator_node = GraphIteratorNode (
Original file line number Diff line number Diff line change @@ -76,7 +76,8 @@ def _create_graph(self) -> BaseGraph:
76
76
output = ["urls" ],
77
77
node_config = {
78
78
"llm_model" : self .llm_model ,
79
- "max_results" : self .max_results
79
+ "max_results" : self .max_results ,
80
+ "search_engine" : self .copy_config .get ("search_engine" )
80
81
}
81
82
)
82
83
graph_iterator_node = GraphIteratorNode (
Original file line number Diff line number Diff line change @@ -41,7 +41,11 @@ def __init__(
41
41
self .verbose = (
42
42
False if node_config is None else node_config .get ("verbose" , False )
43
43
)
44
- self .search_engine = node_config .get ("search_engine" , "google" )
44
+ self .search_engine = (
45
+ node_config ["search_engine" ]
46
+ if node_config .get ("search_engine" )
47
+ else "google"
48
+ )
45
49
self .max_results = node_config .get ("max_results" , 3 )
46
50
47
51
def execute (self , state : dict ) -> dict :
You can’t perform that action at this time.
0 commit comments