fix(rag): Missing the dimension parameter for embedding (#188)

Author: guanxuc

agentscope-extensions/agentscope-extensions-rag-simple/src/main/java/io/agentscope/core/embedding/EmbeddingModel.java

@@ -57,6 +57,7 @@ public interface EmbeddingModel {
 
     /**
      * Get the dimension of embedding vectors produced by this model.
+     * The default value is 1024 if not specified.
      *
      * @return the dimension of embedding vectors
      */

agentscope-extensions/agentscope-extensions-rag-simple/src/main/java/io/agentscope/core/embedding/dashscope/DashScopeMultiModalEmbedding.java

@@ -30,6 +30,7 @@
 import io.agentscope.core.message.TextBlock;
 import io.agentscope.core.model.ExecutionConfig;
 import java.util.List;
+import java.util.Map;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
 import reactor.core.publisher.Mono;
@@ -202,6 +203,7 @@ private MultiModalEmbeddingParam buildTextParam(String text) {
         return MultiModalEmbeddingParam.builder()
                 .apiKey(apiKey)
                 .model(modelName)
+                .parameters(Map.of("dimension", dimensions))
                 .contents(List.of(MultiModalEmbeddingItemText.builder().text(text).build()))
                 .build();
     }
@@ -216,6 +218,7 @@ private MultiModalEmbeddingParam buildImageParam(String imageUrl) {
         return MultiModalEmbeddingParam.builder()
                 .apiKey(apiKey)
                 .model(modelName)
+                .parameters(Map.of("dimension", dimensions))
                 .contents(List.of(MultiModalEmbeddingItemImage.builder().image(imageUrl).build()))
                 .build();
     }
@@ -305,7 +308,7 @@ public int getDimensions() {
     public static class Builder {
         private String apiKey;
         private String modelName;
-        private int dimensions;
+        private int dimensions = 1024;
         private ExecutionConfig defaultExecutionConfig;
         private String baseUrl;
 

agentscope-extensions/agentscope-extensions-rag-simple/src/main/java/io/agentscope/core/embedding/dashscope/DashScopeTextEmbedding.java

@@ -119,6 +119,7 @@ public Mono<double[]> embed(ContentBlock block) {
                                                 TextEmbeddingParam.builder()
                                                         .apiKey(apiKey)
                                                         .model(modelName)
+                                                        .dimension(dimensions)
                                                         .texts(List.of(text))
                                                         .build();
 
@@ -216,7 +217,7 @@ public int getDimensions() {
     public static class Builder {
         private String apiKey;
         private String modelName;
-        private int dimensions;
+        private int dimensions = 1024;
         private ExecutionConfig defaultExecutionConfig;
         private String baseUrl;