mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-31 15:11:35 -05:00
536fbd0399
* introduce rescore stage to handle fusion in shard properly * make shard-stage and collection-stage rescoring explicit * review fixes
141 lines
3.0 KiB
Python
Executable File
141 lines
3.0 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(f"Point: {point.id}, vector: {point.vector}, payload: {point.payload}")
|
|
|
|
|
|
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(
|
|
prefetches = [],
|
|
query = Query.Nearest([6.0, 9.0, 4.0, 2.0]),
|
|
filter = None,
|
|
score_threshold = None,
|
|
limit = 10,
|
|
offset = 0,
|
|
params = None,
|
|
with_vector = True,
|
|
with_payload = True,
|
|
))
|
|
|
|
for point in result:
|
|
print(f"Point: {point.id}, vector: {point.vector}, payload: {point.payload}, score: {point.score}")
|
|
|
|
|
|
print("---- Search ----")
|
|
|
|
points = shard.search(SearchRequest(
|
|
query=Query.Nearest([1.0, 1.0, 1.0, 1.0]),
|
|
filter=None,
|
|
params=None,
|
|
limit=10,
|
|
offset=0,
|
|
with_vector=True,
|
|
with_payload=True,
|
|
score_threshold=None,
|
|
))
|
|
|
|
for point in points:
|
|
print(f"Point: {point.id}, vector: {point.vector}, payload: {point.payload}, score: {point.score}")
|
|
|
|
|
|
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,
|
|
params=None,
|
|
limit=10,
|
|
offset=0,
|
|
with_vector=True,
|
|
with_payload=True,
|
|
score_threshold=None,
|
|
))
|
|
|
|
for point in points:
|
|
print(f"Point: {point.id}, vector: {point.vector}, payload: {point.payload}, score: {point.score}")
|
|
|
|
|
|
print("---- Retrieve ----")
|
|
|
|
points = shard.retrieve(point_ids=[1], with_vector=True, with_payload=True)
|
|
|
|
for point in points:
|
|
print(f"Point: {point.id}, vector: {point.vector}, payload: {point.payload}")
|