Fix import path and query engine call (run-llama#15796)

ssmanjunath · Sep 3, 2024 · 1735a2e · 1735a2e
1 parent 1b618fe
commit 1735a2e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/docs/docs/examples/property_graph/property_graph_kuzu.ipynb b/docs/docs/examples/property_graph/property_graph_kuzu.ipynb
@@ -6,7 +6,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "# %pip install llama-index llama-index-graph-stores-kuzu"
+    "# %pip install llama-index llama-index-embeddings-openai llama-index-graph-stores-kuzu"
    ]
   },
   {
@@ -121,7 +121,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from kuzu_property_graph import KuzuPropertyGraphStore\n",
+    "from llama_index.graph_stores.kuzu import KuzuPropertyGraphStore\n",
     "\n",
     "graph_store = KuzuPropertyGraphStore(db)"
    ]
@@ -238,7 +238,7 @@
     "# Switch to the generate LLM during retrieval\n",
     "Settings.llm = generate_llm\n",
     "\n",
-    "query_engine = index.as_query_engine(include_text=True)\n",
+    "query_engine = index.as_query_engine(include_text=False)\n",
     "\n",
     "response = query_engine.query(\"Tell me more about Interleaf and Viaweb\")\n",
     "\n",
@@ -375,7 +375,7 @@
     "# Switch to the generate LLM during retrieval\n",
     "Settings.llm = generate_llm\n",
     "\n",
-    "query_engine = index.as_query_engine(include_text=True)\n",
+    "query_engine = index.as_query_engine(include_text=False)\n",
     "\n",
     "response2 = query_engine.query(\"Tell me more about Interleaf and Viaweb\")\n",
     "print(str(response2))"
@@ -413,7 +413,7 @@
     "    property_graph_store=graph_store,\n",
     ")\n",
     "\n",
-    "query_engine = index.as_query_engine(include_text=True)\n",
+    "query_engine = index.as_query_engine(include_text=False)\n",
     "\n",
     "response3 = query_engine.query(\"When was Viaweb founded, and by whom?\")\n",
     "print(str(response3))"

diff --git a/llama-index-integrations/graph_stores/llama-index-graph-stores-kuzu/pyproject.toml b/llama-index-integrations/graph_stores/llama-index-graph-stores-kuzu/pyproject.toml
@@ -28,7 +28,7 @@ exclude = ["**/BUILD"]
 license = "MIT"
 name = "llama-index-graph-stores-kuzu"
 readme = "README.md"
-version = "0.3.0"
+version = "0.3.1"
 
 [tool.poetry.dependencies]
 python = ">=3.8.1,<4.0"