chore(api/services): apply ruff reformatting (#7599)

Co-authored-by: -LAN- <laipz8200@outlook.com>

api/services/model_provider_service.py

@@ -73,8 +73,8 @@ class ModelProviderService:
                 system_configuration=SystemConfigurationResponse(
                     enabled=provider_configuration.system_configuration.enabled,
                     current_quota_type=provider_configuration.system_configuration.current_quota_type,
-                    quota_configurations=provider_configuration.system_configuration.quota_configurations
-                )
+                    quota_configurations=provider_configuration.system_configuration.quota_configurations,
+                ),
             )
 
             provider_responses.append(provider_response)
@@ -95,9 +95,9 @@ class ModelProviderService:
         provider_configurations = self.provider_manager.get_configurations(tenant_id)
 
         # Get provider available models
-        return [ModelWithProviderEntityResponse(model) for model in provider_configurations.get_models(
-            provider=provider
-        )]
+        return [
+            ModelWithProviderEntityResponse(model) for model in provider_configurations.get_models(provider=provider)
+        ]
 
     def get_provider_credentials(self, tenant_id: str, provider: str) -> dict:
         """
@@ -195,13 +195,12 @@ class ModelProviderService:
 
         # Get model custom credentials from ProviderModel if exists
         return provider_configuration.get_custom_model_credentials(
-            model_type=ModelType.value_of(model_type),
-            model=model,
-            obfuscated=True
+            model_type=ModelType.value_of(model_type), model=model, obfuscated=True
         )
 
-    def model_credentials_validate(self, tenant_id: str, provider: str, model_type: str, model: str,
-                                   credentials: dict) -> None:
+    def model_credentials_validate(
+        self, tenant_id: str, provider: str, model_type: str, model: str, credentials: dict
+    ) -> None:
         """
         validate model credentials.
 
@@ -222,13 +221,12 @@ class ModelProviderService:
 
         # Validate model credentials
         provider_configuration.custom_model_credentials_validate(
-            model_type=ModelType.value_of(model_type),
-            model=model,
-            credentials=credentials
+            model_type=ModelType.value_of(model_type), model=model, credentials=credentials
         )
 
-    def save_model_credentials(self, tenant_id: str, provider: str, model_type: str, model: str,
-                               credentials: dict) -> None:
+    def save_model_credentials(
+        self, tenant_id: str, provider: str, model_type: str, model: str, credentials: dict
+    ) -> None:
         """
         save model credentials.
 
@@ -249,9 +247,7 @@ class ModelProviderService:
 
         # Add or update custom model credentials
         provider_configuration.add_or_update_custom_model_credentials(
-            model_type=ModelType.value_of(model_type),
-            model=model,
-            credentials=credentials
+            model_type=ModelType.value_of(model_type), model=model, credentials=credentials
         )
 
     def remove_model_credentials(self, tenant_id: str, provider: str, model_type: str, model: str) -> None:
@@ -273,10 +269,7 @@ class ModelProviderService:
             raise ValueError(f"Provider {provider} does not exist.")
 
         # Remove custom model credentials
-        provider_configuration.delete_custom_model_credentials(
-            model_type=ModelType.value_of(model_type),
-            model=model
-        )
+        provider_configuration.delete_custom_model_credentials(model_type=ModelType.value_of(model_type), model=model)
 
     def get_models_by_model_type(self, tenant_id: str, model_type: str) -> list[ProviderWithModelsResponse]:
         """
@@ -290,9 +283,7 @@ class ModelProviderService:
         provider_configurations = self.provider_manager.get_configurations(tenant_id)
 
         # Get provider available models
-        models = provider_configurations.get_models(
-            model_type=ModelType.value_of(model_type)
-        )
+        models = provider_configurations.get_models(model_type=ModelType.value_of(model_type))
 
         # Group models by provider
         provider_models = {}
@@ -323,16 +314,19 @@ class ModelProviderService:
                     icon_small=first_model.provider.icon_small,
                     icon_large=first_model.provider.icon_large,
                     status=CustomConfigurationStatus.ACTIVE,
-                    models=[ProviderModelWithStatusEntity(
-                        model=model.model,
-                        label=model.label,
-                        model_type=model.model_type,
-                        features=model.features,
-                        fetch_from=model.fetch_from,
-                        model_properties=model.model_properties,
-                        status=model.status,
-                        load_balancing_enabled=model.load_balancing_enabled
-                    ) for model in models]
+                    models=[
+                        ProviderModelWithStatusEntity(
+                            model=model.model,
+                            label=model.label,
+                            model_type=model.model_type,
+                            features=model.features,
+                            fetch_from=model.fetch_from,
+                            model_properties=model.model_properties,
+                            status=model.status,
+                            load_balancing_enabled=model.load_balancing_enabled,
+                        )
+                        for model in models
+                    ],
                 )
             )
 
@@ -361,19 +355,13 @@ class ModelProviderService:
         model_type_instance = cast(LargeLanguageModel, model_type_instance)
 
         # fetch credentials
-        credentials = provider_configuration.get_current_credentials(
-            model_type=ModelType.LLM,
-            model=model
-        )
+        credentials = provider_configuration.get_current_credentials(model_type=ModelType.LLM, model=model)
 
         if not credentials:
             return []
 
         # Call get_parameter_rules method of model instance to get model parameter rules
-        return model_type_instance.get_parameter_rules(
-            model=model,
-            credentials=credentials
-        )
+        return model_type_instance.get_parameter_rules(model=model, credentials=credentials)
 
     def get_default_model_of_model_type(self, tenant_id: str, model_type: str) -> Optional[DefaultModelResponse]:
         """
@@ -384,22 +372,23 @@ class ModelProviderService:
         :return:
         """
         model_type_enum = ModelType.value_of(model_type)
-        result = self.provider_manager.get_default_model(
-            tenant_id=tenant_id,
-            model_type=model_type_enum
-        )
+        result = self.provider_manager.get_default_model(tenant_id=tenant_id, model_type=model_type_enum)
         try:
-            return DefaultModelResponse(
-                model=result.model,
-                model_type=result.model_type,
-                provider=SimpleProviderEntityResponse(
-                    provider=result.provider.provider,
-                    label=result.provider.label,
-                    icon_small=result.provider.icon_small,
-                    icon_large=result.provider.icon_large,
-                    supported_model_types=result.provider.supported_model_types
+            return (
+                DefaultModelResponse(
+                    model=result.model,
+                    model_type=result.model_type,
+                    provider=SimpleProviderEntityResponse(
+                        provider=result.provider.provider,
+                        label=result.provider.label,
+                        icon_small=result.provider.icon_small,
+                        icon_large=result.provider.icon_large,
+                        supported_model_types=result.provider.supported_model_types,
+                    ),
                 )
-            ) if result else None
+                if result
+                else None
+            )
         except Exception as e:
             logger.info(f"get_default_model_of_model_type error: {e}")
             return None
@@ -416,13 +405,12 @@ class ModelProviderService:
         """
         model_type_enum = ModelType.value_of(model_type)
         self.provider_manager.update_default_model_record(
-            tenant_id=tenant_id,
-            model_type=model_type_enum,
-            provider=provider,
-            model=model
+            tenant_id=tenant_id, model_type=model_type_enum, provider=provider, model=model
         )
 
-    def get_model_provider_icon(self, provider: str, icon_type: str, lang: str) -> tuple[Optional[bytes], Optional[str]]:
+    def get_model_provider_icon(
+        self, provider: str, icon_type: str, lang: str
+    ) -> tuple[Optional[bytes], Optional[str]]:
         """
         get model provider icon.
 
@@ -434,11 +422,11 @@ class ModelProviderService:
         provider_instance = model_provider_factory.get_provider_instance(provider)
         provider_schema = provider_instance.get_provider_schema()
 
-        if icon_type.lower() == 'icon_small':
+        if icon_type.lower() == "icon_small":
             if not provider_schema.icon_small:
                 raise ValueError(f"Provider {provider} does not have small icon.")
 
-            if lang.lower() == 'zh_hans':
+            if lang.lower() == "zh_hans":
                 file_name = provider_schema.icon_small.zh_Hans
             else:
                 file_name = provider_schema.icon_small.en_US
@@ -446,13 +434,15 @@ class ModelProviderService:
             if not provider_schema.icon_large:
                 raise ValueError(f"Provider {provider} does not have large icon.")
 
-            if lang.lower() == 'zh_hans':
+            if lang.lower() == "zh_hans":
                 file_name = provider_schema.icon_large.zh_Hans
             else:
                 file_name = provider_schema.icon_large.en_US
 
         root_path = current_app.root_path
-        provider_instance_path = os.path.dirname(os.path.join(root_path, provider_instance.__class__.__module__.replace('.', '/')))
+        provider_instance_path = os.path.dirname(
+            os.path.join(root_path, provider_instance.__class__.__module__.replace(".", "/"))
+        )
         file_path = os.path.join(provider_instance_path, "_assets")
         file_path = os.path.join(file_path, file_name)
 
@@ -460,10 +450,10 @@ class ModelProviderService:
             return None, None
 
         mimetype, _ = mimetypes.guess_type(file_path)
-        mimetype = mimetype or 'application/octet-stream'
+        mimetype = mimetype or "application/octet-stream"
 
         # read binary from file
-        with open(file_path, 'rb') as f:
+        with open(file_path, "rb") as f:
             byte_data = f.read()
             return byte_data, mimetype
 
@@ -509,10 +499,7 @@ class ModelProviderService:
             raise ValueError(f"Provider {provider} does not exist.")
 
         # Enable model
-        provider_configuration.enable_model(
-            model=model,
-            model_type=ModelType.value_of(model_type)
-        )
+        provider_configuration.enable_model(model=model, model_type=ModelType.value_of(model_type))
 
     def disable_model(self, tenant_id: str, provider: str, model: str, model_type: str) -> None:
         """
@@ -533,78 +520,49 @@ class ModelProviderService:
             raise ValueError(f"Provider {provider} does not exist.")
 
         # Enable model
-        provider_configuration.disable_model(
-            model=model,
-            model_type=ModelType.value_of(model_type)
-        )
+        provider_configuration.disable_model(model=model, model_type=ModelType.value_of(model_type))
 
     def free_quota_submit(self, tenant_id: str, provider: str):
         api_key = os.environ.get("FREE_QUOTA_APPLY_API_KEY")
         api_base_url = os.environ.get("FREE_QUOTA_APPLY_BASE_URL")
-        api_url = api_base_url + '/api/v1/providers/apply'
+        api_url = api_base_url + "/api/v1/providers/apply"
 
-        headers = {
-            'Content-Type': 'application/json',
-            'Authorization': f"Bearer {api_key}"
-        }
-        response = requests.post(api_url, headers=headers, json={'workspace_id': tenant_id, 'provider_name': provider})
+        headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
+        response = requests.post(api_url, headers=headers, json={"workspace_id": tenant_id, "provider_name": provider})
         if not response.ok:
             logger.error(f"Request FREE QUOTA APPLY SERVER Error: {response.status_code} ")
             raise ValueError(f"Error: {response.status_code} ")
 
-        if response.json()["code"] != 'success':
-            raise ValueError(
-                f"error: {response.json()['message']}"
-            )
+        if response.json()["code"] != "success":
+            raise ValueError(f"error: {response.json()['message']}")
 
         rst = response.json()
 
-        if rst['type'] == 'redirect':
-            return {
-                'type': rst['type'],
-                'redirect_url': rst['redirect_url']
-            }
+        if rst["type"] == "redirect":
+            return {"type": rst["type"], "redirect_url": rst["redirect_url"]}
         else:
-            return {
-                'type': rst['type'],
-                'result': 'success'
-            }
+            return {"type": rst["type"], "result": "success"}
 
     def free_quota_qualification_verify(self, tenant_id: str, provider: str, token: Optional[str]):
         api_key = os.environ.get("FREE_QUOTA_APPLY_API_KEY")
         api_base_url = os.environ.get("FREE_QUOTA_APPLY_BASE_URL")
-        api_url = api_base_url + '/api/v1/providers/qualification-verify'
+        api_url = api_base_url + "/api/v1/providers/qualification-verify"
 
-        headers = {
-            'Content-Type': 'application/json',
-            'Authorization': f"Bearer {api_key}"
-        }
-        json_data = {'workspace_id': tenant_id, 'provider_name': provider}
+        headers = {"Content-Type": "application/json", "Authorization": f"Bearer {api_key}"}
+        json_data = {"workspace_id": tenant_id, "provider_name": provider}
         if token:
-            json_data['token'] = token
-        response = requests.post(api_url, headers=headers,
-                                 json=json_data)
+            json_data["token"] = token
+        response = requests.post(api_url, headers=headers, json=json_data)
         if not response.ok:
             logger.error(f"Request FREE QUOTA APPLY SERVER Error: {response.status_code} ")
             raise ValueError(f"Error: {response.status_code} ")
 
         rst = response.json()
-        if rst["code"] != 'success':
-            raise ValueError(
-                f"error: {rst['message']}"
-            )
+        if rst["code"] != "success":
+            raise ValueError(f"error: {rst['message']}")
 
-        data = rst['data']
-        if data['qualified'] is True:
-            return {
-                'result': 'success',
-                'provider_name': provider,
-                'flag': True
-            }
+        data = rst["data"]
+        if data["qualified"] is True:
+            return {"result": "success", "provider_name": provider, "flag": True}
         else:
-            return {
-                'result': 'success',
-                'provider_name': provider,
-                'flag': False,
-                'reason': data['reason']
-            }
+            return {"result": "success", "provider_name": provider, "flag": False, "reason": data["reason"]}