2024版本idea集成SpringBoot + Ai 手写一个chatgpt 【推荐】

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

题目：SpringBoot + OpenAi 代码获取

在这里获取key和url：获取免费key

base-url为这两个：

话不多说直接来！

一、简介

Spring AI 是 AI 工程的应用框架。其目标是将 Spring 生态系统设计原则（如可移植性和模块化设计）应用于 AI，并推广使用 POJO 作为 AI 领域应用程序的构建块。

跨 AI 提供商的可移植 API 支持，适用于聊天、文本到图像和嵌入模型。支持同步和流 API 选项。还支持下拉以访问特定于模型的功能

二、Ai聊天程序代码

1、创建项目工程

在父工程下面创建新的模块

勾选上依赖然后创建

具体的依赖如下



    4.0.0
    
        org.springframework.boot
        spring-boot-starter-parent
        3.2.4
         
    
    
    
    com.ysl
    SpringAi
    0.0.1-SNAPSHOT
    pom
    SpringAi
    SpringAi
    
        spring-ai-01-chat
    
    
        17

        1.0.0-SNAPSHOT
    
    
        
            org.springframework.boot
            spring-boot-starter-web
        
        
            org.springframework.ai
            spring-ai-openai-spring-boot-starter
        
        
            org.springframework.boot
            spring-boot-devtools
            runtime
            true
        
        
            org.projectlombok
            lombok
            true
        
        
            org.springframework.boot
            spring-boot-starter-test
            test
        
    

    
        
            
                org.springframework.ai
                spring-ai-bom
                ${spring-ai.version}
                pom
                import
            
        
    
    
        
            
                org.springframework.boot
                spring-boot-maven-plugin
                
                    
                        paketobuildpacks/builder-jammy-base:latest
                    
                    
                        
                            org.projectlombok
                            lombok
                        
                    
                
            
        
    
    
    
        

            spring-snapshot
            Spring Snapshots
            https://repo.spring.io/snapshot
            
                false

编写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 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
**/
@RestController
public 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 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
 **/
@RestController
public 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
 **/
@RestController
public 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、访问链接