package com.ard.agent.tool;

import jakarta.annotation.Resource;
import lombok.extern.slf4j.Slf4j;
import org.springframework.ai.document.Document;
import org.springframework.ai.tool.annotation.Tool;
import org.springframework.ai.tool.annotation.ToolParam;
import org.springframework.ai.vectorstore.SearchRequest;
import org.springframework.ai.vectorstore.VectorStore;
import org.springframework.stereotype.Component;

import java.util.List;
import java.util.stream.Collectors;

/**
 * 知识库工具
 *
 * @author lsy
 * @date 2026/5/11
 */

@Component
@Slf4j
public class KnowledgeTools {
    
    @Resource
    private VectorStore vectorStore;  // 你的向量数据库
    
    @Tool(description = """
        从内部知识库中检索相关文档。当你需要查询公司政策、技术文档、产品说明等静态知识时，
        应该优先调用此工具。此工具适用于非实时、非动态的数据查询。
        """)
    public String searchKnowledgeBase(
            @ToolParam(description = "搜索关键词或问题") String query) {
        log.info("AI调用searchKnowledgeBase工具，参数: query={}", query);
        // 执行向量检索
        List<Document> results = vectorStore.similaritySearch(
            SearchRequest.builder().query(query) .topK(3)
                    .build()
        );

        if (results.isEmpty()) {
            return "未找到相关知识";
        }
        
        return results.stream()
            .map(Document::getText)
            .collect(Collectors.joining("\n---\n"));
    }
}