mirror of
https://github.com/milvus-io/milvus.git
synced 2026-07-21 18:25:44 +00:00
yanliang567andGitHub
dc5b3d069a
Related: #49046 ## What changed - Add REST v2 searchAggregation request parsing, validation, and proxy conversion for search requests. - Return aggregation buckets with top hits, metrics, optional recall information, cost, and storage metadata. - Reject searchAggregation on unsupported REST v1 / low-level search paths and hybrid sub-search inputs. - Add Go unit coverage and RESTful v2 e2e coverage for happy paths, nested aggregations, metrics/order, filters, and validation failures. ## Verification - /Users/yanliang.qiao/fork/.venv-py312/bin/python -m ruff check tests/restful_client_v2/testcases/test_vector_operations.py - /Users/yanliang.qiao/fork/.venv-py312/bin/python -m ruff format --check tests/restful_client_v2/testcases/test_vector_operations.py - /Users/yanliang.qiao/fork/.venv-py312/bin/python -m py_compile tests/restful_client_v2/testcases/test_vector_operations.py - /Users/yanliang.qiao/fork/.venv-py312/bin/python -m pytest --collect-only testcases/test_vector_operations.py::TestSearchAggregationVector -q - git diff --check Note: local Go unit test build is blocked by missing native Milvus dependencies on this machine: rocksdb.pc and milvus_core.pc. Signed-off-by: Yanliang Qiao <yanliang.qiao@zilliz.com>
97 lines
3.9 KiB
Go
97 lines
3.9 KiB
Go
// Licensed to the LF AI & Data foundation under one
|
|
// or more contributor license agreements. See the NOTICE file
|
|
// distributed with this work for additional information
|
|
// regarding copyright ownership. The ASF licenses this file
|
|
// to you under the Apache License, Version 2.0 (the
|
|
// "License"); you may not use this file except in compliance
|
|
// with the License. You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package httpserver
|
|
|
|
type CreateCollectionReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
Dimension int32 `json:"dimension" validate:"required"`
|
|
Description string `json:"description"`
|
|
MetricType string `json:"metricType"`
|
|
PrimaryField string `json:"primaryField"`
|
|
VectorField string `json:"vectorField"`
|
|
EnableDynamicField bool `json:"enableDynamicField"`
|
|
}
|
|
|
|
type DropCollectionReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
}
|
|
|
|
type QueryReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
OutputFields []string `json:"outputFields"`
|
|
Filter string `json:"filter" validate:"required"`
|
|
Limit int32 `json:"limit"`
|
|
Offset int32 `json:"offset"`
|
|
}
|
|
|
|
type GetReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
OutputFields []string `json:"outputFields"`
|
|
ID interface{} `json:"id" validate:"required"`
|
|
}
|
|
|
|
type DeleteReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
ID interface{} `json:"id"`
|
|
Filter string `json:"filter"`
|
|
}
|
|
|
|
type InsertReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
Data []map[string]interface{} `json:"data" validate:"required"`
|
|
}
|
|
|
|
type SingleInsertReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
Data map[string]interface{} `json:"data" validate:"required"`
|
|
}
|
|
|
|
type UpsertReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
Data []map[string]interface{} `json:"data" validate:"required"`
|
|
PartialUpdate bool `json:"partialUpdate"`
|
|
FieldOps []FieldPartialUpdateOpReq `json:"fieldOps"`
|
|
}
|
|
|
|
type SingleUpsertReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
Data map[string]interface{} `json:"data" validate:"required"`
|
|
PartialUpdate bool `json:"partialUpdate"`
|
|
FieldOps []FieldPartialUpdateOpReq `json:"fieldOps"`
|
|
}
|
|
|
|
type SearchReq struct {
|
|
DbName string `json:"dbName"`
|
|
CollectionName string `json:"collectionName" validate:"required"`
|
|
Filter string `json:"filter"`
|
|
Limit int32 `json:"limit"`
|
|
Offset int32 `json:"offset"`
|
|
OutputFields []string `json:"outputFields"`
|
|
Vector []float32 `json:"vector"`
|
|
Params map[string]float64 `json:"params"`
|
|
SearchAggregation *SearchAggregationReq `json:"searchAggregation"`
|
|
}
|