Fix rerank benchmark script (#107)

Signed-off-by: Li Gang <[email protected]>
opea-project · Sep 5, 2024 · 8edda1c · 8edda1c
1 parent 58502c5
commit 8edda1c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/evals/benchmark/stresscli/locust/rerankingfixed.py b/evals/benchmark/stresscli/locust/rerankingfixed.py
@@ -14,7 +14,7 @@ def getUrl():
 
 
 def getReqData():
-    return ({"initial_query": my_query, "retrieved_docs": [{"text": query_rerank_1}, {"text": query_rerank_2}]},)
+    return {"initial_query": my_query, "retrieved_docs": [{"text": query_rerank_1}, {"text": query_rerank_2}]}
 
 
 def respStatics(environment, resp):

diff --git a/evals/benchmark/stresscli/locust/rerankservefixed.py b/evals/benchmark/stresscli/locust/rerankservefixed.py
@@ -14,7 +14,7 @@ def getUrl():
 
 
 def getReqData():
-    return ({"query": my_query, "texts": [query_rerank_1, query_rerank_2]},)
+    return {"query": my_query, "texts": [query_rerank_1, query_rerank_2]}
 
 
 def respStatics(environment, resp):