feat: change dashboard commercial logic

Author: zgqgit

.gitignore

@@ -264,3 +264,4 @@ docker/mysql/data/*
 docker/office/bisheng/*.gz
 
 CLAUDE.md
+!src/backend/bisheng/telemetry_search/*.pyc

src/backend/bisheng/api/router.py

@@ -17,12 +17,7 @@
                                                knowledge_router_rpc, workflow_router_rpc,
                                                filelib_router_rpc)
 from bisheng.share_link.api.router import router as share_link_router
-
-try:
-    from bisheng.telemetry_search.api.router import router as telemetry_search_router
-except ImportError:
-    telemetry_search_router = None
-    pass
+from bisheng.telemetry_search.api.router import router as telemetry_search_router
 
 router = APIRouter(prefix='/api/v1', )
 router.include_router(chat_router)
@@ -52,8 +47,7 @@
 router.include_router(invite_code_router)
 router.include_router(session_router)
 router.include_router(share_link_router)
-if telemetry_search_router:
-    router.include_router(telemetry_search_router)
+router.include_router(telemetry_search_router)
 
 router_rpc = APIRouter(prefix='/api/v2', )
 router_rpc.include_router(knowledge_router_rpc)

src/backend/bisheng/api/services/role_group_service.py

@@ -22,6 +22,7 @@
 from bisheng.database.models.role import RoleDao
 from bisheng.database.models.user_group import UserGroupCreate, UserGroupDao, UserGroupRead
 from bisheng.knowledge.domain.models.knowledge import KnowledgeDao
+from bisheng.telemetry_search.domain.services.dashboard import DashboardService
 from bisheng.tool.domain.models.gpts_tools import GptsToolsDao
 from bisheng.user.domain.models.user import User, UserDao
 from bisheng.user.domain.models.user_role import UserRoleDao
@@ -348,7 +349,6 @@ def get_group_tool(self, group_id: int, keyword: str, page_size: int, page_num:
 
     async def get_group_dashboards(self, group_id: int, keyword: str, page_size: int, page_num: int) -> (List[Any],
                                                                                                          int):
-        from bisheng.telemetry_search.domain.services.dashboard import DashboardService
 
         """ 获取用户组下的仪表盘列表 """
         # 查询用户组下的仪表盘ID列表

src/backend/bisheng/common/init_data.py

@@ -16,6 +16,7 @@
 from bisheng.database.models.role import Role
 from bisheng.database.models.role_access import RoleAccess, AccessType, WebMenuResource
 from bisheng.database.models.template import Template
+from bisheng.telemetry_search.domain.init_dataset import init_dashboard_datasets
 from bisheng.tool.domain.models.gpts_tools import GptsTools
 from bisheng.tool.domain.models.gpts_tools import GptsToolsType
 from bisheng.user.domain.models.user import User
@@ -132,13 +133,9 @@ async def init_default_data():
 
             # 初始化数据库config
             await settings.init_config()
-            try:
-                from bisheng.telemetry_search.domain.init_dataset import init_dashboard_datasets
 
-                # init dashboard data
-                await init_dashboard_datasets()
-            except Exception as e:
-                logger.error(f"dashboard data init error: {e}")
+            # init dashboard data
+            await init_dashboard_datasets()
         except Exception as exc:
             # if the exception involves tables already existing
             # we can ignore it

src/backend/bisheng/common/services/config_service.py

@@ -198,12 +198,7 @@ def get_system_login_method(self) -> SystemLoginMethod:
         # 获取密码相关的配置项
         all_config = self.get_all_config()
         tmp = SystemLoginMethod(**all_config.get('system_login_method', {}))
-        try:
-            from bisheng.telemetry_search.domain.utils import is_commercial
-            tmp.bisheng_pro = is_commercial()
-        except Exception as e:
-            logger.error("import dashboard error")
-            tmp.bisheng_pro = False
+        tmp.bisheng_pro = os.getenv('BISHENG_PRO') == 'true'
         return tmp
 
     def get_workflow_conf(self) -> WorkflowConf: