Pārlūkot izejas kodu

Add log statements

atsachlaris 2 dienas atpakaļ
vecāks
revīzija
0742a3c40d

+ 4 - 0
src/main/java/es/uv/saic/llm/GroqProxy.java

@@ -2,6 +2,7 @@ package es.uv.saic.llm;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.openai.OpenAiChatOptions;
 import org.springframework.beans.factory.annotation.Qualifier;
@@ -11,6 +12,7 @@ import org.springframework.stereotype.Component;
 import static es.uv.saic.service.SystemPrompt.SCORE_AND_COMMENT_ANALYSIS_PROMPT;
 
 @Component
+@Slf4j
 public class GroqProxy implements AiProxy {
 
     private final ChatClient chatClient;
@@ -29,6 +31,8 @@ public class GroqProxy implements AiProxy {
 
     @SneakyThrows
     public String calculateScoreAndProduceComments(String asCsv, String model) {
+        log.info("Calculating using Groq Proxy with model: {}", model);
+
         ChatClient.ChatClientRequestSpec prompt = chatClient.prompt();
 
         if (StringUtils.hasText(model)) {

+ 3 - 0
src/main/java/es/uv/saic/llm/LocalLlmProxy.java

@@ -2,6 +2,7 @@ package es.uv.saic.llm;
 
 import com.fasterxml.jackson.databind.ObjectMapper;
 import lombok.SneakyThrows;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.ai.openai.OpenAiChatOptions;
 import org.springframework.beans.factory.annotation.Qualifier;
@@ -11,6 +12,7 @@ import org.springframework.util.StringUtils;
 import static es.uv.saic.service.SystemPrompt.SCORE_AND_COMMENT_ANALYSIS_PROMPT;
 
 @Component
+@Slf4j
 public class LocalLlmProxy implements AiProxy {
 
     private final ChatClient localChatClient;
@@ -29,6 +31,7 @@ public class LocalLlmProxy implements AiProxy {
 
     @SneakyThrows
     public String calculateScoreAndProduceComments(String asCsv, String model) {
+        log.info("Calculating using Local LLM Proxy with model: {}", model);
         ChatClient.ChatClientRequestSpec prompt = localChatClient.prompt();
 
         if (StringUtils.hasText(model)) {

+ 13 - 0
src/main/java/es/uv/saic/web/EnhancementController.java

@@ -3,6 +3,7 @@ package es.uv.saic.web;
 import es.uv.saic.extractor.ExtractionRequest;
 import es.uv.saic.service.EnhancementService;
 import lombok.RequiredArgsConstructor;
+import lombok.extern.slf4j.Slf4j;
 import org.springframework.http.MediaType;
 import org.springframework.web.bind.annotation.PostMapping;
 import org.springframework.web.bind.annotation.RequestBody;
@@ -20,6 +21,7 @@ import static org.springframework.http.HttpStatus.BAD_REQUEST;
 @RestController
 @RequestMapping("/enhancements")
 @RequiredArgsConstructor
+@Slf4j
 public class EnhancementController {
 
     private final EnhancementService enhancementService;
@@ -28,10 +30,13 @@ public class EnhancementController {
     public String singleStepAnalysis(@RequestBody String html,
                                      @RequestParam(value = "provider", required = false) String provider,
                                      @RequestParam(value = "model", required = false) String model) {
+        log.info("====== EnhancementController.singleStepAnalysis start ======");
         try {
             return enhancementService.calculateScoreAndProduceCommentsWithSingleCall(ExtractionRequest.fromHtml(html), provider, model);
         } catch (IllegalArgumentException e) {
             throw new ResponseStatusException(BAD_REQUEST, e.getMessage(), e);
+        } finally {
+            log.info("====== EnhancementController.singleStepAnalysis end ======");
         }
     }
 
@@ -39,6 +44,8 @@ public class EnhancementController {
     public String singleStepAnalysisFile(@RequestPart("file") MultipartFile file,
                                          @RequestParam(value = "provider", required = false) String provider,
                                          @RequestParam(value = "model", required = false) String model) {
+        log.info("====== EnhancementController.singleStepAnalysisFile start ======");
+
         if (file.isEmpty()) {
             throw new ResponseStatusException(BAD_REQUEST, "Uploaded file is empty");
         }
@@ -53,6 +60,8 @@ public class EnhancementController {
             throw new ResponseStatusException(BAD_REQUEST, e.getMessage(), e);
         } catch (Exception e) {
             throw new ResponseStatusException(BAD_REQUEST, "Unable to read uploaded file", e);
+        } finally {
+            log.info("====== EnhancementController.singleStepAnalysisFile end ======");
         }
     }
 
@@ -60,6 +69,8 @@ public class EnhancementController {
     public String singleStepAnalysisCsvFile(@RequestPart("file") MultipartFile file,
                                             @RequestParam(value = "provider", required = false) String provider,
                                             @RequestParam(value = "model", required = false) String model) {
+        log.info("====== EnhancementController.singleStepAnalysisCsvFile start ======");
+
         if (file.isEmpty()) {
             throw new ResponseStatusException(BAD_REQUEST, "Uploaded file is empty");
         }
@@ -71,6 +82,8 @@ public class EnhancementController {
             throw new ResponseStatusException(BAD_REQUEST, e.getMessage(), e);
         } catch (Exception e) {
             throw new ResponseStatusException(BAD_REQUEST, "Unable to read uploaded CSV file", e);
+        } finally {
+            log.info("====== EnhancementController.singleStepAnalysisCsvFile end ======");
         }
     }