SpringBoot整合零一万物模型API进行多轮对话

前期准备工作

零一万物官网:https://www.01.ai/cn

零一万物大模型开放平台:https://platform.lingyiwanwu.com/

选择理由

性价比高很高,模型整体不错,新用户送60元,非常适合研究学习。

开发

只提供了http接口和python的sdk。

所以Java如果想使用,可以采用http接口调用。

java 复制代码
package com.hao.controller;

import cn.hutool.json.JSONUtil;
import lombok.Data;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.http.HttpEntity;
import org.springframework.http.HttpHeaders;
import org.springframework.http.MediaType;
import org.springframework.http.ResponseEntity;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
import org.springframework.web.client.RestTemplate;

import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Scanner;

@RestController
public class TestController {
    @Autowired
    RestTemplate restTemplate;

    @GetMapping("/admin/test")
    public Object test(String[] args) {
        HttpHeaders headers = new HttpHeaders();
        headers.setContentType(MediaType.APPLICATION_JSON);
        headers.setBearerAuth("{your-own-key}");
        Input input = new Input();
        input.setModel("yi-34b-chat-0205");
        input.setTemperature(0.7);

        List<Message> messages = new ArrayList<>();

        Scanner sc = new Scanner(System.in);
        while (true) {
            System.out.println("You:请输入...(exit退出)");
            String question = sc.next();
            if ("exit".equals(question)) {
                break;
            }
            if (messages.size() == 7) {
                messages.remove(0);
            }
            Message message = new Message();
            message.setContent(question);
            message.setRole("user");
            messages.add(message);
            input.setMessages(messages);

            HttpEntity<Input> httpEntity = new HttpEntity<>(input,headers);
            ResponseEntity<Object> obj = restTemplate.postForEntity("https://api.lingyiwanwu.com/v1/chat/completions", httpEntity, Object.class);

            // 添加assistant返回到messages列表,user/assistant消息必须交替出现
            if (messages.size() == 7) {
                messages.remove(0);
            }
//            System.out.println(obj.getBody());
            String content = JSONUtil.parseObj(obj.getBody()).getByPath("choices.0.message.content", String.class);
            Message message1 = new Message();
            message1.setContent(content);
            message1.setRole("assistant");
            messages.add(message1);
            System.out.println("Assistant: " + content);

        }

        return "OK";
    }
}

@Data
class Input{
    private String model;
    private Double temperature;
    private List<Message> messages;
}

@Data
class Message{
    private String content;
    private String role;
}

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

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

测试结果

shell 复制代码
You:请输入...(exit退出)
西红柿炒鸡蛋的做饭?
Assistant: 西红柿炒鸡蛋是一道简单而美味的家常菜,下面是它的基本做法:

材料:
- 西红柿 2-3个
- 鸡蛋 2-3个
- 大蒜 2-3瓣,切末
- 盐 适量
- 白糖 适量(可选)
- 植物油 适量

做法:
1. 西红柿洗净,切成小块或片状。如果西红柿比较硬,可以先用沸水烫一下,使其更容易剥皮和切块。
2. 将鸡蛋打入碗中,加入少许盐,用筷子或打蛋器打散,使蛋液均匀。
3. 热锅凉油,油温升高后倒入打好的鸡蛋液,待底部凝固后,用铲子轻轻推动,使鸡蛋液均匀受热,形成蛋块,然后盛出备用。
4. 锅中再加入少许油,放入大蒜末炒香。
5. 加入切好的西红柿块,翻炒均匀。
6. 根据个人口味,加入适量盐和白糖调味(如果西红柿比较酸,可以适当加一点糖来中和)。
7. 西红柿炒软后,加入之前炒好的鸡蛋块,快速翻炒均匀,使鸡蛋块吸收西红柿的汁液。
8. 炒至西红柿完全软烂,鸡蛋块均匀裹上西红柿汁即可出锅。

这道菜酸甜适口,色彩鲜艳,非常开胃。你可以根据个人口味调整西红柿和鸡蛋的比例,以及糖和盐的用量。如果你喜欢口感更浓郁的,可以在炒西红柿的时候加一点番茄酱。
You:请输入...(exit退出)
不放西红柿可以吗?
Assistant: 当然可以。不放西红柿的话,这道菜就变成了单纯的炒鸡蛋。以下是基本的炒鸡蛋做法:

材料:
- 鸡蛋 2-3个
- 盐 适量
- 植物油 适量

做法:
1. 将鸡蛋打入碗中,加入少许盐,用筷子或打蛋器打散,使蛋液均匀。
2. 热锅凉油,油温升高后倒入打好的鸡蛋液,待底部凝固后,用铲子轻轻推动,使鸡蛋液均匀受热,形成蛋块。
3. 待鸡蛋块成型后,用铲子继续翻炒,使鸡蛋块大小均匀。
4. 炒至鸡蛋块完全熟透,表面金黄即可出锅。

如果你想要更多的风味,可以在炒鸡蛋的时候加入一些其他蔬菜或者调料,比如葱花、火腿丁、辣椒等,增加口感和风味。
You:请输入...(exit退出)
一份可以卖多少钱?
Assistant: 西红柿炒鸡蛋的价格会根据所在地区、餐厅档次、原料质量和份量等因素而有所不同。在中国的大部分地区,如果是在街边小店或者快餐店,这样一份简单的西红柿炒鸡蛋价格可能在10-20元人民币左右。在中高档餐厅,同样的菜品价格可能会翻倍甚至更高。

在其他国家和地区,价格也会有所不同。例如,在美国的中餐馆,一份西红柿炒鸡蛋的价格可能在8-15美元之间。在欧洲或者澳大利亚等地,价格可能会更高。

需要注意的是,这只是一个大致的价格范围,具体价格还需根据当地市场情况来确定。此外,如果是在家中自己烹饪,成本会低很多,因为不需要考虑租金、人工、税费等额外开支。
You:请输入...(exit退出)
exit

最后,感谢李开复及其团队。

相关推荐
昵称为空C36 分钟前
SpringBoot3 http接口调用新方式RestClient + @HttpExchange像使用Feign一样调用
spring boot·后端
大模型教程2 小时前
8GB显存笔记本能跑多大AI模型?这个计算公式90%的人都不知道!
程序员·llm·agent
大模型教程2 小时前
大模型应用开发到底有多赚钱?看完这5个真实案例,你会惊掉下巴
程序员·llm·agent
AI大模型2 小时前
别乱装!Ollama×DeepSeek×AnythingLLM一键本地AI知识库,快人10倍
程序员·llm·agent
聚客AI4 小时前
🌟大模型为什么产生幻觉?预训练到推理的漏洞全揭秘
人工智能·llm·掘金·日新计划
麦兜*10 小时前
MongoDB Atlas 云数据库实战:从零搭建全球多节点集群
java·数据库·spring boot·mongodb·spring·spring cloud
麦兜*10 小时前
MongoDB 在物联网(IoT)中的应用:海量时序数据处理方案
java·数据库·spring boot·物联网·mongodb·spring
汤姆yu10 小时前
基于springboot的毕业旅游一站式定制系统
spring boot·后端·旅游
302AI10 小时前
302.AI 实战指南丨将任意 API 转为 MCP Server,释放 AI 生产力
llm·api·mcp
计算机毕业设计木哥11 小时前
计算机毕设选题推荐:基于Java+SpringBoot物品租赁管理系统【源码+文档+调试】
java·vue.js·spring boot·mysql·spark·毕业设计·课程设计