Skip to content

Commit a8a7978

Browse files
committed
client fix routing strtategy/bug
Signed-off-by: Le Xu <[email protected]>
1 parent 30bc9c6 commit a8a7978

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

benchmarks/client/client.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -238,8 +238,7 @@ async def benchmark_batch(client: openai.AsyncOpenAI,
238238
task = asyncio.create_task(
239239
send_request_batch(client = client,
240240
model = requests[i]["model"] if "model" in requests[i] else default_model,
241-
endpoint = endpoint,
242-
formatted_prompt = formatted_prompts[i],
241+
prompt = formatted_prompts[i],
243242
output_file = output_file,
244243
request_id = request_id)
245244
)
@@ -296,7 +295,7 @@ def main(args):
296295
parser.add_argument("--api-key", type=str, required=True, help="API key to the service. ")
297296
parser.add_argument('--output-file-path', type=str, default="output.jsonl")
298297
parser.add_argument("--streaming", action="store_true", help="Use streaming client.")
299-
parser.add_argument("--routing-strategy", type=str, required=False, default=None, help="Routing strategy to use.")
298+
parser.add_argument("--routing-strategy", type=str, required=False, default="random", help="Routing strategy to use.")
300299

301300
args = parser.parse_args()
302301
main(args)

0 commit comments

Comments
 (0)