Browse Source

Fix: enlarge the timeout limits. (#9201)

### What problem does this PR solve?

#9189

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.20.1
Kevin Hu 2 months ago
parent
commit
30e9212db9
No account linked to committer's email address
2 changed files with 2 additions and 2 deletions
  1. 1
    1
      api/utils/api_utils.py
  2. 1
    1
      graphrag/general/extractor.py

+ 1
- 1
api/utils/api_utils.py View File

@@ -687,7 +687,7 @@ def timeout(seconds: float | int = None, attempts: int = 2, *, exception: Option


async def is_strong_enough(chat_model, embedding_model):
@timeout(30, 2)
@timeout(60, 2)
async def _is_strong_enough():
nonlocal chat_model, embedding_model
if embedding_model:

+ 1
- 1
graphrag/general/extractor.py View File

@@ -47,7 +47,7 @@ class Extractor:
self._language = language
self._entity_types = entity_types or DEFAULT_ENTITY_TYPES

@timeout(60*3)
@timeout(60*5)
def _chat(self, system, history, gen_conf={}):
hist = deepcopy(history)
conf = deepcopy(gen_conf)

Loading…
Cancel
Save