Java? ??????? GeminiAI? ??? ? ?? ??? ? ??? ??? ?? ???? ? ????. ???? Rest API?? Gemini? ??? ??? JSON? ???? ??? ?? ??? ???? ???????. ???

?? ??? ???? ???? ?

• ?? 17
• ????? 2024.1.1
• ??? ??
• GeminiAI ??? ?

---

?????

??? ??? ???? ??? ????? ???? 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>

??? ???? ?? Lombok, RestTemplate ? ObjectMapper? ??? ? ????.

??: ???? ??(??? ???)? ???? ?? ???? ??? ??

RestTemplate: GeminiAI API? http ??

ObjectMapper: Gemini API ??? JSON?? ??

?

???? ??

Java ?????? RestTemplate? ??? ?????. ?? ?? @Configuration ??? ?? ???? Bean? ???? ???? ?????.

@Configuration
public class RestTemplateConfig {

    @Bean
    public RestTemplate restTemplate(RestTemplateBuilder builder) {
        return builder.build();
    }
}

?

??? ???

GeminiAI? ???? ?? ??? ???? ??? ?????. ? ???? Gemini? ??? ?? ?? ??? ??? ???? ??? ????.

@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";
    }
}

? ????? POJO? GeminiRequest ? GeminiResponse? ???? ???, ??? ?? ???? ?????

?????

@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;
    }
}

?

???? ??

?? Rest ??? ???? ???? ?? ???? ????? ??? ?????.

@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;
    }
}

?? ?????? POJO? ????. ???? ?? ??? ?????

????

@Getter
@Setter
@AllArgsConstructor
public class TalkRequest {
    private String chat;
}

???

@Getter
@Setter
@AllArgsConstructor
@NoArgsConstructor
public class TalkResponse {
    private String response;
}

?

??? ?? ??

GeminiAI ??? ?????? ??? ?? ??? ???. ??? ??? ???? ?? ????? ???? ? ??? ?? ??? ??????. ??? ??? ??? ?? ??? ?????

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

?

??????????

GeminiAI? ?? ??? ??????? ?? Postman? ???? ??????? ???? ? ????. ?? ?? Intellij?? ??????? ???? ?? ???? ?? Postman?? ??? ??? ? ????.

---

?? ?
? ??? ??? Java ??????? GeminiAI? Java ??????? ???? ??? ??? ?? ???? ???? ??? ???? ??????. ????? ???, ??? ? ???! ?

? ??? GeminiAI? JAVA ??? ?? ?????. ??? ??? PHP ??? ????? ?? ?? ??? ?????!