Passing extra attributes to tool definitions

[anil-gurbuz]

Hi,

I have a tool definition as below ( I wanted to add extra attributes to tool definition by writing the init method) .
I tried below but apparently the created Tool object is not passed the execution environment but rather the class definition is passed.

class ShopifyProductsTool(Tool):
    name = "shopify_products"
    description = (
        "Fetches all products (title and ID) from user's Shopify store."
        "This tool retrieves every product by paginating through the Shopify GraphQL API. "
        "Returns a list of dictionaries with 'id' and 'title' for each product. "
        "Handles pagination automatically until all products are fetched."
    )
    inputs = {}  # No runtime inputs; configuration is required at initialization
    output_type = "array"
 
    def __init__(self, config={}, *args, **kwargs):

        super().__init__(*args, **kwargs)
        self.config = config or {}
        self.graphql_url = f"https://{self.config.get('shop_domain')}/admin/api/2025-07/graphql.json"
        self.headers = { 'X-Shopify-Access-Token': self.config.get("access_token",""), 'Content-Type': 'application/json' }

    def validate_arguments(self):
        super().validate_arguments()
        if not self.config.get("access_token",None):
            raise AttributeError("access_token is required in the config.")
        if not self.config.get("shop_domain",None):
            raise AttributeError("shop_domain is required in the config.")

    def forward(self):
        import requests


        query = """
        query ($first: Int!, $after: String) {
          products(first: $first, after: $after) {
            pageInfo {
              hasNextPage
              endCursor
            }
            edges {
              node {
                id
                title
              }
            }
          }
        }
        """
        products = []
        has_next_page = True
        after = None
        first = 100  # Shopify allows up to 250, but 100 is safe for most stores
        while has_next_page:
            variables = {"first": first, "after": after}
            response = requests.post(
                self.graphql_url,
                headers=self.headers,
                json={"query": query, "variables": variables},
                verify=False
            )
            if response.status_code != 200:
                raise Exception(f"Failed to fetch products: {response.status_code} {response.text}")
            data = response.json()
            product_data = data.get("data", {}).get("products", {})
            edges = product_data.get("edges", [])
            products.extend([
                {"id": edge.get("node", {}).get("id"), "title": edge.get("node", {}).get("title")}
                for edge in edges
            ])
            page_info = product_data.get("pageInfo", {})
            has_next_page = page_info.get("hasNextPage", False)
            after = page_info.get("endCursor")
        return products 

My question is;

1. Is it possible to pass extra attributes to tool definition?
2. If not how should I pass authentication tokens etc. to the tool call -- I guess env variables would be one way but is that the right way?

Would appreciate any help with that!
Thanks

[albertvillanova]

I am not sure of the issue you found.

Yes, you can create additional attributes in the __init__ method. You can check that, for example in the ApiWebSearchTool builtin tool in the default_tools module, e.g.

smolagents/src/smolagents/default_tools.py

Lines 275 to 294 in 7329ef7

	def __init__(
	self,
	endpoint: str = "",
	api_key: str = "",
	api_key_name: str = "",
	headers: dict = None,
	params: dict = None,
	rate_limit: float | None = 1.0,
	):
	import os
	super().__init__()
	self.endpoint = endpoint or "https://api.search.brave.com/res/v1/web/search"
	self.api_key_name = api_key_name or "BRAVE_API_KEY"
	self.api_key = api_key or os.getenv(self.api_key_name)
	self.headers = headers or {"X-Subscription-Token": self.api_key}
	self.params = params or {"count": 10}
	self.rate_limit = rate_limit
	self._min_interval = 1.0 / rate_limit if rate_limit else 0.0
	self._last_request_time = 0.0