Merge pull request #15 from edge2992/feature/llm-vs-human

feat: LLM vs Human online play with Claude bot

Author: edge2992

.gitignore

@@ -1,2 +1,2 @@
 yatzcli
-yatz
+/yatz

Makefile

@@ -0,0 +1,16 @@
+.PHONY: build test vet clean
+
+BINARY := yatz
+BUILD_DIR := ./cmd/yatz
+
+build:
+	go build -o $(BINARY) $(BUILD_DIR)
+
+test:
+	go test ./...
+
+vet:
+	go vet ./...
+
+clean:
+	rm -f $(BINARY)

bot/bot.go

@@ -0,0 +1,69 @@
+package bot
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+)
+
+type Bot struct {
+	addr     string
+	name     string
+	strategy string
+	model    string
+}
+
+func New(addr, name, strategy, model string) *Bot {
+	return &Bot{
+		addr:     addr,
+		name:     name,
+		strategy: strategy,
+		model:    model,
+	}
+}
+
+func (b *Bot) Run() error {
+	claudePath, err := exec.LookPath("claude")
+	if err != nil {
+		return fmt.Errorf("claude CLI is required. Install it first: %w", err)
+	}
+
+	selfPath, err := os.Executable()
+	if err != nil {
+		return fmt.Errorf("resolve executable path: %w", err)
+	}
+
+	configFile, err := os.CreateTemp("", "yatz-mcp-*.json")
+	if err != nil {
+		return fmt.Errorf("create temp config: %w", err)
+	}
+	defer os.Remove(configFile.Name())
+
+	if _, err := configFile.WriteString(BuildMCPConfig(selfPath)); err != nil {
+		configFile.Close()
+		return fmt.Errorf("write MCP config: %w", err)
+	}
+	configFile.Close()
+
+	prompt := BuildPrompt(b.addr, b.name, b.strategy)
+	systemPrompt := BuildSystemPrompt(b.strategy)
+
+	args := []string{"-p",
+		"--mcp-config", configFile.Name(),
+		"--allowedTools", "mcp__yatzcli__*",
+		"--system-prompt", systemPrompt,
+	}
+	if b.model != "" {
+		args = append(args, "--model", b.model)
+	}
+	args = append(args, prompt)
+
+	cmd := exec.Command(claudePath, args...)
+	cmd.Stdout = os.Stdout
+	cmd.Stderr = os.Stderr
+
+	if err := cmd.Run(); err != nil {
+		return fmt.Errorf("claude exited with error: %w", err)
+	}
+	return nil
+}

bot/prompt.go

@@ -0,0 +1,46 @@
+package bot
+
+import (
+	"encoding/json"
+	"fmt"
+)
+
+func BuildMCPConfig(yatzBinaryPath string) string {
+	config := map[string]interface{}{
+		"mcpServers": map[string]interface{}{
+			"yatzcli": map[string]interface{}{
+				"command": yatzBinaryPath,
+				"args":    []string{"mcp"},
+			},
+		},
+	}
+	b, _ := json.MarshalIndent(config, "", "  ")
+	return string(b)
+}
+
+func BuildSystemPrompt(strategy string) string {
+	return fmt.Sprintf(`ヤッツィー対戦プレイヤー。考えすぎず素早くプレイすること。
+
+戦略: %s`, strategy)
+}
+
+func BuildPrompt(addr, name, strategy string) string {
+	return fmt.Sprintf(`ヤッツィー対戦に参加してプレイせよ。説明や分析は不要。ツールを呼ぶだけでよい。
+
+手順:
+1. join_game で %s に接続（名前: %s）
+2. 自分のターンでなければ wait_for_turn
+3. 自分のターン: まず roll_dice → ダイスを見て hold_dice か score を選ぶ
+4. score の返り値が次ターンの状態。score 後に wait_for_turn を呼ぶな（デッドロックする）
+5. Phase が "Finished" なら終了。そうでなければ 3 に戻る
+
+ルール:
+- 毎ターン最初に必ず roll_dice を呼ぶこと
+- hold_dice でキープするダイスのインデックス(0-4)を指定、残りを振り直す
+- 最大3回ロール（roll_dice 1回 + hold_dice 最大2回）
+- score でカテゴリを選んで得点確定
+
+戦略: %s
+
+score 後に send_chat で一言コメント（日本語、10文字以内）。`, addr, name, strategy)
+}

bot/prompt_test.go

@@ -0,0 +1,45 @@
+package bot
+
+import (
+	"encoding/json"
+	"testing"
+
+	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
+)
+
+func TestBuildMCPConfig(t *testing.T) {
+	config := BuildMCPConfig("/usr/local/bin/yatz")
+
+	var parsed map[string]interface{}
+	err := json.Unmarshal([]byte(config), &parsed)
+	require.NoError(t, err)
+
+	servers := parsed["mcpServers"].(map[string]interface{})
+	yatzcli := servers["yatzcli"].(map[string]interface{})
+	assert.Equal(t, "/usr/local/bin/yatz", yatzcli["command"])
+
+	args := yatzcli["args"].([]interface{})
+	assert.Equal(t, []interface{}{"mcp"}, args)
+}
+
+func TestBuildPrompt(t *testing.T) {
+	prompt := BuildPrompt("localhost:9876", "TestBot", "test strategy")
+
+	assert.Contains(t, prompt, "localhost:9876")
+	assert.Contains(t, prompt, "TestBot")
+	assert.Contains(t, prompt, "test strategy")
+	assert.Contains(t, prompt, "join_game")
+	assert.Contains(t, prompt, "wait_for_turn")
+	assert.Contains(t, prompt, "roll_dice")
+	assert.Contains(t, prompt, "score")
+	assert.Contains(t, prompt, "send_chat")
+	// Critical: instruction not to call wait_for_turn after score
+	assert.Contains(t, prompt, "score 後に wait_for_turn を呼ぶな")
+}
+
+func TestBuildSystemPrompt(t *testing.T) {
+	prompt := BuildSystemPrompt("my strategy")
+	assert.Contains(t, prompt, "ヤッツィー")
+	assert.Contains(t, prompt, "my strategy")
+}

bot/strategy.go

@@ -0,0 +1,8 @@
+package bot
+
+const DefaultStrategy = `- 上段ボーナス（63点以上）を最優先で狙う
+- 同じ数字が3つ以上あればキープしてヤッツィーやフルハウスを狙う
+- ストレート（1-2-3-4, 2-3-4-5, 3-4-5-6）が見えたらキープ
+- 3回目のロールでは最も得点が高いカテゴリにスコアする
+- 捨てカテゴリ（0点で埋める）は ones を優先
+- 終盤は残りカテゴリの期待値を考慮する`