superagent请求问题修复
This commit is contained in:
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,
|
||||
|
||||
Reference in new issue
Block a user