Se voce programa em Java e nunca 'brincou' com o GeminiAI, este artigo vai ser um ótimo guia introdutório, aqui vou apresentar de forma bem simples como enviar requisições ao Gemini e retornar um JSON, como uma API Rest. 🐱👤
O que estou usando? 👀
- Java 17
- Intellij Community 2024.1.1
- Postman
- Chave de acesso ao GeminiAI
VAMOS COMECAR 🚀
Inicie um projeto simples pelo spring initializer, e inclua as seguintes dependências no seu POM
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-web</artifactId>
</dependency>
<dependency>
<groupId>org.projectlombok</groupId>
<artifactId>lombok</artifactId>
<version>1.18.36</version>
<scope>provided</scope>
</dependency>
<dependency>
<groupId>com.fasterxml.jackson.core</groupId>
<artifactId>jackson-databind</artifactId>
<version>2.18.2</version>
</dependency>
Essas dependencias vao possibilitar o uso do Lombok, RestTemplate e ObjectMapper.
Lombok: para evitarmos códigos repetitivos (famosos boilerplates) e para melhorar a legibilidade do nosso código
RestTemplate: para realizar a requisição http a API do GeminiAI
ObjectMapper: para convertermos o retorno da api do Gemini em JSON
CONFIGURANDO O RESTTEMPLATE
Vamos configurar o RestTemplate no nosso projeto Java, para isso criamos uma classe com a anotação @Configuration e com o Bean para defini-lo:
@Configuration
public class RestTemplateConfig {
@Bean
public RestTemplate restTemplate(RestTemplateBuilder builder) {
return builder.build();
}
}
CRIANDO O SERVICE
Vamos criar uma classe service para realizar a comunicação com o GeminiAI, esta classe será responsável por toda comunicação e tratamento da resposta do Gemini, e deverá ficar da seguinte forma:
@Service
public class TalkService {
private final RestTemplate restTemplate;
private final ObjectMapper objectMapper;
@Value("${gemini.ai.api.url}")
private String geminiApiUrl;
@Value("${gemini.ai.api.key}")
private String geminiApiKey;
public TalkService(RestTemplate restTemplate, ObjectMapper objectMapper) {
this.restTemplate = restTemplate;
this.objectMapper = objectMapper;
}
public String callGeminiAI(TalkRequest input) {
String url = geminiApiUrl geminiApiKey;
GeminiRequest request = new GeminiRequest();
GeminiRequest.Content content = new GeminiRequest.Content();
GeminiRequest.Part part = new GeminiRequest.Part();
part.setText(input.getChat());
content.setParts(Collections.singletonList(part));
request.setContents(Collections.singletonList(content));
HttpHeaders headers = new HttpHeaders();
headers.setContentType(MediaType.APPLICATION_JSON);
HttpEntity<GeminiRequest> entity = new HttpEntity<>(request, headers);
ResponseEntity<String> response = restTemplate.exchange(url, HttpMethod.POST, entity, String.class);
try {
GeminiResponse geminiAIResponse = objectMapper.readValue(response.getBody(), GeminiResponse.class);
if (geminiAIResponse.getCandidates() != null && !geminiAIResponse.getCandidates().isEmpty()) {
GeminiResponse.Candidate candidate = geminiAIResponse.getCandidates().get(0);
if (candidate.getContent() != null && candidate.getContent().getParts() != null && !candidate.getContent().getParts().isEmpty()) {
return candidate.getContent().getParts().get(0).getText();
}
}
} catch (Exception e) {
e.printStackTrace();
}
return "Falha ao processar resposta da API";
}
}
Note que nesta classe estamos usando os POJO's GeminiRequest e GeminiResponse, segue abaixo o código para cria-los
GeminiRequest
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
public class GeminiRequest {
private List<Content> contents;
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
public static class Content {
private List<Part> parts;
}
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
public static class Part {
private String text;
}
}
GeminiResponse
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public class GeminiResponse {
private List<Candidate> candidates;
private UsageMetadata usageMetadata;
private String modelVersion;
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public static class Candidate {
private Content content;
private String finishReason;
private double avgLogprobs;
}
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public static class Content {
private List<Part> parts;
}
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public static class Part {
private String text;
}
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
@JsonIgnoreProperties(ignoreUnknown = true)
public static class UsageMetadata {
private int promptTokenCount;
private int candidatesTokenCount;
private int totalTokenCount;
}
}
CRIANDO UM CONTROLLER
Vamos agora criar um controller para escutar uma requisição Rest e fazer o processamento através do nosso Service
@RestController
@RequestMapping("v1")
public class TalkController {
private final TalkService talkService;
@Autowired
public TalkController(final TalkService talkService) {
this.talkService = talkService;
}
@PostMapping("/chat-gemini")
public TalkResponse talk(@RequestBody TalkRequest talkRequest) {
TalkResponse response = new TalkResponse();
response.setResponse(talkService.callGeminiAI(talkRequest));
return response;
}
}
Nosso controller também possui POJO's, confira o código deles abaixo
TalkRequest
@Getter
@Setter
@AllArgsConstructor
public class TalkRequest {
private String chat;
}
TalkResponse
@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
public class TalkResponse {
private String response;
}
CONFIGURANDO VARIAVEIS NOS PROPERTIES
Voce precisará informar o endpoint de acesso ao GeminiAI e também a sua chave de acesso. Essas informações armazenei no arquivo properties, dado que estamos falando de um simples teste. Confira o arquivo properties com as variáveis necessárias
spring.application.name=NOME_DA_SUA_APLICACAO
gemini.ai.api.url=https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-flash:generateContent?key=
gemini.ai.api.key=SUA_CHAVE_DE_ACESSO
BORA TESTAR 🐱💻
Já temos uma comunicação com o GeminiAI, agora podemos testar nossa aplicação usando o postman, para isso inicie a sua aplicação no Intellij e execute a requisição no postman conforme a imagem abaixo:
CONCLUSÃO ✔
O intuito deste artigo foi introduzir programadores Java a conexao do GeminiAI com uma aplicacao Java, criando infinitas novas possibilidades de uso. Espero que tenham gostado, até a próxima! 😊
Top comments (0)