123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115 |
- #!/bin/bash
- # Exit immediately for non zero status
- set -e
- # Print commands
- set -x
- SOURCE="${BASH_SOURCE[0]}"
- while [ -h "$SOURCE" ]; do # resolve $SOURCE until the file is no longer a symlink
- DIR="$( cd -P "$( dirname "$SOURCE" )" && pwd )"
- SOURCE="$(readlink "$SOURCE")"
- [[ $SOURCE != /* ]] && SOURCE="$DIR/$SOURCE" # if $SOURCE was a relative symlink, we need to resolve it relative to the path where the symlink file was located
- done
- ROOT="$( cd -P "$( dirname "$SOURCE" )/../.." && pwd )"
- DATA_PATH="${ROOT}/tests/scripts/restful-data/"
- MILVUS_CLUSTER_ENABLED="${MILVUS_CLUSTER_ENABLED:-false}"
- MILVUS_SERVICE_NAME=$(echo "${MILVUS_HELM_RELEASE_NAME}-milvus.${MILVUS_HELM_NAMESPACE}" | tr -d '\n')
- MILVUS_SERVICE_ADDRESS="${MILVUS_SERVICE_NAME}:9091"
- # Create a collection
- curl -X 'POST' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/collection" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d @${DATA_PATH}/create-collection.json
- # Has collection
- curl -X 'GET' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/collection/existence" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book"
- }'
- # Check collection details
- curl -X 'GET' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/collection" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book"
- }'
- # Load collection
- curl -X 'POST' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/collection/load" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book"
- }'
- ### Data
- # Insert Data
- curl -X 'POST' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/entities" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d @${DATA_PATH}/insert-data.json
- # Build Index
- curl -X 'POST' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/index" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book",
- "field_name": "book_intro",
- "extra_params":[
- {"key": "metric_type", "value": "L2"},
- {"key": "index_type", "value": "IVF_FLAT"},
- {"key": "params", "value": "{\"nlist\":1024}"}
- ]
- }'
- # KNN Search
- curl -X 'POST' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/search" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d @${DATA_PATH}/search.json
- # Drop Index
- curl -X 'DELETE' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/index" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book",
- "field_name": "book_intro"
- }'
- # Release collection
- curl -X 'DELETE' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/collection/load" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book"
- }'
- # Drop collection
- curl -X 'DELETE' \
- "http://${MILVUS_SERVICE_ADDRESS}/api/v1/collection" \
- -H 'accept: application/json' \
- -H 'Content-Type: application/json' \
- -d '{
- "collection_name": "book"
- }'
|