28 lines
1 KiB
Python
28 lines
1 KiB
Python
from __future__ import annotations
|
|
|
|
from asyncio import Queue
|
|
from typing import Any, Protocol
|
|
|
|
from api.repositories.task_store import TaskRecord
|
|
|
|
|
|
class TaskServiceProtocol(Protocol):
|
|
async def submit_task(self, task: str, timeout: int, metadata: dict | None) -> TaskRecord: ...
|
|
|
|
async def get_task(self, task_id: str) -> TaskRecord | None: ...
|
|
|
|
async def create_run(self, thread_id: str, user_input: str, timeout: int, metadata: dict | None) -> TaskRecord: ...
|
|
|
|
async def get_run(self, run_id: str) -> TaskRecord | None: ...
|
|
|
|
async def list_thread_runs(self, thread_id: str) -> list[TaskRecord]: ...
|
|
|
|
async def cancel_run(self, run_id: str) -> TaskRecord | None: ...
|
|
|
|
async def delete_run(self, run_id: str) -> tuple[bool, bool]: ...
|
|
|
|
async def wait_run(self, run_id: str, timeout: float | None = None) -> TaskRecord | None: ...
|
|
|
|
async def subscribe_run_stream(self, run_id: str) -> Queue[dict[str, Any]] | None: ...
|
|
|
|
async def unsubscribe_run_stream(self, run_id: str, queue: Queue[dict[str, Any]]) -> None: ...
|