mirror of
https://github.com/qdrant/qdrant.git
synced 2026-07-23 11:11:00 -05:00
68e88a56e4
* Inference Service, GRPC (batch processing) --------- Co-authored-by: generall <andrey@vasnetsov.com> Co-authored-by: n0x29a <n0x29a@example.com>
119 lines
2.5 KiB
Bash
119 lines
2.5 KiB
Bash
#!/usr/bin/env bash
|
|
# This test checks that Qdrant answers to all API mentioned in README.md as expected
|
|
|
|
set -ex
|
|
|
|
# Ensure current path is project root
|
|
cd "$(dirname "$0")/../"
|
|
|
|
QDRANT_HOST=${QDRANT_HOST:-'127.0.0.1:6334'}
|
|
GRPCURL_PATH="/opt/homebrew/bin/grpcurl"
|
|
|
|
# Check if local grpcurl exists and is executable
|
|
if [ -x "$GRPCURL_PATH" ]; then
|
|
echo "Using local grpcurl installation"
|
|
grpcurl_base=(
|
|
"$GRPCURL_PATH"
|
|
"-plaintext"
|
|
"-import-path"
|
|
"./lib/api/src/grpc/proto"
|
|
"-proto"
|
|
"./lib/api/src/grpc/proto/qdrant.proto"
|
|
)
|
|
else
|
|
echo "Local grpcurl not found, using Docker container"
|
|
# Define base grpcurl command for Docker in case local grpcurl is not found
|
|
grpcurl_base=(
|
|
"docker" "run" "--rm" "--network=host"
|
|
"-v" "${PWD}/lib/api/src/grpc/proto:/proto"
|
|
"fullstorydev/grpcurl"
|
|
"-plaintext"
|
|
"-import-path" "/proto"
|
|
"-proto" "qdrant.proto"
|
|
)
|
|
fi
|
|
|
|
# Add headers if they exist
|
|
if [ -n "${QDRANT_HOST_HEADERS}" ]; then
|
|
while read h; do
|
|
grpcurl_base+=("-H" "$h")
|
|
done <<< $(echo "${QDRANT_HOST_HEADERS}" | jq -r 'to_entries|map("\(.key): \(.value)")[]')
|
|
fi
|
|
|
|
# Function to execute grpcurl commands
|
|
execute_grpcurl() {
|
|
"${grpcurl_base[@]}" "$@"
|
|
}
|
|
|
|
# Upsert first point
|
|
execute_grpcurl -d '{
|
|
"collection_name": "sparse_charts",
|
|
"wait": true,
|
|
"ordering": null,
|
|
"points": [
|
|
{
|
|
"id": { "num": 1 },
|
|
"vectors": {
|
|
"vectors": {
|
|
"vectors": {
|
|
"keywords": {
|
|
"document": {
|
|
"text": "my text",
|
|
"model": "Qdrant/bm25"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"payload": {
|
|
"city": { "string_value": "Berlin" }
|
|
}
|
|
}
|
|
]
|
|
}' $QDRANT_HOST qdrant.Points/Upsert
|
|
|
|
# Upsert multiple points
|
|
execute_grpcurl -d '{
|
|
"collection_name": "sparse_charts",
|
|
"wait": true,
|
|
"ordering": null,
|
|
"points": [
|
|
{
|
|
"id": { "num": 1 },
|
|
"vectors": {
|
|
"vectors": {
|
|
"vectors": {
|
|
"keywords": {
|
|
"document": {
|
|
"text": "my text",
|
|
"model": "Qdrant/bm25"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"payload": {
|
|
"city": { "string_value": "Berlin" }
|
|
}
|
|
},
|
|
{
|
|
"id": { "num": 2 },
|
|
"vectors": {
|
|
"vectors": {
|
|
"vectors": {
|
|
"keywords": {
|
|
"document": {
|
|
"text": "my text another",
|
|
"model": "Qdrant/bm25"
|
|
}
|
|
}
|
|
}
|
|
}
|
|
},
|
|
"payload": {
|
|
"city": { "string_value": "Amsterdam" }
|
|
}
|
|
}
|
|
]
|
|
}' $QDRANT_HOST qdrant.Points/Upsert
|