Skip to content

Commit afd9160

Browse files
committed
Improved example [skip ci]
1 parent 5b3bef1 commit afd9160

1 file changed

Lines changed: 3 additions & 3 deletions

File tree

src/test/java/com/pgvector/CohereTest.java

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ private List<byte[]> fetchEmbeddings(String[] texts, String inputType, String ap
7575
}
7676
root.put("model", "embed-english-v3.0");
7777
root.put("input_type", inputType);
78-
root.withArray("embedding_types").add("ubinary");
78+
root.withArray("embedding_types").add("binary");
7979
String json = mapper.writeValueAsString(root);
8080

8181
HttpClient client = HttpClient.newHttpClient();
@@ -88,11 +88,11 @@ private List<byte[]> fetchEmbeddings(String[] texts, String inputType, String ap
8888
HttpResponse<String> response = client.send(request, BodyHandlers.ofString());
8989

9090
List<byte[]> embeddings = new ArrayList<>();
91-
for (JsonNode n : mapper.readTree(response.body()).get("embeddings").get("ubinary")) {
91+
for (JsonNode n : mapper.readTree(response.body()).get("embeddings").get("binary")) {
9292
byte[] embedding = new byte[n.size()];
9393
int i = 0;
9494
for (JsonNode v : n) {
95-
embedding[i++] = (byte) v.asDouble();
95+
embedding[i++] = (byte) v.asInt();
9696
}
9797
embeddings.add(embedding);
9898
}

0 commit comments

Comments
 (0)