安装下面的依赖
pip install -U pymilvus -i https://pypi.tuna.tsinghua.edu.cn/simple
pip install "pymilvus[model]"
下面可以运行milvus-lite版本作为简单的学习
运行程序
from pymilvus import MilvusClient
from pymilvus import model
# 不提供镜像就会报错下载不了embedding模型文件
import os
os.environ['HF_ENDPOINT'] = 'https://hf-mirror.com'
client = MilvusClient("milvus_demo.db")
if client.has_collection(collection_name="demo_collection"):
client.drop_collection(collection_name="demo_collection") # 有collection就删除collection,相当于mysql中的一张表
client.create_collection(
collection_name="demo_collection",
dimension=768, # The vectors we will use in this demo has 768 dimensions,嵌入式向量的长度为768,一个句子共有768个数表示
)
#使用模型将文本变成向量
embedding_fn = model.DefaultEmbeddingFunction()
docs = [
"Artificial intelligence was founded as an academic discipline in