init

2026-03-07 21:25:55 +08:00
commit 27bd85ef90
60 changed files with 5658 additions and 0 deletions
--- a/app/agents.go
+++ b/app/agents.go
@@ -0,0 +1,84 @@
+package app
+
+import (
+	"github.com/voocel/agentcore"
+	"github.com/voocel/ainovel-cli/state"
+	"github.com/voocel/ainovel-cli/tools"
+)
+
+// BuildCoordinator 组装 Coordinator Agent 及其 SubAgent。
+func BuildCoordinator(
+	cfg Config,
+	store *state.Store,
+	model agentcore.ChatModel,
+	refs tools.References,
+	prompts Prompts,
+	styles map[string]string,
+) *agentcore.Agent {
+	// 共享工具
+	contextTool := tools.NewContextTool(store, refs, cfg.Style)
+
+	// Architect SubAgent 工具
+	architectTools := []agentcore.Tool{
+		contextTool,
+		tools.NewSaveFoundationTool(store),
+	}
+
+	// Writer SubAgent 工具（V1: +polish_chapter +check_consistency）
+	writerTools := []agentcore.Tool{
+		contextTool,
+		tools.NewPlanChapterTool(store),
+		tools.NewWriteSceneTool(store),
+		tools.NewPolishChapterTool(store),
+		tools.NewCheckConsistencyTool(store),
+		tools.NewCommitChapterTool(store),
+	}
+
+	// Editor SubAgent 工具（V1）
+	editorTools := []agentcore.Tool{
+		contextTool,
+		tools.NewSaveReviewTool(store),
+	}
+
+	architect := agentcore.SubAgentConfig{
+		Name:         "architect",
+		Description:  "世界构建师：生成小说前提、大纲和角色档案",
+		Model:        model,
+		SystemPrompt: prompts.Architect,
+		Tools:        architectTools,
+		MaxTurns:     10,
+	}
+
+	// 动态拼接风格指令到 Writer prompt
+	writerPrompt := prompts.Writer
+	if style, ok := styles[cfg.Style]; ok {
+		writerPrompt += "\n\n" + style
+	}
+
+	writer := agentcore.SubAgentConfig{
+		Name:         "writer",
+		Description:  "场景写作者：逐场景完成一章的创作，包含打磨和一致性检查",
+		Model:        model,
+		SystemPrompt: writerPrompt,
+		Tools:        writerTools,
+		MaxTurns:     25,
+	}
+
+	editor := agentcore.SubAgentConfig{
+		Name:         "editor",
+		Description:  "全局审阅者：发现跨章结构问题，输出审阅结果",
+		Model:        model,
+		SystemPrompt: prompts.Editor,
+		Tools:        editorTools,
+		MaxTurns:     10,
+	}
+
+	subagentTool := agentcore.NewSubAgentTool(architect, writer, editor)
+
+	return agentcore.NewAgent(
+		agentcore.WithModel(model),
+		agentcore.WithSystemPrompt(prompts.Coordinator),
+		agentcore.WithTools(subagentTool, contextTool),
+		agentcore.WithMaxTurns(60),
+	)
+}