Mirror/ollama4j
1
0
Fork 1
mirror of https://github.com/amithkoujalgi/ollama4j.git synced 2025-11-02 17:40:41 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Added generateEmbeddings API

Browse Source
This commit is contained in:
Amith Koujalgi
2023-11-12 21:14:54 +05:30
parent e78609727e
commit 7933154462
3 changed files with 21 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/io/github/amithkoujalgi/ollama4j/core/OllamaAPI.java
View File

@@ -293,12 +293,12 @@ public class OllamaAPI {
*
* @param model name of model to generate embeddings from
* @param prompt text to generate embeddings for
* @return embeddings as double[]
* @return embeddings
* @throws IOException
* @throws ParseException
* @throws OllamaBaseException
*/
public double[] generateEmbeddings(String model, String prompt) throws IOException, ParseException, OllamaBaseException {
public List<Double> generateEmbeddings(String model, String prompt) throws IOException, ParseException, OllamaBaseException {
String url = this.host + "/api/embeddings";
String jsonData = String.format("{\"model\": \"%s\", \"prompt\": \"%s\"}", model, prompt);
final HttpPost httpPost = new HttpPost(url);

8
src/main/java/io/github/amithkoujalgi/ollama4j/core/models/EmbeddingResponse.java
View File

@@ -2,18 +2,20 @@ package io.github.amithkoujalgi.ollama4j.core.models;
import com.fasterxml.jackson.annotation.JsonProperty;
import java.util.List;
public class EmbeddingResponse {
@JsonProperty("embedding")
private double[] embedding;
private List<Double> embedding;
public EmbeddingResponse() {
}
public double[] getEmbedding() {
public List<Double> getEmbedding() {
return embedding;
}
public void setEmbedding(double[] embedding) {
public void setEmbedding(List<Double> embedding) {
this.embedding = embedding;
}
}