Mem0 1.0.0 (#3545)

tests/memory/test_main.py

@@ -43,7 +43,7 @@ class TestAddToVectorStoreErrors:
     def test_empty_llm_response_fact_extraction(self, mocker, mock_memory, caplog):
         """Test empty response from LLM during fact extraction"""
         # Setup
-        mock_memory.llm.generate_response.return_value = ""
+        mock_memory.llm.generate_response.return_value = "invalid json"  # This will trigger a JSON decode error
         mock_capture_event = mocker.MagicMock()
         mocker.patch("mem0.memory.main.capture_event", mock_capture_event)
 
@@ -56,7 +56,8 @@ class TestAddToVectorStoreErrors:
         # Verify
         assert mock_memory.llm.generate_response.call_count == 1
         assert result == []  # Should return empty list when no memories processed
-        assert "Error in new_retrieved_facts" in caplog.text
+        # Check for error message in any of the log records
+        assert any("Error in new_retrieved_facts" in record.msg for record in caplog.records), "Expected error message not found in logs"
         assert mock_capture_event.call_count == 1
 
     def test_empty_llm_response_memory_actions(self, mock_memory, caplog):
@@ -96,7 +97,7 @@ class TestAsyncAddToVectorStoreErrors:
     async def test_async_empty_llm_response_fact_extraction(self, mock_async_memory, caplog, mocker):
         """Test empty response in AsyncMemory._add_to_vector_store"""
         mocker.patch("mem0.utils.factory.EmbedderFactory.create", return_value=MagicMock())
-        mock_async_memory.llm.generate_response.return_value = ""
+        mock_async_memory.llm.generate_response.return_value = "invalid json"  # This will trigger a JSON decode error
         mock_capture_event = mocker.MagicMock()
         mocker.patch("mem0.memory.main.capture_event", mock_capture_event)
 
@@ -106,7 +107,8 @@ class TestAsyncAddToVectorStoreErrors:
             )
         assert mock_async_memory.llm.generate_response.call_count == 1
         assert result == []
-        assert "Error in new_retrieved_facts" in caplog.text
+        # Check for error message in any of the log records
+        assert any("Error in new_retrieved_facts" in record.msg for record in caplog.records), "Expected error message not found in logs"
         assert mock_capture_event.call_count == 1
 
     @pytest.mark.asyncio