Skip to content

Commit 37e2cac

Browse files
committed
fix(test): check fmt.Fprint error in rerank mock handlers
1 parent a36269f commit 37e2cac

1 file changed

Lines changed: 6 additions & 2 deletions

File tree

rerank_test.go

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,9 @@ func TestRerankInvokesCohereRerankAndMapsScores(t *testing.T) {
5858
}
5959

6060
w.Header().Set("Content-Type", "application/json")
61-
fmt.Fprint(w, `{"results":[{"index":1,"relevance_score":0.94},{"index":0,"relevance_score":0.42}]}`)
61+
if _, err := fmt.Fprint(w, `{"results":[{"index":1,"relevance_score":0.94},{"index":0,"relevance_score":0.42}]}`); err != nil {
62+
t.Errorf("failed to write mock rerank response: %v", err)
63+
}
6264
}))
6365
defer server.Close()
6466

@@ -139,7 +141,9 @@ func TestRerankDefaultsTopNToDocumentCount(t *testing.T) {
139141
return
140142
}
141143
w.Header().Set("Content-Type", "application/json")
142-
fmt.Fprint(w, `{"results":[]}`)
144+
if _, err := fmt.Fprint(w, `{"results":[]}`); err != nil {
145+
t.Errorf("failed to write mock rerank response: %v", err)
146+
}
143147
}))
144148
defer server.Close()
145149

0 commit comments

Comments
 (0)