mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-23 11:11:00 -05:00
92 lines
2.1 KiB
Bash
Executable File
92 lines
2.1 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
# This test checks that Qdrant answers to all API mentioned in README.md as expected
|
|
|
|
|
|
set -ex
|
|
|
|
QDRANT_HOST='localhost:6333'
|
|
|
|
# cleanup collection if it exists
|
|
curl -X DELETE "http://$QDRANT_HOST/collections/test_collection" \
|
|
-H 'Content-Type: application/json' \
|
|
--fail -s | jq
|
|
|
|
# create collection
|
|
curl -X PUT "http://$QDRANT_HOST/collections/test_collection" \
|
|
-H 'Content-Type: application/json' \
|
|
--fail -s \
|
|
--data-raw '{
|
|
"vectors": {
|
|
"size": 4,
|
|
"distance": "Dot"
|
|
},
|
|
"optimizers_config": {
|
|
"default_segment_number": 2
|
|
}
|
|
}' | jq
|
|
|
|
|
|
for i in {1..100}
|
|
do
|
|
IDX=$i
|
|
|
|
IDX2=$(expr $IDX + 1000)
|
|
|
|
PAYLOAD=$( jq -n \
|
|
--argjson point_id "$IDX" \
|
|
--argjson point_id_2 "$IDX2" \
|
|
'{ "points": [
|
|
{"id": $point_id, "vector": [0.19, 0.81, 0.75, 0.11], "payload": {"city": "London" }},
|
|
{"id": $point_id_2, "vector": [0.05, 0.61, 0.76, 0.74], "payload": {"city": "Berlin" }}
|
|
]}')
|
|
|
|
# insert points
|
|
curl -L -X PUT "http://$QDRANT_HOST/collections/test_collection/points?wait=true" \
|
|
-H 'Content-Type: application/json' \
|
|
--fail -s \
|
|
--data-raw "$PAYLOAD" | jq
|
|
done
|
|
|
|
PAYLOAD='
|
|
{
|
|
"filter": {
|
|
"should": [
|
|
{
|
|
"key": "city",
|
|
"match": {
|
|
"value": "London"
|
|
}
|
|
}
|
|
]
|
|
}
|
|
}
|
|
';
|
|
|
|
# insert points
|
|
curl -L -X POST "http://$QDRANT_HOST/collections/test_collection/points/delete?wait=true" \
|
|
-H 'Content-Type: application/json' \
|
|
--fail -s \
|
|
--data-raw "$PAYLOAD" | jq
|
|
|
|
|
|
for i in {1..10}
|
|
do
|
|
IDX=$i
|
|
|
|
IDX2=$(expr $IDX + 1000)
|
|
|
|
PAYLOAD=$( jq -n \
|
|
--argjson point_id "$IDX" \
|
|
--argjson point_id_2 "$IDX2" \
|
|
'{ "points": [
|
|
{"id": $point_id, "vector": [0.19, 0.81, 0.75, 0.11], "payload": {"city": "London" }},
|
|
{"id": $point_id_2, "vector": [0.05, 0.61, 0.76, 0.74], "payload": {"city": "Berlin" }}
|
|
]}')
|
|
|
|
# insert points
|
|
curl -L -X PUT "http://$QDRANT_HOST/collections/test_collection/points?wait=true" \
|
|
-H 'Content-Type: application/json' \
|
|
--fail -s \
|
|
--data-raw "$PAYLOAD" | jq
|
|
done
|