File size: 518 Bytes
2ebf9ad
 
 
 
 
 
 
 
 
 
 
70bc77d
2ebf9ad
70bc77d
2ebf9ad
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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")