replay.go RawHistoryFolder

1package provider
2
3// WithMessageOrigin records the producing route without changing shared metadata.
4// Explicit provenance survives session replay, compaction, and model switches.
5func WithMessageOrigin(msg Message, providerName, model string) Message {
6	if msg.Role != RoleAssistant || (providerName == "" && model == "") {
7		return msg
8	}
9	meta := make(map[string]string, len(msg.Meta)+2)
10	for k, v := range msg.Meta {
11		meta[k] = v
12	}
13	if meta["provider"] == "" && providerName != "" {
14		meta["provider"] = providerName
15	}
16	if meta["model"] == "" && model != "" {
17		meta["model"] = model
18	}
19	msg.Meta = meta
20	return msg
21}
22
23// replayContent projects model-private state out of foreign assistant turns.
24// The stored transcript and ordinary text/tool blocks remain intact. Unknown
25// provenance retains historical behavior; each wire adapter still validates
26// the shape of opaque items it accepts.
27func replayContent(msg Message, providerName, model string) []Content {
28	if (msg.Meta["provider"] == "" || msg.Meta["provider"] == providerName) &&
29		(msg.Meta["model"] == "" || msg.Meta["model"] == model) {
30		return msg.Content
31	}
32	content := make([]Content, 0, len(msg.Content))
33	for _, block := range msg.Content {
34		switch v := block.(type) {
35		case ReasoningBlock:
36			continue
37		case TextBlock:
38			v.ThoughtSignature = ""
39			block = v
40		case ImageBlock:
41			v.ThoughtSignature = ""
42			block = v
43		case ToolCallBlock:
44			v.ThoughtSignature = ""
45			block = v
46		}
47		content = append(content, block)
48	}
49	return content
50}