mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-01 16:50:12 +08:00
fix: not timeout for sikpped tasks
This commit is contained in:
@@ -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()
|
||||
|
||||
@@ -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}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user