Procházet zdrojové kódy

add retry count to task (#2152)

### What problem does this PR solve?


### Type of change

- [x] New Feature (non-breaking change which adds functionality)
tags/v0.11.0
Kevin Hu před 1 rokem
rodič
revize
212bb8e601
Žádný účet není propojen s e-mailovou adresou tvůrce revize
2 změnil soubory, kde provedl 21 přidání a 2 odebrání
  1. 8
    0
      api/db/db_models.py
  2. 13
    2
      api/db/services/task_service.py

+ 8
- 0
api/db/db_models.py Zobrazit soubor

null=True, null=True,
help_text="process message", help_text="process message",
default="") default="")
retry_count = IntegerField(default=0)




class Dialog(DataBaseModel): class Dialog(DataBaseModel):
DB.execute_sql('ALTER TABLE llm ADD PRIMARY KEY (llm_name,fid);') DB.execute_sql('ALTER TABLE llm ADD PRIMARY KEY (llm_name,fid);')
except Exception as e: except Exception as e:
pass pass
try:
migrate(
migrator.add_column('task', 'retry_count', IntegerField(default=0))
)
except Exception as e:
pass


+ 13
- 2
api/db/services/task_service.py Zobrazit soubor

docs = list(docs.dicts()) docs = list(docs.dicts())
if not docs: return [] if not docs: return []


cls.model.update(progress_msg=cls.model.progress_msg + "\n" + "Task has been received.",
progress=random.random() / 10.).where(
msg = "\nTask has been received."
prog = random.random() / 10.
if docs[0]["retry_count"] >= 3:
msg = "\nERROR: Task is abandoned after 3 times attempts."
prog = -1

cls.model.update(progress_msg=cls.model.progress_msg + msg,
progress=prog,
retry_count=docs[0]["retry_count"]+1
).where(
cls.model.id == docs[0]["id"]).execute() cls.model.id == docs[0]["id"]).execute()

if docs[0]["retry_count"] >= 3: return []

return docs return docs


@classmethod @classmethod

Načítá se…
Zrušit
Uložit