AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-14 03:06:02 +08:00
Code Issues 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", description="Tidb project id",
default=None, 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 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
@ -208,7 +209,7 @@ class TidbService:
} }
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("-", "")