From 6acfb162e4e58ee8aab141fd1ca8421ad1046e3c Mon Sep 17 00:00:00 2001 From: Fangyin Cheng Date: Tue, 4 Mar 2025 21:05:33 +0800 Subject: [PATCH] fix(datasource): Fix datasource resource error --- packages/dbgpt-core/src/dbgpt/agent/resource/database.py | 1 + packages/dbgpt-core/src/dbgpt/core/interface/variables.py | 2 +- 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/packages/dbgpt-core/src/dbgpt/agent/resource/database.py b/packages/dbgpt-core/src/dbgpt/agent/resource/database.py index 2cf452768..ed307325e 100644 --- a/packages/dbgpt-core/src/dbgpt/agent/resource/database.py +++ b/packages/dbgpt-core/src/dbgpt/agent/resource/database.py @@ -39,6 +39,7 @@ class DBResource(Resource[P], Generic[P]): dialect: Optional[str] = None, executor: Optional[Executor] = None, prompt_template: str = _DEFAULT_PROMPT_TEMPLATE, + **kwargs, ): """Initialize the DB resource.""" self._name = name diff --git a/packages/dbgpt-core/src/dbgpt/core/interface/variables.py b/packages/dbgpt-core/src/dbgpt/core/interface/variables.py index e6d6d0293..f3b1daeab 100644 --- a/packages/dbgpt-core/src/dbgpt/core/interface/variables.py +++ b/packages/dbgpt-core/src/dbgpt/core/interface/variables.py @@ -66,7 +66,7 @@ def _generate_key_from_password( algorithm=hashes.SHA256(), length=32, salt=salt, - iterations=100000, + iterations=800000, ) key = base64.urlsafe_b64encode(kdf.derive(password)) return key, salt