diff --git a/tests/unit/images/test_main.py b/tests/unit/images/test_main.py index 84a04405483..055e661ae76 100644 --- a/tests/unit/images/test_main.py +++ b/tests/unit/images/test_main.py @@ -46,7 +46,7 @@ async def test_router_image_edit_bills_the_deployment_price_with_a_logger_built_ logging_obj, routed_request = litellm.utils.function_setup( original_function="aimage_edit", rules_obj=litellm.utils.Rules(), - start_time=datetime.now(), + start_time=datetime(2026, 1, 1), **request, ) diff --git a/tests/unit/llms/azure_ai/image_edit/test_azure_ai_image_edit_transformation.py b/tests/unit/llms/azure_ai/image_edit/test_azure_ai_image_edit_transformation.py index 0517d693b32..749ce0596da 100644 --- a/tests/unit/llms/azure_ai/image_edit/test_azure_ai_image_edit_transformation.py +++ b/tests/unit/llms/azure_ai/image_edit/test_azure_ai_image_edit_transformation.py @@ -604,7 +604,7 @@ async def test_flux2_router_image_edit_bills_the_deployment_rates_with_a_logger_ "litellm_call_id": "proxy-call-id", } logging_obj, routed_request = litellm.utils.function_setup( - original_function="aimage_edit", rules_obj=litellm.utils.Rules(), start_time=datetime.now(), **request + original_function="aimage_edit", rules_obj=litellm.utils.Rules(), start_time=datetime(2026, 1, 1), **request ) response: Final = await router.aimage_edit(**routed_request, litellm_logging_obj=logging_obj)