| import importlib |
|
|
| module_of_models = { |
| "gemini-pro": {"module_name": "gemini_api", "class_name": "gemini"}, |
| "gemini-1.5-pro": {"module_name": "gemini_api", "class_name": "gemini"}, |
| "gemini-1.5-flash": {"module_name": "gemini_api", "class_name": "gemini"} |
| } |
|
|
| def class_by_modelname(modelname): |
| module_obj = module_of_models[modelname] |
| module_name = module_obj["module_name"] |
| class_name = module_obj["class_name"] |
| module = importlib.import_module(module_name) |
| retclass = getattr(module, class_name) |
| return retclass |
