Skip to content

Commit

Permalink
Reuse project directory var in api notebook
Browse files Browse the repository at this point in the history
  • Loading branch information
natoverse committed Feb 4, 2025
1 parent 64da5b8 commit 2990cea
Showing 1 changed file with 6 additions and 5 deletions.
11 changes: 6 additions & 5 deletions docs/examples_notebooks/api_overview.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -49,7 +49,8 @@
"source": [
"import yaml\n",
"\n",
"settings = yaml.safe_load(open(\"<project_directory>/settings.yaml\")) # noqa: PTH123, SIM115"
"PROJECT_DIRECTORY = \"<project_directory>\"\n",
"settings = yaml.safe_load(open(f\"{PROJECT_DIRECTORY}/settings.yaml\")) # noqa: PTH123, SIM115"
]
},
{
Expand All @@ -75,7 +76,7 @@
"from graphrag.config.create_graphrag_config import create_graphrag_config\n",
"\n",
"graphrag_config = create_graphrag_config(\n",
" values=settings, root_dir=\"<project_directory>\"\n",
" values=settings, root_dir=PROJECT_DIRECTORY\n",
")"
]
},
Expand Down Expand Up @@ -126,10 +127,10 @@
"source": [
"import pandas as pd\n",
"\n",
"final_entities = pd.read_parquet(\"<project_directory>/output/entities.parquet\")\n",
"final_communities = pd.read_parquet(\"<project_directory>/output/communities.parquet\")\n",
"final_entities = pd.read_parquet(f\"{PROJECT_DIRECTORY}/output/entities.parquet\")\n",
"final_communities = pd.read_parquet(f\"{PROJECT_DIRECTORY}/output/communities.parquet\")\n",
"final_community_reports = pd.read_parquet(\n",
" \"<project_directory>/output/community_reports.parquet\"\n",
" f\"{PROJECT_DIRECTORY}/output/community_reports.parquet\"\n",
")\n",
"\n",
"response, context = await api.global_search(\n",
Expand Down

0 comments on commit 2990cea

Please sign in to comment.