MCPcopy Create free account
hub / github.com/alibaba/proxima / init_request

Method init_request

tests/query/query_service_test.cc:56–85  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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());

Callers

nothing calls this directly

Calls 12

set_topkMethod · 0.80
set_featuresMethod · 0.80
set_radiusMethod · 0.80
set_keyMethod · 0.80
set_valueMethod · 0.80
resetMethod · 0.45
set_collection_nameMethod · 0.45
set_debug_modeMethod · 0.45
set_column_nameMethod · 0.45
set_dimensionMethod · 0.45
set_data_typeMethod · 0.45
set_primary_keyMethod · 0.45

Tested by

no test coverage detected