-
Notifications
You must be signed in to change notification settings - Fork 3.2k
Description
Do you need to file an issue?
- I have searched the existing issues and this bug is not already filed.
- My model is hosted on OpenAI or Azure. If not, please look at the "model providers" issue and don't file a new one here.
- I believe this is a legitimate bug, not just a question. If this is a question, please use the Discussions area.
Describe the bug
I have encountered a problem. If there is no entity that can be recognized after the document is cut, subsequent errors will be generated. How can I solve this problem?
File "/Users/PycharmProjects/graphrag/graphrag/index/run/run_pipeline.py", line 143, in _run_pipeline
result = await workflow_function(config, context)
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/PycharmProjects/graphrag/graphrag/index/workflows/extract_graph.py", line 46, in run_workflow
entities, relationships = await extract_graph(
^^^^^^^^^^^^^^^^^^^^
File "/Users/PycharmProjects/graphrag/graphrag/index/workflows/extract_graph.py", line 82, in extract_graph
extracted_entities, extracted_relationships = await extractor(
^^^^^^^^^^^^^^^^
File "/Users/PycharmProjects/graphrag/graphrag/index/operations/extract_graph/extract_graph.py", line 151, in extract_graph
entities = _merge_entities(entity_dfs)
^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/Users/PycharmProjects/graphrag/graphrag/index/operations/extract_graph/extract_graph.py", line 179, in _merge_entities
all_entities.groupby(["title", "type"], sort=False)
File "/opt/anaconda3/envs/graphrag/lib/python3.11/site-packages/pandas/core/frame.py", line 9183, in groupby
return DataFrameGroupBy(
^^^^^^^^^^^^^^^^^
File "/opt/anaconda3/envs/graphrag/lib/python3.11/site-packages/pandas/core/groupby/groupby.py", line 1329, in init
grouper, exclusions, obj = get_grouper(
^^^^^^^^^^^^
File "/opt/anaconda3/envs/graphrag/lib/python3.11/site-packages/pandas/core/groupby/grouper.py", line 1043, in get_grouper
raise KeyError(gpr)
KeyError: 'title'
Steps to reproduce
No response
Expected Behavior
No response
GraphRAG Config Used
# Paste your config here
Logs and screenshots
No response
Additional Information
- GraphRAG Version:
- Operating System:
- Python Version:
- Related Issues: