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

refactor: enhance XML tool call parsing to support nested structures, CDATA, and repeated tags

Browse Source
This commit is contained in:
CJACK
2026-04-19 19:58:45 +08:00
parent 26d195f2a6
commit 0f2b5fee23
16 changed files with 550 additions and 140 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
internal/util/util_edge_test.go
View File

@@ -195,9 +195,12 @@ func TestMessagesPrepareAssistantMarkers(t *testing.T) {
if strings.Count(got, "<｜end▁of▁sentence｜>") != 1 {
t.Fatalf("expected one end_of_sentence (assistant only), got %q", got)
}
if !strings.Contains(got, "<｜Assistant｜></think>Hello!<｜end▁of▁sentence｜>") {
if !strings.Contains(got, "<｜Assistant｜>Hello!<｜end▁of▁sentence｜>") {
t.Fatalf("expected assistant EOS suffix, got %q", got)
}
if strings.Contains(got, "<think>") || strings.Contains(got, "</think>") {
t.Fatalf("did not expect think tags in prompt, got %q", got)
}
if strings.Contains(got, "<system_instructions>") {
t.Fatalf("did not expect legacy system marker, got %q", got)
}