util/configuration.py

from pydantic import BaseModel

class InferenceConfiguration(BaseModel):
    model_path_keyfigure: str = "danielsteinigen/KeyFiTax"
    spacy_model: str = "de_core_news_sm"
    transformer_model: str = "xlm-roberta-large"
    merge_entities: bool = True
    split_len: int = 200
    extract_relations: bool = True