搜索模块 - 高级特性
1. 多数据源聚合搜索
同时搜索多个数据源并聚合结果：
# Agent 调用多源搜索
results = search.multi_source(
query="订单处理流程",
sources=["PROJECT", "KNOWLEDGE", "EXPERIENCE"],
top_k_per_source=5
)
# 结果按数据源分组
for source, items in results.items():
print(f"=== {source} ===")
for item in items:
print(f" - {item['title']}: {item['score']}")
2. 向量搜索集成
与 Spring AI VectorStore 集成：
import org.springframework.ai.vectorstore.VectorStore;
import org.springframework.ai.vectorstore.SearchRequest;
@Component
public class VectorSearchProvider implements SearchProvider {
private final VectorStore vectorStore;
@Override
public boolean supports(SearchSourceType type) {
return type == SearchSourceType.KNOWLEDGE;
}
@Override
public List<SearchResultItem> search(com.alibaba.assistant.agent.extension.search.model.SearchRequest request) {
// 使用向量搜索
List<Document> docs = vectorStore.similaritySearch(
SearchRequest.query(request.getQuery())
.withTopK(request.getTopK())
.withSimilarityThreshold(0.7)
);
return docs.stream()
.map(this::toSearchResultItem)
.collect(Collectors.toList());
}
}
3. Web 搜索
启用并配置 Web 搜索：
spring:
ai:
alibaba:
codeact:
extension:
search:
web-search-enabled: true
baidu-api-key: ${BAIDU_SEARCH_API_KEY}
自定义 Web 搜索提供者：
@Component
public class BaiduSearchProvider implements SearchProvider {
private final String apiKey;
private final RestTemplate restTemplate;
@Override
public boolean supports(SearchSourceType type) {
return type == SearchSourceType.WEB;
}
@Override
public List<SearchResultItem> search(SearchRequest request) {
// 调用百度搜索 API
String url = String.format(
"https://api.baidu.com/search?key=%s&q=%s",
apiKey,
URLEncoder.encode(request.getQuery(), StandardCharsets.UTF_8)
);
BaiduSearchResponse response = restTemplate.getForObject(url, BaiduSearchResponse.class);
return response.getResults().stream()
.map(r -> {
SearchResultItem item = new SearchResultItem();
item.setTitle(r.getTitle());
item.setContent(r.getSnippet());
item.setUri(r.getUrl());
item.setSourceType(SearchSourceType.WEB);
return item;
})
.collect(Collectors.toList());
}
}
4. 项目上下文搜索
搜索当前项目的代码和配置：
@Component
public class ProjectSearchProvider implements SearchProvider {
private final ProjectIndexer projectIndexer;
@Override
public boolean supports(SearchSourceType type) {
return type == SearchSourceType.PROJECT;
}
@Override
public List<SearchResultItem> search(SearchRequest request) {
String projectId = (String) request.getFilters().get("projectId");
// 搜索代码索引
List<CodeSnippet> snippets = projectIndexer.search(
projectId,
request.getQuery(),
request.getTopK()
);
return snippets.stream()
.map(s -> {
SearchResultItem item = new SearchResultItem();
item.setTitle(s.getFilePath());
item.setContent(s.getCode());
item.setSourceType(SearchSourceType.PROJECT);
item.setScore(s.getScore());
// 设置元数据
SearchMetadata metadata = new SearchMetadata();
metadata.setLanguage(s.getLanguage());
metadata.getExtensions().put("lineNumber", s.getLineNumber());
item.setMetadata(metadata);
return item;
})
.collect(Collectors.toList());
}
}
5. 搜索结果缓存
@Component
public class CachedSearchProvider implements SearchProvider {
private final SearchProvider delegate;
private final Cache<String, List<SearchResultItem>> cache;
@Override
public List<SearchResultItem> search(SearchRequest request) {
String cacheKey = generateCacheKey(request);
return cache.get(cacheKey, () -> delegate.search(request));
}
private String generateCacheKey(SearchRequest request) {
return request.getSourceType() + ":" +
request.getQuery() + ":" +
request.getTopK();
}
}
6. 搜索结果排序
自定义排序策略：
@Component
public class SearchResultSorter {
public List<SearchResultItem> sort(List<SearchResultItem> items, String strategy) {
switch (strategy) {
case "score":
return items.stream()
.sorted(Comparator.comparingDouble(SearchResultItem::getScore).reversed())
.collect(Collectors.toList());
case "recency":
return items.stream()
.sorted(Comparator.comparing(i ->
(Instant) i.getMetadata().get("updatedAt"),
Comparator.reverseOrder()))
.collect(Collectors.toList());
default:
return items;
}
}
}
7. 搜索过滤器
@Component
public class SearchFilterChain {
private final List<SearchFilter> filters;
public List<SearchResultItem> filter(List<SearchResultItem> items, SearchContext context) {
List<SearchResultItem> result = items;
for (SearchFilter filter : filters) {
result = filter.apply(result, context);
}
return result;
}
}
// 权限过滤器示例
@Component
public class PermissionFilter implements SearchFilter {
@Override
public List<SearchResultItem> apply(List<SearchResultItem> items, SearchContext context) {
String userId = context.getUserId();
return items.stream()
.filter(item -> hasPermission(userId, item))
.collect(Collectors.toList());
}
}
8. 搜索上下文
SearchContext context = SearchContext.builder()
.userId("user-123")
.projectId("project-456")
.sessionId("session-789")
.metadata(Map.of(
"language", "java",
"framework", "spring-boot"
))
.build();
SearchRequest request = new SearchRequest();
request.setQuery("配置文件加载");
request.setContext(context);
9. 自定义搜索工具
为 Agent 提供自定义搜索工具：
@Component
public class DatabaseSearchTool implements CodeactTool {
private final JdbcTemplate jdbcTemplate;
@Override
public String call(String toolInput) {
Map<String, Object> params = parseInput(toolInput);
String table = (String) params.get("table");
String keyword = (String) params.get("keyword");
String sql = "SELECT * FROM " + table + " WHERE content LIKE ?";
List<Map<String, Object>> results = jdbcTemplate.queryForList(
sql, "%" + keyword + "%"
);
return toJson(results);
}
@Override
public CodeactToolDefinition getCodeactDefinition() {
return DefaultCodeactToolDefinition.builder()
.name("search_database")
.description("搜索数据库表")
.parameterTree(ParameterTree.builder()
.addParameter(ParameterNode.builder()
.name("table")
.type(ParameterType.STRING)
.required(true)
.build())
.addParameter(ParameterNode.builder()
.name("keyword")
.type(ParameterType.STRING)
.required(true)
.build())
.build())
.build();
}
@Override
public CodeactToolMetadata getCodeactMetadata() {
return DefaultCodeactToolMetadata.builder()
.targetClassName("search")
.build();
}
}
10. 搜索指标监控
@Component
public class SearchMetrics {
private final MeterRegistry meterRegistry;
public void recordSearch(SearchSourceType type, long durationMs, int resultCount) {
meterRegistry.timer("search.duration", "source", type.name())
.record(Duration.ofMillis(durationMs));
meterRegistry.counter("search.results", "source", type.name())
.increment(resultCount);
}
}
11. 日志配置
logging:
level:
com.alibaba.assistant.agent.extension.search: DEBUG