JDSS/mem0
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

perf: skip ThreadPoolExecutor when graph store is disabled

Browse Source 
Avoids spawning 2 threads per add() call when graph is not configured.
Fixes thread leakage under high concurrency (was accumulating 3500+ threads
per worker). Calls _add_to_vector_store directly in the current thread.

Applied to both sync Memory and async AsyncMemory classes.
This commit is contained in:
Soumil Rathi
2026-04-03 07:42:32 -07:00
parent cb64d978fc
commit a6130c5436
1 changed files with 18 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mem0/memory/main.py
+18 -16
View File
@@ -454,21 +454,22 @@ class Memory(MemoryBase):
else:
messages = parse_vision_messages(messages)
with concurrent.futures.ThreadPoolExecutor() as executor:
future1 = executor.submit(self._add_to_vector_store, messages, processed_metadata, effective_filters, infer, timestamp)
future2 = executor.submit(self._add_to_graph, messages, effective_filters)
concurrent.futures.wait([future1, future2])
vector_store_result = future1.result()
graph_result = future2.result()
if self.enable_graph:
with concurrent.futures.ThreadPoolExecutor() as executor:
future1 = executor.submit(self._add_to_vector_store, messages, processed_metadata, effective_filters, infer, timestamp)
future2 = executor.submit(self._add_to_graph, messages, effective_filters)
concurrent.futures.wait([future1, future2])
vector_store_result = future1.result()
graph_result = future2.result()
return {
"results": vector_store_result,
"relations": graph_result,
}
vector_store_result = self._add_to_vector_store(messages, processed_metadata, effective_filters, infer, timestamp)
return {"results": vector_store_result}
def _add_to_vector_store(self, messages, metadata, filters, infer, timestamp=None):
@@ -1737,19 +1738,20 @@ class AsyncMemory(MemoryBase):
else:
messages = parse_vision_messages(messages)
vector_store_task = asyncio.create_task(
self._add_to_vector_store(messages, processed_metadata, effective_filters, infer)
)
graph_task = asyncio.create_task(self._add_to_graph(messages, effective_filters))
vector_store_result, graph_result = await asyncio.gather(vector_store_task, graph_task)
if self.enable_graph:
vector_store_task = asyncio.create_task(
self._add_to_vector_store(messages, processed_metadata, effective_filters, infer)
)
graph_task = asyncio.create_task(self._add_to_graph(messages, effective_filters))
vector_store_result, graph_result = await asyncio.gather(vector_store_task, graph_task)
return {
"results": vector_store_result,
"relations": graph_result,
}
vector_store_result = await self._add_to_vector_store(messages, processed_metadata, effective_filters, infer)
return {"results": vector_store_result}
async def _add_to_vector_store(