79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
|
package copymission
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"strings"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
type ResearchSource struct {
|
|||
|
|
Title string
|
|||
|
|
URL string
|
|||
|
|
Snippet string
|
|||
|
|
Query string
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// FormatResearchItemsForMatrix renders Brave research snippets for the matrix prompt.
|
|||
|
|
func FormatResearchItemsForMatrix(items []ResearchSource) string {
|
|||
|
|
if len(items) == 0 {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
lines := make([]string, 0, len(items))
|
|||
|
|
seen := map[string]struct{}{}
|
|||
|
|
for _, item := range items {
|
|||
|
|
title := strings.TrimSpace(item.Title)
|
|||
|
|
url := strings.TrimSpace(item.URL)
|
|||
|
|
snippet := strings.TrimSpace(item.Snippet)
|
|||
|
|
if title == "" && url == "" && snippet == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
key := strings.ToLower(url)
|
|||
|
|
if key == "" {
|
|||
|
|
key = strings.ToLower(title + "|" + snippet)
|
|||
|
|
}
|
|||
|
|
if _, ok := seen[key]; ok {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
seen[key] = struct{}{}
|
|||
|
|
var parts []string
|
|||
|
|
if title != "" {
|
|||
|
|
parts = append(parts, title)
|
|||
|
|
}
|
|||
|
|
if url != "" {
|
|||
|
|
parts = append(parts, url)
|
|||
|
|
}
|
|||
|
|
if snippet != "" {
|
|||
|
|
parts = append(parts, snippet)
|
|||
|
|
}
|
|||
|
|
lines = append(lines, strings.Join(parts, "｜"))
|
|||
|
|
}
|
|||
|
|
if len(lines) == 0 {
|
|||
|
|
return ""
|
|||
|
|
}
|
|||
|
|
return strings.Join(lines, "\n")
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// MergeReferenceURLs combines graph-selected URLs with mission research item URLs.
|
|||
|
|
func MergeReferenceURLs(selected []string, researchItems []ResearchSource) []string {
|
|||
|
|
if len(researchItems) == 0 {
|
|||
|
|
return selected
|
|||
|
|
}
|
|||
|
|
out := append([]string(nil), selected...)
|
|||
|
|
seen := map[string]struct{}{}
|
|||
|
|
for _, raw := range selected {
|
|||
|
|
if key := normalizeURLKey(raw); key != "" {
|
|||
|
|
seen[key] = struct{}{}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
for _, item := range researchItems {
|
|||
|
|
raw := strings.TrimSpace(item.URL)
|
|||
|
|
key := normalizeURLKey(raw)
|
|||
|
|
if key == "" {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
if _, ok := seen[key]; ok {
|
|||
|
|
continue
|
|||
|
|
}
|
|||
|
|
seen[key] = struct{}{}
|
|||
|
|
out = append(out, raw)
|
|||
|
|
}
|
|||
|
|
return out
|
|||
|
|
}
|