utils/env_utils.py

import os

import repository.repository
from repository.repository_abc import Model, ModelRoles


def in_hf() -> bool:
    return os.getenv("env") == "hf"


def build_repo_from_environment(system_prompt: str):
    implementation = os.getenv("implementation")
    model_name = os.getenv("model_name")

    if implementation:
        return repository.repository.get_repository(implementation, Model(model_name, ModelRoles("system",
                                                                                                 "user",
                                                                                                 "assistant")),
                                                    system_prompt)
    else:
        return None