Spring AI学习：工具调用实践（基于和风天气api）

工具调用总体实现：

设计工具类：

1. @Component
2. public class WeatherInquiryTools {
4.     @Autowired
5.     private WeatherService weatherService;
7.     @Tool(description = "根据城市名称查询城市LocationID")
8.     public String getLocationId(@ToolParam(description = "城市名称") String cityName){
9.         return weatherService.getLocationId(cityName).getLocation().get(0).getId();
10.     }
12.     @Tool(description = "根据城市LocationID查询实时温度")
13.     public String getWeather(@ToolParam(description = "城市LocationID") String locationId){
14.         return weatherService.getWeather(locationId).getNow().getTemp();
15.     }
16. }

复制代码

将工具交给ai客户端：

 对话测试：

 具体实现：

实体类：

1. @NoArgsConstructor
2. @AllArgsConstructor
3. @Data
4. public class WeatherResponse {
5.     @JsonProperty("code")
6.     private String code;
8.     @JsonProperty("updateTime")
9.     private String updateTime;
11.     @JsonProperty("fxLink")
12.     private String fxLink;
14.     @JsonProperty("now")
15.     private NowData now;
17.     @JsonProperty("refer")
18.     private Refer refer;
19. }

复制代码

1. @Data
2. @AllArgsConstructor
3. @NoArgsConstructor
4. public class NowData {
6.     // 观测时间
7.     @JsonProperty("obsTime")
8.     private String obsTime;
10.     // 温度，单位：摄氏度
11.     @JsonProperty("temp")
12.     private String temp;
14.     // 体感温度，单位：摄氏度
15.     @JsonProperty("feelsLike")
16.     private String feelsLike;
18.     // 天气图标代码
19.     @JsonProperty("icon")
20.     private String icon;
22.     // 天气描述
23.     @JsonProperty("text")
24.     private String text;
26.     // 风向360角度
27.     @JsonProperty("wind360")
28.     private String wind360;
30.     // 风向
31.     @JsonProperty("windDir")
32.     private String windDir;
34.     // 风力等级
35.     @JsonProperty("windScale")
36.     private String windScale;
38.     // 风速，单位：公里/小时
39.     @JsonProperty("windSpeed")
40.     private String windSpeed;
42.     // 相对湿度，单位：%
43.     @JsonProperty("humidity")
44.     private String humidity;
46.     // 降水量，单位：毫米
47.     @JsonProperty("precip")
48.     private String precip;
50.     // 大气压强，单位：百帕
51.     @JsonProperty("pressure")
52.     private String pressure;
54.     // 能见度，单位：公里
55.     @JsonProperty("vis")
56.     private String vis;
58.     // 云量，单位：%
59.     @JsonProperty("cloud")
60.     private String cloud;
62.     //露点温度，单位：摄氏度
63.     @JsonProperty("dew")
64.     private String dew;
65. }

复制代码

1. @NoArgsConstructor
2. @AllArgsConstructor
3. @Data
4. public class LocationResponse {
6.     @JsonProperty("code")
7.     private String code;
9.     @JsonProperty("location")
10.     private List<Location> location;
12.     @JsonProperty("refer")
13.     private Refer refer;
14. }

复制代码

1. @AllArgsConstructor
2. @NoArgsConstructor
3. @Data
4. public class Location {
6.     @JsonProperty("name")
7.     private String name;
9.     @JsonProperty("id")
10.     private String id;
12.     @JsonProperty("lat")
13.     private String lat;
15.     @JsonProperty("lon")
16.     private String lon;
18.     @JsonProperty("adm2")
19.     private String adm2;
21.     @JsonProperty("adm1")
22.     private String adm1;
24.     @JsonProperty("country")
25.     private String country;
27.     @JsonProperty("tz")
28.     private String timezone;
30.     @JsonProperty("utcOffset")
31.     private String utcOffset;
33.     @JsonProperty("isDst")
34.     private String isDst;
36.     @JsonProperty("type")
37.     private String type;
39.     @JsonProperty("rank")
40.     private String rank;
42.     @JsonProperty("fxLink")
43.     private String fxLink;
44. }

复制代码

1. @NoArgsConstructor
2. @AllArgsConstructor
3. @Data
4. public class Refer {
6.     @JsonProperty("sources")
7.     private List<String> sources;
9.     @JsonProperty("license")
10.     private List<String> license;
11. }

复制代码

以上实体类基于 和风天气api 设计。
配置类：

1. @Component
2. @Data
3. @ConfigurationProperties(prefix = "frog.weather")
4. public class WeatherProperties {
6.     private String apiKey;
8.     private String apiHost;
9. }<br>

复制代码

以上配置类基于 和风天气api 设计。
序列化与反序列化工具类：

1. @Component
2. public class JasonUtils {
4.     private static final ObjectMapper mapper = new ObjectMapper()
5.             .configure(DeserializationFeature.FAIL_ON_UNKNOWN_PROPERTIES, false)
6.             .setSerializationInclusion(JsonInclude.Include.NON_NULL)
7.             .registerModule(new JavaTimeModule());
9.     public static String toJson(Object obj) {
10.         try {
11.             return mapper.writeValueAsString(obj);
12.         } catch (JsonProcessingException e) {
13.             throw new RuntimeException("序列化失败", e);
14.         }
15.     }
17.     public static <T> T fromJson(String json, Class<T> clazz) {
18.         try {
19.             return mapper.readValue(json, clazz);
20.         } catch (JsonProcessingException e) {
21.             throw new RuntimeException("反序列化失败", e);
22.         }
23.     }
24. }

复制代码

查询功能具体实现：

1. @Service
2. @Slf4j
3. public class WeatherServiceImpl implements WeatherService {
5.     @Autowired
6.     private WeatherProperties weatherProperties;
8.     /**
9.      * 根据城市LocationID查询实时天气状况
10.      */
11.     @Override
12.     public WeatherResponse getWeather(String locationId) {
13.         HttpClient client = HttpClient.newHttpClient();
14.         String apiUrl = weatherProperties.getApiHost() + "/v7/weather/now?location=" + locationId;
15.         try {
16.             HttpRequest request = HttpRequest.newBuilder()
17.                     .uri(URI.create(apiUrl))
18.                     .header("X-QW-Api-Key", weatherProperties.getApiKey())
19.                     .GET()
20.                     .build();
21.             HttpResponse<byte[]> response = client.send(request, HttpResponse.BodyHandlers.ofByteArray());
22.             byte[] compressedBody = response.body();
23.             // 解压缩响应体
24.             String decompressedBody = GzipDecompressor.decompressToString(compressedBody);
25.             log.info("查询天气成功, locationId: {}", locationId);
26.             WeatherResponse weatherResponse = JasonUtils.fromJson(decompressedBody, WeatherResponse.class);
27.             return weatherResponse;
28.         } catch (Exception e) {
29.             log.error("查询天气失败, locationId: {}", locationId, e);
30.             return null;
31.         }
32.     }
34.     /**
35.      * 根据城市名称查询城市LocationID
36.      */
37.     @Override
38.     public LocationResponse getLocationId(String cityName) {
39.         HttpClient client = HttpClient.newHttpClient();
40.         String apiUrl = weatherProperties.getApiHost() + "/geo/v2/city/lookup?location=" + cityName;
41.         try {
42.             HttpRequest request = HttpRequest.newBuilder()
43.                     .uri(URI.create(apiUrl))
44.                     .header("X-QW-Api-Key", weatherProperties.getApiKey())
45.                     .GET()
46.                     .build();
47.             HttpResponse<byte[]> response = client.send(request, HttpResponse.BodyHandlers.ofByteArray());
48.             byte[] compressedBody = response.body();
49.             // 解压缩响应体
50.             String decompressedBody = GzipDecompressor.decompressToString(compressedBody);
51.             log.info("查询城市LocationID成功, cityName: {}", cityName);
52.             LocationResponse locationResponse = JasonUtils.fromJson(decompressedBody, LocationResponse.class);
53.             return locationResponse;
54.         } catch (Exception e) {
55.             log.error("查询城市LocationID失败, cityName: {}", cityName, e);
56.             return null;
57.         }
58.     }
59. }

复制代码

目前实现的tool查询功能局限温度查询，但根据工具需求可拓展（service返回了所有查询所得，tool内对数据重新整理可以产生丰富功能）。

来源：程序园用户自行投稿发布，如果侵权，请联系站长删除
免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！

喜欢鼓捣这些软件，现在用得少，谢谢分享！

过来提前占个楼

yyds。多谢分享

东西不错很实用谢谢分享

鼓励转贴优秀软件安全工具和文档！

用心讨论，共获提升！

前排留名，哈哈哈

收藏一下   不知道什么时候能用到

热心回复！

这个有用。

感谢发布原创作品，程序园因你更精彩

前排留名，哈哈哈

感谢分享，学习下。

东西不错很实用谢谢分享

这个好，看起来很实用

很好很强大  我过来先占个楼 待编辑

这个好，看起来很实用