| 54 | |
| 55 | private: |
| 56 | void init_request() { |
| 57 | request_.reset(new (std::nothrow) QueryRequest()); |
| 58 | request_->set_query_type(proxima::be::proto::QueryRequest_QueryType_QT_KNN); |
| 59 | request_->set_collection_name(collection_); |
| 60 | // not supported yet |
| 61 | request_->set_debug_mode(false); |
| 62 | // Allocate KNN Param |
| 63 | param_ = request_->mutable_knn_param(); |
| 64 | |
| 65 | param_->set_column_name("column_name"); |
| 66 | param_->set_topk(1); |
| 67 | param_->set_dimension(10); |
| 68 | param_->set_data_type(proxima::be::proto::DataType::DT_VECTOR_FP16); |
| 69 | param_->set_features("features"); |
| 70 | param_->set_batch_count(1); |
| 71 | param_->set_radius(0.1f); |
| 72 | |
| 73 | auto kv = param_->add_extra_params(); |
| 74 | kv->set_key("string_key1"); |
| 75 | kv->set_value("value1"); |
| 76 | kv = param_->add_extra_params(); |
| 77 | kv->set_key("int_key1"); |
| 78 | kv->set_value("10"); |
| 79 | |
| 80 | equal_request_.reset(new (std::nothrow) GetDocumentRequest()); |
| 81 | equal_request_->set_collection_name(collection_); |
| 82 | // not supported yet |
| 83 | equal_request_->set_debug_mode(false); |
| 84 | equal_request_->set_primary_key(1); |
| 85 | } |
| 86 | |
| 87 | void init_response() { |
| 88 | response_.reset(new (std::nothrow) QueryResponse()); |
nothing calls this directly
no test coverage detected