mirror of
https://github.com/qdrant/landing_page.git
synced 2026-10-05 02:48:30 +02:00
Abdon Pijpelinkandxzfc
ea0de4c12e
* Initial Rust snippets * Update content for Rust * Upgrade Rust to 1.94.0; test Edge code snippets against dev * Temporarily test against Edge shim crate * Move to EdgeShardConfig and EdgeVectorParams * Add docs for payload indexing, filtering, facet(), and optimize() * De-emphasize on-device use case * Update link to Rust examples on Github * Flattened API * Switch to new/create and load to initialize shards * Fixups for released packages * Mention recover_partial_snapshot on method list * Link to Github dev branch for examples --------- Co-authored-by: xzfc <xzfcpw@gmail.com>
130 lines
2.9 KiB
Python
130 lines
2.9 KiB
Python
# @block-start create-storage-directory
|
|
from pathlib import Path
|
|
|
|
SHARD_DIRECTORY = "./qdrant-edge-directory"
|
|
|
|
Path(SHARD_DIRECTORY).mkdir(parents=True, exist_ok=True)
|
|
# @block-end create-storage-directory
|
|
|
|
# @block-start configure-edge-shard
|
|
from qdrant_edge import (
|
|
Distance,
|
|
EdgeConfig,
|
|
EdgeVectorParams,
|
|
)
|
|
|
|
VECTOR_NAME="my-vector"
|
|
VECTOR_DIMENSION=4
|
|
|
|
config = EdgeConfig(
|
|
vectors={
|
|
VECTOR_NAME: EdgeVectorParams(
|
|
size=VECTOR_DIMENSION,
|
|
distance=Distance.Cosine,
|
|
)
|
|
}
|
|
)
|
|
# @block-end configure-edge-shard
|
|
|
|
# @block-start initialize-edge-shard
|
|
from qdrant_edge import EdgeShard
|
|
|
|
edge_shard = EdgeShard.create(SHARD_DIRECTORY, config)
|
|
# @block-end initialize-edge-shard
|
|
|
|
# @block-start upsert-points
|
|
from qdrant_edge import ( Point, UpdateOperation )
|
|
|
|
point = Point(
|
|
id=1,
|
|
vector={VECTOR_NAME: [0.1, 0.2, 0.3, 0.4]},
|
|
payload={"color": "red"}
|
|
)
|
|
|
|
edge_shard.update(UpdateOperation.upsert_points([point]))
|
|
# @block-end upsert-points
|
|
|
|
# @block-start retrieve-point
|
|
records = edge_shard.retrieve(
|
|
point_ids=[1],
|
|
with_payload=True,
|
|
with_vector=False
|
|
)
|
|
# @block-end retrieve-point
|
|
|
|
# @block-start query-points
|
|
from qdrant_edge import Query, QueryRequest
|
|
|
|
results = edge_shard.query(
|
|
QueryRequest(
|
|
query=Query.Nearest([0.2, 0.1, 0.9, 0.7], using=VECTOR_NAME),
|
|
limit=10,
|
|
with_vector=False,
|
|
with_payload=True
|
|
)
|
|
)
|
|
# @block-end query-points
|
|
|
|
# @block-start filter
|
|
from qdrant_edge import FieldCondition, Filter, MatchValue
|
|
|
|
results = edge_shard.query(
|
|
QueryRequest(
|
|
query=Query.Nearest([0.2, 0.1, 0.9, 0.7], using=VECTOR_NAME),
|
|
filter=Filter(
|
|
must=[
|
|
FieldCondition(
|
|
key="color",
|
|
match=MatchValue(value="red"),
|
|
)
|
|
]
|
|
),
|
|
limit=10,
|
|
with_vector=False,
|
|
with_payload=True
|
|
)
|
|
)
|
|
# @block-end filter
|
|
|
|
# @block-start facet
|
|
from qdrant_edge import FacetRequest
|
|
|
|
facet_response = edge_shard.facet(FacetRequest(key="color", limit=10, exact=False))
|
|
# @block-end facet
|
|
|
|
# @block-start optimize
|
|
edge_shard.optimize()
|
|
# @block-end optimize
|
|
|
|
# @block-start configure-optimizer
|
|
from qdrant_edge import EdgeOptimizersConfig
|
|
|
|
config = EdgeConfig(
|
|
vectors={
|
|
VECTOR_NAME: EdgeVectorParams(
|
|
size=VECTOR_DIMENSION,
|
|
distance=Distance.Cosine,
|
|
)
|
|
},
|
|
optimizers=EdgeOptimizersConfig(
|
|
deleted_threshold=0.2,
|
|
vacuum_min_vector_number=100,
|
|
default_segment_number=2,
|
|
),
|
|
)
|
|
# @block-end configure-optimizer
|
|
|
|
# @block-start create-payload-index
|
|
from qdrant_edge import PayloadSchemaType
|
|
|
|
edge_shard.update(UpdateOperation.create_field_index("color", PayloadSchemaType.Keyword))
|
|
# @block-end create-payload-index
|
|
|
|
# @block-start close-edge-shard
|
|
edge_shard.close()
|
|
# @block-end close-edge-shard
|
|
|
|
# @block-start load-edge-shard
|
|
edge_shard = EdgeShard.load(SHARD_DIRECTORY)
|
|
# @block-end load-edge-shard
|