chore: run modernize

Author: andreynering

api_client.go

@@ -575,7 +575,7 @@ func (ac *apiClient) uploadFile(ctx context.Context, r io.Reader, uploadURL stri
 		} else if err != nil {
 			return nil, fmt.Errorf("Failed to read bytes from file at offset %d: %w. Bytes actually read: %d", offset, err, bytesRead)
 		}
-		for attempt := 0; attempt < maxRetryCount; attempt++ {
+		for attempt := range maxRetryCount {
 			patchedHTTPOptions, err := patchHTTPOptions(ac.clientConfig.HTTPOptions, *httpOptions)
 			if err != nil {
 				return nil, err

api_client_test.go

@@ -1310,7 +1310,7 @@ func createTestFile(t *testing.T, size int64) (string, func()) {
 
 	buf := make([]byte, 1024*1024) // 1MB buffer
 	pattern := []byte("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ0123456789!@#$%^&*()")
-	for i := 0; i < len(buf); i++ {
+	for i := range buf {
 		buf[i] = pattern[i%len(pattern)]
 	}
 

common.go

@@ -21,6 +21,7 @@ import (
 	"fmt"
 	"iter"
 	"log"
+	"maps"
 	"net/http"
 	"net/url"
 	"reflect"
@@ -135,9 +136,7 @@ func setValueByPath(data map[string]any, keys []string, value any) {
 			if newMap, ok2 := value.(map[string]any); ok2 {
 				// Instead of overwriting dictionary with another dictionary, merge them.
 				// This is important for handling training and validation datasets in tuning.
-				for k, v := range newMap {
-					existingMap[k] = v
-				}
+				maps.Copy(existingMap, newMap)
 				data[finalKey] = existingMap // Assign the updated map back
 			}
 		} else {
@@ -147,9 +146,7 @@ func setValueByPath(data map[string]any, keys []string, value any) {
 		if finalKey == "_self" && reflect.TypeOf(value).Kind() == reflect.Map {
 			// Iterate through the `value` map and copy its contents to `data`.
 			if valMap, ok := value.(map[string]any); ok {
-				for k, v := range valMap {
-					data[k] = v
-				}
+				maps.Copy(data, valMap)
 			}
 		} else {
 			// If existing_data is None (or key doesn't exist), set the value directly.

pages.go

@@ -18,6 +18,7 @@ import (
 	"context"
 	"errors"
 	"iter"
+	"maps"
 )
 
 // ErrPageDone is the error returned by an iterator's Next method when no more pages are available.
@@ -88,9 +89,7 @@ func (p Page[T]) Next(ctx context.Context) (Page[T], error) {
 		return p, ErrPageDone
 	}
 	c := make(map[string]any)
-	for k, v := range p.config {
-		c[k] = v
-	}
+	maps.Copy(c, p.config)
 	c["PageToken"] = p.NextPageToken
 
 	return newPage[T](ctx, p.Name, c, p.listFunc)

replay_sanitizer.go

@@ -60,7 +60,7 @@ func sanitizeMapWithSourceType(t *testing.T, sourceType reflect.Type, m any) {
 	for _, path := range paths {
 		if sourceType.Kind() == reflect.Slice {
 			data := m.([]any)
-			for i := 0; i < len(data); i++ {
+			for i := range data {
 				sanitizeMapByPath(data[i], path, stdBase64Handler, false)
 			}
 		} else {

transformer.go

@@ -220,8 +220,8 @@ func tFileName(name any) (string, error) {
 					return "", fmt.Errorf("could not extract file name from URI: %s", name)
 				}
 				name = match[0]
-			} else if strings.HasPrefix(name, "files/") {
-				name = strings.TrimPrefix(name, "files/")
+			} else if after, ok := strings.CutPrefix(name, "files/"); ok {
+				name = after
 			}
 			return name, nil
 		}