Add Perplexity chat client integration with API updates

Integrated Perplexity chat client into the application by implementing its configuration and exposing endpoints in PdfController. Updated related properties and existing test setup to align with the new integration. Simplified prompts in PodcastServiceImpl and enabled configuration properties for Perplexity.

Author: maxiplux

src/main/java/app/quantun/summary/config/ai/AiConfig.java

@@ -4,16 +4,21 @@
 import org.springframework.ai.anthropic.AnthropicChatModel;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.openai.OpenAiChatModel;
+import org.springframework.ai.openai.OpenAiChatOptions;
+import org.springframework.ai.openai.api.OpenAiApi;
 import org.springframework.ai.vertexai.gemini.VertexAiGeminiChatModel;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.context.properties.EnableConfigurationProperties;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
+import org.springframework.context.annotation.Primary;
 
 /**
  * Configuration class for AI chat clients.
  * This class provides beans for OpenAI and Gemini chat clients.
  */
 @Configuration
-//@EnableConfigurationProperties(PerplexityProperties.class)
+@EnableConfigurationProperties(PerplexityProperties.class)
 public class AiConfig {
 
     /**
@@ -24,6 +29,7 @@ public class AiConfig {
      */
     //@Primary
     @Bean
+    @Primary
     ChatClient openAiChatClient(OpenAiChatModel chatModel) {
         return ChatClient.create(chatModel);
     }
@@ -41,4 +47,17 @@ ChatClient anthropicChatClient(AnthropicChatModel chatModel) {
     }
 
 
+    @Bean
+    public ChatClient perplexityChatClient(@Qualifier("perplexityProperties")
+                                           PerplexityProperties perplexityProperties) {
+        OpenAiApi openAiApi = new OpenAiApi(perplexityProperties.getBaseUrl(), perplexityProperties.getApiKey());
+        OpenAiChatOptions chatOptions = OpenAiChatOptions.builder()
+                .model(perplexityProperties.getChat().getModel())
+                .temperature(perplexityProperties.getChat().getTemperature())
+                .build();
+        OpenAiChatModel chatModel = new OpenAiChatModel(openAiApi, chatOptions);
+        return ChatClient.create(chatModel);
+    }
+
+
 }

src/main/java/app/quantun/summary/config/ai/PerplexityProperties.java

@@ -1,10 +1,14 @@
 package app.quantun.summary.config.ai;
 
 import lombok.Data;
+import org.springframework.boot.context.properties.ConfigurationProperties;
+import org.springframework.context.annotation.Configuration;
 
-// Configuration Properties Class
-//@ConfigurationProperties(prefix = "spring.ai.perplexity")
-//@Data
+@Configuration
+@ConfigurationProperties(prefix = "spring.ai.perplexity")
+
+
+@Data
 public class PerplexityProperties {
     private String apiKey;
     private String baseUrl;

src/main/java/app/quantun/summary/controller/PdfController.java

@@ -8,8 +8,13 @@
 import io.swagger.v3.oas.annotations.responses.ApiResponse;
 import io.swagger.v3.oas.annotations.tags.Tag;
 import lombok.RequiredArgsConstructor;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.ai.chat.client.advisor.SimpleLoggerAdvisor;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
 import org.springframework.http.MediaType;
 import org.springframework.http.ResponseEntity;
+import org.springframework.web.bind.annotation.GetMapping;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestParam;
 import org.springframework.web.bind.annotation.RestController;
@@ -21,6 +26,15 @@
 public class PdfController {
 
     private final PdfServices pdfServices;
+    @Autowired
+    @Qualifier("anthropicChatClient")
+    private ChatClient anthropicChatClient;
+    @Autowired
+    @Qualifier("openAiChatClient")
+    private ChatClient openAiChatClient;
+    @Autowired
+    @Qualifier("perplexityChatClient")
+    private ChatClient perplexityChatClient;
 
     /**
      * Handles the upload of a PDF file.
@@ -59,5 +73,20 @@ public TableIndexContent getTableOfContent(String message) {
         return this.pdfServices.getBookTableOfContentPages(message);
     }
 
+    @GetMapping("/anthropic")
+    public String getAnthropicChatClient() {
+        return anthropicChatClient.prompt().user("GIVE A NUMBER BETWEEN ONE TO FIVE").call().content();
+    }
+
+    @GetMapping("/openai")
+    public String getOpenAiChatClient() {
+        return openAiChatClient.prompt().user("GIVE A NUMBER BETWEEN ONE TO FIVE").call().content();
+    }
+
+    @GetMapping("/perplexity")
+    public String getPerplexityChatClient() {
+        return perplexityChatClient.prompt().advisors(new SimpleLoggerAdvisor()).user("GIVE A NUMBER BETWEEN ONE TO FIVE").call().content();
+    }
+
 
 }

src/main/java/app/quantun/summary/service/impl/PodcastServiceImpl.java

@@ -57,12 +57,11 @@ public PodCast createScript(Book book) {
 
             BeanOutputConverter<PodCast> format = new BeanOutputConverter<>(PodCast.class);
 
-            Prompt systemPrompt = systemPromptTemplate.create(Map.of("summary", bookSummaryJson, "format", format.getFormat()));
+            Prompt systemPrompt = systemPromptTemplate.create();
 
             Prompt userPrompt = userPromptTemplate.create(Map.of("summary", bookSummaryJson, "format", format.getFormat()));
 
 
-
             String aiResponse = chatClient.prompt(systemPrompt).advisors(new SimpleLoggerAdvisor()).user(userPrompt.getContents()).call().content();
 
 
@@ -77,7 +76,5 @@ public PodCast createScript(Book book) {
         }
 
 
-
-
     }
 }

src/main/resources/application.properties

@@ -75,8 +75,8 @@ spring.ai.anthropic.chat.options.max-tokens=8192
 spring.ai.perplexity.api-key=${PERPLEXITY_API_KEY}
 spring.ai.perplexity.base-url=https://api.perplexity.ai
 spring.ai.perplexity.chat.completions-path=/chat/completions
-spring.ai.perplexity.chat.model=llama-3.1-sonar-small-128k-online
-spring.ai.perplexity.chat.temperature=0.7
+spring.ai.perplexity.chat.model=sonar-pro
+spring.ai.perplexity.chat.temperature=0.9
 ##############################################################################################
 # File Upload Configuration
 ############################################################
@@ -91,16 +91,16 @@ azure.tts.server-url=${AZURE_TTS_SERVER_URL}
 # Application-Specific File Paths
 ############################################################
 # Path for PDF uploads
-app.config.pdf.upload.path=c:/tmp/summary-ai-pdf/
+app.config.pdf.upload.path=${UPLOAD_CONTENT_PATH}
 # Path where TTS files are saved
-app.config.tts.save.path=c:/tmp/summary-ai-tts/
+app.config.tts.save.path=${UPLOAD_TTS_PATH}
 ############################################################
 # Logging Configuration
 ############################################################
 logging.level.root=INFO
 logging.level.org.springframework.web.reactive.function.client=INFO
-logging.level.org.springframework.http.codec=DEBUG
-logging.level.org.springframework.web.service=DEBUG
+logging.level.org.springframework.http.codec=INFO
+logging.level.org.springframework.web.service=INFO
 # Spring AI advisor logging
 logging.level.org.springframework.ai.chat.client.advisor=DEBUG
 # Optional: Full request/response logging

src/test/java/app/quantun/summary/message/consumer/BookConsumerTest.java

@@ -1,6 +1,8 @@
 package app.quantun.summary.message.consumer;
 
 import app.quantun.summary.repository.BookRepository;
+import app.quantun.summary.service.PodcastService;
+import app.quantun.summary.service.SummaryServices;
 import org.junit.jupiter.api.BeforeEach;
 import org.junit.jupiter.api.Test;
 import org.mockito.Mockito;
@@ -26,23 +28,18 @@ class BookConsumerTest {
 
     @BeforeEach
     public void setUp(EmbeddedKafkaBroker embeddedKafkaBroker) {
-/*
-        // Create a mock SummaryBookRepository to satisfy the dependency of SummaryBookConsumerImpl.
-        SummaryBookRepository repository = mock(SummaryBookRepository.class);
-        summaryBookConsumer = Mockito.spy(new SummaryBookConsumerImpl(repository));
-
         Map<String, Object> producerProps = KafkaTestUtils.producerProps(embeddedKafkaBroker);
-        ProducerFactory<String, Map<String, String>> producerFactory = new DefaultKafkaProducerFactory<>(producerProps);
-        kafkaTemplate = new KafkaTemplate<>(producerFactory);
-*/
 
-// Create a mock SummaryBookRepository to satisfy the dependency of SummaryBookConsumerImpl.
         BookRepository repository = Mockito.mock(BookRepository.class);
-        summaryBookConsumer = Mockito.spy(new SummaryBookConsumerImpl(repository));
+        PodcastService podcastService = Mockito.mock(PodcastService.class);
+
+        SummaryServices summaryServices = Mockito.mock(SummaryServices.class);
+
+        summaryBookConsumer = Mockito.spy(new SummaryBookConsumerImpl(repository, summaryServices, podcastService));
 
-        Map<String, Object> producerProps = KafkaTestUtils.producerProps(embeddedKafkaBroker);
         ProducerFactory<String, Map<String, String>> producerFactory = new DefaultKafkaProducerFactory<>(producerProps);
         KafkaTemplate<String, Map<String, String>> kafkaTemplate = new KafkaTemplate<>(producerFactory);
+
     }
 
     @Test