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

引入依赖

注意引入 LangChain4j-tavily 依赖：

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

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

配置 WebSearchEngine

新建 WebSearchConfig 后，可以使用 @Value 注解将配置注入。

@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,
    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);
}