From af3ec08f71076d7ba8f6ce309207dca9a552682e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Andreas=20K=C3=B6pf?= Date: Fri, 3 Feb 2023 09:43:19 +0000 Subject: [PATCH] fix: not timeout for sikpped tasks --- backend/oasst_backend/task_repository.py | 4 +++- backend/oasst_backend/tree_manager.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/backend/oasst_backend/task_repository.py b/backend/oasst_backend/task_repository.py index b721caf5..5fe84b24 100644 --- a/backend/oasst_backend/task_repository.py +++ b/backend/oasst_backend/task_repository.py @@ -205,7 +205,7 @@ class TaskRepository: return task def fetch_recent_reply_tasks( - self, max_age: timedelta = timedelta(minutes=5), done: bool = False, limit: int = 100 + self, max_age: timedelta = timedelta(minutes=5), done: bool = False, skipped: bool = False, limit: int = 100 ) -> list[Task]: qry = self.db.query(Task).filter( func.age(Task.created_date) < max_age, @@ -213,6 +213,8 @@ class TaskRepository: ) if done is not None: qry = qry.filter(Task.done == done) + if skipped is not None: + qry = qry.filter(Task.skipped == skipped) if limit: qry = qry.limit(limit) return qry.all() diff --git a/backend/oasst_backend/tree_manager.py b/backend/oasst_backend/tree_manager.py index e09f22d1..44560e15 100644 --- a/backend/oasst_backend/tree_manager.py +++ b/backend/oasst_backend/tree_manager.py @@ -496,7 +496,7 @@ class TreeManager: case TaskType.REPLY: recent_reply_tasks = self.pr.task_repository.fetch_recent_reply_tasks( - max_age=timedelta(seconds=self.cfg.recent_tasks_span_sec), done=False + max_age=timedelta(seconds=self.cfg.recent_tasks_span_sec), done=False, skipped=False, limit=500 ) recent_reply_task_parents = {t.parent_message_id for t in recent_reply_tasks}