mirror of
https://github.com/qdrant/qdrant.git
synced 2026-08-03 00:20:57 -05:00
e7fe24bb1b
* Cleanup `edge::Shard::retrieve` * Cleanup Python errors * Cleanup `PointId` conversions * Rename `ids` into `point_ids` in `PyShard::retrieve` * WIP: Prototype native conversions for more types
100 lines
2.2 KiB
Python
100 lines
2.2 KiB
Python
import os
|
|
import shutil
|
|
import uuid
|
|
|
|
from qdrant_edge import *
|
|
|
|
print("---- Load shard ----")
|
|
|
|
DATA_DIRECTORY = "./data"
|
|
|
|
# Clear and recreate data directory
|
|
if os.path.exists(DATA_DIRECTORY):
|
|
shutil.rmtree(DATA_DIRECTORY)
|
|
|
|
os.makedirs(DATA_DIRECTORY)
|
|
|
|
# Load Qdrant Edge shard
|
|
config = SegmentConfig(
|
|
vector_data={
|
|
"": VectorDataConfig(
|
|
size=4,
|
|
distance=Distance.DOT,
|
|
storage_type=VectorStorageType.CHUNKED_MMAP,
|
|
index=Indexes.PLAIN,
|
|
quantization_config=None,
|
|
multivector_config=None,
|
|
datatype=None,
|
|
),
|
|
},
|
|
sparse_vector_data={},
|
|
payload_storage_type=PayloadStorageType.IN_RAM_MMAP,
|
|
)
|
|
|
|
shard = Shard(DATA_DIRECTORY, config)
|
|
|
|
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"
|
|
},
|
|
),
|
|
Point(
|
|
uuid.uuid4(),
|
|
[1.0, 6.0, 4.0, 2.0],
|
|
{
|
|
"hello": "world"
|
|
},
|
|
),
|
|
]))
|
|
|
|
print("---- Search ----")
|
|
|
|
points = shard.search(SearchRequest(
|
|
query=[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("---- 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}")
|