Method Extract

pkg/memory/dialog/extractor.go:90–120

Extract 从消息中提取偏好

(message string)

// Extract 从消息中提取偏好
func (e *Extractor) Extract(message string) []*ExtractedPreference {
	var preferences []*ExtractedPreference

	for category, keywords := range e.keywords {
		for _, keyword := range keywords {
			if strings.Contains(message, keyword) {
				// 提取包含关键词的句子
				content := e.extractSentence(message, keyword)
				if content == "" {
					continue
				}

				pref := &ExtractedPreference{
					Category:   category,
					Content:    content,
					Keywords:   []string{keyword},
					Confidence: e.calculateConfidence(message, keyword, category),
					Timestamp:  time.Now(),
				}

				if pref.Confidence >= e.config.MinConfidence {
					preferences = append(preferences, pref)
				}

				break // 每个类别只提取一次
			}
		}
	}

	return preferences
}

// HasPreferenceKeyword 检查消息是否包含偏好关键词
func (e *Extractor) HasPreferenceKeyword(message string) bool {