add tidb spend limit config (#9999)

2024-10-29 17:51:13 +08:00
parent 12adcf8925
commit 5580bcf870
2 changed files with 7 additions and 1 deletions
--- a/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py
+++ b/api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py
@@ -4,6 +4,7 @@ import uuid
 import requests
 from requests.auth import HTTPDigestAuth

+from configs import dify_config
 from extensions.ext_database import db
 from extensions.ext_redis import redis_client
 from models.dataset import TidbAuthBinding
@@ -208,7 +209,7 @@ class TidbService:
            }

            spending_limit = {
-                "monthly": 10,
+                "monthly": dify_config.TIDB_SPEND_LIMIT,
            }
            password = str(uuid.uuid4()).replace("-", "")[:16]
            display_name = str(uuid.uuid4()).replace("-", "")