File tree Expand file tree Collapse file tree 3 files changed +2
-4
lines changed
index/operations/summarize_communities Expand file tree Collapse file tree 3 files changed +2
-4
lines changed Original file line number Diff line number Diff line change 4949from graphrag .vector_stores .base import BaseVectorStore
5050from graphrag .vector_stores .factory import VectorStoreFactory
5151
52- if TYPE_CHECKING :
53- from graphrag .query .structured_search .base import SearchResult
54-
5552if TYPE_CHECKING :
5653 from graphrag .query .structured_search .base import SearchResult
5754
Original file line number Diff line number Diff line change @@ -45,6 +45,7 @@ async def summarize_communities(
4545 reports : list [CommunityReport | None ] = []
4646 tick = progress_ticker (callbacks .progress , len (local_contexts ))
4747 runner = load_strategy (strategy ["type" ])
48+
4849 for level in levels :
4950 level_contexts = prep_community_report_context (
5051 pd .DataFrame (reports ),
Original file line number Diff line number Diff line change @@ -27,7 +27,7 @@ class StorageFactory:
2727
2828 @classmethod
2929 def register (cls , storage_type : str , storage : type ):
30- """Register a vector store type ."""
30+ """Register a custom storage implementation ."""
3131 cls .storage_types [storage_type ] = storage
3232
3333 @classmethod
You can’t perform that action at this time.
0 commit comments