| 
						
						
							
								
							
						
						
					 | 
				
				 | 
				
					@ -12,16 +12,23 @@ class MilvusClient(PyOperator): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    Milvus ANN index class. | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    """ | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					
 | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    def __init__(self, host: str, port: int, collection_name: str, user: str = None, password: str = None): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    def __init__( | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self, host: str = 'localhost', port: int = 19530, collection_name: str = None,  uri: str = None, user: str = None, password: str = None, token: str = None | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    ): | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self._host = host | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self._port = port | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self._uri = uri | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self._collection_name = collection_name | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self._connect_name = uuid.uuid4().hex | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        if None in [user, password]: | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					            connections.connect(alias=self._connect_name, host=self._host, port=self._port) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        else: | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        if uri and token: | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					            connections.connect(alias=self._connect_name, uri=self._uri, token=token, secure=True) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        elif user and password: | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					            connections.connect(alias=self._connect_name, host=self._host, port=self._port, | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					                                user=user, password=password, secure=True) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        else: | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					            connections.connect(alias=self._connect_name, host=self._host, port=self._port) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					         | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					     | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					        self._collection = Collection(self._collection_name, using=self._connect_name) | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					
 | 
				
			
			
		
	
		
			
				
					 | 
					 | 
				
				 | 
				
					    def __call__(self, *data): | 
				
			
			
		
	
	
		
			
				
					| 
						
							
								
							
						
						
						
					 | 
				
				 | 
				
					
  |