使用说明
具有LLM集成的文件搜索助手
该项目将语义搜索功能与大型语言模型(LLM)相结合,在Linux系统中提供智能文件搜索功能。
特性
- 自然语言文件搜索查询
- 基于BERT嵌入的语义搜索
- 文件系统与MCP服务器集成
- Gemini LLM集成用于查询理解
- 自动推断文件扩展名
先决条件
- Python 3.8或更高版本
- CUDA兼容GPU(可选,处理速度更快)
- Linux 操作系统
安装
- 克隆存储库:
git clone <repository-url>
cd LLm_To_agent/data_dir
- 安装依赖项:
pip install -r requirements.txt
- 配置环境变量:
- 创建一个
.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
用法
- 启动应用程序:
python main.py
- 出现提示时输入您的搜索查询:
What file are you looking for?
> "find a python script that handles file operations"
- 该系统将:
- 使用LLM提取相关文件名
- 使用语义搜索搜索文件系统
- 显示匹配的文件位置
详细功能
- LLM集成:使用谷歌的Gemini进行自然语言理解
- 语义搜索:采用BERT嵌入进行上下文感知文件匹配
- MCP服务器:高效处理文件系统操作
- 可扩展架构:易于添加新的搜索功能
故障排除
如果遇到错误“未找到嵌入文件”,请运行:
cd ~ && find / -type f 2>/dev/null >> Desktop/LLm_To_agent
许可证
[您选择的许可证]
贡献
[您的贡献指南]
