test/dify
1
0
Fork 0
mirror of https://gitee.com/dify_ai/dify.git synced 2024-12-03 11:48:08 +08:00
Code Issues Actions 4 Packages Projects Releases Wiki Activity

add tidb spend limit config (#9999)

Browse Source
This commit is contained in:
Jyong 2024-10-29 17:51:13 +08:00 committed by GitHub
parent 12adcf8925
commit 5580bcf870
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 7 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
api/configs/middleware/vdb/tidb_on_qdrant_config.py
View File

@ -63,3 +63,8 @@ class TidbOnQdrantConfig(BaseSettings):
description="Tidb project id",
default=None,
)
TIDB_SPEND_LIMIT: Optional[int] = Field(
description="Tidb spend limit",
default=100,
)

3
api/core/rag/datasource/vdb/tidb_on_qdrant/tidb_service.py
View File

@ -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("-", "")