release 0.2.31

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "veadk-python"
-version = "0.2.30"
+version = "0.2.31"
 description = "Volcengine agent development kit, integrations with Volcengine cloud services."
 readme = "README.md"
 requires-python = ">=3.10"

veadk/knowledgebase/backends/vikingdb_knowledge_backend.py

@@ -22,16 +22,18 @@
 import requests
 from pydantic import Field
 from typing_extensions import override
-from veadk.utils.misc import getenv
+from volcengine.viking_knowledgebase import VikingKnowledgeBaseService
+
 import veadk.config  # noqa E401
 from veadk.auth.veauth.utils import get_credential_from_vefaas_iam
 from veadk.configs.database_configs import NormalTOSConfig, TOSConfig
 from veadk.knowledgebase.backends.base_backend import BaseKnowledgebaseBackend
-from veadk.knowledgebase.backends.utils import build_vikingdb_knowledgebase_request
+from veadk.knowledgebase.backends.utils import (
+    build_vikingdb_knowledgebase_request,
+)
 from veadk.knowledgebase.entry import KnowledgebaseEntry
 from veadk.utils.logger import get_logger
-from veadk.utils.misc import formatted_timestamp
-from volcengine.viking_knowledgebase import VikingKnowledgeBaseService
+from veadk.utils.misc import formatted_timestamp, getenv
 
 try:
     from veadk.integrations.ve_tos.ve_tos import VeTOS
@@ -111,9 +113,14 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
     )
     session_token: str = ""
 
-    volcengine_project: str = "default"
+    volcengine_project: str = Field(
+        default_factory=lambda: os.getenv("DATABASE_VIKING_PROJECT", "default")
+    )
+
+    region: str = Field(
+        default_factory=lambda: os.getenv("DATABASE_VIKING_REGION", "cn-beijing")
+    )
 
-    region: str = "cn-beijing"
     base_url: str = "https://api-knowledgebase.mlp.cn-beijing.volces.com"
     host: str = "api-knowledgebase.mlp.cn-beijing.volces.com"
     schema: str = "https"
@@ -562,7 +569,9 @@ def _search_knowledge(
 
     def _set_service_info(self):
         env_host = getenv(
-            "DATABASE_VIKING_BASE_URL", default_value=None, allow_false_values=True
+            "DATABASE_VIKING_BASE_URL",
+            default_value=None,
+            allow_false_values=True,
         )
         if env_host:
             if env_host.startswith("http://") or env_host.startswith("https://"):

veadk/version.py

@@ -12,4 +12,4 @@
 # See the License for the specific language governing permissions and
 # limitations under the License.
 
-VERSION = "0.2.30"
+VERSION = "0.2.31"