Import AITURK IDE 1.0.0-beta.1 from Hermes 63279301; preserve MIT license
This commit is contained in:
@@ -0,0 +1,118 @@
|
||||
#!/usr/bin/env python3
|
||||
"""Propose an MCP server to the user as an inline card in the desktop chat.
|
||||
|
||||
The card (install / enable / authorize + decline) lives in the desktop
|
||||
renderer, so this tool round-trips through the gateway's blocking-prompt
|
||||
bridge — the same one ``clarify`` uses: tui_gateway emits
|
||||
``mcp.setup.request``, the renderer walks the user through the flow via the
|
||||
existing REST endpoints (catalog install, enable, OAuth), and answers with
|
||||
``mcp.setup.respond`` once the flow settles. This module is just schema + a
|
||||
thin dispatcher over the platform-injected callback.
|
||||
|
||||
Lives in the ``desktop_ui`` toolset, which the GUI gateway enables only for
|
||||
desktop-sourced sessions — on every other surface the agent falls back to
|
||||
``hermes mcp install <name>`` in the terminal.
|
||||
"""
|
||||
|
||||
import json
|
||||
from typing import Callable, Optional
|
||||
|
||||
from tools.registry import registry, tool_error
|
||||
|
||||
_ACTIONS = ("install", "enable", "authorize")
|
||||
|
||||
|
||||
def setup_mcp_tool(
|
||||
server: str = "",
|
||||
action: str = "install",
|
||||
reason: str = "",
|
||||
callback: Optional[Callable] = None,
|
||||
) -> str:
|
||||
"""Ask the desktop GUI to run an MCP setup flow; return its JSON outcome."""
|
||||
if callback is None:
|
||||
return tool_error(
|
||||
"setup_mcp is only available in the Hermes desktop app. Use the "
|
||||
"terminal instead: `hermes mcp install <name>` for catalog entries, "
|
||||
"`hermes mcp login <name>` for OAuth."
|
||||
)
|
||||
|
||||
name = (server or "").strip()
|
||||
if not name:
|
||||
return tool_error("server is required — the catalog or config name of the MCP server.")
|
||||
|
||||
action = (action or "install").strip().lower()
|
||||
if action not in _ACTIONS:
|
||||
return tool_error(f"action must be one of {', '.join(_ACTIONS)}.")
|
||||
|
||||
try:
|
||||
raw = callback(name, action, (reason or "").strip())
|
||||
except Exception as exc:
|
||||
return tool_error(f"MCP setup flow failed: {exc}")
|
||||
|
||||
if not raw:
|
||||
# The renderer never answered (timeout / closed window). Distinct from
|
||||
# an explicit decline, which arrives as {"status": "declined"}.
|
||||
return json.dumps(
|
||||
{
|
||||
"status": "unanswered",
|
||||
"server": name,
|
||||
"note": (
|
||||
"The user did not respond to the setup card. Do not retry "
|
||||
"immediately; continue without the server or ask in chat."
|
||||
),
|
||||
},
|
||||
ensure_ascii=False,
|
||||
)
|
||||
|
||||
# Desktop answers with a JSON object; pass it through, else wrap the raw text.
|
||||
try:
|
||||
return json.dumps(json.loads(raw), ensure_ascii=False)
|
||||
except (TypeError, ValueError):
|
||||
return json.dumps({"status": "error", "detail": str(raw)}, ensure_ascii=False)
|
||||
|
||||
|
||||
SETUP_MCP_SCHEMA = {
|
||||
"name": "setup_mcp",
|
||||
"description": (
|
||||
"Propose an MCP server as an inline consent card (install a catalog "
|
||||
"entry, re-enable a disabled server, or run OAuth); blocks until the "
|
||||
"user acts. Use when they ask to add an MCP or a task clearly needs "
|
||||
"a missing one. Never hand-edit mcp_servers config for them — always "
|
||||
"use this tool. Never re-ask after a decline — on declined/"
|
||||
"unanswered, continue without it. Catalog names: `hermes mcp "
|
||||
"catalog` in the terminal."
|
||||
),
|
||||
"parameters": {
|
||||
"type": "object",
|
||||
"properties": {
|
||||
"server": {
|
||||
"type": "string",
|
||||
"description": "Catalog name (install) or mcp_servers config name (enable/authorize).",
|
||||
},
|
||||
"action": {
|
||||
"type": "string",
|
||||
"enum": ["install", "enable", "authorize"],
|
||||
"description": "Defaults to install.",
|
||||
},
|
||||
"reason": {
|
||||
"type": "string",
|
||||
"description": "One sentence on the card: why this helps right now.",
|
||||
},
|
||||
},
|
||||
"required": ["server"],
|
||||
},
|
||||
}
|
||||
|
||||
|
||||
registry.register(
|
||||
name="setup_mcp",
|
||||
toolset="desktop_ui",
|
||||
schema=SETUP_MCP_SCHEMA,
|
||||
handler=lambda args, **kw: setup_mcp_tool(
|
||||
server=args.get("server", ""),
|
||||
action=args.get("action", "install"),
|
||||
reason=args.get("reason", ""),
|
||||
callback=kw.get("callback"),
|
||||
),
|
||||
emoji="🔌",
|
||||
)
|
||||
Reference in New Issue
Block a user