feat: support hyperparameters in distillation tuning

PiperOrigin-RevId: 890622141

Author: speedstorm1

replay_sanitizer.go

@@ -67,6 +67,36 @@ func sanitizeMapWithSourceType(t *testing.T, sourceType reflect.Type, m any) {
 			sanitizeMapByPath(m.(map[string]any), path, stdBase64Handler, false)
 		}
 	}
+
+	int32Paths := make([]string, 0)
+	visitedTypesInt32 := make(map[string]bool)
+	if err := getFieldPath(st, reflect.TypeOf(int32(0)), &int32Paths, "", visitedTypesInt32, false); err != nil {
+		t.Fatal(err)
+	}
+
+	numericStringHandler := func(data any, path string) any {
+		s, ok := data.(string)
+		if !ok {
+			return data
+		}
+		f, err := strconv.ParseFloat(s, 64)
+		if err != nil {
+			t.Errorf("invalid numeric string %s at path %s", s, path)
+			return data
+		}
+		return f
+	}
+
+	for _, path := range int32Paths {
+		if sourceType.Kind() == reflect.Slice {
+			data := m.([]any)
+			for i := 0; i < len(data); i++ {
+				sanitizeMapByPath(data[i], path, numericStringHandler, false)
+			}
+		} else {
+			sanitizeMapByPath(m.(map[string]any), path, numericStringHandler, false)
+		}
+	}
 }
 
 // sanitizeMapByPath sanitizes a value within a nested map structure based on the given path.

table_test.go

@@ -330,6 +330,16 @@ func TestTable(t *testing.T) {
 								}
 							} else {
 								method := extractMethod(t, &testTableFile, client)
+
+								// Sanitize replay response body segments based on the method's return type.
+								// This is needed because some numeric fields in the replay files might be
+								// represented as strings, which causes unmarshalling errors in the Go SDK.
+								for _, interaction := range replayClient.ReplayFile.Interactions {
+									for _, bodySegment := range interaction.Response.BodySegments {
+										sanitizeMapWithSourceType(t, method.Type().Out(0), bodySegment)
+									}
+								}
+
 								args := extractArgs(ctx, t, method, &testTableFile, testTableItem)
 
 								// Inject unknown fields to the replay file to simulate the case where the SDK adds