import com.volcengine.tos.TOSVectorsClientBuilder;
import com.volcengine.tos.TOSVectors;
import com.volcengine.tos.TosClientException;
import com.volcengine.tos.TosServerException;
import com.volcengine.tos.model.vectors.PutVectorsInput;
import com.volcengine.tos.model.vectors.PutVectorsOutput;
import com.volcengine.tos.model.vectors.Vector;
import com.volcengine.tos.model.vectors.VectorData;
import java.util.ArrayList;
import java.util.HashMap;
public class PutVectorsExample {
public static void main(String[] args) {
String endpoint = "Provide your endpoint";
String region = "Provide your region";
String accessKey = System.getenv("TOS_ACCESS_KEY");
String secretKey = System.getenv("TOS_SECRET_KEY");
String accountID = System.getenv("TOS_ACCOUNT_ID");
String vectorBucketName = "Provide your vector bucket name";
String indexName = "testindex";
TOSVectors client = new TOSVectorsClientBuilder().build(region, endpoint, accessKey, secretKey);
List<Vector> vectors = new ArrayList<>();
float[] vectorData1 = new float[128];
for (int i = 0; i < 128; i++) {
vectorData1[i] = (float) Math.random();
Map<String, Object> metadata1 = new HashMap<>();
metadata1.put("category", "electronics");
metadata1.put("timestamp", System.currentTimeMillis());
metadata1.put("source", "user-upload");
Vector vector1 = Vector.builder()
.key("product-vector-001")
.data(VectorData.builder().float32(vectorData1).build())
float[] vectorData2 = new float[128];
for (int i = 0; i < 128; i++) {
vectorData2[i] = (float) Math.random();
Map<String, Object> metadata2 = new HashMap<>();
metadata2.put("category", "clothing");
metadata2.put("timestamp", System.currentTimeMillis());
metadata2.put("source", "batch-import");
Vector vector2 = Vector.builder()
.key("product-vector-002")
.data(VectorData.builder().float32(vectorData2).build())
float[] vectorData3 = new float[128];
for (int i = 0; i < 128; i++) {
vectorData3[i] = (float) Math.random();
Map<String, Object> metadata3 = new HashMap<>();
metadata3.put("category", "books");
metadata3.put("timestamp", System.currentTimeMillis());
metadata3.put("source", "manual-entry");
Vector vector3 = Vector.builder()
.key("product-vector-003")
.data(VectorData.builder().float32(vectorData3).build())
PutVectorsInput input = PutVectorsInput.builder()
.vectorBucketName(vectorBucketName)
PutVectorsOutput output = client.putVectors(input);
System.out.println("批量插入向量成功");
System.out.println("RequestID: " + output.getRequestInfo().getRequestId());
System.out.println("StatusCode: " + output.getRequestInfo().getStatusCode());
System.out.println("成功插入向量数量: " + vectors.size());
for (int i = 0; i < vectors.size(); i++) {
Vector vector = vectors.get(i);
System.out.println("向量 " + (i + 1) + ":");
System.out.println(" Key: " + vector.getKey());
System.out.println(" Dimension: " + vector.getData().getFloat32().length);
System.out.println(" Metadata: " + vector.getMetadata());
} catch (TosClientException e) {
System.out.println("操作失败");
System.out.println("Message: " + e.getMessage());
if (e.getCause() != null) {
e.getCause().printStackTrace();
} catch (TosServerException e) {
System.out.println("操作失败");
System.out.println("StatusCode: " + e.getStatusCode());
System.out.println("Code: " + e.getCode());
System.out.println("Message: " + e.getMessage());
System.out.println("RequestID: " + e.getRequestID());
System.out.println("操作失败");
System.out.println("unexpected exception, message: " + t.getMessage());