ishantvivek/codegen

1from abc import ABC, abstractmethod2 3 4class ModelVisitor(ABC):5    """6    Abstract base class for model visitors.7 8    This class defines the interface for visiting a model generator. 9    Subclasses must implement the visit method to define 10    specific behaviors for different types of model generators.11 12    Methods:13        visit(generator, *args, **kwargs):14            Abstract method to visit the model generator. Subclasses 15            must override this method to provide specific functionality.16 17    Example:18        class IbmTextGenerator(ModelVisitor):19            def visit(self, model_generator, *args, **kwargs):20                # Implement specific behavior here21                pass22    """23 24    @abstractmethod25    def visit(self, generator, *args, **kwargs):26        pass27