from typing import Protocol


class AIEngineAdapter(Protocol):
    def run_structured_task(self, payload: dict) -> dict: ...

    def stream_chat(self, payload: dict): ...

    def create_knowledge_base(self, payload: dict) -> dict: ...

    def upload_knowledge_document(self, payload: dict) -> dict: ...

    def get_knowledge_document_status(self, payload: dict) -> dict: ...