mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-29 22:21:08 -05:00
536fbd0399
* introduce rescore stage to handle fusion in shard properly * make shard-stage and collection-stage rescoring explicit * review fixes
55 lines
1.0 KiB
Python
55 lines
1.0 KiB
Python
from qdrant_edge import *
|
|
from common import *
|
|
|
|
|
|
|
|
shard = load_new_shard()
|
|
fill_dummy_data(shard)
|
|
|
|
|
|
search_filter = Filter(
|
|
must=[
|
|
FieldCondition(
|
|
key="color",
|
|
match=MatchValue(value="red"),
|
|
)
|
|
]
|
|
)
|
|
|
|
|
|
result = shard.query(QueryRequest(
|
|
prefetches = [
|
|
Prefetch(
|
|
prefetches=[],
|
|
query=Query.Nearest([6.0, 9.0, 4.0, 2.0]),
|
|
limit=5,
|
|
params=None,
|
|
filter=None,
|
|
score_threshold=None,
|
|
),
|
|
Prefetch(
|
|
prefetches=[],
|
|
query=Query.Nearest([1.0, -3.0, 2.0, 8.0]),
|
|
limit=5,
|
|
params=None,
|
|
filter=search_filter,
|
|
score_threshold=None,
|
|
)
|
|
],
|
|
query = Fusion.rrfk(2),
|
|
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}")
|
|
|
|
|
|
|