基层 open ai 添加 chat 聊天

cherishsince
1 parent f4b3bb5a
Showing 8 changed files with 168 additions and 0 deletions
yudao-module-ai/yudao-module-ai-api/pom.xml
yudao-module-ai/yudao-module-ai-api/src/main/java/cn/iocoder/yudao/module/ai/enums/AiModelEnum.java
yudao-module-ai/yudao-module-ai-biz/pom.xml
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/ChatController.java
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/admin/vo/AiChatReqVO.java
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/controller/package-info.java
yudao-module-ai/yudao-module-ai-biz/src/main/resources/application.properties
yudao-module-ai/yudao-module-ai-biz/src/main/resources/application.yaml
@@ -17,4 +17,22 @@
         <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
     </properties>
+    <dependencies>
+        <!-- yudao-common -->
+        <dependency>
+            <groupId>cn.iocoder.boot</groupId>
+            <artifactId>yudao-common</artifactId>
+        </dependency>
+        <!-- lombok -->
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+        </dependency>
+        <!-- 参数校验 -->
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-validation</artifactId>
+            <optional>true</optional>
+        </dependency>
+    </dependencies>
 </project>
 \ No newline at end of file
+package cn.iocoder.yudao.module.ai.enums;
+
+import lombok.Getter;
+
+/**
+ * author: fansili
+ * time: 2024/3/4 12:36
+ */
+@Getter
+public enum AiModelEnum {
+
+    OPEN_AI_GPT_3_5("gpt-3.5-turbo", "GPT3.5"),
+    OPEN_AI_GPT_4("gpt-4-turbo", "GPT4")
+
+    ;
+
+    AiModelEnum(String value, String message) {
+        this.value = value;
+        this.message = message;
+    }
+
+    private String value;
+
+    private String message;
+}
@@ -19,10 +19,38 @@
     <dependencies>
+        <dependency>
+            <groupId>cn.iocoder.boot</groupId>
+            <artifactId>yudao-module-ai-api</artifactId>
+            <version>${revision}</version>
+        </dependency>
+        <dependency>
+            <groupId>cn.iocoder.boot</groupId>
+            <artifactId>yudao-common</artifactId>
+        </dependency>
+
         <!-- 增加 openai -->
         <dependency>
             <groupId>org.springframework.ai</groupId>
             <artifactId>spring-ai-openai-spring-boot-starter</artifactId>
         </dependency>
+
+        <dependency>
+            <groupId>cn.iocoder.boot</groupId>
+            <artifactId>yudao-spring-boot-starter-security</artifactId>
+        </dependency>
+
     </dependencies>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.apache.maven.plugins</groupId>
+                <artifactId>maven-compiler-plugin</artifactId>
+                <configuration>
+                    <source>17</source>
+                    <target>17</target>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
 </project>
 \ No newline at end of file
+package cn.iocoder.yudao.module.ai.controller.admin;
+
+import cn.iocoder.yudao.framework.common.exception.util.ServiceExceptionUtil;
+import cn.iocoder.yudao.framework.common.pojo.CommonResult;
+import cn.iocoder.yudao.module.ai.ErrorCodeConstants;
+import cn.iocoder.yudao.module.ai.controller.admin.vo.AiChatReqVO;
+import cn.iocoder.yudao.module.ai.enums.AiModelEnum;
+import io.swagger.v3.oas.annotations.Operation;
+import io.swagger.v3.oas.annotations.tags.Tag;
+import org.springframework.ai.chat.ChatClient;
+import org.springframework.ai.openai.OpenAiChatClient;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.context.ApplicationContext;
+import org.springframework.validation.annotation.Validated;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestBody;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+/**
+ * AI模块
+ *
+ * author: fansili
+ * time: 2024/3/3 20:28
+ */
+@Tag(name = "AI模块")
+@RestController
+@RequestMapping("/ai-api")
+public class ChatController {
+
+    @Autowired
+    private ApplicationContext applicationContext;
+
+    @PostMapping("/chat")
+    @Operation(summary = "对话聊天", description = "简单的ai聊天")
+    public CommonResult chat(@RequestBody @Validated AiChatReqVO reqVO) {
+        ChatClient chatClient = getChatClient(reqVO.getAiModel());
+        String res;
+        try {
+            res = chatClient.call(reqVO.getInputText());
+        } catch (Exception e) {
+            res = e.getMessage();
+        }
+        return CommonResult.success(res);
+    }
+
+    /**
+     * 根据 ai模型 获取对于的 模型实现类
+     *
+     * @param aiModelEnum
+     * @return
+     */
+    private ChatClient getChatClient(AiModelEnum aiModelEnum) {
+        if (AiModelEnum.OPEN_AI_GPT_3_5 == aiModelEnum) {
+            return applicationContext.getBean(OpenAiChatClient.class);
+        }
+        // AI模型暂不支持
+        throw ServiceExceptionUtil.exception(ErrorCodeConstants.AI_MODULE_NOT_SUPPORTED);
+    }
+}
+package cn.iocoder.yudao.module.ai.controller.admin.vo;
+
+import cn.iocoder.yudao.module.ai.enums.AiModelEnum;
+import io.swagger.v3.oas.annotations.media.Schema;
+import jakarta.validation.constraints.NotNull;
+import lombok.Data;
+
+/**
+ * ai 聊天 req
+ *
+ * author: fansili
+ * time: 2024/3/4 12:33
+ */
+@Schema(description = "用户 App - 上传文件 Request VO")
+@Data
+public class AiChatReqVO {
+
+    @Schema(description = "输入内容", requiredMode = Schema.RequiredMode.REQUIRED)
+    @NotNull(message = "输入内容不能为空")
+    private String inputText;
+
+    @Schema(description = "AI模型", requiredMode = Schema.RequiredMode.REQUIRED)
+    @NotNull(message = "AI模型不能为空")
+    private AiModelEnum aiModel;
+
+}
+/**
+ * author: fansili
+ * time: 2024/3/4 13:08
+ */
+package cn.iocoder.yudao.module.ai.controller;
 \ No newline at end of file
+# open ai
+
+# openAI https://openai.com/
+spring.ai.openai.api-key=${OPEN_AI_KEY}
+spring.ai.openai.chat.options.model=gpt-3.5-turbo
+spring.ai.openai.chat.options.temperature=0.7
 \ No newline at end of file