Compare commits

..

9 Commits

Author SHA1 Message Date
amithkoujalgi
11201bc7c7 [maven-release-plugin] prepare release v1.0.30 2023-12-29 03:57:58 +00:00
Amith Koujalgi
3a8b5257c0 Merge pull request #12 from wyona/main
Basic Auth
2023-12-29 09:26:59 +05:30
Michael Wechner
7481c2ba0e method introduced to get request builder default settings 2023-12-27 08:42:45 +01:00
Michael Wechner
9d336e257c basic auth implemented for generate request 2023-12-26 20:49:54 +01:00
Michael Wechner
2027171cb9 version updated 2023-12-26 18:43:47 +01:00
amithkoujalgi
e06baf0d29 [maven-release-plugin] prepare for next development iteration 2023-12-26 15:01:10 +00:00
amithkoujalgi
5d6a68a5bb [maven-release-plugin] prepare release v1.0.29 2023-12-26 15:01:08 +00:00
Amith Koujalgi
41a5bb70bf Updated docs 2023-12-26 20:30:07 +05:30
amithkoujalgi
c2ec62ba08 [maven-release-plugin] prepare for next development iteration 2023-12-26 14:47:27 +00:00
9 changed files with 136 additions and 20 deletions

View File

@@ -60,7 +60,7 @@ In your Maven project, add this dependency:
<dependency>
<groupId>io.github.amithkoujalgi</groupId>
<artifactId>ollama4j</artifactId>
<version>1.0.20</version>
<version>1.0.29</version>
</dependency>
```

View File

@@ -3,6 +3,6 @@
"position": 10,
"link": {
"type": "generated-index",
"description": "Specs of APIs and their usage."
"description": "Details of APIs to interact with LLMs."
}
}

View File

@@ -0,0 +1,8 @@
{
"label": "APIs - Extras",
"position": 10,
"link": {
"type": "generated-index",
"description": "Details of APIs to handle bunch of extra stuff."
}
}

View File

@@ -0,0 +1,20 @@
---
sidebar_position: 3
---
# Ping
This API lets you check the reachability of Ollama server.
```java
public class Main {
public static void main(String[] args) {
String host = "http://localhost:11434/";
OllamaAPI ollamaAPI = new OllamaAPI(host);
ollamaAPI.ping();
}
}
```

View File

@@ -0,0 +1,21 @@
---
sidebar_position: 2
---
# Set Request Timeout
This API lets you set the request timeout for the Ollama client.
```java
public class Main {
public static void main(String[] args) {
String host = "http://localhost:11434/";
OllamaAPI ollamaAPI = new OllamaAPI(host);
ollamaAPI.setRequestTimeoutSeconds(10);
}
}
```

View File

@@ -0,0 +1,23 @@
---
sidebar_position: 1
---
# Set Verbosity
This API lets you set the verbosity of the Ollama client.
## Try asking a question about the model.
```java
public class Main {
public static void main(String[] args) {
String host = "http://localhost:11434/";
OllamaAPI ollamaAPI = new OllamaAPI(host);
ollamaAPI.setVerbose(true);
}
}
```

View File

@@ -3,6 +3,6 @@
"position": 4,
"link": {
"type": "generated-index",
"description": "Specs of APIs and their usage."
"description": "Details of APIs to manage LLMs."
}
}

View File

@@ -4,7 +4,7 @@
<groupId>io.github.amithkoujalgi</groupId>
<artifactId>ollama4j</artifactId>
<version>1.0.28</version>
<version>1.0.30</version>
<name>Ollama4j</name>
<description>Java library for interacting with Ollama API.</description>
@@ -39,7 +39,7 @@
<connection>scm:git:git@github.com:amithkoujalgi/ollama4j.git</connection>
<developerConnection>scm:git:https://github.com/amithkoujalgi/ollama4j.git</developerConnection>
<url>https://github.com/amithkoujalgi/ollama4j</url>
<tag>v1.0.28</tag>
<tag>v1.0.30</tag>
</scm>
<build>

View File

@@ -37,6 +37,8 @@ public class OllamaAPI {
private final String host;
private long requestTimeoutSeconds = 3;
private boolean verbose = true;
private String username;
private String password;
/**
* Instantiates the Ollama API.
@@ -64,6 +66,14 @@ public class OllamaAPI {
this.verbose = verbose;
}
/**
*
*/
public void setBasicAuth(String username, String password) {
this.username = username;
this.password = password;
}
/**
* API to check the reachability of Ollama server.
*
@@ -306,17 +316,13 @@ public class OllamaAPI {
*/
public List<Double> generateEmbeddings(String model, String prompt)
throws IOException, InterruptedException, OllamaBaseException {
String url = this.host + "/api/embeddings";
URI uri = URI.create(this.host + "/api/embeddings");
String jsonData = new ModelEmbeddingsRequest(model, prompt).toString();
HttpClient httpClient = HttpClient.newHttpClient();
HttpRequest request =
HttpRequest.newBuilder()
.uri(URI.create(url))
HttpRequest.Builder requestBuilder = getRequestBuilderDefault(uri)
.header("Accept", "application/json")
.header("Content-type", "application/json")
.timeout(Duration.ofSeconds(requestTimeoutSeconds))
.POST(HttpRequest.BodyPublishers.ofString(jsonData))
.build();
.POST(HttpRequest.BodyPublishers.ofString(jsonData));
HttpRequest request = requestBuilder.build();
HttpResponse<String> response = httpClient.send(request, HttpResponse.BodyHandlers.ofString());
int statusCode = response.statusCode();
String responseBody = response.body();
@@ -426,14 +432,12 @@ public class OllamaAPI {
long startTime = System.currentTimeMillis();
HttpClient httpClient = HttpClient.newHttpClient();
URI uri = URI.create(this.host + "/api/generate");
HttpRequest request =
HttpRequest.newBuilder(uri)
HttpRequest.Builder requestBuilder = getRequestBuilderDefault(uri)
.POST(
HttpRequest.BodyPublishers.ofString(
Utils.getObjectMapper().writeValueAsString(ollamaRequestModel)))
.header("Content-Type", "application/json")
.timeout(Duration.ofSeconds(requestTimeoutSeconds))
.build();
Utils.getObjectMapper().writeValueAsString(ollamaRequestModel)));
HttpRequest request = requestBuilder.build();
logger.debug("Ask model '" + ollamaRequestModel + "' ...");
HttpResponse<InputStream> response =
httpClient.send(request, HttpResponse.BodyHandlers.ofInputStream());
int statusCode = response.statusCode();
@@ -444,9 +448,15 @@ public class OllamaAPI {
String line;
while ((line = reader.readLine()) != null) {
if (statusCode == 404) {
logger.warn("Status code: 404 (Not Found)");
OllamaErrorResponseModel ollamaResponseModel =
Utils.getObjectMapper().readValue(line, OllamaErrorResponseModel.class);
responseBuffer.append(ollamaResponseModel.getError());
} else if (statusCode == 401) {
logger.warn("Status code: 401 (Unauthorized)");
OllamaErrorResponseModel ollamaResponseModel =
Utils.getObjectMapper().readValue("{\"error\":\"Unauthorized\"}", OllamaErrorResponseModel.class);
responseBuffer.append(ollamaResponseModel.getError());
}else {
OllamaResponseModel ollamaResponseModel =
Utils.getObjectMapper().readValue(line, OllamaResponseModel.class);
@@ -457,10 +467,44 @@ public class OllamaAPI {
}
}
if (statusCode != 200) {
logger.error("Status code " + statusCode + " instead 200");
throw new OllamaBaseException(responseBuffer.toString());
} else {
long endTime = System.currentTimeMillis();
return new OllamaResult(responseBuffer.toString().trim(), endTime - startTime, statusCode);
}
}
/**
*
*/
private HttpRequest.Builder getRequestBuilderDefault(URI uri) {
HttpRequest.Builder requestBuilder =
HttpRequest.newBuilder(uri)
.header("Content-Type", "application/json")
.timeout(Duration.ofSeconds(requestTimeoutSeconds));
if (basicAuthCredentialsSet()) {
requestBuilder.header("Authorization", getBasicAuthHeaderValue());
}
return requestBuilder;
}
/**
* @return basic authentication header value (encoded credentials)
*/
private String getBasicAuthHeaderValue() {
String credentialsToEncode = username + ":" + password;
return "Basic " + Base64.getEncoder().encodeToString(credentialsToEncode.getBytes());
}
/**
* @return true when Basic Auth credentials set
*/
private boolean basicAuthCredentialsSet() {
if (username != null && password != null) {
return true;
} else {
return false;
}
}
}