fixes #29193 | handles case for no documents

langchain-ai · Jan 14, 2025 · 328808e · 328808e
1 parent c55af44
commit 328808e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/docs/docs/versions/migrating_memory/long_term_memory_agent.ipynb b/docs/docs/versions/migrating_memory/long_term_memory_agent.ipynb
@@ -52,7 +52,7 @@
    "metadata": {},
    "outputs": [
     {
-     "name": "stdin",
+     "name": "stdout",
      "output_type": "stream",
      "text": [
       "OPENAI_API_KEY:  ········\n",
@@ -182,7 +182,10 @@
     "    documents = recall_vector_store.similarity_search(\n",
     "        query, k=3, filter=_filter_function\n",
     "    )\n",
-    "    return [document.page_content for document in documents]"
+    "    if not documents:\n",
+    "        return []\n",
+    "    else:\n",
+    "        return [document.page_content for document in documents]"
    ]
   },
   {