notebooks: update code for llamaindex 0.10.x (#388)

* notebooks: update code for llamaindex 0.10.x

* fix

* fix

Author: nicoloboschi

examples/notebooks/llama-astra.ipynb

@@ -130,7 +130,7 @@
    },
    "outputs": [],
    "source": [
-    "from llama_index.llama_dataset import download_llama_dataset\n",
+    "from llama_index.core.llama_dataset import download_llama_dataset\n",
     "\n",
     "!mkdir -p 'data'\n",
     "\n",
@@ -158,12 +158,8 @@
    },
    "outputs": [],
    "source": [
-    "from llama_index.vector_stores import AstraDBVectorStore\n",
-    "from llama_index import (\n",
-    "    VectorStoreIndex,\n",
-    "    SimpleDirectoryReader,\n",
-    "    StorageContext,\n",
-    ")\n",
+    "from llama_index.vector_stores.astra_db import AstraDBVectorStore\n",
+    "from llama_index.core import VectorStoreIndex, SimpleDirectoryReader, StorageContext\n",
     "\n",
     "documents = SimpleDirectoryReader(\"./data/source_files\").load_data()\n",
     "print(f\"Total documents: {len(documents)}\")\n",
@@ -196,7 +192,7 @@
    },
    "outputs": [],
    "source": [
-    "from llama_index.vector_stores import AstraDBVectorStore\n",
+    "from llama_index.vector_stores.astra_db import AstraDBVectorStore\n",
     "import os\n",
     "\n",
     "astra_db_store = AstraDBVectorStore(\n",

examples/notebooks/llama-parse-astra.ipynb

@@ -38,8 +38,7 @@
     },
     "slideshow": {
      "slide_type": ""
-    },
-    "tags": []
+    }
    },
    "outputs": [
     {
@@ -230,7 +229,7 @@
    "outputs": [],
    "source": [
     "import os\n",
-    "from llama_index.vector_stores import AstraDBVectorStore\n",
+    "from llama_index.vector_stores.astra_db import AstraDBVectorStore\n",
     "\n",
     "astra_db_store = AstraDBVectorStore(\n",
     "    token=os.environ[\"ASTRA_DB_APPLICATION_TOKEN\"],\n",
@@ -246,8 +245,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_index.node_parser import SimpleNodeParser\n",
-    "from llama_index.llms import OpenAI\n",
+    "from llama_index.core.node_parser import SimpleNodeParser\n",
+    "from llama_index.llms.openai import OpenAI\n",
     "\n",
     "node_parser = SimpleNodeParser()\n",
     "\n",
@@ -260,12 +259,8 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from llama_index import (\n",
-    "    OpenAIEmbedding,\n",
-    "    VectorStoreIndex,\n",
-    "    StorageContext,\n",
-    "    ServiceContext,\n",
-    ")\n",
+    "from llama_index.embeddings.openai import OpenAIEmbedding\n",
+    "from llama_index.core import VectorStoreIndex, StorageContext, ServiceContext\n",
     "\n",
     "storage_context = StorageContext.from_defaults(vector_store=astra_db_store)\n",
     "\n",