Browse Source

add tidb spend limit config (#9999)

tags/0.11.0
Jyong 1 year ago
parent
commit
5580bcf870
No account linked to committer's email address

+ 5
- 0
api/configs/middleware/vdb/tidb_on_qdrant_config.py View File

description="Tidb project id", description="Tidb project id",
default=None, default=None,
) )

TIDB_SPEND_LIMIT: Optional[int] = Field(
description="Tidb spend limit",
default=100,
)

+ 2
- 1
api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py View File

import requests import requests
from requests.auth import HTTPDigestAuth from requests.auth import HTTPDigestAuth


from configs import dify_config
from extensions.ext_database import db from extensions.ext_database import db
from extensions.ext_redis import redis_client from extensions.ext_redis import redis_client
from models.dataset import TidbAuthBinding from models.dataset import TidbAuthBinding
} }


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

Loading…
Cancel
Save