QAI项目的初始化

.gitignore

@@ -0,0 +1,143 @@
+target/
+!.mvn/wrapper/maven-wrapper.jar
+!**/src/main/**/target/
+!**/src/test/**/target/
+
+### IntelliJ IDEA ###
+.idea/modules.xml
+.idea/jarRepositories.xml
+.idea/compiler.xml
+.idea/libraries/
+.idea
+*.iws
+*.iml
+*.ipr
+
+### Eclipse ###
+.apt_generated
+.classpath
+.factorypath
+.project
+.settings
+.springBeans
+.sts4-cache
+
+### NetBeans ###
+/nbproject/private/
+/nbbuild/
+/dist/
+/nbdist/
+/.nb-gradle/
+build/
+!**/src/main/**/build/
+!**/src/test/**/build/
+
+### VS Code ###
+.vscode/
+
+### Mac OS ###
+.DS_Store### Java template
+# Compiled class file
+*.class
+
+# Log file
+*.log
+
+# BlueJ files
+*.ctxt
+
+# Mobile Tools for Java (J2ME)
+.mtj.tmp/
+
+# Package Files #
+*.jar
+*.war
+*.nar
+*.ear
+*.zip
+*.tar.gz
+*.rar
+
+# virtual machine crash logs, see http://www.java.com/en/download/help/error_hotspot.xml
+hs_err_pid*
+replay_pid*
+
+### JetBrains template
+# Covers JetBrains IDEs: IntelliJ, RubyMine, PhpStorm, AppCode, PyCharm, CLion, Android Studio, WebStorm and Rider
+# Reference: https://intellij-support.jetbrains.com/hc/en-us/articles/206544839
+
+# User-specific stuff
+.idea/**/workspace.xml
+.idea/**/tasks.xml
+.idea/**/usage.statistics.xml
+.idea/**/dictionaries
+.idea/**/shelf
+
+# AWS User-specific
+.idea/**/aws.xml
+
+# Generated files
+.idea/**/contentModel.xml
+
+# Sensitive or high-churn files
+.idea/**/dataSources/
+.idea/**/dataSources.ids
+.idea/**/dataSources.local.xml
+.idea/**/sqlDataSources.xml
+.idea/**/dynamic.xml
+.idea/**/uiDesigner.xml
+.idea/**/dbnavigator.xml
+
+# Gradle
+.idea/**/gradle.xml
+.idea/**/libraries
+
+# Gradle and Maven with auto-import
+# When using Gradle or Maven with auto-import, you should exclude module files,
+# since they will be recreated, and may cause churn.  Uncomment if using
+# auto-import.
+# .idea/artifacts
+# .idea/compiler.xml
+# .idea/jarRepositories.xml
+# .idea/modules.xml
+# .idea/*.iml
+# .idea/modules
+# *.iml
+# *.ipr
+
+# CMake
+cmake-build-*/
+
+# Mongo Explorer plugin
+.idea/**/mongoSettings.xml
+
+# File-based project format
+*.iws
+
+# IntelliJ
+out/
+
+# mpeltonen/sbt-idea plugin
+.idea_modules/
+
+# JIRA plugin
+atlassian-ide-plugin.xml
+
+# Cursive Clojure plugin
+.idea/replstate.xml
+
+# SonarLint plugin
+.idea/sonarlint/
+
+# Crashlytics plugin (for Android Studio and IntelliJ)
+com_crashlytics_export_strings.xml
+crashlytics.properties
+crashlytics-build.properties
+fabric.properties
+
+# Editor-based Rest Client
+.idea/httpRequests
+
+# Android studio 3.1+ serialized cache file
+.idea/caches/build_file_checksums.ser
+

pom.xml

@@ -0,0 +1,91 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project xmlns="http://maven.apache.org/POM/4.0.0"
+         xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
+         xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 http://maven.apache.org/xsd/maven-4.0.0.xsd">
+    <modelVersion>4.0.0</modelVersion>
+
+    <parent>
+        <groupId>org.springframework.boot</groupId>
+        <artifactId>spring-boot-starter-parent</artifactId>
+        <version>2.7.14</version>
+        <relativePath/>
+    </parent>
+
+    <groupId>org.example</groupId>
+    <artifactId>QAI</artifactId>
+    <version>1.0-SNAPSHOT</version>
+
+    <dependencies>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-web</artifactId>
+            <version>3.1.2</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-aop</artifactId>
+            <version>3.1.2</version>
+        </dependency>
+        <dependency>
+            <groupId>io.github.gemingjia</groupId>
+            <artifactId>gear-wenxinworkshop-starter</artifactId>
+            <version>0.0.7</version>
+        </dependency>
+        <dependency>
+            <groupId>com.github.xiaoymin</groupId>
+            <artifactId>knife4j-spring-boot-starter</artifactId>
+            <version>3.0.3</version>
+        </dependency>
+        <dependency>
+            <groupId>org.projectlombok</groupId>
+            <artifactId>lombok</artifactId>
+            <version>1.18.28</version>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-starter-test</artifactId>
+            <version>2.7.14</version>
+            <scope>test</scope>
+        </dependency>
+        <!-- https://mvnrepository.com/artifact/junit/junit -->
+        <dependency>
+            <groupId>junit</groupId>
+            <artifactId>junit</artifactId>
+            <version>4.13.2</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-devtools</artifactId>
+            <scope>runtime</scope>
+            <optional>true</optional>
+        </dependency>
+        <dependency>
+            <groupId>org.springframework.boot</groupId>
+            <artifactId>spring-boot-configuration-processor</artifactId>
+            <optional>true</optional>
+        </dependency>
+    </dependencies>
+
+    <properties>
+        <maven.compiler.source>17</maven.compiler.source>
+        <maven.compiler.target>17</maven.compiler.target>
+        <project.build.sourceEncoding>UTF-8</project.build.sourceEncoding>
+    </properties>
+    <build>
+        <plugins>
+            <plugin>
+                <groupId>org.springframework.boot</groupId>
+                <artifactId>spring-boot-maven-plugin</artifactId>
+                <configuration>
+                    <excludes>
+                        <exclude>
+                            <groupId>org.projectlombok</groupId>
+                            <artifactId>lombok</artifactId>
+                        </exclude>
+                    </excludes>
+                </configuration>
+            </plugin>
+        </plugins>
+    </build>
+</project>

src/main/java/cn/honour/QAIApplication.java

@@ -0,0 +1,14 @@
+package cn.honour;
+
+import org.springframework.boot.SpringApplication;
+import org.springframework.boot.autoconfigure.SpringBootApplication;
+
+/**
+ * @author qiaoxiaoxu
+ */
+@SpringBootApplication
+public class QAIApplication {
+    public static void main(String[] args) {
+        SpringApplication.run(QAIApplication.class, args);
+    }
+}

src/main/java/cn/honour/common/BaseResponse.java

@@ -0,0 +1,27 @@
+package cn.honour.common;
+
+import com.gearwenxin.common.ErrorCode;
+import lombok.Data;
+
+import java.io.Serializable;
+
+@Data
+public class BaseResponse<T> implements Serializable {
+    private int code;
+    private T data;
+    private String message;
+
+    public BaseResponse(int code, T data, String message) {
+        this.code = code;
+        this.data = data;
+        this.message = message;
+    }
+
+    public BaseResponse(int code, T data) {
+        this(code, data, "");
+    }
+
+    public BaseResponse(ErrorCode errorCode) {
+        this(errorCode.getCode(), null, errorCode.getMessage());
+    }
+}

src/main/java/cn/honour/common/LogInterceptor.java

@@ -0,0 +1,103 @@
+package cn.honour.common;
+
+import cn.honour.exception.ResultUtils;
+import com.gearwenxin.common.ErrorCode;
+import lombok.extern.slf4j.Slf4j;
+import org.apache.commons.lang3.StringUtils;
+import org.aspectj.lang.ProceedingJoinPoint;
+import org.aspectj.lang.annotation.Around;
+import org.aspectj.lang.annotation.Aspect;
+import org.springframework.stereotype.Component;
+import org.springframework.web.context.request.RequestAttributes;
+import org.springframework.web.context.request.RequestContextHolder;
+import org.springframework.web.context.request.ServletRequestAttributes;
+
+import javax.servlet.http.HttpServletRequest;
+import java.util.*;
+import java.util.concurrent.ConcurrentHashMap;
+import java.util.concurrent.Executors;
+import java.util.concurrent.ScheduledExecutorService;
+import java.util.concurrent.TimeUnit;
+
+/**
+ * 请求响应日志
+ */
+@Aspect
+@Component
+@Slf4j
+public class LogInterceptor {
+    private final Map<String, List<Long>> ipRequestRecords = new ConcurrentHashMap<>();
+    private final Map<String, Integer> ipHourlyRequestCounts = new ConcurrentHashMap<>();
+    private final ScheduledExecutorService scheduler = Executors.newScheduledThreadPool(1);
+
+    public LogInterceptor() {
+        // 设置定时任务，每小时清理一次请求记录
+        scheduler.scheduleAtFixedRate(this::clearExpiredRecords, 1, 1, TimeUnit.HOURS);
+    }
+
+    private Map<String, String> ipListMap = new ConcurrentHashMap<>();
+
+    @Around("execution(* cn.honour.controller.*.*(..))")
+    public Object doInterceptor(ProceedingJoinPoint point) throws Throwable {
+        HttpServletRequest httpServletRequest = getHttpServletRequest();
+        String ip = httpServletRequest.getRemoteAddr();
+        // 获取亲求路径
+        RequestAttributes requestAttributes = RequestContextHolder.currentRequestAttributes();
+        HttpServletRequest httpServletRequest1 = ((ServletRequestAttributes) requestAttributes).getRequest();
+        String requestId = UUID.randomUUID().toString();
+        String uri = httpServletRequest1.getRequestURI();
+
+        Object[] args = point.getArgs();
+        String reqParam = "[" + StringUtils.join(args, ", ") + "]";
+        ipListMap.put("120.216.179.82", "blacklist");
+        // 输出请求日志
+        log.info("request start，id: {}, path: {}, ip: {}, params: {}", requestId, uri,
+                httpServletRequest1.getRemoteHost(), reqParam);
+        // 检查名单
+        String listType = ipListMap.get(ip);
+        if ("blacklist".equals(listType)) {
+            log.warn("IP {} is in the blacklist.", ip);
+            return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "IP已被加入黑名单！");
+        } else if ("whitelist".equals(listType)) {
+            // 在白名单中处理方式，这里示意直接放行
+            return point.proceed();
+        }
+
+        // 检测每小时内请求的次数
+        Integer hourlyRequestCounts = ipHourlyRequestCounts.getOrDefault(ip, 0);
+        if (hourlyRequestCounts >= 30) {
+            log.warn("IP {} has been blacklist due to excessive requests.", ip);
+            return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "次数过多");
+        }
+
+        // 记录请求时间
+        long currentTimeMillis = System.currentTimeMillis();
+        List<Long> requests = ipRequestRecords.computeIfAbsent(ip, k -> new ArrayList<>());
+        requests.add(currentTimeMillis);
+
+        // 清理1分钟内请求次数
+        requests.removeIf(timestap -> timestap < currentTimeMillis - 60000);
+
+        // 检测1分钟内的请求次数
+        if(requests.size() > 8) {
+            log.warn("IP {} has made too many request in the last minute.", ip);
+            return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "次数过快！");
+        }
+
+        // 执行原方法
+        return point.proceed();
+    }
+
+    private void clearExpiredRecords() {
+        long currentTimeMillis = System.currentTimeMillis();
+        ipRequestRecords.forEach((ip, requests) -> {
+            requests.removeIf(timestamp -> timestamp  < currentTimeMillis - 3600000);
+            ipHourlyRequestCounts.put(ip, requests.size());
+        });
+    }
+
+    private HttpServletRequest  getHttpServletRequest() {
+        RequestAttributes requestAttributes = RequestContextHolder.currentRequestAttributes();
+        return ((ServletRequestAttributes)requestAttributes).getRequest();
+    }
+}

src/main/java/cn/honour/config/CorsConfig.java

@@ -0,0 +1,23 @@
+package cn.honour.config;
+
+import org.springframework.beans.factory.annotation.Configurable;
+import org.springframework.web.servlet.config.annotation.CorsRegistry;
+import org.springframework.web.servlet.config.annotation.WebMvcConfigurer;
+
+/**
+ * 全局跨域配置
+ */
+@Configurable
+public class CorsConfig implements WebMvcConfigurer {
+    @Override
+    public void addCorsMappings(CorsRegistry registry) {
+        registry.addMapping("/**")
+                // 允许发送 Cookie
+                .allowCredentials(true)
+                // 放行哪些域名（必须用 patterns，否则 * 会和 allowCredentials 冲突）
+                .allowedOriginPatterns("*")
+                .allowedMethods("GET", "POST", "DELETE", "OPTIONS")
+                .allowedHeaders("*")
+                .exposedHeaders("*");
+    }
+}

src/main/java/cn/honour/config/Knife4jConfig.java

@@ -0,0 +1,30 @@
+package cn.honour.config;
+
+import org.springframework.beans.factory.annotation.Configurable;
+import org.springframework.context.annotation.Profile;
+import springfox.documentation.builders.ApiInfoBuilder;
+import springfox.documentation.builders.PathSelectors;
+import springfox.documentation.builders.RequestHandlerSelectors;
+import springfox.documentation.service.ApiInfo;
+import springfox.documentation.spi.DocumentationType;
+import springfox.documentation.spring.web.plugins.Docket;
+import springfox.documentation.swagger2.annotations.EnableSwagger2;
+
+@Configurable
+@EnableSwagger2
+@Profile("dev")
+public class Knife4jConfig {
+    public Docket defaultApi2() {
+        return new Docket(DocumentationType.SWAGGER_2)
+                .apiInfo(new ApiInfoBuilder()
+                        .title("wenxin")
+                        .description("wenxin")
+                        .version("1.0")
+                        .build())
+                .select()
+                // 指定 Controller 扫描包路径
+                .apis(RequestHandlerSelectors.basePackage("cn.honour.controller"))
+                .paths(PathSelectors.any())
+                .build();
+    }
+}

src/main/java/cn/honour/controller/BloomZ7BController.java

@@ -0,0 +1,75 @@
+package cn.honour.controller;
+
+import cn.honour.common.BaseResponse;
+import cn.honour.exception.ResultUtils;
+import com.gearwenxin.client.BloomZ7BClient;
+import com.gearwenxin.entity.chatmodel.ChatBaseRequest;
+import com.gearwenxin.entity.response.ChatResponse;
+import org.springframework.http.MediaType;
+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;
+import reactor.core.publisher.Flux;
+
+import javax.annotation.Resource;
+
+@RestController
+@RequestMapping("/bloomz7b")
+public class BloomZ7BController {
+    @Resource
+    private BloomZ7BClient bloomz7BClient;
+
+    // 单轮对话
+    @PostMapping("/chat")
+    public BaseResponse<ChatResponse> chatSingle(String msg) {
+        ChatResponse response = bloomz7BClient.chatSingle(msg).block();
+        return ResultUtils.success(response);
+    }
+
+    // 连续对话
+    @PostMapping("/chats")
+    public BaseResponse<ChatResponse> chatCont(String msg, String msgUid) {
+        ChatResponse response = bloomz7BClient.chatCont(msg, msgUid).block();
+        return ResultUtils.success(response);
+    }
+
+    // 流式返回，单次对话
+    @PostMapping(value = "/stream/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> chatSingleStream(String msg) {
+        return bloomz7BClient.chatSingleOfStream(msg);
+    }
+
+    // 流式返回，连续对话
+    @PostMapping(value = "/stream/chats", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> chatContStream(String msg, String msgUid) {
+        return bloomz7BClient.chatContOfStream(msg, msgUid);
+    }
+
+    // 模板对话
+    @PostMapping("/param/chat")
+    public BaseResponse<ChatResponse> pChatSingle(@RequestBody ChatBaseRequest chatBaseRequest) {
+        ChatResponse chatResponse = bloomz7BClient.chatSingle(chatBaseRequest).block();
+        return ResultUtils.success(chatResponse);
+    }
+
+    // 连续对话
+    @PostMapping("/param/chats")
+    public BaseResponse<ChatResponse> pChatCont(@RequestBody ChatBaseRequest chatBaseRequest) {
+        ChatResponse response = bloomz7BClient.chatCont(chatBaseRequest, chatBaseRequest.getUserId()).block();
+        return ResultUtils.success(response);
+    }
+
+    // 流式返回，单次对话
+    @PostMapping(value = "/param/stream/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> pChatSingleStream(@RequestBody ChatBaseRequest chatBaseRequest) {
+        return bloomz7BClient.chatSingleOfStream(chatBaseRequest);
+    }
+
+    // 流式返回，连续对话
+    @PostMapping(value = "/param/stream/chats", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> pChatContStream(@RequestBody ChatBaseRequest chatBaseRequest, String msgUid) {
+        return bloomz7BClient.chatContOfStream(chatBaseRequest, msgUid);
+    }
+
+}

src/main/java/cn/honour/controller/ErnieBotController.java

@@ -0,0 +1,77 @@
+package cn.honour.controller;
+
+import cn.honour.common.BaseResponse;
+import cn.honour.exception.ResultUtils;
+import com.gearwenxin.client.ernie.ErnieBotClient;
+import com.gearwenxin.entity.chatmodel.ChatErnieRequest;
+import com.gearwenxin.entity.response.ChatResponse;
+import org.springframework.http.MediaType;
+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;
+import reactor.core.publisher.Flux;
+
+import javax.annotation.Resource;
+
+/**
+ * 文心一言的相关接口
+ * @author qiaoxiaoxu
+ */
+@RestController
+@RequestMapping("/ernie")
+public class ErnieBotController {
+    @Resource
+    private ErnieBotClient ernieBotClient;
+
+    // 单轮对话
+    @PostMapping("/chat")
+    public BaseResponse<ChatResponse> chatSingle(String msg) {
+        ChatResponse chatResponse = ernieBotClient.chatSingle(msg).block();
+        return ResultUtils.success(chatResponse);
+    }
+    // 连续对话
+    @PostMapping("/chats")
+    public BaseResponse<ChatResponse> chatCont(String msg, String msgUid) {
+        ChatResponse chatResponseMono = ernieBotClient.chatCont(msg, msgUid).block();
+        return ResultUtils.success(chatResponseMono);
+    }
+
+    // 流式返回，单次对话
+    @PostMapping(value = "/stream/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> chatSingleStream(String msg) {
+        return ernieBotClient.chatSingleOfStream(msg);
+    }
+
+    // 流式返回，连续对话
+    @PostMapping(value = "/stream/chats", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> chatContStream(String msg, String msgUid) {
+        return ernieBotClient.chatContOfStream(msg, msgUid);
+    }
+
+    // 模板对话
+    @PostMapping("/param/chat")
+    public BaseResponse<ChatResponse> pChatSingle(@RequestBody ChatErnieRequest chatErnieRequest) {
+        ChatResponse chatResponse = ernieBotClient.chatSingle(chatErnieRequest).block();
+        return ResultUtils.success(chatResponse);
+    }
+
+    // 连续对话
+    @PostMapping("/param/chats")
+    public BaseResponse<ChatResponse> pChatCont(@RequestBody ChatErnieRequest chatErnieRequest) {
+        ChatResponse response = ernieBotClient.chatCont(chatErnieRequest, chatErnieRequest.getUserId()).block();
+        return ResultUtils.success(response);
+    }
+
+    // 流式返回，单次对话
+    @PostMapping(value = "/param/stream/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> pChatSingleStream(@RequestBody ChatErnieRequest chatErnieRequest) {
+        return ernieBotClient.chatSingleOfStream(chatErnieRequest);
+    }
+
+    // 流式返回，连续对话
+    @PostMapping(value = "/param/stream/chats", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> pChatContStream(@RequestBody ChatErnieRequest chatErnieRequest, String msgUid) {
+        return ernieBotClient.chatContOfStream(chatErnieRequest, msgUid);
+    }
+}

src/main/java/cn/honour/controller/ErnieBotTurboController.java

@@ -0,0 +1,77 @@
+package cn.honour.controller;
+
+import cn.honour.common.BaseResponse;
+import cn.honour.exception.ResultUtils;
+import com.gearwenxin.client.ernie.ErnieBotTurboClient;
+import com.gearwenxin.entity.chatmodel.ChatBaseRequest;
+import com.gearwenxin.entity.response.ChatResponse;
+import org.springframework.http.MediaType;
+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;
+import reactor.core.publisher.Flux;
+
+import javax.annotation.Resource;
+
+/**
+ * @author qiaoxiaoxu
+ */
+@RestController
+@RequestMapping("/turbo")
+public class ErnieBotTurboController {
+    @Resource
+    private ErnieBotTurboClient ernieBotTurboClient;
+
+    // 单轮对话
+    @PostMapping("/chat")
+    public BaseResponse<ChatResponse> chatSingle(String msg) {
+        ChatResponse chatResponse = ernieBotTurboClient.chatSingle(msg).block();
+        return ResultUtils.success(chatResponse);
+    }
+
+    // 连续对话
+    @PostMapping("/chats")
+    public BaseResponse<ChatResponse> chatCont(String msg, String msgUid) {
+        ChatResponse response = ernieBotTurboClient.chatCont(msg, msgUid).block();
+        return ResultUtils.success(response);
+    }
+
+    // 流式返回，单次对话
+    @PostMapping(value = "/stream/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> chatSingleStream(String msg) {
+        return ernieBotTurboClient.chatSingleOfStream(msg);
+    }
+
+    // 流式返回，连续对话
+    @PostMapping(value = "/stream/chats", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> chatContStream(String msg, String msgUid) {
+        return ernieBotTurboClient.chatContOfStream(msg, msgUid);
+    }
+
+    // 模板对话
+    @PostMapping("/param/chat")
+    public BaseResponse<ChatResponse> pChatSingle(@RequestBody ChatBaseRequest chatBaseRequest) {
+        ChatResponse chatResponse = ernieBotTurboClient.chatSingle(chatBaseRequest).block();
+        return ResultUtils.success(chatResponse);
+    }
+
+    // 连续对话
+    @PostMapping("/param/chats")
+    public BaseResponse<ChatResponse> pChatCont(@RequestBody ChatBaseRequest chatBaseRequest) {
+        ChatResponse response = ernieBotTurboClient.chatCont(chatBaseRequest, chatBaseRequest.getUserId()).block();
+        return ResultUtils.success(response);
+    }
+
+    // 流式返回，单次对话
+    @PostMapping(value = "/param/stream/chat", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> pChatSingleStream(@RequestBody ChatBaseRequest chatBaseRequest) {
+        return ernieBotTurboClient.chatSingleOfStream(chatBaseRequest);
+    }
+
+    // 流式返回，连续对话
+    @PostMapping(value = "/param/stream/chats", produces = MediaType.TEXT_EVENT_STREAM_VALUE)
+    public Flux<ChatResponse> pChatContStream(@RequestBody ChatBaseRequest chatBaseRequest, String msgUid) {
+        return ernieBotTurboClient.chatContOfStream(chatBaseRequest, msgUid);
+    }
+}

src/main/java/cn/honour/controller/PromptController.java

@@ -0,0 +1,39 @@
+package cn.honour.controller;
+
+import cn.honour.common.BaseResponse;
+import cn.honour.exception.ResultUtils;
+import com.gearwenxin.client.PromptBotClient;
+import com.gearwenxin.entity.chatmodel.ChatPromptRequest;
+import com.gearwenxin.entity.response.PromptResponse;
+import org.springframework.web.bind.annotation.PostMapping;
+import org.springframework.web.bind.annotation.RequestMapping;
+import org.springframework.web.bind.annotation.RestController;
+
+import javax.annotation.Resource;
+import java.util.HashMap;
+import java.util.Map;
+
+@RestController
+@RequestMapping("/propt")
+public class PromptController {
+    @Resource
+    private PromptBotClient promptClient;
+
+    // 模板对话
+    @PostMapping
+    public BaseResponse<PromptResponse> chatSingle(int id) {
+
+        Map<String, String> map = new HashMap<>();
+        map.put("article", "人们都说：“桂林山水甲天下。”我们乘着木船，荡漾在漓江上，来观赏桂林的山水。\n" +
+                "我看见过波澜壮阔的大海，玩赏过水平如镜的西湖，却从没看见过漓江这样的水。漓江的水真静啊，静得让你感觉不到它在流动；漓江的水真清啊，清得可以看见江底的沙石；漓江的水真绿啊，绿得仿佛那是一块无瑕的翡翠。船桨激起的微波扩散出一道道水纹，才让你感觉到船在前进，岸在后移。\n" +
+                "我攀登过峰峦雄伟的泰山，游览过红叶似火的香山，却从没看见过桂林这一带的.山，桂林的山真奇啊，一座座拔地而起，各不相连，像老人，像巨象，像骆驼，奇峰罗列，形态万千；桂林的山真秀啊，像翠绿的屏障，像新生的竹笋，色彩明丽，倒映水中；桂林的山真险啊，危峰兀立，怪石嶙峋，好像一不小心就会栽倒下来。\n" +
+                "这样的山围绕着这样的水，这样的水倒映着这样的山，再加上空中云雾迷蒙，山间绿树红花，江上竹筏小舟，让你感到像是走进了连绵不断的画卷，真是“舟行碧波上，人在画中游”。");
+        map.put("number", "20");
+        ChatPromptRequest promptRequest = new ChatPromptRequest();
+        promptRequest.setId(id);
+        promptRequest.setParamMap(map);
+        PromptResponse promptResponse = promptClient.chatPrompt(promptRequest).block();
+
+        return ResultUtils.success(promptResponse);
+    }
+}

src/main/java/cn/honour/exception/BusinessException.java

@@ -0,0 +1,30 @@
+package cn.honour.exception;
+
+import com.gearwenxin.common.ErrorCode;
+
+/**
+ * 自定义异常类
+ * @author qiaoxiaoxu
+ */
+public class BusinessException extends RuntimeException {
+    private final int code;
+
+    public BusinessException(int code, String message) {
+        super(message);
+        this.code = code;
+    }
+
+    public BusinessException(ErrorCode errorCode){
+        super(errorCode.getMessage());
+        this.code = errorCode.getCode();
+    }
+
+    public BusinessException(ErrorCode errorCode, String message){
+        super(message);
+        this.code = errorCode.getCode();
+    }
+
+    public int getCode() {
+        return code;
+    }
+}

src/main/java/cn/honour/exception/GlobalExceptionHandler.java

@@ -0,0 +1,29 @@
+package cn.honour.exception;
+
+import cn.honour.common.BaseResponse;
+import com.gearwenxin.common.ErrorCode;
+import lombok.extern.slf4j.Slf4j;
+import org.springframework.web.bind.annotation.ExceptionHandler;
+import org.springframework.web.bind.annotation.RestControllerAdvice;
+
+/**
+ * 全局异常处理器
+ * @author qiaoxiaoxu
+ */
+
+@RestControllerAdvice
+@Slf4j
+public class GlobalExceptionHandler {
+
+    @ExceptionHandler(BusinessException.class)
+    public BaseResponse<?> businessExceptionHandler(BusinessException e) {
+        log.error("BusinessException", e);
+        return ResultUtils.error(e.getCode(), e.getMessage());
+    }
+
+    @ExceptionHandler(RuntimeException.class)
+    public BaseResponse<?> runtimeExceptionHandler(RuntimeException e) {
+        log.error("RuntimeException", e);
+        return ResultUtils.error(ErrorCode.SYSTEM_ERROR, "系统错误");
+    }
+}

src/main/java/cn/honour/exception/ResultUtils.java

@@ -0,0 +1,45 @@
+package cn.honour.exception;
+
+import cn.honour.common.BaseResponse;
+import com.gearwenxin.common.ErrorCode;
+
+public class ResultUtils {
+    /**
+     * 成功
+     * @param data
+     * @return
+     * @param <T>
+     */
+    public static <T>BaseResponse<T> success(T data) {
+        return new BaseResponse<>(0, data, "ok");
+    }
+
+    /**
+     * 失败
+     * @param errorCode
+     * @return
+     */
+    public static BaseResponse error(ErrorCode errorCode) {
+        return new BaseResponse<>(errorCode);
+    }
+
+    /**
+     * 失败
+     * @param code
+     * @param message
+     * @return
+     */
+    public static BaseResponse error(int code, String message) {
+        return new BaseResponse<>(code, null, message);
+    }
+
+    /**
+     * 失败
+     * @param errorCode
+     * @param message
+     * @return
+     */
+    public static BaseResponse error(ErrorCode errorCode, String message) {
+        return new BaseResponse<>(errorCode.getCode(), null, message);
+    }
+}

src/main/java/cn/honour/exception/ThrowUtils.java

@@ -0,0 +1,39 @@
+package cn.honour.exception;
+
+import com.gearwenxin.common.ErrorCode;
+
+/**
+ * 抛弃异常工具类
+ * @author qiaoxiaoxu
+ */
+public class ThrowUtils {
+    /**
+     * 条件成立则抛弃异常
+     * @param condition
+     * @param runtimeException
+     */
+    public static void throwIf(boolean condition, RuntimeException runtimeException) {
+        if (condition) {
+            throw runtimeException;
+        }
+    }
+
+    /**
+     * 条件成立则抛异常
+     * @param condition
+     * @param errorCode
+     */
+    public static void throwIf(boolean condition, ErrorCode errorCode) {
+        throwIf(condition, new BusinessException(errorCode));
+    }
+
+    /**
+     * 条件成立则抛异常
+     * @param condition
+     * @param errorCode
+     * @param message
+     */
+    public static void throwIf(boolean condition, ErrorCode errorCode, String message) {
+        throwIf(condition, new BusinessException(errorCode, message));
+    }
+}

src/main/resources/application.yml

@@ -0,0 +1,17 @@
+spring:
+  application:
+    name: qai
+  mvc:
+    pathmatch:
+      matching-strategy: ANT_PATH_MATCHER
+server:
+  port: 7529
+  servlet:
+    context-path: /api
+
+gear:
+  wenxin:
+    api-key: kQnoazc1Ph9ApVsiKLO4N3Hi
+    secret-key: 6XIDixfpvffvsDkm9ZGv4Bip008QXgBP
+    chat-glm2-6b-url: http://asdhkjashdkjas
+    star-coder-url: http://asdhkjashdkjas