From 2cfdd0b3f57f37a6a9f741317641be2cfe9e9cae Mon Sep 17 00:00:00 2001 From: bennykok Date: Wed, 28 Aug 2024 19:25:35 -0700 Subject: [PATCH] chore: add log --- custom_routes.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/custom_routes.py b/custom_routes.py index 128b89b..9af7b80 100644 --- a/custom_routes.py +++ b/custom_routes.py @@ -1191,10 +1191,13 @@ async def upload_file(prompt_id, filename, subfolder=None, content_type="image/p target_url = f"{file_upload_endpoint}?file_name={filename}&run_id={prompt_id}&type={content_type}&version=v2" start_time = time.time() # Start timing here + logger.info(f"Target URL: {target_url}") result = await async_request_with_retry("GET", target_url, disable_timeout=True) end_time = time.time() # End timing after the request is complete logger.info("Time taken for getting file upload endpoint: {:.2f} seconds".format(end_time - start_time)) ok = await result.json() + + logger.info(f"Result: {ok}") start_time = time.time() # Start timing here headers = {