fix: Fixes to RAG Engine Intro notebook (#1398)

GoogleCloudPlatform · Nov 12, 2024 · 01da1f7 · 01da1f7
1 parent 1bf1f70
commit 01da1f7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gemini/rag-engine/intro_rag_engine.ipynb b/gemini/rag-engine/intro_rag_engine.ipynb
@@ -260,6 +260,7 @@
       },
       "outputs": [],
       "source": [
+        "from IPython.display import Markdown\n",
         "from vertexai.preview import rag\n",
         "from vertexai.preview.generative_models import GenerativeModel, Tool"
       ]
@@ -374,7 +375,7 @@
         ")\n",
         "\n",
         "response = rag.import_files(\n",
-        "    corpus_name=rag_corpus.corpus_name,\n",
+        "    corpus_name=rag_corpus.name,\n",
         "    paths=[INPUT_GCS_BUCKET],\n",
         "    chunk_size=1024,  # Optional\n",
         "    chunk_overlap=100,  # Optional\n",