fix: correctly handles array of token arrays in embedding request (#1404)

**Description**

This fixes the embeddings request parser so that it will not reject
"array of token arrays" input values which is already supported.

Signed-off-by: Takeshi Yoneda <[email protected]>

Author: mathetake

internal/apischema/openai/openai.go

@@ -311,9 +311,6 @@ func (s *EmbeddingRequestInput) UnmarshalJSON(data []byte) (err error) {
 	if err != nil {
 		return
 	}
-	if _, ok := s.Value.([][]int64); ok {
-		return fmt.Errorf("input has unsupported type [][]int64")
-	}
 	return
 }
 

internal/apischema/openai/openai_test.go

@@ -627,11 +627,6 @@ func TestEmbeddingRequestInputUnmarshal(t *testing.T) {
 			data:        []byte{},
 			expectedErr: "truncated input data",
 		},
-		{
-			name:        "nested int array (unsupported type)",
-			data:        []byte(`[[1, 2, 3], [4, 5, 6]]`),
-			expectedErr: "input has unsupported type [][]int64",
-		},
 	}
 
 	for _, tc := range errorCases {

internal/apischema/openai/union_test.go

@@ -52,6 +52,16 @@ func TestUnmarshalJSONNestedUnion(t *testing.T) {
 			data:     []byte(" \t\n\r\"test\""),
 			expected: "test",
 		},
+		{
+			name:     "array of token arrays",
+			data:     []byte(`[[-1, -2, -3], [1, 2, 3]]`),
+			expected: [][]int64{{-1, -2, -3}, {1, 2, 3}},
+		},
+		{
+			name:     "array of strings",
+			data:     []byte(`[ "aa", "bb", "cc" ]`),
+			expected: []string{"aa", "bb", "cc"},
+		},
 	}
 
 	allCases := append(promptUnionBenchmarkCases, additionalSuccessCases...) //nolint:gocritic // intentionally creating new slice