Die Vektordatenbankkomponente von BabyAGI speichert die Datensätze und Ergebnisse der abgeschlossenen Aufgaben und stellt das Gedächtnis des Agenten dar. BabyAGI kann die Ergebnisse der ersten Aufgabe verwenden, um die zweite Aufgabe zu informieren, und iteriert diesen Prozess, während er in der Aufgabenliste nach unten geht.
Vektordatenbanken speichern Daten als mathematische Darstellungen, die als Einbetten bezeichnet werden. Datenpunkte, die im hochdimensionalen Vektorraum näher beieinander liegen, gelten als semantisch ähnlicher. BabyAGI verwendet die semantische Suche, um relevante Informationen in der Datenbank zu finden.
Die kanonische Implementierung verwendet Pinecone, aber alternative Vektorspeicher wie Metas Facebook AI Similarity Search (FAISS) und Chroma werden manchmal in Varianten oder Forks verwendet. FAISS und Chroma sind Open Source, Pinecone hingegen, wie viele OpenAI-Produkte, nicht.