@@ -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