From 4955e29f7d2da54620656c4c14fb4aad1a4ae60d Mon Sep 17 00:00:00 2001
From: yumoqing <yumoqing@gmail.com>
Date: Fri, 31 Jul 2026 15:48:43 +0800
Subject: [PATCH] fix: re-export reserve/finalize/refund_balance from
 balance.py in utils.py (init.py imports them from .utils)

---
 llmage/utils.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/llmage/utils.py b/llmage/utils.py
index 5514447..ecfc953 100644
--- a/llmage/utils.py
+++ b/llmage/utils.py
@@ -638,3 +638,6 @@ async def update_model_max_cost(llmid, new_max):
 	async with get_sor_context(env, 'llmage') as sor:
 		await sor.U('llm', {'id': llmid, 'max_cost': float(new_max)})
 
+
+# Redis balance reservation (implemented in balance.py)
+from .balance import reserve_balance, finalize_balance, refund_balance