89 lines
2.8 KiB
Go
89 lines
2.8 KiB
Go
package ai
|
||
|
||
import (
|
||
"context"
|
||
"fmt"
|
||
"strings"
|
||
"time"
|
||
|
||
"nl-game-api-gin/internal/gamecore/xiangqi"
|
||
)
|
||
|
||
// llmChessAI 大模型象棋 AI：服务端先用搜索引擎给候选着法排序，
|
||
// 再按难度截取候选交给 LLM 挑选（困难=只给高分着法，简单=全量随机）
|
||
type llmChessAI struct {
|
||
client *LLMClient // LLM 客户端
|
||
difficulty string // 难度
|
||
fallback ChessAI // 规则 AI 兜底
|
||
name string // 显示名
|
||
}
|
||
|
||
// Name AI 显示名
|
||
func (a *llmChessAI) Name() string { return a.name }
|
||
|
||
// boardText 把棋盘渲染成文本（LLM 提示词用）
|
||
func boardText(b *xiangqi.Board) string {
|
||
var sb strings.Builder
|
||
sb.WriteString(" 0 1 2 3 4 5 6 7 8\n")
|
||
for r := 0; r < 10; r++ {
|
||
sb.WriteString(fmt.Sprintf("%d ", r))
|
||
for c := 0; c < 9; c++ {
|
||
p := b[r][c]
|
||
if p == 0 {
|
||
sb.WriteString("． ")
|
||
} else {
|
||
sb.WriteString(xiangqi.PieceName(p) + " ")
|
||
}
|
||
}
|
||
sb.WriteString("\n")
|
||
}
|
||
return sb.String()
|
||
}
|
||
|
||
// DecideMove 走子：候选着法按引擎评分排序后交 LLM 选择
|
||
func (a *llmChessAI) DecideMove(b *xiangqi.Board, side int) (xiangqi.Move, string) {
|
||
// 用 2 层搜索为候选排序（困难截取前6，中等前12，简单全部）
|
||
ranked := b.RankedMoves(side, 2)
|
||
if len(ranked) == 0 {
|
||
return xiangqi.Move{}, ""
|
||
}
|
||
limit := len(ranked)
|
||
switch a.difficulty {
|
||
case DiffHard:
|
||
limit = 6
|
||
case DiffMedium:
|
||
limit = 12
|
||
}
|
||
if limit > len(ranked) {
|
||
limit = len(ranked)
|
||
}
|
||
shown := ranked[:limit]
|
||
// 组装候选清单（附带吃子信息帮助 LLM 判断）
|
||
options := make([]string, 0, len(shown))
|
||
for i, m := range shown {
|
||
target := b[m.ToR][m.ToC]
|
||
capture := ""
|
||
if target != 0 {
|
||
capture = "（吃" + xiangqi.PieceName(target) + "）"
|
||
}
|
||
options = append(options, fmt.Sprintf("%d: %s从(%d,%d)走到(%d,%d)%s",
|
||
i, b.MoveDesc(m), m.FromR, m.FromC, m.ToR, m.ToC, capture))
|
||
}
|
||
sideName := "红方"
|
||
if side == xiangqi.Black {
|
||
sideName = "黑方"
|
||
}
|
||
system := styleFor(a.difficulty) + "你在下中国象棋，执" + sideName + "。" +
|
||
"只输出 JSON：{\"choice\":候选编号,\"say\":\"一句台词\"}，不要输出其他内容。"
|
||
user := fmt.Sprintf("当前棋盘（行0在上为黑方，行9在下为红方）：\n%s\n候选着法（已按棋力排序，越靠前越好）：\n%s\n请选择一个候选编号。",
|
||
boardText(b), strings.Join(options, "\n"))
|
||
ctx, cancel := context.WithTimeout(context.Background(), 12*time.Second)
|
||
defer cancel()
|
||
// 带返回校验与重试的决策调用（传输层自动重试 + 输出非法自动纠错重问）
|
||
if d, err := a.client.ChatDecision(ctx, system, user, temperatureFor(a.difficulty), len(shown)-1); err == nil {
|
||
return shown[d.Choice], d.Say
|
||
}
|
||
// LLM 多次重试仍失败：走规则兜底
|
||
return a.fallback.DecideMove(b, side)
|
||
}
|