获取模型实例 @param provider: 供应商 @param model_type: 模型类型 @param model_name: 模型名称 @param credential: 认证信息 @param model_id: 模型id @param use_local: 是否调用本地模型 只适用于本地供应商 @return: 模型实例
(provider, model_type, model_name, credential, model_id, use_local=False, **kwargs)
| 23 | |
| 24 | |
| 25 | def get_model_(provider, model_type, model_name, credential, model_id, use_local=False, **kwargs): |
| 26 | """ |
| 27 | 获取模型实例 |
| 28 | @param provider: 供应商 |
| 29 | @param model_type: 模型类型 |
| 30 | @param model_name: 模型名称 |
| 31 | @param credential: 认证信息 |
| 32 | @param model_id: 模型id |
| 33 | @param use_local: 是否调用本地模型 只适用于本地供应商 |
| 34 | @return: 模型实例 |
| 35 | """ |
| 36 | model = get_provider(provider).get_model(model_type, model_name, |
| 37 | json.loads( |
| 38 | rsa_long_decrypt(credential)), |
| 39 | model_id=model_id, |
| 40 | use_local=use_local, |
| 41 | streaming=True, **kwargs) |
| 42 | return model |
| 43 | |
| 44 | |
| 45 | def get_model(model, **kwargs): |
no test coverage detected