SpringBoot整合Qwen(通义千问)模型API进行多轮对话

SpringBoot整合Qwen(通义千问)模型API进行多轮对话

码农世界 2024-06-20 后端 86 次浏览 0个评论

前期准备工作

在阿里云DashScope灵积模型服务注册账号,申请api-key。

每个模型会有试用体验,详见:https://dashscope.console.aliyun.com/billing

开发

此处采用SDK模式,即引入依赖的形式进行接口的调用。

pom添加


        
            com.alibaba
            dashscope-sdk-java
            2.13.0
        

java代码

package com.hao;
import com.alibaba.dashscope.aigc.generation.Generation;
import com.alibaba.dashscope.aigc.generation.GenerationParam;
import com.alibaba.dashscope.aigc.generation.GenerationResult;
import com.alibaba.dashscope.common.Message;
import com.alibaba.dashscope.common.Role;
import com.alibaba.dashscope.exception.ApiException;
import com.alibaba.dashscope.exception.InputRequiredException;
import com.alibaba.dashscope.exception.NoApiKeyException;
import com.alibaba.dashscope.utils.JsonUtils;
import java.util.ArrayList;
import java.util.List;
import java.util.Scanner;
public class Main {
    public static void callWithMessage() throws NoApiKeyException, ApiException, InputRequiredException {
        Generation gen = new Generation();
        Scanner sc = new Scanner(System.in);
        List messages = new ArrayList<>();
        Message systemMsg = Message.builder().role(Role.SYSTEM.getValue()).content("You are a helpful assistant.").build();
        messages.add(systemMsg);
        while (true) {
            System.out.println("You:请输入...(exit退出)");
            String input = sc.next();
            if ("exit".equals(input)) {
                break;
            }
            Message userMsg = Message.builder().role(Role.USER.getValue()).content(input).build();
            if (messages.size() == 7) {
                messages.remove(0);
            }
            messages.add(userMsg);
            GenerationParam param = GenerationParam.builder().model("qwen-turbo").messages(messages).apiKey("your-own-key").resultFormat(GenerationParam.ResultFormat.MESSAGE).topP(0.8).build();
            GenerationResult result = gen.call(param);
            System.out.println(result.getOutput().getChoices().get(0).getMessage().getContent());
            // 添加assistant返回到messages列表,user/assistant消息必须交替出现
            if (messages.size() == 7) {
                messages.remove(0);
            }
            messages.add(result.getOutput().getChoices().get(0).getMessage());
        }
    }
    public static void main(String[] args) {
        try {
            callWithMessage();
        } catch (ApiException | NoApiKeyException | InputRequiredException e) {
            System.out.println(e.getMessage());
        }
        System.exit(0);
    }
}

需将模型、key根据实际情况进行替换

设置历史对话保留三轮,超出部分会移除,加入新对话,可根据实际情况调整

效果

You:请输入...(exit退出)
西红柿炒鸡蛋的做法,只罗列步骤即可,100字。
1. 鸡蛋打入碗中搅拌均匀,西红柿切块备用。
2. 热锅凉油,倒入鸡蛋液炒至凝固,盛出备用。
3. 锅中再加油,放入西红柿翻煎出汁。
4. 加入炒好的鸡蛋,加盐、糖调味,快速翻炒均匀。
5. 汤汁收浓后出锅,撒上葱花点缀即可。
You:请输入...(exit退出)
可以不放西红柿吗?
当然可以,如果你不喜欢或者没有西红柿,可以用其他蔬菜替代,比如青椒、洋葱等,步骤大同小异:先炒蔬菜,再倒入打散的鸡蛋,加盐、胡椒粉等调料,炒匀即可。
You:请输入...(exit退出)
按照目前市场行情,一份可卖多少钱?
我是一个AI助手,无法提供实时的市场价格信息,因为菜品的价格会因地区、季节、食材成本、餐厅档次等多种因素而变化。如果你想了解当前市场上的西红柿炒鸡蛋价格,建议你查询当地的餐饮价格或询问餐馆。
You:请输入...(exit退出)
小孩儿可以吃吗
当然可以,西红柿炒鸡蛋是一道营养丰富且口感适宜的家常菜,适合大多数人群,包括小孩儿。不过,如果孩子对某些食材过敏或者有特殊饮食需求,请根据实际情况调整。对于较小的孩子,建议将食材切成更小的块状,烹饪时尽量少油少盐,以保证健康。
You:请输入...(exit退出)
exit
进程已结束,退出代码为 0

转载请注明来自码农世界,本文标题:《SpringBoot整合Qwen(通义千问)模型API进行多轮对话》

百度分享代码,如果开启HTTPS请参考李洋个人博客
每一天,每一秒,你所做的决定都会改变你的人生!

发表评论

快捷回复:

评论列表 (暂无评论,86人围观)参与讨论

还没有评论,来说两句吧...

Top