GraphRAG
GraphRAG copied to clipboard
use GGraphRAG.yml and query failed
- code version :commit d5a2653830c0d5360de2bdb414c2985dfbedac4a
- bug log :
File "/home/oliver/notebook/rag/others/GraphRAG/test.py", line 41, in <module> asyncio.run(main()) File "/home/oliver/.conda/envs/digimon/lib/python3.10/asyncio/runners.py", line 44, in run return loop.run_until_complete(main) File "/home/oliver/.conda/envs/digimon/lib/python3.10/asyncio/base_events.py", line 649, in run_until_complete return future.result() File "/home/oliver/notebook/rag/others/GraphRAG/test.py", line 35, in main response = await digimon.query(query) File "/home/oliver/notebook/rag/others/GraphRAG/Core/GraphRAG.py", line 292, in query response = await self._querier.query(query) File "/home/oliver/notebook/rag/others/GraphRAG/Core/Query/BaseQuery.py", line 22, in query context = await self._retrieve_relevant_contexts(query=query) File "/home/oliver/notebook/rag/others/GraphRAG/Core/Query/BasicQuery.py", line 21, in _retrieve_relevant_contexts return await self._retrieve_relevant_contexts_global(query) File "/home/oliver/notebook/rag/others/GraphRAG/Core/Query/BasicQuery.py", line 128, in _retrieve_relevant_contexts_global community_datas = await self._retriever.retrieve_relevant_content(type=Retriever.COMMUNITY, mode="from_level") File "/home/oliver/notebook/rag/others/GraphRAG/Core/Retriever/MixRetriever.py", line 19, in retrieve_relevant_content return await self.retrievers[type].retrieve_relevant_content(mode=mode, **kwargs) File "/home/oliver/notebook/rag/others/GraphRAG/Core/Retriever/BaseRetriever.py", line 25, in retrieve_relevant_content return await retrieve_fun(self, **kwargs) File "/home/oliver/notebook/rag/others/GraphRAG/Core/Retriever/CommunityRetriever.py", line 71, in find_relevant_community_by_level community_schema = { File "/home/oliver/notebook/rag/others/GraphRAG/Core/Retriever/CommunityRetriever.py", line 72, in <dictcomp> k: v for k, v in community_schema.items() if v.level <= self.config.level File "/home/oliver/.conda/envs/digimon/lib/python3.10/site-packages/pydantic/main.py", line 984, in __getattr__ raise AttributeError(f'{type(self).__name__!r} object has no attribute {item!r}') AttributeError: 'RetrieverConfig' object has no attribute 'level' -
the error came from codes below
@register_retriever_method(type="community", method_name="from_level") async def find_relevant_community_by_level(self, seed=None): community_schema = self.community.community_schemalen(community_schema) is 0 Why does that happen ? - GGraphRAG.yml
- Config2.yml
llm:
api_type: "open_llm" # or openai
base_url: 'http://2.ndsl:11434/v1'
model: "myqwen2.5"
api_key: None
embedding:
api_type: "ollama" # or ollama / openai.
# base_url: "https://cfcus02.opapi.win/v1" # or forward url / other llm url
base_url: 'http://2.ndsl:11434/v1'
api_key: None
model: "nomic-embed-text"
cache_dir: ""
dimensions: 1024
max_token_size: 8102
embed_batch_size: 128
embedding_func_max_async: 16
data_root: "./test/datasets" # Root directory for data
working_dir: "./exp" # Result directory for the experiment
exp_name: "test1" # Experiment name
#