diff --git a/backend/app/prompt_repository.py b/backend/app/prompt_repository.py index 9be665d7..54faed2b 100644 --- a/backend/app/prompt_repository.py +++ b/backend/app/prompt_repository.py @@ -202,7 +202,7 @@ class PromptRepository: return reaction def store_task(self, task: protocol_schema.Task) -> WorkPackage: - payload: TaskPayload = None + payload: TaskPayload match type(task): case protocol_schema.SummarizeStoryTask: payload = SummarizationStoryPayload(story=task.story)