Просмотр исходного кода

Fix: restore task limiter. (#7844)

### What problem does this PR solve?

Close #7828

### Type of change

- [x] Bug Fix (non-breaking change which fixes an issue)
tags/v0.19.1
Kevin Hu 5 месяцев назад
Родитель
Сommit
be83074131
Аккаунт пользователя с таким Email не найден
1 измененных файлов: 1 добавлений и 0 удалений
  1. 1
    0
      rag/svr/task_executor.py

+ 1
- 0
rag/svr/task_executor.py Просмотреть файл

with_community = graphrag_conf.get("community", False) with_community = graphrag_conf.get("community", False)
await run_graphrag(task, task_language, with_resolution, with_community, chat_model, embedding_model, progress_callback) await run_graphrag(task, task_language, with_resolution, with_community, chat_model, embedding_model, progress_callback)
progress_callback(prog=1.0, msg="Knowledge Graph done ({:.2f}s)".format(timer() - start_ts)) progress_callback(prog=1.0, msg="Knowledge Graph done ({:.2f}s)".format(timer() - start_ts))
task_limiter = trio.CapacityLimiter(MAX_CONCURRENT_TASKS)
return return
else: else:
# Standard chunking methods # Standard chunking methods

Загрузка…
Отмена
Сохранить