MCPcopy Create free account
hub / github.com/AQ-MedAI/MedMemoryBench / memorize

Method memorize

methods/memos_agent.py:231–351  ·  view source on GitHub ↗

Store memory using MemOS's official extract() and add() methods. This uses MemOS's native LLM-based memory extraction pipeline. Args: text: Input text to memorize. **kwargs: Additional arguments (unused). Returns: MemoryBuildRes

(self, text: str, **kwargs)

Source from the content-addressed store, hash-verified

229 return system
230
231 def memorize(self, text: str, **kwargs) -> MemoryBuildResult:
232 """Store memory using MemOS's official extract() and add() methods.
233
234 This uses MemOS's native LLM-based memory extraction pipeline.
235
236 Args:
237 text: Input text to memorize.
238 **kwargs: Additional arguments (unused).
239
240 Returns:
241 MemoryBuildResult with extraction and storage details.
242 """
243 context_id = self._get_context_id()
244 memory_system = self._get_memory_system(context_id)
245
246 # Truncate input text
247 bounded_text = self._truncate_to_tokens(text, self.max_input_tokens)
248
249 # Convert text to message format for MemOS extract()
250 messages = [{"role": "user", "content": bounded_text}]
251
252 start_time = time.time()
253 all_extraction_results: List[Dict[str, Any]] = []
254 all_stored_memories: List[Dict[str, Any]] = []
255 extraction_time = 0.0
256 add_time = 0.0
257
258 try:
259 # Use MemOS's official extract() method
260 # This calls MemOS's LLM-based extractor internally
261 extracted_items = memory_system.extract(messages)
262 extraction_time = time.time() - start_time
263
264 # Record extraction results
265 for item in extracted_items:
266 memory_text = getattr(item, "memory", str(item))
267 metadata = getattr(item, "metadata", {})
268 if hasattr(metadata, "model_dump"):
269 metadata = metadata.model_dump()
270
271 all_extraction_results.append({
272 "memory": memory_text,
273 "metadata": metadata,
274 })
275
276 # Use MemOS's official add() method to store extracted memories
277 if extracted_items:
278 add_start = time.time()
279 memory_system.add(extracted_items)
280 add_time = time.time() - add_start
281
282 for item in extracted_items:
283 memory_text = getattr(item, "memory", str(item))
284 metadata = getattr(item, "metadata", {})
285 if hasattr(metadata, "model_dump"):
286 metadata = metadata.model_dump()
287
288 all_stored_memories.append({

Callers

nothing calls this directly

Calls 9

_get_context_idMethod · 0.95
_get_memory_systemMethod · 0.95
_truncate_to_tokensMethod · 0.95
MemoryBuildResultClass · 0.85
errorMethod · 0.65
extractMethod · 0.45
model_dumpMethod · 0.45
addMethod · 0.45
getMethod · 0.45

Tested by

no test coverage detected