MongoDB Atlas Integration for Retrieval-Augmented Generation (RAG)

MongoDB + RAG + Splunk Agent Observability overview

MongoDB Atlas Vector Search lets you keep your knowledge base inside MongoDB while enjoying hybrid search and AI-ready workloads. In this guide you’ll:

Embed and store vectors in Atlas
Setup Splunk Agent Observability
Stream LangGraph traces to Splunk Agent Observability for end-to-end observability

Set up MongoDB

from langchain_mongodb import MongoDBAtlasVectorSearch
from langchain_openai import OpenAIEmbeddings
from pymongo import MongoClient

client          = MongoClient(MONGODB_URI)
DB_NAME         = "rag_demo"
COLLECTION_NAME = "blog_vectors"

collection = client[DB_NAME][COLLECTION_NAME]
embeddings = OpenAIEmbeddings()

vector_store = MongoDBAtlasVectorSearch(
    collection    = collection,
    embedding     = embeddings,
    index_name    = "rag-index",
    relevance_score_fn = "cosine"
)

Use Splunk Agent Observability for logging

Install python-dotenv if it is not already available:

pip install python-dotenv

Create a .env file for your deployment.

Splunk Observability Cloud
On-premises, standalone, or custom

.env

SPLUNK_AO_REALM="us1"
SPLUNK_AO_O11Y_TOKEN="your-o11y-token"

.env

SPLUNK_AO_API_KEY="your-splunk-ao-api-key"
SPLUNK_AO_CONSOLE_URL="https://your-splunk-ao-console-url"

Add the project and Agent Stream to the same .env file:

.env

SPLUNK_AO_PROJECT="your-splunk-ao-project-name"
SPLUNK_AO_AGENT_STREAM="mongo_rag_demo"

from dotenv import load_dotenv
from splunk_ao.handlers.langchain import SplunkAOCallback

load_dotenv()

splunk_ao_handler = SplunkAOCallback()

Once the callback is created we can add it to the agent.

query = "What types of agent memory does Lilian Weng describe?"
inputs = {"messages": [HumanMessage(content=query)]}

result = graph.invoke(
    inputs,
    config={"recursion_limit": 5, "callbacks": [splunk_ao_handler]}
)

Find the full notebook in Google Colab.

Overview

Agentic AI

Conversational AI

Retrieval-Augmented Generation

MongoDB Atlas Integration for Retrieval-Augmented Generation (RAG)

MongoDB + RAG + Splunk Agent Observability overview

Set up MongoDB

Use Splunk Agent Observability for logging

​MongoDB + RAG + Splunk Agent Observability overview

​Set up MongoDB

​Use Splunk Agent Observability for logging

MongoDB + RAG + Splunk Agent Observability overview

Set up MongoDB

Use Splunk Agent Observability for logging