基于Spring AI alibaba组件AI问答功能开发示例

基于Spring AI alibaba组件AI问答功能开发示例
功能效果图：

http://localhost:9999/ai/test
http://localhost:9999/ai/chat?input=ai
设置 application.yaml 指定 API-KEY（可通过访问阿里云百炼模子服务平台获取，有免费额度可用。）
阿里云百炼
阿里云百炼模子服务平台
https://ai.aliyun.com/
架构版本
JDK17
SpringBoot3.3.3
Spring-AI 1.0.0-M3
Spring-AI-Alibaba1.0.0-M3.1
vue3

1. 架构版本JDK17
2. SpringBoot3.3.3
3. Spring-AI 1.0.0-M3
4. Spring-AI-Alibaba1.0.0-M3.1
5. vue3

复制代码

aiApplication.java入口类

1. package com.cwgis;
3. import com.cwgis.pg.function.MockOrderService;
4. import com.cwgis.pg.function.Response;
5. import org.springframework.boot.SpringApplication;
6. import org.springframework.boot.autoconfigure.SpringBootApplication;
7. import org.springframework.context.annotation.Bean;
8. import org.springframework.context.annotation.Description;
10. import java.util.function.Function;
12. /**
13. * Hello world!
14. */
15. @SpringBootApplication
16. public class aiApplication {
17.    
18.    
19.     public static void main(String[] args) {
20.    
21.    
22.         SpringApplication.run(aiApplication.class, args);
23.         System.out.println("===============");
24.         System.out.println("ai服务正在运行中!");
25.         System.out.println("===============");
26.     }
28.     @Bean
29.     @Description("根据用户编号和订单编号查询订单信息")
30.     public Function<MockOrderService.Request, Response> getOrderFunction(
31.             MockOrderService mockOrderService) {
32.    
33.    
34.         return mockOrderService::getOrder;
35.     }
36. }

复制代码

ChatController.java问答服务类

1. package com.cwgis;
3. import com.cwgis.pg.core.AjaxResult;
4. import com.cwgis.pg.function.MJDLTBService;
5. import com.cwgis.pg.function.MockWeatherService;
6. import org.springframework.ai.chat.client.ChatClient;
7. import org.springframework.web.bind.annotation.GetMapping;
8. import org.springframework.web.bind.annotation.PostMapping;
9. import org.springframework.web.bind.annotation.RequestMapping;
10. import org.springframework.web.bind.annotation.RestController;
11. import reactor.core.publisher.Flux;
13. import java.util.Objects;
15. @RestController
16. @RequestMapping("/ai")
17. public class ChatController {
18.    
19.    
20.     private final ChatClient chatClient;
22.     public ChatController(ChatClient.Builder builder) {
23.    
24.    
25.         this.chatClient = builder.build();
26.     }
28.     @GetMapping("/test")
29.     public AjaxResult test() {
30.    
31.    
32.         String r= this.chatClient.prompt().user("测试").call().content();
33.         return AjaxResult.success(r);
34.     }
36.     //自动加载自定义function函数
37.     @PostMapping("/chat")
38.     public AjaxResult chat(String input) {
39.    
40.    
41.         String r= this.chatClient.prompt()
42.                 .function("getWeather", "根据城市查询天气", new MockWeatherService())
43.                 .functions("getOrderFunction")     //获取aiApplication类中注册的Bean函数
44.                 .function("MJDLTBService","根据城市和地类名称查询地类图斑面积",new MJDLTBService())
45.                 .user(input).call().content();
46.         return AjaxResult.success(r);
47.     }
49.     @PostMapping("/stream")
50.     public AjaxResult stream(String input) {
51.    
52.    
54.         Flux<String> content = this.chatClient.prompt().user(input).stream().content();
55.         String r= Objects.requireNonNull(content.collectList().block()).stream().reduce((a, b) -> a + b).get();
56.         return AjaxResult.success(r);
57.     }
58.     //函数调用
59.     @PostMapping("/weather-service")
60.     public AjaxResult weatherService(String subject) {
61.    
62.    
63.         String r= chatClient.prompt()
64.                 .function("getWeather", "根据城市查询天气", new MockWeatherService())
65.                 .user(subject)
66.                 .call()
67.                 .content();
68.         return AjaxResult.success(r);
69.     }
70.     @PostMapping("/order-detail")
71.     public AjaxResult orderDetail() {
72.    
73.    
74.         String r= chatClient.prompt()
75.                 .functions("getOrderFunction")     //获取aiApplication类中注册的Bean函数
76.                 .user("帮我查询一下订单, 用户编号为1001, 订单编号为2001")
77.                 .call()
78.                 .content();
79.         return AjaxResult.success(r);
80.     }
81.     //
82. }

复制代码

MJDLTBService.java自界说服务函数

1. package com.cwgis.pg.function;
3. import com.fasterxml.jackson.annotation.JsonClassDescription;
4. import com.fasterxml.jackson.annotation

复制代码