test(backend): 修复 usage 类型断言未检查
This commit is contained in:
@@ -163,11 +163,9 @@ func TestStreamingReconcile_MessageDelta(t *testing.T) {
|
|||||||
require.Equal(t, "message_delta", eventType)
|
require.Equal(t, "message_delta", eventType)
|
||||||
|
|
||||||
// 模拟 processSSEEvent 中的 reconcile 逻辑
|
// 模拟 processSSEEvent 中的 reconcile 逻辑
|
||||||
if u, ok := event["usage"].(map[string]any); ok {
|
usage, ok := event["usage"].(map[string]any)
|
||||||
reconcileCachedTokens(u)
|
require.True(t, ok)
|
||||||
}
|
reconcileCachedTokens(usage)
|
||||||
|
|
||||||
usage := event["usage"].(map[string]any)
|
|
||||||
assert.Equal(t, float64(15), usage["cache_read_input_tokens"])
|
assert.Equal(t, float64(15), usage["cache_read_input_tokens"])
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -183,11 +181,9 @@ func TestStreamingReconcile_MessageDelta_NativeClaude(t *testing.T) {
|
|||||||
var event map[string]any
|
var event map[string]any
|
||||||
require.NoError(t, json.Unmarshal([]byte(eventJSON), &event))
|
require.NoError(t, json.Unmarshal([]byte(eventJSON), &event))
|
||||||
|
|
||||||
if u, ok := event["usage"].(map[string]any); ok {
|
usage, ok := event["usage"].(map[string]any)
|
||||||
reconcileCachedTokens(u)
|
require.True(t, ok)
|
||||||
}
|
reconcileCachedTokens(usage)
|
||||||
|
|
||||||
usage := event["usage"].(map[string]any)
|
|
||||||
_, hasCacheRead := usage["cache_read_input_tokens"]
|
_, hasCacheRead := usage["cache_read_input_tokens"]
|
||||||
assert.False(t, hasCacheRead, "不应为原生 Claude 响应注入 cache_read_input_tokens")
|
assert.False(t, hasCacheRead, "不应为原生 Claude 响应注入 cache_read_input_tokens")
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user