Merge 'add_embedding_cases' into 'main'

add embedding cases

See merge request: !5

Author: ByteDanceLiuYang

src/test/java/com/volcengine/vikingdb/runtime/vector/TestVar.java

@@ -3,10 +3,10 @@
 import com.volcengine.vikingdb.runtime.vector.model.request.EmbeddingModel;
 
 public class TestVar {
-    public static final String COLLECTION_NAME = "test_liuyang_coll_2";
-    public static final String COLLECTION_NAME_VECTORIZE = "test_liuyang_coll_vtz_2";
+    public static final String COLLECTION_NAME = "test_coll_for_java_sdk_with_vector";
+    public static final String COLLECTION_NAME_VECTORIZE = "test_coll_for_java_sdk_with_vectorize";
     public static final String INDEX_NAME = "idx_1";
     public static final String INDEX_NAME_SORT = "idx_rs_1";
-    public static final EmbeddingModel EMBEDDING_DENSE = new EmbeddingModel("doubao-embedding-large", "240915", 1024);
+    public static final EmbeddingModel EMBEDDING_DENSE = new EmbeddingModel("doubao-embedding-vision", "250615", 1024);
     public static final EmbeddingModel EMBEDDING_SPARSE = new EmbeddingModel("bge-m3", "default");
 }

src/test/java/com/volcengine/vikingdb/runtime/vector/embedding/EmbeddingTest.java

@@ -6,6 +6,7 @@
 import com.volcengine.vikingdb.runtime.vector.TestVar;
 import com.volcengine.vikingdb.runtime.vector.model.request.EmbeddingDataItem;
 import com.volcengine.vikingdb.runtime.vector.model.request.EmbeddingRequest;
+import com.volcengine.vikingdb.runtime.vector.model.request.FullModalData;
 import com.volcengine.vikingdb.runtime.vector.model.response.DataApiResponse;
 import com.volcengine.vikingdb.runtime.vector.model.response.EmbeddingResult;
 import com.volcengine.vikingdb.runtime.vector.service.*;
@@ -15,6 +16,7 @@
 import java.io.IOException;
 import java.io.InputStream;
 import java.util.Arrays;
+import java.util.Collections;
 import java.util.Properties;
 
 public class EmbeddingTest {
@@ -38,7 +40,20 @@ public static void setUp() throws Exception {
     }
 
     @Test
-    public void testSearchByScalarV1() throws Exception {
+    public void testEmbeddingForText() throws Exception {
+        EmbeddingRequest request = EmbeddingRequest.builder()
+                .denseModel(TestVar.EMBEDDING_DENSE)
+                .data(Arrays.asList(
+                        EmbeddingDataItem.builder().text("我喜欢听歌").build(),
+                        EmbeddingDataItem.builder().text("我喜欢雨天").build()
+                ))
+                .build();
+        DataApiResponse<EmbeddingResult> response = service.embedding(request);
+        System.out.println(response);
+    }
+
+    @Test
+    public void testEmbeddingForTextWithSparse() throws Exception {
         EmbeddingRequest request = EmbeddingRequest.builder()
                 .denseModel(TestVar.EMBEDDING_DENSE)
                 .sparseModel(TestVar.EMBEDDING_SPARSE)
@@ -50,4 +65,31 @@ public void testSearchByScalarV1() throws Exception {
         DataApiResponse<EmbeddingResult> response = service.embedding(request);
         System.out.println(response);
     }
+
+    @Test
+    public void testEmbeddingForTextAndImage() throws Exception {
+        EmbeddingRequest request = EmbeddingRequest.builder()
+                .denseModel(TestVar.EMBEDDING_DENSE)
+                .data(Arrays.asList(
+                        EmbeddingDataItem.builder().text("我喜欢听歌").image("tos://your_bucket/1.jpg").build(),
+                        EmbeddingDataItem.builder().text("我喜欢雨天").build()
+                ))
+                .build();
+        DataApiResponse<EmbeddingResult> response = service.embedding(request);
+        System.out.println(response);
+    }
+
+    @Test
+    public void testEmbeddingForFullModalSeq() throws Exception {
+        EmbeddingRequest request = EmbeddingRequest.builder()
+                .denseModel(TestVar.EMBEDDING_DENSE)
+                .sparseModel(TestVar.EMBEDDING_SPARSE)
+                .data(Collections.singletonList(EmbeddingDataItem.builder().fullModalSeq(Arrays.asList(
+                            FullModalData.builder().text("我喜欢听歌").build(),
+                            FullModalData.builder().image("tos://your_bucket/.1.jpg").build(),
+                            FullModalData.builder().video("tos://your_bucket/.1.mp4").build()))
+                .build())).build();
+        DataApiResponse<EmbeddingResult> response = service.embedding(request);
+        System.out.println(response);
+    }
 }