Skip to content

Commit 386d033

Browse files
mjacobsmariusvniekerk
authored andcommitted
fix(server): preserve prompt source on upload
1 parent 34e3df2 commit 386d033

2 files changed

Lines changed: 21 additions & 0 deletions

File tree

internal/server/upload.go

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -216,6 +216,7 @@ func sessionBatchWriteFromParsed(
216216
ContentLength: m.ContentLength,
217217
Model: m.Model,
218218
TokenUsage: m.TokenUsage,
219+
PromptSource: m.PromptSource,
219220
ContextTokens: m.ContextTokens,
220221
OutputTokens: m.OutputTokens,
221222
HasContextTokens: hasCtx,

internal/server/upload_internal_test.go

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -48,3 +48,23 @@ func TestSessionBatchWriteFromParsedPreservesSessionIdentity(t *testing.T) {
4848
assert.Equal(t, "Claude Code", result.Session.AgentLabel)
4949
assert.Equal(t, "claude-sdk", result.Session.Entrypoint)
5050
}
51+
52+
func TestSessionBatchWriteFromParsedPreservesClaudeProvenance(t *testing.T) {
53+
sess := parser.ParsedSession{
54+
ID: "test-claude-provenance",
55+
Agent: parser.AgentClaude,
56+
SessionKind: "bg",
57+
}
58+
msgs := []parser.ParsedMessage{{
59+
Ordinal: 1,
60+
Role: parser.RoleUser,
61+
Content: "queued prompt",
62+
PromptSource: "queued",
63+
}}
64+
65+
result := sessionBatchWriteFromParsed(sess, msgs)
66+
67+
assert.Equal(t, "bg", result.Session.SessionKind)
68+
require.Len(t, result.Messages, 1)
69+
assert.Equal(t, "queued", result.Messages[0].PromptSource)
70+
}

0 commit comments

Comments
 (0)