change the planning.
Browse files- basic_agent.py +3 -3
basic_agent.py
CHANGED
@@ -74,7 +74,7 @@ class BasicAgent:
|
|
74 |
|
75 |
if not self.manager_model:
|
76 |
self.manager_model = smolagents.HfApiModel(
|
77 |
-
max_tokens=
|
78 |
temperature=0.3,
|
79 |
model_id=self.manager_model_id,
|
80 |
custom_role_conversions=None,
|
@@ -148,9 +148,9 @@ class BasicAgent:
|
|
148 |
tools=manager_tools,
|
149 |
model=self.manager_model,
|
150 |
add_base_tools=True,
|
151 |
-
max_steps=
|
152 |
verbosity_level=2,
|
153 |
-
planning_interval=
|
154 |
additional_authorized_imports=[
|
155 |
"duckduckgo_search",
|
156 |
"wikipedia-api",
|
|
|
74 |
|
75 |
if not self.manager_model:
|
76 |
self.manager_model = smolagents.HfApiModel(
|
77 |
+
max_tokens=64000,
|
78 |
temperature=0.3,
|
79 |
model_id=self.manager_model_id,
|
80 |
custom_role_conversions=None,
|
|
|
148 |
tools=manager_tools,
|
149 |
model=self.manager_model,
|
150 |
add_base_tools=True,
|
151 |
+
max_steps=15,
|
152 |
verbosity_level=2,
|
153 |
+
planning_interval=3,
|
154 |
additional_authorized_imports=[
|
155 |
"duckduckgo_search",
|
156 |
"wikipedia-api",
|