易达系统中使用智谱AI流程

引入依赖

1
2
3
4
5
<dependency>
    <groupId>cn.bigmodel.openapi</groupId>
    <artifactId>oapi-java-sdk</artifactId>
    <version>release-V4-2.0.2</version>
</dependency>

yml文件中进行配置

  • 首先打开智谱AI开放平台注册账号
    智谱AI注册结束后
  • 将图片中的API Key复制一下填写入下图的yml配置文件里 
    1
    2
    3
    # AI 配置
    ai:
      appKey: xxx

代码中的调用方式

  • 定义一个工具类 
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    @Configuration
    @ConfigurationProperties(prefix = "ai")   //通过该注解引入yml中配置的ai-appKey参数
    @Data
    public class AiConfig {

        private String appKey;

        @Bean
        public ClientV4 getClientV4(){
           return new ClientV4.Builder(appKey).build();  // 将ClientV4注入到Bean中， 调用类就可以通过@Resource获取
        }

    }
  • 代码调用需要按需传递参数，参考这里 
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    public void test() {
         // 构建请求
        ChatCompletionRequest chatCompletionRequest = ChatCompletionRequest.builder()
                .model(Constants.ModelChatGLM4)
                .stream(stream)
                .temperature(temperature)
                .invokeMethod(Constants.invokeMethod)
                .messages(messages)
                .build();
        try {
            ModelApiResponse invokeModelApiResp = clientV4.invokeModelApi(chatCompletionRequest);
            return invokeModelApiResp.getData().getChoices().get(0).toString(); // 该段返回的内容则为智谱AI生成的内容
        } catch (Exception e) {
            e.printStackTrace();
            throw new BusinessException(ErrorCode.SYSTEM_ERROR, e.getMessage());
        }
    }