Skip to content

Commit 51542c1

Browse files
committed
fix ltm checks
1 parent f613950 commit 51542c1

File tree

1 file changed

+14
-17
lines changed

1 file changed

+14
-17
lines changed

veadk/memory/long_term_memory.py

Lines changed: 14 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -84,24 +84,21 @@ def model_post_init(self, __context: Any) -> None:
8484
logger.info(
8585
f"Initialized long term memory with provided backend instance {self._backend.__class__.__name__}"
8686
)
87+
elif self.backend_config:
88+
logger.info(
89+
f"Initialized long term memory backend {self.backend} with config."
90+
)
91+
self._backend = BACKEND_CLS[self.backend](**self.backend_config)
92+
elif self.app_name and self.user_id:
93+
self.index = build_long_term_memory_index(
94+
app_name=self.app_name, user_id=self.user_id
95+
)
96+
logger.info(f"Long term memory index set to {self.index}.")
97+
self._backend = BACKEND_CLS[self.backend](index=self.index)
8798
else:
88-
if self.backend_config:
89-
logger.info(
90-
f"Initialized long term memory backend {self.backend} with config."
91-
)
92-
self._backend = BACKEND_CLS[self.backend](**self.backend_config)
93-
elif self.app_name and self.user_id:
94-
self.index = build_long_term_memory_index(
95-
app_name=self.app_name, user_id=self.user_id
96-
)
97-
logger.info(f"Long term memory index set to {self.index}.")
98-
self._backend = BACKEND_CLS[self.backend](
99-
**self.backend_config, index=self.index
100-
)
101-
else:
102-
logger.warning(
103-
"Neither `backend_instance`, `backend_config`, nor `app_name`/`user_id` is provided, the long term memory storage will initialize when adding a session."
104-
)
99+
logger.warning(
100+
"Neither `backend_instance`, `backend_config`, nor `app_name`/`user_id` is provided, the long term memory storage will initialize when adding a session."
101+
)
105102

106103
def _filter_and_convert_events(self, events: list[Event]) -> list[str]:
107104
final_events = []

0 commit comments

Comments
 (0)