lisone/ds2api
1
0
Fork 0
mirror of https://github.com/CJackHwang/ds2api.git synced 2026-05-09 02:45:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix zero-based citation index normalization

Browse Source
This commit is contained in:
CJACK.
2026-04-20 18:18:00 +08:00
parent aa7f821151
commit ae37654893
2 changed files with 24 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
internal/sse/citation_links.go
View File

@@ -66,7 +66,11 @@ func (c *citationLinkCollector) captureURLAndIndex(m map[string]any) {
if !hasIdx {
return
}
if idx <= 0 {
// DeepSeek citation indices in search results are zero-based (0,1,2,...),
// while visible markers are one-based ([citation:1], [citation:2], ...).
// Normalize all non-negative explicit indices to one-based to avoid
// misalignment when 3+ citations are present.
if idx >= 0 {
idx = idx + 1
}
if idx <= 0 {

19
internal/sse/consumer_edge_test.go
View File

@@ -131,6 +131,25 @@ func TestCollectStreamExtractsCitationLinks(t *testing.T) {
}
}
func TestCollectStreamExtractsCitationLinksForSequentialZeroBasedIndices(t *testing.T) {
resp := makeHTTPResponse(
"data: {\"p\":\"response/fragments/-1/results\",\"v\":[{\"url\":\"https://example.com/a\",\"cite_index\":0},{\"url\":\"https://example.com/b\",\"cite_index\":1},{\"url\":\"https://example.com/c\",\"cite_index\":2}]}\n" +
"data: {\"p\":\"response/content\",\"v\":\"结论[citation:1][citation:2][citation:3]\"}\n" +
"data: [DONE]\n",
)
result := CollectStream(resp, false, false)
if got := result.CitationLinks[1]; got != "https://example.com/a" {
t.Fatalf("expected citation 1 link, got %q", got)
}
if got := result.CitationLinks[2]; got != "https://example.com/b" {
t.Fatalf("expected citation 2 link, got %q", got)
}
if got := result.CitationLinks[3]; got != "https://example.com/c" {
t.Fatalf("expected citation 3 link, got %q", got)
}
}
func TestCollectStreamMultipleThinkingChunks(t *testing.T) {
resp := makeHTTPResponse(
"data: {\"p\":\"response/thinking_content\",\"v\":\"part1\"}\n" +