gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

small changes to load test

Browse source
This commit is contained in:
Andrej Milicevic 2025-10-28 09:27:33 +01:00
parent 897fbd2f09
commit 2b083dd0f1
1 changed files with 7 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
cognee/tests/test_load.py
View file

@ -48,15 +48,15 @@ async def main():
upper_boundary_minutes = 10
average_minutes = 8
await cognee.prune.prune_data()
await cognee.prune.prune_system(metadata=True)
recorded_times = []
for _ in range(num_of_reps):
await cognee.prune.prune_data()
await cognee.prune.prune_system(metadata=True)
s3_input = "s3://cognee-load-test-s3-bucket"
await cognee.add(s3_input)
s3_input = "s3://cognee-test-load-s3-bucket"
await cognee.add(s3_input)
recorded_times = await asyncio.gather(
*[process_and_search(num_of_pdfs) for _ in range(num_of_reps)]
)
recorded_times.append(await process_and_search(num_of_pdfs))
average_recorded_time = sum(recorded_times) / len(recorded_times)