remove Chunk generic

Author: jan-elastic

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/EmbeddingResults.java

@@ -19,9 +19,7 @@
  * A call to the inference service may contain multiple input texts, so this results may
  * contain multiple results.
  */
-public interface EmbeddingResults<C extends EmbeddingResults.Chunk, E extends EmbeddingResults.Embedding<C, E>>
-    extends
-        InferenceServiceResults {
+public interface EmbeddingResults<E extends EmbeddingResults.Embedding<E>> extends InferenceServiceResults {
 
     /**
      * A resulting embedding together with the offset into the input text.
@@ -35,11 +33,11 @@ interface Chunk {
     /**
      * A resulting embedding for one of the input texts to the inference service.
      */
-    interface Embedding<C extends Chunk, E extends EmbeddingResults.Embedding<C, E>> {
+    interface Embedding<E extends Embedding<E>> {
         /**
          * Combines the resulting embedding with the offset into the input text into a chunk.
          */
-        C toChunk(ChunkedInference.TextOffset offset);
+        Chunk toChunk(ChunkedInference.TextOffset offset);
 
         /**
          * Merges the existing embedding and provided embedding into a new embedding.

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/SparseEmbeddingResults.java

@@ -37,9 +37,7 @@
 
 import static org.elasticsearch.xpack.core.ml.inference.trainedmodel.InferenceConfig.DEFAULT_RESULTS_FIELD;
 
-public record SparseEmbeddingResults(List<Embedding> embeddings)
-    implements
-        EmbeddingResults<SparseEmbeddingResults.Chunk, SparseEmbeddingResults.Embedding> {
+public record SparseEmbeddingResults(List<Embedding> embeddings) implements EmbeddingResults<SparseEmbeddingResults.Embedding> {
 
     public static final String NAME = "sparse_embedding_results";
     public static final String SPARSE_EMBEDDING = TaskType.SPARSE_EMBEDDING.toString();
@@ -126,7 +124,7 @@ public record Embedding(List<WeightedToken> tokens, boolean isTruncated)
         implements
             Writeable,
             ToXContentObject,
-            EmbeddingResults.Embedding<Chunk, Embedding> {
+            EmbeddingResults.Embedding<Embedding> {
 
         public static final String EMBEDDING = "embedding";
         public static final String IS_TRUNCATED = "is_truncated";

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/TextEmbeddingByteResults.java

@@ -124,7 +124,7 @@ public record Embedding(byte[] values, int[] sumMergedValues, int numberOfMerged
         implements
             Writeable,
             ToXContentObject,
-            EmbeddingResults.Embedding<Chunk, Embedding> {
+            EmbeddingResults.Embedding<Embedding> {
 
         public static final String EMBEDDING = "embedding";
 

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/TextEmbeddingFloatResults.java

@@ -161,7 +161,7 @@ public record Embedding(float[] values, int numberOfMergedEmbeddings)
         implements
             Writeable,
             ToXContentObject,
-            EmbeddingResults.Embedding<Chunk, Embedding> {
+            EmbeddingResults.Embedding<Embedding> {
         public static final String EMBEDDING = "embedding";
 
         public Embedding(float[] values) {

x-pack/plugin/core/src/main/java/org/elasticsearch/xpack/core/inference/results/TextEmbeddingResults.java

@@ -7,9 +7,9 @@
 
 package org.elasticsearch.xpack.core.inference.results;
 
-public interface TextEmbeddingResults<C extends EmbeddingResults.Chunk, E extends EmbeddingResults.Embedding<C, E>>
+public interface TextEmbeddingResults<C extends EmbeddingResults.Chunk, E extends EmbeddingResults.Embedding<E>>
     extends
-        EmbeddingResults<C, E> {
+        EmbeddingResults<E> {
 
     /**
      * Returns the first text embedding entry in the result list's array size.

x-pack/plugin/inference/src/main/java/org/elasticsearch/xpack/inference/chunking/EmbeddingRequestChunker.java

@@ -36,7 +36,7 @@
  * processing and map the results back to the original element
  * in the input list.
  */
-public class EmbeddingRequestChunker<C extends EmbeddingResults.Chunk, E extends EmbeddingResults.Embedding<C, E>> {
+public class EmbeddingRequestChunker<E extends EmbeddingResults.Embedding<E>> {
 
     // Visible for testing
     record Request(int inputIndex, int chunkIndex, ChunkOffset chunk, List<String> inputs) {
@@ -150,12 +150,12 @@ private class DebatchingListener implements ActionListener<InferenceServiceResul
 
         @Override
         public void onResponse(InferenceServiceResults inferenceServiceResults) {
-            if (inferenceServiceResults instanceof EmbeddingResults<?, ?> == false) {
+            if (inferenceServiceResults instanceof EmbeddingResults<?> == false) {
                 onFailure(unexpectedResultTypeException(inferenceServiceResults.getWriteableName()));
                 return;
             }
             @SuppressWarnings("unchecked")
-            EmbeddingResults<C, E> embeddingResults = (EmbeddingResults<C, E>) inferenceServiceResults;
+            EmbeddingResults<E> embeddingResults = (EmbeddingResults<E>) inferenceServiceResults;
             if (embeddingResults.embeddings().size() != request.requests.size()) {
                 onFailure(numResultsDoesntMatchException(embeddingResults.embeddings().size(), request.requests.size()));
                 return;