update locomo_rag bash

Author: CSLiuPeng

evaluation/scripts/locomo/locomo_rag.py

@@ -272,6 +272,7 @@ def generate_response_file(file_path):
     )
     parser.add_argument("--chunk_size", type=int, default=2000, help="Chunk size for processing")
     parser.add_argument("--num_chunks", type=int, default=2, help="Number of chunks to process")
+    parser.add_argument("--frame", type=str, default="rag")
 
     args = parser.parse_args()
     with open(file_path) as f:

evaluation/scripts/run_rag_eval.sh

@@ -2,14 +2,19 @@
 LIB="rag"
 VERSION="default"
 DATA_SET="locomo"
-
+export HF_ENDPOINT=https://hf-mirror.com
 mkdir -p results/$DATA_SET/$LIB-$VERSION/
 echo "The result saved in：results/$DATA_SET/$LIB-$VERSION/"
 
 echo "The complete evaluation steps for generating the RAG and full context!"
 
 echo "Running locomo_rag.py..."
-python scripts/locomo/locomo_rag.py --chunk_size 128 --num_chunks 1 --output_folder results/$DATA_SET/$LIB-$VERSION/
+python scripts/locomo/locomo_rag.py \
+    --chunk_size 128 \
+    --num_chunks 1 \
+    --frame $LIB \
+    --output_folder "results/$DATA_SET/$LIB-$VERSION/"
+
 if [ $? -ne 0 ]; then
     echo "Error running locomo_rag.py"
     exit 1