【源码】基于Python的无需休眠检索增强生成系统

项目简介

本项目致力于解决检索增强生成（RAG）过程中 time.sleep 的问题。通过任务细分、图像和文本嵌入、向量数据库以及图生成技术，达成了高效且准确的 RAG 模型，构建向量数据库时采用了 coco 部分数据集。

项目的主要特性和功能

特性

• 任务细分：借助 GPT - 4o 模型把复杂任务细化成可管理的小任务。
• 多模态嵌入：利用 CLIP 实现图像和文本的嵌入。
• 高效存储检索：使用 Qdrant 构建高效的向量数据库，用于存储和检索嵌入数据。
• 图生图技术：运用 Stable Diffusion 从给定图像生成新图像。

功能

实现高效准确的检索增强生成功能，避免传统 RAG 过程中不必要的休眠操作。

安装使用步骤

安装步骤

1. 复制仓库： bash cd NoSleepRAG
2. 安装依赖： bash pip3 install -r requirements.txt
3. 下载 coco 数据集： bash python3 download_dataset.py
4. 启动 Qdrant： bash docker pull qdrant/qdrant docker run -p 6333:6333 qdrant/qdrant
5. 配置环境变量： bash export OPENAI_API_KEY="your api key"

使用方法

1. 初始化向量数据库： bash python3 QdrantDatabase.py
2. 执行 RAG： bash python3 main.py

下载地址

点击下载 【提取码: 4003】【解压密码: www.makuang.net】