| from models.Gemini import Gemini |
| from models.OpenAI import ChatGPT |
| from models.OpenAI import GPT4 |
| from models.DeepSeek import DeepSeekModel |
| from models.Pangu import Pangu |
| from models.Qwen import Qwen |
| from models.Pangu72B import Pangu72B |
|
|
| class ModelFactory: |
| @staticmethod |
| def get_model_class(model_name): |
| if model_name == "Gemini": |
| return Gemini |
| elif model_name == "ChatGPT": |
| return ChatGPT |
| elif model_name == "GPT4": |
| return GPT4 |
| elif model_name == "DeepSeek": |
| return DeepSeekModel |
| elif model_name == "Pangu": |
| return Pangu |
| elif model_name == "Qwen": |
| return Qwen |
| elif model_name == "Pangu72B": |
| return Pangu72B |
| else: |
| raise Exception(f"Unknown model name {model_name}") |
|
|
