bug -fix : use of probability db instead of vector db