feat: add supported_model_types field and filter in provider list (#1581)

2023-11-21 15:06:47 +08:00
parent 0d9ce1bab0
commit c9368925a3
18 changed files with 78 additions and 10 deletions
--- a/api/controllers/console/workspace/model_providers.py
+++ b/api/controllers/console/workspace/model_providers.py
@@ -21,8 +21,12 @@ class ModelProviderListApi(Resource):
    def get(self):
        tenant_id = current_user.current_tenant_id

+        parser = reqparse.RequestParser()
+        parser.add_argument('model_type', type=str, required=False, nullable=True, location='args')
+        args = parser.parse_args()
+
        provider_service = ProviderService()
-        provider_list = provider_service.get_provider_list(tenant_id)
+        provider_list = provider_service.get_provider_list(tenant_id=tenant_id, model_type=args.get('model_type'))

        return provider_list