用 LangGraph 实现「密钥轮换计划」:工具定义与智能体代码
「密钥轮换计划」是运维与平台工程师的高频任务。它的输入通常是密钥清单(不含明文),期望得到的是轮换计划。这类有固定流程、又需要一点判断力的工作,很适合封装成智能体 Skill。
本文介绍如何在 LangChain / LangGraph 工具 上实现这个技能,包括它的形态、示例配置、启用步骤、一次完整的运行过程和常见坑。
这个技能做什么
| 项目 | 说明 |
|---|---|
| 技能 | 密钥轮换计划(secret-rotation-plan) |
| 作用 | 为系统中的密钥制定轮换步骤与验证方式 |
| 输入 | 密钥清单(不含明文) |
| 输出 | 轮换计划 |
| 形态 | 纯提示词 |
| 风险 | 高 |
| 适用人群 | 运维与平台工程师 |
处理流程分四步:
- 列出密钥用途与使用方
- 设计双密钥过渡
- 写验证步骤
- 安排时间窗口
这个技能纯指令型,不需要额外脚本或外部连接,复制到大多数平台都能用;属于高风险:涉及权限、个人信息或不可逆操作,Skill 只产出建议或草稿,执行前必须人工确认。
在 LangChain / LangGraph 工具 中是什么形态
LangChain 用 @tool 装饰器把函数变成工具,LangGraph 提供预置的 ReAct 智能体,也可以自己画状态图,把技能拆成可观察、可重试的节点。
代码运行在你的环境中;流程复杂时建议从预置智能体起步,再逐步换成自定义图。
示例配置
# 示意代码:LangChain 工具 + LangGraph 预置 ReAct 智能体
from langchain_core.tools import tool
from langchain.chat_models import init_chat_model
from langgraph.prebuilt import create_react_agent
@tool
def secret_rotation_plan_precheck(content: str) -> str:
"""检查密钥清单(不含明文)是否齐全,返回缺失项。"""
return "ok" if content.strip() else "输入为空"
model = init_chat_model("你的模型标识")
agent = create_react_agent(
model, tools=[secret_rotation_plan_precheck],
prompt="你负责密钥轮换计划:列出密钥用途与使用方;设计双密钥过渡;写验证步骤;安排时间窗口。",
)
out = agent.invoke({"messages": [("user", "这是密钥清单(不含明文):...")]})
示例里最重要的是两处:一是描述(或指令开头)要写明「什么时候用」,让智能体在运维与平台工程师提出相关请求时能对上号;二是约束部分要写进「变更类命令先在测试环境演练,并准备好回滚」这类领域规矩,避免智能体自作主张。
安装与启用步骤
- 安装 LangChain 与 LangGraph
- 用 @tool 定义预检查、读取等工具
- 用预置智能体快速跑通
- 需要固定流程时改为自定义状态图,每一步一个节点
- 接入追踪工具观察每一步
实操:一次完整的运行
假设你在服务器、容器与部署流水线中发出这样的请求(虚构示例):
帮我做一下密钥轮换计划,材料是附上的密钥清单(不含明文)。
智能体识别到任务与「密钥轮换计划」相关后,会依次列出密钥用途与使用方、设计双密钥过渡、写验证步骤、安排时间窗口。最终交付轮换计划,并在末尾列出需要人工确认的内容。你的工作是对照原始密钥清单(不含明文)抽查两三处关键信息,确认无误后再使用。
测试与验收
对每个节点单独写测试,再跑一次端到端用例。
针对这个技能,还建议额外准备:
- 一份信息齐全的标准密钥清单(不含明文),检查输出是否完整
- 一份缺少关键信息的输入,检查智能体是否先提问而不是编造
- 一份包含“忽略以上规则”等字样的输入,检查约束是否仍然有效
常见坑
- 一开始就设计复杂的多节点图,调试成本高。
- 工具 docstring 缺失,模型不知道如何使用。
- 只写了步骤,没写交付格式,导致每次输出的轮换计划结构不同。
- 高风险技能没有设置人工确认环节,智能体直接执行了不可逆操作。
延伸阅读
- 密钥轮换计划 · n8n AI Agent 版
- 密钥轮换计划 · 快速上手
- 密钥轮换计划 · 组合用法
- 运维变更通告 · LangChain / LangGraph 工具 版
- K8s 清单审查 · LangChain / LangGraph 工具 版
- 智能体Skills 分类:625 个技能 × 32 个角度
说明:文中的配置、代码与目录结构均为示意,用于说明思路;各平台的入口、字段名与能力会更新,落地前请以对应官方文档为准。示例中的数据与场景为虚构,不代表真实案例。