Skip to content

Commit 05628ab

Browse files
authored
fix(feat0): fix issues of mem0 backend (#211)
* support mem0 as LTM * reset test_long_term_memory.py * add mem0 dependency * support host/base_url configuration of mem0
1 parent df4b1e8 commit 05628ab

File tree

2 files changed

+9
-4
lines changed

2 files changed

+9
-4
lines changed

config.yaml.full

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -120,8 +120,8 @@ database:
120120
region: cn-beijing # default Volcengine TOS region
121121
bucket:
122122
mem0:
123-
base_url:
124-
api_key:
123+
base_url: # default "https://api.mem0.ai/v1", using full https url including port
124+
api_key: #api_key
125125

126126

127127

veadk/memory/long_term_memory_backends/mem0_backend.py

Lines changed: 7 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -46,12 +46,17 @@ def model_post_init(self, __context: Any) -> None:
4646

4747
try:
4848
self._mem0_client = MemoryClient(
49-
# base_url=self.mem0_config.base_url, # mem0 endpoint
49+
host=self.mem0_config.base_url, # mem0 endpoint
5050
api_key=self.mem0_config.api_key, # mem0 API key
5151
)
52+
logger.info(
53+
f"Initialized Mem0 client for host: {self.mem0_config.base_url}"
54+
)
5255
logger.info(f"Initialized Mem0 client for index: {self.index}")
5356
except Exception as e:
54-
logger.error(f"Failed to initialize Mem0 client: {str(e)}")
57+
logger.error(
58+
f"Failed to initialize Mem0 client for host {self.mem0_config.base_url} : {str(e)}"
59+
)
5560
raise
5661

5762
def precheck_index_naming(self):

0 commit comments

Comments
 (0)