From 4ea26c7e81cf56eb2b5c695352d236bab2eb5157 Mon Sep 17 00:00:00 2001 From: cherishsince Date: Sun, 2 Mar 2025 15:03:14 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90=E4=BC=98=E5=8C=96=E3=80=91=E5=A2=9E?= =?UTF-8?q?=E5=8A=A0=E6=B3=A8=E9=87=8A?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ai/service/WebSearchServiceTests.java | 20 +++++++++++++++++++ 1 file changed, 20 insertions(+) diff --git a/yudao-module-ai/yudao-module-ai-biz/src/test/java/cn/iocoder/yudao/module/ai/service/WebSearchServiceTests.java b/yudao-module-ai/yudao-module-ai-biz/src/test/java/cn/iocoder/yudao/module/ai/service/WebSearchServiceTests.java index fe97c9dfd1..a929de4463 100644 --- a/yudao-module-ai/yudao-module-ai-biz/src/test/java/cn/iocoder/yudao/module/ai/service/WebSearchServiceTests.java +++ b/yudao-module-ai/yudao-module-ai-biz/src/test/java/cn/iocoder/yudao/module/ai/service/WebSearchServiceTests.java @@ -9,30 +9,50 @@ import org.junit.jupiter.api.Test; import java.util.List; import java.util.Map; +/** + * 网络搜索服务测试类 + * 用于测试网页爬取和谷歌搜索功能 + */ public class WebSearchServiceTests { + /** + * 测试网页爬取功能 + * 爬取指定URL的网页内容并打印结果 + */ @Test public void webCrawlerTest() { + // 创建网络搜索服务实例 WebSearchServiceImpl webSearchService = new WebSearchServiceImpl(); + // 调用网页爬取方法,获取长沙天气页面内容 Map webCrawlerRes = webSearchService.webCrawler( Lists.newArrayList("https://tianqi.eastday.com/changsha/40/")); + // 遍历并打印爬取结果 for (Map.Entry entry : webCrawlerRes.entrySet()) { System.err.println(entry.getValue()); } } + /** + * 测试谷歌搜索功能 + * 搜索指定关键词,获取搜索结果并爬取对应网页内容 + */ @Test public void googleSearchTest() { + // 创建网络搜索服务实例 WebSearchServiceImpl webSearchService = new WebSearchServiceImpl(); + // 调用谷歌搜索方法,搜索"长沙今天天气",限制返回6条结果 List webSearchRespList = webSearchService.googleSearch("长沙今天天气", 6); + // 从搜索结果中提取URL,并爬取对应网页内容 Map webCrawlerRespMap = webSearchService.webCrawler(webSearchRespList.stream().map(WebSearchRespVO::getUrl).toList()); + // 打印搜索结果 for (WebSearchRespVO webSearchRespVO : webSearchRespList) { System.err.println(JSON.toJSONString(webSearchRespVO)); } + // 打印爬取的网页内容 for (Map.Entry entry : webCrawlerRespMap.entrySet()) { System.err.println("url:" + entry.getKey()); System.err.println("value" + entry.getValue());