SpringBoot + 九天大模型(文生图接口)

目录

1、先到九天大模型的官网(LLM Studio)上订阅模型的推理服务,得到APIKey,后期需要使用它生成token才能调用模型的推理服务。

2、在SpringBoot项目里面的pom.xml文件中添加九天大模型的相关依赖,后面会使用到其中的一些函数。

3、 在项目中的实体类里面新建一个JiuTian.java文件,用于存储调用模型功能时必须要设置的参数。

4、 在项目的控制层中新建一个JiuTianController.java文件,编写项目的功能接口。九天大模型文生图功能有两类接口,所以我也写了两个接口。

5、 在服务层新建JiuTianService.interface文件,两个项目接口对应的两个服务接口。

6、在工具层新建JiuTianUtil.java文件,里面编写一些功能函数。

7、在服务层中的实现层中新建JiuTianServiceImpl.java文件,把上述两个接口给实现。这里参考了九天大模型官网的调用示例。

 8、在apifox接口软件中调用项目接口测试。

9、查看图片保存结果。图片挺好看的


1、先到九天大模型的官网(LLM Studio)上订阅模型的推理服务,得到APIKey,后期需要使用它生成token才能调用模型的推理服务。

2、在SpringBoot项目里面的pom.xml文件中添加九天大模型的相关依赖,后面会使用到其中的一些函数。

        <!-- 九天大模型的依赖 --><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-starter-webflux</artifactId></dependency><dependency><groupId>com.nimbusds</groupId><artifactId>nimbus-jose-jwt</artifactId><version>8.16</version></dependency><dependency><groupId>com.fasterxml.jackson.core</groupId><artifactId>jackson-databind</artifactId></dependency><dependency><groupId>com.alibaba.fastjson2</groupId><artifactId>fastjson2</artifactId><version>2.0.26</version></dependency>

3、 在项目中的实体类里面新建一个JiuTian.java文件,用于存储调用模型功能时必须要设置的参数。

public class JiuTian {//调用文生图功能时,给模型的提示语:画一条鱼private String genImgMessage;调用文生图功能时,图像保存的地址private String genImgSavePath;public String getGenImgMessage() {return genImgMessage;}public void setGenImgMessage(String genImgMessage) {this.genImgMessage = genImgMessage;}public String getGenImgSavePath() {return genImgSavePath;}public void setGenImgSavePath(String genImgSavePath) {this.genImgSavePath = genImgSavePath;}
}

4、 在项目的控制层中新建一个JiuTianController.java文件,编写项目的功能接口。九天大模型文生图功能有两类接口,所以我也写了两个接口。

import com.knowledgeBase.entity.JiuTian;
import com.knowledgeBase.permission.PassToken;
import com.knowledgeBase.service.JiuTianService;
import com.nimbusds.jose.JOSEException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.web.bind.annotation.*;@RestController
@RequestMapping("/JiuTian")
public class JiuTianController {@AutowiredJiuTianService jiuTianService;@PassToken@RequestMapping(value = "/gen_img_cha", method = RequestMethod.POST)public String generate_img_Chat(@RequestBody JiuTian jiuTian) throws JOSEException {return jiuTianService.generate_img_Chat(jiuTian);}@PassToken@RequestMapping(value = "/gen_img_com", method = RequestMethod.POST)public String generate_img_Completions(@RequestBody JiuTian jiuTian) throws JOSEException {return jiuTianService.generate_img_Completions(jiuTian);}
}

5、 在服务层新建JiuTianService.interface文件,两个项目接口对应的两个服务接口。

import com.knowledgeBase.entity.JiuTian;
import com.nimbusds.jose.JOSEException;public interface JiuTianService {String generate_img_Chat(JiuTian jiuTian) throws JOSEException;String generate_img_Completions(JiuTian jiuTian) throws JOSEException;
}

6、在工具层新建JiuTianUtil.java文件,里面编写一些功能函数。

import com.alibaba.fastjson2.JSONObject;
import com.nimbusds.jose.*;
import com.nimbusds.jose.crypto.MACSigner;import java.time.LocalDateTime;
import java.util.Date;public class JiuTianUtils {//获取九天大模型的tokenpublic static String generateToken(String apikey,long expSeconds) throws JOSEException {String[] apikeyArr=apikey.split("\\.",2);//kid.secretDate now=new Date();JSONObject payload=new JSONObject();payload.put("api_key",apikeyArr[0]);payload.put("exp",now.getTime()/1000+expSeconds);payload.put("timestamp",now.getTime()/1000);//创建JWS对象JWSObject jwsObject = new JWSObject(new JWSHeader.Builder(JWSAlgorithm.HS256).type(JOSEObjectType.JWT).build(),new Payload(payload.toJSONString()));//签名jwsObject.sign(new MACSigner(apikeyArr[1]));return jwsObject.serialize();}//获取当前时间设置成字符串public static String getNowTime(){LocalDateTime now = LocalDateTime.now();int year = now.getYear();int month = now.getMonthValue();int day = now.getDayOfMonth();int hour = now.getHour();int minute = now.getMinute();int second = now.getSecond();return year + "-" + month + "-" + day + "-" + hour + "-" + minute + "-" + second;}
}

7、在服务层中的实现层中新建JiuTianServiceImpl.java文件,把上述两个接口给实现。这里参考了九天大模型官网的调用示例。

import com.alibaba.fastjson2.JSONObject;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.knowledgeBase.entity.JiuTian;
import com.knowledgeBase.service.JiuTianService;
import com.knowledgeBase.utils.JiuTianUtils;
import com.nimbusds.jose.JOSEException;
import org.springframework.stereotype.Service;
import org.springframework.web.reactive.function.client.ExchangeStrategies;
import org.springframework.web.reactive.function.client.WebClient;
import reactor.core.publisher.Mono;import javax.imageio.ImageIO;
import java.awt.image.BufferedImage;
import java.io.*;
import java.util.*;
import java.util.concurrent.ExecutorService;
import java.util.concurrent.Executors;
import java.util.concurrent.atomic.AtomicReference;@Service
public class JiuTianServiceImpl implements JiuTianService {// 九天大模型的apikeyprivate static String apiKey = "到九天大模型官网订阅服务里面复制apikey";// 九天大模型的token,有效期需要设置为项目用户登录token一样的有效期private static String jwt_token = null;// 相关工具类private static JiuTianUtils jiuTianUtils = new JiuTianUtils();// 九天大模型token有效期,两天private static Integer token_time = 60 * 60 * 24 * 2;@Overridepublic String generate_img_Completions(JiuTian jiuTian) throws JOSEException {if(jwt_token == null){jwt_token = "Bearer " + jiuTianUtils.generateToken(apiKey, token_time);System.out.println("九天大模型的token已生成: " + jwt_token);}if(jiuTian.getGenImgMessage().isEmpty()){System.out.println("提示语为空!");return "提示语为空!";}if(jiuTian.getGenImgSavePath().isEmpty()){System.out.println("图片存储路径为空!");return "图片存储路径为空!";}// 构建请求体数据JSONObject obj = new JSONObject();obj.put("model", "StableDiffusion");obj.put("prompt", jiuTian.getGenImgMessage());obj.put("returnImageUrl", false);// 格式转换器ObjectMapper objectMapper = new ObjectMapper();//处理结果AtomicReference<String> dealResult = new AtomicReference<>();// 构造线程池ExecutorService nonBlockingService = Executors.newCachedThreadPool();nonBlockingService.execute(() -> {try {// 设置WebClient处理数据时的内存限制ExchangeStrategies strategies = ExchangeStrategies.builder().codecs(codecs -> codecs.defaultCodecs().maxInMemorySize(Integer.MAX_VALUE)).build();Mono<String> mono = WebClient.builder().baseUrl("https://jiutian.10086.cn/largemodel/api/v2/completions").defaultHeader("Authorization", jwt_token).defaultHeader("Content-Type", "application/json").exchangeStrategies(strategies).build().post().bodyValue(obj)//请求体数据.retrieve()//获取响应体.bodyToMono(String.class);//响应数据类型转换mono.subscribe(result -> {System.out.println("响应报文已转成字符串!下面开始解析:");try {JsonNode responseJson = objectMapper.readTree(result);//字符串转JosnString base64_img = responseJson.get("choices").get(0).get("text").asText();//取图像base64字符串byte[] imageBytes = Base64.getDecoder().decode(base64_img);//解码InputStream in = new ByteArrayInputStream(imageBytes);//把字符数组转换成BufferedImageBufferedImage image = ImageIO.read(in);File outputfile = new File(jiuTian.getGenImgSavePath() + "\\output_img_Completions_" + jiuTianUtils.getNowTime() + ".png");//保存图像ImageIO.write(image, "PNG", outputfile);System.out.println("解析完成,生成的图像存储在" + outputfile.getAbsolutePath());dealResult.set("解析完成,生成的图像存储在" + outputfile.getAbsolutePath());} catch (JsonProcessingException e) {throw new RuntimeException(e);} catch (IOException e) {throw new RuntimeException(e);}},error -> {System.out.println("Error: "+ error);dealResult.set("Error: "+ error);},() -> {System.out.println("Completed!");dealResult.set("Completed!");});} catch (Exception e) {throw new RuntimeException(e);};});return dealResult.get();}@Overridepublic String generate_img_Chat(JiuTian jiuTian) throws JOSEException {if (jwt_token == null) {jwt_token = jiuTianUtils.generateToken(apiKey, token_time);System.out.println("九天大模型的token已生成: " + jwt_token);}if(jiuTian.getGenImgMessage().isEmpty()){System.out.println("提示语为空!");return "提示语为空!";}if(jiuTian.getGenImgSavePath().isEmpty()){System.out.println("图片存储路径为空!");return "图片存储路径为空!";}// 构建请求体数据JSONObject obj = new JSONObject();obj.put("model", "StableDiffusion");JSONObject params1 = new JSONObject();params1.put("role", "system");params1.put("content", "你是一位漫画家");JSONObject params2 = new JSONObject();params2.put("role", "user");params2.put("content", jiuTian.getGenImgMessage());List<JSONObject> messages = new ArrayList<>();messages.add(params1);messages.add(params2);obj.put("messages", messages);obj.put("returnImageUrl", false);// 格式转换器ObjectMapper objectMapper = new ObjectMapper();// 处理结果AtomicReference<String> dealResult = new AtomicReference<>();// 构造线程池ExecutorService nonBlockingService = Executors.newCachedThreadPool();nonBlockingService.execute(() -> {try {// 设置WebClient处理数据时的内存限制ExchangeStrategies strategies = ExchangeStrategies.builder().codecs(codecs -> codecs.defaultCodecs().maxInMemorySize(Integer.MAX_VALUE)).build();Mono<String> mono = WebClient.builder().baseUrl("https://jiutian.10086.cn/largemodel/api/v2/chat/completions").defaultHeader("Authorization", "Bearer " + jwt_token).defaultHeader("Content-Type", "application/json").exchangeStrategies(strategies).build().post().bodyValue(obj)//请求体数据.retrieve()//获取响应体.bodyToMono(String.class);//响应数据类型转换mono.subscribe(result -> {System.out.println("响应报文已转成字符串!下面开始解析:");try {JsonNode responseJson = objectMapper.readTree(result);//字符串转JosnString base64_img = responseJson.get("choices").get(0).get("message").get("content").asText();//取图像base64字符串byte[] imageBytes = Base64.getDecoder().decode(base64_img);//解码InputStream in = new ByteArrayInputStream(imageBytes);//把字符数组转换成BufferedImageBufferedImage image = ImageIO.read(in);File outputfile = new File(jiuTian.getGenImgSavePath() + "\\output_img_Completions_" + jiuTianUtils.getNowTime() + ".png");//保存图像ImageIO.write(image, "PNG", outputfile);System.out.println("解析完成,生成的图像存储在" + outputfile.getAbsolutePath());dealResult.set("解析完成,生成的图像存储在" + outputfile.getAbsolutePath());} catch (JsonProcessingException e) {throw new RuntimeException(e);} catch (IOException e) {throw new RuntimeException(e);}},error -> {System.out.println("Error: "+ error);dealResult.set("Error: "+ error);},() -> {System.out.println("Completed!");dealResult.set("Completed!");});} catch (Exception e) {throw new RuntimeException(e);};});return dealResult.get();}
}

 8、在apifox接口软件中调用项目接口测试。

9、查看图片保存结果。图片挺好看的。

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.mzph.cn/pingmian/66771.shtml

如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈email:809451989@qq.com,一经查实,立即删除!

相关文章

1. npm 常用命令详解

npm 常用命令详解 npm&#xff08;Node Package Manager&#xff09;是 Node.js 的包管理工具&#xff0c;用于安装和管理 Node.js 应用中的依赖库。下面是 npm 的一些常用命令及其详细解释和示例代码。 镜像源 # 查询当前使用的镜像源 npm get registry# 设置为淘宝镜像源 …

12. C语言 数组与指针(深入理解)

本章目录: 前言1. 什么是数组&#xff1f;2. 数组的声明与初始化声明数组初始化数组 3. 访问数组元素遍历数组 4. 获取数组长度使用 sizeof 获取长度使用宏定义简化 5. 数组与指针数组名与指针的区别使用指针操作数组 6. 多维数组遍历多维数组 7. 数组作为函数参数8. 高级技巧与…

Leetcode 120. 三角形最小路径和 动态规划

原题链接&#xff1a;Leetcode 120. 三角形最小路径和 class Solution { public:int minimumTotal(vector<vector<int>>& triangle) {int n triangle.size();if (n 1)return triangle[0][0];int dp[n][n];dp[0][0] triangle[0][0];int res INT_MAX;for (in…

网络安全概论

网络安全概论--网络基础知识--防火墙技术--身份识别技术--虚拟专用网络--入侵检测技术--病毒和恶意代码 --业务连续性计划--安全管理--信息系统安全方案设计方法 一、网络安全面临的威胁 物理安全威胁、操作系统的安全缺陷、网络协议的安全缺陷、应用软件的实现缺陷、用户使…

设计模式 行为型 状态模式(State Pattern)与 常见技术框架应用 解析

状态模式&#xff08;State Pattern&#xff09;是一种行为型设计模式&#xff0c;它允许对象在内部状态改变时改变其行为&#xff0c;使得对象看起来好像修改了它的类。这种设计模式的核心思想是将对象的状态和行为封装成不同的状态类&#xff0c;通过状态对象的行为改变来避免…

Java 线程池 ThreadPoolExecutor 底层原理与源码分析

引言 我们探讨了 ThreadPoolExecutor 的基本概念、内部机制以及部分源码实现。本文将继续深入研究该类的更多细节&#xff0c;并结合提供的文档内容&#xff0c;进一步解析线程池的工作流程、任务提交和执行的具体过程&#xff0c;以及如何通过自定义配置来优化性能。 一、线…

Mysql--基础篇--视图,存储过程,触发器

1、视图 MySQL视图&#xff08;View&#xff09;是一个虚拟表&#xff0c;其内容由查询定义。同真实的表一样&#xff0c;视图包含一系列带有名称的列和行数据。但是&#xff0c;视图并不在数据库中以存储的数据值集形式存在。行和列数据来自由定义视图的查询所引用的表&#…

某团 mtgsig1.2 | sdkVersion: 3.0.0 签名算法分析记录(2025/1/9)

【作者主页】&#xff1a;小鱼神1024 【擅长领域】&#xff1a;JS逆向、小程序逆向、AST还原、验证码突防、Python开发、浏览器插件开发、React前端开发、NestJS后端开发等等 本文章中所有内容仅供学习交流使用&#xff0c;不用于其他任何目的&#xff0c;不提供完整代码&#…

(二)最长公共子序列、最长上升子序列、最大子段和、三角形最小路径和、矩阵连乘、0-1背包

最近刚考完算法设计分析课的考试&#xff0c;复习总结一下期末考试的几道算法题吧 目录 LCR 095. 最长公共子序列 300. 最长递增子序列 53. 最大子数组和 LCR 100. 三角形最小路径和 矩阵连乘问题 0-1背包 LCR 095. 最长公共子序列 给定两个字符串 text1 和 text2&#xff…

聚类系列 (二)——HDBSCAN算法详解

在进行组会汇报的时候&#xff0c;为了引出本研究动机&#xff08;论文尚未发表&#xff0c;暂不介绍&#xff09;&#xff0c;需要对DBSCAN、OPTICS、和HDBSCAN算法等进行详细介绍。在查询相关资料的时候&#xff0c;发现网络上对于DBSCAN算法的介绍非常多与细致&#xff0c;但…

通义灵码在跨领域应用拓展之物联网篇

目录 一.引言 二.通义灵码简介 三.通义灵码在物联网领域的设备端应用 1.传感器数据采集 (1).不同类型传感器的数据读取 (2).数据转换与预处理 2.设备控制指令接收和执行 (1).指令解析与处理 (2).设备动作执行 四.通义灵码在物联网领域的云端平台应用 1.数据存储和管…

python_excel列表单元格字符合并、填充、复制操作

读取指定sheet页&#xff0c;根据规则合并指定列&#xff0c;填充特定字符&#xff0c;删除多余的列&#xff0c;每行复制四次&#xff0c;最后写入新的文件中。 import pandas as pd""" 读取指定sheet页&#xff0c;根据规则合并指定列&#xff0c;填充特定字…

C++ STL 中的 vector 总结

1. 什么是 std::vector&#xff1f; std::vector 是 C STL 提供的动态数组容器&#xff0c;可以动态调整大小并存储任意类型的元素。 与普通数组相比&#xff0c;std::vector 更加灵活&#xff0c;提供了丰富的操作接口。 2. 基本特性 动态大小&#xff1a;支持在运行时动态增…

DolphinScheduler自身容错导致的服务器持续崩溃重大问题的排查与解决

01 问题复现 在DolphinScheduler中有如下一个Shell任务&#xff1a; current_timestamp() { date "%Y-%m-%d %H:%M:%S" }TIMESTAMP$(current_timestamp) echo $TIMESTAMP sleep 60 在DolphinScheduler将工作流执行策略设置为并行&#xff1a; 定时周期调度设置…

【stm32+K210项目】基于K210与STM32协同工作的智能垃圾分类系统设计与实现(完整工程资料源码)

视频效果演示: 基于K210与STM32协同工作的智能垃圾分类系统设计与实现 目录: 目录 视频效果演示: 目录:

CISAW-ES应急服务方向信息安全事件分级

网络安全事件事件分级 网络安全事件分为四级&#xff1a;特别重大网络安全事件、重大网络安全事大网络安全事件、一般网络安全事件。 1&#xff0e;特别重大网络安全事件 符合下列情形之一的&#xff0c;为特别重大网络安全事件。 &#xff08;1&#xff09;重要网络和信息系…

油猴支持阿里云自动登陆插件

遇到的以下问题,都已在脚本中解决: 获取到的元素赋值在页面显示,但是底层的value并没有改写,导致请求就是获取不到数据元素的加载时机不定,尤其是弱网情况下,只靠延迟还是有可能获取不到,且登陆不丝滑,通过元素发现机制,解决此问题并做到丝滑登陆根据密钥计算校验码之…

简聊MySQL的顺序读写和随机读写

在MySQL数据库中&#xff0c;顺序读写和随机读写的应用区别主要体现在以下几个方面&#xff1a; 一、定义回顾 顺序读写&#xff1a;数据按照物理地址的连续性进行读写操作&#xff0c;通常用于处理大型文件或连续的数据块。随机读写&#xff1a;数据分散在磁盘的不同位置进行…

[Git] git reset --hard / git reset --soft

git reset --hard 功能&#xff1a;重置索引&#xff08;暂存区&#xff09;和工作目录到指定的提交状态。这意味着它会丢弃所有未提交的更改和已暂存的更改。 适用场景&#xff1a;当你想要完全放弃当前工作目录中的所有更改并回退到某个特定提交状态时&#xff0c;可以使用这…

C语言基本知识复习浓缩版:scanf函数

C语言基本知识复习浓缩版&#xff1a;scanf函数 1 scanf()函数用于读取用户的键盘输入 2 scanf()函数的基本形式&#xff1a;scanf("参数列表1",参数列表2): 参数列表1&#xff1a;用户键盘输入的数据的类型&#xff0c;用占位符表示。 参数列表2&#xff1a;用户键…