feat(extensions): add register_tools() for raw Agno toolkits & providers

Lets custom/*.py attach any Agno tool that is not a single decorated function —
a Toolkit instance or a context provider's get_tools() list (Drive, Gmail,
Calendar, Slack, etc.) — to the agent without editing core code:

    from dango.extensions import register_tools
    from agno.context.gdrive import GoogleDriveContextProvider
    register_tools(*GoogleDriveContextProvider().get_tools())

register_tools(*objs) flattens lists/tuples and stores raw objects; they flow
into the agent via get_custom_tools() alongside decorated tools, and are exposed
to the agent only (never slash commands). Docs cover usage and note that Agno's
Scheduler is a runtime/cron layer that does not fit Dango's model. 44 tests pass.

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

Author: chiao-ke

custom/tools.py.example

@@ -49,3 +49,17 @@ def whoami(ctx: Ctx) -> str:
 #     import re
 #     m = re.search(r"<title>(.*?)</title>", html, re.I | re.S)
 #     return m.group(1).strip() if m else "(no title found)"
+
+
+# ── Attach a built-in Agno toolkit or context provider ───────────────────────
+# Use register_tools() for anything that isn't your own single function — an
+# Agno Toolkit instance, or a context provider's get_tools() list. These are
+# exposed to the agent only (never a slash command), with no core code changes.
+#
+# Some providers need extra packages / credentials, e.g. Drive:
+#   uv add google-api-python-client google-auth-httplib2 google-auth-oauthlib
+#
+# from dango.extensions import register_tools
+# from agno.context.gdrive import GoogleDriveContextProvider
+#
+# register_tools(*GoogleDriveContextProvider(corpora="user").get_tools())

dango/extensions/__init__.py

@@ -9,13 +9,15 @@
 
 from .context import Ctx
 from .loader import get_custom_tools, load_custom_modules, register_custom_commands
-from .registry import agent_tool, command, command_and_tool
+from .registry import agent_tool, command, command_and_tool, register_tools
 
 __all__ = [
     # Decorators (used in custom/*.py)
     "command",
     "agent_tool",
     "command_and_tool",
+    # Attach raw Agno toolkits / context providers (used in custom/*.py)
+    "register_tools",
     # Call context
     "Ctx",
     # Loader (used by the app at startup)

dango/extensions/loader.py

@@ -19,7 +19,7 @@
 from typing import Any, Callable, get_type_hints
 
 from .context import Ctx
-from .registry import ExtensionSpec, command_specs, tool_specs
+from .registry import ExtensionSpec, command_specs, raw_tools, tool_specs
 
 _loaded = False
 
@@ -112,20 +112,26 @@ def wrapper(*args, **kwargs):
 
 
 def get_custom_tools() -> list:
-    """Build Agno tool objects from the registered tool specs."""
+    """Build the agent's custom tool list.
+
+    Combines decorated @agent_tool / @command_and_tool functions with any raw
+    tools/toolkits/context-provider tools registered via register_tools().
+    """
+    tools = []
+
     specs = tool_specs()
-    if not specs:
-        return []
+    if specs:
+        from agno.tools import tool
 
-    from agno.tools import tool
+        for spec in specs:
+            wrapper = _make_tool_wrapper(spec)
+            try:
+                tools.append(tool(name=spec.name, description=spec.description or None)(wrapper))
+            except Exception as e:
+                print(f"⚠️  [custom] could not register tool '{spec.name}': {e}")
 
-    tools = []
-    for spec in specs:
-        wrapper = _make_tool_wrapper(spec)
-        try:
-            tools.append(tool(name=spec.name, description=spec.description or None)(wrapper))
-        except Exception as e:
-            print(f"⚠️  [custom] could not register tool '{spec.name}': {e}")
+    # Raw Agno toolkits / context-provider tools registered via register_tools().
+    tools.extend(raw_tools())
     return tools
 
 

dango/extensions/registry.py

@@ -31,6 +31,9 @@ class ExtensionSpec:
 # Populated at import time, when the loader imports each custom/*.py file.
 _REGISTRY: list[ExtensionSpec] = []
 
+# Raw Agno tools/toolkits/context-provider tools registered via register_tools().
+_RAW_TOOLS: list = []
+
 
 def _first_doc_line(fn: Callable) -> str:
     doc = (fn.__doc__ or "").strip()
@@ -87,6 +90,30 @@ def wrap(fn: Callable) -> Callable:
 """Register a function as BOTH a Discord slash command and an Agent tool."""
 
 
+def register_tools(*tools) -> None:
+    """Attach raw Agno tools/toolkits/context providers to the agent.
+
+    For anything that is not a single decorated function — an Agno ``Toolkit``
+    instance, or the list produced by a context provider's ``get_tools()`` — call
+    this from a ``custom/*.py`` file so it reaches the agent without editing core
+    code. Accepts plain functions, ``@tool`` functions, Toolkit instances, or
+    lists/tuples of any of these::
+
+        from dango.extensions import register_tools
+        from agno.context.gdrive import GoogleDriveContextProvider
+
+        register_tools(*GoogleDriveContextProvider().get_tools())
+
+    These are exposed to the agent only; they are never turned into Discord slash
+    commands.
+    """
+    for t in tools:
+        if isinstance(t, (list, tuple)):
+            _RAW_TOOLS.extend(t)
+        else:
+            _RAW_TOOLS.append(t)
+
+
 def command_specs() -> list[ExtensionSpec]:
     return [s for s in _REGISTRY if s.expose_command]
 
@@ -95,6 +122,11 @@ def tool_specs() -> list[ExtensionSpec]:
     return [s for s in _REGISTRY if s.expose_tool]
 
 
+def raw_tools() -> list:
+    return list(_RAW_TOOLS)
+
+
 def clear_registry() -> None:
     """Reset the registry. Used by tests and (later) hot-reload."""
     _REGISTRY.clear()
+    _RAW_TOOLS.clear()

docs/features/extensions.md

@@ -158,6 +158,37 @@ own Discord permissions before doing anything:
     permissions, return them for the model to reason about) but cannot perform
     member/role mutations directly.
 
+## Using built-in Agno toolkits & context providers
+
+Beyond writing your own functions, you can attach any
+[Agno toolkit](https://docs.agno.com/tools/toolkits) or
+[context provider](https://docs.agno.com/context-providers) to the agent from a
+custom file with `register_tools()` — no core code changes:
+
+```python
+from dango.extensions import register_tools
+from agno.context.gdrive import GoogleDriveContextProvider
+
+# A context provider exposes its tools via get_tools()
+register_tools(*GoogleDriveContextProvider(corpora="user").get_tools())
+```
+
+`register_tools()` accepts plain functions, `@tool` functions, `Toolkit`
+instances, or lists (such as a provider's `get_tools()`). Everything you pass is
+exposed to the **agent only** — it never becomes a Discord slash command. This is
+the same mechanism Dango uses internally for its SQL database tools.
+
+Agno ships context providers for Google Drive, Gmail, Calendar, Slack, web, wiki,
+the filesystem, and more. Some need extra packages and credentials — e.g. Drive
+needs `google-api-python-client google-auth-httplib2 google-auth-oauthlib` plus
+OAuth. Install those in your clone the usual way (`uv add ...`).
+
+!!! note "Agno's Scheduler does not apply"
+    Agno's [Scheduler](https://docs.agno.com/scheduler/overview) is a runtime/cron
+    orchestration layer that drives AgentOS endpoints — it is not a tool and does
+    not plug into Dango's discord.py + Workflow model. For scheduled behaviour,
+    use `discord.ext.tasks` or an external cron job.
+
 ## Interactive UI (modals, buttons, selects) — not supported here
 
 This SDK is built around a simple contract: **your function takes arguments and

docs/llms.txt

@@ -17,7 +17,7 @@
 
 - [Model Providers & Routing](https://raw.githubusercontent.com/zhiro-labs/dango/main/docs/features/models.md): `provider:model_id` format, supported providers, dual-model AUTO_ROUTE, error fallback, local models, context token budget.
 - [Tools](https://raw.githubusercontent.com/zhiro-labs/dango/main/docs/features/tools.md): DuckDuckGo search, website fetching, workspace file access, custom HTTP APIs, SQL databases.
-- [Custom Commands & Tools](https://raw.githubusercontent.com/zhiro-labs/dango/main/docs/features/extensions.md): Write your own Discord slash commands and agent tools in Python in a gitignored custom/ directory. Three decorators — @command, @agent_tool, @command_and_tool — with per-function opt-in for agent access. One function can be both a slash command and an LLM-callable tool; a normalized Ctx works across both call paths.
+- [Custom Commands & Tools](https://raw.githubusercontent.com/zhiro-labs/dango/main/docs/features/extensions.md): Write your own Discord slash commands and agent tools in Python in a gitignored custom/ directory. Three decorators — @command, @agent_tool, @command_and_tool — with per-function opt-in for agent access. One function can be both a slash command and an LLM-callable tool; a normalized Ctx works across both call paths. register_tools() attaches any Agno toolkit or context provider (Drive, Gmail, Calendar, Slack, etc.) to the agent without editing core code.
 - [Workflow Architecture](https://raw.githubusercontent.com/zhiro-labs/dango/main/docs/features/workflow.md): Four-step Agno Workflow — FetchHistory → LLMChat → ExtractRenderTables → SendResponse. Entry point: `on_message` in ChatCog (and the `/deep` command), which builds message_data and calls workflow.arun().
 
 ## Usage

docs/reference/api.md

@@ -90,6 +90,20 @@ When `name` is omitted the function name is used; when `description` is omitted
 the first line of the docstring is used. A leading `ctx` parameter is optional
 and is stripped from the public command/tool schema.
 
+### `register_tools(*tools)`
+
+Attach raw Agno tools to the agent — plain functions, `@tool` functions,
+`Toolkit` instances, or lists such as a context provider's `get_tools()`. Lists
+and tuples are flattened. Call it from a `custom/*.py` file; everything passed is
+exposed to the agent only (never a slash command).
+
+```python
+from dango.extensions import register_tools
+from agno.context.gdrive import GoogleDriveContextProvider
+
+register_tools(*GoogleDriveContextProvider().get_tools())
+```
+
 ### `Ctx`
 
 Dataclass describing where a function was invoked from, passed as the first

tests/test_extensions.py

@@ -5,7 +5,7 @@
 
 import pytest
 
-from dango.extensions import agent_tool, command, command_and_tool
+from dango.extensions import agent_tool, command, command_and_tool, register_tools
 from dango.extensions.context import (
     Ctx,
     reset_request_context,
@@ -182,6 +182,36 @@ def test_no_tools_returns_empty_list(self):
         assert get_custom_tools() == []
 
 
+# ── register_tools (raw toolkits / context providers) ──────────────────────────
+class TestRegisterTools:
+    def test_single_object_passes_through(self):
+        sentinel = object()
+        register_tools(sentinel)
+        assert sentinel in get_custom_tools()
+
+    def test_list_is_flattened(self):
+        # Mirrors register_tools(*provider.get_tools()) and register_tools(provider.get_tools()).
+        a, b = object(), object()
+        register_tools([a, b])
+        tools = get_custom_tools()
+        assert a in tools and b in tools
+
+    def test_raw_tools_are_not_slash_commands(self):
+        register_tools(object())
+        assert command_specs() == []
+
+    def test_combined_with_decorated_tools(self):
+        @agent_tool(name="decorated")
+        def decorated(x: str):
+            return x
+
+        sentinel = object()
+        register_tools(sentinel)
+        tools = get_custom_tools()
+        assert sentinel in tools
+        assert "decorated" in [getattr(t, "name", None) for t in tools]
+
+
 # ── Integration with discord.py ────────────────────────────────────────────────
 class TestDiscordIntegration:
     def test_make_app_command_strips_ctx_and_interaction(self):