可被 import 的
领域无关 Agent 框架

import (
    "context"
    superagent "github.com/tokengravity/superagent"
    "github.com/tokengravity/superagent/llm"
)

ag := superagent.New(superagent.Config{
    LLM:          llm.NewClient(llm.Config{APIUrl: "https://your-gateway/v1", APIKey: "sk-...", Model: "your-model"}),
    SystemPrompt: "你是 {{product}} 的运维助手，需要数据时调用工具。",
    PromptVars:   map[string]string{"product": "支付系统"},
    Memory:       superagent.NewInMemoryMemory(3),     // 会话记忆
    Hooks: superagent.Hooks{                       // 护栏：危险操作走人工审批
        BeforeTool: func(ctx context.Context, key, args string) (string, bool) {
            if key == "delete_db" { return "危险操作，需人工审批", true }
            return "", false
        },
    },
})

ag.Tools().Register(superagent.ToolSpec{Key: "restart_pod", Description: "重启异常 pod",
    Fn: func(ctx context.Context, args map[string]any) (superagent.SkillResult, error) {
        return superagent.SkillResult{Content: "已重启 3 个 pod"}, nil
    }})

res, _ := ag.Run(context.Background(), "重启支付服务的异常 pod", nil)
fmt.Println(res.FinalContent)

from superagent import Agent, Config, ToolSpec, SkillResult, Hooks, InMemory
from superagent.openai_client import OpenAIClient

ag = Agent(Config(
    llm=OpenAIClient(api_url="https://your-gateway/v1", api_key="sk-...", model="your-model"),
    system_prompt="你是 {{dept}} 的财务助手，依据制度处理报销与查询。",
    prompt_vars={"dept": "研发中心"},
    memory=InMemory(3),                              # 会话记忆
    # 大额付款不直接执行，挂人工审批门
    hooks=Hooks(before_tool=lambda ctx, k, a: ("大额付款需财务经理审批", True) if k == "make_payment" else ("", False)),
))

ag.tools().register(ToolSpec(key="submit_reimbursement", description="发起一笔报销",
    fn=lambda ctx, args: SkillResult(content="报销单 R2026-0012 已创建，待审批")))

res = ag.run(None, "帮我报销上周差旅 2300 元")
print(res.final_content)

# 异步 + 流式（FastAPI 友好）
async for ev in ag.arun_stream(None, "..."):
    if ev.kind == "delta": print(ev.delta, end="")