mirror of
https://github.com/qdrant/qdrant.git
synced 2026-08-07 02:20:55 -05:00
fb9d36e913
* Cleanup `shard` crate module declarations * Move `ScrollRequestInternal` into `shard` crate * fixup! Move `ScrollRequestInternal` into `shard` crate Fix imports * fixup! Move `ScrollRequestInternal` into `shard` crate `const fn default_*` * Implement `edge::Shard::scroll` * fixup! Implement `edge::Shard::scroll` Re-export `OrderByInterface` * Cleanup `edge` module declarations * Cleanup `qdrant-edge-py` module declarations * Move `PyWithPayload` and `PyWithVector` into `types::query` * Add `PyScrollRequest` type * Implement `PyShard::scroll` * Move `CountRequestInternal` into `shard` crate * fixup! Move `CountRequestInternal` into `shard` crate Fix imports * fixup! Move `CountRequestInternal` into `shard` crate Rename `default_exact_count` into `CountRequestInternal::default_exact` * Implement `edge::Shard::count` * Implement `PyShard::count` * review: offset for scroll, default values, examples * ai review --------- Co-authored-by: generall <andrey@vasnetsov.com>
93 lines
2.2 KiB
Rust
93 lines
2.2 KiB
Rust
use bytemuck::TransparentWrapper as _;
|
|
use derive_more::Into;
|
|
use pyo3::prelude::*;
|
|
use segment::types::*;
|
|
use shard::scroll::*;
|
|
|
|
use crate::query::PyOrderBy;
|
|
use crate::repr::*;
|
|
use crate::types::*;
|
|
|
|
#[pyclass(name = "ScrollRequest")]
|
|
#[derive(Clone, Debug, Into)]
|
|
pub struct PyScrollRequest(ScrollRequestInternal);
|
|
|
|
#[pyclass_repr]
|
|
#[pymethods]
|
|
impl PyScrollRequest {
|
|
#[new]
|
|
#[pyo3(signature = (
|
|
offset = None,
|
|
limit = None,
|
|
filter = None,
|
|
with_payload = None,
|
|
with_vector = None,
|
|
order_by = None,
|
|
))]
|
|
pub fn new(
|
|
offset: Option<PyPointId>,
|
|
limit: Option<usize>,
|
|
filter: Option<PyFilter>,
|
|
with_payload: Option<PyWithPayload>,
|
|
with_vector: Option<PyWithVector>,
|
|
order_by: Option<PyOrderBy>,
|
|
) -> Self {
|
|
Self(ScrollRequestInternal {
|
|
offset: offset.map(PointIdType::from),
|
|
limit,
|
|
filter: filter.map(Filter::from),
|
|
with_payload: with_payload.map(WithPayloadInterface::from),
|
|
with_vector: with_vector.map(WithVector::from).unwrap_or_default(),
|
|
order_by: order_by.map(OrderByInterface::from),
|
|
})
|
|
}
|
|
|
|
#[getter]
|
|
pub fn offset(&self) -> Option<PyPointId> {
|
|
self.0.offset.map(PyPointId)
|
|
}
|
|
|
|
#[getter]
|
|
pub fn limit(&self) -> Option<usize> {
|
|
self.0.limit
|
|
}
|
|
|
|
#[getter]
|
|
pub fn filter(&self) -> Option<&PyFilter> {
|
|
self.0.filter.as_ref().map(PyFilter::wrap_ref)
|
|
}
|
|
|
|
#[getter]
|
|
pub fn with_payload(&self) -> Option<&PyWithPayload> {
|
|
self.0.with_payload.as_ref().map(PyWithPayload::wrap_ref)
|
|
}
|
|
|
|
#[getter]
|
|
pub fn with_vector(&self) -> &PyWithVector {
|
|
PyWithVector::wrap_ref(&self.0.with_vector)
|
|
}
|
|
|
|
#[getter]
|
|
pub fn order_by(&self) -> Option<PyOrderBy> {
|
|
self.0.order_by.clone().map(PyOrderBy::from)
|
|
}
|
|
|
|
pub fn __repr__(&self) -> String {
|
|
self.repr()
|
|
}
|
|
}
|
|
|
|
impl PyScrollRequest {
|
|
fn _getters(self) {
|
|
// Every field should have a getter method
|
|
let ScrollRequestInternal {
|
|
offset: _,
|
|
limit: _,
|
|
filter: _,
|
|
with_payload: _,
|
|
with_vector: _,
|
|
order_by: _,
|
|
} = self.0;
|
|
}
|
|
}
|