mldr/models/__init__.py

from .default import PROMPT as default_prompt
from .retrieva import PROMPT as retrieva_prompt 
from .retrieva_en import PROMPT as retrieva_en_prompt


PROMPTS = {
    "default": default_prompt,
    "retrieva": retrieva_prompt,
    "retrieva-en": retrieva_en_prompt,
}