MCP：图片搜索服务
This commit is contained in:
parent
497bb33098
commit
03c70db3ab
2
long-image-search-mcp-server/.gitattributes
vendored
Normal file
2
long-image-search-mcp-server/.gitattributes
vendored
Normal file
@ -0,0 +1,2 @@
|
||||
/mvnw text eol=lf
|
||||
*.cmd text eol=crlf
|
||||
33
long-image-search-mcp-server/.gitignore
vendored
Normal file
33
long-image-search-mcp-server/.gitignore
vendored
Normal file
@ -0,0 +1,33 @@
|
||||
HELP.md
|
||||
target/
|
||||
.mvn/wrapper/maven-wrapper.jar
|
||||
!**/src/main/**/target/
|
||||
!**/src/test/**/target/
|
||||
|
||||
### STS ###
|
||||
.apt_generated
|
||||
.classpath
|
||||
.factorypath
|
||||
.project
|
||||
.settings
|
||||
.springBeans
|
||||
.sts4-cache
|
||||
|
||||
### IntelliJ IDEA ###
|
||||
.idea
|
||||
*.iws
|
||||
*.iml
|
||||
*.ipr
|
||||
|
||||
### NetBeans ###
|
||||
/nbproject/private/
|
||||
/nbbuild/
|
||||
/dist/
|
||||
/nbdist/
|
||||
/.nb-gradle/
|
||||
build/
|
||||
!**/src/main/**/build/
|
||||
!**/src/test/**/build/
|
||||
|
||||
### VS Code ###
|
||||
.vscode/
|
||||
84
long-image-search-mcp-server/pom.xml
Normal file
84
long-image-search-mcp-server/pom.xml
Normal file
@ -0,0 +1,84 @@
|
||||
<?xml version="1.0" encoding="UTF-8"?>
|
||||
<project xmlns="http://maven.apache.org/POM/4.0.0" xmlns:xsi="http://www.w3.org/2001/XMLSchema-instance"
|
||||
xsi:schemaLocation="http://maven.apache.org/POM/4.0.0 https://maven.apache.org/xsd/maven-4.0.0.xsd">
|
||||
<modelVersion>4.0.0</modelVersion>
|
||||
<parent>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-parent</artifactId>
|
||||
<version>3.5.0</version>
|
||||
<relativePath/> <!-- lookup parent from repository -->
|
||||
</parent>
|
||||
<groupId>com.huangge1199</groupId>
|
||||
<artifactId>long-image-search-mcp-server</artifactId>
|
||||
<version>0.0.1-SNAPSHOT</version>
|
||||
<name>long-image-search-mcp</name>
|
||||
<description>long-image-search-mcp-server</description>
|
||||
<url/>
|
||||
<licenses>
|
||||
<license/>
|
||||
</licenses>
|
||||
<developers>
|
||||
<developer/>
|
||||
</developers>
|
||||
<scm>
|
||||
<connection/>
|
||||
<developerConnection/>
|
||||
<tag/>
|
||||
<url/>
|
||||
</scm>
|
||||
<properties>
|
||||
<java.version>21</java.version>
|
||||
</properties>
|
||||
<dependencies>
|
||||
<dependency>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-starter-test</artifactId>
|
||||
<scope>test</scope>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.projectlombok</groupId>
|
||||
<artifactId>lombok</artifactId>
|
||||
<optional>true</optional>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>cn.hutool</groupId>
|
||||
<artifactId>hutool-all</artifactId>
|
||||
<version>5.8.37</version>
|
||||
</dependency>
|
||||
<dependency>
|
||||
<groupId>org.springframework.ai</groupId>
|
||||
<artifactId>spring-ai-mcp-server-webmvc-spring-boot-starter</artifactId>
|
||||
<version>1.0.0-M6</version>
|
||||
</dependency>
|
||||
</dependencies>
|
||||
|
||||
<build>
|
||||
<plugins>
|
||||
<plugin>
|
||||
<groupId>org.apache.maven.plugins</groupId>
|
||||
<artifactId>maven-compiler-plugin</artifactId>
|
||||
<configuration>
|
||||
<annotationProcessorPaths>
|
||||
<path>
|
||||
<groupId>org.projectlombok</groupId>
|
||||
<artifactId>lombok</artifactId>
|
||||
</path>
|
||||
</annotationProcessorPaths>
|
||||
</configuration>
|
||||
</plugin>
|
||||
<plugin>
|
||||
<groupId>org.springframework.boot</groupId>
|
||||
<artifactId>spring-boot-maven-plugin</artifactId>
|
||||
<configuration>
|
||||
<excludes>
|
||||
<exclude>
|
||||
<groupId>org.projectlombok</groupId>
|
||||
<artifactId>lombok</artifactId>
|
||||
</exclude>
|
||||
</excludes>
|
||||
</configuration>
|
||||
</plugin>
|
||||
</plugins>
|
||||
</build>
|
||||
|
||||
</project>
|
||||
@ -0,0 +1,24 @@
|
||||
package com.huangge1199.longimagesearchmcp;
|
||||
|
||||
import com.huangge1199.longimagesearchmcp.tools.ImageSearchTool;
|
||||
import org.springframework.ai.tool.ToolCallbackProvider;
|
||||
import org.springframework.ai.tool.method.MethodToolCallbackProvider;
|
||||
import org.springframework.boot.SpringApplication;
|
||||
import org.springframework.boot.autoconfigure.SpringBootApplication;
|
||||
import org.springframework.context.annotation.Bean;
|
||||
|
||||
@SpringBootApplication
|
||||
public class LongImageSearchMcpApplication {
|
||||
|
||||
public static void main(String[] args) {
|
||||
SpringApplication.run(LongImageSearchMcpApplication.class, args);
|
||||
}
|
||||
|
||||
@Bean
|
||||
public ToolCallbackProvider imageSearchTools(ImageSearchTool imageSearchTool) {
|
||||
return MethodToolCallbackProvider.builder()
|
||||
.toolObjects(imageSearchTool)
|
||||
.build();
|
||||
}
|
||||
|
||||
}
|
||||
@ -0,0 +1,74 @@
|
||||
package com.huangge1199.longimagesearchmcp.tools;
|
||||
|
||||
import cn.hutool.core.util.StrUtil;
|
||||
import cn.hutool.http.HttpUtil;
|
||||
import cn.hutool.json.JSONObject;
|
||||
import cn.hutool.json.JSONUtil;
|
||||
import org.springframework.ai.tool.annotation.Tool;
|
||||
import org.springframework.ai.tool.annotation.ToolParam;
|
||||
import org.springframework.stereotype.Service;
|
||||
|
||||
import java.util.Collections;
|
||||
import java.util.HashMap;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.stream.Collectors;
|
||||
|
||||
/**
|
||||
* ImageSearchTool
|
||||
*
|
||||
* @author huangge1199
|
||||
* @since 2025/6/4 15:48:11
|
||||
*/
|
||||
@Service
|
||||
public class ImageSearchTool {
|
||||
|
||||
// 替换为你的 Pexels API 密钥（需从官网申请）
|
||||
private static final String API_KEY = "你的api-key";
|
||||
|
||||
// Pexels 常规搜索接口（请以文档为准）
|
||||
private static final String API_URL = "https://api.pexels.com/v1/search";
|
||||
|
||||
@Tool(description = "search image from web")
|
||||
public String searchImage(@ToolParam(description = "Search query keyword") String query) {
|
||||
try {
|
||||
return String.join(",", searchMediumImages(query));
|
||||
} catch (Exception e) {
|
||||
return "Error search image: " + e.getMessage();
|
||||
}
|
||||
}
|
||||
|
||||
/**
|
||||
* 搜索中等尺寸的图片列表
|
||||
*
|
||||
* @param query
|
||||
* @return
|
||||
*/
|
||||
public List<String> searchMediumImages(String query) {
|
||||
// 设置请求头（包含API密钥）
|
||||
Map<String, String> headers = new HashMap<>();
|
||||
headers.put("Authorization", API_KEY);
|
||||
|
||||
// 设置请求参数（仅包含query，可根据文档补充page、per_page等参数）
|
||||
Map<String, Object> params = new HashMap<>();
|
||||
params.put("query", query);
|
||||
|
||||
// 发送 GET 请求
|
||||
String response = HttpUtil.createGet(API_URL)
|
||||
.addHeaders(headers)
|
||||
.form(params)
|
||||
.execute()
|
||||
.body();
|
||||
|
||||
// 解析响应JSON（假设响应结构包含"photos"数组，每个元素包含"medium"字段）
|
||||
return JSONUtil.parseObj(response)
|
||||
.getJSONArray("photos")
|
||||
.stream()
|
||||
.map(photoObj -> (JSONObject) photoObj)
|
||||
.map(photoObj -> photoObj.getJSONObject("src"))
|
||||
.map(photo -> photo.getStr("medium"))
|
||||
.filter(StrUtil::isNotBlank)
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
}
|
||||
|
||||
@ -0,0 +1,9 @@
|
||||
spring:
|
||||
ai:
|
||||
mcp:
|
||||
server:
|
||||
name: long-image-search-mcp-server
|
||||
version: 0.0.1
|
||||
type: SYNC
|
||||
# sse
|
||||
stdio: false
|
||||
@ -0,0 +1,13 @@
|
||||
spring:
|
||||
ai:
|
||||
mcp:
|
||||
server:
|
||||
name: long-image-search-mcp-server
|
||||
version: 0.0.1
|
||||
type: SYNC
|
||||
# stdio
|
||||
stdio: true
|
||||
# stdio
|
||||
main:
|
||||
web-application-type: none
|
||||
banner-mode: off
|
||||
@ -0,0 +1,7 @@
|
||||
spring:
|
||||
application:
|
||||
name: long-image-search-mcp-server
|
||||
profiles:
|
||||
active: stdio
|
||||
server:
|
||||
port: 8127
|
||||
@ -0,0 +1,28 @@
|
||||
package com.huangge1199.longimagesearchmcp.tools;
|
||||
|
||||
import jakarta.annotation.Resource;
|
||||
import org.junit.jupiter.api.Assertions;
|
||||
import org.junit.jupiter.api.Test;
|
||||
import org.springframework.boot.test.context.SpringBootTest;
|
||||
|
||||
import static org.junit.jupiter.api.Assertions.*;
|
||||
|
||||
/**
|
||||
* ImageSearchToolTest
|
||||
*
|
||||
* @author huangge1199
|
||||
* @since 2025/6/4 15:51:37
|
||||
*/
|
||||
@SpringBootTest
|
||||
class ImageSearchToolTest {
|
||||
|
||||
@Resource
|
||||
private ImageSearchTool imageSearchTool;
|
||||
|
||||
@Test
|
||||
void searchImage() {
|
||||
String result = imageSearchTool.searchImage("computer");
|
||||
System.out.println(result);
|
||||
Assertions.assertNotNull(result);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user