提示信息

利用 pymilvus 使用milvus

李霖 已关注

在Python环境中使用pymilvus库可以方便地操作Milvus向量数据库。下面是一个基本的例子,演示了如何创建集合,插入数据和执行搜索。

首先,确保你已经安装了pymilvus库:

  1. pip3 install pymilvus

然后,连接到运行的Milvus服务,并进行操作:

  1. from pymilvus import connections, FieldSchema, CollectionSchema, DataType, Collection
  2. # 连接到Milvus服务
  3. connections.connect("default", host="localhost", port="19530")
  4. # 定义字段和集合模式
  5. dim = 128
  6. fields = [
  7. FieldSchema(name="id", dtype=DataType.INT64, is_primary=True),
  8. FieldSchema(name="embedding", dtype=DataType.FLOAT_VECTOR, dim=dim)
  9. ]
  10. schema = CollectionSchema(fields=fields, description="test collection")
  11. # 创建向量数据库
  12. collection = Collection(name="my_collection", schema=schema)
  13. # 插入一些向量数据
  14. import numpy as np
  15. vectors = np.random.rand(100, dim).tolist()
  16. ids = [i for i in range(len(vectors))]
  17. mr = collection.insert([ids, vectors])
  18. # 执行查询
  19. topK = 10
  20. query_embedding = vectors[0]
  21. res = collection.search(query_embedding, "embedding", param={'nprobe': 10}, limit=topK)

这个代码片段首先连接到运行在本地的Milvus服务,然后创建一个新的集合并插入一些随机生成的向量数据。之后,它将执行一个查询,返回查询向量的最近邻。

请注意,这只是一个非常基本的例子。Milvus提供了许多其他功能,包括分区、索引管理、数据管理和高级搜索等。具体信息可以参考pymilvus的官方文档

    遇到难题? "AI大模型GPT4.0、GPT" 是你的私人解答专家! 点击按钮去提问......
李霖 关注 已关注

最近一次登录:2024-06-17 13:12:03   

暂时还没有签名,请关注我或评论我的文章
×
免费图表工具,画流程图、架构图