mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-07-13 00:50:06 +08:00
Yannic Kilcher
64 lines
1.4 KiB
Python
64 lines
1.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
from typing import Literal, Optional
|
|
from uuid import UUID
|
|
|
|
import pydantic
|
|
from pydantic import BaseModel
|
|
|
|
|
|
class TaskRequest(BaseModel):
|
|
"""The frontend asks the backend for a task."""
|
|
|
|
type: Literal
|
|
user_id: Optional[str] = None
|
|
|
|
|
|
class GenericTaskRequest(TaskRequest):
|
|
type: Literal["generic"] = "generic"
|
|
|
|
|
|
class Task(BaseModel):
|
|
"""A task is a unit of work that the backend gives to the frontend."""
|
|
|
|
id: UUID = pydantic.Field(default_factory=UUID)
|
|
type: Literal
|
|
addressed_users: Optional[list[str]] = None
|
|
|
|
|
|
class TaskResponse(BaseModel):
|
|
"""A task response is a message from the frontend to acknowledge the given task."""
|
|
|
|
type: Literal
|
|
status: Literal["success", "failure"]
|
|
|
|
|
|
class PostCreatedTaskResponse(TaskResponse):
|
|
type: Literal["post_created"] = "post_created"
|
|
post_id: UUID
|
|
|
|
|
|
class SummarizeStoryTask(Task):
|
|
type: Literal["summarize_story"] = "summarize_story"
|
|
story: str
|
|
|
|
|
|
class TaskDone(Task):
|
|
type: Literal["task_done"] = "task_done"
|
|
reply_to_post_id: UUID
|
|
|
|
|
|
class Interaction(BaseModel):
|
|
"""An interaction is a message from the frontend to the backend."""
|
|
|
|
type: Literal
|
|
user_id: str
|
|
|
|
|
|
class TextReplyToPost(Interaction):
|
|
"""A user has replied to a post with text."""
|
|
|
|
type: Literal["text_reply_to_post"] = "text_reply_to_post"
|
|
post_id: UUID
|
|
user_post_id: UUID
|
|
text: str
|