Spring AI 初学

Spring AI 初学

Spring AI 官方地址
”spring 不生产 AI，只是 AI 工具的搬运工“
项目可以查看gitee
Open AI

前期准备

Open AI官方地址，需要使用魔法才能打开，同时购买很麻烦，发起淘宝举行购买，只需要购买 open ai 的 apikey 即可。
apikey 形如 sk-xxxxxxxxxxxxxxxxx
项目创建

Idea 创建 SpringBoot Maven 项目（基于1.0-SNAPSHOT版本，SpringBoot 3.2.6），依赖选择Spring Web、 OpenAI。其他可以自行选择
修改项目仓库地址，中心仓库暂时还没 Spring AI 相关 jar 包。仓库地址改成快照仓库地址，官方说明

1.     <repository>
2.         <id>spring-snapshots</id>
3.         <name>Spring Snapshots</name>
4.         <url>https://repo.spring.io/snapshot</url>
5.         <releases>
6.             <enabled>false</enabled>
7.         </releases>
8.     </repository>

复制代码

项目中找到 pom.xml 文件，将 0.8.1 改为 1.0.0-SNAPSHOT
yaml 配置文件中添加，openai 更多配置可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration。

1. spring:
2.   ai:
3.     openai:
4.       # 购买的 api-key
5.       api-key: sk-xxxx
6.       # 如果是官方地址，则可以不填，默认为 https://api.openai.com
7.       base-url:

复制代码

谈天

基础使用

主要类 org.springframework.ai.openai.OpenAiChatModel，快照版本不同，可能名字不一样，可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration 中的谈天类是哪个。

1. import jakarta.annotation.Resource;
2. import org.junit.jupiter.api.Test;
3. import org.springframework.ai.openai.OpenAiChatModel;
4. import org.springframework.boot.test.context.SpringBootTest;
6. @SpringBootTest
7. public class ChatTest {
9.     @Resource
10.     private OpenAiChatModel chatModel;
12.     @Test
13.     public void chat1(){
14.         String msg = "你是谁？";
15.         //返回string数据
16.         String res = chatModel.call(msg);
17.         System.out.println(res);
18.     }
19.    
20.     @Test
21.     public void chat2(){
22.         String msg = "你是谁？";
23.         //返回对象
24.         ChatResponse res = chatModel.call(new Prompt(msg));
25.         System.out.println(res);
26.         //获取对话返回结果
27.         System.out.println(res.getResult().getOutput().getContent());
28.     }
30. }

复制代码

配置属性

1.     @Test
2.     public void test3(){
3.         String msg = "你是谁";
4.         //采用 gpt-4-turbo 模型
5.         ChatResponse res = chatModel.call(new Prompt(msg, OpenAiChatOptions.builder()
6.                                                      .withModel("gpt-4-turbo")
7.                                                      .build()));
8.         System.out.println(res);
9.         //获取对话返回结果
10.         System.out.println(res.getResult().getOutput().getContent());
11.     }

复制代码

谈天模子配置属性可以查看 org.springframework.ai.autoconfigure.openai.OpenAiChatProperties，也可以在官网查看更详细的信息。配置属性也可以放在 yml 配置文件中，如 OpenAiChatProperties 的注解，需要以 spring.ai.openai.chat 开头，例如将 gpt-4-turbo 配置在配置文件中，就是 OpenAiChatProperties 中 options 中的属性。

1. spring:
2.   ai:
3.     openai:
4.       chat:
5.         options:
6.           model: gpt-4-turbo

复制代码

多模子

可以共同图片等让谈天模子举行回答。

1.     //给图片来进行聊天
2.     @Test
3.     public void test4() {
4.         //获取图片资源
5.         ClassPathResource resource = new ClassPathResource("2024052701.png");
6.         UserMessage userMessage = new UserMessage("说说你看到了什么",
7.                 List.of(new Media(MimeTypeUtils.IMAGE_PNG, resource)));
8.         ChatResponse res = chatModel.call(new Prompt(userMessage, OpenAiChatOptions.builder()
9.                 .withModel("gpt-4-turbo")
10.                 .build()));
11.         System.out.println(res);
12.         //获取回答
13.         System.out.println(res.getResult().getOutput().getContent());
14.     }

复制代码

图像

基础使用

主要类  org.springframework.ai.openai.OpenAiImageModel，快照版本不同，可能类不一样。可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration 中详细的图像类是哪个。

1. import jakarta.annotation.Resource;
2. import org.junit.jupiter.api.Test;
3. import org.springframework.ai.image.ImagePrompt;
4. import org.springframework.ai.image.ImageResponse;
5. import org.springframework.ai.openai.OpenAiImageModel;
6. import org.springframework.boot.test.context.SpringBootTest;
8. @SpringBootTest
9. public class ImageTest {
11.     @Resource
12.     private OpenAiImageModel imageModel;
14.     @Test
15.     public void test(){
16.         //调用 image 模型的 call 方法获取图片
17.         ImageResponse res = imageModel.call(new ImagePrompt("山水画"));
18.         //AI 绘制的图片路径
19.         String url = res.getResult().getOutput().getUrl();
20.         System.out.println(url);
21.     }
23. }

复制代码

配置属性

1.     @Test
2.     public void test2(){
3.         //使用 dall-e-2 绘画
4.         OpenAiImageOptions options = OpenAiImageOptions.builder()
5.             .withModel(OpenAiImageApi.ImageModel.DALL_E_2.getValue())
6.             .build();
7.         ImageResponse res = imageModel.call(new ImagePrompt("山水画", options));
8.         //获取 AI 绘画路径
9.         String url = res.getResult().getOutput().getUrl();
10.         System.out.println(url);
11.     }

复制代码

图像模子属性配置可以查看 org.springframework.ai.autoconfigure.openai.OpenAiImageProperties，也可以查看官网获取更详细的信息。当然配置属性也可以在 yml 中定义，如 OpenAiImageProperties 上的注解，需要以 spring.ai.openai.image 开头，例如使用 dall-e-2 模子举行绘画

1. spring:
2.   ai:
3.     openai:
4.       image:
5.         options:
6.           model: dall-e-2

复制代码

语音

语音转文字

基础使用

主要类 org.springframework.ai.openai.OpenAiAudioTranscriptionModel，快照版本不同，可能名字不一样，可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration 中的语音转文字翻译类是哪个。

1. import jakarta.annotation.Resource;
2. import org.junit.jupiter.api.Test;
3. import org.springframework.ai.openai.OpenAiAudioTranscriptionModel;
4. import org.springframework.boot.test.context.SpringBootTest;
5. import org.springframework.core.io.ClassPathResource;
7. @SpringBootTest
8. public class AudioTest {
10.     //语音转文字
11.     @Resource
12.     private OpenAiAudioTranscriptionModel transcriptionModel;
14.     @Test
15.     public void testTranscription1(){
16.         String res = transcriptionModel.call(new ClassPathResource("2024052702.mp3"));
17.         System.out.println(res);
18.     }
20. }

复制代码

配置属性

1.     @Test
2.     public void testTranscription2(){
3.         //创建模型属性，采用 whisper-1 语音模型
4.         OpenAiAudioTranscriptionOptions options = new OpenAiAudioTranscriptionOptions().builder()
5.                 .withModel(OpenAiAudioApi.WhisperModel.WHISPER_1.getValue())
6.                 .build();
7.         AudioTranscriptionResponse res = transcriptionModel.call(
8.             new AudioTranscriptionPrompt(new ClassPathResource("2024052702.mp3"), options));
9.         //获取翻译内容
10.         String output = res.getResult().getOutput();
11.         System.out.println(output);
12.     }

复制代码

语音转文字模子属性可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAudioTranscriptionProperties，也可以在官网查看更详细信息。当然可以在 yml 配置中配置属性，如 OpenAiAudioTranscriptionProperties 上的注解，以 spring.ai.openai.audio.transcription 开头，例如接纳 whisper-1 模子

1. spring:
2.   ai:
3.     openai:
4.       audio:
5.         transcription:
6.           options:
7.             model: whisper-1

复制代码

文字转语音

基础使用

主要类 org.springframework.ai.openai.OpenAiAudioSpeechModel，快照版本不同，可能名字不一样，可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAutoConfiguration 中的文字转语音类是哪个。

1. import jakarta.annotation.Resource;
2. import org.junit.jupiter.api.Test;
3. import org.springframework.ai.openai.OpenAiAudioSpeechModel;
4. import org.springframework.boot.test.context.SpringBootTest;
5. import java.io.FileOutputStream;
6. import java.io.IOException;
8. @SpringBootTest
9. public class AudioTest2 {
11.     @Resource
12.     private OpenAiAudioSpeechModel speechModel;
14.     //byte数组转文件
15.     private void byteArrayToFile(byte[] byteArray, String filePath) throws IOException {
16.         FileOutputStream fos = new FileOutputStream(filePath);
17.         fos.write(byteArray);
18.         fos.close();
19.     }
21.     @Test
22.     public void testSpeech() throws IOException {
23.         byte[] res = speechModel.call("我爱北京");
24.         byteArrayToFile(res,"D:\\project\\AI\\openai\\speech\\1.mp3");
25.     }
27. }

复制代码

属性配置

1.     @Test
2.     public void testSpeech2() throws IOException {
3.         //采用 tts-1-hd 模型
4.         OpenAiAudioSpeechOptions options = new OpenAiAudioSpeechOptions().builder()
5.                 .withModel(OpenAiAudioApi.TtsModel.TTS_1_HD.getValue())
6.                 .build();
7.         SpeechPrompt prompt = new SpeechPrompt("我爱北京", options);
8.         SpeechResponse res = speechModel.call(prompt);
9.         byte[] bytes = res.getResult().getOutput();
10.         byteArrayToFile(bytes,"D:\\project\\AI\\openai\\speech\\1-hd.mp3");
11.     }

复制代码

文字转语音模子属性可以查看 org.springframework.ai.autoconfigure.openai.OpenAiAudioSpeechProperties，也可以在官网查看更详细信息。当然可以在 yml 配置中配置属性，如 OpenAiAudioSpeechProperties 上的注解，以 spring.ai.openai.audio.speech 开头，例如接纳 tts-1-hd 模子

1. spring:
2.   ai:
3.     openai:
4.       audio:
5.         speech:
6.           options:
7.             model: tts-1-hd

复制代码

Ollama


免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！更多信息从访问主页：qidao123.com:ToB企服之家，中国第一个企服评测及商务社交产业平台。