From 48b9250a3d022df7ded71fc732337cf3fc51b6d1 Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Fri, 22 Mar 2024 19:44:16 -0700 Subject: [PATCH] (fix) update load test used --- litellm/proxy/proxy_load_test/locustfile.py | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/litellm/proxy/proxy_load_test/locustfile.py b/litellm/proxy/proxy_load_test/locustfile.py index f439f7274..263c871ef 100644 --- a/litellm/proxy/proxy_load_test/locustfile.py +++ b/litellm/proxy/proxy_load_test/locustfile.py @@ -6,11 +6,11 @@ import time class MyUser(HttpUser): wait_time = between(1, 5) - @task(3) + @task def chat_completion(self): headers = { "Content-Type": "application/json", - "Authorization": f"Bearer sk-mh3YNUDs1d_f6fMXfvEqBA", + "Authorization": f"Bearer sk-1234", # Include any additional headers you may need for authentication, etc. } @@ -28,11 +28,3 @@ class MyUser(HttpUser): response = self.client.post("chat/completions", json=payload, headers=headers) # Print or log the response if needed - - @task(10) - def health_readiness(self): - response = self.client.get("health/readiness") - - @task(10) - def health_liveliness(self): - response = self.client.get("health/liveliness")