Custom tools

​

Tool contract

def main(inputs, env_variables):
    return {
        "output": { ... },
        "captured_variables": { ... }
    }

​

Example

def main(inputs, env_variables):
    try:
        product_id = inputs.get("product_id")
        api_key = env_variables.get("INVENTORY_API_KEY")
        return {
            "output": {"stock": 42, "product_name": "Example"},
            "captured_variables": {"current_stock": 42}
        }
    except Exception as e:
        return {"output": {"error": str(e)}, "captured_variables": {}}

​

Test tools

1. Open the tool in Dev Studio → Test tab.
2. Enter realistic sample input JSON.
3. Choose Dev, UAT, or Prod for env_variables.
4. Review stdout, errors, and returned output / captured_variables.
5. Fix and re-test before Publish.

​

Publish versions

1. Draft — edit handler and parameters.
2. Test — validate in Dev (and UAT per your process).
3. Publish — immutable version with release notes.
4. Build — pin when you build the Agent Graph.

​

Link to Graph Studio

1. Open Graph Studio → select a node → Tools tab.
2. Add tool → pick a published tool.
3. Map input variables from the session where shown.
4. Save the graph before Build.

​

Related

• Methods
• Tools & Dev Studio overview
• Env. variables