# milvus-use

**Repository Path**: alpbeta/milvus-use

## Basic Information

- **Project Name**: milvus-use
- **Description**: milvus-use教程 python
- **Primary Language**: Unknown
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2025-03-06
- **Last Updated**: 2025-03-08

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

## 需求描述
参考vanna项目，获取数据库元数据和问题sql对，存入Milvus向量数据库，之后进行检索，返回相似的数据库表和问题对。本项目采用的嵌入模型为m3e-large。该该项目milvus版本较新，与vanna项目版本不同，大量api不兼容

## 功能介绍
DatabaseManager类对数据库连接持久化，具体的连接信息如host、用户名密码等在.env文件中。现支持mysql，milvus数据库连接；
MilvusVectorStore类封装对milvus数据库操作方法，如创建数据库元数据和问题sql对的collection（类似于mysql中的表），添加数据库元数据和问题对，和基于问题查询collections中最相似的数据。
另通过init_collections、delete_collections、add_db_metadata、add_qa_pairs、api_search文件对collections进行初始化、删除、添加数据库元数据、添加问题对、对输入的问题查询数据库

## 运行(linux环境下)
### 准备工作
下载魔搭
下载m3e-large
```shell
pip install modelscope -i https://pypi.tuna.tsinghua.edu.cn/simple
modelscope download --model 'AI-ModelScope/m3e-large' --local_dir '/home/models/m3e-large'
```

下载miniconda
```shell
mkdir -p ~/miniconda3 
wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh -O ~/miniconda3/miniconda.sh
bash ~/miniconda3/miniconda.sh -b -u -p ~/miniconda3
rm ~/miniconda3/miniconda.sh
#配置环境变量
ls ~/miniconda3 #查看是否有安装文件
export PATH="$HOME/miniconda3/bin:$PATH" #配置路径
source ~/.bashrc #刷新配置
conda --version #版本
```

### 运行
拉取gitee仓库的代码
git clone https://gitee.com/alpbeta/milvus-use.git

创建、激活环境
```shell
conda create -n milvus python=3.10
conda activate milvus
```

打开目录，安装依赖
```shell
cd /home/milvus/milvus-use
pip install -r requirements.txt
```

启动api_search进行查询
```shell
cd /home/milvus/milvus-use/
python -m milvus_vector.api_search
```