소스 검색

Rename graphrag task lock (#6576)

### What problem does this PR solve?

Rename graphrag task lock

### Type of change

- [x] Refactoring
tags/v0.18.0
Zhichang Yu 7 달 전
부모
커밋
c4998d0e09
No account linked to committer's email address
1개의 변경된 파일5개의 추가작업 그리고 3개의 파일을 삭제
  1. 5
    3
      graphrag/general/index.py

+ 5
- 3
graphrag/general/index.py 파일 보기

@@ -190,7 +190,7 @@ async def merge_subgraph(
embedding_model,
callback,
):
graphrag_task_lock = RedisDistributedLock("graphrag_task", lock_value=doc_id, timeout=600)
graphrag_task_lock = RedisDistributedLock(f"graphrag_task_{kb_id}", lock_value=doc_id, timeout=600)
while True:
if graphrag_task_lock.acquire():
break
@@ -230,10 +230,11 @@ async def resolve_entities(
embed_bdl,
callback,
):
graphrag_task_lock = RedisDistributedLock("graphrag_task", lock_value=doc_id, timeout=600)
graphrag_task_lock = RedisDistributedLock(f"graphrag_task_{kb_id}", lock_value=doc_id, timeout=600)
while True:
if graphrag_task_lock.acquire():
break
callback(msg=f"resolve_entities {doc_id} is waiting graphrag_task_lock")
await trio.sleep(10)

start = trio.current_time()
@@ -261,10 +262,11 @@ async def extract_community(
embed_bdl,
callback,
):
graphrag_task_lock = RedisDistributedLock("graphrag_task", lock_value=doc_id, timeout=600)
graphrag_task_lock = RedisDistributedLock(f"graphrag_task_{kb_id}", lock_value=doc_id, timeout=600)
while True:
if graphrag_task_lock.acquire():
break
callback(msg=f"extract_community {doc_id} is waiting graphrag_task_lock")
await trio.sleep(10)

start = trio.current_time()

Loading…
취소
저장