Skip to content

Commit 3a02bf9

Browse files
fix(memory): fix env name of viking db memory type (#249)
1 parent 8d34b39 commit 3a02bf9

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

veadk/memory/long_term_memory_backends/vikingdb_memory_backend.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def model_post_init(self, __context: Any) -> None:
5757
# 2. environment variable
5858
# 3. default value
5959
if not self.memory_type:
60-
env_memory_type = os.getenv("DATABASE_VIKING_MEMORY_TYPE")
60+
env_memory_type = os.getenv("DATABASE_VIKINGMEM_MEMORY_TYPE")
6161
if env_memory_type:
6262
# "event_1, event_2" -> ["event_1", "event_2"]
6363
self.memory_type = [x.strip() for x in env_memory_type.split(",")]
@@ -84,17 +84,23 @@ def _collection_exist(self) -> bool:
8484
try:
8585
client = self._get_client()
8686
client.get_collection(collection_name=self.index)
87+
logger.info(f"Collection {self.index} exist.")
8788
return True
8889
except Exception:
90+
logger.info(f"Collection {self.index} not exist.")
8991
return False
9092

9193
def _create_collection(self) -> None:
94+
logger.info(
95+
f"Create collection with collection_name={self.index}, builtin_event_types={self.memory_type}"
96+
)
9297
client = self._get_client()
9398
response = client.create_collection(
9499
collection_name=self.index,
95100
description="Created by Volcengine Agent Development Kit VeADK",
96-
builtin_event_types=["sys_event_v1"],
101+
builtin_event_types=self.memory_type,
97102
)
103+
logger.debug(f"Create collection with response {response}")
98104
return response
99105

100106
def _get_client(self) -> VikingDBMemoryClient:

0 commit comments

Comments
 (0)