From e688fba761d896a80dfa6b05b7c2221620e7b8be Mon Sep 17 00:00:00 2001 From: chuckbutkus Date: Fri, 30 Jan 2026 19:31:46 -0500 Subject: [PATCH] Fix check for already migrated (#12700) --- enterprise/storage/lite_llm_manager.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/enterprise/storage/lite_llm_manager.py b/enterprise/storage/lite_llm_manager.py index 92c57cccd8..7517c3633f 100644 --- a/enterprise/storage/lite_llm_manager.py +++ b/enterprise/storage/lite_llm_manager.py @@ -153,10 +153,13 @@ class LiteLlmManager: max_budget *= billing_margin spend *= billing_margin - # Check if max_budget is None (not 0.0) to determine if already migrated + # Check if max_budget is None (not 0.0) or set to unlimited to determine if already migrated # A user with max_budget=0.0 is different from max_budget=None - if original_max_budget is None: - # if max_budget is None, then we've already migrated the User + if ( + original_max_budget is None + or original_max_budget == UNLIMITED_BUDGET_SETTING + ): + # if max_budget is None or UNLIMITED, then we've already migrated the User logger.info( 'LiteLlmManager:migrate_lite_llm_entries:already_migrated', extra={