「学习笔记」Milvus 向量数据库入门

from milvus import default_server
from pymilvus import connections, Collection, utility

# 1. 启动 Milvus Lite 服务器
print("Starting Milvus Lite server...")
default_server.start()

# 2. 连接到服务器
try:
    connections.connect(alias="default", host="127.0.0.1", port=default_server.listen_port)
    print(f"Connected to Milvus. Port: {default_server.listen_port}")

    # 3. 检查服务器状态
    print(f"Server is running: {utility.get_server_version()}")

    # 4. 创建一个简单的集合（这里只是一个示例，实际使用需要定义Schema）
    # 注意：以下代码仅为演示连接成功，创建复杂Schema需要更多步骤。
    if utility.has_collection("hello_milvus"):
        utility.drop_collection("hello_milvus")
        print("Dropped existing 'hello_milvus' collection.")

    # 在实际应用中，您需要在这里定义 fields, schema 等。
    # 此处仅用 has_collection 和 drop_collection 来演示与服务器的交互是成功的。

    print("Milvus Lite is working correctly!")

except Exception as e:
    print(f"Error: {e}")

finally:
    # 5. 停止服务器 (重要!)
    print("Stopping Milvus Lite server...")
    default_server.stop()
    print("Server stopped.")

python test_milvus.py

Starting Milvus Lite server...
Connected to Milvus. Port: 19530
Server is running: v2.2.16-lite
Milvus Lite is working correctly!
Stopping Milvus Lite server...
Server stopped.

from milvus import default_server
from pymilvus import connections, Collection, CollectionSchema, FieldSchema, DataType, utility
import random

# 设置基础目录和端口
default_server.set_base_dir('./milvus_data')  # 指定数据存储目录
default_server.cleanup()  # 清理之前的运行状态

# 启动 Milvus Lite 服务器
print("Starting Milvus Lite server...")
default_server.start()

# 连接到服务器
try:
    connections.connect(alias="default", host="127.0.0.1", port=default_server.listen_port)
    print(f"Connected to Milvus. Port: {default_server.listen_port}")
    # 检查服务器状态
    print(f"Server is running: {utility.get_server_version()}")

    # 1. 定义 Schema
    # 我们创建三个字段：电影ID（主键）、电影名称（标量）、电影简介的向量
    fields = [
        FieldSchema(name="movie_id", dtype=DataType.INT64, is_primary=True, auto_id=True),
        FieldSchema(name="title", dtype=DataType.VARCHAR, max_length=200),
        FieldSchema(name="embedding", dtype=DataType.FLOAT_VECTOR, dim=8) # 使用8维向量方便演示
    ]
    schema = CollectionSchema(fields=fields, description="Movie search demo")

    # 2. 创建集合
    collection_name = "demo_movies"
    # 删除已存在的同名集合（如果是第一次运行，可以注释掉这几行）
    if utility.has_collection(collection_name):
        utility.drop_collection(collection_name)
        print(f"Dropped existing collection: {collection_name}")
    # 执行创建
    collection = Collection(name=collection_name, schema=schema)
    print(f"Collection '{collection_name}' created.")

    # 3. 创建索引（在插入数据后创建效率更高，但这里先创建以演示完整流程）
    index_params = {
        "index_type": "AUTOINDEX", # 对于 Milvus Lite，使用 AUTOINDEX 即可
        "metric_type": "L2",       # 使用 L2 距离（欧氏距离）
        "params": {}
    }
    collection.create_index(field_name="embedding", index_params=index_params)
    print("Index on 'embedding' field created.")

    # 4. 插入数据
    movies = [ # 模拟一些电影数据和对应的8维向量
        {"title": "The Matrix", "embedding": [0.1, 0.3, 0.5, 0.2, 0.9, 0.1, 0.6, 0.4]},
        {"title": "Inception", "embedding": [0.2, 0.4, 0.5, 0.3, 0.8, 0.2, 0.5, 0.3]},
        {"title": "The Godfather", "embedding": [0.8, 0.1, 0.1, 0.9, 0.2, 0.8, 0.1, 0.9]},
        {"title": "Pulp Fiction", "embedding": [0.7, 0.2, 0.2, 0.8, 0.3, 0.7, 0.2, 0.8]},
        {"title": "The Dark Knight", "embedding": [0.3, 0.5, 0.6, 0.4, 0.7, 0.3, 0.4, 0.2]},
    ]
    # 准备插入的数据
    titles = [movie["title"] for movie in movies]
    embeddings = [movie["embedding"] for movie in movies]
    # 注意：movie_id 是 auto_id，所以我们不需要提供
    entities = [titles, embeddings]
    # 执行插入
    insert_result = collection.insert(entities)
    print(f"Inserted {len(insert_result.primary_keys)} movies.")

    # 5. 将集合加载到内存（搜索前必须执行此步骤）
    collection.load()
    print("Collection loaded into memory.")

    # 6. 执行向量搜索
    print("\n--- Starting Vector Search ---")
    # 假设我们想找与 "Inception" 相似的电影，使用它的向量作为查询向量
    query_embedding = [0.2, 0.4, 0.5, 0.3, 0.8, 0.2, 0.5, 0.3] # Inception 的向量
    # 定义搜索参数
    search_params = {
        "metric_type": "L2",     # 使用 L2 距离（欧氏距离）
        "params": {"nprobe": 10} # 搜索时探查的聚类数
    }
    # 执行搜索，返回最相似的2部电影
    results = collection.search(
        data=[query_embedding],
        anns_field="embedding",
        param=search_params,
        limit=3,                # 返回前3个结果
        output_fields=["title"] # 指定要返回的标量字段
    )
    # 处理并打印搜索结果
    print(f"Search for movies similar to 'Inception':")
    for i, hits in enumerate(results):
        print(f"Query {i+1} results:")
        for hit in hits:
            print(f"  Title: {hit.entity.get('title')}, Distance: {hit.distance:.4f}")

    # 7. 执行混合搜索（向量搜索 + 标量过滤）
    print("\n--- Starting Hybrid Search (with filter) ---")
    # 假设我们想找相似的电影，但标题中不能包含 "Pulp"
    filter_expression = "title not in [\"Pulp Fiction\"]"
    results_hybrid = collection.search(
        data=[query_embedding],
        anns_field="embedding",
        param=search_params,
        limit=3,
        expr=filter_expression, # 这里添加过滤条件
        output_fields=["title"]
    )
    # 处理并打印搜索结果
    print(f"Hybrid search for movies similar to 'Inception' but not 'Pulp Fiction':")
    for i, hits in enumerate(results_hybrid):
        print(f"Query {i+1} results:")
        for hit in hits:
            print(f"  Title: {hit.entity.get('title')}, Distance: {hit.distance:.4f}")

    # 8. 清理资源
    print("\nCleaning up...")
    collection.release()
    print("Demo finished!")

except Exception as e:
    print(f"Error: {e}")

finally:
    # 停止服务器 (重要!)
    print("Stopping Milvus Lite server...")
    default_server.stop()
    print("Server stopped.")

Starting Milvus Lite server...
Connected to Milvus. Port: 19530
Server is running: v2.2.16-lite
Collection 'demo_movies' created.
Index on 'embedding' field created.
Inserted 5 movies.
Collection loaded into memory.

--- Starting Vector Search ---
Search for movies similar to 'Inception':
Query 1 results:
  Title: Inception, Distance: 0.0000
  Title: The Matrix, Distance: 0.0700
  Title: The Dark Knight, Distance: 0.0800

--- Starting Hybrid Search (with filter) ---
Hybrid search for movies similar to 'Inception' but not 'Pulp Fiction':
Query 1 results:
  Title: Inception, Distance: 0.0000
  Title: The Matrix, Distance: 0.0700
  Title: The Dark Knight, Distance: 0.0800

Cleaning up...
Demo finished!
Stopping Milvus Lite server...
Server stopped.