Models/Utils.py

from Models.Gemini import Gemini
from Models.GPT import GPT

message_user = lambda x: ({"role": "user", "content": x})
message_system = lambda x: ({"role": "system", "content": x})


# Core Functions here ,if want more write in backend
def get_specific_model(model_name):
    if model_name in ["gemini-1.5-flash","gemini"]:
        return Gemini()
    elif model_name in ["openai/gpt-oss-120b","gpt 120","gbt20"]:
        return GPT("openai/gpt-oss-20b")
        
    else :
        return GPT("openai/gpt-oss-120b")