superagent请求问题修复

This commit is contained in:
andy committed 2026-09-06 01:40:15 +08:00
1 parent ee9d2d7cf1
commit 7243319bbb
19 files changed
+388 -74

No files matched your search

@@ -105,6 +105,39 @@ func TestHTTPClientCreatesSessionAndStreamsMessage(t *testing.T) {
}
}
func TestHTTPClientStreamsWithoutTraceWhenDisabled(t *testing.T) {
server := httptest.NewServer(http.HandlerFunc(func(writer http.ResponseWriter, request *http.Request) {
if got := request.URL.Query().Get("include_trace"); got != "false" {
t.Errorf("include_trace = %q, want false", got)
}
writer.Header().Set("Content-Type", "text/event-stream")
fmt.Fprint(writer, "event: messages\ndata: [{\"type\":\"AIMessageChunk\",\"content\":\"\",\"response_metadata\":{\"finish_reason\":\"stop\"}}]\n\n")
fmt.Fprint(writer, "event: message.final\ndata: {\"run_id\":\"run-no-trace\",\"text\":\"no-trace answer\"}\n\n")
fmt.Fprint(writer, "event: end\n\n")
}))
defer server.Close()
client, err := NewHTTPClient(Config{
Enabled: true,
BaseURL: server.URL,
APIKey: testAPIKey,
IncludeTrace: false,
ConnectTimeout: time.Second,
RecoveryInitialBackoff: time.Millisecond,
MaxMessageBytes: 1024,
})
if err != nil {
t.Fatalf("NewHTTPClient() error = %v", err)
}
result, err := client.StreamMessage(context.Background(), validMessageRequest(), nil)
if err != nil {
t.Fatalf("StreamMessage() error = %v", err)
}
if result.Answer != "no-trace answer" {
t.Fatalf("Answer = %q, want no-trace answer", result.Answer)
}
}
func TestHTTPClientRecoversWithoutRepostingMessage(t *testing.T) {
var messagePosts atomic.Int32
var runQueries atomic.Int32
@@ -405,6 +438,7 @@ func newTestHTTPClient(t *testing.T, baseURL string, recoveryAttempts int, backo
Enabled: true,
BaseURL: baseURL,
APIKey: testAPIKey,
IncludeTrace: true,
ConnectTimeout: time.Second,
RecoveryMaxAttempts: recoveryAttempts,
RecoveryInitialBackoff: backoff,