SpringBoot 整合 LangChain4j 与 Tavily 实现联网搜索及 API Key 获取

引入依赖

确保项目中引入了对应的 Web Search Engine 依赖：

<dependency>
  <groupId>dev.langchain4j</groupId>
  <artifactId>langchain4j-web-search-engine-tavily</artifactId>
  <version>1.0.0-beta3</version>
</dependency>

网络搜索引擎可以作为 RAG 管道中的 ContentRetriever 使用，也可以作为 @Tool 使用。LangChain4j 支持的检索图途径主要有以下几种：

配置 Bean

新建 WebSearchConfig 类，配置联网搜索需要的 Bean。可以使用 @Value 注解将配置注入。

@Value("${langchain4j.web-search-engine.tavily.api-key}")
private String tavilyApiKey;

@Value("${langchain4j.web-search-engine.tavily.max-results:3}")
private Integer maxResults;

随后完成 WebSearchEngine 类，创建集中管理 Tavily 搜索引擎配置，统一 API 密钥等参数：

@Bean
public WebSearchEngine webSearchEngine() {
    return TavilyWebSearchEngine.builder()
            .apiKey(tavilyApiKey)
            .build();
}

接着就是完成 ContentRetriever 内容检索器接口，核心作用是从数据源（如网络、向量库）中，根据查询语义精准召回相关内容片段，为大模型提供上下文支持（适配 RAG 场景）。也可以专门通过 WebSearchEngine（如 Tavily）从网络检索相关内容，返回给大模型辅助回答。

@Bean
public ContentRetriever webContentRetriever(WebSearchEngine webSearchEngine) {
    return WebSearchContentRetriever.builder()
            .webSearchEngine(webSearchEngine)
            .maxResults(maxResults)
            .build();
}

完整代码示例如下：

package com.example.kui.config;

import dev.langchain4j.rag.content.retriever.ContentRetriever;
import dev.langchain4j.rag.content.retriever.WebSearchContentRetriever;
import dev.langchain4j.web.search.WebSearchEngine;
import dev.langchain4j.web.search.tavily.TavilyWebSearchEngine;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;

@Configuration
public class WebSearchConfig {

    @Value("${langchain4j.web-search-engine.tavily.api-key}")
    private String tavilyApiKey;

    @Value("${langchain4j.web-search-engine.tavily.max-results:3}")
    private Integer maxResults;

    @Bean
    public WebSearchEngine webSearchEngine() {
        return TavilyWebSearchEngine.builder()
                .apiKey(tavilyApiKey)
                .build();
    }

    @Bean
    public ContentRetriever webContentRetriever(WebSearchEngine webSearchEngine) {
        return WebSearchContentRetriever.builder()
                .webSearchEngine(webSearchEngine)
                .maxResults(maxResults)
                .build();
    }
}

创建 Agent 接口

我们可以创建一个专门用于网络检索的 Agent 接口，命名为 WebSearchAgent。借助 @AiService 注解配置好我们的联网搜索的 ContentRetriever，名称应该与类名相同。

@AiService(wiringMode = EXPLICIT, 
           /* 模型，会话记忆等其他配置 */ 
           contentRetriever = "webContentRetriever")

在接口中实现方法：

@SystemMessage("根据检索结果回答")
String chat(@UserMessage String message);

完整代码：

@AiService(wiringMode = EXPLICIT, 
           chatModel = "openAiChatModel", 
           streamingChatModel = "openAiStreamingChatModel", 
           chatMemory = "chatMemory", 
           chatMemoryProvider = "chatMemoryProvider", 
           contentRetriever = "webContentRetriever")
public interface WebSearchAgent {
    @SystemMessage("根据检索结果回答")
    String chat(@UserMessage String message);
}

此时我们只需要调用 WebSearchAgent 的 chat 方法就能得到网络搜索结果了。

@PostMapping("/search")
public String webSearch(@RequestBody String userMessage) {
    return webSearchAgent.chat(userMessage);
}