87 lines
2.3 KiB
Go
87 lines
2.3 KiB
Go
// agent_cli_history.go - Agent CLI 历史会话恢复为 LLM 对话上下文 // Logan.Li
|
||
package controller
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
|
||
"ezops/model"
|
||
"ezops/pkg/agentcli/llm"
|
||
)
|
||
|
||
// buildConvHistoryFromMessages 将 DB 消息重建为追问用的 prior 历史（不含 system） // Logan.Li
|
||
func buildConvHistoryFromMessages(msgs []model.AIRunMessage) []llm.Message {
|
||
var out []llm.Message
|
||
var pendingThought string
|
||
var pendingToolCalls []llm.ToolCall
|
||
var pendingToolCallIDs []string
|
||
|
||
flushAssistant := func() {
|
||
if pendingThought == "" && len(pendingToolCalls) == 0 {
|
||
return
|
||
}
|
||
out = append(out, llm.Message{
|
||
Role: "assistant",
|
||
Content: pendingThought,
|
||
ToolCalls: append([]llm.ToolCall(nil), pendingToolCalls...),
|
||
})
|
||
for _, tc := range pendingToolCalls {
|
||
pendingToolCallIDs = append(pendingToolCallIDs, tc.ID)
|
||
}
|
||
pendingThought = ""
|
||
pendingToolCalls = nil
|
||
}
|
||
|
||
for _, m := range msgs {
|
||
switch m.Type {
|
||
case "user":
|
||
flushAssistant()
|
||
out = append(out, llm.Message{Role: "user", Content: m.Content})
|
||
case "thought":
|
||
pendingThought = m.Content
|
||
case "tool_call":
|
||
args := validateToolArgsJSON(m.Content)
|
||
tcID := fmt.Sprintf("hist-%d", m.ID)
|
||
pendingToolCalls = append(pendingToolCalls, llm.ToolCall{
|
||
ID: tcID, Type: "function",
|
||
Function: llm.FunctionCall{Name: m.ToolName, Arguments: args},
|
||
})
|
||
case "tool_result":
|
||
if pendingThought != "" || len(pendingToolCalls) > 0 {
|
||
flushAssistant()
|
||
}
|
||
toolID := ""
|
||
if len(pendingToolCallIDs) > 0 {
|
||
toolID = pendingToolCallIDs[0]
|
||
pendingToolCallIDs = pendingToolCallIDs[1:]
|
||
}
|
||
content := m.Content
|
||
out = append(out, llm.Message{
|
||
Role: "tool", Name: m.ToolName, Content: content, ToolCallID: toolID,
|
||
})
|
||
case "done":
|
||
flushAssistant()
|
||
out = append(out, llm.Message{Role: "assistant", Content: m.Content})
|
||
case "error":
|
||
flushAssistant()
|
||
out = append(out, llm.Message{Role: "assistant", Content: "错误: " + m.Content})
|
||
default:
|
||
// 忽略 progress/session 等 // Logan.Li
|
||
}
|
||
}
|
||
flushAssistant()
|
||
return out
|
||
}
|
||
|
||
// validateToolArgsJSON 校验工具参数 JSON，非法则回退空对象 // Logan.Li
|
||
func validateToolArgsJSON(raw string) string {
|
||
if raw == "" {
|
||
return "{}"
|
||
}
|
||
var tmp map[string]interface{}
|
||
if json.Unmarshal([]byte(raw), &tmp) != nil {
|
||
return "{}"
|
||
}
|
||
return raw
|
||
}
|