Mirror/ollama4j
1
0
Fork 1
mirror of https://github.com/amithkoujalgi/ollama4j.git synced 2025-05-15 20:07:10 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #3 from reckart/bugfix/2-Certain-requests-fail-with-a-400-Bad-Request

Browse Source 
Thanks for the fix!
This commit is contained in:
Amith Koujalgi 2023-11-12 23:21:42 +05:30 committed by GitHub
commit dd911029d6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,8 +1,16 @@
package io.github.amithkoujalgi.ollama4j.core; package io.github.amithkoujalgi.ollama4j.core;
import com.fasterxml.jackson.databind.ObjectMapper; import java.io.BufferedReader;
import io.github.amithkoujalgi.ollama4j.core.exceptions.OllamaBaseException; import java.io.DataOutputStream;
import io.github.amithkoujalgi.ollama4j.core.models.*; import java.io.IOException;
import java.io.InputStreamReader;
import java.io.OutputStream;
import java.net.HttpURLConnection;
import java.net.URL;
import java.nio.charset.StandardCharsets;
import java.util.List;
import java.util.stream.Collectors;
import org.apache.hc.client5.http.classic.methods.HttpDelete; import org.apache.hc.client5.http.classic.methods.HttpDelete;
import org.apache.hc.client5.http.classic.methods.HttpGet; import org.apache.hc.client5.http.classic.methods.HttpGet;
import org.apache.hc.client5.http.classic.methods.HttpPost; import org.apache.hc.client5.http.classic.methods.HttpPost;
@ -16,14 +24,15 @@ import org.apache.hc.core5.http.io.entity.StringEntity;
import org.slf4j.Logger; import org.slf4j.Logger;
import org.slf4j.LoggerFactory; import org.slf4j.LoggerFactory;
import java.io.BufferedReader; import com.fasterxml.jackson.databind.ObjectMapper;
import java.io.DataOutputStream;
import java.io.IOException; import io.github.amithkoujalgi.ollama4j.core.exceptions.OllamaBaseException;
import java.io.InputStreamReader; import io.github.amithkoujalgi.ollama4j.core.models.ListModelsResponse;
import java.net.HttpURLConnection; import io.github.amithkoujalgi.ollama4j.core.models.Model;
import java.net.URL; import io.github.amithkoujalgi.ollama4j.core.models.ModelDetail;
import java.util.List; import io.github.amithkoujalgi.ollama4j.core.models.OllamaAsyncResultCallback;
import java.util.stream.Collectors; import io.github.amithkoujalgi.ollama4j.core.models.OllamaRequestModel;
import io.github.amithkoujalgi.ollama4j.core.models.OllamaResponseModel;
/** /**
* The base Ollama API class. * The base Ollama API class.
@ -242,8 +251,8 @@ public class OllamaAPI {
con.setRequestMethod("POST"); con.setRequestMethod("POST");
con.setDoOutput(true); con.setDoOutput(true);
con.setRequestProperty("Content-Type", "application/json"); con.setRequestProperty("Content-Type", "application/json");
try (DataOutputStream wr = new DataOutputStream(con.getOutputStream())) { try (OutputStream out = con.getOutputStream()) {
wr.writeBytes(ollamaRequestModel.toString()); out.write(ollamaRequestModel.toString().getBytes(StandardCharsets.UTF_8));
} }
int responseCode = con.getResponseCode(); int responseCode = con.getResponseCode();
if (responseCode == HttpURLConnection.HTTP_OK) { if (responseCode == HttpURLConnection.HTTP_OK) {