-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Add GraphRAG LLM package. #2174
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
base: v3/main
Are you sure you want to change the base?
Conversation
a3dd470 to
ab68b9f
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Pull request overview
This PR introduces the GraphRAG LLM package, which consolidates and refactors the language model infrastructure. The changes replace the existing graphrag.language_model module with a new graphrag-llm package, standardizing the interface for completion and embedding models across the codebase.
Changes:
- Removed the existing language model implementation and replaced it with the new
graphrag-llmpackage - Updated configuration models to use
completion_model_idandembedding_model_idinstead of genericmodel_id - Refactored all workflows and operations to use the new LLM interfaces (
LLMCompletionandLLMEmbedding)
Reviewed changes
Copilot reviewed 257 out of 260 changed files in this pull request and generated 4 comments.
Show a summary per file
| File | Description |
|---|---|
| tests/verbs/util.py | Removed deprecated model configuration constants |
| tests/verbs/test_*.py | Updated test files to use new configuration utility functions |
| tests/unit/config/utils.py | Added new model config assertion helpers and updated default configs |
| tests/unit/config/test_config.py | Removed validation tests that are now handled by the LLM package |
| packages/graphrag/graphrag/config/models/*_config.py | Updated field names from model_id to completion_model_id/embedding_model_id |
| packages/graphrag/graphrag/index/workflows/*.py | Refactored to use new LLM creation functions and interfaces |
| packages/graphrag/graphrag/query/**/*.py | Updated query components to use new LLM interfaces |
| packages/graphrag/graphrag/language_model/** | Removed old language model implementation |
💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.
Add GraphRAG LLM package.