Solved: Choosing a vector database - Page 2

atanu585 · 06-12-2024 05:29 PM

I am building a RAG architecture for an AI Agent and want to store my data in the form of vectors in a relational database. I am confused with which database should I opt for (BigQuery, CloudSQL or AlloyDB). I see that most of the tutorials in the Google gen-AI repo use BigQuery to connect Gen-AI models like Gemini to BigQuery.

I would want to know which database I should choose for seamless connection of the database with Vertex-AI and Gemini models and deploying it in a production environment.

jaia

Hello,
Thank You for contacting us!
You're right, BigQuery is a popular choice for storing vector embeddings in a RAG architecture using Vertex AI and Gemini models.
Since most tutorials use BigQuery, it's a safe starting point if data size and real-time needs aren't critical factors. However, you can explore Cloud SQL or AlloyDB if data management expertise, schema flexibility, or lower latency is a priority.

I hope I have answered your question.

View solution in original post