Quellcode durchsuchen

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 vor 2 Monaten
Ursprung
Commit
30e9212db9
Es ist kein Account mit der E-Mail-Adresse des Committers verbunden
2 geänderte Dateien mit 2 neuen und 2 gelöschten Zeilen
  1. 1
    1
      api/utils/api_utils.py
  2. 1
    1
      graphrag/general/extractor.py

+ 1
- 1
api/utils/api_utils.py Datei anzeigen





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

+ 1
- 1
graphrag/general/extractor.py Datei anzeigen

self._language = language self._language = language
self._entity_types = entity_types or DEFAULT_ENTITY_TYPES self._entity_types = entity_types or DEFAULT_ENTITY_TYPES


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

Laden…
Abbrechen
Speichern