Fix test

cohere-ai · Feb 7, 2024 · cd29b51 · cd29b51
1 parent d7975a5
commit cd29b51
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/tests/sdk_test.go b/tests/sdk_test.go
@@ -7,6 +7,7 @@ import (
 	"os"
 	"strings"
 	"testing"
+	"time"
 
 	cohere "github.com/cohere-ai/cohere-go/v2"
 	client "github.com/cohere-ai/cohere-go/v2/client"
@@ -227,9 +228,13 @@ func TestNewClient(t *testing.T) {
 	})
 
 	t.Run("TestListDatasets", func(t *testing.T) {
+		timee := time.Now()
 		datasets, err := client.Datasets.List(
 			context.TODO(),
-			&cohere.DatasetsListRequest{})
+			&cohere.DatasetsListRequest{
+				Before: &timee,
+				After:  &timee,
+			})
 
 		require.NoError(t, err)
 		print(datasets)