kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-01-14 09:07:20 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

Deploying to gh-pages from @ microsoft/graphrag@e84df28e64 🚀

Browse Source
This commit is contained in:
AlonsoGuevara 2025-07-10 00:33:22 +00:00
parent 9ae919afbc
commit 7d183e4cf4
45 changed files with 1561 additions and 1614 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
404.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="/assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1540,7 +1540,7 @@
<script id="__config" type="application/json">{"base": "/", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "/assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="/assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="/assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="/scripts/create_cookie_banner.js"></script>

4
assets/javascripts/bundle.13a4f30d.min.js → assets/javascripts/bundle.56ea9cef.min.js vendored
View File

File diff suppressed because one or more lines are too long

4
assets/javascripts/bundle.13a4f30d.min.js.map → assets/javascripts/bundle.56ea9cef.min.js.map
View File

File diff suppressed because one or more lines are too long

4
blog_posts/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1663,7 +1663,7 @@
<script id="__config" type="application/json">{"base": "..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../scripts/create_cookie_banner.js"></script>

141
cli/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1550,17 +1550,15 @@
<a id="__codelineno-3-7" name="__codelineno-3-7" href="#__codelineno-3-7"></a> logging
<a id="__codelineno-3-8" name="__codelineno-3-8" href="#__codelineno-3-8"></a> --memprofile Run the indexing pipeline with memory
<a id="__codelineno-3-9" name="__codelineno-3-9" href="#__codelineno-3-9"></a> profiling
<a id="__codelineno-3-10" name="__codelineno-3-10" href="#__codelineno-3-10"></a> --logger [rich|print|none] The progress logger to use. \[default:
<a id="__codelineno-3-11" name="__codelineno-3-11" href="#__codelineno-3-11"></a> rich]
<a id="__codelineno-3-12" name="__codelineno-3-12" href="#__codelineno-3-12"></a> --dry-run Run the indexing pipeline without executing
<a id="__codelineno-3-13" name="__codelineno-3-13" href="#__codelineno-3-13"></a> any steps to inspect and validate the
<a id="__codelineno-3-14" name="__codelineno-3-14" href="#__codelineno-3-14"></a> configuration.
<a id="__codelineno-3-15" name="__codelineno-3-15" href="#__codelineno-3-15"></a> --cache / --no-cache Use LLM cache. \[default: cache]
<a id="__codelineno-3-16" name="__codelineno-3-16" href="#__codelineno-3-16"></a> --skip-validation Skip any preflight validation. Useful when
<a id="__codelineno-3-17" name="__codelineno-3-17" href="#__codelineno-3-17"></a> running no LLM steps.
<a id="__codelineno-3-18" name="__codelineno-3-18" href="#__codelineno-3-18"></a> -o, --output PATH Indexing pipeline output directory.
<a id="__codelineno-3-19" name="__codelineno-3-19" href="#__codelineno-3-19"></a> Overrides output.base_dir in the
<a id="__codelineno-3-20" name="__codelineno-3-20" href="#__codelineno-3-20"></a> configuration file.
<a id="__codelineno-3-10" name="__codelineno-3-10" href="#__codelineno-3-10"></a> --dry-run Run the indexing pipeline without executing
<a id="__codelineno-3-11" name="__codelineno-3-11" href="#__codelineno-3-11"></a> any steps to inspect and validate the
<a id="__codelineno-3-12" name="__codelineno-3-12" href="#__codelineno-3-12"></a> configuration.
<a id="__codelineno-3-13" name="__codelineno-3-13" href="#__codelineno-3-13"></a> --cache / --no-cache Use LLM cache. \[default: cache]
<a id="__codelineno-3-14" name="__codelineno-3-14" href="#__codelineno-3-14"></a> --skip-validation Skip any preflight validation. Useful when
<a id="__codelineno-3-15" name="__codelineno-3-15" href="#__codelineno-3-15"></a> running no LLM steps.
<a id="__codelineno-3-16" name="__codelineno-3-16" href="#__codelineno-3-16"></a> -o, --output PATH Indexing pipeline output directory.
<a id="__codelineno-3-17" name="__codelineno-3-17" href="#__codelineno-3-17"></a> Overrides output.base_dir in the
<a id="__codelineno-3-18" name="__codelineno-3-18" href="#__codelineno-3-18"></a> configuration file.
</code></pre></div>
<h2 id="init">init</h2>
<p>Generate a default configuration file.</p>
@ -1581,43 +1579,41 @@
<a id="__codelineno-7-2" name="__codelineno-7-2" href="#__codelineno-7-2"></a> -c, --config PATH The configuration to use.
<a id="__codelineno-7-3" name="__codelineno-7-3" href="#__codelineno-7-3"></a> -v, --verbose Run the prompt tuning pipeline with verbose
<a id="__codelineno-7-4" name="__codelineno-7-4" href="#__codelineno-7-4"></a> logging.
<a id="__codelineno-7-5" name="__codelineno-7-5" href="#__codelineno-7-5"></a> --logger [rich|print|none] The progress logger to use. \[default:
<a id="__codelineno-7-6" name="__codelineno-7-6" href="#__codelineno-7-6"></a> rich]
<a id="__codelineno-7-7" name="__codelineno-7-7" href="#__codelineno-7-7"></a> --domain TEXT The domain your input data is related to.
<a id="__codelineno-7-8" name="__codelineno-7-8" href="#__codelineno-7-8"></a> For example &#39;space science&#39;, &#39;microbiology&#39;,
<a id="__codelineno-7-9" name="__codelineno-7-9" href="#__codelineno-7-9"></a> &#39;environmental news&#39;. If not defined, a
<a id="__codelineno-7-10" name="__codelineno-7-10" href="#__codelineno-7-10"></a> domain will be inferred from the input data.
<a id="__codelineno-7-11" name="__codelineno-7-11" href="#__codelineno-7-11"></a> --selection-method [all|random|top|auto]
<a id="__codelineno-7-12" name="__codelineno-7-12" href="#__codelineno-7-12"></a> The text chunk selection method. \[default:
<a id="__codelineno-7-13" name="__codelineno-7-13" href="#__codelineno-7-13"></a> random]
<a id="__codelineno-7-14" name="__codelineno-7-14" href="#__codelineno-7-14"></a> --n-subset-max INTEGER The number of text chunks to embed when
<a id="__codelineno-7-15" name="__codelineno-7-15" href="#__codelineno-7-15"></a> --selection-method=auto. \[default: 300]
<a id="__codelineno-7-16" name="__codelineno-7-16" href="#__codelineno-7-16"></a> --k INTEGER The maximum number of documents to select
<a id="__codelineno-7-17" name="__codelineno-7-17" href="#__codelineno-7-17"></a> from each centroid when --selection-
<a id="__codelineno-7-18" name="__codelineno-7-18" href="#__codelineno-7-18"></a> method=auto. \[default: 15]
<a id="__codelineno-7-19" name="__codelineno-7-19" href="#__codelineno-7-19"></a> --limit INTEGER The number of documents to load when
<a id="__codelineno-7-20" name="__codelineno-7-20" href="#__codelineno-7-20"></a> --selection-method={random,top}. \[default:
<a id="__codelineno-7-21" name="__codelineno-7-21" href="#__codelineno-7-21"></a> 15]
<a id="__codelineno-7-22" name="__codelineno-7-22" href="#__codelineno-7-22"></a> --max-tokens INTEGER The max token count for prompt generation.
<a id="__codelineno-7-23" name="__codelineno-7-23" href="#__codelineno-7-23"></a> \[default: 2000]
<a id="__codelineno-7-24" name="__codelineno-7-24" href="#__codelineno-7-24"></a> --min-examples-required INTEGER
<a id="__codelineno-7-25" name="__codelineno-7-25" href="#__codelineno-7-25"></a> The minimum number of examples to
<a id="__codelineno-7-26" name="__codelineno-7-26" href="#__codelineno-7-26"></a> generate/include in the entity extraction
<a id="__codelineno-7-27" name="__codelineno-7-27" href="#__codelineno-7-27"></a> prompt. \[default: 2]
<a id="__codelineno-7-28" name="__codelineno-7-28" href="#__codelineno-7-28"></a> --chunk-size INTEGER The size of each example text chunk.
<a id="__codelineno-7-29" name="__codelineno-7-29" href="#__codelineno-7-29"></a> Overrides chunks.size in the configuration
<a id="__codelineno-7-30" name="__codelineno-7-30" href="#__codelineno-7-30"></a> file. \[default: 1200]
<a id="__codelineno-7-31" name="__codelineno-7-31" href="#__codelineno-7-31"></a> --overlap INTEGER The overlap size for chunking documents.
<a id="__codelineno-7-32" name="__codelineno-7-32" href="#__codelineno-7-32"></a> Overrides chunks.overlap in the
<a id="__codelineno-7-33" name="__codelineno-7-33" href="#__codelineno-7-33"></a> configuration file. \[default: 100]
<a id="__codelineno-7-34" name="__codelineno-7-34" href="#__codelineno-7-34"></a> --language TEXT The primary language used for inputs and
<a id="__codelineno-7-35" name="__codelineno-7-35" href="#__codelineno-7-35"></a> outputs in graphrag prompts.
<a id="__codelineno-7-36" name="__codelineno-7-36" href="#__codelineno-7-36"></a> --discover-entity-types / --no-discover-entity-types
<a id="__codelineno-7-37" name="__codelineno-7-37" href="#__codelineno-7-37"></a> Discover and extract unspecified entity
<a id="__codelineno-7-38" name="__codelineno-7-38" href="#__codelineno-7-38"></a> types. \[default: discover-entity-types]
<a id="__codelineno-7-39" name="__codelineno-7-39" href="#__codelineno-7-39"></a> -o, --output PATH The directory to save prompts to, relative
<a id="__codelineno-7-40" name="__codelineno-7-40" href="#__codelineno-7-40"></a> to the project root directory. \[default:
<a id="__codelineno-7-41" name="__codelineno-7-41" href="#__codelineno-7-41"></a> prompts]
<a id="__codelineno-7-5" name="__codelineno-7-5" href="#__codelineno-7-5"></a> --domain TEXT The domain your input data is related to.
<a id="__codelineno-7-6" name="__codelineno-7-6" href="#__codelineno-7-6"></a> For example &#39;space science&#39;, &#39;microbiology&#39;,
<a id="__codelineno-7-7" name="__codelineno-7-7" href="#__codelineno-7-7"></a> &#39;environmental news&#39;. If not defined, a
<a id="__codelineno-7-8" name="__codelineno-7-8" href="#__codelineno-7-8"></a> domain will be inferred from the input data.
<a id="__codelineno-7-9" name="__codelineno-7-9" href="#__codelineno-7-9"></a> --selection-method [all|random|top|auto]
<a id="__codelineno-7-10" name="__codelineno-7-10" href="#__codelineno-7-10"></a> The text chunk selection method. \[default:
<a id="__codelineno-7-11" name="__codelineno-7-11" href="#__codelineno-7-11"></a> random]
<a id="__codelineno-7-12" name="__codelineno-7-12" href="#__codelineno-7-12"></a> --n-subset-max INTEGER The number of text chunks to embed when
<a id="__codelineno-7-13" name="__codelineno-7-13" href="#__codelineno-7-13"></a> --selection-method=auto. \[default: 300]
<a id="__codelineno-7-14" name="__codelineno-7-14" href="#__codelineno-7-14"></a> --k INTEGER The maximum number of documents to select
<a id="__codelineno-7-15" name="__codelineno-7-15" href="#__codelineno-7-15"></a> from each centroid when --selection-
<a id="__codelineno-7-16" name="__codelineno-7-16" href="#__codelineno-7-16"></a> method=auto. \[default: 15]
<a id="__codelineno-7-17" name="__codelineno-7-17" href="#__codelineno-7-17"></a> --limit INTEGER The number of documents to load when
<a id="__codelineno-7-18" name="__codelineno-7-18" href="#__codelineno-7-18"></a> --selection-method={random,top}. \[default:
<a id="__codelineno-7-19" name="__codelineno-7-19" href="#__codelineno-7-19"></a> 15]
<a id="__codelineno-7-20" name="__codelineno-7-20" href="#__codelineno-7-20"></a> --max-tokens INTEGER The max token count for prompt generation.
<a id="__codelineno-7-21" name="__codelineno-7-21" href="#__codelineno-7-21"></a> \[default: 2000]
<a id="__codelineno-7-22" name="__codelineno-7-22" href="#__codelineno-7-22"></a> --min-examples-required INTEGER
<a id="__codelineno-7-23" name="__codelineno-7-23" href="#__codelineno-7-23"></a> The minimum number of examples to
<a id="__codelineno-7-24" name="__codelineno-7-24" href="#__codelineno-7-24"></a> generate/include in the entity extraction
<a id="__codelineno-7-25" name="__codelineno-7-25" href="#__codelineno-7-25"></a> prompt. \[default: 2]
<a id="__codelineno-7-26" name="__codelineno-7-26" href="#__codelineno-7-26"></a> --chunk-size INTEGER The size of each example text chunk.
<a id="__codelineno-7-27" name="__codelineno-7-27" href="#__codelineno-7-27"></a> Overrides chunks.size in the configuration
<a id="__codelineno-7-28" name="__codelineno-7-28" href="#__codelineno-7-28"></a> file. \[default: 1200]
<a id="__codelineno-7-29" name="__codelineno-7-29" href="#__codelineno-7-29"></a> --overlap INTEGER The overlap size for chunking documents.
<a id="__codelineno-7-30" name="__codelineno-7-30" href="#__codelineno-7-30"></a> Overrides chunks.overlap in the
<a id="__codelineno-7-31" name="__codelineno-7-31" href="#__codelineno-7-31"></a> configuration file. \[default: 100]
<a id="__codelineno-7-32" name="__codelineno-7-32" href="#__codelineno-7-32"></a> --language TEXT The primary language used for inputs and
<a id="__codelineno-7-33" name="__codelineno-7-33" href="#__codelineno-7-33"></a> outputs in graphrag prompts.
<a id="__codelineno-7-34" name="__codelineno-7-34" href="#__codelineno-7-34"></a> --discover-entity-types / --no-discover-entity-types
<a id="__codelineno-7-35" name="__codelineno-7-35" href="#__codelineno-7-35"></a> Discover and extract unspecified entity
<a id="__codelineno-7-36" name="__codelineno-7-36" href="#__codelineno-7-36"></a> types. \[default: discover-entity-types]
<a id="__codelineno-7-37" name="__codelineno-7-37" href="#__codelineno-7-37"></a> -o, --output PATH The directory to save prompts to, relative
<a id="__codelineno-7-38" name="__codelineno-7-38" href="#__codelineno-7-38"></a> to the project root directory. \[default:
<a id="__codelineno-7-39" name="__codelineno-7-39" href="#__codelineno-7-39"></a> prompts]
</code></pre></div>
<h2 id="query">query</h2>
<p>Query a knowledge graph index.</p>
@ -1629,21 +1625,22 @@
<a id="__codelineno-9-2" name="__codelineno-9-2" href="#__codelineno-9-2"></a> The query algorithm to use. \[required]
<a id="__codelineno-9-3" name="__codelineno-9-3" href="#__codelineno-9-3"></a> -q, --query TEXT The query to execute. \[required]
<a id="__codelineno-9-4" name="__codelineno-9-4" href="#__codelineno-9-4"></a> -c, --config PATH The configuration to use.
<a id="__codelineno-9-5" name="__codelineno-9-5" href="#__codelineno-9-5"></a> -d, --data PATH Index output directory (contains the parquet
<a id="__codelineno-9-6" name="__codelineno-9-6" href="#__codelineno-9-6"></a> files).
<a id="__codelineno-9-7" name="__codelineno-9-7" href="#__codelineno-9-7"></a> -r, --root PATH The project root directory. \[default: .]
<a id="__codelineno-9-8" name="__codelineno-9-8" href="#__codelineno-9-8"></a> --community-level INTEGER Leiden hierarchy level from which to load
<a id="__codelineno-9-9" name="__codelineno-9-9" href="#__codelineno-9-9"></a> community reports. Higher values represent
<a id="__codelineno-9-10" name="__codelineno-9-10" href="#__codelineno-9-10"></a> smaller communities. \[default: 2]
<a id="__codelineno-9-11" name="__codelineno-9-11" href="#__codelineno-9-11"></a> --dynamic-community-selection / --no-dynamic-selection
<a id="__codelineno-9-12" name="__codelineno-9-12" href="#__codelineno-9-12"></a> Use global search with dynamic community
<a id="__codelineno-9-13" name="__codelineno-9-13" href="#__codelineno-9-13"></a> selection. \[default: no-dynamic-selection]
<a id="__codelineno-9-14" name="__codelineno-9-14" href="#__codelineno-9-14"></a> --response-type TEXT Free-form description of the desired
<a id="__codelineno-9-15" name="__codelineno-9-15" href="#__codelineno-9-15"></a> response format (e.g. &#39;Single Sentence&#39;,
<a id="__codelineno-9-16" name="__codelineno-9-16" href="#__codelineno-9-16"></a> &#39;List of 3-7 Points&#39;, etc.). \[default:
<a id="__codelineno-9-17" name="__codelineno-9-17" href="#__codelineno-9-17"></a> Multiple Paragraphs]
<a id="__codelineno-9-18" name="__codelineno-9-18" href="#__codelineno-9-18"></a> --streaming / --no-streaming Print the response in a streaming manner.
<a id="__codelineno-9-19" name="__codelineno-9-19" href="#__codelineno-9-19"></a> \[default: no-streaming]
<a id="__codelineno-9-5" name="__codelineno-9-5" href="#__codelineno-9-5"></a> -v, --verbose Run the query with verbose logging.
<a id="__codelineno-9-6" name="__codelineno-9-6" href="#__codelineno-9-6"></a> -d, --data PATH Index output directory (contains the parquet
<a id="__codelineno-9-7" name="__codelineno-9-7" href="#__codelineno-9-7"></a> files).
<a id="__codelineno-9-8" name="__codelineno-9-8" href="#__codelineno-9-8"></a> -r, --root PATH The project root directory. \[default: .]
<a id="__codelineno-9-9" name="__codelineno-9-9" href="#__codelineno-9-9"></a> --community-level INTEGER Leiden hierarchy level from which to load
<a id="__codelineno-9-10" name="__codelineno-9-10" href="#__codelineno-9-10"></a> community reports. Higher values represent
<a id="__codelineno-9-11" name="__codelineno-9-11" href="#__codelineno-9-11"></a> smaller communities. \[default: 2]
<a id="__codelineno-9-12" name="__codelineno-9-12" href="#__codelineno-9-12"></a> --dynamic-community-selection / --no-dynamic-selection
<a id="__codelineno-9-13" name="__codelineno-9-13" href="#__codelineno-9-13"></a> Use global search with dynamic community
<a id="__codelineno-9-14" name="__codelineno-9-14" href="#__codelineno-9-14"></a> selection. \[default: no-dynamic-selection]
<a id="__codelineno-9-15" name="__codelineno-9-15" href="#__codelineno-9-15"></a> --response-type TEXT Free-form description of the desired
<a id="__codelineno-9-16" name="__codelineno-9-16" href="#__codelineno-9-16"></a> response format (e.g. &#39;Single Sentence&#39;,
<a id="__codelineno-9-17" name="__codelineno-9-17" href="#__codelineno-9-17"></a> &#39;List of 3-7 Points&#39;, etc.). \[default:
<a id="__codelineno-9-18" name="__codelineno-9-18" href="#__codelineno-9-18"></a> Multiple Paragraphs]
<a id="__codelineno-9-19" name="__codelineno-9-19" href="#__codelineno-9-19"></a> --streaming / --no-streaming Print the response in a streaming manner.
<a id="__codelineno-9-20" name="__codelineno-9-20" href="#__codelineno-9-20"></a> \[default: no-streaming]
</code></pre></div>
<h2 id="update">update</h2>
<p>Update an existing knowledge graph index.</p>
@ -1661,14 +1658,12 @@
<a id="__codelineno-11-7" name="__codelineno-11-7" href="#__codelineno-11-7"></a> logging.
<a id="__codelineno-11-8" name="__codelineno-11-8" href="#__codelineno-11-8"></a> --memprofile Run the indexing pipeline with memory
<a id="__codelineno-11-9" name="__codelineno-11-9" href="#__codelineno-11-9"></a> profiling.
<a id="__codelineno-11-10" name="__codelineno-11-10" href="#__codelineno-11-10"></a> --logger [rich|print|none] The progress logger to use. \[default:
<a id="__codelineno-11-11" name="__codelineno-11-11" href="#__codelineno-11-11"></a> rich]
<a id="__codelineno-11-12" name="__codelineno-11-12" href="#__codelineno-11-12"></a> --cache / --no-cache Use LLM cache. \[default: cache]
<a id="__codelineno-11-13" name="__codelineno-11-13" href="#__codelineno-11-13"></a> --skip-validation Skip any preflight validation. Useful when
<a id="__codelineno-11-14" name="__codelineno-11-14" href="#__codelineno-11-14"></a> running no LLM steps.
<a id="__codelineno-11-15" name="__codelineno-11-15" href="#__codelineno-11-15"></a> -o, --output PATH Indexing pipeline output directory.
<a id="__codelineno-11-16" name="__codelineno-11-16" href="#__codelineno-11-16"></a> Overrides output.base_dir in the
<a id="__codelineno-11-17" name="__codelineno-11-17" href="#__codelineno-11-17"></a> configuration file.
<a id="__codelineno-11-10" name="__codelineno-11-10" href="#__codelineno-11-10"></a> --cache / --no-cache Use LLM cache. \[default: cache]
<a id="__codelineno-11-11" name="__codelineno-11-11" href="#__codelineno-11-11"></a> --skip-validation Skip any preflight validation. Useful when
<a id="__codelineno-11-12" name="__codelineno-11-12" href="#__codelineno-11-12"></a> running no LLM steps.
<a id="__codelineno-11-13" name="__codelineno-11-13" href="#__codelineno-11-13"></a> -o, --output PATH Indexing pipeline output directory.
<a id="__codelineno-11-14" name="__codelineno-11-14" href="#__codelineno-11-14"></a> Overrides output.base_dir in the
<a id="__codelineno-11-15" name="__codelineno-11-15" href="#__codelineno-11-15"></a> configuration file.
</code></pre></div>
@ -1766,7 +1761,7 @@
<script id="__config" type="application/json">{"base": "..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../scripts/create_cookie_banner.js"></script>

8
config/env_vars/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2445,7 +2445,7 @@
</tbody>
</table>
<h3 id="reporting">Reporting</h3>
<p>This section controls the reporting mechanism used by the pipeline, for common events and error messages. The default is to write reports to a file in the output directory. However, you can also choose to write reports to the console or to an Azure Blob Storage container.</p>
<p>This section controls the reporting mechanism used by the pipeline, for common events and error messages. The default is to write reports to a file in the output directory. However, you can also choose to write reports to an Azure Blob Storage container.</p>
<table>
<thead>
<tr>
@ -2459,7 +2459,7 @@
<tbody>
<tr>
<td><code>GRAPHRAG_REPORTING_TYPE</code></td>
<td>The type of reporter to use. Options are <code>file</code>, <code>console</code>, or <code>blob</code></td>
<td>The type of reporter to use. Options are <code>file</code> or <code>blob</code></td>
<td><code>str</code></td>
<td>optional</td>
<td><code>file</code></td>
@ -2699,7 +2699,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
config/init/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1776,7 +1776,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
config/models/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1845,7 +1845,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
config/overview/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1691,7 +1691,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

8
config/yaml/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1654,10 +1654,10 @@
<li><code>cosmosdb_account_blob_url</code> <strong>str</strong> - (cosmosdb only) The CosmosDB account blob URL to use.</li>
</ul>
<h3 id="reporting">reporting</h3>
<p>This section controls the reporting mechanism used by the pipeline, for common events and error messages. The default is to write reports to a file in the output directory. However, you can also choose to write reports to the console or to an Azure Blob Storage container.</p>
<p>This section controls the reporting mechanism used by the pipeline, for common events and error messages. The default is to write reports to a file in the output directory. However, you can also choose to write reports to an Azure Blob Storage container.</p>
<h4 id="fields_6">Fields</h4>
<ul>
<li><code>type</code> <strong>file|console|blob</strong> - The reporting type to use. Default=<code>file</code></li>
<li><code>type</code> <strong>file|blob</strong> - The reporting type to use. Default=<code>file</code></li>
<li><code>base_dir</code> <strong>str</strong> - The base directory to write reports to, relative to the root.</li>
<li><code>connection_string</code> <strong>str</strong> - (blob only) The Azure Storage connection string.</li>
<li><code>container_name</code> <strong>str</strong> - (blob only) The Azure Storage container name.</li>
@ -1961,7 +1961,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
data/operation_dulce/ABOUT/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1628,7 +1628,7 @@
<script id="__config" type="application/json">{"base": "../../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../../scripts/create_cookie_banner.js"></script>

4
data/operation_dulce/Operation Dulce v2 1 1/index.html
View File

@ -14,7 +14,7 @@
<link rel="icon" href="../../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2447,7 +2447,7 @@
<script id="__config" type="application/json">{"base": "../../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../../scripts/create_cookie_banner.js"></script>

4
developing/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1697,7 +1697,7 @@ to reduce concurrency. Please refer to the <a href="../config/overview/">Configu
<script id="__config" type="application/json">{"base": "..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../scripts/create_cookie_banner.js"></script>

203
examples_notebooks/api_overview/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2051,8 +2051,8 @@ span.linenos.special { color: #000000; background-color: #ffffc0; padding-left:
</div>
</div>
</div>
</div><div class="jp-Cell jp-CodeCell jp-Notebook-cell">
<div class="jp-Cell jp-CodeCell jp-Notebook-cell">
</div><div class="jp-Cell jp-CodeCell jp-Notebook-cell jp-mod-noOutputs">
<div class="jp-Cell jp-CodeCell jp-Notebook-cell jp-mod-noOutputs">
<div class="jp-Cell-inputWrapper" tabindex="0">
<div class="jp-Collapser jp-InputCollapser jp-Cell-inputCollapser">
</div>
@ -2091,19 +2091,6 @@ from graphrag.index.typing.pipeline_run_result import PipelineRunResult</div>
</div>
</div>
</div>
<div class="jp-Cell-outputWrapper">
<div class="jp-Collapser jp-OutputCollapser jp-Cell-outputCollapser">
</div>
<div class="jp-OutputArea jp-Cell-outputArea">
<div class="jp-OutputArea-child">
<div class="jp-OutputPrompt jp-OutputArea-prompt"></div>
<div class="jp-RenderedText jp-OutputArea-output" data-mime-type="text/plain" tabindex="0">
<pre>
</pre>
</div>
</div>
</div>
</div>
</div>
</div><div class="jp-Cell jp-CodeCell jp-Notebook-cell jp-mod-noOutputs">
<div class="jp-Cell jp-CodeCell jp-Notebook-cell jp-mod-noOutputs">
@ -2191,34 +2178,33 @@ from graphrag.index.typing.pipeline_run_result import PipelineRunResult</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[4], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> graphrag_config <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">load_config</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">Path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[4]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> graphrag_config = <span class="ansi-yellow-bg">load_config</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">Path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg">)</span>
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:183</span>, in <span class="ansi-cyan-fg">load_config</span><span class="ansi-blue-fg">(root_dir, config_filepath, cli_overrides)</span>
<span class="ansi-green-intense-fg ansi-bold"> 151</span> <span style="color: rgb(175,0,0)">"""Load configuration from a file.</span>
<span class="ansi-green-intense-fg ansi-bold"> 152</span>
<span class="ansi-green-intense-fg ansi-bold"> 153</span> <span style="color: rgb(175,0,0)">Parameters</span>
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 180</span> <span style="color: rgb(175,0,0)"> If there are pydantic validation errors when instantiating the config.</span>
<span class="ansi-green-intense-fg ansi-bold"> 181</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 182</span> root <span style="color: rgb(98,98,98)">=</span> root_dir<span style="color: rgb(98,98,98)">.</span>resolve()
<span class="ansi-green-fg">--&gt; 183</span> config_path <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_config_path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">config_filepath</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 184</span> _load_dotenv(config_path)
<span class="ansi-green-intense-fg ansi-bold"> 185</span> config_extension <span style="color: rgb(98,98,98)">=</span> config_path<span style="color: rgb(98,98,98)">.</span>suffix
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:183</span>, in <span class="ansi-cyan-fg">load_config</span><span class="ansi-blue-fg">(root_dir, config_filepath, cli_overrides)</span>
<span class="ansi-green-fg"> 151</span> <span class="ansi-yellow-fg">"""Load configuration from a file.</span>
<span class="ansi-green-fg"> 152</span>
<span class="ansi-green-fg"> 153</span> <span class="ansi-yellow-fg">Parameters</span>
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 180</span> <span class="ansi-yellow-fg"> If there are pydantic validation errors when instantiating the config.</span>
<span class="ansi-green-fg"> 181</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 182</span> root = root_dir.resolve()
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">183</span> config_path = <span class="ansi-yellow-bg">_get_config_path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">config_filepath</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 184</span> _load_dotenv(config_path)
<span class="ansi-green-fg"> 185</span> config_extension = config_path.suffix
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:106</span>, in <span class="ansi-cyan-fg">_get_config_path</span><span class="ansi-blue-fg">(root_dir, config_filepath)</span>
<span class="ansi-green-intense-fg ansi-bold"> 104</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-intense-fg ansi-bold"> 105</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">--&gt; 106</span> config_path <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_search_for_config_in_root_dir</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root_dir</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 108</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> config_path:
<span class="ansi-green-intense-fg ansi-bold"> 109</span> msg <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Config file not found in root directory: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root_dir<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:106</span>, in <span class="ansi-cyan-fg">_get_config_path</span><span class="ansi-blue-fg">(root_dir, config_filepath)</span>
<span class="ansi-green-fg"> 104</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-fg"> 105</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">106</span> config_path = <span class="ansi-yellow-bg">_search_for_config_in_root_dir</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root_dir</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 108</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> config_path:
<span class="ansi-green-fg"> 109</span> msg = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Config file not found in root directory: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root_dir<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">"</span>
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:40</span>, in <span class="ansi-cyan-fg">_search_for_config_in_root_dir</span><span class="ansi-blue-fg">(root)</span>
<span class="ansi-green-intense-fg ansi-bold"> 38</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> root<span style="color: rgb(98,98,98)">.</span>is_dir():
<span class="ansi-green-intense-fg ansi-bold"> 39</span> msg <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Invalid config path: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)"> is not a directory</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-fg">---&gt; 40</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-intense-fg ansi-bold"> 42</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> file <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> _default_config_files:
<span class="ansi-green-intense-fg ansi-bold"> 43</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (root <span style="color: rgb(98,98,98)">/</span> file)<span style="color: rgb(98,98,98)">.</span>is_file():
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:40</span>, in <span class="ansi-cyan-fg">_search_for_config_in_root_dir</span><span class="ansi-blue-fg">(root)</span>
<span class="ansi-green-fg"> 38</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> root.is_dir():
<span class="ansi-green-fg"> 39</span> msg = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Invalid config path: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg"> is not a directory</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">40</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-fg"> 42</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> file <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> _default_config_files:
<span class="ansi-green-fg"> 43</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (root / file).is_file():
<span class="ansi-red-fg">FileNotFoundError</span>: Invalid config path: /home/runner/work/graphrag/graphrag/docs/examples_notebooks/&lt;your project directory&gt; is not a directory</pre>
</div>
@ -2294,10 +2280,10 @@ for workflow_result in index_result:
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[5], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> index_result: <span style="color: rgb(0,135,0)">list</span>[PipelineRunResult] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> api<span style="color: rgb(98,98,98)">.</span>build_index(config<span style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">graphrag_config</span>)
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span style="color: rgb(95,135,135)"># index_result is a list of workflows that make up the indexing pipeline that was run</span>
<span class="ansi-green-intense-fg ansi-bold"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> workflow_result <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> index_result:
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[5]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> index_result: <span style="color: rgb(0,135,0)">list</span>[PipelineRunResult] = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> api.build_index(config=<span class="ansi-yellow-bg">graphrag_config</span>)
<span class="ansi-green-fg"> 3</span> <span style="color: rgb(95,135,135)"># index_result is a list of workflows that make up the indexing pipeline that was run</span>
<span class="ansi-green-fg"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> workflow_result <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> index_result:
<span class="ansi-red-fg">NameError</span>: name 'graphrag_config' is not defined</pre>
</div>
@ -2382,70 +2368,67 @@ response, context = await api.global_search(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[6], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/output/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>PROJECT_DIRECTORY<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/output/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-intense-fg ansi-bold"> 3</span> community_reports <span style="color: rgb(98,98,98)">=</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(
<span class="ansi-green-intense-fg ansi-bold"> 4</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>PROJECT_DIRECTORY<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/output/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 5</span> )
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[6]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = <span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/output/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 2</span> communities = pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>PROJECT_DIRECTORY<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/output/communities.parquet</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg"> 3</span> community_reports = pd.read_parquet(
<span class="ansi-green-fg"> 4</span> <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>PROJECT_DIRECTORY<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/output/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 5</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:667</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 664</span> use_nullable_dtypes <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-intense-fg ansi-bold"> 665</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; 667</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 668</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 669</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 676</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:669</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-fg"> 666</span> use_nullable_dtypes = <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-fg"> 667</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">669</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 676</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 677</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 678</span> <span class="ansi-yellow-bg">)</span>
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:267</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">array</span><span style="color: rgb(175,0,0)">"</span>:
<span class="ansi-green-intense-fg ansi-bold"> 265</span> to_pandas_kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">split_blocks</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">True</span> <span style="color: rgb(95,135,135)"># type: ignore[assignment]</span>
<span class="ansi-green-fg">--&gt; 267</span> path_or_handle, handles, filesystem <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 268</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 269</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 270</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 271</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">rb</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 272</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 273</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 274</span> pa_table <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>api<span style="color: rgb(98,98,98)">.</span>parquet<span style="color: rgb(98,98,98)">.</span>read_table(
<span class="ansi-green-intense-fg ansi-bold"> 275</span> path_or_handle,
<span class="ansi-green-intense-fg ansi-bold"> 276</span> columns<span style="color: rgb(98,98,98)">=</span>columns,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 279</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs,
<span class="ansi-green-intense-fg ansi-bold"> 280</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:258</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-fg"> 256</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager == <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">array</span><span class="ansi-yellow-fg">"</span>:
<span class="ansi-green-fg"> 257</span> to_pandas_kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">split_blocks</span><span class="ansi-yellow-fg">"</span>] = <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">258</span> path_or_handle, handles, filesystem = <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 259</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 260</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 261</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 262</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">rb</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 263</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 265</span> pa_table = <span style="color: rgb(0,135,0)">self</span>.api.parquet.read_table(
<span class="ansi-green-fg"> 266</span> path_or_handle,
<span class="ansi-green-fg"> 267</span> columns=columns,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 270</span> **kwargs,
<span class="ansi-green-fg"> 271</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:140</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-intense-fg ansi-bold"> 130</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 131</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-intense-fg ansi-bold"> 132</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-intense-fg ansi-bold"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 138</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-intense-fg ansi-bold"> 139</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; 140</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 141</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 143</span> fs <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 144</span> path_or_handle <span style="color: rgb(98,98,98)">=</span> handles<span style="color: rgb(98,98,98)">.</span>handle
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:141</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-fg"> 131</span> handles = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 132</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-fg"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-fg"> 134</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 139</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-fg"> 140</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">141</span> handles = <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-fg"> 143</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 144</span> fs = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 145</span> path_or_handle = handles.handle
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-intense-fg ansi-bold"> 873</span> handle <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-intense-fg ansi-bold"> 874</span> handle,
<span class="ansi-green-intense-fg ansi-bold"> 875</span> ioargs<span style="color: rgb(98,98,98)">.</span>mode,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 878</span> newline<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 879</span> )
<span class="ansi-green-intense-fg ansi-bold"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-intense-fg ansi-bold"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; 882</span> handle <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 883</span> handles<span style="color: rgb(98,98,98)">.</span>append(handle)
<span class="ansi-green-intense-fg ansi-bold"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-fg"> 873</span> handle = <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-fg"> 874</span> handle,
<span class="ansi-green-fg"> 875</span> ioargs.mode,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 878</span> newline=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 879</span> )
<span class="ansi-green-fg"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">882</span> handle = <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 883</span> handles.append(handle)
<span class="ansi-green-fg"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-red-fg">FileNotFoundError</span>: [Errno 2] No such file or directory: '&lt;your project directory&gt;/output/entities.parquet'</pre>
</div>
@ -2500,8 +2483,8 @@ File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[7], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">response</span>)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[7]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">response</span>)
<span class="ansi-red-fg">NameError</span>: name 'response' is not defined</pre>
</div>
@ -2556,8 +2539,8 @@ Cell <span class="ansi-green-fg">In[7], line 1</span>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[8], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> pprint(<span class="ansi-yellow-bg">context</span>) <span style="color: rgb(95,135,135)"># noqa: T203</span>
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[8]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> pprint(<span class="ansi-yellow-bg">context</span>) <span style="color: rgb(95,135,135)"># noqa: T203</span>
<span class="ansi-red-fg">NameError</span>: name 'context' is not defined</pre>
</div>
@ -2631,7 +2614,7 @@ Cell <span class="ansi-green-fg">In[8], line 1</span>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

350
examples_notebooks/drift_search/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2514,193 +2514,189 @@ search = DRIFTSearch(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">AuthenticationError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[6], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> resp <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> search<span style="color: rgb(98,98,98)">.</span>search(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Who is agent Mercer?</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[6]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> resp = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> search.search(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Who is agent Mercer?</span><span class="ansi-yellow-fg">"</span>)
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/search.py:213</span>, in <span class="ansi-cyan-fg">DRIFTSearch.search</span><span class="ansi-blue-fg">(self, query, conversation_history, reduce, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 210</span> <span style="color: rgb(95,135,135)"># Check if query state is empty</span>
<span class="ansi-green-intense-fg ansi-bold"> 211</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>query_state<span style="color: rgb(98,98,98)">.</span>graph:
<span class="ansi-green-intense-fg ansi-bold"> 212</span> <span style="color: rgb(95,135,135)"># Prime the search with the primer</span>
<span class="ansi-green-fg">--&gt; 213</span> primer_context, token_ct <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>context_builder<span style="color: rgb(98,98,98)">.</span>build_context(query)
<span class="ansi-green-intense-fg ansi-bold"> 214</span> llm_calls[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> token_ct[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">llm_calls</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-green-intense-fg ansi-bold"> 215</span> prompt_tokens[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> token_ct[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">prompt_tokens</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/search.py:213</span>, in <span class="ansi-cyan-fg">DRIFTSearch.search</span><span class="ansi-blue-fg">(self, query, conversation_history, reduce, **kwargs)</span>
<span class="ansi-green-fg"> 210</span> <span style="color: rgb(95,135,135)"># Check if query state is empty</span>
<span class="ansi-green-fg"> 211</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span style="color: rgb(0,135,0)">self</span>.query_state.graph:
<span class="ansi-green-fg"> 212</span> <span style="color: rgb(95,135,135)"># Prime the search with the primer</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">213</span> primer_context, token_ct = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.context_builder.build_context(query)
<span class="ansi-green-fg"> 214</span> llm_calls[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>] = token_ct[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">llm_calls</span><span class="ansi-yellow-fg">"</span>]
<span class="ansi-green-fg"> 215</span> prompt_tokens[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>] = token_ct[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">prompt_tokens</span><span class="ansi-yellow-fg">"</span>]
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/drift_context.py:199</span>, in <span class="ansi-cyan-fg">DRIFTSearchContextBuilder.build_context</span><span class="ansi-blue-fg">(self, query, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 190</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">ValueError</span>(missing_reports_error)
<span class="ansi-green-intense-fg ansi-bold"> 192</span> query_processor <span style="color: rgb(98,98,98)">=</span> PrimerQueryProcessor(
<span class="ansi-green-intense-fg ansi-bold"> 193</span> chat_model<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model,
<span class="ansi-green-intense-fg ansi-bold"> 194</span> text_embedder<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>text_embedder,
<span class="ansi-green-intense-fg ansi-bold"> 195</span> token_encoder<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>token_encoder,
<span class="ansi-green-intense-fg ansi-bold"> 196</span> reports<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>reports,
<span class="ansi-green-intense-fg ansi-bold"> 197</span> )
<span class="ansi-green-fg">--&gt; 199</span> query_embedding, token_ct <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> query_processor(query)
<span class="ansi-green-intense-fg ansi-bold"> 201</span> report_df <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>convert_reports_to_df(<span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>reports)
<span class="ansi-green-intense-fg ansi-bold"> 203</span> <span style="color: rgb(95,135,135)"># Check compatibility between query embedding and document embeddings</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/drift_context.py:199</span>, in <span class="ansi-cyan-fg">DRIFTSearchContextBuilder.build_context</span><span class="ansi-blue-fg">(self, query, **kwargs)</span>
<span class="ansi-green-fg"> 190</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">ValueError</span>(missing_reports_error)
<span class="ansi-green-fg"> 192</span> query_processor = PrimerQueryProcessor(
<span class="ansi-green-fg"> 193</span> chat_model=<span style="color: rgb(0,135,0)">self</span>.model,
<span class="ansi-green-fg"> 194</span> text_embedder=<span style="color: rgb(0,135,0)">self</span>.text_embedder,
<span class="ansi-green-fg"> 195</span> token_encoder=<span style="color: rgb(0,135,0)">self</span>.token_encoder,
<span class="ansi-green-fg"> 196</span> reports=<span style="color: rgb(0,135,0)">self</span>.reports,
<span class="ansi-green-fg"> 197</span> )
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">199</span> query_embedding, token_ct = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> query_processor(query)
<span class="ansi-green-fg"> 201</span> report_df = <span style="color: rgb(0,135,0)">self</span>.convert_reports_to_df(<span style="color: rgb(0,135,0)">self</span>.reports)
<span class="ansi-green-fg"> 203</span> <span style="color: rgb(95,135,135)"># Check compatibility between query embedding and document embeddings</span>
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/primer.py:96</span>, in <span class="ansi-cyan-fg">PrimerQueryProcessor.__call__</span><span class="ansi-blue-fg">(self, query)</span>
<span class="ansi-green-intense-fg ansi-bold"> 85</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span style="color: rgb(0,0,255)">__call__</span>(<span style="color: rgb(0,135,0)">self</span>, query: <span style="color: rgb(0,135,0)">str</span>) <span style="color: rgb(98,98,98)">-</span><span style="color: rgb(98,98,98)">&gt;</span> <span style="color: rgb(0,135,0)">tuple</span>[<span style="color: rgb(0,135,0)">list</span>[<span style="color: rgb(0,135,0)">float</span>], <span style="color: rgb(0,135,0)">dict</span>[<span style="color: rgb(0,135,0)">str</span>, <span style="color: rgb(0,135,0)">int</span>]]:
<span class="ansi-green-intense-fg ansi-bold"> 86</span> <span style="color: rgb(188,188,188)"> </span><span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 87</span> <span style="color: rgb(175,0,0)"> Call method to process the query, expand it, and embed the result.</span>
<span class="ansi-green-intense-fg ansi-bold"> 88</span>
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 94</span> <span style="color: rgb(175,0,0)"> tuple[list[float], int]: List of embeddings for the expanded query and the token count.</span>
<span class="ansi-green-intense-fg ansi-bold"> 95</span> <span style="color: rgb(175,0,0)"> """</span>
<span class="ansi-green-fg">---&gt; 96</span> hyde_query, token_ct <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>expand_query(query)
<span class="ansi-green-intense-fg ansi-bold"> 97</span> log<span style="color: rgb(98,98,98)">.</span>info(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Expanded query: </span><span class="ansi-bold" style="color: rgb(175,95,135)">%s</span><span style="color: rgb(175,0,0)">"</span>, hyde_query)
<span class="ansi-green-intense-fg ansi-bold"> 98</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>text_embedder<span style="color: rgb(98,98,98)">.</span>embed(hyde_query), token_ct
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/primer.py:96</span>, in <span class="ansi-cyan-fg">PrimerQueryProcessor.__call__</span><span class="ansi-blue-fg">(self, query)</span>
<span class="ansi-green-fg"> 85</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-fg">__call__</span>(<span style="color: rgb(0,135,0)">self</span>, query: <span style="color: rgb(0,135,0)">str</span>) -&gt; <span style="color: rgb(0,135,0)">tuple</span>[<span style="color: rgb(0,135,0)">list</span>[<span style="color: rgb(0,135,0)">float</span>], <span style="color: rgb(0,135,0)">dict</span>[<span style="color: rgb(0,135,0)">str</span>, <span style="color: rgb(0,135,0)">int</span>]]:
<span class="ansi-green-fg"> 86</span> <span style="color: rgb(188,188,188)"> </span><span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 87</span> <span class="ansi-yellow-fg"> Call method to process the query, expand it, and embed the result.</span>
<span class="ansi-green-fg"> 88</span>
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 94</span> <span class="ansi-yellow-fg"> tuple[list[float], int]: List of embeddings for the expanded query and the token count.</span>
<span class="ansi-green-fg"> 95</span> <span class="ansi-yellow-fg"> """</span>
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">96</span> hyde_query, token_ct = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.expand_query(query)
<span class="ansi-green-fg"> 97</span> logger.debug(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Expanded query: </span><span class="ansi-bold" style="color: rgb(175,95,135)">%s</span><span class="ansi-yellow-fg">"</span>, hyde_query)
<span class="ansi-green-fg"> 98</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span style="color: rgb(0,135,0)">self</span>.text_embedder.embed(hyde_query), token_ct
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/primer.py:70</span>, in <span class="ansi-cyan-fg">PrimerQueryProcessor.expand_query</span><span class="ansi-blue-fg">(self, query)</span>
<span class="ansi-green-intense-fg ansi-bold"> 63</span> template <span style="color: rgb(98,98,98)">=</span> secrets<span style="color: rgb(98,98,98)">.</span>choice(<span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>reports)<span style="color: rgb(98,98,98)">.</span>full_content <span style="color: rgb(95,135,135)"># nosec S311</span>
<span class="ansi-green-intense-fg ansi-bold"> 65</span> prompt <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"""</span><span style="color: rgb(175,0,0)">Create a hypothetical answer to the following query: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>query<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span>
<span class="ansi-green-intense-fg ansi-bold"> 66</span> <span style="color: rgb(175,0,0)"> Format it to follow the structure of the template below:</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span>
<span class="ansi-green-intense-fg ansi-bold"> 67</span> <span style="color: rgb(175,0,0)"> </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>template<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 68</span> <span style="color: rgb(175,0,0)"> Ensure that the hypothetical answer does not reference new named entities that are not present in the original query.</span><span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-fg">---&gt; 70</span> model_response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>chat_model<span style="color: rgb(98,98,98)">.</span>achat(prompt)
<span class="ansi-green-intense-fg ansi-bold"> 71</span> text <span style="color: rgb(98,98,98)">=</span> model_response<span style="color: rgb(98,98,98)">.</span>output<span style="color: rgb(98,98,98)">.</span>content
<span class="ansi-green-intense-fg ansi-bold"> 73</span> prompt_tokens <span style="color: rgb(98,98,98)">=</span> num_tokens(prompt, <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>token_encoder)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/drift_search/primer.py:70</span>, in <span class="ansi-cyan-fg">PrimerQueryProcessor.expand_query</span><span class="ansi-blue-fg">(self, query)</span>
<span class="ansi-green-fg"> 63</span> template = secrets.choice(<span style="color: rgb(0,135,0)">self</span>.reports).full_content <span style="color: rgb(95,135,135)"># nosec S311</span>
<span class="ansi-green-fg"> 65</span> prompt = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"""</span><span class="ansi-yellow-fg">Create a hypothetical answer to the following query: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>query<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span>
<span class="ansi-green-fg"> 66</span> <span class="ansi-yellow-fg"> Format it to follow the structure of the template below:</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span>
<span class="ansi-green-fg"> 67</span> <span class="ansi-yellow-fg"> </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>template<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-bold" style="color: rgb(175,95,0)">\n</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 68</span> <span class="ansi-yellow-fg"> Ensure that the hypothetical answer does not reference new named entities that are not present in the original query.</span><span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">70</span> model_response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.chat_model.achat(prompt)
<span class="ansi-green-fg"> 71</span> text = model_response.output.content
<span class="ansi-green-fg"> 73</span> prompt_tokens = num_tokens(prompt, <span style="color: rgb(0,135,0)">self</span>.token_encoder)
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/language_model/providers/fnllm/models.py:82</span>, in <span class="ansi-cyan-fg">OpenAIChatFNLLM.achat</span><span class="ansi-blue-fg">(self, prompt, history, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 70</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 71</span> <span style="color: rgb(175,0,0)">Chat with the Model using the given prompt.</span>
<span class="ansi-green-intense-fg ansi-bold"> 72</span>
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 79</span> <span style="color: rgb(175,0,0)"> The response from the Model.</span>
<span class="ansi-green-intense-fg ansi-bold"> 80</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 81</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> history <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>:
<span class="ansi-green-fg">---&gt; 82</span> response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 83</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-intense-fg ansi-bold"> 84</span> response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model(prompt, history<span style="color: rgb(98,98,98)">=</span>history, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/language_model/providers/fnllm/models.py:82</span>, in <span class="ansi-cyan-fg">OpenAIChatFNLLM.achat</span><span class="ansi-blue-fg">(self, prompt, history, **kwargs)</span>
<span class="ansi-green-fg"> 70</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 71</span> <span class="ansi-yellow-fg">Chat with the Model using the given prompt.</span>
<span class="ansi-green-fg"> 72</span>
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 79</span> <span class="ansi-yellow-fg"> The response from the Model.</span>
<span class="ansi-green-fg"> 80</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 81</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> history <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>:
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">82</span> response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.model(prompt, **kwargs)
<span class="ansi-green-fg"> 83</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg"> 84</span> response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.model(prompt, history=history, **kwargs)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_chat_llm.py:94</span>, in <span class="ansi-cyan-fg">OpenAIChatLLMImpl.__call__</span><span class="ansi-blue-fg">(self, prompt, stream, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 91</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream:
<span class="ansi-green-intense-fg ansi-bold"> 92</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_streaming_chat_llm(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-fg">---&gt; 94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_text_chat_llm(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_chat_llm.py:94</span>, in <span class="ansi-cyan-fg">OpenAIChatLLMImpl.__call__</span><span class="ansi-blue-fg">(self, prompt, stream, **kwargs)</span>
<span class="ansi-green-fg"> 91</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream:
<span class="ansi-green-fg"> 92</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._streaming_chat_llm(prompt, **kwargs)
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._text_chat_llm(prompt, **kwargs)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/services/openai_tools_parsing.py:130</span>, in <span class="ansi-cyan-fg">OpenAIParseToolsLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 127</span> tools <span style="color: rgb(98,98,98)">=</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">tools</span><span style="color: rgb(175,0,0)">"</span>, [])
<span class="ansi-green-intense-fg ansi-bold"> 129</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> tools:
<span class="ansi-green-fg">--&gt; 130</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 132</span> completion_parameters <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_add_tools_to_parameters(kwargs, tools)
<span class="ansi-green-intense-fg ansi-bold"> 134</span> result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>completion_parameters)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/services/openai_tools_parsing.py:130</span>, in <span class="ansi-cyan-fg">OpenAIParseToolsLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-fg"> 127</span> tools = kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">tools</span><span class="ansi-yellow-fg">"</span>, [])
<span class="ansi-green-fg"> 129</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> tools:
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">130</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._delegate(prompt, **kwargs)
<span class="ansi-green-fg"> 132</span> completion_parameters = <span style="color: rgb(0,135,0)">self</span>._add_tools_to_parameters(kwargs, tools)
<span class="ansi-green-fg"> 134</span> result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._delegate(prompt, **completion_parameters)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:144</span>, in <span class="ansi-cyan-fg">BaseLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 143</span> prompt, kwargs <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_rewrite_input(prompt, kwargs)
<span class="ansi-green-fg">--&gt; 144</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_decorated_target(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 145</span> <span class="ansi-bold" style="color: rgb(0,135,0)">except</span> <span class="ansi-bold" style="color: rgb(215,95,95)">BaseException</span> <span class="ansi-bold" style="color: rgb(0,135,0)">as</span> e:
<span class="ansi-green-intense-fg ansi-bold"> 146</span> stack_trace <span style="color: rgb(98,98,98)">=</span> traceback<span style="color: rgb(98,98,98)">.</span>format_exc()
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:144</span>, in <span class="ansi-cyan-fg">BaseLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 143</span> prompt, kwargs = <span style="color: rgb(0,135,0)">self</span>._rewrite_input(prompt, kwargs)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">144</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._decorated_target(prompt, **kwargs)
<span class="ansi-green-fg"> 145</span> <span class="ansi-bold" style="color: rgb(0,135,0)">except</span> <span class="ansi-bold" style="color: rgb(215,95,95)">BaseException</span> <span class="ansi-bold" style="color: rgb(0,135,0)">as</span> e:
<span class="ansi-green-fg"> 146</span> stack_trace = traceback.format_exc()
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:78</span>, in <span class="ansi-cyan-fg">JsonReceiver.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">json_model</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">json</span><span style="color: rgb(175,0,0)">"</span>):
<span class="ansi-green-intense-fg ansi-bold"> 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> this<span style="color: rgb(98,98,98)">.</span>invoke_json(delegate, prompt, kwargs)
<span class="ansi-green-fg">---&gt; 78</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:78</span>, in <span class="ansi-cyan-fg">JsonReceiver.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **kwargs)</span>
<span class="ansi-green-fg"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">json_model</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">json</span><span class="ansi-yellow-fg">"</span>):
<span class="ansi-green-fg"> 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> this.invoke_json(delegate, prompt, kwargs)
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">78</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, **kwargs)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/rate_limiter.py:75</span>, in <span class="ansi-cyan-fg">RateLimiter.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **args)</span>
<span class="ansi-green-intense-fg ansi-bold"> 73</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">with</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_limiter<span style="color: rgb(98,98,98)">.</span>use(manifest):
<span class="ansi-green-intense-fg ansi-bold"> 74</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_events<span style="color: rgb(98,98,98)">.</span>on_limit_acquired(manifest)
<span class="ansi-green-fg">---&gt; 75</span> result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>args)
<span class="ansi-green-intense-fg ansi-bold"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">finally</span>:
<span class="ansi-green-intense-fg ansi-bold"> 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_events<span style="color: rgb(98,98,98)">.</span>on_limit_released(manifest)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/rate_limiter.py:75</span>, in <span class="ansi-cyan-fg">RateLimiter.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **args)</span>
<span class="ansi-green-fg"> 73</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">with</span> <span style="color: rgb(0,135,0)">self</span>._limiter.use(manifest):
<span class="ansi-green-fg"> 74</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._events.on_limit_acquired(manifest)
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">75</span> result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, **args)
<span class="ansi-green-fg"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">finally</span>:
<span class="ansi-green-fg"> 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._events.on_limit_released(manifest)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:126</span>, in <span class="ansi-cyan-fg">BaseLLM._decorator_target</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 121</span> <span style="color: rgb(175,0,0)">"""Target for the decorator chain.</span>
<span class="ansi-green-intense-fg ansi-bold"> 122</span>
<span class="ansi-green-intense-fg ansi-bold"> 123</span> <span style="color: rgb(175,0,0)">Leave signature alone as prompt, kwargs.</span>
<span class="ansi-green-intense-fg ansi-bold"> 124</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 125</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_events<span style="color: rgb(98,98,98)">.</span>on_execute_llm()
<span class="ansi-green-fg">--&gt; 126</span> output <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_execute_llm(prompt, kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 127</span> result <span style="color: rgb(98,98,98)">=</span> LLMOutput(output<span style="color: rgb(98,98,98)">=</span>output)
<span class="ansi-green-intense-fg ansi-bold"> 128</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_inject_usage(result)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:126</span>, in <span class="ansi-cyan-fg">BaseLLM._decorator_target</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-fg"> 121</span> <span class="ansi-yellow-fg">"""Target for the decorator chain.</span>
<span class="ansi-green-fg"> 122</span>
<span class="ansi-green-fg"> 123</span> <span class="ansi-yellow-fg">Leave signature alone as prompt, kwargs.</span>
<span class="ansi-green-fg"> 124</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 125</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._events.on_execute_llm()
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">126</span> output = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._execute_llm(prompt, kwargs)
<span class="ansi-green-fg"> 127</span> result = LLMOutput(output=output)
<span class="ansi-green-fg"> 128</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._inject_usage(result)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_text_chat_llm.py:166</span>, in <span class="ansi-cyan-fg">OpenAITextChatLLMImpl._execute_llm</span><span class="ansi-blue-fg">(self, prompt, kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 163</span> local_model_parameters <span style="color: rgb(98,98,98)">=</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model_parameters</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-intense-fg ansi-bold"> 164</span> parameters <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_build_completion_parameters(local_model_parameters)
<span class="ansi-green-fg">--&gt; 166</span> raw_response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_client<span style="color: rgb(98,98,98)">.</span>chat<span style="color: rgb(98,98,98)">.</span>completions<span style="color: rgb(98,98,98)">.</span>with_raw_response<span style="color: rgb(98,98,98)">.</span>create(
<span class="ansi-green-intense-fg ansi-bold"> 167</span> messages<span style="color: rgb(98,98,98)">=</span>cast(Iterator[ChatCompletionMessageParam], messages),
<span class="ansi-green-intense-fg ansi-bold"> 168</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>parameters,
<span class="ansi-green-intense-fg ansi-bold"> 169</span> )
<span class="ansi-green-intense-fg ansi-bold"> 170</span> completion <span style="color: rgb(98,98,98)">=</span> raw_response<span style="color: rgb(98,98,98)">.</span>parse()
<span class="ansi-green-intense-fg ansi-bold"> 171</span> headers <span style="color: rgb(98,98,98)">=</span> raw_response<span style="color: rgb(98,98,98)">.</span>headers
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_text_chat_llm.py:166</span>, in <span class="ansi-cyan-fg">OpenAITextChatLLMImpl._execute_llm</span><span class="ansi-blue-fg">(self, prompt, kwargs)</span>
<span class="ansi-green-fg"> 163</span> local_model_parameters = kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model_parameters</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg"> 164</span> parameters = <span style="color: rgb(0,135,0)">self</span>._build_completion_parameters(local_model_parameters)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">166</span> raw_response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._client.chat.completions.with_raw_response.create(
<span class="ansi-green-fg"> 167</span> messages=cast(Iterator[ChatCompletionMessageParam], messages),
<span class="ansi-green-fg"> 168</span> **parameters,
<span class="ansi-green-fg"> 169</span> )
<span class="ansi-green-fg"> 170</span> completion = raw_response.parse()
<span class="ansi-green-fg"> 171</span> headers = raw_response.headers
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_legacy_response.py:381</span>, in <span class="ansi-cyan-fg">async_to_raw_response_wrapper.&lt;locals&gt;.wrapped</span><span class="ansi-blue-fg">(*args, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 377</span> extra_headers[RAW_RESPONSE_HEADER] <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">true</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 379</span> kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">extra_headers</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> extra_headers
<span class="ansi-green-fg">--&gt; 381</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> cast(LegacyAPIResponse[R], <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> func(<span style="color: rgb(98,98,98)">*</span>args, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs))
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_legacy_response.py:381</span>, in <span class="ansi-cyan-fg">async_to_raw_response_wrapper.&lt;locals&gt;.wrapped</span><span class="ansi-blue-fg">(*args, **kwargs)</span>
<span class="ansi-green-fg"> 377</span> extra_headers[RAW_RESPONSE_HEADER] = <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">true</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 379</span> kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">extra_headers</span><span class="ansi-yellow-fg">"</span>] = extra_headers
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">381</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> cast(LegacyAPIResponse[R], <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> func(*args, **kwargs))
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/resources/chat/completions/completions.py:2028</span>, in <span class="ansi-cyan-fg">AsyncCompletions.create</span><span class="ansi-blue-fg">(self, messages, model, audio, frequency_penalty, function_call, functions, logit_bias, logprobs, max_completion_tokens, max_tokens, metadata, modalities, n, parallel_tool_calls, prediction, presence_penalty, reasoning_effort, response_format, seed, service_tier, stop, store, stream, stream_options, temperature, tool_choice, tools, top_logprobs, top_p, user, web_search_options, extra_headers, extra_query, extra_body, timeout)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1985</span> <span style="color: rgb(175,0,255)">@required_args</span>([<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">messages</span><span style="color: rgb(175,0,0)">"</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model</span><span style="color: rgb(175,0,0)">"</span>], [<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">messages</span><span style="color: rgb(175,0,0)">"</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model</span><span style="color: rgb(175,0,0)">"</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stream</span><span style="color: rgb(175,0,0)">"</span>])
<span class="ansi-green-intense-fg ansi-bold"> 1986</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span style="color: rgb(0,0,255)">create</span>(
<span class="ansi-green-intense-fg ansi-bold"> 1987</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 2025</span> timeout: <span style="color: rgb(0,135,0)">float</span> <span style="color: rgb(98,98,98)">|</span> httpx<span style="color: rgb(98,98,98)">.</span>Timeout <span style="color: rgb(98,98,98)">|</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span style="color: rgb(98,98,98)">|</span> NotGiven <span style="color: rgb(98,98,98)">=</span> NOT_GIVEN,
<span class="ansi-green-intense-fg ansi-bold"> 2026</span> ) <span style="color: rgb(98,98,98)">-</span><span style="color: rgb(98,98,98)">&gt;</span> ChatCompletion <span style="color: rgb(98,98,98)">|</span> AsyncStream[ChatCompletionChunk]:
<span class="ansi-green-intense-fg ansi-bold"> 2027</span> validate_response_format(response_format)
<span class="ansi-green-fg">-&gt; 2028</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_post(
<span class="ansi-green-intense-fg ansi-bold"> 2029</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">/chat/completions</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 2030</span> body<span style="color: rgb(98,98,98)">=</span><span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_maybe_transform(
<span class="ansi-green-intense-fg ansi-bold"> 2031</span> {
<span class="ansi-green-intense-fg ansi-bold"> 2032</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">messages</span><span style="color: rgb(175,0,0)">"</span>: messages,
<span class="ansi-green-intense-fg ansi-bold"> 2033</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model</span><span style="color: rgb(175,0,0)">"</span>: model,
<span class="ansi-green-intense-fg ansi-bold"> 2034</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">audio</span><span style="color: rgb(175,0,0)">"</span>: audio,
<span class="ansi-green-intense-fg ansi-bold"> 2035</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">frequency_penalty</span><span style="color: rgb(175,0,0)">"</span>: frequency_penalty,
<span class="ansi-green-intense-fg ansi-bold"> 2036</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">function_call</span><span style="color: rgb(175,0,0)">"</span>: function_call,
<span class="ansi-green-intense-fg ansi-bold"> 2037</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">functions</span><span style="color: rgb(175,0,0)">"</span>: functions,
<span class="ansi-green-intense-fg ansi-bold"> 2038</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">logit_bias</span><span style="color: rgb(175,0,0)">"</span>: logit_bias,
<span class="ansi-green-intense-fg ansi-bold"> 2039</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">logprobs</span><span style="color: rgb(175,0,0)">"</span>: logprobs,
<span class="ansi-green-intense-fg ansi-bold"> 2040</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">max_completion_tokens</span><span style="color: rgb(175,0,0)">"</span>: max_completion_tokens,
<span class="ansi-green-intense-fg ansi-bold"> 2041</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">max_tokens</span><span style="color: rgb(175,0,0)">"</span>: max_tokens,
<span class="ansi-green-intense-fg ansi-bold"> 2042</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">metadata</span><span style="color: rgb(175,0,0)">"</span>: metadata,
<span class="ansi-green-intense-fg ansi-bold"> 2043</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">modalities</span><span style="color: rgb(175,0,0)">"</span>: modalities,
<span class="ansi-green-intense-fg ansi-bold"> 2044</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">n</span><span style="color: rgb(175,0,0)">"</span>: n,
<span class="ansi-green-intense-fg ansi-bold"> 2045</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">parallel_tool_calls</span><span style="color: rgb(175,0,0)">"</span>: parallel_tool_calls,
<span class="ansi-green-intense-fg ansi-bold"> 2046</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">prediction</span><span style="color: rgb(175,0,0)">"</span>: prediction,
<span class="ansi-green-intense-fg ansi-bold"> 2047</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">presence_penalty</span><span style="color: rgb(175,0,0)">"</span>: presence_penalty,
<span class="ansi-green-intense-fg ansi-bold"> 2048</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reasoning_effort</span><span style="color: rgb(175,0,0)">"</span>: reasoning_effort,
<span class="ansi-green-intense-fg ansi-bold"> 2049</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">response_format</span><span style="color: rgb(175,0,0)">"</span>: response_format,
<span class="ansi-green-intense-fg ansi-bold"> 2050</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">seed</span><span style="color: rgb(175,0,0)">"</span>: seed,
<span class="ansi-green-intense-fg ansi-bold"> 2051</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">service_tier</span><span style="color: rgb(175,0,0)">"</span>: service_tier,
<span class="ansi-green-intense-fg ansi-bold"> 2052</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stop</span><span style="color: rgb(175,0,0)">"</span>: stop,
<span class="ansi-green-intense-fg ansi-bold"> 2053</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">store</span><span style="color: rgb(175,0,0)">"</span>: store,
<span class="ansi-green-intense-fg ansi-bold"> 2054</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stream</span><span style="color: rgb(175,0,0)">"</span>: stream,
<span class="ansi-green-intense-fg ansi-bold"> 2055</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stream_options</span><span style="color: rgb(175,0,0)">"</span>: stream_options,
<span class="ansi-green-intense-fg ansi-bold"> 2056</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">temperature</span><span style="color: rgb(175,0,0)">"</span>: temperature,
<span class="ansi-green-intense-fg ansi-bold"> 2057</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">tool_choice</span><span style="color: rgb(175,0,0)">"</span>: tool_choice,
<span class="ansi-green-intense-fg ansi-bold"> 2058</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">tools</span><span style="color: rgb(175,0,0)">"</span>: tools,
<span class="ansi-green-intense-fg ansi-bold"> 2059</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">top_logprobs</span><span style="color: rgb(175,0,0)">"</span>: top_logprobs,
<span class="ansi-green-intense-fg ansi-bold"> 2060</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">top_p</span><span style="color: rgb(175,0,0)">"</span>: top_p,
<span class="ansi-green-intense-fg ansi-bold"> 2061</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">user</span><span style="color: rgb(175,0,0)">"</span>: user,
<span class="ansi-green-intense-fg ansi-bold"> 2062</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">web_search_options</span><span style="color: rgb(175,0,0)">"</span>: web_search_options,
<span class="ansi-green-intense-fg ansi-bold"> 2063</span> },
<span class="ansi-green-intense-fg ansi-bold"> 2064</span> completion_create_params<span style="color: rgb(98,98,98)">.</span>CompletionCreateParamsStreaming
<span class="ansi-green-intense-fg ansi-bold"> 2065</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream
<span class="ansi-green-intense-fg ansi-bold"> 2066</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> completion_create_params<span style="color: rgb(98,98,98)">.</span>CompletionCreateParamsNonStreaming,
<span class="ansi-green-intense-fg ansi-bold"> 2067</span> ),
<span class="ansi-green-intense-fg ansi-bold"> 2068</span> options<span style="color: rgb(98,98,98)">=</span>make_request_options(
<span class="ansi-green-intense-fg ansi-bold"> 2069</span> extra_headers<span style="color: rgb(98,98,98)">=</span>extra_headers, extra_query<span style="color: rgb(98,98,98)">=</span>extra_query, extra_body<span style="color: rgb(98,98,98)">=</span>extra_body, timeout<span style="color: rgb(98,98,98)">=</span>timeout
<span class="ansi-green-intense-fg ansi-bold"> 2070</span> ),
<span class="ansi-green-intense-fg ansi-bold"> 2071</span> cast_to<span style="color: rgb(98,98,98)">=</span>ChatCompletion,
<span class="ansi-green-intense-fg ansi-bold"> 2072</span> stream<span style="color: rgb(98,98,98)">=</span>stream <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>,
<span class="ansi-green-intense-fg ansi-bold"> 2073</span> stream_cls<span style="color: rgb(98,98,98)">=</span>AsyncStream[ChatCompletionChunk],
<span class="ansi-green-intense-fg ansi-bold"> 2074</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/resources/chat/completions/completions.py:2454</span>, in <span class="ansi-cyan-fg">AsyncCompletions.create</span><span class="ansi-blue-fg">(self, messages, model, audio, frequency_penalty, function_call, functions, logit_bias, logprobs, max_completion_tokens, max_tokens, metadata, modalities, n, parallel_tool_calls, prediction, presence_penalty, reasoning_effort, response_format, seed, service_tier, stop, store, stream, stream_options, temperature, tool_choice, tools, top_logprobs, top_p, user, web_search_options, extra_headers, extra_query, extra_body, timeout)</span>
<span class="ansi-green-fg"> 2411</span> <span style="color: rgb(175,0,255)">@required_args</span>([<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">messages</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model</span><span class="ansi-yellow-fg">"</span>], [<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">messages</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stream</span><span class="ansi-yellow-fg">"</span>])
<span class="ansi-green-fg"> 2412</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-fg">create</span>(
<span class="ansi-green-fg"> 2413</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 2451</span> timeout: <span style="color: rgb(0,135,0)">float</span> | httpx.Timeout | <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> | NotGiven = NOT_GIVEN,
<span class="ansi-green-fg"> 2452</span> ) -&gt; ChatCompletion | AsyncStream[ChatCompletionChunk]:
<span class="ansi-green-fg"> 2453</span> validate_response_format(response_format)
<span class="ansi-green-fg">-&gt; </span><span class="ansi-green-fg">2454</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._post(
<span class="ansi-green-fg"> 2455</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">/chat/completions</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 2456</span> body=<span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_maybe_transform(
<span class="ansi-green-fg"> 2457</span> {
<span class="ansi-green-fg"> 2458</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">messages</span><span class="ansi-yellow-fg">"</span>: messages,
<span class="ansi-green-fg"> 2459</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model</span><span class="ansi-yellow-fg">"</span>: model,
<span class="ansi-green-fg"> 2460</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">audio</span><span class="ansi-yellow-fg">"</span>: audio,
<span class="ansi-green-fg"> 2461</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">frequency_penalty</span><span class="ansi-yellow-fg">"</span>: frequency_penalty,
<span class="ansi-green-fg"> 2462</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">function_call</span><span class="ansi-yellow-fg">"</span>: function_call,
<span class="ansi-green-fg"> 2463</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">functions</span><span class="ansi-yellow-fg">"</span>: functions,
<span class="ansi-green-fg"> 2464</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">logit_bias</span><span class="ansi-yellow-fg">"</span>: logit_bias,
<span class="ansi-green-fg"> 2465</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">logprobs</span><span class="ansi-yellow-fg">"</span>: logprobs,
<span class="ansi-green-fg"> 2466</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">max_completion_tokens</span><span class="ansi-yellow-fg">"</span>: max_completion_tokens,
<span class="ansi-green-fg"> 2467</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">max_tokens</span><span class="ansi-yellow-fg">"</span>: max_tokens,
<span class="ansi-green-fg"> 2468</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">metadata</span><span class="ansi-yellow-fg">"</span>: metadata,
<span class="ansi-green-fg"> 2469</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">modalities</span><span class="ansi-yellow-fg">"</span>: modalities,
<span class="ansi-green-fg"> 2470</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">n</span><span class="ansi-yellow-fg">"</span>: n,
<span class="ansi-green-fg"> 2471</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">parallel_tool_calls</span><span class="ansi-yellow-fg">"</span>: parallel_tool_calls,
<span class="ansi-green-fg"> 2472</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">prediction</span><span class="ansi-yellow-fg">"</span>: prediction,
<span class="ansi-green-fg"> 2473</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">presence_penalty</span><span class="ansi-yellow-fg">"</span>: presence_penalty,
<span class="ansi-green-fg"> 2474</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reasoning_effort</span><span class="ansi-yellow-fg">"</span>: reasoning_effort,
<span class="ansi-green-fg"> 2475</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">response_format</span><span class="ansi-yellow-fg">"</span>: response_format,
<span class="ansi-green-fg"> 2476</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">seed</span><span class="ansi-yellow-fg">"</span>: seed,
<span class="ansi-green-fg"> 2477</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">service_tier</span><span class="ansi-yellow-fg">"</span>: service_tier,
<span class="ansi-green-fg"> 2478</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stop</span><span class="ansi-yellow-fg">"</span>: stop,
<span class="ansi-green-fg"> 2479</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">store</span><span class="ansi-yellow-fg">"</span>: store,
<span class="ansi-green-fg"> 2480</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stream</span><span class="ansi-yellow-fg">"</span>: stream,
<span class="ansi-green-fg"> 2481</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stream_options</span><span class="ansi-yellow-fg">"</span>: stream_options,
<span class="ansi-green-fg"> 2482</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">temperature</span><span class="ansi-yellow-fg">"</span>: temperature,
<span class="ansi-green-fg"> 2483</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">tool_choice</span><span class="ansi-yellow-fg">"</span>: tool_choice,
<span class="ansi-green-fg"> 2484</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">tools</span><span class="ansi-yellow-fg">"</span>: tools,
<span class="ansi-green-fg"> 2485</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">top_logprobs</span><span class="ansi-yellow-fg">"</span>: top_logprobs,
<span class="ansi-green-fg"> 2486</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">top_p</span><span class="ansi-yellow-fg">"</span>: top_p,
<span class="ansi-green-fg"> 2487</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">user</span><span class="ansi-yellow-fg">"</span>: user,
<span class="ansi-green-fg"> 2488</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">web_search_options</span><span class="ansi-yellow-fg">"</span>: web_search_options,
<span class="ansi-green-fg"> 2489</span> },
<span class="ansi-green-fg"> 2490</span> completion_create_params.CompletionCreateParamsStreaming
<span class="ansi-green-fg"> 2491</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream
<span class="ansi-green-fg"> 2492</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> completion_create_params.CompletionCreateParamsNonStreaming,
<span class="ansi-green-fg"> 2493</span> ),
<span class="ansi-green-fg"> 2494</span> options=make_request_options(
<span class="ansi-green-fg"> 2495</span> extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
<span class="ansi-green-fg"> 2496</span> ),
<span class="ansi-green-fg"> 2497</span> cast_to=ChatCompletion,
<span class="ansi-green-fg"> 2498</span> stream=stream <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>,
<span class="ansi-green-fg"> 2499</span> stream_cls=AsyncStream[ChatCompletionChunk],
<span class="ansi-green-fg"> 2500</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1742</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.post</span><span class="ansi-blue-fg">(self, path, cast_to, body, files, options, stream, stream_cls)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1728</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span style="color: rgb(0,0,255)">post</span>(
<span class="ansi-green-intense-fg ansi-bold"> 1729</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-intense-fg ansi-bold"> 1730</span> path: <span style="color: rgb(0,135,0)">str</span>,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1737</span> stream_cls: <span style="color: rgb(0,135,0)">type</span>[_AsyncStreamT] <span style="color: rgb(98,98,98)">|</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>,
<span class="ansi-green-intense-fg ansi-bold"> 1738</span> ) <span style="color: rgb(98,98,98)">-</span><span style="color: rgb(98,98,98)">&gt;</span> ResponseT <span style="color: rgb(98,98,98)">|</span> _AsyncStreamT:
<span class="ansi-green-intense-fg ansi-bold"> 1739</span> opts <span style="color: rgb(98,98,98)">=</span> FinalRequestOptions<span style="color: rgb(98,98,98)">.</span>construct(
<span class="ansi-green-intense-fg ansi-bold"> 1740</span> method<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">post</span><span style="color: rgb(175,0,0)">"</span>, url<span style="color: rgb(98,98,98)">=</span>path, json_data<span style="color: rgb(98,98,98)">=</span>body, files<span style="color: rgb(98,98,98)">=</span><span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_to_httpx_files(files), <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>options
<span class="ansi-green-intense-fg ansi-bold"> 1741</span> )
<span class="ansi-green-fg">-&gt; 1742</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>request(cast_to, opts, stream<span style="color: rgb(98,98,98)">=</span>stream, stream_cls<span style="color: rgb(98,98,98)">=</span>stream_cls)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1784</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.post</span><span class="ansi-blue-fg">(self, path, cast_to, body, files, options, stream, stream_cls)</span>
<span class="ansi-green-fg"> 1770</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-fg">post</span>(
<span class="ansi-green-fg"> 1771</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-fg"> 1772</span> path: <span style="color: rgb(0,135,0)">str</span>,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 1779</span> stream_cls: <span style="color: rgb(0,135,0)">type</span>[_AsyncStreamT] | <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>,
<span class="ansi-green-fg"> 1780</span> ) -&gt; ResponseT | _AsyncStreamT:
<span class="ansi-green-fg"> 1781</span> opts = FinalRequestOptions.construct(
<span class="ansi-green-fg"> 1782</span> method=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">post</span><span class="ansi-yellow-fg">"</span>, url=path, json_data=body, files=<span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_to_httpx_files(files), **options
<span class="ansi-green-fg"> 1783</span> )
<span class="ansi-green-fg">-&gt; </span><span class="ansi-green-fg">1784</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.request(cast_to, opts, stream=stream, stream_cls=stream_cls)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1549</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.request</span><span class="ansi-blue-fg">(self, cast_to, options, stream, stream_cls)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1546</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> err<span style="color: rgb(98,98,98)">.</span>response<span style="color: rgb(98,98,98)">.</span>aread()
<span class="ansi-green-intense-fg ansi-bold"> 1548</span> log<span style="color: rgb(98,98,98)">.</span>debug(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Re-raising status error</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-fg">-&gt; 1549</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_make_status_error_from_response(err<span style="color: rgb(98,98,98)">.</span>response) <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 1551</span> <span class="ansi-bold" style="color: rgb(0,135,0)">break</span>
<span class="ansi-green-intense-fg ansi-bold"> 1553</span> <span class="ansi-bold" style="color: rgb(0,135,0)">assert</span> response <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">could not resolve response (should never happen)</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1584</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.request</span><span class="ansi-blue-fg">(self, cast_to, options, stream, stream_cls)</span>
<span class="ansi-green-fg"> 1581</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> err.response.aread()
<span class="ansi-green-fg"> 1583</span> log.debug(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Re-raising status error</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg">-&gt; </span><span class="ansi-green-fg">1584</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span style="color: rgb(0,135,0)">self</span>._make_status_error_from_response(err.response) <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 1586</span> <span class="ansi-bold" style="color: rgb(0,135,0)">break</span>
<span class="ansi-green-fg"> 1588</span> <span class="ansi-bold" style="color: rgb(0,135,0)">assert</span> response <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">could not resolve response (should never happen)</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-red-fg">AuthenticationError</span>: Error code: 401 - {'error': {'message': 'Incorrect API key provided: sk-proj-********************************************************************************************************************************************************zWYA. You can find your API key at https://platform.openai.com/account/api-keys.', 'type': 'invalid_request_error', 'param': None, 'code': 'invalid_api_key'}}</pre>
</div>
@ -2744,8 +2740,8 @@ File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[7], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span class="ansi-yellow-bg">resp</span><span style="color: rgb(98,98,98)">.</span>response
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[7]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span class="ansi-yellow-bg">resp</span>.response
<span class="ansi-red-fg">NameError</span>: name 'resp' is not defined</pre>
</div>
@ -2789,8 +2785,8 @@ Cell <span class="ansi-green-fg">In[7], line 1</span>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[8], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">resp</span><span style="color: rgb(98,98,98)">.</span>context_data)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[8]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">resp</span>.context_data)
<span class="ansi-red-fg">NameError</span>: name 'resp' is not defined</pre>
</div>
@ -2902,7 +2898,7 @@ Cell <span class="ansi-green-fg">In[8], line 1</span>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

18
examples_notebooks/global_search/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2705,8 +2705,8 @@ print(result.response)</div>
<div class="jp-OutputArea jp-Cell-outputArea">
<div class="jp-OutputArea-child">
<div class="jp-OutputPrompt jp-OutputArea-prompt"></div>
<div class="jp-RenderedText jp-OutputArea-output" data-mime-type="application/vnd.jupyter.stderr" tabindex="0">
<pre>Exception in _map_response_single_batch
<div class="jp-RenderedText jp-OutputArea-output" data-mime-type="text/plain" tabindex="0">
<pre>2025-07-10 00:30:51.0481 - ERROR - graphrag.query.structured_search.global_search.search - Exception in _map_response_single_batch
Traceback (most recent call last):
File "/home/runner/work/graphrag/graphrag/graphrag/query/structured_search/global_search/search.py", line 227, in _map_response_single_batch
model_response = await self.model.achat(
@ -2744,13 +2744,13 @@ Traceback (most recent call last):
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_legacy_response.py", line 381, in wrapped
return cast(LegacyAPIResponse[R], await func(*args, **kwargs))
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/resources/chat/completions/completions.py", line 2028, in create
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/resources/chat/completions/completions.py", line 2454, in create
return await self._post(
^^^^^^^^^^^^^^^^^
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py", line 1742, in post
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py", line 1784, in post
return await self.request(cast_to, opts, stream=stream, stream_cls=stream_cls)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py", line 1549, in request
File "/home/runner/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py", line 1584, in request
raise self._make_status_error_from_response(err.response) from None
openai.AuthenticationError: Error code: 401 - {'error': {'message': 'Incorrect API key provided: sk-proj-********************************************************************************************************************************************************zWYA. You can find your API key at https://platform.openai.com/account/api-keys.', 'type': 'invalid_request_error', 'param': None, 'code': 'invalid_api_key'}}
</pre>
@ -2758,8 +2758,8 @@ openai.AuthenticationError: Error code: 401 - {'error': {'message': 'Incorrect A
</div>
<div class="jp-OutputArea-child">
<div class="jp-OutputPrompt jp-OutputArea-prompt"></div>
<div class="jp-RenderedText jp-OutputArea-output" data-mime-type="application/vnd.jupyter.stderr" tabindex="0">
<pre>Warning: All map responses have score 0 (i.e., no relevant information found from the dataset), returning a canned 'I do not know' answer. You can try enabling `allow_general_knowledge` to encourage the LLM to incorporate relevant general knowledge, at the risk of increasing hallucinations.
<div class="jp-RenderedText jp-OutputArea-output" data-mime-type="text/plain" tabindex="0">
<pre>2025-07-10 00:30:51.0485 - WARNING - graphrag.query.structured_search.global_search.search - Warning: All map responses have score 0 (i.e., no relevant information found from the dataset), returning a canned 'I do not know' answer. You can try enabling `allow_general_knowledge` to encourage the LLM to incorporate relevant general knowledge, at the risk of increasing hallucinations.
</pre>
</div>
</div>
@ -3008,7 +3008,7 @@ print(
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

413
examples_notebooks/global_search_with_dynamic_community_selection/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2627,222 +2627,219 @@ print(result.response)</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">AuthenticationError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[9], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> search_engine<span style="color: rgb(98,98,98)">.</span>search(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">What is operation dulce?</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span style="color: rgb(0,135,0)">print</span>(result<span style="color: rgb(98,98,98)">.</span>response)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[9]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> search_engine.search(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">What is operation dulce?</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg"> 3</span> <span style="color: rgb(0,135,0)">print</span>(result.response)
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/global_search/search.py:153</span>, in <span class="ansi-cyan-fg">GlobalSearch.search</span><span class="ansi-blue-fg">(self, query, conversation_history, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 150</span> llm_calls, prompt_tokens, output_tokens <span style="color: rgb(98,98,98)">=</span> {}, {}, {}
<span class="ansi-green-intense-fg ansi-bold"> 152</span> start_time <span style="color: rgb(98,98,98)">=</span> time<span style="color: rgb(98,98,98)">.</span>time()
<span class="ansi-green-fg">--&gt; 153</span> context_result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>context_builder<span style="color: rgb(98,98,98)">.</span>build_context(
<span class="ansi-green-intense-fg ansi-bold"> 154</span> query<span style="color: rgb(98,98,98)">=</span>query,
<span class="ansi-green-intense-fg ansi-bold"> 155</span> conversation_history<span style="color: rgb(98,98,98)">=</span>conversation_history,
<span class="ansi-green-intense-fg ansi-bold"> 156</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>context_builder_params,
<span class="ansi-green-intense-fg ansi-bold"> 157</span> )
<span class="ansi-green-intense-fg ansi-bold"> 158</span> llm_calls[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> context_result<span style="color: rgb(98,98,98)">.</span>llm_calls
<span class="ansi-green-intense-fg ansi-bold"> 159</span> prompt_tokens[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> context_result<span style="color: rgb(98,98,98)">.</span>prompt_tokens
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/global_search/search.py:153</span>, in <span class="ansi-cyan-fg">GlobalSearch.search</span><span class="ansi-blue-fg">(self, query, conversation_history, **kwargs)</span>
<span class="ansi-green-fg"> 150</span> llm_calls, prompt_tokens, output_tokens = {}, {}, {}
<span class="ansi-green-fg"> 152</span> start_time = time.time()
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">153</span> context_result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.context_builder.build_context(
<span class="ansi-green-fg"> 154</span> query=query,
<span class="ansi-green-fg"> 155</span> conversation_history=conversation_history,
<span class="ansi-green-fg"> 156</span> **<span style="color: rgb(0,135,0)">self</span>.context_builder_params,
<span class="ansi-green-fg"> 157</span> )
<span class="ansi-green-fg"> 158</span> llm_calls[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>] = context_result.llm_calls
<span class="ansi-green-fg"> 159</span> prompt_tokens[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>] = context_result.prompt_tokens
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/global_search/community_context.py:98</span>, in <span class="ansi-cyan-fg">GlobalCommunityContext.build_context</span><span class="ansi-blue-fg">(self, query, conversation_history, use_community_summary, column_delimiter, shuffle_data, include_community_rank, min_community_rank, community_rank_name, include_community_weight, community_weight_name, normalize_community_weight, max_context_tokens, context_name, conversation_history_user_turns_only, conversation_history_max_turns, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 93</span> community_reports <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>community_reports
<span class="ansi-green-intense-fg ansi-bold"> 94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>dynamic_community_selection <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>:
<span class="ansi-green-intense-fg ansi-bold"> 95</span> (
<span class="ansi-green-intense-fg ansi-bold"> 96</span> community_reports,
<span class="ansi-green-intense-fg ansi-bold"> 97</span> dynamic_info,
<span class="ansi-green-fg">---&gt; 98</span> ) <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>dynamic_community_selection<span style="color: rgb(98,98,98)">.</span>select(query)
<span class="ansi-green-intense-fg ansi-bold"> 99</span> llm_calls <span style="color: rgb(98,98,98)">+</span><span style="color: rgb(98,98,98)">=</span> dynamic_info[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">llm_calls</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-green-intense-fg ansi-bold"> 100</span> prompt_tokens <span style="color: rgb(98,98,98)">+</span><span style="color: rgb(98,98,98)">=</span> dynamic_info[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">prompt_tokens</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/structured_search/global_search/community_context.py:98</span>, in <span class="ansi-cyan-fg">GlobalCommunityContext.build_context</span><span class="ansi-blue-fg">(self, query, conversation_history, use_community_summary, column_delimiter, shuffle_data, include_community_rank, min_community_rank, community_rank_name, include_community_weight, community_weight_name, normalize_community_weight, max_context_tokens, context_name, conversation_history_user_turns_only, conversation_history_max_turns, **kwargs)</span>
<span class="ansi-green-fg"> 93</span> community_reports = <span style="color: rgb(0,135,0)">self</span>.community_reports
<span class="ansi-green-fg"> 94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span style="color: rgb(0,135,0)">self</span>.dynamic_community_selection <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>:
<span class="ansi-green-fg"> 95</span> (
<span class="ansi-green-fg"> 96</span> community_reports,
<span class="ansi-green-fg"> 97</span> dynamic_info,
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">98</span> ) = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.dynamic_community_selection.select(query)
<span class="ansi-green-fg"> 99</span> llm_calls += dynamic_info[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">llm_calls</span><span class="ansi-yellow-fg">"</span>]
<span class="ansi-green-fg"> 100</span> prompt_tokens += dynamic_info[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">prompt_tokens</span><span class="ansi-yellow-fg">"</span>]
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/context_builder/dynamic_community_selection.py:91</span>, in <span class="ansi-cyan-fg">DynamicCommunitySelection.select</span><span class="ansi-blue-fg">(self, query)</span>
<span class="ansi-green-intense-fg ansi-bold"> 88</span> relevant_communities <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">set</span>()
<span class="ansi-green-intense-fg ansi-bold"> 90</span> <span class="ansi-bold" style="color: rgb(0,135,0)">while</span> queue:
<span class="ansi-green-fg">---&gt; 91</span> gather_results <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> asyncio<span style="color: rgb(98,98,98)">.</span>gather(<span style="color: rgb(98,98,98)">*</span>[
<span class="ansi-green-intense-fg ansi-bold"> 92</span> rate_relevancy(
<span class="ansi-green-intense-fg ansi-bold"> 93</span> query<span style="color: rgb(98,98,98)">=</span>query,
<span class="ansi-green-intense-fg ansi-bold"> 94</span> description<span style="color: rgb(98,98,98)">=</span>(
<span class="ansi-green-intense-fg ansi-bold"> 95</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>reports[community]<span style="color: rgb(98,98,98)">.</span>summary
<span class="ansi-green-intense-fg ansi-bold"> 96</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>use_summary
<span class="ansi-green-intense-fg ansi-bold"> 97</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>reports[community]<span style="color: rgb(98,98,98)">.</span>full_content
<span class="ansi-green-intense-fg ansi-bold"> 98</span> ),
<span class="ansi-green-intense-fg ansi-bold"> 99</span> model<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model,
<span class="ansi-green-intense-fg ansi-bold"> 100</span> token_encoder<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>token_encoder,
<span class="ansi-green-intense-fg ansi-bold"> 101</span> rate_query<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>rate_query,
<span class="ansi-green-intense-fg ansi-bold"> 102</span> num_repeats<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>num_repeats,
<span class="ansi-green-intense-fg ansi-bold"> 103</span> semaphore<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>semaphore,
<span class="ansi-green-intense-fg ansi-bold"> 104</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span><span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model_params,
<span class="ansi-green-intense-fg ansi-bold"> 105</span> )
<span class="ansi-green-intense-fg ansi-bold"> 106</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> community <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> queue
<span class="ansi-green-intense-fg ansi-bold"> 107</span> ])
<span class="ansi-green-intense-fg ansi-bold"> 109</span> communities_to_rate <span style="color: rgb(98,98,98)">=</span> []
<span class="ansi-green-intense-fg ansi-bold"> 110</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> community, result <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span style="color: rgb(0,135,0)">zip</span>(queue, gather_results, strict<span style="color: rgb(98,98,98)">=</span><span class="ansi-bold" style="color: rgb(0,135,0)">True</span>):
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/context_builder/dynamic_community_selection.py:91</span>, in <span class="ansi-cyan-fg">DynamicCommunitySelection.select</span><span class="ansi-blue-fg">(self, query)</span>
<span class="ansi-green-fg"> 88</span> relevant_communities = <span style="color: rgb(0,135,0)">set</span>()
<span class="ansi-green-fg"> 90</span> <span class="ansi-bold" style="color: rgb(0,135,0)">while</span> queue:
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">91</span> gather_results = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> asyncio.gather(*[
<span class="ansi-green-fg"> 92</span> rate_relevancy(
<span class="ansi-green-fg"> 93</span> query=query,
<span class="ansi-green-fg"> 94</span> description=(
<span class="ansi-green-fg"> 95</span> <span style="color: rgb(0,135,0)">self</span>.reports[community].summary
<span class="ansi-green-fg"> 96</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span style="color: rgb(0,135,0)">self</span>.use_summary
<span class="ansi-green-fg"> 97</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> <span style="color: rgb(0,135,0)">self</span>.reports[community].full_content
<span class="ansi-green-fg"> 98</span> ),
<span class="ansi-green-fg"> 99</span> model=<span style="color: rgb(0,135,0)">self</span>.model,
<span class="ansi-green-fg"> 100</span> token_encoder=<span style="color: rgb(0,135,0)">self</span>.token_encoder,
<span class="ansi-green-fg"> 101</span> rate_query=<span style="color: rgb(0,135,0)">self</span>.rate_query,
<span class="ansi-green-fg"> 102</span> num_repeats=<span style="color: rgb(0,135,0)">self</span>.num_repeats,
<span class="ansi-green-fg"> 103</span> semaphore=<span style="color: rgb(0,135,0)">self</span>.semaphore,
<span class="ansi-green-fg"> 104</span> **<span style="color: rgb(0,135,0)">self</span>.model_params,
<span class="ansi-green-fg"> 105</span> )
<span class="ansi-green-fg"> 106</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> community <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> queue
<span class="ansi-green-fg"> 107</span> ])
<span class="ansi-green-fg"> 109</span> communities_to_rate = []
<span class="ansi-green-fg"> 110</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> community, result <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span style="color: rgb(0,135,0)">zip</span>(queue, gather_results, strict=<span class="ansi-bold" style="color: rgb(0,135,0)">True</span>):
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/context_builder/rate_relevancy.py:53</span>, in <span class="ansi-cyan-fg">rate_relevancy</span><span class="ansi-blue-fg">(query, description, model, token_encoder, rate_query, num_repeats, semaphore, **model_params)</span>
<span class="ansi-green-intense-fg ansi-bold"> 51</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> _ <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span style="color: rgb(0,135,0)">range</span>(num_repeats):
<span class="ansi-green-intense-fg ansi-bold"> 52</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">with</span> semaphore <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> semaphore <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> nullcontext():
<span class="ansi-green-fg">---&gt; 53</span> model_response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> model<span style="color: rgb(98,98,98)">.</span>achat(
<span class="ansi-green-intense-fg ansi-bold"> 54</span> prompt<span style="color: rgb(98,98,98)">=</span>query, history<span style="color: rgb(98,98,98)">=</span>messages, model_parameters<span style="color: rgb(98,98,98)">=</span>model_params, json<span style="color: rgb(98,98,98)">=</span><span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-intense-fg ansi-bold"> 55</span> )
<span class="ansi-green-intense-fg ansi-bold"> 56</span> response <span style="color: rgb(98,98,98)">=</span> model_response<span style="color: rgb(98,98,98)">.</span>output<span style="color: rgb(98,98,98)">.</span>content
<span class="ansi-green-intense-fg ansi-bold"> 57</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/query/context_builder/rate_relevancy.py:53</span>, in <span class="ansi-cyan-fg">rate_relevancy</span><span class="ansi-blue-fg">(query, description, model, token_encoder, rate_query, num_repeats, semaphore, **model_params)</span>
<span class="ansi-green-fg"> 51</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> _ <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span style="color: rgb(0,135,0)">range</span>(num_repeats):
<span class="ansi-green-fg"> 52</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">with</span> semaphore <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> semaphore <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> nullcontext():
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">53</span> model_response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> model.achat(
<span class="ansi-green-fg"> 54</span> prompt=query, history=messages, model_parameters=model_params, json=<span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg"> 55</span> )
<span class="ansi-green-fg"> 56</span> response = model_response.output.content
<span class="ansi-green-fg"> 57</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/language_model/providers/fnllm/models.py:84</span>, in <span class="ansi-cyan-fg">OpenAIChatFNLLM.achat</span><span class="ansi-blue-fg">(self, prompt, history, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 82</span> response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 83</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">---&gt; 84</span> response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>model(prompt, history<span style="color: rgb(98,98,98)">=</span>history, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 85</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> BaseModelResponse(
<span class="ansi-green-intense-fg ansi-bold"> 86</span> output<span style="color: rgb(98,98,98)">=</span>BaseModelOutput(
<span class="ansi-green-intense-fg ansi-bold"> 87</span> content<span style="color: rgb(98,98,98)">=</span>response<span style="color: rgb(98,98,98)">.</span>output<span style="color: rgb(98,98,98)">.</span>content,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 94</span> metrics<span style="color: rgb(98,98,98)">=</span>response<span style="color: rgb(98,98,98)">.</span>metrics,
<span class="ansi-green-intense-fg ansi-bold"> 95</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/language_model/providers/fnllm/models.py:84</span>, in <span class="ansi-cyan-fg">OpenAIChatFNLLM.achat</span><span class="ansi-blue-fg">(self, prompt, history, **kwargs)</span>
<span class="ansi-green-fg"> 82</span> response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.model(prompt, **kwargs)
<span class="ansi-green-fg"> 83</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">84</span> response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.model(prompt, history=history, **kwargs)
<span class="ansi-green-fg"> 85</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> BaseModelResponse(
<span class="ansi-green-fg"> 86</span> output=BaseModelOutput(
<span class="ansi-green-fg"> 87</span> content=response.output.content,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 94</span> metrics=response.metrics,
<span class="ansi-green-fg"> 95</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_chat_llm.py:94</span>, in <span class="ansi-cyan-fg">OpenAIChatLLMImpl.__call__</span><span class="ansi-blue-fg">(self, prompt, stream, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 91</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream:
<span class="ansi-green-intense-fg ansi-bold"> 92</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_streaming_chat_llm(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-fg">---&gt; 94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_text_chat_llm(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_chat_llm.py:94</span>, in <span class="ansi-cyan-fg">OpenAIChatLLMImpl.__call__</span><span class="ansi-blue-fg">(self, prompt, stream, **kwargs)</span>
<span class="ansi-green-fg"> 91</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream:
<span class="ansi-green-fg"> 92</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._streaming_chat_llm(prompt, **kwargs)
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._text_chat_llm(prompt, **kwargs)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/services/openai_tools_parsing.py:130</span>, in <span class="ansi-cyan-fg">OpenAIParseToolsLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 127</span> tools <span style="color: rgb(98,98,98)">=</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">tools</span><span style="color: rgb(175,0,0)">"</span>, [])
<span class="ansi-green-intense-fg ansi-bold"> 129</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> tools:
<span class="ansi-green-fg">--&gt; 130</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 132</span> completion_parameters <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_add_tools_to_parameters(kwargs, tools)
<span class="ansi-green-intense-fg ansi-bold"> 134</span> result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>completion_parameters)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/services/openai_tools_parsing.py:130</span>, in <span class="ansi-cyan-fg">OpenAIParseToolsLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-fg"> 127</span> tools = kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">tools</span><span class="ansi-yellow-fg">"</span>, [])
<span class="ansi-green-fg"> 129</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> tools:
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">130</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._delegate(prompt, **kwargs)
<span class="ansi-green-fg"> 132</span> completion_parameters = <span style="color: rgb(0,135,0)">self</span>._add_tools_to_parameters(kwargs, tools)
<span class="ansi-green-fg"> 134</span> result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._delegate(prompt, **completion_parameters)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:144</span>, in <span class="ansi-cyan-fg">BaseLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 143</span> prompt, kwargs <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_rewrite_input(prompt, kwargs)
<span class="ansi-green-fg">--&gt; 144</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_decorated_target(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 145</span> <span class="ansi-bold" style="color: rgb(0,135,0)">except</span> <span class="ansi-bold" style="color: rgb(215,95,95)">BaseException</span> <span class="ansi-bold" style="color: rgb(0,135,0)">as</span> e:
<span class="ansi-green-intense-fg ansi-bold"> 146</span> stack_trace <span style="color: rgb(98,98,98)">=</span> traceback<span style="color: rgb(98,98,98)">.</span>format_exc()
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:144</span>, in <span class="ansi-cyan-fg">BaseLLM.__call__</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 143</span> prompt, kwargs = <span style="color: rgb(0,135,0)">self</span>._rewrite_input(prompt, kwargs)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">144</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._decorated_target(prompt, **kwargs)
<span class="ansi-green-fg"> 145</span> <span class="ansi-bold" style="color: rgb(0,135,0)">except</span> <span class="ansi-bold" style="color: rgb(215,95,95)">BaseException</span> <span class="ansi-bold" style="color: rgb(0,135,0)">as</span> e:
<span class="ansi-green-fg"> 146</span> stack_trace = traceback.format_exc()
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:77</span>, in <span class="ansi-cyan-fg">JsonReceiver.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 72</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span style="color: rgb(0,0,255)">invoke</span>(
<span class="ansi-green-intense-fg ansi-bold"> 73</span> prompt: TInput,
<span class="ansi-green-intense-fg ansi-bold"> 74</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs: Unpack[LLMInput[TJsonModel, THistoryEntry, TModelParameters]],
<span class="ansi-green-intense-fg ansi-bold"> 75</span> ) <span style="color: rgb(98,98,98)">-</span><span style="color: rgb(98,98,98)">&gt;</span> LLMOutput[TOutput, TJsonModel, THistoryEntry]:
<span class="ansi-green-intense-fg ansi-bold"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">json_model</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">json</span><span style="color: rgb(175,0,0)">"</span>):
<span class="ansi-green-fg">---&gt; 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> this<span style="color: rgb(98,98,98)">.</span>invoke_json(delegate, prompt, kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 78</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:77</span>, in <span class="ansi-cyan-fg">JsonReceiver.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **kwargs)</span>
<span class="ansi-green-fg"> 72</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-fg">invoke</span>(
<span class="ansi-green-fg"> 73</span> prompt: TInput,
<span class="ansi-green-fg"> 74</span> **kwargs: Unpack[LLMInput[TJsonModel, THistoryEntry, TModelParameters]],
<span class="ansi-green-fg"> 75</span> ) -&gt; LLMOutput[TOutput, TJsonModel, THistoryEntry]:
<span class="ansi-green-fg"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">json_model</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">json</span><span class="ansi-yellow-fg">"</span>):
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> this.invoke_json(delegate, prompt, kwargs)
<span class="ansi-green-fg"> 78</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, **kwargs)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:96</span>, in <span class="ansi-cyan-fg">JsonReceiver.invoke_json</span><span class="ansi-blue-fg">(self, delegate, prompt, kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> attempt <span style="color: rgb(98,98,98)">&gt;</span> <span style="color: rgb(98,98,98)">0</span>:
<span class="ansi-green-intense-fg ansi-bold"> 95</span> kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">bust_cache</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">---&gt; 96</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>try_receive_json(delegate, prompt, kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 97</span> <span class="ansi-bold" style="color: rgb(0,135,0)">except</span> FailedToGenerateValidJsonError <span class="ansi-bold" style="color: rgb(0,135,0)">as</span> e:
<span class="ansi-green-intense-fg ansi-bold"> 98</span> error <span style="color: rgb(98,98,98)">=</span> e
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:96</span>, in <span class="ansi-cyan-fg">JsonReceiver.invoke_json</span><span class="ansi-blue-fg">(self, delegate, prompt, kwargs)</span>
<span class="ansi-green-fg"> 94</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> attempt &gt; <span class="ansi-green-fg">0</span>:
<span class="ansi-green-fg"> 95</span> kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">bust_cache</span><span class="ansi-yellow-fg">"</span>] = <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">96</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.try_receive_json(delegate, prompt, kwargs)
<span class="ansi-green-fg"> 97</span> <span class="ansi-bold" style="color: rgb(0,135,0)">except</span> FailedToGenerateValidJsonError <span class="ansi-bold" style="color: rgb(0,135,0)">as</span> e:
<span class="ansi-green-fg"> 98</span> error = e
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:162</span>, in <span class="ansi-cyan-fg">LooseModeJsonReceiver.try_receive_json</span><span class="ansi-blue-fg">(self, delegate, prompt, kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 159</span> <span style="color: rgb(175,0,0)">"""Invoke the JSON decorator."""</span>
<span class="ansi-green-intense-fg ansi-bold"> 160</span> json_model <span style="color: rgb(98,98,98)">=</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">json_model</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-fg">--&gt; 162</span> result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 163</span> json_string <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_marshaler<span style="color: rgb(98,98,98)">.</span>extract_json_string(result)
<span class="ansi-green-intense-fg ansi-bold"> 164</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/json.py:162</span>, in <span class="ansi-cyan-fg">LooseModeJsonReceiver.try_receive_json</span><span class="ansi-blue-fg">(self, delegate, prompt, kwargs)</span>
<span class="ansi-green-fg"> 159</span> <span class="ansi-yellow-fg">"""Invoke the JSON decorator."""</span>
<span class="ansi-green-fg"> 160</span> json_model = kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">json_model</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">162</span> result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, **kwargs)
<span class="ansi-green-fg"> 163</span> json_string = <span style="color: rgb(0,135,0)">self</span>._marshaler.extract_json_string(result)
<span class="ansi-green-fg"> 164</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/rate_limiter.py:75</span>, in <span class="ansi-cyan-fg">RateLimiter.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **args)</span>
<span class="ansi-green-intense-fg ansi-bold"> 73</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">with</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_limiter<span style="color: rgb(98,98,98)">.</span>use(manifest):
<span class="ansi-green-intense-fg ansi-bold"> 74</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_events<span style="color: rgb(98,98,98)">.</span>on_limit_acquired(manifest)
<span class="ansi-green-fg">---&gt; 75</span> result <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>args)
<span class="ansi-green-intense-fg ansi-bold"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">finally</span>:
<span class="ansi-green-intense-fg ansi-bold"> 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_events<span style="color: rgb(98,98,98)">.</span>on_limit_released(manifest)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/services/rate_limiter.py:75</span>, in <span class="ansi-cyan-fg">RateLimiter.decorate.&lt;locals&gt;.invoke</span><span class="ansi-blue-fg">(prompt, **args)</span>
<span class="ansi-green-fg"> 73</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">with</span> <span style="color: rgb(0,135,0)">self</span>._limiter.use(manifest):
<span class="ansi-green-fg"> 74</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._events.on_limit_acquired(manifest)
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">75</span> result = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> delegate(prompt, **args)
<span class="ansi-green-fg"> 76</span> <span class="ansi-bold" style="color: rgb(0,135,0)">finally</span>:
<span class="ansi-green-fg"> 77</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._events.on_limit_released(manifest)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:126</span>, in <span class="ansi-cyan-fg">BaseLLM._decorator_target</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 121</span> <span style="color: rgb(175,0,0)">"""Target for the decorator chain.</span>
<span class="ansi-green-intense-fg ansi-bold"> 122</span>
<span class="ansi-green-intense-fg ansi-bold"> 123</span> <span style="color: rgb(175,0,0)">Leave signature alone as prompt, kwargs.</span>
<span class="ansi-green-intense-fg ansi-bold"> 124</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 125</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_events<span style="color: rgb(98,98,98)">.</span>on_execute_llm()
<span class="ansi-green-fg">--&gt; 126</span> output <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_execute_llm(prompt, kwargs)
<span class="ansi-green-intense-fg ansi-bold"> 127</span> result <span style="color: rgb(98,98,98)">=</span> LLMOutput(output<span style="color: rgb(98,98,98)">=</span>output)
<span class="ansi-green-intense-fg ansi-bold"> 128</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_inject_usage(result)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/base/base_llm.py:126</span>, in <span class="ansi-cyan-fg">BaseLLM._decorator_target</span><span class="ansi-blue-fg">(self, prompt, **kwargs)</span>
<span class="ansi-green-fg"> 121</span> <span class="ansi-yellow-fg">"""Target for the decorator chain.</span>
<span class="ansi-green-fg"> 122</span>
<span class="ansi-green-fg"> 123</span> <span class="ansi-yellow-fg">Leave signature alone as prompt, kwargs.</span>
<span class="ansi-green-fg"> 124</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 125</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._events.on_execute_llm()
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">126</span> output = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._execute_llm(prompt, kwargs)
<span class="ansi-green-fg"> 127</span> result = LLMOutput(output=output)
<span class="ansi-green-fg"> 128</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._inject_usage(result)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_text_chat_llm.py:166</span>, in <span class="ansi-cyan-fg">OpenAITextChatLLMImpl._execute_llm</span><span class="ansi-blue-fg">(self, prompt, kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 163</span> local_model_parameters <span style="color: rgb(98,98,98)">=</span> kwargs<span style="color: rgb(98,98,98)">.</span>get(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model_parameters</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-intense-fg ansi-bold"> 164</span> parameters <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_build_completion_parameters(local_model_parameters)
<span class="ansi-green-fg">--&gt; 166</span> raw_response <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_client<span style="color: rgb(98,98,98)">.</span>chat<span style="color: rgb(98,98,98)">.</span>completions<span style="color: rgb(98,98,98)">.</span>with_raw_response<span style="color: rgb(98,98,98)">.</span>create(
<span class="ansi-green-intense-fg ansi-bold"> 167</span> messages<span style="color: rgb(98,98,98)">=</span>cast(Iterator[ChatCompletionMessageParam], messages),
<span class="ansi-green-intense-fg ansi-bold"> 168</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>parameters,
<span class="ansi-green-intense-fg ansi-bold"> 169</span> )
<span class="ansi-green-intense-fg ansi-bold"> 170</span> completion <span style="color: rgb(98,98,98)">=</span> raw_response<span style="color: rgb(98,98,98)">.</span>parse()
<span class="ansi-green-intense-fg ansi-bold"> 171</span> headers <span style="color: rgb(98,98,98)">=</span> raw_response<span style="color: rgb(98,98,98)">.</span>headers
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/fnllm/openai/llm/openai_text_chat_llm.py:166</span>, in <span class="ansi-cyan-fg">OpenAITextChatLLMImpl._execute_llm</span><span class="ansi-blue-fg">(self, prompt, kwargs)</span>
<span class="ansi-green-fg"> 163</span> local_model_parameters = kwargs.get(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model_parameters</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg"> 164</span> parameters = <span style="color: rgb(0,135,0)">self</span>._build_completion_parameters(local_model_parameters)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">166</span> raw_response = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._client.chat.completions.with_raw_response.create(
<span class="ansi-green-fg"> 167</span> messages=cast(Iterator[ChatCompletionMessageParam], messages),
<span class="ansi-green-fg"> 168</span> **parameters,
<span class="ansi-green-fg"> 169</span> )
<span class="ansi-green-fg"> 170</span> completion = raw_response.parse()
<span class="ansi-green-fg"> 171</span> headers = raw_response.headers
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_legacy_response.py:381</span>, in <span class="ansi-cyan-fg">async_to_raw_response_wrapper.&lt;locals&gt;.wrapped</span><span class="ansi-blue-fg">(*args, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 377</span> extra_headers[RAW_RESPONSE_HEADER] <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">true</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 379</span> kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">extra_headers</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> extra_headers
<span class="ansi-green-fg">--&gt; 381</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> cast(LegacyAPIResponse[R], <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> func(<span style="color: rgb(98,98,98)">*</span>args, <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs))
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_legacy_response.py:381</span>, in <span class="ansi-cyan-fg">async_to_raw_response_wrapper.&lt;locals&gt;.wrapped</span><span class="ansi-blue-fg">(*args, **kwargs)</span>
<span class="ansi-green-fg"> 377</span> extra_headers[RAW_RESPONSE_HEADER] = <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">true</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 379</span> kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">extra_headers</span><span class="ansi-yellow-fg">"</span>] = extra_headers
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">381</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> cast(LegacyAPIResponse[R], <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> func(*args, **kwargs))
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/resources/chat/completions/completions.py:2028</span>, in <span class="ansi-cyan-fg">AsyncCompletions.create</span><span class="ansi-blue-fg">(self, messages, model, audio, frequency_penalty, function_call, functions, logit_bias, logprobs, max_completion_tokens, max_tokens, metadata, modalities, n, parallel_tool_calls, prediction, presence_penalty, reasoning_effort, response_format, seed, service_tier, stop, store, stream, stream_options, temperature, tool_choice, tools, top_logprobs, top_p, user, web_search_options, extra_headers, extra_query, extra_body, timeout)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1985</span> <span style="color: rgb(175,0,255)">@required_args</span>([<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">messages</span><span style="color: rgb(175,0,0)">"</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model</span><span style="color: rgb(175,0,0)">"</span>], [<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">messages</span><span style="color: rgb(175,0,0)">"</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model</span><span style="color: rgb(175,0,0)">"</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stream</span><span style="color: rgb(175,0,0)">"</span>])
<span class="ansi-green-intense-fg ansi-bold"> 1986</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span style="color: rgb(0,0,255)">create</span>(
<span class="ansi-green-intense-fg ansi-bold"> 1987</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 2025</span> timeout: <span style="color: rgb(0,135,0)">float</span> <span style="color: rgb(98,98,98)">|</span> httpx<span style="color: rgb(98,98,98)">.</span>Timeout <span style="color: rgb(98,98,98)">|</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span style="color: rgb(98,98,98)">|</span> NotGiven <span style="color: rgb(98,98,98)">=</span> NOT_GIVEN,
<span class="ansi-green-intense-fg ansi-bold"> 2026</span> ) <span style="color: rgb(98,98,98)">-</span><span style="color: rgb(98,98,98)">&gt;</span> ChatCompletion <span style="color: rgb(98,98,98)">|</span> AsyncStream[ChatCompletionChunk]:
<span class="ansi-green-intense-fg ansi-bold"> 2027</span> validate_response_format(response_format)
<span class="ansi-green-fg">-&gt; 2028</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_post(
<span class="ansi-green-intense-fg ansi-bold"> 2029</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">/chat/completions</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 2030</span> body<span style="color: rgb(98,98,98)">=</span><span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_maybe_transform(
<span class="ansi-green-intense-fg ansi-bold"> 2031</span> {
<span class="ansi-green-intense-fg ansi-bold"> 2032</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">messages</span><span style="color: rgb(175,0,0)">"</span>: messages,
<span class="ansi-green-intense-fg ansi-bold"> 2033</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">model</span><span style="color: rgb(175,0,0)">"</span>: model,
<span class="ansi-green-intense-fg ansi-bold"> 2034</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">audio</span><span style="color: rgb(175,0,0)">"</span>: audio,
<span class="ansi-green-intense-fg ansi-bold"> 2035</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">frequency_penalty</span><span style="color: rgb(175,0,0)">"</span>: frequency_penalty,
<span class="ansi-green-intense-fg ansi-bold"> 2036</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">function_call</span><span style="color: rgb(175,0,0)">"</span>: function_call,
<span class="ansi-green-intense-fg ansi-bold"> 2037</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">functions</span><span style="color: rgb(175,0,0)">"</span>: functions,
<span class="ansi-green-intense-fg ansi-bold"> 2038</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">logit_bias</span><span style="color: rgb(175,0,0)">"</span>: logit_bias,
<span class="ansi-green-intense-fg ansi-bold"> 2039</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">logprobs</span><span style="color: rgb(175,0,0)">"</span>: logprobs,
<span class="ansi-green-intense-fg ansi-bold"> 2040</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">max_completion_tokens</span><span style="color: rgb(175,0,0)">"</span>: max_completion_tokens,
<span class="ansi-green-intense-fg ansi-bold"> 2041</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">max_tokens</span><span style="color: rgb(175,0,0)">"</span>: max_tokens,
<span class="ansi-green-intense-fg ansi-bold"> 2042</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">metadata</span><span style="color: rgb(175,0,0)">"</span>: metadata,
<span class="ansi-green-intense-fg ansi-bold"> 2043</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">modalities</span><span style="color: rgb(175,0,0)">"</span>: modalities,
<span class="ansi-green-intense-fg ansi-bold"> 2044</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">n</span><span style="color: rgb(175,0,0)">"</span>: n,
<span class="ansi-green-intense-fg ansi-bold"> 2045</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">parallel_tool_calls</span><span style="color: rgb(175,0,0)">"</span>: parallel_tool_calls,
<span class="ansi-green-intense-fg ansi-bold"> 2046</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">prediction</span><span style="color: rgb(175,0,0)">"</span>: prediction,
<span class="ansi-green-intense-fg ansi-bold"> 2047</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">presence_penalty</span><span style="color: rgb(175,0,0)">"</span>: presence_penalty,
<span class="ansi-green-intense-fg ansi-bold"> 2048</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reasoning_effort</span><span style="color: rgb(175,0,0)">"</span>: reasoning_effort,
<span class="ansi-green-intense-fg ansi-bold"> 2049</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">response_format</span><span style="color: rgb(175,0,0)">"</span>: response_format,
<span class="ansi-green-intense-fg ansi-bold"> 2050</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">seed</span><span style="color: rgb(175,0,0)">"</span>: seed,
<span class="ansi-green-intense-fg ansi-bold"> 2051</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">service_tier</span><span style="color: rgb(175,0,0)">"</span>: service_tier,
<span class="ansi-green-intense-fg ansi-bold"> 2052</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stop</span><span style="color: rgb(175,0,0)">"</span>: stop,
<span class="ansi-green-intense-fg ansi-bold"> 2053</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">store</span><span style="color: rgb(175,0,0)">"</span>: store,
<span class="ansi-green-intense-fg ansi-bold"> 2054</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stream</span><span style="color: rgb(175,0,0)">"</span>: stream,
<span class="ansi-green-intense-fg ansi-bold"> 2055</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">stream_options</span><span style="color: rgb(175,0,0)">"</span>: stream_options,
<span class="ansi-green-intense-fg ansi-bold"> 2056</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">temperature</span><span style="color: rgb(175,0,0)">"</span>: temperature,
<span class="ansi-green-intense-fg ansi-bold"> 2057</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">tool_choice</span><span style="color: rgb(175,0,0)">"</span>: tool_choice,
<span class="ansi-green-intense-fg ansi-bold"> 2058</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">tools</span><span style="color: rgb(175,0,0)">"</span>: tools,
<span class="ansi-green-intense-fg ansi-bold"> 2059</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">top_logprobs</span><span style="color: rgb(175,0,0)">"</span>: top_logprobs,
<span class="ansi-green-intense-fg ansi-bold"> 2060</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">top_p</span><span style="color: rgb(175,0,0)">"</span>: top_p,
<span class="ansi-green-intense-fg ansi-bold"> 2061</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">user</span><span style="color: rgb(175,0,0)">"</span>: user,
<span class="ansi-green-intense-fg ansi-bold"> 2062</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">web_search_options</span><span style="color: rgb(175,0,0)">"</span>: web_search_options,
<span class="ansi-green-intense-fg ansi-bold"> 2063</span> },
<span class="ansi-green-intense-fg ansi-bold"> 2064</span> completion_create_params<span style="color: rgb(98,98,98)">.</span>CompletionCreateParamsStreaming
<span class="ansi-green-intense-fg ansi-bold"> 2065</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream
<span class="ansi-green-intense-fg ansi-bold"> 2066</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> completion_create_params<span style="color: rgb(98,98,98)">.</span>CompletionCreateParamsNonStreaming,
<span class="ansi-green-intense-fg ansi-bold"> 2067</span> ),
<span class="ansi-green-intense-fg ansi-bold"> 2068</span> options<span style="color: rgb(98,98,98)">=</span>make_request_options(
<span class="ansi-green-intense-fg ansi-bold"> 2069</span> extra_headers<span style="color: rgb(98,98,98)">=</span>extra_headers, extra_query<span style="color: rgb(98,98,98)">=</span>extra_query, extra_body<span style="color: rgb(98,98,98)">=</span>extra_body, timeout<span style="color: rgb(98,98,98)">=</span>timeout
<span class="ansi-green-intense-fg ansi-bold"> 2070</span> ),
<span class="ansi-green-intense-fg ansi-bold"> 2071</span> cast_to<span style="color: rgb(98,98,98)">=</span>ChatCompletion,
<span class="ansi-green-intense-fg ansi-bold"> 2072</span> stream<span style="color: rgb(98,98,98)">=</span>stream <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>,
<span class="ansi-green-intense-fg ansi-bold"> 2073</span> stream_cls<span style="color: rgb(98,98,98)">=</span>AsyncStream[ChatCompletionChunk],
<span class="ansi-green-intense-fg ansi-bold"> 2074</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/resources/chat/completions/completions.py:2454</span>, in <span class="ansi-cyan-fg">AsyncCompletions.create</span><span class="ansi-blue-fg">(self, messages, model, audio, frequency_penalty, function_call, functions, logit_bias, logprobs, max_completion_tokens, max_tokens, metadata, modalities, n, parallel_tool_calls, prediction, presence_penalty, reasoning_effort, response_format, seed, service_tier, stop, store, stream, stream_options, temperature, tool_choice, tools, top_logprobs, top_p, user, web_search_options, extra_headers, extra_query, extra_body, timeout)</span>
<span class="ansi-green-fg"> 2411</span> <span style="color: rgb(175,0,255)">@required_args</span>([<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">messages</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model</span><span class="ansi-yellow-fg">"</span>], [<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">messages</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stream</span><span class="ansi-yellow-fg">"</span>])
<span class="ansi-green-fg"> 2412</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-fg">create</span>(
<span class="ansi-green-fg"> 2413</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 2451</span> timeout: <span style="color: rgb(0,135,0)">float</span> | httpx.Timeout | <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> | NotGiven = NOT_GIVEN,
<span class="ansi-green-fg"> 2452</span> ) -&gt; ChatCompletion | AsyncStream[ChatCompletionChunk]:
<span class="ansi-green-fg"> 2453</span> validate_response_format(response_format)
<span class="ansi-green-fg">-&gt; </span><span class="ansi-green-fg">2454</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>._post(
<span class="ansi-green-fg"> 2455</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">/chat/completions</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 2456</span> body=<span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_maybe_transform(
<span class="ansi-green-fg"> 2457</span> {
<span class="ansi-green-fg"> 2458</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">messages</span><span class="ansi-yellow-fg">"</span>: messages,
<span class="ansi-green-fg"> 2459</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">model</span><span class="ansi-yellow-fg">"</span>: model,
<span class="ansi-green-fg"> 2460</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">audio</span><span class="ansi-yellow-fg">"</span>: audio,
<span class="ansi-green-fg"> 2461</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">frequency_penalty</span><span class="ansi-yellow-fg">"</span>: frequency_penalty,
<span class="ansi-green-fg"> 2462</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">function_call</span><span class="ansi-yellow-fg">"</span>: function_call,
<span class="ansi-green-fg"> 2463</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">functions</span><span class="ansi-yellow-fg">"</span>: functions,
<span class="ansi-green-fg"> 2464</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">logit_bias</span><span class="ansi-yellow-fg">"</span>: logit_bias,
<span class="ansi-green-fg"> 2465</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">logprobs</span><span class="ansi-yellow-fg">"</span>: logprobs,
<span class="ansi-green-fg"> 2466</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">max_completion_tokens</span><span class="ansi-yellow-fg">"</span>: max_completion_tokens,
<span class="ansi-green-fg"> 2467</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">max_tokens</span><span class="ansi-yellow-fg">"</span>: max_tokens,
<span class="ansi-green-fg"> 2468</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">metadata</span><span class="ansi-yellow-fg">"</span>: metadata,
<span class="ansi-green-fg"> 2469</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">modalities</span><span class="ansi-yellow-fg">"</span>: modalities,
<span class="ansi-green-fg"> 2470</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">n</span><span class="ansi-yellow-fg">"</span>: n,
<span class="ansi-green-fg"> 2471</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">parallel_tool_calls</span><span class="ansi-yellow-fg">"</span>: parallel_tool_calls,
<span class="ansi-green-fg"> 2472</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">prediction</span><span class="ansi-yellow-fg">"</span>: prediction,
<span class="ansi-green-fg"> 2473</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">presence_penalty</span><span class="ansi-yellow-fg">"</span>: presence_penalty,
<span class="ansi-green-fg"> 2474</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reasoning_effort</span><span class="ansi-yellow-fg">"</span>: reasoning_effort,
<span class="ansi-green-fg"> 2475</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">response_format</span><span class="ansi-yellow-fg">"</span>: response_format,
<span class="ansi-green-fg"> 2476</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">seed</span><span class="ansi-yellow-fg">"</span>: seed,
<span class="ansi-green-fg"> 2477</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">service_tier</span><span class="ansi-yellow-fg">"</span>: service_tier,
<span class="ansi-green-fg"> 2478</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stop</span><span class="ansi-yellow-fg">"</span>: stop,
<span class="ansi-green-fg"> 2479</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">store</span><span class="ansi-yellow-fg">"</span>: store,
<span class="ansi-green-fg"> 2480</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stream</span><span class="ansi-yellow-fg">"</span>: stream,
<span class="ansi-green-fg"> 2481</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">stream_options</span><span class="ansi-yellow-fg">"</span>: stream_options,
<span class="ansi-green-fg"> 2482</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">temperature</span><span class="ansi-yellow-fg">"</span>: temperature,
<span class="ansi-green-fg"> 2483</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">tool_choice</span><span class="ansi-yellow-fg">"</span>: tool_choice,
<span class="ansi-green-fg"> 2484</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">tools</span><span class="ansi-yellow-fg">"</span>: tools,
<span class="ansi-green-fg"> 2485</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">top_logprobs</span><span class="ansi-yellow-fg">"</span>: top_logprobs,
<span class="ansi-green-fg"> 2486</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">top_p</span><span class="ansi-yellow-fg">"</span>: top_p,
<span class="ansi-green-fg"> 2487</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">user</span><span class="ansi-yellow-fg">"</span>: user,
<span class="ansi-green-fg"> 2488</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">web_search_options</span><span class="ansi-yellow-fg">"</span>: web_search_options,
<span class="ansi-green-fg"> 2489</span> },
<span class="ansi-green-fg"> 2490</span> completion_create_params.CompletionCreateParamsStreaming
<span class="ansi-green-fg"> 2491</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> stream
<span class="ansi-green-fg"> 2492</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span> completion_create_params.CompletionCreateParamsNonStreaming,
<span class="ansi-green-fg"> 2493</span> ),
<span class="ansi-green-fg"> 2494</span> options=make_request_options(
<span class="ansi-green-fg"> 2495</span> extra_headers=extra_headers, extra_query=extra_query, extra_body=extra_body, timeout=timeout
<span class="ansi-green-fg"> 2496</span> ),
<span class="ansi-green-fg"> 2497</span> cast_to=ChatCompletion,
<span class="ansi-green-fg"> 2498</span> stream=stream <span class="ansi-bold" style="color: rgb(175,0,255)">or</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>,
<span class="ansi-green-fg"> 2499</span> stream_cls=AsyncStream[ChatCompletionChunk],
<span class="ansi-green-fg"> 2500</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1742</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.post</span><span class="ansi-blue-fg">(self, path, cast_to, body, files, options, stream, stream_cls)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1728</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span style="color: rgb(0,0,255)">post</span>(
<span class="ansi-green-intense-fg ansi-bold"> 1729</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-intense-fg ansi-bold"> 1730</span> path: <span style="color: rgb(0,135,0)">str</span>,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1737</span> stream_cls: <span style="color: rgb(0,135,0)">type</span>[_AsyncStreamT] <span style="color: rgb(98,98,98)">|</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>,
<span class="ansi-green-intense-fg ansi-bold"> 1738</span> ) <span style="color: rgb(98,98,98)">-</span><span style="color: rgb(98,98,98)">&gt;</span> ResponseT <span style="color: rgb(98,98,98)">|</span> _AsyncStreamT:
<span class="ansi-green-intense-fg ansi-bold"> 1739</span> opts <span style="color: rgb(98,98,98)">=</span> FinalRequestOptions<span style="color: rgb(98,98,98)">.</span>construct(
<span class="ansi-green-intense-fg ansi-bold"> 1740</span> method<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">post</span><span style="color: rgb(175,0,0)">"</span>, url<span style="color: rgb(98,98,98)">=</span>path, json_data<span style="color: rgb(98,98,98)">=</span>body, files<span style="color: rgb(98,98,98)">=</span><span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_to_httpx_files(files), <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>options
<span class="ansi-green-intense-fg ansi-bold"> 1741</span> )
<span class="ansi-green-fg">-&gt; 1742</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>request(cast_to, opts, stream<span style="color: rgb(98,98,98)">=</span>stream, stream_cls<span style="color: rgb(98,98,98)">=</span>stream_cls)
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1784</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.post</span><span class="ansi-blue-fg">(self, path, cast_to, body, files, options, stream, stream_cls)</span>
<span class="ansi-green-fg"> 1770</span> <span class="ansi-bold" style="color: rgb(0,135,0)">async</span> <span class="ansi-bold" style="color: rgb(0,135,0)">def</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-fg">post</span>(
<span class="ansi-green-fg"> 1771</span> <span style="color: rgb(0,135,0)">self</span>,
<span class="ansi-green-fg"> 1772</span> path: <span style="color: rgb(0,135,0)">str</span>,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 1779</span> stream_cls: <span style="color: rgb(0,135,0)">type</span>[_AsyncStreamT] | <span class="ansi-bold" style="color: rgb(0,135,0)">None</span> = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>,
<span class="ansi-green-fg"> 1780</span> ) -&gt; ResponseT | _AsyncStreamT:
<span class="ansi-green-fg"> 1781</span> opts = FinalRequestOptions.construct(
<span class="ansi-green-fg"> 1782</span> method=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">post</span><span class="ansi-yellow-fg">"</span>, url=path, json_data=body, files=<span class="ansi-bold" style="color: rgb(0,135,0)">await</span> async_to_httpx_files(files), **options
<span class="ansi-green-fg"> 1783</span> )
<span class="ansi-green-fg">-&gt; </span><span class="ansi-green-fg">1784</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> <span style="color: rgb(0,135,0)">self</span>.request(cast_to, opts, stream=stream, stream_cls=stream_cls)
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1549</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.request</span><span class="ansi-blue-fg">(self, cast_to, options, stream, stream_cls)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1546</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> err<span style="color: rgb(98,98,98)">.</span>response<span style="color: rgb(98,98,98)">.</span>aread()
<span class="ansi-green-intense-fg ansi-bold"> 1548</span> log<span style="color: rgb(98,98,98)">.</span>debug(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Re-raising status error</span><span style="color: rgb(175,0,0)">"</span>)
<span class="ansi-green-fg">-&gt; 1549</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>_make_status_error_from_response(err<span style="color: rgb(98,98,98)">.</span>response) <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 1551</span> <span class="ansi-bold" style="color: rgb(0,135,0)">break</span>
<span class="ansi-green-intense-fg ansi-bold"> 1553</span> <span class="ansi-bold" style="color: rgb(0,135,0)">assert</span> response <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">could not resolve response (should never happen)</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/openai/_base_client.py:1584</span>, in <span class="ansi-cyan-fg">AsyncAPIClient.request</span><span class="ansi-blue-fg">(self, cast_to, options, stream, stream_cls)</span>
<span class="ansi-green-fg"> 1581</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> err.response.aread()
<span class="ansi-green-fg"> 1583</span> log.debug(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Re-raising status error</span><span class="ansi-yellow-fg">"</span>)
<span class="ansi-green-fg">-&gt; </span><span class="ansi-green-fg">1584</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span style="color: rgb(0,135,0)">self</span>._make_status_error_from_response(err.response) <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 1586</span> <span class="ansi-bold" style="color: rgb(0,135,0)">break</span>
<span class="ansi-green-fg"> 1588</span> <span class="ansi-bold" style="color: rgb(0,135,0)">assert</span> response <span class="ansi-bold" style="color: rgb(175,0,255)">is</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">could not resolve response (should never happen)</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-red-fg">AuthenticationError</span>: Error code: 401 - {'error': {'message': 'Incorrect API key provided: sk-proj-********************************************************************************************************************************************************zWYA. You can find your API key at https://platform.openai.com/account/api-keys.', 'type': 'invalid_request_error', 'param': None, 'code': 'invalid_api_key'}}</pre>
</div>
@ -2888,9 +2885,9 @@ result.context_data["reports"]</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[10], line 2</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span style="color: rgb(95,135,135)"># inspect the data used to build the context for the LLM responses</span>
<span class="ansi-green-fg">----&gt; 2</span> <span class="ansi-yellow-bg">result</span><span style="color: rgb(98,98,98)">.</span>context_data[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reports</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[10]</span><span class="ansi-green-fg">, line 2</span>
<span class="ansi-green-fg"> 1</span> <span style="color: rgb(95,135,135)"># inspect the data used to build the context for the LLM responses</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> <span class="ansi-yellow-bg">result</span>.context_data[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reports</span><span class="ansi-yellow-fg">"</span>]
<span class="ansi-red-fg">NameError</span>: name 'result' is not defined</pre>
</div>
@ -2968,11 +2965,11 @@ print(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[11], line 2</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span style="color: rgb(95,135,135)"># inspect number of LLM calls and tokens in dynamic community selection</span>
<span class="ansi-green-fg">----&gt; 2</span> llm_calls <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">result</span><span style="color: rgb(98,98,98)">.</span>llm_calls_categories[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-green-intense-fg ansi-bold"> 3</span> prompt_tokens <span style="color: rgb(98,98,98)">=</span> result<span style="color: rgb(98,98,98)">.</span>prompt_tokens_categories[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-green-intense-fg ansi-bold"> 4</span> output_tokens <span style="color: rgb(98,98,98)">=</span> result<span style="color: rgb(98,98,98)">.</span>output_tokens_categories[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">build_context</span><span style="color: rgb(175,0,0)">"</span>]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[11]</span><span class="ansi-green-fg">, line 2</span>
<span class="ansi-green-fg"> 1</span> <span style="color: rgb(95,135,135)"># inspect number of LLM calls and tokens in dynamic community selection</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> llm_calls = <span class="ansi-yellow-bg">result</span>.llm_calls_categories[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>]
<span class="ansi-green-fg"> 3</span> prompt_tokens = result.prompt_tokens_categories[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>]
<span class="ansi-green-fg"> 4</span> output_tokens = result.output_tokens_categories[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">build_context</span><span class="ansi-yellow-fg">"</span>]
<span class="ansi-red-fg">NameError</span>: name 'result' is not defined</pre>
</div>
@ -3046,7 +3043,7 @@ Cell <span class="ansi-green-fg">In[11], line 2</span>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

95
examples_notebooks/index_migration_to_v1/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2091,41 +2091,40 @@ storage = StorageFactory().create_storage(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[3], line 6</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">config</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">load_config</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> load_config
<span class="ansi-green-intense-fg ansi-bold"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">storage</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">factory</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> StorageFactory
<span class="ansi-green-fg">----&gt; 6</span> config <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">load_config</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">Path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 7</span> storage_config <span style="color: rgb(98,98,98)">=</span> config<span style="color: rgb(98,98,98)">.</span>output<span style="color: rgb(98,98,98)">.</span>model_dump()
<span class="ansi-green-intense-fg ansi-bold"> 8</span> storage <span style="color: rgb(98,98,98)">=</span> StorageFactory()<span style="color: rgb(98,98,98)">.</span>create_storage(
<span class="ansi-green-intense-fg ansi-bold"> 9</span> storage_type<span style="color: rgb(98,98,98)">=</span>storage_config[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">type</span><span style="color: rgb(175,0,0)">"</span>],
<span class="ansi-green-intense-fg ansi-bold"> 10</span> kwargs<span style="color: rgb(98,98,98)">=</span>storage_config,
<span class="ansi-green-intense-fg ansi-bold"> 11</span> )
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[3]</span><span class="ansi-green-fg">, line 6</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">config</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">load_config</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> load_config
<span class="ansi-green-fg"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">storage</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">factory</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> StorageFactory
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">6</span> config = <span class="ansi-yellow-bg">load_config</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">Path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 7</span> storage_config = config.output.model_dump()
<span class="ansi-green-fg"> 8</span> storage = StorageFactory().create_storage(
<span class="ansi-green-fg"> 9</span> storage_type=storage_config[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">type</span><span class="ansi-yellow-fg">"</span>],
<span class="ansi-green-fg"> 10</span> kwargs=storage_config,
<span class="ansi-green-fg"> 11</span> )
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:183</span>, in <span class="ansi-cyan-fg">load_config</span><span class="ansi-blue-fg">(root_dir, config_filepath, cli_overrides)</span>
<span class="ansi-green-intense-fg ansi-bold"> 151</span> <span style="color: rgb(175,0,0)">"""Load configuration from a file.</span>
<span class="ansi-green-intense-fg ansi-bold"> 152</span>
<span class="ansi-green-intense-fg ansi-bold"> 153</span> <span style="color: rgb(175,0,0)">Parameters</span>
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 180</span> <span style="color: rgb(175,0,0)"> If there are pydantic validation errors when instantiating the config.</span>
<span class="ansi-green-intense-fg ansi-bold"> 181</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 182</span> root <span style="color: rgb(98,98,98)">=</span> root_dir<span style="color: rgb(98,98,98)">.</span>resolve()
<span class="ansi-green-fg">--&gt; 183</span> config_path <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_config_path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">config_filepath</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 184</span> _load_dotenv(config_path)
<span class="ansi-green-intense-fg ansi-bold"> 185</span> config_extension <span style="color: rgb(98,98,98)">=</span> config_path<span style="color: rgb(98,98,98)">.</span>suffix
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:183</span>, in <span class="ansi-cyan-fg">load_config</span><span class="ansi-blue-fg">(root_dir, config_filepath, cli_overrides)</span>
<span class="ansi-green-fg"> 151</span> <span class="ansi-yellow-fg">"""Load configuration from a file.</span>
<span class="ansi-green-fg"> 152</span>
<span class="ansi-green-fg"> 153</span> <span class="ansi-yellow-fg">Parameters</span>
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 180</span> <span class="ansi-yellow-fg"> If there are pydantic validation errors when instantiating the config.</span>
<span class="ansi-green-fg"> 181</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 182</span> root = root_dir.resolve()
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">183</span> config_path = <span class="ansi-yellow-bg">_get_config_path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">config_filepath</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 184</span> _load_dotenv(config_path)
<span class="ansi-green-fg"> 185</span> config_extension = config_path.suffix
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:106</span>, in <span class="ansi-cyan-fg">_get_config_path</span><span class="ansi-blue-fg">(root_dir, config_filepath)</span>
<span class="ansi-green-intense-fg ansi-bold"> 104</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-intense-fg ansi-bold"> 105</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">--&gt; 106</span> config_path <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_search_for_config_in_root_dir</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root_dir</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 108</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> config_path:
<span class="ansi-green-intense-fg ansi-bold"> 109</span> msg <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Config file not found in root directory: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root_dir<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:106</span>, in <span class="ansi-cyan-fg">_get_config_path</span><span class="ansi-blue-fg">(root_dir, config_filepath)</span>
<span class="ansi-green-fg"> 104</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-fg"> 105</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">106</span> config_path = <span class="ansi-yellow-bg">_search_for_config_in_root_dir</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root_dir</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 108</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> config_path:
<span class="ansi-green-fg"> 109</span> msg = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Config file not found in root directory: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root_dir<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">"</span>
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:40</span>, in <span class="ansi-cyan-fg">_search_for_config_in_root_dir</span><span class="ansi-blue-fg">(root)</span>
<span class="ansi-green-intense-fg ansi-bold"> 38</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> root<span style="color: rgb(98,98,98)">.</span>is_dir():
<span class="ansi-green-intense-fg ansi-bold"> 39</span> msg <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Invalid config path: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)"> is not a directory</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-fg">---&gt; 40</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-intense-fg ansi-bold"> 42</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> file <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> _default_config_files:
<span class="ansi-green-intense-fg ansi-bold"> 43</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (root <span style="color: rgb(98,98,98)">/</span> file)<span style="color: rgb(98,98,98)">.</span>is_file():
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:40</span>, in <span class="ansi-cyan-fg">_search_for_config_in_root_dir</span><span class="ansi-blue-fg">(root)</span>
<span class="ansi-green-fg"> 38</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> root.is_dir():
<span class="ansi-green-fg"> 39</span> msg = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Invalid config path: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg"> is not a directory</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">40</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-fg"> 42</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> file <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> _default_config_files:
<span class="ansi-green-fg"> 43</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (root / file).is_file():
<span class="ansi-red-fg">FileNotFoundError</span>: Invalid config path: /home/runner/work/graphrag/graphrag/docs/examples_notebooks/&lt;your project directory is not a directory</pre>
</div>
@ -2453,13 +2452,13 @@ await write_table_to_storage(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[6], line 8</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">utils</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">storage</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> load_table_from_storage, write_table_to_storage
<span class="ansi-green-intense-fg ansi-bold"> 5</span> <span style="color: rgb(95,135,135)"># First we'll go through any parquet files that had model changes and update them</span>
<span class="ansi-green-intense-fg ansi-bold"> 6</span> <span style="color: rgb(95,135,135)"># The new data model may have removed excess columns as well, but we will only make the minimal changes required for compatibility</span>
<span class="ansi-green-fg">----&gt; 8</span> final_documents <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">create_final_documents</span><span style="color: rgb(175,0,0)">"</span>, <span class="ansi-yellow-bg">storage</span>)
<span class="ansi-green-intense-fg ansi-bold"> 9</span> final_text_units <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">create_final_text_units</span><span style="color: rgb(175,0,0)">"</span>, storage)
<span class="ansi-green-intense-fg ansi-bold"> 10</span> final_entities <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">create_final_entities</span><span style="color: rgb(175,0,0)">"</span>, storage)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[6]</span><span class="ansi-green-fg">, line 8</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">utils</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">storage</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> load_table_from_storage, write_table_to_storage
<span class="ansi-green-fg"> 5</span> <span style="color: rgb(95,135,135)"># First we'll go through any parquet files that had model changes and update them</span>
<span class="ansi-green-fg"> 6</span> <span style="color: rgb(95,135,135)"># The new data model may have removed excess columns as well, but we will only make the minimal changes required for compatibility</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">8</span> final_documents = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">create_final_documents</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-bg">storage</span>)
<span class="ansi-green-fg"> 9</span> final_text_units = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">create_final_text_units</span><span class="ansi-yellow-fg">"</span>, storage)
<span class="ansi-green-fg"> 10</span> final_entities = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">create_final_entities</span><span class="ansi-yellow-fg">"</span>, storage)
<span class="ansi-red-fg">NameError</span>: name 'storage' is not defined</pre>
</div>
@ -2565,13 +2564,13 @@ await generate_text_embeddings(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">ImportError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[7], line 3</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">cache</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">factory</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> CacheFactory
<span class="ansi-green-intense-fg ansi-bold"> 2</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">callbacks</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">noop_workflow_callbacks</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> NoopWorkflowCallbacks
<span class="ansi-green-fg">----&gt; 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">config</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">embeddings</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> get_embedded_fields, get_embedding_settings
<span class="ansi-green-intense-fg ansi-bold"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">index</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">flows</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">generate_text_embeddings</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> generate_text_embeddings
<span class="ansi-green-intense-fg ansi-bold"> 6</span> <span style="color: rgb(95,135,135)"># We only need to re-run the embeddings workflow, to ensure that embeddings for all required search fields are in place</span>
<span class="ansi-green-intense-fg ansi-bold"> 7</span> <span style="color: rgb(95,135,135)"># We'll construct the context and run this function flow directly to avoid everything else</span>
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[7]</span><span class="ansi-green-fg">, line 3</span>
<span class="ansi-green-fg"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">cache</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">factory</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> CacheFactory
<span class="ansi-green-fg"> 2</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">callbacks</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">noop_workflow_callbacks</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> NoopWorkflowCallbacks
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">config</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">embeddings</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> get_embedded_fields, get_embedding_settings
<span class="ansi-green-fg"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">index</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">flows</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">generate_text_embeddings</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> generate_text_embeddings
<span class="ansi-green-fg"> 6</span> <span style="color: rgb(95,135,135)"># We only need to re-run the embeddings workflow, to ensure that embeddings for all required search fields are in place</span>
<span class="ansi-green-fg"> 7</span> <span style="color: rgb(95,135,135)"># We'll construct the context and run this function flow directly to avoid everything else</span>
<span class="ansi-red-fg">ImportError</span>: cannot import name 'get_embedded_fields' from 'graphrag.config.embeddings' (/home/runner/work/graphrag/graphrag/graphrag/config/embeddings.py)</pre>
</div>
@ -2645,7 +2644,7 @@ Cell <span class="ansi-green-fg">In[7], line 3</span>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

87
examples_notebooks/index_migration_to_v2/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2087,41 +2087,40 @@ storage = StorageFactory().create_storage(
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[3], line 6</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">config</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">load_config</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> load_config
<span class="ansi-green-intense-fg ansi-bold"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">storage</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">factory</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> StorageFactory
<span class="ansi-green-fg">----&gt; 6</span> config <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">load_config</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">Path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 7</span> storage_config <span style="color: rgb(98,98,98)">=</span> config<span style="color: rgb(98,98,98)">.</span>output<span style="color: rgb(98,98,98)">.</span>model_dump()
<span class="ansi-green-intense-fg ansi-bold"> 8</span> storage <span style="color: rgb(98,98,98)">=</span> StorageFactory()<span style="color: rgb(98,98,98)">.</span>create_storage(
<span class="ansi-green-intense-fg ansi-bold"> 9</span> storage_type<span style="color: rgb(98,98,98)">=</span>storage_config[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">type</span><span style="color: rgb(175,0,0)">"</span>],
<span class="ansi-green-intense-fg ansi-bold"> 10</span> kwargs<span style="color: rgb(98,98,98)">=</span>storage_config,
<span class="ansi-green-intense-fg ansi-bold"> 11</span> )
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[3]</span><span class="ansi-green-fg">, line 6</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">config</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">load_config</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> load_config
<span class="ansi-green-fg"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">storage</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">factory</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> StorageFactory
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">6</span> config = <span class="ansi-yellow-bg">load_config</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">Path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">PROJECT_DIRECTORY</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 7</span> storage_config = config.output.model_dump()
<span class="ansi-green-fg"> 8</span> storage = StorageFactory().create_storage(
<span class="ansi-green-fg"> 9</span> storage_type=storage_config[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">type</span><span class="ansi-yellow-fg">"</span>],
<span class="ansi-green-fg"> 10</span> kwargs=storage_config,
<span class="ansi-green-fg"> 11</span> )
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:183</span>, in <span class="ansi-cyan-fg">load_config</span><span class="ansi-blue-fg">(root_dir, config_filepath, cli_overrides)</span>
<span class="ansi-green-intense-fg ansi-bold"> 151</span> <span style="color: rgb(175,0,0)">"""Load configuration from a file.</span>
<span class="ansi-green-intense-fg ansi-bold"> 152</span>
<span class="ansi-green-intense-fg ansi-bold"> 153</span> <span style="color: rgb(175,0,0)">Parameters</span>
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 180</span> <span style="color: rgb(175,0,0)"> If there are pydantic validation errors when instantiating the config.</span>
<span class="ansi-green-intense-fg ansi-bold"> 181</span> <span style="color: rgb(175,0,0)">"""</span>
<span class="ansi-green-intense-fg ansi-bold"> 182</span> root <span style="color: rgb(98,98,98)">=</span> root_dir<span style="color: rgb(98,98,98)">.</span>resolve()
<span class="ansi-green-fg">--&gt; 183</span> config_path <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_config_path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">config_filepath</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 184</span> _load_dotenv(config_path)
<span class="ansi-green-intense-fg ansi-bold"> 185</span> config_extension <span style="color: rgb(98,98,98)">=</span> config_path<span style="color: rgb(98,98,98)">.</span>suffix
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:183</span>, in <span class="ansi-cyan-fg">load_config</span><span class="ansi-blue-fg">(root_dir, config_filepath, cli_overrides)</span>
<span class="ansi-green-fg"> 151</span> <span class="ansi-yellow-fg">"""Load configuration from a file.</span>
<span class="ansi-green-fg"> 152</span>
<span class="ansi-green-fg"> 153</span> <span class="ansi-yellow-fg">Parameters</span>
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 180</span> <span class="ansi-yellow-fg"> If there are pydantic validation errors when instantiating the config.</span>
<span class="ansi-green-fg"> 181</span> <span class="ansi-yellow-fg">"""</span>
<span class="ansi-green-fg"> 182</span> root = root_dir.resolve()
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">183</span> config_path = <span class="ansi-yellow-bg">_get_config_path</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">config_filepath</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 184</span> _load_dotenv(config_path)
<span class="ansi-green-fg"> 185</span> config_extension = config_path.suffix
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:106</span>, in <span class="ansi-cyan-fg">_get_config_path</span><span class="ansi-blue-fg">(root_dir, config_filepath)</span>
<span class="ansi-green-intense-fg ansi-bold"> 104</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-intense-fg ansi-bold"> 105</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">--&gt; 106</span> config_path <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_search_for_config_in_root_dir</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root_dir</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 108</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> config_path:
<span class="ansi-green-intense-fg ansi-bold"> 109</span> msg <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Config file not found in root directory: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root_dir<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:106</span>, in <span class="ansi-cyan-fg">_get_config_path</span><span class="ansi-blue-fg">(root_dir, config_filepath)</span>
<span class="ansi-green-fg"> 104</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-fg"> 105</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">106</span> config_path = <span class="ansi-yellow-bg">_search_for_config_in_root_dir</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">root_dir</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 108</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> config_path:
<span class="ansi-green-fg"> 109</span> msg = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Config file not found in root directory: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root_dir<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">"</span>
File <span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:40</span>, in <span class="ansi-cyan-fg">_search_for_config_in_root_dir</span><span class="ansi-blue-fg">(root)</span>
<span class="ansi-green-intense-fg ansi-bold"> 38</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> root<span style="color: rgb(98,98,98)">.</span>is_dir():
<span class="ansi-green-intense-fg ansi-bold"> 39</span> msg <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">Invalid config path: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)"> is not a directory</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-fg">---&gt; 40</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-intense-fg ansi-bold"> 42</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> file <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> _default_config_files:
<span class="ansi-green-intense-fg ansi-bold"> 43</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (root <span style="color: rgb(98,98,98)">/</span> file)<span style="color: rgb(98,98,98)">.</span>is_file():
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/work/graphrag/graphrag/graphrag/config/load_config.py:40</span>, in <span class="ansi-cyan-fg">_search_for_config_in_root_dir</span><span class="ansi-blue-fg">(root)</span>
<span class="ansi-green-fg"> 38</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> root.is_dir():
<span class="ansi-green-fg"> 39</span> msg = <span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">Invalid config path: </span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>root<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg"> is not a directory</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg">---&gt; </span><span class="ansi-green-fg">40</span> <span class="ansi-bold" style="color: rgb(0,135,0)">raise</span> <span class="ansi-bold" style="color: rgb(215,95,95)">FileNotFoundError</span>(msg)
<span class="ansi-green-fg"> 42</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> file <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> _default_config_files:
<span class="ansi-green-fg"> 43</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (root / file).is_file():
<span class="ansi-red-fg">FileNotFoundError</span>: Invalid config path: /home/runner/work/graphrag/graphrag/docs/examples_notebooks/&lt;your project directory&gt; is not a directory</pre>
</div>
@ -2344,16 +2343,16 @@ await delete_table_from_storage("create_final_community_reports", storage)</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[5], line 9</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">import</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">numpy</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">as</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">np</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,0,255)">graphrag</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">utils</span><span class="ansi-bold" style="color: rgb(0,0,255)">.</span><span class="ansi-bold" style="color: rgb(0,0,255)">storage</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> (
<span class="ansi-green-intense-fg ansi-bold"> 4</span> delete_table_from_storage,
<span class="ansi-green-intense-fg ansi-bold"> 5</span> load_table_from_storage,
<span class="ansi-green-intense-fg ansi-bold"> 6</span> write_table_to_storage,
<span class="ansi-green-intense-fg ansi-bold"> 7</span> )
<span class="ansi-green-fg">----&gt; 9</span> final_documents <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">create_final_documents</span><span style="color: rgb(175,0,0)">"</span>, <span class="ansi-yellow-bg">storage</span>)
<span class="ansi-green-intense-fg ansi-bold"> 10</span> final_text_units <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">create_final_text_units</span><span style="color: rgb(175,0,0)">"</span>, storage)
<span class="ansi-green-intense-fg ansi-bold"> 11</span> final_entities <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">create_final_entities</span><span style="color: rgb(175,0,0)">"</span>, storage)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[5]</span><span class="ansi-green-fg">, line 9</span>
<span class="ansi-green-fg"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">import</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">numpy</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">as</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">np</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-bold" style="color: rgb(0,135,0)">from</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-blue-intense-fg ansi-bold">graphrag</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">utils</span><span class="ansi-blue-intense-fg ansi-bold">.</span><span class="ansi-blue-intense-fg ansi-bold">storage</span><span style="color: rgb(188,188,188)"> </span><span class="ansi-bold" style="color: rgb(0,135,0)">import</span> (
<span class="ansi-green-fg"> 4</span> delete_table_from_storage,
<span class="ansi-green-fg"> 5</span> load_table_from_storage,
<span class="ansi-green-fg"> 6</span> write_table_to_storage,
<span class="ansi-green-fg"> 7</span> )
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">9</span> final_documents = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">create_final_documents</span><span class="ansi-yellow-fg">"</span>, <span class="ansi-yellow-bg">storage</span>)
<span class="ansi-green-fg"> 10</span> final_text_units = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">create_final_text_units</span><span class="ansi-yellow-fg">"</span>, storage)
<span class="ansi-green-fg"> 11</span> final_entities = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> load_table_from_storage(<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">create_final_entities</span><span class="ansi-yellow-fg">"</span>, storage)
<span class="ansi-red-fg">NameError</span>: name 'storage' is not defined</pre>
</div>
@ -2427,7 +2426,7 @@ Cell <span class="ansi-green-fg">In[5], line 9</span>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
examples_notebooks/inputs/operation dulce/ABOUT/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1567,7 +1567,7 @@
<script id="__config" type="application/json">{"base": "../../../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../../../scripts/create_cookie_banner.js"></script>

4
examples_notebooks/inputs/operation dulce/Operation Dulce v2 1 1/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2222,7 +2222,7 @@
<script id="__config" type="application/json">{"base": "../../../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../../../scripts/create_cookie_banner.js"></script>

1091
examples_notebooks/local_search/index.html
View File

File diff suppressed because it is too large Load Diff

631
examples_notebooks/multi_index_search/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2203,8 +2203,7 @@ vector_store_configs = {
<div class="jp-OutputArea-child">
<div class="jp-OutputPrompt jp-OutputArea-prompt"></div>
<div class="jp-RenderedText jp-OutputArea-output" data-mime-type="text/plain" tabindex="0">
<pre>
['alaska', 'california', 'dc', 'maryland', 'ny', 'washington']
<pre>['alaska', 'california', 'dc', 'maryland', 'ny', 'washington']
</pre>
</div>
</div>
@ -2415,81 +2414,78 @@ results = await task</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[4], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">[</span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span><span class="ansi-yellow-bg">]</span>
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 3</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> community_reports <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 6</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[4]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = <span class="ansi-yellow-bg">[</span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span><span class="ansi-yellow-bg">]</span>
<span class="ansi-green-fg"> 2</span> communities = [
<span class="ansi-green-fg"> 3</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/communities.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> community_reports = [
<span class="ansi-green-fg"> 6</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 7</span> ]
Cell <span class="ansi-green-fg">In[4], line 1</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> [<span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes]
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 3</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> community_reports <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 6</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[4]</span><span class="ansi-green-fg">, line 1</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = [<span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes]
<span class="ansi-green-fg"> 2</span> communities = [
<span class="ansi-green-fg"> 3</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/communities.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> community_reports = [
<span class="ansi-green-fg"> 6</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 7</span> ]
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:667</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 664</span> use_nullable_dtypes <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-intense-fg ansi-bold"> 665</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; 667</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 668</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 669</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 676</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:669</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-fg"> 666</span> use_nullable_dtypes = <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-fg"> 667</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">669</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 676</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 677</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 678</span> <span class="ansi-yellow-bg">)</span>
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:267</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">array</span><span style="color: rgb(175,0,0)">"</span>:
<span class="ansi-green-intense-fg ansi-bold"> 265</span> to_pandas_kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">split_blocks</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">True</span> <span style="color: rgb(95,135,135)"># type: ignore[assignment]</span>
<span class="ansi-green-fg">--&gt; 267</span> path_or_handle, handles, filesystem <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 268</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 269</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 270</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 271</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">rb</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 272</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 273</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 274</span> pa_table <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>api<span style="color: rgb(98,98,98)">.</span>parquet<span style="color: rgb(98,98,98)">.</span>read_table(
<span class="ansi-green-intense-fg ansi-bold"> 275</span> path_or_handle,
<span class="ansi-green-intense-fg ansi-bold"> 276</span> columns<span style="color: rgb(98,98,98)">=</span>columns,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 279</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs,
<span class="ansi-green-intense-fg ansi-bold"> 280</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:258</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-fg"> 256</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager == <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">array</span><span class="ansi-yellow-fg">"</span>:
<span class="ansi-green-fg"> 257</span> to_pandas_kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">split_blocks</span><span class="ansi-yellow-fg">"</span>] = <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">258</span> path_or_handle, handles, filesystem = <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 259</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 260</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 261</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 262</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">rb</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 263</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 265</span> pa_table = <span style="color: rgb(0,135,0)">self</span>.api.parquet.read_table(
<span class="ansi-green-fg"> 266</span> path_or_handle,
<span class="ansi-green-fg"> 267</span> columns=columns,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 270</span> **kwargs,
<span class="ansi-green-fg"> 271</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:140</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-intense-fg ansi-bold"> 130</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 131</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-intense-fg ansi-bold"> 132</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-intense-fg ansi-bold"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 138</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-intense-fg ansi-bold"> 139</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; 140</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 141</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 143</span> fs <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 144</span> path_or_handle <span style="color: rgb(98,98,98)">=</span> handles<span style="color: rgb(98,98,98)">.</span>handle
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:141</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-fg"> 131</span> handles = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 132</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-fg"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-fg"> 134</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 139</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-fg"> 140</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">141</span> handles = <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-fg"> 143</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 144</span> fs = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 145</span> path_or_handle = handles.handle
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-intense-fg ansi-bold"> 873</span> handle <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-intense-fg ansi-bold"> 874</span> handle,
<span class="ansi-green-intense-fg ansi-bold"> 875</span> ioargs<span style="color: rgb(98,98,98)">.</span>mode,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 878</span> newline<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 879</span> )
<span class="ansi-green-intense-fg ansi-bold"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-intense-fg ansi-bold"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; 882</span> handle <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 883</span> handles<span style="color: rgb(98,98,98)">.</span>append(handle)
<span class="ansi-green-intense-fg ansi-bold"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-fg"> 873</span> handle = <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-fg"> 874</span> handle,
<span class="ansi-green-fg"> 875</span> ioargs.mode,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 878</span> newline=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 879</span> )
<span class="ansi-green-fg"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">882</span> handle = <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 883</span> handles.append(handle)
<span class="ansi-green-fg"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-red-fg">FileNotFoundError</span>: [Errno 2] No such file or directory: 'inputs/alaska/entities.parquet'</pre>
</div>
@ -2544,8 +2540,8 @@ File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[5], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">0</span>])
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[5]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">0</span>])
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -2632,15 +2628,15 @@ Cell <span class="ansi-green-fg">In[5], line 1</span>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[6], line 2</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> report_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span style="color: rgb(98,98,98)">120</span>, <span style="color: rgb(98,98,98)">129</span>, <span style="color: rgb(98,98,98)">40</span>, <span style="color: rgb(98,98,98)">16</span>, <span style="color: rgb(98,98,98)">204</span>, <span style="color: rgb(98,98,98)">143</span>, <span style="color: rgb(98,98,98)">85</span>, <span style="color: rgb(98,98,98)">122</span>, <span style="color: rgb(98,98,98)">83</span>]:
<span class="ansi-green-fg">----&gt; 2</span> index_name <span style="color: rgb(98,98,98)">=</span> [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">1</span>][<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reports</span><span style="color: rgb(175,0,0)">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">id</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(0,135,0)">str</span>(report_id)][<span style="color: rgb(98,98,98)">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">index_name</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> index_id <span style="color: rgb(98,98,98)">=</span> [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> results[<span style="color: rgb(98,98,98)">1</span>][<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reports</span><span style="color: rgb(175,0,0)">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">id</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(0,135,0)">str</span>(report_id)][<span style="color: rgb(98,98,98)">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-intense-fg ansi-bold"> 6</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">index_id</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 8</span> <span style="color: rgb(0,135,0)">print</span>(report_id, index_name, index_id)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[6]</span><span class="ansi-green-fg">, line 2</span>
<span class="ansi-green-fg"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> report_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span class="ansi-green-fg">120</span>, <span class="ansi-green-fg">129</span>, <span class="ansi-green-fg">40</span>, <span class="ansi-green-fg">16</span>, <span class="ansi-green-fg">204</span>, <span class="ansi-green-fg">143</span>, <span class="ansi-green-fg">85</span>, <span class="ansi-green-fg">122</span>, <span class="ansi-green-fg">83</span>]:
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> index_name = [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">1</span>][<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reports</span><span class="ansi-yellow-fg">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">id</span><span class="ansi-yellow-fg">"</span>] == <span style="color: rgb(0,135,0)">str</span>(report_id)][<span class="ansi-green-fg">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">index_name</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> index_id = [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> results[<span class="ansi-green-fg">1</span>][<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reports</span><span class="ansi-yellow-fg">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">id</span><span class="ansi-yellow-fg">"</span>] == <span style="color: rgb(0,135,0)">str</span>(report_id)][<span class="ansi-green-fg">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-fg"> 6</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">index_id</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 7</span> ]
<span class="ansi-green-fg"> 8</span> <span style="color: rgb(0,135,0)">print</span>(report_id, index_name, index_id)
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -2761,81 +2757,78 @@ results = await task</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[7], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">[</span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span><span class="ansi-yellow-bg">]</span>
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 3</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> community_reports <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 6</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[7]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = <span class="ansi-yellow-bg">[</span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span><span class="ansi-yellow-bg">]</span>
<span class="ansi-green-fg"> 2</span> communities = [
<span class="ansi-green-fg"> 3</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/communities.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> community_reports = [
<span class="ansi-green-fg"> 6</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 7</span> ]
Cell <span class="ansi-green-fg">In[7], line 1</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> [<span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes]
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 3</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> community_reports <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 6</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[7]</span><span class="ansi-green-fg">, line 1</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = [<span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes]
<span class="ansi-green-fg"> 2</span> communities = [
<span class="ansi-green-fg"> 3</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/communities.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> community_reports = [
<span class="ansi-green-fg"> 6</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 7</span> ]
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:667</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 664</span> use_nullable_dtypes <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-intense-fg ansi-bold"> 665</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; 667</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 668</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 669</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 676</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:669</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-fg"> 666</span> use_nullable_dtypes = <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-fg"> 667</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">669</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 676</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 677</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 678</span> <span class="ansi-yellow-bg">)</span>
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:267</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">array</span><span style="color: rgb(175,0,0)">"</span>:
<span class="ansi-green-intense-fg ansi-bold"> 265</span> to_pandas_kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">split_blocks</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">True</span> <span style="color: rgb(95,135,135)"># type: ignore[assignment]</span>
<span class="ansi-green-fg">--&gt; 267</span> path_or_handle, handles, filesystem <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 268</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 269</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 270</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 271</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">rb</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 272</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 273</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 274</span> pa_table <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>api<span style="color: rgb(98,98,98)">.</span>parquet<span style="color: rgb(98,98,98)">.</span>read_table(
<span class="ansi-green-intense-fg ansi-bold"> 275</span> path_or_handle,
<span class="ansi-green-intense-fg ansi-bold"> 276</span> columns<span style="color: rgb(98,98,98)">=</span>columns,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 279</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs,
<span class="ansi-green-intense-fg ansi-bold"> 280</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:258</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-fg"> 256</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager == <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">array</span><span class="ansi-yellow-fg">"</span>:
<span class="ansi-green-fg"> 257</span> to_pandas_kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">split_blocks</span><span class="ansi-yellow-fg">"</span>] = <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">258</span> path_or_handle, handles, filesystem = <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 259</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 260</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 261</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 262</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">rb</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 263</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 265</span> pa_table = <span style="color: rgb(0,135,0)">self</span>.api.parquet.read_table(
<span class="ansi-green-fg"> 266</span> path_or_handle,
<span class="ansi-green-fg"> 267</span> columns=columns,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 270</span> **kwargs,
<span class="ansi-green-fg"> 271</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:140</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-intense-fg ansi-bold"> 130</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 131</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-intense-fg ansi-bold"> 132</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-intense-fg ansi-bold"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 138</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-intense-fg ansi-bold"> 139</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; 140</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 141</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 143</span> fs <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 144</span> path_or_handle <span style="color: rgb(98,98,98)">=</span> handles<span style="color: rgb(98,98,98)">.</span>handle
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:141</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-fg"> 131</span> handles = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 132</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-fg"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-fg"> 134</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 139</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-fg"> 140</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">141</span> handles = <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-fg"> 143</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 144</span> fs = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 145</span> path_or_handle = handles.handle
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-intense-fg ansi-bold"> 873</span> handle <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-intense-fg ansi-bold"> 874</span> handle,
<span class="ansi-green-intense-fg ansi-bold"> 875</span> ioargs<span style="color: rgb(98,98,98)">.</span>mode,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 878</span> newline<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 879</span> )
<span class="ansi-green-intense-fg ansi-bold"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-intense-fg ansi-bold"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; 882</span> handle <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 883</span> handles<span style="color: rgb(98,98,98)">.</span>append(handle)
<span class="ansi-green-intense-fg ansi-bold"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-fg"> 873</span> handle = <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-fg"> 874</span> handle,
<span class="ansi-green-fg"> 875</span> ioargs.mode,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 878</span> newline=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 879</span> )
<span class="ansi-green-fg"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">882</span> handle = <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 883</span> handles.append(handle)
<span class="ansi-green-fg"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-red-fg">FileNotFoundError</span>: [Errno 2] No such file or directory: 'inputs/alaska/entities.parquet'</pre>
</div>
@ -2890,8 +2883,8 @@ File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[8], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">0</span>])
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[8]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">0</span>])
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -3100,15 +3093,15 @@ for claim_id in [100]:
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[9], line 2</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> report_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span style="color: rgb(98,98,98)">47</span>, <span style="color: rgb(98,98,98)">213</span>]:
<span class="ansi-green-fg">----&gt; 2</span> index_name <span style="color: rgb(98,98,98)">=</span> [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">1</span>][<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reports</span><span style="color: rgb(175,0,0)">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">id</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(0,135,0)">str</span>(report_id)][<span style="color: rgb(98,98,98)">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">index_name</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> index_id <span style="color: rgb(98,98,98)">=</span> [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> results[<span style="color: rgb(98,98,98)">1</span>][<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reports</span><span style="color: rgb(175,0,0)">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">id</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(0,135,0)">str</span>(report_id)][<span style="color: rgb(98,98,98)">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-intense-fg ansi-bold"> 6</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">index_id</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 8</span> <span style="color: rgb(0,135,0)">print</span>(report_id, index_name, index_id)
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[9]</span><span class="ansi-green-fg">, line 2</span>
<span class="ansi-green-fg"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> report_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span class="ansi-green-fg">47</span>, <span class="ansi-green-fg">213</span>]:
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> index_name = [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">1</span>][<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reports</span><span class="ansi-yellow-fg">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">id</span><span class="ansi-yellow-fg">"</span>] == <span style="color: rgb(0,135,0)">str</span>(report_id)][<span class="ansi-green-fg">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">index_name</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> index_id = [i <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> i <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> results[<span class="ansi-green-fg">1</span>][<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reports</span><span class="ansi-yellow-fg">"</span>] <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> i[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">id</span><span class="ansi-yellow-fg">"</span>] == <span style="color: rgb(0,135,0)">str</span>(report_id)][<span class="ansi-green-fg">0</span>][ <span style="color: rgb(95,135,135)"># noqa: RUF015</span>
<span class="ansi-green-fg"> 6</span> <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">index_id</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 7</span> ]
<span class="ansi-green-fg"> 8</span> <span style="color: rgb(0,135,0)">print</span>(report_id, index_name, index_id)
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -3221,81 +3214,78 @@ results = await task</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[10], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">[</span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span><span class="ansi-yellow-bg">]</span>
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 3</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> community_reports <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 6</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[10]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = <span class="ansi-yellow-bg">[</span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span><span class="ansi-yellow-bg">]</span>
<span class="ansi-green-fg"> 2</span> communities = [
<span class="ansi-green-fg"> 3</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/communities.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> community_reports = [
<span class="ansi-green-fg"> 6</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 7</span> ]
Cell <span class="ansi-green-fg">In[10], line 1</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg">----&gt; 1</span> entities <span style="color: rgb(98,98,98)">=</span> [<span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/entities.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes]
<span class="ansi-green-intense-fg ansi-bold"> 2</span> communities <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 3</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/communities.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 4</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> community_reports <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-intense-fg ansi-bold"> 6</span> pd<span style="color: rgb(98,98,98)">.</span>read_parquet(<span style="color: rgb(175,0,0)">f</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span style="color: rgb(175,0,0)">/community_reports.parquet</span><span style="color: rgb(175,0,0)">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 7</span> ]
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[10]</span><span class="ansi-green-fg">, line 1</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> entities = [<span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/entities.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes]
<span class="ansi-green-fg"> 2</span> communities = [
<span class="ansi-green-fg"> 3</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/communities.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 4</span> ]
<span class="ansi-green-fg"> 5</span> community_reports = [
<span class="ansi-green-fg"> 6</span> pd.read_parquet(<span class="ansi-yellow-fg">f</span><span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">inputs/</span><span class="ansi-bold" style="color: rgb(175,95,135)">{</span>index<span class="ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg">/community_reports.parquet</span><span class="ansi-yellow-fg">"</span>) <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 7</span> ]
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:667</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 664</span> use_nullable_dtypes <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-intense-fg ansi-bold"> 665</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; 667</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 668</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 669</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 676</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:669</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-fg"> 666</span> use_nullable_dtypes = <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-fg"> 667</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">669</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 676</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 677</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 678</span> <span class="ansi-yellow-bg">)</span>
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:267</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">array</span><span style="color: rgb(175,0,0)">"</span>:
<span class="ansi-green-intense-fg ansi-bold"> 265</span> to_pandas_kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">split_blocks</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">True</span> <span style="color: rgb(95,135,135)"># type: ignore[assignment]</span>
<span class="ansi-green-fg">--&gt; 267</span> path_or_handle, handles, filesystem <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 268</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 269</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 270</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 271</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">rb</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 272</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 273</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 274</span> pa_table <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>api<span style="color: rgb(98,98,98)">.</span>parquet<span style="color: rgb(98,98,98)">.</span>read_table(
<span class="ansi-green-intense-fg ansi-bold"> 275</span> path_or_handle,
<span class="ansi-green-intense-fg ansi-bold"> 276</span> columns<span style="color: rgb(98,98,98)">=</span>columns,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 279</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs,
<span class="ansi-green-intense-fg ansi-bold"> 280</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:258</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-fg"> 256</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager == <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">array</span><span class="ansi-yellow-fg">"</span>:
<span class="ansi-green-fg"> 257</span> to_pandas_kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">split_blocks</span><span class="ansi-yellow-fg">"</span>] = <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">258</span> path_or_handle, handles, filesystem = <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 259</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 260</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 261</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 262</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">rb</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 263</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 265</span> pa_table = <span style="color: rgb(0,135,0)">self</span>.api.parquet.read_table(
<span class="ansi-green-fg"> 266</span> path_or_handle,
<span class="ansi-green-fg"> 267</span> columns=columns,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 270</span> **kwargs,
<span class="ansi-green-fg"> 271</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:140</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-intense-fg ansi-bold"> 130</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 131</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-intense-fg ansi-bold"> 132</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-intense-fg ansi-bold"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 138</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-intense-fg ansi-bold"> 139</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; 140</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 141</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 143</span> fs <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 144</span> path_or_handle <span style="color: rgb(98,98,98)">=</span> handles<span style="color: rgb(98,98,98)">.</span>handle
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:141</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-fg"> 131</span> handles = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 132</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-fg"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-fg"> 134</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 139</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-fg"> 140</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">141</span> handles = <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-fg"> 143</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 144</span> fs = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 145</span> path_or_handle = handles.handle
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-intense-fg ansi-bold"> 873</span> handle <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-intense-fg ansi-bold"> 874</span> handle,
<span class="ansi-green-intense-fg ansi-bold"> 875</span> ioargs<span style="color: rgb(98,98,98)">.</span>mode,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 878</span> newline<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 879</span> )
<span class="ansi-green-intense-fg ansi-bold"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-intense-fg ansi-bold"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; 882</span> handle <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 883</span> handles<span style="color: rgb(98,98,98)">.</span>append(handle)
<span class="ansi-green-intense-fg ansi-bold"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-fg"> 873</span> handle = <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-fg"> 874</span> handle,
<span class="ansi-green-fg"> 875</span> ioargs.mode,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 878</span> newline=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 879</span> )
<span class="ansi-green-fg"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">882</span> handle = <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 883</span> handles.append(handle)
<span class="ansi-green-fg"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-red-fg">FileNotFoundError</span>: [Errno 2] No such file or directory: 'inputs/alaska/entities.parquet'</pre>
</div>
@ -3350,8 +3340,8 @@ File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[11], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">0</span>])
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[11]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">0</span>])
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -3496,11 +3486,11 @@ for source_id in [10, 16, 19, 20, 21, 22, 24, 29, 93, 95]:
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[12], line 2</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> report_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span style="color: rgb(98,98,98)">47</span>, <span style="color: rgb(98,98,98)">236</span>]:
<span class="ansi-green-fg">----&gt; 2</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> question <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">1</span>]:
<span class="ansi-green-intense-fg ansi-bold"> 3</span> resq <span style="color: rgb(98,98,98)">=</span> results[<span style="color: rgb(98,98,98)">1</span>][question]
<span class="ansi-green-intense-fg ansi-bold"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span style="color: rgb(0,135,0)">len</span>(resq[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">reports</span><span style="color: rgb(175,0,0)">"</span>]) <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(98,98,98)">0</span>:
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[12]</span><span class="ansi-green-fg">, line 2</span>
<span class="ansi-green-fg"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> report_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span class="ansi-green-fg">47</span>, <span class="ansi-green-fg">236</span>]:
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> question <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> <span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">1</span>]:
<span class="ansi-green-fg"> 3</span> resq = results[<span class="ansi-green-fg">1</span>][question]
<span class="ansi-green-fg"> 4</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> <span style="color: rgb(0,135,0)">len</span>(resq[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">reports</span><span class="ansi-yellow-fg">"</span>]) == <span class="ansi-green-fg">0</span>:
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -3573,85 +3563,82 @@ results = await task</div>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">FileNotFoundError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[13], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> text_units <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">[</span>
<span class="ansi-green-intense-fg ansi-bold"> 2</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/text_units.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span>
<span class="ansi-green-intense-fg ansi-bold"> 3</span> <span class="ansi-yellow-bg">]</span>
<span class="ansi-green-intense-fg ansi-bold"> 5</span> task <span style="color: rgb(98,98,98)">=</span> loop<span style="color: rgb(98,98,98)">.</span>create_task(
<span class="ansi-green-intense-fg ansi-bold"> 6</span> multi_index_basic_search(
<span class="ansi-green-intense-fg ansi-bold"> 7</span> parameters, text_units, indexes, <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">industry in maryland</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 8</span> )
<span class="ansi-green-intense-fg ansi-bold"> 9</span> )
<span class="ansi-green-intense-fg ansi-bold"> 10</span> results <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> task
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[13]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> text_units = <span class="ansi-yellow-bg">[</span>
<span class="ansi-green-fg"> 2</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/text_units.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">for</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,0,255)">in</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">indexes</span>
<span class="ansi-green-fg"> 3</span> <span class="ansi-yellow-bg">]</span>
<span class="ansi-green-fg"> 5</span> task = loop.create_task(
<span class="ansi-green-fg"> 6</span> multi_index_basic_search(
<span class="ansi-green-fg"> 7</span> parameters, text_units, indexes, <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">industry in maryland</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 8</span> )
<span class="ansi-green-fg"> 9</span> )
<span class="ansi-green-fg"> 10</span> results = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> task
Cell <span class="ansi-green-fg">In[13], line 2</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> text_units <span style="color: rgb(98,98,98)">=</span> [
<span class="ansi-green-fg">----&gt; 2</span> <span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">f</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">/text_units.parquet</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-intense-fg ansi-bold"> 3</span> ]
<span class="ansi-green-intense-fg ansi-bold"> 5</span> task <span style="color: rgb(98,98,98)">=</span> loop<span style="color: rgb(98,98,98)">.</span>create_task(
<span class="ansi-green-intense-fg ansi-bold"> 6</span> multi_index_basic_search(
<span class="ansi-green-intense-fg ansi-bold"> 7</span> parameters, text_units, indexes, <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>, <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">industry in maryland</span><span style="color: rgb(175,0,0)">"</span>
<span class="ansi-green-intense-fg ansi-bold"> 8</span> )
<span class="ansi-green-intense-fg ansi-bold"> 9</span> )
<span class="ansi-green-intense-fg ansi-bold"> 10</span> results <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> task
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[13]</span><span class="ansi-green-fg">, line 2</span>, in <span class="ansi-cyan-fg">&lt;listcomp&gt;</span><span class="ansi-blue-fg">(.0)</span>
<span class="ansi-green-fg"> 1</span> text_units = [
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> <span class="ansi-yellow-bg">pd</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read_parquet</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-fg ansi-yellow-bg">f</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">inputs/</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">{</span><span class="ansi-yellow-bg">index</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(175,95,135)">}</span><span class="ansi-yellow-fg ansi-yellow-bg">/text_units.parquet</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">)</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> index <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> indexes
<span class="ansi-green-fg"> 3</span> ]
<span class="ansi-green-fg"> 5</span> task = loop.create_task(
<span class="ansi-green-fg"> 6</span> multi_index_basic_search(
<span class="ansi-green-fg"> 7</span> parameters, text_units, indexes, <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>, <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">industry in maryland</span><span class="ansi-yellow-fg">"</span>
<span class="ansi-green-fg"> 8</span> )
<span class="ansi-green-fg"> 9</span> )
<span class="ansi-green-fg"> 10</span> results = <span class="ansi-bold" style="color: rgb(0,135,0)">await</span> task
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:667</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 664</span> use_nullable_dtypes <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-intense-fg ansi-bold"> 665</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; 667</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 668</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 669</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 676</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:669</span>, in <span class="ansi-cyan-fg">read_parquet</span><span class="ansi-blue-fg">(path, engine, columns, storage_options, use_nullable_dtypes, dtype_backend, filesystem, filters, **kwargs)</span>
<span class="ansi-green-fg"> 666</span> use_nullable_dtypes = <span class="ansi-bold" style="color: rgb(0,135,0)">False</span>
<span class="ansi-green-fg"> 667</span> check_dtype_backend(dtype_backend)
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">669</span> <span class="ansi-bold" style="color: rgb(0,135,0)">return</span> <span class="ansi-yellow-bg">impl</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">read</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 670</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 671</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">columns</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 672</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filters</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 673</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 674</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">use_nullable_dtypes</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 675</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">dtype_backend</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 676</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 677</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">*</span><span class="ansi-yellow-bg">kwargs</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 678</span> <span class="ansi-yellow-bg">)</span>
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:267</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-intense-fg ansi-bold"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager <span style="color: rgb(98,98,98)">==</span> <span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">array</span><span style="color: rgb(175,0,0)">"</span>:
<span class="ansi-green-intense-fg ansi-bold"> 265</span> to_pandas_kwargs[<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">split_blocks</span><span style="color: rgb(175,0,0)">"</span>] <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">True</span> <span style="color: rgb(95,135,135)"># type: ignore[assignment]</span>
<span class="ansi-green-fg">--&gt; 267</span> path_or_handle, handles, filesystem <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 268</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 269</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 270</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 271</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">rb</span><span class="ansi-yellow-bg" style="color: rgb(175,0,0)">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-intense-fg ansi-bold"> 272</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 273</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-intense-fg ansi-bold"> 274</span> pa_table <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">self</span><span style="color: rgb(98,98,98)">.</span>api<span style="color: rgb(98,98,98)">.</span>parquet<span style="color: rgb(98,98,98)">.</span>read_table(
<span class="ansi-green-intense-fg ansi-bold"> 275</span> path_or_handle,
<span class="ansi-green-intense-fg ansi-bold"> 276</span> columns<span style="color: rgb(98,98,98)">=</span>columns,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 279</span> <span style="color: rgb(98,98,98)">*</span><span style="color: rgb(98,98,98)">*</span>kwargs,
<span class="ansi-green-intense-fg ansi-bold"> 280</span> )
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:258</span>, in <span class="ansi-cyan-fg">PyArrowImpl.read</span><span class="ansi-blue-fg">(self, path, columns, filters, use_nullable_dtypes, dtype_backend, storage_options, filesystem, **kwargs)</span>
<span class="ansi-green-fg"> 256</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> manager == <span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">array</span><span class="ansi-yellow-fg">"</span>:
<span class="ansi-green-fg"> 257</span> to_pandas_kwargs[<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">split_blocks</span><span class="ansi-yellow-fg">"</span>] = <span class="ansi-bold" style="color: rgb(0,135,0)">True</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">258</span> path_or_handle, handles, filesystem = <span class="ansi-yellow-bg">_get_path_or_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 259</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 260</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">filesystem</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 261</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 262</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-fg ansi-yellow-bg">rb</span><span class="ansi-yellow-fg ansi-yellow-bg">"</span><span class="ansi-yellow-bg">,</span>
<span class="ansi-green-fg"> 263</span> <span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 264</span> <span class="ansi-bold" style="color: rgb(0,135,0)">try</span>:
<span class="ansi-green-fg"> 265</span> pa_table = <span style="color: rgb(0,135,0)">self</span>.api.parquet.read_table(
<span class="ansi-green-fg"> 266</span> path_or_handle,
<span class="ansi-green-fg"> 267</span> columns=columns,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 270</span> **kwargs,
<span class="ansi-green-fg"> 271</span> )
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:140</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-intense-fg ansi-bold"> 130</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 131</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-intense-fg ansi-bold"> 132</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-intense-fg ansi-bold"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 138</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-intense-fg ansi-bold"> 139</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; 140</span> handles <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-intense-fg ansi-bold"> 141</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-intense-fg ansi-bold"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 143</span> fs <span style="color: rgb(98,98,98)">=</span> <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-intense-fg ansi-bold"> 144</span> path_or_handle <span style="color: rgb(98,98,98)">=</span> handles<span style="color: rgb(98,98,98)">.</span>handle
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/parquet.py:141</span>, in <span class="ansi-cyan-fg">_get_path_or_handle</span><span class="ansi-blue-fg">(path, fs, storage_options, mode, is_dir)</span>
<span class="ansi-green-fg"> 131</span> handles = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 132</span> <span class="ansi-bold" style="color: rgb(0,135,0)">if</span> (
<span class="ansi-green-fg"> 133</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> fs
<span class="ansi-green-fg"> 134</span> <span class="ansi-bold" style="color: rgb(175,0,255)">and</span> <span class="ansi-bold" style="color: rgb(175,0,255)">not</span> is_dir
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 139</span> <span style="color: rgb(95,135,135)"># fsspec resources can also point to directories</span>
<span class="ansi-green-fg"> 140</span> <span style="color: rgb(95,135,135)"># this branch is used for example when reading from non-fsspec URLs</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">141</span> handles = <span class="ansi-yellow-bg">get_handle</span><span class="ansi-yellow-bg">(</span>
<span class="ansi-green-fg"> 142</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">path_or_handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">is_text</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg ansi-bold" style="color: rgb(0,135,0)">False</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">storage_options</span><span class="ansi-yellow-bg">=</span><span class="ansi-yellow-bg">storage_options</span>
<span class="ansi-green-fg"> 143</span> <span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 144</span> fs = <span class="ansi-bold" style="color: rgb(0,135,0)">None</span>
<span class="ansi-green-fg"> 145</span> path_or_handle = handles.handle
File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-intense-fg ansi-bold"> 873</span> handle <span style="color: rgb(98,98,98)">=</span> <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-intense-fg ansi-bold"> 874</span> handle,
<span class="ansi-green-intense-fg ansi-bold"> 875</span> ioargs<span style="color: rgb(98,98,98)">.</span>mode,
<span class="ansi-green-fg"> (...)</span>
<span class="ansi-green-intense-fg ansi-bold"> 878</span> newline<span style="color: rgb(98,98,98)">=</span><span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">"</span>,
<span class="ansi-green-intense-fg ansi-bold"> 879</span> )
<span class="ansi-green-intense-fg ansi-bold"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-intense-fg ansi-bold"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; 882</span> handle <span style="color: rgb(98,98,98)">=</span> <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg" style="color: rgb(98,98,98)">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-intense-fg ansi-bold"> 883</span> handles<span style="color: rgb(98,98,98)">.</span>append(handle)
<span class="ansi-green-intense-fg ansi-bold"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-cyan-fg">File </span><span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7-py3.11/lib/python3.11/site-packages/pandas/io/common.py:882</span>, in <span class="ansi-cyan-fg">get_handle</span><span class="ansi-blue-fg">(path_or_buf, mode, encoding, compression, memory_map, is_text, errors, storage_options)</span>
<span class="ansi-green-fg"> 873</span> handle = <span style="color: rgb(0,135,0)">open</span>(
<span class="ansi-green-fg"> 874</span> handle,
<span class="ansi-green-fg"> 875</span> ioargs.mode,
<span class="ansi-green-fg"> (...)</span><span class="ansi-green-fg"> 878</span> newline=<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">"</span>,
<span class="ansi-green-fg"> 879</span> )
<span class="ansi-green-fg"> 880</span> <span class="ansi-bold" style="color: rgb(0,135,0)">else</span>:
<span class="ansi-green-fg"> 881</span> <span style="color: rgb(95,135,135)"># Binary mode</span>
<span class="ansi-green-fg">--&gt; </span><span class="ansi-green-fg">882</span> handle = <span class="ansi-yellow-bg" style="color: rgb(0,135,0)">open</span><span class="ansi-yellow-bg">(</span><span class="ansi-yellow-bg">handle</span><span class="ansi-yellow-bg">,</span><span class="ansi-yellow-bg"> </span><span class="ansi-yellow-bg">ioargs</span><span class="ansi-yellow-bg">.</span><span class="ansi-yellow-bg">mode</span><span class="ansi-yellow-bg">)</span>
<span class="ansi-green-fg"> 883</span> handles.append(handle)
<span class="ansi-green-fg"> 885</span> <span style="color: rgb(95,135,135)"># Convert BytesIO or file objects passed with an encoding</span>
<span class="ansi-red-fg">FileNotFoundError</span>: [Errno 2] No such file or directory: 'inputs/alaska/text_units.parquet'</pre>
</div>
@ -3706,8 +3693,8 @@ File <span class="ansi-green-fg">~/.cache/pypoetry/virtualenvs/graphrag-F2jvqev7
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[14], line 1</span>
<span class="ansi-green-fg">----&gt; 1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">0</span>])
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[14]</span><span class="ansi-green-fg">, line 1</span>
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">1</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">0</span>])
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -3764,9 +3751,9 @@ Cell <span class="ansi-green-fg">In[14], line 1</span>
<pre>
<span class="ansi-red-fg">---------------------------------------------------------------------------</span>
<span class="ansi-red-fg">NameError</span> Traceback (most recent call last)
Cell <span class="ansi-green-fg">In[15], line 2</span>
<span class="ansi-green-intense-fg ansi-bold"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> source_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span style="color: rgb(98,98,98)">0</span>, <span style="color: rgb(98,98,98)">1</span>]:
<span class="ansi-green-fg">----&gt; 2</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span style="color: rgb(98,98,98)">1</span>][<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">sources</span><span style="color: rgb(175,0,0)">"</span>][source_id][<span style="color: rgb(175,0,0)">"</span><span style="color: rgb(175,0,0)">text</span><span style="color: rgb(175,0,0)">"</span>][:<span style="color: rgb(98,98,98)">250</span>])
<span class="ansi-cyan-fg">Cell</span><span class="ansi-cyan-fg"> </span><span class="ansi-green-fg">In[15]</span><span class="ansi-green-fg">, line 2</span>
<span class="ansi-green-fg"> 1</span> <span class="ansi-bold" style="color: rgb(0,135,0)">for</span> source_id <span class="ansi-bold" style="color: rgb(175,0,255)">in</span> [<span class="ansi-green-fg">0</span>, <span class="ansi-green-fg">1</span>]:
<span class="ansi-green-fg">----&gt; </span><span class="ansi-green-fg">2</span> <span style="color: rgb(0,135,0)">print</span>(<span class="ansi-yellow-bg">results</span>[<span class="ansi-green-fg">1</span>][<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">sources</span><span class="ansi-yellow-fg">"</span>][source_id][<span class="ansi-yellow-fg">"</span><span class="ansi-yellow-fg">text</span><span class="ansi-yellow-fg">"</span>][:<span class="ansi-green-fg">250</span>])
<span class="ansi-red-fg">NameError</span>: name 'results' is not defined</pre>
</div>
@ -3840,7 +3827,7 @@ Cell <span class="ansi-green-fg">In[15], line 2</span>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
get_started/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1756,7 +1756,7 @@ Once the pipeline is complete, you should see a new folder called <code>./ragtes
<script id="__config" type="application/json">{"base": "..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../scripts/create_cookie_banner.js"></script>

4
index.html
View File

@ -14,7 +14,7 @@
<link rel="icon" href="assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1829,7 +1829,7 @@ We strongly recommend to fine-tune your prompts following the <a href="prompt_tu
<script id="__config" type="application/json">{"base": ".", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="scripts/create_cookie_banner.js"></script>

4
index/architecture/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1639,7 +1639,7 @@ This allows our indexer to be more resilient to network issues, to act idempoten
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
index/byog/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1893,7 +1893,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
index/default_dataflow/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2259,7 +2259,7 @@ flowchart LR
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
index/inputs/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2355,7 +2355,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
index/methods/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1749,7 +1749,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
index/outputs/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2209,7 +2209,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
index/overview/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1825,7 +1825,7 @@ After you have a config file you can run the pipeline using the CLI or the Pytho
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
prompt_tuning/auto_prompt_tuning/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1872,7 +1872,7 @@ After that, it uses one of the following selection methods to pick a sample to w
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
prompt_tuning/manual_prompt_tuning/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -2128,7 +2128,7 @@ We provide a means for you to do this by allowing you to specify a custom prompt
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
prompt_tuning/overview/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1726,7 +1726,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
query/drift_search/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1779,7 +1779,7 @@ Figure 1. An entire DRIFT search hierarchy highlighting the three core phases of
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
query/global_search/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1796,7 +1796,7 @@ flowchart LR
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
query/local_search/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1787,7 +1787,7 @@ flowchart LR
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
query/multi_index_search/index.html
View File

@ -12,7 +12,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1609,7 +1609,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
query/notebooks/overview/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1638,7 +1638,7 @@
<script id="__config" type="application/json">{"base": "../../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../../scripts/create_cookie_banner.js"></script>

4
query/overview/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1777,7 +1777,7 @@ It is responsible for the following tasks:</p>
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

4
query/question_generation/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1753,7 +1753,7 @@
<script id="__config" type="application/json">{"base": "../..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../../scripts/create_cookie_banner.js"></script>

2
search/search_index.json
View File

File diff suppressed because one or more lines are too long

BIN
sitemap.xml.gz
View File

Binary file not shown.

4
visualization_guide/index.html
View File

@ -16,7 +16,7 @@
<link rel="icon" href="../assets/images/favicon.png">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.14">
<meta name="generator" content="mkdocs-1.6.1, mkdocs-material-9.6.15">
@ -1951,7 +1951,7 @@ After running the indexing pipeline over your data, there will be an output fold
<script id="__config" type="application/json">{"base": "..", "features": ["content.code.copy", "content.code.select", "navigation.footer", "navigation.tabs"], "search": "../assets/javascripts/workers/search.d50fe291.min.js", "tags": null, "translations": {"clipboard.copied": "Copied to clipboard", "clipboard.copy": "Copy to clipboard", "search.result.more.one": "1 more on this page", "search.result.more.other": "# more on this page", "search.result.none": "No matching documents", "search.result.one": "1 matching document", "search.result.other": "# matching documents", "search.result.placeholder": "Type to start searching", "search.result.term.missing": "Missing", "select.version": "Select version"}, "version": null}</script>
<script src="../assets/javascripts/bundle.13a4f30d.min.js"></script>
<script src="../assets/javascripts/bundle.56ea9cef.min.js"></script>
<script src="../scripts/create_cookie_banner.js"></script>