✨Feat: 构建对话终端

internal/ai/prompt.go

@@ -0,0 +1,71 @@
+package ai
+
+import (
+	"embed"
+	"fmt"
+	"strings"
+	"text/template"
+)
+
+//go:embed prompts/*.txt
+var promptFS embed.FS
+
+// PromptManager 提示词管理器
+type PromptManager struct {
+	templates map[string]*template.Template
+}
+
+// NewPromptManager 创建提示词管理器
+func NewPromptManager() *PromptManager {
+	pm := &PromptManager{
+		templates: make(map[string]*template.Template),
+	}
+
+	pm.loadTemplates()
+	return pm
+}
+
+func (pm *PromptManager) loadTemplates() {
+	templateFiles := map[string]string{
+		"mermaid":   "prompts/mermaid.txt",
+		"guodegang": "prompts/guodegang.txt",
+		"quiz":      "prompts/quiz.txt",
+		"card":      "prompts/card.txt",
+		"scenario":  "prompts/scenario.txt",
+	}
+
+	for name, path := range templateFiles {
+		content, err := promptFS.ReadFile(path)
+		if err != nil {
+			continue
+		}
+
+		tmpl, err := template.New(name).Parse(string(content))
+		if err != nil {
+			continue
+		}
+
+		pm.templates[name] = tmpl
+	}
+}
+
+// GeneratePrompt 生成提示词
+func (pm *PromptManager) GeneratePrompt(templateType string, knowledge string) (string, error) {
+	tmpl, ok := pm.templates[templateType]
+	if !ok {
+		return "", fmt.Errorf("template type %s not found", templateType)
+	}
+
+	var buf strings.Builder
+	data := struct {
+		Knowledge string
+	}{
+		Knowledge: knowledge,
+	}
+
+	if err := tmpl.Execute(&buf, data); err != nil {
+		return "", fmt.Errorf("failed to execute template: %w", err)
+	}
+
+	return buf.String(), nil
+}