index_search_c_api.h

Go to the documentation of this file.

#ifndef _INDEX_SEARCH_C_API_H_
#define _INDEX_SEARCH_C_API_H_

#include "constants.h"
#include "api_types.h"

// Creates the instance of the class with saving hnsw index
extern "C" {
void *rf_index_search_init_from_vectors(rf_float_vector_type enrolled_vectors,
                                        const char* hnsw_file_name,
                                        rf_error_type *error);



#if 0
void *rf_index_search_init_from_vectors_ex(rf_float_vector_type enrolled_vectors,
                                            rf_object_list_type external_ids,
                                           const char* hnsw_file_name,
                                           rf_error_type *error);
#endif

void *rf_index_search_init_from_vectors_list(rf_object_list_type vectors_list,
                                             rf_object_list_type external_ids,
                                             const char* hnsw_file_name,
                                             rf_error_type *error);
void *rf_index_search_init_empty(const char* hnsw_file_name,
                                 rf_error_type *error);

void *rf_index_search_init_from_file(const char* hnsw_file_name,
                                     rf_error_type *error);

void rf_index_search_close(void *index_search_handle, rf_error_type *error);


void rf_index_search_by_index(void *index_search_handle, rf_float_vector_type query_vector,
                              int top_k,
                              float search_threshold,
                              rf_int_vector_type *output_vector_ids, rf_error_type *error);

void rf_index_search_by_index_external_ids(void *index_search_handle, rf_float_vector_type query_vector,
                                           int top_k,
                                           float search_threshold,
                                           rf_int_vector_type *output_vector_ids,
                                           rf_object_list_type *external_ids_out,
                                           rf_float_vector_type *confidence, rf_error_type *error);


void rf_face_get_enrolled_vectors(void *index_search_handle, rf_float_vector_type *enrolled_vectors_out, rf_error_type *error);

void rf_face_get_enrolled_vectors_list(void *index_search_handle, rf_object_list_type *enrolled_vectors_out, rf_error_type *error);
void rf_face_get_external_ids(void *index_search_handle, rf_object_list_type *enrolled_vectors_external_ids_out, rf_error_type *error);

void rf_face_enroll(void *index_search_handle, rf_object_list_type vectors_to_enroll, rf_object_list_type external_ids_to_enroll, rf_error_type *error);
void rf_face_remove_by_external_id(void *index_search_handle, const char *external_id_to_remove, rf_error_type *error);
void rf_face_remove_by_external_ids(void *index_search_handle, rf_object_list_type external_ids_to_remove, rf_error_type *error);
#if 0
void rf_face_remove_by_id(void *index_search_handle, int id_to_remove, rf_error_type *error);
#endif

#if 0
void rf_face_remove_by_ids(void *index_search_handle, rf_int_vector_type ids_to_remove, rf_error_type *error);
#endif

float rf_face_get_simularity(void *index_search_handle, rf_float_vector_type item1, rf_float_vector_type item2, rf_error_type *error);

void rf_face_cluster_embeddings(rf_float_vector_type embedding_vectors,
                                rf_int_vector_type *out_ids,
                                rf_error_type *error);
}

#endif //_index_search_C_API_H_