diff --git a/docs/docs/examples/property_graph/property_graph_kuzu.ipynb b/docs/docs/examples/property_graph/property_graph_kuzu.ipynb index 56d79f71edcf9..47ecb197a8d6e 100644 --- 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 index 6b76d8234b071..d88f6daea1d8d 100644 --- 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"