Skip to content

Commit

Permalink
Fix MultiStepAgent docstring (#336)
Browse files Browse the repository at this point in the history
* Fix MultiStepAgent docstring

* Force PR doc build
  • Loading branch information
albertvillanova authored Jan 23, 2025
1 parent b333e08 commit 0217d3f
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 7 deletions.
3 changes: 0 additions & 3 deletions docs/source/en/reference/agents.md
Original file line number Diff line number Diff line change
Expand Up @@ -35,17 +35,14 @@ We provide two types of agents, based on the main [`Agent`] class.

Both require arguments `model` and list of tools `tools` at initialization.


### Classes of agents


[[autodoc]] MultiStepAgent

[[autodoc]] CodeAgent

[[autodoc]] ToolCallingAgent


### ManagedAgent

[[autodoc]] ManagedAgent
Expand Down
8 changes: 4 additions & 4 deletions src/smolagents/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,17 +164,17 @@ class MultiStepAgent:
While the objective is not reached, the agent will perform a cycle of action (given by the LLM) and observation (obtained from the environment).
Args:
tools (`list[[`Tool`]]`): List of tools that the agent can use.
model (Callable[[list[dict[str, str]]], [`ChatMessage`]]): Model that will generate the agent's actions.
tools (`list[Tool]`): [`Tool`]s that the agent can use.
model (`Callable[[list[dict[str, str]]], ChatMessage]`): Model that will generate the agent's actions.
system_prompt (`str`, *optional*): System prompt that will be used to generate the agent's actions.
tool_description_template (`str`, *optional*): Template used to describe the tools in the system prompt.
max_steps (`int`, default `6`): Maximum number of steps the agent can take to solve the task.
tool_parser (`Callable`, *optional*): Function used to parse the tool calls from the LLM output.
add_base_tools (`bool`, default `False`): Whether to add the base tools to the agent's tools.
verbosity_level (`int`, default `1`): Level of verbosity of the agent's logs.
grammar (`dict[str, str]`, *optional*): Grammar used to parse the LLM output.
managed_agents (`list`, *optional*): List of managed agents that the agent can call.
step_callbacks (`list[Callable]`, *optional*): List of callbacks that will be called at each step.
managed_agents (`list`, *optional*): Managed agents that the agent can call.
step_callbacks (`list[Callable]`, *optional*): Callbacks that will be called at each step.
planning_interval (`int`, *optional*): Interval at which the agent will run a planning step.
"""

Expand Down

0 comments on commit 0217d3f

Please sign in to comment.