An Introduction to Milvus Python SDK and API

Cover image.

Background

The following illustration depicts the interaction between SDKs and Milvus through gRPC. Imagine that Milvus is a black box. Protocol Buffers are used to define the interfaces of the server and the structure of the information they carry. Therefore, all operations in the black box Milvus are defined by Protocol API.

The interaction between SDKs and Milvus through gRPC

The interaction between SDKs and Milvus through gRPC

Milvus Protocol API

Milvus Protocol API consists of milvus.proto, common.proto, and schema.proto, which are Protocol Buffers files suffixed with .proto. SDKs must interact with Milvus with these Protocol Buffers files to ensure proper operation.

milvus.proto

milvus.proto is the vital component of Milvus Protocol API because it defines the, which further defines all RPC interfaces of Milvus.

The following code sample shows the interface CreatePartitionRequest. It has two major string-type parameters collection_name and partition_name, based on which you can start a partition creation request.

CreatePartitionRequest interface

CreatePartitionRequest interface

Check an example of Protocol in PyMilvus GitHub Repository on line 19.

An example of Protocol

An example of a Protocol

You can find the definition  CreatePartitionRequest here.

The definition of CreatePartitionRequest

The definition of CreatePartitionRequest

Contributors who wish to develop a feature of Milvus or an SDK in a different programming language are welcome to find all interfaces Milvus offers via RPC.

common.proto

common.proto defines the common types of information, including, and Status.

common.proto

common.proto

schema.proto

schema.proto defines the schema in the parameters. The following code sample is an example of CollectionSchema.

schema.proto

schema.proto

milvus.proto, common.proto, and schema.proto together constitutes the API of Milvus, representing all operations that can be called via RPC.

If you dig into the source code and observe, you will find that when interfaces like create_index are called, they call multiple RPC interfaces such as describe_collection and describe_index. Many of the outward interfaces of Milvus is a combination of multiple RPC interfaces.

Having understood the behaviors of RPC, you can then develop new features for Milvus through combination. You are more than welcome to use your imagination and creativeness and contribute to the Milvus community.

PyMilvus 2.0

Object-Relational Mapping (ORM)

To put it in a nutshell, Object-relational mapping (ORM) refers to that when you operate on a local object; such operations will affect the corresponding object on the server. PyMilvus ORM-style API features the following characteristics:

  1. First, it operates directly on objects.
  2. It isolates service logic and data access details.
  3. It hides the complexity of implementation, and you can run the same scripts across different Milvus instances regardless of their deployment approaches or implementation.

ORM-Style API

One of the essences of ORM-style API lies in the control of the Milvus connection. For example, you can specify aliases for multiple Milvus servers and connect to or disconnect from them merely with their aliases. You can even delete the local server address and precisely control certain objects via a specific connection.

Control connections to Milvus

Control connections to Milvus

Another feature of ORM-style API is that, after abstraction, all operations can be performed directly on objects, including collection, partition, and index.

You can abstract a collection object by getting an existing one or creating a new one. You can also assign a Milvus connection to specific objects using a connection alias so that you can operate on these objects locally.

To create a partition object, you can either create it with its parent collection object or do it just like when you create a collection object. These methods can be employed on an index object as well.

In the case that these partition or index objects exist, you can get them through their parent collection object.

What's More

It is recommended to learn about PyMilvus through our technical document. The document consists of two major sections, including an automated API reference and usage instructions made by our contributors.

 

 

 

 

Top