Skip to content

This is the source code of the Milvus vector database for Ballerina

License

Notifications You must be signed in to change notification settings

ballerina-platform/module-ballerinax-milvus

Repository files navigation

Ballerina Milvus Connector

Build Trivy GraalVM Check GitHub Last Commit GitHub Issues License

Milvus is an open-source vector database built for scalable similarity search and AI applications. It provides high-performance vector storage and retrieval capabilities, making it ideal for applications involving machine learning, deep learning, and similarity search scenarios.

The Ballerina Milvus connector provides seamless integration with Milvus vector database, enabling developers to perform operations such as creating collections, inserting vector data, building indexes, and conducting similarity searches within Ballerina applications. It supports Milvus SDK Java v2.6.1.

Setup guide

To utilize the Milvus connector, you must have access to a running Milvus instance. You can use one of the following methods for that.

Option 1: Using Docker

  1. Make sure Docker is installed on your system.

  2. Use the following command to start a Milvus standalone instance in docker

# Download the installation script
$ curl -sfL https://raw.githubusercontent.com/milvus-io/milvus/master/scripts/standalone_embed.sh -o standalone_embed.sh

#Start the Docker container
$ bash standalone_embed.sh start

For detailed installation instructions, refer to the official Milvus documentation.

Option 2: Using Milvus Cloud by Zilliz

Zilliz Cloud provides a fully managed Milvus service. Follow these steps to set up your cloud instance:

  1. Sign up to Zilliz Cloud: Visit Zilliz Cloud and create an account.

    Zilliz Cloud Sign Up
  2. Set up your account: Complete the account setup process with your details.

    Account Setup
  3. Create a new cluster: From the welcome page, select "Create Cluster" to start setting up your Milvus instance.

    Welcome Page
  4. Configure cluster details: Provide the necessary configuration details for your cluster, including cluster name, cloud provider, and region.

    Create Cluster
  5. Download credentials: Once your cluster is created, download the authentication credentials and connection details.

    Cluster Creation Complete
  6. Generate API Key: Navigate to the API Keys section in your cluster dashboard and generate an API key for authentication.

Quickstart

To use the Milvus connector in your Ballerina project, modify the .bal file as follows.

Step 1: Import the module

Import the ballerinax/milvus module into your Ballerina project.

import ballerinax/milvus;

Step 2: Instantiate a new connector

Create a milvus:Client with the Milvus server URL and optional configuration.

configurable string milvusUrl = ?;

milvus:Client milvusClient = check new(milvusUrl);

For authenticated connections, you can provide additional configuration.

milvus:Client milvusClient = check new(milvusUrl);

Step 3: Invoke the connector operations

You can now utilize the operations available within the connector.

public function main() returns error? {
    check milvusClient->createCollection({
        collectionName: "example_collection",
        dimension: 2
    });
}

Step 4: Run the Ballerina application

Use the following command to compile and run the Ballerina program.

bal run

Issues and projects

The Issues and Projects tabs are disabled for this repository as this is part of the Ballerina library. To report bugs, request new features, start new discussions, view project boards, etc., visit the Ballerina library parent repository.

This repository only contains the source code for the package.

Building from the source

Prerequisites

  1. Download and install Java SE Development Kit (JDK) version 17. You can download it from either of the following sources:

    Note: After installation, remember to set the JAVA_HOME environment variable to the directory where JDK was installed.

  2. Download and install Ballerina Swan Lake.

  3. Download and install Docker.

    Note: Ensure that the Docker daemon is running before executing any tests.

  4. Generate a Github access token with read package permissions, then set the following env variables:

    export packageUser=<Your GitHub Username>
    export packagePAT=<GitHub Personal Access Token>

Build options

Execute the commands below to build from the source.

  1. To build the package:

    ./gradlew clean build
  2. To run the tests:

    ./gradlew clean test
  3. To build the without the tests:

    ./gradlew clean build -x test
  4. To debug package with a remote debugger:

    ./gradlew clean build -Pdebug=<port>
  5. To debug with Ballerina language:

    ./gradlew clean build -PbalJavaDebug=<port>
  6. Publish the generated artifacts to the local Ballerina central repository:

    ./gradlew clean build -PpublishToLocalCentral=true
  7. Publish the generated artifacts to the Ballerina central repository:

    ./gradlew clean build -PpublishToCentral=true

Contributing to Ballerina

As an open source project, Ballerina welcomes contributions from the community.

For more information, go to the contribution guidelines.

Code of conduct

All contributors are encouraged to read the Ballerina Code of Conduct.

Useful links

About

This is the source code of the Milvus vector database for Ballerina

Topics

Resources

License

Stars

Watchers

Forks

Packages

 
 
 

Contributors 5