diff --git a/pyproject.toml b/pyproject.toml index 58796189..249611bf 100644 --- a/pyproject.toml +++ b/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" diff --git a/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py b/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py index b98fb2c9..7f803ef8 100644 --- a/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py +++ b/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://"): diff --git a/veadk/version.py b/veadk/version.py index 0ab2d4d5..7b43d374 100644 --- a/veadk/version.py +++ b/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"