Spaces:
Sleeping
Sleeping
File size: 2,022 Bytes
1b7e88c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 |
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
|