Files
sub2api/backend/internal/service/openai_content_session_seed_benchmark_test.go
T
李建琦 6d655c9903
Release / update-version (push) Has been cancelled
Release / build-frontend (push) Has been cancelled
Release / release (push) Has been cancelled
Release / sync-version-file (push) Has been cancelled
CI / shell (push) Canceled after 0s
CI / test (push) Canceled after 0s
CI / frontend (push) Canceled after 0s
CI / golangci-lint (push) Canceled after 0s
Security Scan / backend-security (push) Canceled after 0s
Security Scan / frontend-security (push) Canceled after 0s
Sub2API v1.0 - AI API 网关(二开初始版本,基于上游 Wei-Shaw/sub2api)
2026-08-21 18:30:13 +08:00

30 lines
1.1 KiB
Go

package service
import (
"strings"
"testing"
)
var benchmarkOpenAIContentSessionSeed string
func BenchmarkDeriveOpenAIContentSessionSeedLargeBody(b *testing.B) {
largeHistory := strings.Repeat("payload", 1<<17)
tests := []struct {
name string
body []byte
}{
{name: "ChatCompletions", body: []byte(`{"model":"gpt-5.4","tools":[{"type":"function","function":{"name":"lookup"}}],"messages":[{"role":"system","content":"Be concise."},{"role":"user","content":"Hello"},{"role":"assistant","content":"` + largeHistory + `"},{"role":"user","content":"Follow-up"}]}`)},
{name: "Responses", body: []byte(`{"model":"gpt-5.4","instructions":"Be concise.","tools":[{"type":"function","name":"lookup"}],"input":[{"role":"system","content":"System prompt"},{"role":"user","content":"Hello"},{"role":"assistant","content":"` + largeHistory + `"}]}`)},
}
for _, test := range tests {
b.Run(test.name, func(b *testing.B) {
b.ReportAllocs()
b.SetBytes(int64(len(test.body)))
for range b.N {
benchmarkOpenAIContentSessionSeed = deriveOpenAIContentSessionSeed(test.body)
}
})
}
}