from typing import Protocol


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

    def stream_legal_answer(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: ...