osmarks/autobotrobot
1
0
Fork 0
mirror of https://github.com/osmarks/autobotrobot synced 2024-12-22 06:20:28 +00:00
Code Issues Releases Wiki Activity

fix failovers

Browse Source
This commit is contained in:
osmarks 2024-10-17 21:18:32 +01:00
parent a311f30602
commit dd3776ea1e
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/util.py
View File

@ -386,6 +386,8 @@ async def generate(sess: aiohttp.ClientSession, prompt, stop=["\n"]):
try:
result = await generate_raw(sess, backend, prompt, stop)
backend_successes.labels(backend["url"]).inc()
failure_stats = last_failures[backend["url"]]
failure_stats.consecutive_failures = 0
return result
except Exception as e:
backend_failures.labels(backend["url"]).inc()