2024版本idea集成SpringBoot + Ai 手写一个chatgpt 【推荐】
Ylik~ 2024-06-22 15:31:08 阅读 52
题目:SpringBoot + OpenAi 代码获取
在这里获取key和url:获取免费key
base-url为这两个:
话不多说直接来!
一、简介
Spring AI 是 AI 工程的应用框架。其目标是将 Spring 生态系统设计原则(如可移植性和模块化设计)应用于 AI,并推广使用 POJO 作为 AI 领域应用程序的构建块。
跨 AI 提供商的可移植 API 支持,适用于聊天、文本到图像和嵌入模型。支持同步和流 API 选项。还支持下拉以访问特定于模型的功能
二、Ai聊天程序代码
1、 创建项目工程
在父工程下面创建新的模块勾选上依赖然后创建
具体的依赖如下
<?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 https://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>3.2.4</version> <relativePath/> <!-- lookup parent from repository --> </parent> <!-- Generated by https://start.springboot.io --> <!-- 优质的 spring/boot/data/security/cloud 框架中文文档尽在 => https://springdoc.cn --> <groupId>com.ysl</groupId> <artifactId>SpringAi</artifactId> <version>0.0.1-SNAPSHOT</version> <packaging>pom</packaging> <name>SpringAi</name> <description>SpringAi</description> <modules> <module>spring-ai-01-chat</module> </modules> <properties> <java.version>17</java.version><!-- 快照版本--> <spring-ai.version>1.0.0-SNAPSHOT</spring-ai.version> </properties> <dependencies> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-web</artifactId> </dependency> <dependency> <groupId>org.springframework.ai</groupId> <artifactId>spring-ai-openai-spring-boot-starter</artifactId> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-devtools</artifactId> <scope>runtime</scope> <optional>true</optional> </dependency> <dependency> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> <optional>true</optional> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-starter-test</artifactId> <scope>test</scope> </dependency> </dependencies><!-- 相当于继承一个父项目:spring-ai-bom--> <dependencyManagement> <dependencies> <dependency> <groupId>org.springframework.ai</groupId> <artifactId>spring-ai-bom</artifactId> <version>${ spring-ai.version}</version> <type>pom</type> <scope>import</scope> </dependency> </dependencies> </dependencyManagement> <build> <plugins> <plugin> <groupId>org.springframework.boot</groupId> <artifactId>spring-boot-maven-plugin</artifactId> <configuration> <image> <builder>paketobuildpacks/builder-jammy-base:latest</builder> </image> <excludes> <exclude> <groupId>org.projectlombok</groupId> <artifactId>lombok</artifactId> </exclude> </excludes> </configuration> </plugin> </plugins> </build> <!--~配置本项目的仓库:因maven中心仓库还设有更新spring ai的jar包--> <repositories> <repository><!-- 里程碑版本releases的仓库,改成快照版本的snapshot--> <id>spring-snapshot</id> <name>Spring Snapshots</name> <url>https://repo.spring.io/snapshot</url> <releases> <enabled>false</enabled> </releases> </repository> </repositories></project>
编写yml配置
openai有自动配置类OpenAiAutoConfiguration
其中有聊天客户端,图片客户端…等(看下面源码)
//聊天客户端@Bean@ConditionalOnMissingBean@ConditionalOnProperty(prefix = OpenAiChatProperties.CONFIG_PREFIX, name = "enabled", havingValue = "true",matchIfMissing = true)public OpenAiChatClient openAiChatClient(OpenAiConnectionProperties commonProperties,OpenAiChatProperties chatProperties, RestClient.Builder restClientBuilder,List<FunctionCallback> toolFunctionCallbacks, FunctionCallbackContext functionCallbackContext,RetryTemplate retryTemplate, ResponseErrorHandler responseErrorHandler) { var openAiApi = openAiApi(chatProperties.getBaseUrl(), commonProperties.getBaseUrl(),chatProperties.getApiKey(), commonProperties.getApiKey(), restClientBuilder, responseErrorHandler);if (!CollectionUtils.isEmpty(toolFunctionCallbacks)) { chatProperties.getOptions().getFunctionCallbacks().addAll(toolFunctionCallbacks);}return new OpenAiChatClient(openAiApi, chatProperties.getOptions(), functionCallbackContext, retryTemplate);}
//图片客户端@Bean@ConditionalOnMissingBean@ConditionalOnProperty(prefix = OpenAiImageProperties.CONFIG_PREFIX, name = "enabled", havingValue = "true",matchIfMissing = true)public OpenAiImageClient openAiImageClient(OpenAiConnectionProperties commonProperties,OpenAiImageProperties imageProperties, RestClient.Builder restClientBuilder, RetryTemplate retryTemplate,ResponseErrorHandler responseErrorHandler) { String apiKey = StringUtils.hasText(imageProperties.getApiKey()) ? imageProperties.getApiKey(): commonProperties.getApiKey();String baseUrl = StringUtils.hasText(imageProperties.getBaseUrl()) ? imageProperties.getBaseUrl(): commonProperties.getBaseUrl();Assert.hasText(apiKey, "OpenAI API key must be set");Assert.hasText(baseUrl, "OpenAI base URL must be set");var openAiImageApi = new OpenAiImageApi(baseUrl, apiKey, restClientBuilder, responseErrorHandler);return new OpenAiImageClient(openAiImageApi, imageProperties.getOptions(), retryTemplate);}
二、一个简单的示例
1、直接写一个Controller层就可以
package com.ysl.controller;import jakarta.annotation.Resource;import org.springframework.ai.chat.ChatResponse;import org.springframework.ai.chat.prompt.Prompt;import org.springframework.ai.openai.OpenAiChatClient;import org.springframework.ai.openai.OpenAiChatOptions;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RequestParam;import org.springframework.web.bind.annotation.RestController;import reactor.core.publisher.Flux;/*** @Author Ysl* @Date 2024/5/11* @name SpringAi**/@RestControllerpublic class ChatController { /** * OpenAi自动装配,可以直接注入使用 */ @Resource private OpenAiChatClient openAiChatClient; /** * 调用OpenAi的接口,call方法为同步的api * @param msg 你要问的问题 * @return */ @RequestMapping ("/ai/chat") public String chat(@RequestParam("msg") String msg) { String call = openAiChatClient.call(msg); return call; } /** * 调用OpenAi的接口 * @param msg 你要问的问题 * @return Object--json对象 */ @RequestMapping ("/ai/chat1") public Object chat1(@RequestParam("msg") String msg) { ChatResponse response = openAiChatClient.call(new Prompt(msg)); return response;// return response.getResult().getOutput().getContent();//只拿到内容 } /** * 调用OpenAi的接口 * @param msg 你要问的问题 * @return */ @RequestMapping ("/ai/chat3") public String chat3(@RequestParam("msg") String msg) { //可选参数在yml配置,同时在代码中也配置,那么会以代码为准 ChatResponse response = openAiChatClient.call(new Prompt(msg, OpenAiChatOptions.builder()// .withModel("gpt-4")//使用的模型 .withTemperature(0.3F)//温度越高回答越慢,温度越低回答越快 .build())); return response.getResult().getOutput().getContent(); } /** * 调用OpenAi的接口 stream是流式的api * @param msg 你要问的问题 * @return */ @RequestMapping ("/ai/chat4") public Object chat4(@RequestParam("msg") String msg) { //可选参数在yml配置,同时在代码中也配置,那么会以代码为准 Flux<ChatResponse> flux = openAiChatClient.stream(new Prompt(msg, OpenAiChatOptions.builder()// .withModel("gpt-3.5")//使用的模型 .withTemperature(0.3F)//温度越高回答越慢,温度越低回答越快 .build())); flux.toStream().forEach(chatResponse ->{ System.out.println(chatResponse.getResult().getOutput().getContent()); }); return flux.collectList();//数据的序列 }}
2、直接在浏览器访问
http://localhost:8080/ai/chat?msg=24年经济形势
http://localhost:8080/ai/chat1?msg=24年经济形势
http://localhost:8080/ai/chat3?msg=java怎么学
OpenAi聊天客户端就写到这里,接下来是图片客户端。
三、Ai画图程序代码
首先需要确保你的key支持绘图可以使用DALL·E 模型
1、创建一个子项目(pom文件的依赖和第一个项目一样即可)
2、编写yml配置文件
3、编写controller层
import jakarta.annotation.Resource;import org.springframework.ai.image.ImagePrompt;import org.springframework.ai.image.ImageResponse;import org.springframework.ai.openai.OpenAiImageClient;import org.springframework.ai.openai.OpenAiImageOptions;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RequestParam;import org.springframework.web.bind.annotation.RestController;/** * @Author Ysl * @Date 2024/5/12 * @name SpringAi **/@RestControllerpublic class ImageController { @Resource private OpenAiImageClient openAiImageClient; @RequestMapping("/ai/image") public Object image(@RequestParam("msg") String msg) { ImageResponse imageResponse = openAiImageClient.call(new ImagePrompt(msg));// 对图片进行业务处理(拿到图片的链接) String url = imageResponse.getResult().getOutput().getUrl(); return imageResponse.getResult().getOutput(); } @RequestMapping("/ai/image2") public Object image2(@RequestParam("msg") String msg) { //第二个参数是传递图片的参数,同样可在yml里配置 ImageResponse imageResponse = openAiImageClient.call(new ImagePrompt(msg, OpenAiImageOptions.builder() .withQuality("hd")//高清 .withN(1)//生成1张,可填数字1-10 .withHeight(1024)//高度 .withWidth(1024)//宽度 .build() ));// 对图片进行业务处理(拿到图片的链接) String url = imageResponse.getResult().getOutput().getUrl(); return imageResponse.getResult().getOutput(); }}
3、访问链接
http://localhost:8080/ai/image?msg=画一只蝴蝶http://localhost:8080/ai/image2?msg=画一只蝴蝶
4、结果:
打开链接
Ai画图就讲到这里,下面是音频翻译
四、Ai音频翻译代码
首先确保你的key可以使用whisper 模型
1、创建子项目编写配置文件
2、用到的工具类:将bytes[ ]字节数组写入新建的文件中
import java.io.*;/** * @Author Ysl * @Date 2024/5/12 * @name SpringAi **/public class save2File { /** * 方法功能:将字节数组写入到新建文件中。 * @param fname * @param msg * @return boolean * */ public static boolean save2File(String fname, byte[] msg){ OutputStream fos = null; try{ File file = new File(fname); File parent = file.getParentFile(); boolean bool; if ((!parent.exists()) && (!parent.mkdirs())) { return false; } fos = new FileOutputStream(file); fos.write(msg); fos.flush(); return true; } catch (FileNotFoundException e) { return false; }catch (IOException e){ File parent; return false; } finally{ if (fos != null) { try{ fos.close(); }catch (IOException e) { } } } }}
2、编写controller层
import jakarta.annotation.Resource;import org.springframework.ai.openai.OpenAiAudioSpeechClient;import org.springframework.ai.openai.OpenAiAudioTranscriptionClient;import org.springframework.core.io.ClassPathResource;import org.springframework.web.bind.annotation.RequestMapping;import org.springframework.web.bind.annotation.RestController;import static com.ysl.utisl.save2File.save2File;/** * @Author Ysl * @Date 2024/5/12 * @name SpringAi **/@RestControllerpublic class TranscriptionController { //将音频转文字时使用 @Resource private OpenAiAudioTranscriptionClient transcriptionClient; //将文字转语音时使用 @Resource private OpenAiAudioSpeechClient speechClient; /** * 将音频转文字 * @return */ @RequestMapping("/ai/transcription") public Object transcription() { //读取的是磁盘的路径 //FileSystemResource audioFile = new FileSystemResource("C:\\Users\\DELL\\Desktop\\luyin.m4a"); //读取的是classpath静态资源下的文件 ClassPathResource audioFile = new ClassPathResource("luyin.m4a"); String call = transcriptionClient.call(audioFile); System.out.println(call); return call; } /** * 将文字转音频 * @return */ @RequestMapping("/ai/tts") public Object tts() { String text = "Spring AI 是 AI 工程的应用框架。其目标是将 Spring 生态系统设计原则(如可移植性和模块化设计)应用于 AI,并推广使用 POJO 作为 AI 领域应用程序的构建块。 跨 AI 提供商的可移植 API 支持,适用于聊天、文本到图像和嵌入模型。支持同步和流 API 选项。还支持下拉以访问特定于模型的功能"; byte[] bytes = speechClient.call(text); save2File("C:\\Users\\DELL\\Desktop\\test.mp3",bytes); return "OK"; }}
3、访问链接
1、 http://localhost:8080/ai/transcription将音频中的话转化为文字显示在页面中
2、 http://localhost:8080/ai/tts将text中的文字转为语音后储存在路径C:\Users\DELL\Desktop\test.mp3打开mp3文件即可听到代码中text文本中的内容
声明
本文内容仅代表作者观点,或转载于其他网站,本站不以此文作为商业用途
如有涉及侵权,请联系本站进行删除
转载本站原创文章,请注明来源及作者。