⚡️ Speed up method AsyncMemory._should_use_agent_memory_extraction by 123%

[codeflash-ai]

📄 123% (1.23x) speedup for AsyncMemory._should_use_agent_memory_extraction in mem0/memory/main.py

⏱️ Runtime : 7.89 microseconds → 3.53 microseconds (best of 30 runs)

📝 Explanation and details

The optimization focuses on improving the performance of the _should_use_agent_memory_extraction method through early exit logic and avoiding unnecessary computation.

Key Optimization:

• Early Return Pattern: Instead of computing both conditions (has_agent_id and has_assistant_messages) regardless of the first condition's result, the optimized version immediately returns False if agent_id is missing, avoiding the expensive any() operation entirely.

• Efficient Loop Structure: When agent_id is present, the code uses a simple for loop that returns True immediately upon finding the first assistant message, rather than the any() function which creates a generator expression and potentially scans all messages.

Performance Impact:
The line profiler shows the optimization eliminates the expensive any() call (which took 66.4% of the original execution time at 9764ns). In the optimized version, when agent_id is None (common case), the function exits early after just two operations, reducing total runtime from 14.7µs to 6.8µs.

Minor Optimization:

• Tuple vs List: Changed the membership test from ["faiss", "qdrant"] to ("faiss", "qdrant") for slightly better performance in the in operator.

This optimization is particularly effective for scenarios where agent_id is frequently missing from metadata, as it completely avoids the message scanning loop. Even when agent_id is present, the loop-based approach is more efficient than any() for early termination cases.

✅ Correctness verification report:

Test	Status
⚙️ Existing Unit Tests	🔘 None Found
🌀 Generated Regression Tests	🔘 None Found
⏪ Replay Tests	✅ 3 Passed
🔎 Concolic Coverage Tests	🔘 None Found
📊 Tests Coverage	100.0%

⏪ Replay Tests and Runtime

Test File::Test Function	Original ⏱️	Optimized ⏱️	Speedup
test_pytest_testsconfigstest_prompts_py_testsvector_storestest_weaviate_py_testsllmstest_deepseek_py_test__replay_test_0.py::test_mem0_memory_main_AsyncMemory__should_use_agent_memory_extraction	3.00μs	1.25μs	139%✅
test_pytest_testsvector_storestest_opensearch_py_testsvector_storestest_upstash_vector_py_testsllmstest_l__replay_test_0.py::test_mem0_memory_main_AsyncMemory__should_use_agent_memory_extraction	4.89μs	2.28μs	115%✅

To edit these changes git checkout codeflash/optimize-AsyncMemory._should_use_agent_memory_extraction-mhk2kz7t and push.