Skip to content

Commit d2673c1

Browse files
committed
create default collection using app_name
1 parent bd60240 commit d2673c1

File tree

2 files changed

+6
-1
lines changed

2 files changed

+6
-1
lines changed

veadk/knowledgebase/backends/vikingdb_knowledge_backend.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -124,6 +124,7 @@ def model_post_init(self, __context: Any) -> None:
124124
logger.warning(
125125
f"VikingDB knowledgebase collection {self.index} does not exist, please create it first..."
126126
)
127+
self.create_collection()
127128

128129
def precheck_index_naming(self):
129130
if not (
@@ -443,6 +444,9 @@ def create_collection(self) -> None:
443444
path=CREATE_COLLECTION_PATH,
444445
method="POST",
445446
)
447+
logger.debug(
448+
f"Create collection {self.index} using project {self.volcengine_project} response: {response}"
449+
)
446450

447451
if response.get("code") != 0:
448452
raise ValueError(

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,8 @@ def model_post_init(self, __context: Any) -> None:
6262
# "event_1, event_2" -> ["event_1", "event_2"]
6363
self.memory_type = [x.strip() for x in env_memory_type.split(",")]
6464
else:
65-
self.memory_type = ["sys_event_v1", "event_v1"]
65+
# self.memory_type = ["sys_event_v1", "event_v1"]
66+
self.memory_type = ["sys_event_v1"]
6667

6768
logger.info(f"Using memory type: {self.memory_type}")
6869

0 commit comments

Comments
 (0)