Skip to content

docs: update query notebooks to current model factories - #2350

Open
qianzhu / Lucien (qianzhu18) wants to merge 1 commit into
microsoft:mainfrom
qianzhu18:fix/search-notebook-model-config
Open

docs: update query notebooks to current model factories#2350
qianzhu / Lucien (qianzhu18) wants to merge 1 commit into
microsoft:mainfrom
qianzhu18:fix/search-notebook-model-config

Conversation

@qianzhu18

Copy link
Copy Markdown

Summary

  • update the query notebooks to use the current graphrag_llm model factories
  • replace stale ModelType, LanguageModelConfig, and ModelManager imports with ModelConfig, RetryConfig, create_completion, and create_embedding
  • keep the existing notebook search flows intact while aligning them with the package layout on main

Why

Issue #2301 reports that the example notebooks no longer match the current code structure. On main, graphrag.config.enums no longer exports ModelType, and the notebooks still point at the older model manager entry points.

This update refreshes the affected query notebooks so their model setup matches the current API surface shipped in the repository today.

Scope

  • docs/examples_notebooks/local_search.ipynb
  • docs/examples_notebooks/global_search.ipynb
  • docs/examples_notebooks/global_search_with_dynamic_community_selection.ipynb
  • docs/examples_notebooks/drift_search.ipynb

Closes #2301

@qianzhu18
qianzhu / Lucien (qianzhu18) requested a review from a team as a code owner May 11, 2026 13:33
@qianzhu18

Copy link
Copy Markdown
Author

@microsoft-github-policy-service agree

@qianzhu18

Copy link
Copy Markdown
Author

Hi, just a friendly follow-up on this PR. It updates the official query notebooks from deprecated model setup paths (ModelType / LanguageModelConfig) to the current graphrag_llm model factory APIs, so new users following the examples won't hit import errors on the latest main branch. Happy to adjust if anything needs changing.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

[Bug]: Example notebooks code is incorrect (not updated)

1 participant