Fix pyright errors by removing WorkflowCallbacks from strategy type signatures

Co-authored-by: jgbradley1 <654554+jgbradley1@users.noreply.github.com>

Author: Copilot

graphrag/index/operations/extract_graph/typing.py

@@ -11,7 +11,6 @@
 import networkx as nx
 
 from graphrag.cache.pipeline_cache import PipelineCache
-from graphrag.callbacks.workflow_callbacks import WorkflowCallbacks
 
 ExtractedEntity = dict[str, Any]
 ExtractedRelationship = dict[str, Any]
@@ -40,7 +39,6 @@ class EntityExtractionResult:
     [
         list[Document],
         EntityTypes,
-        WorkflowCallbacks,
         PipelineCache,
         StrategyConfig,
     ],

graphrag/index/operations/summarize_descriptions/typing.py

@@ -9,7 +9,6 @@
 from typing import Any, NamedTuple
 
 from graphrag.cache.pipeline_cache import PipelineCache
-from graphrag.callbacks.workflow_callbacks import WorkflowCallbacks
 
 StrategyConfig = dict[str, Any]
 
@@ -26,7 +25,6 @@ class SummarizedDescriptionResult:
     [
         str | tuple[str, str],
         list[str],
-        WorkflowCallbacks,
         PipelineCache,
         StrategyConfig,
     ],

tests/unit/indexing/verbs/entities/extraction/strategies/graph_intelligence/test_gi_entity_extraction.py

@@ -14,13 +14,6 @@
 class TestRunChain(unittest.IsolatedAsyncioTestCase):
     async def test_run_extract_graph_single_document_correct_entities_returned(self):
         results = await run_extract_graph(
-            docs=[Document("test_text", "1")],
-            entity_types=["person"],
-            callbacks=None,
-            args={
-                "max_gleanings": 0,
-                "summarize_descriptions": False,
-            },
             model=create_mock_llm(
                 responses=[
                     """
@@ -37,6 +30,12 @@ async def test_run_extract_graph_single_document_correct_entities_returned(self)
                 ],
                 name="test_run_extract_graph_single_document_correct_entities_returned",
             ),
+            docs=[Document("test_text", "1")],
+            entity_types=["person"],
+            args={
+                "max_gleanings": 0,
+                "summarize_descriptions": False,
+            },
         )
 
         # self.assertItemsEqual isn't available yet, or I am just silly
@@ -49,13 +48,6 @@ async def test_run_extract_graph_multiple_documents_correct_entities_returned(
         self,
     ):
         results = await run_extract_graph(
-            docs=[Document("text_1", "1"), Document("text_2", "2")],
-            entity_types=["person"],
-            callbacks=None,
-            args={
-                "max_gleanings": 0,
-                "summarize_descriptions": False,
-            },
             model=create_mock_llm(
                 responses=[
                     """
@@ -76,6 +68,12 @@ async def test_run_extract_graph_multiple_documents_correct_entities_returned(
                 ],
                 name="test_run_extract_graph_multiple_documents_correct_entities_returned",
             ),
+            docs=[Document("text_1", "1"), Document("text_2", "2")],
+            entity_types=["person"],
+            args={
+                "max_gleanings": 0,
+                "summarize_descriptions": False,
+            },
         )
 
         # self.assertItemsEqual isn't available yet, or I am just silly
@@ -86,13 +84,6 @@ async def test_run_extract_graph_multiple_documents_correct_entities_returned(
 
     async def test_run_extract_graph_multiple_documents_correct_edges_returned(self):
         results = await run_extract_graph(
-            docs=[Document("text_1", "1"), Document("text_2", "2")],
-            entity_types=["person"],
-            callbacks=None,
-            args={
-                "max_gleanings": 0,
-                "summarize_descriptions": False,
-            },
             model=create_mock_llm(
                 responses=[
                     """
@@ -113,6 +104,12 @@ async def test_run_extract_graph_multiple_documents_correct_edges_returned(self)
                 ],
                 name="test_run_extract_graph_multiple_documents_correct_edges_returned",
             ),
+            docs=[Document("text_1", "1"), Document("text_2", "2")],
+            entity_types=["person"],
+            args={
+                "max_gleanings": 0,
+                "summarize_descriptions": False,
+            },
         )
 
         # self.assertItemsEqual isn't available yet, or I am just silly
@@ -131,13 +128,6 @@ async def test_run_extract_graph_multiple_documents_correct_entity_source_ids_ma
         self,
     ):
         results = await run_extract_graph(
-            docs=[Document("text_1", "1"), Document("text_2", "2")],
-            entity_types=["person"],
-            callbacks=None,
-            args={
-                "max_gleanings": 0,
-                "summarize_descriptions": False,
-            },
             model=create_mock_llm(
                 responses=[
                     """
@@ -158,6 +148,12 @@ async def test_run_extract_graph_multiple_documents_correct_entity_source_ids_ma
                 ],
                 name="test_run_extract_graph_multiple_documents_correct_entity_source_ids_mapped",
             ),
+            docs=[Document("text_1", "1"), Document("text_2", "2")],
+            entity_types=["person"],
+            args={
+                "max_gleanings": 0,
+                "summarize_descriptions": False,
+            },
         )
 
         graph = results.graph
@@ -181,13 +177,6 @@ async def test_run_extract_graph_multiple_documents_correct_edge_source_ids_mapp
         self,
     ):
         results = await run_extract_graph(
-            docs=[Document("text_1", "1"), Document("text_2", "2")],
-            entity_types=["person"],
-            callbacks=None,
-            args={
-                "max_gleanings": 0,
-                "summarize_descriptions": False,
-            },
             model=create_mock_llm(
                 responses=[
                     """
@@ -208,6 +197,12 @@ async def test_run_extract_graph_multiple_documents_correct_edge_source_ids_mapp
                 ],
                 name="test_run_extract_graph_multiple_documents_correct_edge_source_ids_mapped",
             ),
+            docs=[Document("text_1", "1"), Document("text_2", "2")],
+            entity_types=["person"],
+            args={
+                "max_gleanings": 0,
+                "summarize_descriptions": False,
+            },
         )
 
         graph = results.graph