智能文件搜索工具

一个结合语义搜索和大语言模型的智能文件搜索工具,适用于Linux系统,提供自然语言查询和上下文感知的文件匹配功能。

  • by Code-Trees
  • Python
  • 本地部署
  • 更新于
在 GitHub 查看 code-trees/agentic_search

使用说明

具有LLM集成的文件搜索助手

该项目将语义搜索功能与大型语言模型(LLM)相结合,在Linux系统中提供智能文件搜索功能。

特性

  • 自然语言文件搜索查询
  • 基于BERT嵌入的语义搜索
  • 文件系统与MCP服务器集成
  • Gemini LLM集成用于查询理解
  • 自动推断文件扩展名

先决条件

  • Python 3.8或更高版本
  • CUDA兼容GPU(可选,处理速度更快)
  • Linux 操作系统

安装

  1. 克隆存储库:
git clone <repository-url>
cd LLm_To_agent/data_dir
  1. 安装依赖项:
pip install -r requirements.txt
  1. 配置环境变量:
    • 创建一个 .env 项目根目录中的文件
    • 添加您的Gemini API密钥:
    GEMINI_API_KEY="your-api-key-here"
    

项目结构

data_dir/
├── main.py              # Main application entry point
├── tools/
│   └── file_finder.py   # File search implementation
├── utils/
│   └── doc_Search.py    # Document search utilities
├── llm/
│   └── gemini_client.py # LLM integration
├── .env                 # Environment variables
└── OSData_store.pth     # Embedded data storage

用法

  1. 启动应用程序:
python main.py
  1. 出现提示时输入您的搜索查询:
What file are you looking for?
> "find a python script that handles file operations"
  1. 该系统将:
    • 使用LLM提取相关文件名
    • 使用语义搜索搜索文件系统
    • 显示匹配的文件位置

详细功能

  • LLM集成:使用谷歌的Gemini进行自然语言理解
  • 语义搜索:采用BERT嵌入进行上下文感知文件匹配
  • MCP服务器:高效处理文件系统操作
  • 可扩展架构:易于添加新的搜索功能

故障排除

如果遇到错误“未找到嵌入文件”,请运行:

cd ~ && find / -type f 2>/dev/null >> Desktop/LLm_To_agent

许可证

[您选择的许可证]

贡献

[您的贡献指南]