ZhAgent/apps/models_provider/impl/base_tti.py

# coding=utf-8
from abc import abstractmethod

from pydantic import BaseModel


class BaseTextToImage(BaseModel):
    @abstractmethod
    def check_auth(self):
        pass

    @abstractmethod
    def generate_image(self, prompt: str, negative_prompt: str = None):
        pass