site init move to event handler

api/controllers/console/app/app.py

 import json
-import logging
 from datetime import datetime
 from typing import cast
 
@@ -8,29 +7,24 @@ from flask_login import current_user
 from flask_restful import Resource, abort, inputs, marshal_with, reqparse
 from werkzeug.exceptions import Forbidden
 
-from constants.languages import languages
 from constants.model_template import default_app_templates
 from controllers.console import api
-from controllers.console.app.error import ProviderNotInitializeError
 from controllers.console.app.wraps import get_app_model
 from controllers.console.setup import setup_required
 from controllers.console.wraps import account_initialization_required, cloud_edition_billing_resource_check
-from core.errors.error import LLMBadRequestError, ProviderTokenNotInitError
+from core.errors.error import ProviderTokenNotInitError
 from core.model_manager import ModelManager
 from core.model_runtime.entities.model_entities import ModelType, ModelPropertyKey
 from core.model_runtime.model_providers.__base.large_language_model import LargeLanguageModel
-from core.provider_manager import ProviderManager
 from events.app_event import app_was_created, app_was_deleted
 from extensions.ext_database import db
 from fields.app_fields import (
     app_detail_fields,
     app_detail_fields_with_site,
     app_pagination_fields,
-    template_list_fields,
 )
 from libs.login import login_required
-from models.model import App, AppModelConfig, Site, AppMode
-from services.app_model_config_service import AppModelConfigService
+from models.model import App, AppModelConfig, AppMode
 from services.workflow_service import WorkflowService
 
 
@@ -99,52 +93,6 @@ class AppListApi(Resource):
         if not current_user.is_admin_or_owner:
             raise Forbidden()
 
-        # TODO: MOVE TO IMPORT API
-        if args['model_config'] is not None:
-            # validate config
-            model_config_dict = args['model_config']
-
-            # Get provider configurations
-            provider_manager = ProviderManager()
-            provider_configurations = provider_manager.get_configurations(current_user.current_tenant_id)
-
-            # get available models from provider_configurations
-            available_models = provider_configurations.get_models(
-                model_type=ModelType.LLM,
-                only_active=True
-            )
-
-            # check if model is available
-            available_models_names = [f'{model.provider.provider}.{model.model}' for model in available_models]
-            provider_model = f"{model_config_dict['model']['provider']}.{model_config_dict['model']['name']}"
-            if provider_model not in available_models_names:
-                if not default_model_entity:
-                    raise ProviderNotInitializeError(
-                        "No Default System Reasoning Model available. Please configure "
-                        "in the Settings -> Model Provider.")
-                else:
-                    model_config_dict["model"]["provider"] = default_model_entity.provider
-                    model_config_dict["model"]["name"] = default_model_entity.model
-
-            model_configuration = AppModelConfigService.validate_configuration(
-                tenant_id=current_user.current_tenant_id,
-                account=current_user,
-                config=model_config_dict,
-                app_mode=args['mode']
-            )
-
-            app = App(
-                enable_site=True,
-                enable_api=True,
-                is_demo=False,
-                api_rpm=0,
-                api_rph=0,
-                status='normal'
-            )
-
-            app_model_config = AppModelConfig()
-            app_model_config = app_model_config.from_model_config_dict(model_configuration)
-        else:
         if 'mode' not in args or args['mode'] is None:
             abort(400, message="mode is required")
 
@@ -186,8 +134,6 @@ class AppListApi(Resource):
             default_model_config['model'] = json.dumps(default_model_dict)
 
         app = App(**app_template['app'])
-            app_model_config = AppModelConfig(**default_model_config)
-
         app.name = args['name']
         app.mode = args['mode']
         app.icon = args['icon']
@@ -197,26 +143,14 @@ class AppListApi(Resource):
         db.session.add(app)
         db.session.flush()
 
+        app_model_config = AppModelConfig(**default_model_config)
         app_model_config.app_id = app.id
         db.session.add(app_model_config)
         db.session.flush()
 
         app.app_model_config_id = app_model_config.id
 
-        account = current_user
-
-        site = Site(
-            app_id=app.id,
-            title=app.name,
-            default_language=account.interface_language,
-            customize_token_strategy='not_allow',
-            code=Site.generate_code(16)
-        )
-
-        db.session.add(site)
-        db.session.commit()
-
-        app_was_created.send(app)
+        app_was_created.send(app, account=current_user)
 
         return app, 201
 
@@ -259,21 +193,16 @@ class AppImportApi(Resource):
                                  "when mode is advanced-chat or workflow")
 
         app = App(
+            tenant_id=current_user.current_tenant_id,
+            mode=app_data.get('mode'),
+            name=args.get("name") if args.get("name") else app_data.get('name'),
+            icon=args.get("icon") if args.get("icon") else app_data.get('icon'),
+            icon_background=args.get("icon_background") if args.get("icon_background") \
+                else app_data.get('icon_background'),
             enable_site=True,
-            enable_api=True,
-            is_demo=False,
-            api_rpm=0,
-            api_rph=0,
-            status='normal'
+            enable_api=True
         )
 
-        app.tenant_id = current_user.current_tenant_id
-        app.mode = app_data.get('mode')
-        app.name = args.get("name") if args.get("name") else app_data.get('name')
-        app.icon = args.get("icon") if args.get("icon") else app_data.get('icon')
-        app.icon_background = args.get("icon_background") if args.get("icon_background") \
-            else app_data.get('icon_background')
-
         db.session.add(app)
         db.session.commit()
 
@@ -292,20 +221,7 @@ class AppImportApi(Resource):
 
         app.app_model_config_id = app_model_config.id
 
-        account = current_user
-
-        site = Site(
-            app_id=app.id,
-            title=app.name,
-            default_language=account.interface_language,
-            customize_token_strategy='not_allow',
-            code=Site.generate_code(16)
-        )
-
-        db.session.add(site)
-        db.session.commit()
-
-        app_was_created.send(app)
+        app_was_created.send(app, account=current_user)
 
         return app, 201
 

api/events/event_handlers/__init__.py

@@ -2,6 +2,7 @@ from .clean_when_dataset_deleted import handle
 from .clean_when_document_deleted import handle
 from .create_document_index import handle
 from .create_installed_app_when_app_created import handle
+from .create_site_record_when_app_created import handle
 from .deduct_quota_when_messaeg_created import handle
 from .delete_installed_app_when_app_deleted import handle
 from .generate_conversation_name_when_first_message_created import handle

api/events/event_handlers/create_site_record_when_app_created.py

+from events.app_event import app_was_created
+from extensions.ext_database import db
+from models.model import Site
+
+
+@app_was_created.connect
+def handle(sender, **kwargs):
+    """Create site record when an app is created."""
+    app = sender
+    account = kwargs.get('account')
+    site = Site(
+        app_id=app.id,
+        title=app.name,
+        default_language=account.interface_language,
+        customize_token_strategy='not_allow',
+        code=Site.generate_code(16)
+    )
+
+    db.session.add(site)
+    db.session.commit()

api/services/workflow/workflow_converter.py

@@ -93,18 +93,7 @@ class WorkflowConverter:
         new_app.app_model_config_id = new_app_model_config.id
         db.session.commit()
 
-        site = Site(
-            app_id=new_app.id,
-            title=new_app.name,
-            default_language=account.interface_language,
-            customize_token_strategy='not_allow',
-            code=Site.generate_code(16)
-        )
-
-        db.session.add(site)
-        db.session.commit()
-
-        app_was_created.send(new_app)
+        app_was_created.send(new_app, account=account)
 
         return new_app