Logan
1acb0c49af
- Agent CLI: 多轮对话、SSH 长连接、流式推理、Evolver 按用户隔离、历史侧栏 - 终端审计: 回放播放器、命令书签、筛选分页导出、命令输出统计、录制落盘修复 - 修复 session_commands 表迁移、Kill 竞态导致录制丢失、无录制会话回放提示 - 审计历史页只读（移除删除），移除回放弹框录制搜索
111 lines
4.0 KiB
Go
111 lines
4.0 KiB
Go
// Package agentcli 是 Agent CLI 引擎的顶层入口。 // Logan.Li
|
||
// 组装 llm 客户端 + 工具注册中心（含 ssh_execute）+ 安全控制器 + Evolver + ReAct 核心，
|
||
// 对外暴露 Engine，由 controller 通过依赖注入（EventCallback / ConfirmFn / SSH 凭据）驱动。
|
||
//
|
||
// 设计原则（参考 opsxcli，适配为 aiops 后端库）：
|
||
// - 零业务依赖：不 import model/controller，所有持久化与 WS 由 controller 层负责。
|
||
// - 复用 aiops 既有能力：LLM 走 OpenAI 兼容（model.LLMConfig），SSH 走 pkg/remoteexec。
|
||
// - 安全：危险命令黑名单硬拒，高危命令经 ConfirmFn 由前端确认。
|
||
// - 自进化：Evolver 异步学习经验并注入下次 Prompt，失败优雅降级。
|
||
package agentcli
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
|
||
"ezops/pkg/agentcli/core"
|
||
"ezops/pkg/agentcli/evolver"
|
||
"ezops/pkg/agentcli/llm"
|
||
"ezops/pkg/agentcli/safety"
|
||
"ezops/pkg/agentcli/tools"
|
||
"ezops/pkg/remoteexec"
|
||
)
|
||
|
||
// EngineConfig 引擎配置（由 controller 从 model.LLMConfig + model.Server 构建后注入） // Logan.Li
|
||
type EngineConfig struct {
|
||
LLM llm.Config // LLM 客户端配置（endpoint 已解析）
|
||
SSH remoteexec.SSHConfig // 目标主机 SSH 配置（含凭据）
|
||
SafetyMode safety.Mode // 安全模式（默认 balanced）
|
||
Core *core.Config // ReAct 配置（nil 用默认）
|
||
MemoryDir string // Evolver 记忆目录（空则 data/agentcli）
|
||
}
|
||
|
||
// Engine Agent CLI 引擎实例（每个 WebSocket 连接 / 任务一个） // Logan.Li
|
||
type Engine struct {
|
||
agent *core.Agent
|
||
safety *safety.Controller
|
||
tools *tools.Registry
|
||
evolver *evolver.EvolverEngine
|
||
}
|
||
|
||
// NewEngine 组装引擎并注册默认工具集 // Logan.Li
|
||
func NewEngine(cfg EngineConfig) (*Engine, error) {
|
||
coreCfg := cfg.Core
|
||
if coreCfg == nil {
|
||
coreCfg = core.DefaultConfig()
|
||
}
|
||
if cfg.MemoryDir != "" {
|
||
coreCfg.MemoryDir = cfg.MemoryDir
|
||
}
|
||
mode := cfg.SafetyMode
|
||
if mode == "" {
|
||
mode = safety.ModeBalanced
|
||
}
|
||
|
||
// LLM 客户端
|
||
client := llm.NewClient(cfg.LLM)
|
||
|
||
// 工具注册中心：ssh_execute（SSH 长连接）+ analyze_output
|
||
registry := tools.NewRegistry()
|
||
sshTool, err := tools.NewSSHExecuteTool(cfg.SSH, coreCfg.ToolTimeout)
|
||
if err != nil {
|
||
return nil, fmt.Errorf("ssh_execute 初始化失败: %w", err)
|
||
}
|
||
registry.Register(sshTool)
|
||
registry.Register(tools.NewAnalyzeOutputTool())
|
||
|
||
// 安全控制器
|
||
sc := safety.NewController(mode)
|
||
|
||
// Evolver（LLM 可用时启用学习；失败降级）
|
||
ev, err := evolver.NewEvolverEngine(coreCfg.MemoryDir, client)
|
||
if err != nil {
|
||
ev = nil // 降级：无自学习，不影响主流程
|
||
}
|
||
|
||
agent := core.NewAgent(client, registry, coreCfg, sc, ev)
|
||
return &Engine{agent: agent, safety: sc, tools: registry, evolver: ev}, nil
|
||
}
|
||
|
||
// SetCallback 注入事件回调（驱动前端渲染 + DB 持久化） // Logan.Li
|
||
func (e *Engine) SetCallback(cb core.EventCallback) { e.agent.SetCallback(cb) }
|
||
|
||
// SetConfirmFn 注入高危确认回调（controller 推 WS confirm 事件并阻塞等前端回传） // Logan.Li
|
||
func (e *Engine) SetConfirmFn(fn safety.ConfirmFn) { e.safety.SetConfirmFn(fn) }
|
||
|
||
// SetAutoApprove 测试用：自动批准（生产禁用） // Logan.Li
|
||
func (e *Engine) SetAutoApprove(b bool) { e.safety.SetAutoApprove(b) }
|
||
|
||
// Run 执行任务（ReAct 循环）；prior 为上一轮对话历史（不含 system），priorTokens 为累计 token // Logan.Li
|
||
func (e *Engine) Run(ctx context.Context, query, serverInfo string, prior []llm.Message, priorTokens int) (*core.RunResult, error) {
|
||
return e.agent.Run(ctx, query, serverInfo, prior, priorTokens)
|
||
}
|
||
|
||
// EvolveStats 进化统计（供审计展示） // Logan.Li
|
||
func (e *Engine) EvolveStats() map[string]interface{} {
|
||
if e.evolver == nil {
|
||
return map[string]interface{}{"enabled": false}
|
||
}
|
||
return e.evolver.GetStats()
|
||
}
|
||
|
||
// Close 释放资源（含 SSH 长连接） // Logan.Li
|
||
func (e *Engine) Close() {
|
||
if e.tools != nil {
|
||
e.tools.Close()
|
||
}
|
||
if e.agent != nil {
|
||
e.agent.Close()
|
||
}
|
||
}
|