diff --git a/run_retrieve.sh b/run_retrieve.sh new file mode 100755 index 0000000..c56d584 --- /dev/null +++ b/run_retrieve.sh @@ -0,0 +1,29 @@ +#!/bin/bash +# This shell script will handle I/O for the python query engine +# It will take a query and return the formatted results + +# E.M.F. August 2025 + +# Usage: ./run_query.sh + +QUERY_SCRIPT="retrieve.py" + +echo -e "$QUERY_SCRIPT -- retrieve vector store chunks based on similaity + BM25 with reranking.\n" + +# Loop until input is "exit" +while true; do + read -p "Enter your query (or type 'exit' to quit): " query + if [ "$query" == "exit" ] || [ "$query" == "quit" ] || [ "$query" == "" ] ; then + echo "Exiting..." + break + fi + time_start=$(date +%s) + + # Call the python script with the query and format the output + python3 $QUERY_SCRIPT --query "$query" | \ + expand | sed -E 's|(.* )(.*/data)|\1./data|' | fold -s -w 131 + + time_end=$(date +%s) + elapsed=$((time_end - time_start)) + echo -e "Query processed in $elapsed seconds.\n" +done