ivyxivyx
← All ivy-nodes
IVYXSTUDIO · IVY NODE
M

Milvus Client

ivy.node.milvus-client · v0.0.0

ivyx✓

Creates and manages a connection to a Milvus vector database. This node establishes a client connection that can be used for vector similarity search operations. Milvus is used for storing and searching high-dimensional vectors (embeddings). The client output is required by milvus-insert and milvus-search nodes. Use this node at the start of vector database workflows.

#Milvus#Client

Inputs

FieldTypeDescription
urirequiredstringThe connection URI for the Milvus server
featuresobjectFeatures
labelsobjectLabels

Outputs

FieldTypeDescription
clientrequiredobjectAn instance of MilvusClient connected to the specified URI
featuresobjectFeatures
labelsobjectLabels

Source

python

# Input preparation
inp = __ivy_ctx__["nodes"][__ivy_node_id__]["input"]
uri = inp["uri"]

# Compute
from pymilvus import MilvusClient

class MilvusClient:
    """
    A class for creating and managing a Milvus client instance.
    """

    def __init__(self, uri: str):
        """
        Initialize the MilvusClient with a Milvus server URI.

        :param uri: The connection URI for the Milvus server.
        """
        self.uri = uri

    def generate(self) -> MilvusClient:
        """
        Create and return a Milvus client instance.

        :return: An instance of MilvusClient connected to the specified URI.
        """
        self.milvus_client = MilvusClient(self.uri)
        
        return self.milvus_client

milvus_client = MilvusClient(uri=uri)
client = milvus_client.generate()

# Output collection (runner reads __ivy_ctx__)
out = __ivy_ctx__["nodes"][__ivy_node_id__]["output"]
out["client"] = client

Tests

Requires: python:3.11

  • basic-connection

    Create Milvus client with valid URI (should succeed).

    connectionbasic
  • error-invalid-uri

    Invalid URI format should handle error gracefully.

    error-handling