From 934b49323bcb71416fd45ce5788e8b4496df77a5 Mon Sep 17 00:00:00 2001 From: "hanzhi.421" Date: Wed, 17 Dec 2025 20:01:56 +0800 Subject: [PATCH] chore(kb): modify vikingdb knowledge create_collection version field form 4 to 2 --- .../backends/vikingdb_knowledge_backend.py | 10 +++++++++- 1 file changed, 9 insertions(+), 1 deletion(-) diff --git a/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py b/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py index 7f803ef8..11bdb96b 100644 --- a/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py +++ b/veadk/knowledgebase/backends/vikingdb_knowledge_backend.py @@ -117,6 +117,10 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend): default_factory=lambda: os.getenv("DATABASE_VIKING_PROJECT", "default") ) + version: str = Field( + default_factory=lambda: os.getenv("DATABASE_VIKING_VERSION", "2") + ) + region: str = Field( default_factory=lambda: os.getenv("DATABASE_VIKING_REGION", "cn-beijing") ) @@ -457,12 +461,16 @@ def collection_status(self): def create_collection(self) -> None: CREATE_COLLECTION_PATH = "/api/knowledge/collection/create" - + if self.version not in ["2", "4"]: + raise ValueError( + f"The version number must be 2 or 4. The current value: {self.version}. For details, please refer to: `https://www.volcengine.com/docs/84313/1254593?lang=zh`" + ) response = self._do_request( body={ "name": self.index, "project": self.volcengine_project, "description": "Created by Volcengine Agent Development Kit (VeADK).", + "version": int(self.version), }, path=CREATE_COLLECTION_PATH, method="POST",