fix
All checks were successful
Deploy Prod / Build (pull_request) Successful in 15s
Deploy Prod / Push (pull_request) Successful in 7s
Deploy Prod / Deploy prod (pull_request) Successful in 27s

This commit is contained in:
Egor Matveev 2024-12-28 15:33:18 +03:00
parent 789a3298e5
commit 4236d3a7cf
2 changed files with 5 additions and 5 deletions

View File

@ -8,6 +8,6 @@ RUN pip install -r requirements.txt
COPY . .
ENV PYTHONUNBUFFERED 1
ENV PYTHONUNBUFFERED=1
ENTRYPOINT ["python", "main.py"]

View File

@ -23,9 +23,9 @@ class Response(pydantic.BaseModel):
async def execute(queue: typing.Annotated[str, fastapi.Header()]) -> Response:
try:
task = await tasks.take_task(queue)
except Exception as e:
print('GOT ERROR', e)
return Response(task=None)
if not task:
return Response(task=None)
return Response(task=Task(id=str(task._id), attempt=task.attempts, payload=task.payload))
except Exception as e:
print("GOT ERROR", e)
return Response(task=None)