mirror of
https://github.com/qdrant/qdrant.git
synced 2026-08-02 08:00:55 -05:00
8cdc8dea94
* Add pyo3 signature annotations with default values for all Py* structs - Added #[pyo3(signature = (...))] attributes to all Py* structs with optional parameters - Ensured non-default parameters come before parameters with defaults - Follows the pattern established in PyScrollRequest - Updated structs: PySearchRequest, PySearchParams, PyQuantizationSearchParams, PyAcornSearchParams, PyQueryRequest, PyPrefetch, PyOrderBy, PyMmr, PyPoint, PyFormula, PyShard::load, PySparseVectorDataConfig, PySparseIndexConfig * fix params ordering
145 lines
2.9 KiB
Python
Executable File
145 lines
2.9 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
|
|
from qdrant_edge import *
|
|
from common import *
|
|
|
|
|
|
print("---- Point conversions ----")
|
|
|
|
points = [
|
|
Point(10, [[1,2,3], [3, 4, 5]], {}),
|
|
Point(11, { "sparse": SparseVector(indices=[0, 2], values=[1.0, 3.0]) }, {}),
|
|
]
|
|
|
|
# Test points conversion into internal representation and back
|
|
for point in points:
|
|
print(point)
|
|
|
|
|
|
print("---- Load shard ----")
|
|
|
|
shard = load_new_shard()
|
|
|
|
|
|
print("---- Upsert ----")
|
|
|
|
shard.update(UpdateOperation.upsert_points([
|
|
Point(
|
|
1,
|
|
[6.0, 9.0, 4.0, 2.0],
|
|
{
|
|
"null": None,
|
|
"str": "string",
|
|
"uint": 42,
|
|
"int": -69,
|
|
"float": 4.20,
|
|
"bool": True,
|
|
"obj": {
|
|
"null": None,
|
|
"str": "string",
|
|
"uint": 42,
|
|
"int": -69,
|
|
"float": 4.20,
|
|
"bool": True,
|
|
"obj": {},
|
|
"arr": [],
|
|
},
|
|
"arr": [None, "string", 42, -69, 4.20, True, {}, []],
|
|
},
|
|
),
|
|
Point(
|
|
"e9408f2b-b917-4af1-ab75-d97ac6b2c047",
|
|
[6.0, 9.0, 3.0, -2.0],
|
|
{
|
|
"hello": "world",
|
|
"price": 199.99,
|
|
},
|
|
),
|
|
Point(
|
|
uuid.uuid4(),
|
|
[1.0, 6.0, 4.0, 2.0],
|
|
{
|
|
"hello": "world",
|
|
"price": 999.99,
|
|
},
|
|
),
|
|
]))
|
|
|
|
|
|
print("---- Query ----")
|
|
|
|
result = shard.query(QueryRequest(
|
|
query = Query.Nearest([6.0, 9.0, 4.0, 2.0]),
|
|
limit = 10,
|
|
with_vector = True,
|
|
with_payload = True,
|
|
))
|
|
|
|
for point in result:
|
|
print(point)
|
|
|
|
|
|
print("---- Search ----")
|
|
|
|
points = shard.search(SearchRequest(
|
|
query=Query.Nearest([1.0, 1.0, 1.0, 1.0]),
|
|
limit=10,
|
|
with_vector=True,
|
|
with_payload=True,
|
|
))
|
|
|
|
for point in points:
|
|
print(point)
|
|
|
|
|
|
print("---- Search + Filter ----")
|
|
|
|
search_filter = Filter(
|
|
must=[
|
|
FieldCondition(
|
|
key="hello",
|
|
match=MatchTextAny(text_any="world"),
|
|
),
|
|
FieldCondition(
|
|
key="price",
|
|
range=RangeFloat(gte=500.0),
|
|
)
|
|
]
|
|
)
|
|
|
|
points = shard.search(SearchRequest(
|
|
query=Query.Nearest([1.0, 1.0, 1.0, 1.0]),
|
|
filter=search_filter,
|
|
limit=10,
|
|
with_vector=True,
|
|
with_payload=True,
|
|
))
|
|
|
|
for point in points:
|
|
print(point)
|
|
|
|
|
|
print("---- Retrieve ----")
|
|
|
|
points = shard.retrieve(point_ids=[1], with_vector=True, with_payload=True)
|
|
|
|
for point in points:
|
|
print(point)
|
|
|
|
print("---- Scroll ----")
|
|
|
|
scroll_result, next_offset = shard.scroll(ScrollRequest(limit=2))
|
|
for point in scroll_result:
|
|
print(point)
|
|
|
|
while next_offset is not None:
|
|
print(f"--- Next scroll (offset = {next_offset})---")
|
|
scroll_result, next_offset = shard.scroll(ScrollRequest(limit=2, offset=next_offset))
|
|
for point in scroll_result:
|
|
print(point)
|
|
|
|
print("---- Count ----")
|
|
|
|
count = shard.count(CountRequest(exact=True))
|
|
|
|
print(f"Total points count: {count}") |