mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-02-14 14:25:40 -05:00
26 lines
848 B
Python
26 lines
848 B
Python
from abc import ABC, abstractmethod
|
|
|
|
|
|
class ModelRelationshipsServiceABC(ABC):
|
|
"""High-level service for managing model-to-model relationships."""
|
|
|
|
@abstractmethod
|
|
def add_model_relationship(self, model_key_1: str, model_key_2: str) -> None:
|
|
"""Creates a relationship between two models keys."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def remove_model_relationship(self, model_key_1: str, model_key_2: str) -> None:
|
|
"""Removes a relationship between two models keys."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_related_model_keys(self, model_key: str) -> list[str]:
|
|
"""Gets all models keys related to a given model key."""
|
|
pass
|
|
|
|
@abstractmethod
|
|
def get_related_model_keys_batch(self, model_keys: list[str]) -> list[str]:
|
|
"""Get related model keys for multiple models."""
|
|
pass
|