Skip to content

Commit 693cf04

Browse files
authored
chore(kb): modify vikingdb knowledge create_collection version field form 4 to 2 (#387)
1 parent 768c6e4 commit 693cf04

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

veadk/knowledgebase/backends/vikingdb_knowledge_backend.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,6 +117,10 @@ class VikingDBKnowledgeBackend(BaseKnowledgebaseBackend):
117117
default_factory=lambda: os.getenv("DATABASE_VIKING_PROJECT", "default")
118118
)
119119

120+
version: str = Field(
121+
default_factory=lambda: os.getenv("DATABASE_VIKING_VERSION", "2")
122+
)
123+
120124
region: str = Field(
121125
default_factory=lambda: os.getenv("DATABASE_VIKING_REGION", "cn-beijing")
122126
)
@@ -457,12 +461,16 @@ def collection_status(self):
457461

458462
def create_collection(self) -> None:
459463
CREATE_COLLECTION_PATH = "/api/knowledge/collection/create"
460-
464+
if self.version not in ["2", "4"]:
465+
raise ValueError(
466+
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`"
467+
)
461468
response = self._do_request(
462469
body={
463470
"name": self.index,
464471
"project": self.volcengine_project,
465472
"description": "Created by Volcengine Agent Development Kit (VeADK).",
473+
"version": int(self.version),
466474
},
467475
path=CREATE_COLLECTION_PATH,
468476
method="POST",

0 commit comments

Comments
 (0)