feat: add task deletion endpoint

This commit is contained in:
kevin.zhang
2024-04-10 16:14:50 +08:00
parent a880b2b45b
commit d57434e0d3
3 changed files with 47 additions and 1 deletions

View File

@@ -38,6 +38,10 @@ class MemoryState(BaseState):
def get_task(self, task_id: str):
return self._tasks.get(task_id, None)
def delete_task(self, task_id: str):
if task_id in self._tasks:
del self._tasks[task_id]
# Redis state management
class RedisState(BaseState):
@@ -67,6 +71,9 @@ class RedisState(BaseState):
task = {key.decode('utf-8'): self._convert_to_original_type(value) for key, value in task_data.items()}
return task
def delete_task(self, task_id: str):
self._redis.delete(task_id)
@staticmethod
def _convert_to_original_type(value):
"""