diff --git a/api/utils/api_utils.py b/api/utils/api_utils.py index 0bb0931cb..48be1ecae 100644 --- a/api/utils/api_utils.py +++ b/api/utils/api_utils.py @@ -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: diff --git a/graphrag/general/extractor.py b/graphrag/general/extractor.py index a6ceb1d1c..a49074c90 100644 --- a/graphrag/general/extractor.py +++ b/graphrag/general/extractor.py @@ -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)