Update src/oss/langchain/short-term-memory.mdx

Co-authored-by: Copilot <175728472+Copilot@users.noreply.github.com>

Author: TheovanKraay

src/oss/langchain/short-term-memory.mdx

@@ -136,12 +136,11 @@ with CosmosDBSaver(  # [!code highlight]
     )
 ```
 
+:::
+
 <Note>
     Azure Cosmos DB checkpointer supports both sync and async operations. If the database and container already exist, you can use default RBAC credentials (e.g., `az login`) instead of setting the endpoint and key.
 </Note>
-:::
-
-
 ## Customizing agent memory
 
 By default, agents use @[`AgentState`] to manage short term memory, specifically the conversation history via a `messages` key.