From 3dfe2bfd85e6fb619cb4c5148cd91999e486408b Mon Sep 17 00:00:00 2001 From: Ishaan Jaff Date: Mon, 12 Aug 2024 16:33:35 -0700 Subject: [PATCH] tes logging to gcs buckets --- litellm/tests/test_gcs_bucket.py | 13 +++++++++++-- 1 file changed, 11 insertions(+), 2 deletions(-) diff --git a/litellm/tests/test_gcs_bucket.py b/litellm/tests/test_gcs_bucket.py index 4fa9d8ef43..607599d903 100644 --- a/litellm/tests/test_gcs_bucket.py +++ b/litellm/tests/test_gcs_bucket.py @@ -117,8 +117,17 @@ async def test_basic_gcs_logger(): await asyncio.sleep(5) + # Get the current date + # Get the current date + current_date = datetime.now().strftime("%Y-%m-%d") + + # Modify the object_name to include the date-based folder + object_name = f"{current_date}%2F{response.id}" + + print("object_name", object_name) + # Check if object landed on GCS - object_from_gcs = await gcs_logger.download_gcs_object(object_name=response.id) + object_from_gcs = await gcs_logger.download_gcs_object(object_name=object_name) print("object from gcs=", object_from_gcs) # convert object_from_gcs from bytes to DICT parsed_data = json.loads(object_from_gcs) @@ -151,4 +160,4 @@ async def test_basic_gcs_logger(): # Delete Object from GCS print("deleting object from GCS") - await gcs_logger.delete_gcs_object(object_name=object_name) + # await gcs_logger.delete_gcs_object(object_name=response.id)