github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-07-24 20:47:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix(datasource): Fix datasource resource error

Browse Source
This commit is contained in:
Fangyin Cheng 2025-03-04 21:05:33 +08:00
parent 948a93be32
commit 6acfb162e4
2 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
packages/dbgpt-core/src/dbgpt/agent/resource/database.py
View File

@ -39,6 +39,7 @@ class DBResource(Resource[P], Generic[P]):
dialect: Optional[str] = None, dialect: Optional[str] = None,
executor: Optional[Executor] = None, executor: Optional[Executor] = None,
prompt_template: str = _DEFAULT_PROMPT_TEMPLATE, prompt_template: str = _DEFAULT_PROMPT_TEMPLATE,
**kwargs,
): ):
"""Initialize the DB resource.""" """Initialize the DB resource."""
self._name = name self._name = name

2
packages/dbgpt-core/src/dbgpt/core/interface/variables.py
View File

@ -66,7 +66,7 @@ def _generate_key_from_password(
algorithm=hashes.SHA256(), algorithm=hashes.SHA256(),
length=32, length=32,
salt=salt, salt=salt,
iterations=100000, iterations=800000,
) )
key = base64.urlsafe_b64encode(kdf.derive(password)) key = base64.urlsafe_b64encode(kdf.derive(password))
return key, salt return key, salt