Add multi-agent config and registry runtime flow

pkg/agent/subagent_config_intent.go

@@ -0,0 +1,209 @@
+package agent
+
+import (
+	"context"
+	"fmt"
+	"strings"
+
+	"clawgo/pkg/bus"
+	"clawgo/pkg/tools"
+)
+
+func (al *AgentLoop) maybeHandleSubagentConfigIntent(ctx context.Context, msg bus.InboundMessage) (string, bool, error) {
+	_ = ctx
+	if al == nil {
+		return "", false, nil
+	}
+	if msg.Channel == "system" || msg.Channel == "internal" {
+		return "", false, nil
+	}
+	content := strings.TrimSpace(msg.Content)
+	if content == "" {
+		return "", false, nil
+	}
+	if isSubagentConfigConfirm(content) {
+		return al.confirmPendingSubagentDraft(msg.SessionKey)
+	}
+	if isSubagentConfigCancel(content) {
+		return al.cancelPendingSubagentDraft(msg.SessionKey)
+	}
+	if !looksLikeSubagentCreateRequest(content) {
+		return "", false, nil
+	}
+	description := extractSubagentDescription(content)
+	if description == "" {
+		return "", false, nil
+	}
+	draft := tools.DraftConfigSubagent(description, "")
+	al.storePendingSubagentDraft(msg.SessionKey, draft)
+	return formatSubagentDraftForUser(draft), true, nil
+}
+
+func looksLikeSubagentCreateRequest(content string) bool {
+	lower := strings.ToLower(strings.TrimSpace(content))
+	if lower == "" {
+		return false
+	}
+	createMarkers := []string{
+		"创建", "新建", "增加", "添加", "配置一个", "生成一个",
+		"create", "add", "new",
+	}
+	subagentMarkers := []string{
+		"subagent", "sub-agent", "agent", "子代理", "子 agent", "工作代理",
+	}
+	hasCreate := false
+	for _, item := range createMarkers {
+		if strings.Contains(lower, item) {
+			hasCreate = true
+			break
+		}
+	}
+	if !hasCreate {
+		return false
+	}
+	for _, item := range subagentMarkers {
+		if strings.Contains(lower, item) {
+			return true
+		}
+	}
+	return false
+}
+
+func isSubagentConfigConfirm(content string) bool {
+	lower := strings.ToLower(strings.TrimSpace(content))
+	phrases := []string{
+		"确认创建", "确认保存", "确认生成", "保存这个子代理", "创建这个子代理",
+		"confirm create", "confirm save", "save it", "create it",
+	}
+	for _, phrase := range phrases {
+		if lower == phrase || strings.Contains(lower, phrase) {
+			return true
+		}
+	}
+	return false
+}
+
+func isSubagentConfigCancel(content string) bool {
+	lower := strings.ToLower(strings.TrimSpace(content))
+	phrases := []string{
+		"取消创建", "取消保存", "取消这个子代理", "放弃创建",
+		"cancel create", "cancel save", "discard draft", "never mind",
+	}
+	for _, phrase := range phrases {
+		if lower == phrase || strings.Contains(lower, phrase) {
+			return true
+		}
+	}
+	return false
+}
+
+func extractSubagentDescription(content string) string {
+	content = strings.TrimSpace(content)
+	replacers := []string{
+		"请", "帮我", "给我", "创建", "新建", "增加", "添加", "配置", "生成",
+		"a ", "an ", "new ", "create ", "add ",
+	}
+	out := content
+	for _, item := range replacers {
+		out = strings.ReplaceAll(out, item, "")
+	}
+	out = strings.ReplaceAll(out, "子代理", "")
+	out = strings.ReplaceAll(out, "subagent", "")
+	out = strings.ReplaceAll(out, "sub-agent", "")
+	out = strings.TrimSpace(out)
+	if out == "" {
+		return strings.TrimSpace(content)
+	}
+	return out
+}
+
+func formatSubagentDraftForUser(draft map[string]interface{}) string {
+	return fmt.Sprintf(
+		"已生成 subagent 草案。\nagent_id: %v\nrole: %v\ndisplay_name: %v\ntool_allowlist: %v\nrouting_keywords: %v\nsystem_prompt: %v\n\n回复“确认创建”会写入 config.json，回复“取消创建”会丢弃这个草案。",
+		draft["agent_id"],
+		draft["role"],
+		draft["display_name"],
+		draft["tool_allowlist"],
+		draft["routing_keywords"],
+		draft["system_prompt"],
+	)
+}
+
+func (al *AgentLoop) storePendingSubagentDraft(sessionKey string, draft map[string]interface{}) {
+	if al == nil || draft == nil {
+		return
+	}
+	if strings.TrimSpace(sessionKey) == "" {
+		sessionKey = "main"
+	}
+	al.streamMu.Lock()
+	defer al.streamMu.Unlock()
+	if al.pendingSubagentDraft == nil {
+		al.pendingSubagentDraft = map[string]map[string]interface{}{}
+	}
+	copied := make(map[string]interface{}, len(draft))
+	for k, v := range draft {
+		copied[k] = v
+	}
+	al.pendingSubagentDraft[sessionKey] = copied
+	if al.pendingDraftStore != nil {
+		_ = al.pendingDraftStore.Put(sessionKey, copied)
+	}
+}
+
+func (al *AgentLoop) loadPendingSubagentDraft(sessionKey string) map[string]interface{} {
+	if al == nil {
+		return nil
+	}
+	if strings.TrimSpace(sessionKey) == "" {
+		sessionKey = "main"
+	}
+	al.streamMu.Lock()
+	defer al.streamMu.Unlock()
+	draft := al.pendingSubagentDraft[sessionKey]
+	if draft == nil {
+		return nil
+	}
+	copied := make(map[string]interface{}, len(draft))
+	for k, v := range draft {
+		copied[k] = v
+	}
+	return copied
+}
+
+func (al *AgentLoop) deletePendingSubagentDraft(sessionKey string) {
+	if al == nil {
+		return
+	}
+	if strings.TrimSpace(sessionKey) == "" {
+		sessionKey = "main"
+	}
+	al.streamMu.Lock()
+	defer al.streamMu.Unlock()
+	delete(al.pendingSubagentDraft, sessionKey)
+	if al.pendingDraftStore != nil {
+		_ = al.pendingDraftStore.Delete(sessionKey)
+	}
+}
+
+func (al *AgentLoop) confirmPendingSubagentDraft(sessionKey string) (string, bool, error) {
+	draft := al.loadPendingSubagentDraft(sessionKey)
+	if draft == nil {
+		return "", false, nil
+	}
+	result, err := tools.UpsertConfigSubagent(al.configPath, draft)
+	if err != nil {
+		return "", true, err
+	}
+	al.deletePendingSubagentDraft(sessionKey)
+	return fmt.Sprintf("subagent 已写入 config.json。\nagent_id: %v\nrules: %v", result["agent_id"], result["rules"]), true, nil
+}
+
+func (al *AgentLoop) cancelPendingSubagentDraft(sessionKey string) (string, bool, error) {
+	draft := al.loadPendingSubagentDraft(sessionKey)
+	if draft == nil {
+		return "", false, nil
+	}
+	al.deletePendingSubagentDraft(sessionKey)
+	return "已取消这次 subagent 草案，不会写入 config.json。", true, nil
+}