diff --git a/api/core/app/llm/__init__.py b/api/core/app/llm/__init__.py
index d20a5b2344..85f342de5d 100644
--- a/api/core/app/llm/__init__.py
+++ b/api/core/app/llm/__init__.py
@@ -3,13 +3,17 @@
 from .quota import (
     deduct_llm_quota,
     deduct_llm_quota_for_model,
+    deduct_model_quota,
     ensure_llm_quota_available,
     ensure_llm_quota_available_for_model,
+    ensure_model_quota_available,
 )
 
 __all__ = [
     "deduct_llm_quota",
     "deduct_llm_quota_for_model",
+    "deduct_model_quota",
     "ensure_llm_quota_available",
     "ensure_llm_quota_available_for_model",
+    "ensure_model_quota_available",
 ]