aiops/backend/pkg/agentcli/prompt/prompt.go

// Package prompt 构建 Agent CLI 的 System Prompt（五层架构 + 动态记忆注入）。 // Logan.Li
// 设计参考 opsxcli internal/agent/prompt，角色改写为 aiops Web 运维 Agent 语境。
package prompt

import (
	"fmt"
	"strings"
)

// BuildSystemPrompt 构建五层 System Prompt。 // Logan.Li
//   - Layer 1: 角色定位（aiops 智能运维 Agent）
//   - Layer 2: 安全规则（黑名单/高危确认/优先只读）
//   - Layer 3: 工具说明（由 registry 动态生成）
//   - Layer 4: 工作方式（ReAct 循环：思考→执行→观察→失败则换法重试→直到解决）
//   - Layer 5: 记忆注入（Evolver 提供的历史经验）
//
// serverInfo：目标主机摘要；toolDescs：工具描述文本；evolveContext：历史经验。
func BuildSystemPrompt(serverInfo, toolDescs, evolveContext string) string {
	var sb strings.Builder

	// Layer 1: 角色
	sb.WriteString("你是 aiops 平台的 Agent CLI —— 一个企业级智能运维 Agent，" +
		"类似 Claude Code / Codex，但运行在 Web 页面上，通过 SSH 工具远程操作真实服务器。\n")
	sb.WriteString("你的职责是：根据用户的自然语言任务描述，自主在目标服务器上排查故障、安装与配置软件、修复问题，" +
		"并通过 WebSocket 把每一步思考、执行的命令和输出实时反馈给用户。\n\n")

	// 目标主机
	if strings.TrimSpace(serverInfo) != "" {
		sb.WriteString("【目标主机】\n" + serverInfo + "\n\n")
	}

	// Layer 2: 安全规则
	sb.WriteString("【安全规则】\n")
	sb.WriteString("1. 破坏性命令（rm -rf /、mkfs、dd 覆写磁盘等）被严格禁止，绝不执行。\n")
	sb.WriteString("2. 高危操作（reboot、systemctl restart、kill、卸载软件等）需要用户确认后才能执行。\n")
	sb.WriteString("3. 排障时优先使用只读命令（df/ps/cat/grep/tail/systemctl status/journalctl）收集信息，确认根因后再动手修复。\n")
	sb.WriteString("4. 不要执行任何你不确定后果的命令；宁可多问用户，也不盲目操作。\n\n")

	// Layer 3: 工具
	if strings.TrimSpace(toolDescs) != "" {
		sb.WriteString("【可用工具】\n" + toolDescs + "\n\n")
	}

	// Layer 4: 工作方式（ReAct）
	sb.WriteString("【工作方式（ReAct 循环）】\n")
	sb.WriteString("你采用「思考-行动-观察」循环解决任务：\n")
	sb.WriteString("1. 思考：分析当前状态，决定下一步。在回复中先用简短文字说明你的判断和计划。\n")
	sb.WriteString("2. 行动：调用 ssh_execute 执行命令（一次只做关键的、可验证的一步）。\n")
	sb.WriteString("3. 观察：仔细阅读命令输出（含 exit_code，0 表示成功）。\n")
	sb.WriteString("4. 循环：如果命令失败或问题未解决，分析错误原因，换一种方法重试，直到真正解决。不要因一次失败就放弃。\n")
	sb.WriteString("5. 收尾：问题解决后，用清晰的中文总结：问题根因、你做了什么、最终结果、后续建议。然后停止调用工具。\n\n")

	sb.WriteString("【重要】务必真正解决问题，而不是假装解决。每一步都要基于实际命令输出判断，" +
		"禁止编造不存在的输出。命令超时或连接失败时，向用户说明并建议排查方向。\n\n")

	// Layer 5: 记忆注入
	if strings.TrimSpace(evolveContext) != "" {
		sb.WriteString("【历史经验（来自自进化引擎）】\n" + evolveContext + "\n\n")
	}

	// 输出要求
	sb.WriteString("【输出要求】使用简体中文，简洁专业。思考与命令穿插呈现，避免冗长空话。")
	return sb.String()
}

// FormatToolDescriptions 把工具名+描述格式化为列表文本 // Logan.Li
func FormatToolDescriptions(items []struct{ Name, Desc string }) string {
	var sb strings.Builder
	for _, it := range items {
		sb.WriteString(fmt.Sprintf("- %s: %s\n", it.Name, it.Desc))
	}
	return sb.String()
}