468367d617
后端： 1.彻底删除原agent文件夹，并将现agent2文件夹全量重命名为agent(包括全部涉及到的文件以及文档、注释)，迁移工作完美结束 2.修复了重试消息的相关逻辑问题 前端： 1.改善了一些交互体验，修复了一些bug，现在只剩少的功能了，现存的bug基本都修复完毕 全仓库： 1.更新了决策记录和README文档
113 lines
2.6 KiB
Go
113 lines
2.6 KiB
Go
package agentllm
|
||
|
||
import (
|
||
"encoding/json"
|
||
"errors"
|
||
"fmt"
|
||
"strings"
|
||
)
|
||
|
||
// ParseJSONObject 解析模型返回中的 JSON 对象。
|
||
//
|
||
// 职责边界：
|
||
// 1. 负责处理“模型输出前后夹杂解释文字 / markdown 代码块”的常见情况；
|
||
// 2. 负责提取最外层 JSON object 并反序列化为目标结构；
|
||
// 3. 不负责业务字段合法性校验，例如 priority 是否在 1~4，应由上层 node 再校验。
|
||
func ParseJSONObject[T any](raw string) (*T, error) {
|
||
clean := strings.TrimSpace(raw)
|
||
if clean == "" {
|
||
return nil, errors.New("模型返回为空，无法解析 JSON")
|
||
}
|
||
|
||
objectText := ExtractJSONObject(clean)
|
||
if objectText == "" {
|
||
return nil, fmt.Errorf("模型返回中未找到 JSON 对象: %s", truncateForError(clean))
|
||
}
|
||
|
||
var out T
|
||
if err := json.Unmarshal([]byte(objectText), &out); err != nil {
|
||
return nil, fmt.Errorf("JSON 解析失败: %w", err)
|
||
}
|
||
return &out, nil
|
||
}
|
||
|
||
// ExtractJSONObject 从混合文本里提取第一个完整 JSON 对象。
|
||
//
|
||
// 设计说明：
|
||
// 1. LLM 很容易输出“这里是结果：{...}”这种半结构化文本；
|
||
// 2. 这里用括号计数而不是正则，避免嵌套对象一多就误截断；
|
||
// 3. 目前只提取 object，不提取 array，因为当前 agent 的路由/规划契约基本都是对象。
|
||
func ExtractJSONObject(text string) string {
|
||
clean := trimMarkdownCodeFence(strings.TrimSpace(text))
|
||
if clean == "" {
|
||
return ""
|
||
}
|
||
|
||
start := strings.Index(clean, "{")
|
||
if start < 0 {
|
||
return ""
|
||
}
|
||
|
||
depth := 0
|
||
inString := false
|
||
escaped := false
|
||
for idx := start; idx < len(clean); idx++ {
|
||
ch := clean[idx]
|
||
|
||
if escaped {
|
||
escaped = false
|
||
continue
|
||
}
|
||
if ch == '\\' && inString {
|
||
escaped = true
|
||
continue
|
||
}
|
||
if ch == '"' {
|
||
inString = !inString
|
||
continue
|
||
}
|
||
if inString {
|
||
continue
|
||
}
|
||
|
||
switch ch {
|
||
case '{':
|
||
depth++
|
||
case '}':
|
||
depth--
|
||
if depth == 0 {
|
||
return clean[start : idx+1]
|
||
}
|
||
}
|
||
}
|
||
return ""
|
||
}
|
||
|
||
func trimMarkdownCodeFence(text string) string {
|
||
trimmed := strings.TrimSpace(text)
|
||
if !strings.HasPrefix(trimmed, "```") {
|
||
return trimmed
|
||
}
|
||
|
||
lines := strings.Split(trimmed, "\n")
|
||
if len(lines) == 0 {
|
||
return trimmed
|
||
}
|
||
|
||
// 1. 去掉首行 ```json / ```；
|
||
// 2. 若末行是 ```，一并去掉；
|
||
// 3. 中间正文保持原样，避免破坏 JSON 的换行结构。
|
||
body := lines[1:]
|
||
if len(body) > 0 && strings.TrimSpace(body[len(body)-1]) == "```" {
|
||
body = body[:len(body)-1]
|
||
}
|
||
return strings.TrimSpace(strings.Join(body, "\n"))
|
||
}
|
||
|
||
func truncateForError(text string) string {
|
||
if len(text) <= 160 {
|
||
return text
|
||
}
|
||
return text[:160] + "..."
|
||
}
|