phoenix/litellm
1
0
Fork 0
forked from phoenix/litellm-mirror
Code Pull requests Activity

(fix) locustfile used in load test

Browse source
This commit is contained in:
ishaan-jaff 2024-03-15 12:38:37 -07:00
parent 310811681a
commit f3209b63cd
1 changed files with 0 additions and 20 deletions
Download patch file Download diff file Expand all files Collapse all files

20
litellm/proxy/proxy_load_test/locustfile.py
View file

@ -31,28 +31,8 @@ class MyUser(HttpUser):
@task(10) @task(10)
def health_readiness(self): def health_readiness(self):
start_time = time.time()
response = self.client.get("health/readiness") response = self.client.get("health/readiness")
response_time = time.time() - start_time
if response_time > 1:
events.request_failure.fire(
request_type="GET",
name="health/readiness",
response_time=response_time,
exception=None,
response=response,
)
@task(10) @task(10)
def health_liveliness(self): def health_liveliness(self):
start_time = time.time()
response = self.client.get("health/liveliness") response = self.client.get("health/liveliness")
response_time = time.time() - start_time
if response_time > 1:
events.request_failure.fire(
request_type="GET",
name="health/liveliness",
response_time=response_time,
exception=None,
response=response,
)