yjgr-agent-dns

基于 DNS 的 AI Agent 发现与调用 MCP 服务器（DNS-AID 协议），修好了远程 MCP 的鉴权。

这是 dns-aid/dns-aid-core（Apache-2.0）的二次开发版本。 上游的 MCP 插件在调用需要鉴权的远程 Agent 时，Bearer 只有一种精确写法能通， OAuth 对 MCP 场景基本不可用，而且所有 401 都被吞成一句 Invocation failed (status: error)。 我们向上游反馈过，没有回应，所以自己修、自己发。

在 Claude Code / Codex 里使用

发布到 PyPI 后，两边都只需要一段配置，不用预装任何东西（依赖 uv）。

Claude Code（写入项目根目录 .mcp.json，或 ~/.claude.json 的 mcpServers）：

{
  "mcpServers": {
    "yjgr-agent-dns": {
      "command": "uvx",
      "args": ["--from", "yjgr-agent-dns", "yjgr-agent-dns-mcp"]
    }
  }
}

或者一条命令：

claude mcp add yjgr-agent-dns -- uvx --from yjgr-agent-dns yjgr-agent-dns-mcp

Codex（~/.codex/config.toml）：

[mcp_servers.yjgr-agent-dns]
command = "uvx"
args = ["--from", "yjgr-agent-dns", "yjgr-agent-dns-mcp"]

只做发现和调用时不需要任何环境变量。要发布自己的 Agent 到 DNS 时， 按上游文档配置 DNS_AID_BACKEND 等后端凭据（见 docs/UPSTREAM-README.md）。

鉴权怎么用

远程 Agent 有三种常见鉴权方式，现在三种都能用：

判断远程服务用哪种：直接调一次，401 的报错里会带服务端的 WWW-Authenticate 挑战和下一步提示，例如：

HTTP 401 Unauthorized from https://agent.example.com/mcp;
WWW-Authenticate: Bearer resource_metadata="https://agent.example.com/.well-known/oauth-protected-resource";
The server uses MCP OAuth: run the agent_login tool for this endpoint (browser login), then retry.

MCP 规范 OAuth 就是 Claude Code / Codex 连远程 MCP 用的那套： RFC 9728 资源元数据 → RFC 8414 授权服务器元数据 → RFC 7591 动态注册 → 授权码 + PKCE → 回调到本机 → 换 token。 实现直接用官方 mcp SDK 的 OAuthClientProvider，本项目只补了 token 落盘、本机回调服务和浏览器拉起。

新增的 MCP 工具：

• agent_login(endpoint, scopes?, client_id?, client_secret?, timeout_seconds?) 浏览器登录
• agent_logout(endpoint) 忘掉登录
• agent_auth_status(endpoint) 查看登录状态（不返回 token 值）

list_agent_tools 也补上了 auth_type / auth_config / credentials 参数，上游版本连这个入口都没有。

A2A 多轮对话

message/send 返回的是一个 Task，不只是一段文字。智能体可能先回 state="input-required" 要你确认，这时返回里会带 task_id、context_id、state 和 data_parts。 续话要把 task_id 和 context_id 原样传回去，否则每次都是新任务，确认流程永远走不完：

r1 = send_a2a_message(endpoint=EP, message="给我推荐一些商品", credentials={"token": TOKEN})
# r1: state="input-required", data_parts=[{"action_type":"confirmation_required", ...}]
r2 = send_a2a_message(endpoint=EP, message="", credentials={"token": TOKEN},
                      task_id=r1["task_id"], context_id=r1["context_id"],
                      data={"action_type": "confirmed"})

确认的 data 具体是什么形状由对方智能体定义，看它 data_parts 里的 task_intent。

命令行同样可用：

yjgr-agent-dns agent login  https://agent.example.com/mcp
yjgr-agent-dns agent status https://agent.example.com/mcp
yjgr-agent-dns list-tools   https://agent.example.com/mcp            # 自动用已登录的 token
yjgr-agent-dns call         https://agent.example.com/mcp echo -a '{"text":"hi"}' --token XXX

token 存在 ~/.yjgr-agent-dns/oauth/<origin>.json，权限 0600，按服务的 origin 隔离。

相关环境变量

内网部署注意

上游的 SSRF 防护把所有 http:// 和私网地址一律拒绝，连白名单都绕不过， 所以 OAuth2 客户端凭据模式的 token 地址不能指向内网 IdP。 现在：白名单的主机不限 scheme；或者整体开 DNS_AID_ALLOW_PRIVATE_NETWORKS=1。 file:// 和 URL 内嵌账号密码依旧拒绝。

相对上游改了什么

全部改动都在 git log 里逐条可查，这里列结论：

1. 凭据被静默丢弃：auth_type 为空时上游直接不鉴权。现在按凭据推断类型，或自动用已存的登录。
2. list_agent_tools 没有鉴权参数：补上。
3. 401 / 403 被吞：从 anyio 的 ExceptionGroup 里解出 HTTP 状态和 WWW-Authenticate，报错说明是没带凭据还是凭据被拒。
4. 没有 MCP 规范 OAuth：上游只有 client_credentials。新增 sdk/auth/mcp_oauth.py 与三个工具。
5. 内网 IdP 被 SSRF 防护拦死：白名单优先于 scheme 检查，新增 DNS_AID_ALLOW_PRIVATE_NETWORKS。
6. A2A 的 401/403 也被吞：只报 HTTP 401: {body}，没有 WWW-Authenticate、不说该做什么。三个协议处理器现在共用一个报错构造器，会区分「没带凭据」「带了被拒」「缺身份头」。
7. 发不了自定义请求头：请求头写死，需要 X-Platform 这类标识的服务根本调不了。
8. A2A 只能单轮：每次调用都新建 task，data part 被丢弃，确认类流程无法完成。
9. 发现阶段不报鉴权要求：agent card 里明明声明了 bearer，auth_type 却一直是 null；而且 card 里的 http / httpAuth 不是处理器名字，直接用会报 Unknown auth_type。
10. 依赖声明打架：上游 main 允许 mcp<3，装上 mcp 2.x 后客户端直接报 transport unavailable。现钉死 <2.0.0，且 mcp、typer、rich 进核心依赖，uvx --from yjgr-agent-dns yjgr-agent-dns-mcp 不需要任何 extra（--from 是 uvx 的规则：命令名和包名不同就要写）。

包名、模块名、命令名全部改为 yjgr-agent-dns / yjgr_agent_dns。 线上协议标识（X-DNS-AID-* 头、dns-aid-jwks.json、DNS_AID_* 环境变量）刻意没改， 保证和上游工具发布的 DNS 记录互通。

开发

uv venv --python 3.12 && uv pip install -e '.[all,dev]'
scripts/test_per_file.sh                 # 单元测试（每个文件独立进程，上游测试之间有状态泄漏）
.venv/bin/python -m pytest tests/integration/test_mcp_oauth_e2e.py   # OAuth 全流程端到端
.venv/bin/ruff check src/ && .venv/bin/mypy src/yjgr_agent_dns
uv build                                 # dist/ 下出 wheel

tests/testbed/oauth_mcp_target.py 是一个符合 MCP 规范的本地授权服务器 + 受保护 MCP 服务， 可以单独起来手工验证：python tests/testbed/oauth_mcp_target.py 8766，然后 yjgr-agent-dns agent login http://127.0.0.1:8766/mcp。

同步上游：git fetch upstream && git merge upstream/main（remote 名为 upstream）。

许可

Apache-2.0，与上游一致。保留上游 LICENSE 与 NOTICE，修改部分版权归 YJGR。