zyx 2 年之前
父節點
當前提交
d42ceec0a2

+ 0 - 22
src/main/java/com/sxtvs/NetTest.java

@@ -1,22 +0,0 @@
-package com.sxtvs;
-
-import com.fasterxml.jackson.databind.ObjectMapper;
-import com.sxtvs.api.chatgpt.dto.CompletionsParamsDto;
-import lombok.SneakyThrows;
-import org.apache.hc.client5.http.fluent.Request;
-import org.apache.hc.core5.http.ContentType;
-
-public class NetTest {
-    @SneakyThrows
-    public static void main(String[] args) {
-        var objectMapper = new ObjectMapper();
-        var paramsDto = new CompletionsParamsDto("今天天气怎么样");
-        var response = Request.post("https://api.openai.com/v1/completions")
-                .setHeader("Authorization", "Bearer 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")
-                .bodyString(objectMapper.writeValueAsString(paramsDto), ContentType.APPLICATION_JSON)
-                .execute()
-                .returnContent()
-                .toString();
-        System.out.println(response);
-    }
-}

+ 5 - 0
src/main/java/com/sxtvs/api/chatgpt/controller/ChatGptController.java

@@ -5,6 +5,7 @@ import com.sxtvs.api.chatgpt.dto.CompletionsParamsDto;
 import com.sxtvs.api.chatgpt.dto.CompletionsRequestDto;
 import com.sxtvs.api.chatgpt.dto.CompletionsResponseDto;
 import com.sxtvs.api.chatgpt.dto.GptResponse;
+import com.sxtvs.core.sls.AliyunLogger;
 import lombok.SneakyThrows;
 import org.apache.hc.client5.http.fluent.Request;
 import org.apache.hc.core5.http.ContentType;
@@ -21,6 +22,9 @@ public class ChatGptController {
     @Autowired
     private ObjectMapper objectMapper;
 
+    @Autowired
+    private AliyunLogger logger;
+
     @SneakyThrows
     @RequestMapping("completions")
     public CompletionsResponseDto completions(@RequestBody CompletionsRequestDto dto) {
@@ -35,6 +39,7 @@ public class ChatGptController {
         var completionsResponseDto = new CompletionsResponseDto();
         var text = gptResponse.getChoices().stream().map(GptResponse.ChoicesDTO::getText).findFirst().orElse("").trim();
         completionsResponseDto.setResult(text);
+        logger.info("prompt",dto.getPrompt(),"result",text);
         return completionsResponseDto;
     }