[Tiny Agents] Add tools to config #3242
Conversation
|
The docs for this PR live here. All of your documentation changes will be reflected on that endpoint. The docs are available until 30 days after the last update. |
There was a problem hiding this comment.
to be moved to https://huggingface.co/datasets/tiny-agents/tiny-agents
| class StdioServerConfig(TypedDict): | ||
| type: Literal["stdio"] | ||
| command: str | ||
| args: List[str] | ||
| env: Dict[str, str] | ||
| cwd: str | ||
| tools: NotRequired[ToolsConfig] |
There was a problem hiding this comment.
Can you rename it allowed_tools and accept only a list of strings as input? I can see pretty clearly the goal of allowing only a certain subset of tools but not necessarily excluding some. In the future if we really need it we can always add a new forbidden_tools property.
Note: this is taken from OpenAI's specs
| def _filter_tools( | ||
| self, tools: List[Any], tools_config: Optional[Dict[str, Any]], all_tool_names: List[str] | ||
| ) -> List[Any]: |
There was a problem hiding this comment.
Haven't reviewed this logic myself but I feel that if we only have a allowed_tools list then a 1-line list comprehension should be enough
This PR enables tools inclusion/exclusion for a given agent. Users can specify the tools to include/exclude like so:
{ "model": "meta-llama/Meta-Llama-3-8B-Instruct", "provider": "auto", "servers": [ { "type": "stdio", "command": "npx", "args": ["@playwright/mcp@latest"], + "tools": { + "include": ["browser_click", "browser_close"] } } ] }Note: this feature was implemented using Claude Code.