Springboot-starter-gemini

Usage

• Add dependency to your project

<dependency>
<groupId>com.codingapi</groupId>
<artifactId>springboot-starter-gemini</artifactId>
<version>${last.version}</version>
</dependency>

• Add configuration to your application

gemini.api-key={your gemini api key}
gemini.proxy-host={your http proxy host}
gemini.proxy-port={your http proxy port}
gemini.proxy-enable={enable proxy, default is true}
gemini.version={gemini api version, default is v1beta}

• Use GeminiClient in your code

packagecom.codingapi.gemini.client;
importcom.codingapi.gemini.pojo.Embedding;
importcom.codingapi.gemini.pojo.Generate;
importcom.codingapi.gemini.pojo.Model;
importcom.codingapi.gemini.pojo.Models;
importlombok.SneakyThrows;
importorg.junit.jupiter.api.Test;
importorg.springframework.beans.factory.annotation.Autowired;
importorg.springframework.boot.test.context.SpringBootTest;
importjava.io.File;
importjava.io.IOException;
importjava.util.List;
@SpringBootTestclassGeminiClientTest {
@AutowiredprivateGeminiClientclient;
@Testvoidgenerate() {
Generate.Requestrequest = Generate.creatTextChart("你好，请用中文简体回答我，你如何看待区块链？");
Generate.Responseresponse = client.generate(request);
Stringanswer = Generate.toAnswer(response);
System.out.println(answer);
}
@TestvoidgenerateConfiguration() {
Generate.Requestrequest = Generate.creatTextChart("你好，请用中文简体回答我，你如何看待区块链？");
request.setGenerationConfig(newGenerate.GenerationConfig(List.of("Title"), 1.0f, 1000, 0.8f, 10));
request.addSafetySetting("HARM_CATEGORY_DANGEROUS_CONTENT", "BLOCK_ONLY_HIGH");
Generate.Responseresponse = client.generate(request);
Stringanswer = Generate.toAnswer(response);
System.out.println(answer);
}
@Test@SneakyThrowsvoidstream() {
Generate.Requestrequest = Generate.creatTextChart("你好，请用中文简体回答我，你如何看待区块链？");
client.stream(request, response -> {
Stringanswer = Generate.toAnswer(response);
System.out.println(answer);
});
}
@TestvoidgenerateVision() throwsIOException {
Generate.Requestrequest = Generate.creatImageChart("这是一张什么图片？", newFile("./images/test.png"));
Generate.Responseresponse = client.generate(request);
Stringanswer = Generate.toAnswer(response);
System.out.println(answer);
}
@Testvoidcounts() {
Generate.Requestrequest = Generate.creatTextChart("你好，请用中文简体回答我，你如何看待区块链？");
inttokens = client.counts(request);
System.out.println(tokens);
asserttokens > 0;
}
@Testvoidembedding() {
Embedding.Requestrequest = Embedding.creat("你好，我是小强");
Embedding.Responseresponse = client.embedding(request);
List<Double> answer = Embedding.toAnswer(response);
System.out.println(answer);
}
@Testvoidmodel() {
Modelmodel = client.model("models/gemini-pro");
System.out.println(model);
}
@Testvoidmodels() {
Modelsmodels = client.models();
System.out.println(models);
}
}

Implementation

https://ai.google.dev/tutorials/rest_quickstart