From d457ad3cc272b8884a3a9e6cd829323748cc8ae5 Mon Sep 17 00:00:00 2001
From: William FH <13333726+hinthornw@users.noreply.github.com>
Date: Wed, 4 Dec 2024 14:50:23 -0800
Subject: [PATCH] Clean up code snippet (#2637)

---
 docs/docs/concepts/persistence.md              | 5 ++++-
 docs/docs/how-tos/memory/semantic-search.ipynb | 1 -
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/docs/docs/concepts/persistence.md b/docs/docs/concepts/persistence.md
index 8f4aa993a..0ec126316 100644
--- a/docs/docs/concepts/persistence.md
+++ b/docs/docs/concepts/persistence.md
@@ -276,9 +276,11 @@ The attributes it has are:
 Beyond simple retrieval, the store also supports semantic search, allowing you to find memories based on meaning rather than exact matches. To enable this, configure the store with an embedding model:
 
 ```python
+from langchain.embeddings import init_embeddings
+
 store = InMemoryStore(
     index={
-        "embed": "openai:text-embedding-3-small",  # Embedding provider
+        "embed": init_embeddings("openai:text-embedding-3-small"),  # Embedding provider
         "dims": 1536,                              # Embedding dimensions
         "fields": ["food_preference", "$"]              # Fields to embed
     }
@@ -289,6 +291,7 @@ Now when searching, you can use natural language queries to find relevant memori
 
 ```python
 # Find memories about food preferences
+# (This can be done after putting memories into the store)
 memories = store.search(
     namespace_for_memory,
     query="What does the user like to eat?",
diff --git a/docs/docs/how-tos/memory/semantic-search.ipynb b/docs/docs/how-tos/memory/semantic-search.ipynb
index 24905e625..658e4bb29 100644
--- a/docs/docs/how-tos/memory/semantic-search.ipynb
+++ b/docs/docs/how-tos/memory/semantic-search.ipynb
@@ -297,7 +297,6 @@
     }
    ],
    "source": [
-    "embeddings = init_embeddings(\"openai:text-embedding-3-small\")\n",
     "store = InMemoryStore(\n",
     "    index={\n",
     "        \"embed\": embeddings,\n",