improved code coverage and removed unnecessary checks

client.go

@@ -86,20 +86,14 @@ func withBody(body any) requestOption {
 
 func withExtraBody(extraBody map[string]any) requestOption {
 	return func(args *requestOptions) {
-		// Initialize args.body as a map[string]any if it's nil.
-		if args.body == nil {
-			args.body = make(map[string]any)
-		}
 		// Assert that args.body is a map[string]any.
 		bodyMap, ok := args.body.(map[string]any)
-		if !ok {
-			// If it's not, initialize it as a map[string]any.
-			bodyMap = make(map[string]any)
-			args.body = bodyMap
-		}
-		// Add extraBody fields to args.body.
-		for key, value := range extraBody {
-			bodyMap[key] = value
+		if ok {
+			// If it's a map[string]any then only add extraBody
+			// fields to args.body otherwise keep only fields in request struct.
+			for key, value := range extraBody {
+				bodyMap[key] = value
+			}
 		}
 	}
 }

embeddings_test.go

@@ -150,6 +150,7 @@ func TestEmbeddingEndpoint(t *testing.T) {
 				"input_type": "query",
 				"truncate":   "NONE",
 			},
+			Dimensions: 1,
 		},
 	)
 	checks.NoError(t, err, "CreateEmbeddings error")