| from abc import ABC, abstractmethod |
| from typing import List, Optional |
|
|
| from omagent_core.engine.http.models import PollData |
| from omagent_core.engine.http.models.task import Task |
| from omagent_core.engine.http.models.task_exec_log import TaskExecLog |
| from omagent_core.engine.http.models.task_result import TaskResult |
| from omagent_core.engine.http.models.task_result_status import TaskResultStatus |
| from omagent_core.engine.http.models.workflow import Workflow |
|
|
|
|
| class TaskClient(ABC): |
| @abstractmethod |
| def poll_task( |
| self, |
| task_type: str, |
| worker_id: Optional[str] = None, |
| domain: Optional[str] = None, |
| ) -> Optional[Task]: |
| pass |
|
|
| @abstractmethod |
| def batch_poll_tasks( |
| self, |
| task_type: str, |
| worker_id: Optional[str] = None, |
| count: Optional[int] = None, |
| timeout_in_millisecond: Optional[int] = None, |
| domain: Optional[str] = None, |
| ) -> List[Task]: |
| pass |
|
|
| @abstractmethod |
| def get_task(self, task_id: str) -> Task: |
| pass |
|
|
| @abstractmethod |
| def update_task(self, task_result: TaskResult) -> str: |
| pass |
|
|
| @abstractmethod |
| def update_task_by_ref_name( |
| self, |
| workflow_id: str, |
| task_ref_name: str, |
| status: TaskResultStatus, |
| output: object, |
| worker_id: Optional[str] = None, |
| ) -> str: |
| pass |
|
|
| @abstractmethod |
| def update_task_sync( |
| self, |
| workflow_id: str, |
| task_ref_name: str, |
| status: TaskResultStatus, |
| output: object, |
| worker_id: Optional[str] = None, |
| ) -> Workflow: |
| pass |
|
|
| @abstractmethod |
| def get_queue_size_for_task(self, task_type: str) -> int: |
| pass |
|
|
| @abstractmethod |
| def add_task_log(self, task_id: str, log_message: str): |
| pass |
|
|
| @abstractmethod |
| def get_task_logs(self, task_id: str) -> List[TaskExecLog]: |
| pass |
|
|
| @abstractmethod |
| def get_task_poll_data(self, task_type: str) -> List[PollData]: |
| pass |
|
|
