feat: 集成 Spring AI 实现基础的 AI 写作功能

novel-front/src/main/java/com/java2nb/novel/controller/AuthorController.java

@@ -12,9 +12,11 @@ import com.java2nb.novel.entity.AuthorIncomeDetail;
 import com.java2nb.novel.entity.Book;
 import com.java2nb.novel.service.AuthorService;
 import com.java2nb.novel.service.BookService;
+import io.swagger.v3.oas.annotations.Operation;
 import jakarta.servlet.http.HttpServletRequest;
 import lombok.RequiredArgsConstructor;
 import lombok.extern.slf4j.Slf4j;
+import org.springframework.ai.chat.client.ChatClient;
 import org.springframework.web.bind.annotation.*;
 
 import java.util.Date;
@@ -32,6 +34,8 @@ public class AuthorController extends BaseController{
 
     private final BookService bookService;
 
+    private final ChatClient chatClient;
+
     /**
      * 校验笔名是否存在
      * */
@@ -220,6 +224,54 @@ public class AuthorController extends BaseController{
 
     }
 
+    /**
+     * AI扩写
+     */
+    @PostMapping("ai/expand")
+    public RestResult<String> expandText(@RequestParam("text") String text, @RequestParam("ratio") Double ratio) {
+        String prompt = "请将以下文本扩写为原长度的" + ratio/100 + "倍：" + text;
+        return RestResult.ok(chatClient.prompt()
+                .user(prompt)
+                .call()
+                .content());
+    }
+
+    /**
+     * AI缩写
+     */
+    @PostMapping("ai/condense")
+    public RestResult<String> condenseText(@RequestParam("text") String text, @RequestParam("ratio") Integer ratio) {
+        String prompt = "请将以下文本缩写为原长度的" + 100/ratio + "分之一：" + text;
+        return RestResult.ok(chatClient.prompt()
+                .user(prompt)
+                .call()
+                .content());
+    }
+
+    /**
+     * AI续写
+     */
+    @PostMapping("ai/continue")
+    public RestResult<String> continueText(@RequestParam("text") String text, @RequestParam("length") Integer length) {
+        String prompt = "请续写以下文本，续写长度约为" + length + "字：" + text;
+        return RestResult.ok(chatClient.prompt()
+                .user(prompt)
+                .call()
+                .content());
+    }
+
+    /**
+     * AI润色
+     */
+    @PostMapping("ai/polish")
+    public RestResult<String> polishText(@RequestParam("text") String text) {
+        String prompt = "请润色优化以下文本，保持原意：" + text;
+        return RestResult.ok(chatClient.prompt()
+                .user(prompt)
+                .call()
+                .content());
+    }
+
 
 
 

novel-front/src/main/java/com/java2nb/novel/core/config/AiConfig.java

@@ -0,0 +1,47 @@
+package com.java2nb.novel.core.config;
+
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.ai.chat.client.ChatClient;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+import org.springframework.http.client.SimpleClientHttpRequestFactory;
+import org.springframework.web.client.RestClient;
+
+/**
+ * Ai 相关配置
+ *
+ * @author xiongxiaoyang
+ * @date 2025/2/19
+ */
+@Configuration
+@Slf4j
+public class AiConfig {
+
+    /**
+     * 目的：配置自定义的 RestClientBuilder 对象
+     * <p>
+     * 原因：Spring AI 框架的 ChatClient 内部通过 RestClient（Spring Framework 6 和 Spring Boot 3 中引入） 发起 HTTP REST 请求与远程的大模型服务进行通信，
+     * 如果项目中没有配置自定义的 RestClientBuilder 对象， 那么在 RestClient 的自动配置类 org.springframework.boot.autoconfigure.web.client.RestClientAutoConfiguration
+     * 中配置的 RestClientBuilder 对象会使用 Spring 容器中提供的 HttpMessageConverters， 由于本项目中配置了 spring.jackson.generator.write-numbers-as-strings
+     * = true， 所以 Spring 容器中的 HttpMessageConverters 在 RestClient 发起 HTTP REST 请求转换 Java 对象为 JSON 字符串时会自动将 Number 类型的
+     * Java 对象属性转换为字符串而导致请求参数错误
+     * <p>
+     * 示例："temperature": 0.7 =》"temperature": "0.7"
+     * {"code":20015,"message":"The parameter is invalid. Please check again.","data":null}
+     */
+    @Bean
+    public RestClient.Builder restClientBuilder() {
+        SimpleClientHttpRequestFactory factory = new SimpleClientHttpRequestFactory();
+        // 连接超时时间
+        factory.setConnectTimeout(5000);
+        // 读取超时时间
+        factory.setReadTimeout(60000);
+        return RestClient.builder().requestFactory(factory);
+    }
+
+    @Bean
+    public ChatClient chatClient(ChatClient.Builder chatClientBuilder) {
+        return chatClientBuilder.build();
+    }
+
+}