Implementing a Custom LLM in ADK Java

public class OllamaLLM implements BaseLLM {
  private final HttpClient http;
  private final String modelName;
  private final String endpoint;

  public OllamaLLM(String endpoint, String modelName) {
    this.endpoint = endpoint;
    this.modelName = modelName;
    this.http = HttpClient.newHttpClient();
  }
}

public ModelResponse generate(GenerateRequest req) {
  String body = toOllamaFormat(req);
  HttpResponse resp = http.send(
      HttpRequest.newBuilder(URI.create(endpoint + "/api/chat"))
          .POST(HttpRequest.BodyPublishers.ofString(body))
          .build(),
      HttpResponse.BodyHandlers.ofString());
  return fromOllamaFormat(resp.body());
}