TCares commited on
Commit
4f560fd
·
verified ·
1 Parent(s): eecce0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -8,7 +8,10 @@ from tools.final_answer import FinalAnswerTool
8
  from Gradio_UI import GradioUI
9
  from tools.visit_webpage import VisitWebpageTool
10
 
11
- ddg_search_tool = DuckDuckGoSearchTool()
 
 
 
12
 
13
  @tool
14
  def find_job_websites(job:str, location:str)-> str:
@@ -69,7 +72,7 @@ agent = CodeAgent(
69
  visit_webpage,
70
  final_answer
71
  ],
72
- max_steps=6,
73
  verbosity_level=1,
74
  grammar=None,
75
  planning_interval=None,
 
8
  from Gradio_UI import GradioUI
9
  from tools.visit_webpage import VisitWebpageTool
10
 
11
+ MAX_STEPS = 15
12
+ MAX_SEARCH_RESULTS = 10 # should be smaller than MAX_STEPS, as we potentially visit each result, and each visit costs one step
13
+
14
+ ddg_search_tool = DuckDuckGoSearchTool(max_results=MAX_SEARCH_RESULTS)
15
 
16
  @tool
17
  def find_job_websites(job:str, location:str)-> str:
 
72
  visit_webpage,
73
  final_answer
74
  ],
75
+ max_steps=MAX_STEPS,
76
  verbosity_level=1,
77
  grammar=None,
78
  planning_interval=None,