fix: mcp tools check is support stdio conf

Author: zgqgit

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

@@ -8,7 +8,7 @@
 from bisheng.common.repositories.implementations.config_repository_impl import ConfigRepositoryImpl
 from bisheng.core.cache.redis_manager import get_redis_client_sync
 from bisheng.core.config.settings import Settings, PasswordConf, SystemLoginMethod, \
-    WorkflowConf, LinsightConf, KnowledgeConf, IntelligenceCenterConf
+    WorkflowConf, LinsightConf, KnowledgeConf, IntelligenceCenterConf, McpConf
 from bisheng.core.database import get_sync_db_session, get_async_db_session
 
 config_file = os.getenv('config', 'config.yaml')
@@ -262,5 +262,12 @@ async def aget_from_db(self, key: str):
         all_config = await self.aget_all_config()
         return all_config.get(key, {})
 
+    async def get_mcp_conf(self) -> McpConf:
+        all_config = await self.aget_all_config()
+        mcp_conf = all_config.get('mcp', {})
+        if mcp_conf:
+            return McpConf(**mcp_conf)
+        return McpConf()
+
 
 settings = ConfigService.load_settings_from_yaml(config_file)

src/backend/bisheng/core/config/settings.py

@@ -252,6 +252,11 @@ class IntelligenceCenterConf(BaseModel):
     kwargs: Dict = Field(default_factory=dict, description='Additional Arguments')
 
 
+class McpConf(BaseModel):
+    """ MCP Configure """
+    enable_stdio: bool = Field(default=True, description='Whether to enable stdio')
+
+
 class Settings(BaseModel):
     """ Application Settings """
     model_config = ConfigDict(validate_assignment=True, arbitrary_types_allowed=True, extra='ignore')
@@ -305,6 +310,7 @@ class Settings(BaseModel):
     license_str: Optional[str] = None  # license Contents
 
     information_conf: IntelligenceCenterConf = IntelligenceCenterConf()
+    mcp: McpConf = McpConf()
 
     @field_validator('database_url')
     @classmethod

src/backend/bisheng/initdb_config.yaml

@@ -96,3 +96,6 @@ linsight:
   default_temperature: 0
   # 模型json格式失败重试时的温度
   retry_temperature: 1
+
+mcp:
+  enable_stdio: true

src/backend/bisheng/tool/domain/services/tool.py

@@ -16,9 +16,11 @@
 from bisheng.common.errcode.tool import ToolTypeNotExistsError, ToolTypeRepeatError, ToolTypeNameError, \
     ToolTypeIsPresetError, ToolSchemaDownloadError, ToolSchemaEmptyError, ToolSchemaParseError, ToolSchemaServerError, \
     ToolMcpSchemaError
+from bisheng.common.services.config_service import settings
 from bisheng.database.models.group_resource import GroupResourceDao, ResourceTypeEnum, GroupResource
 from bisheng.database.models.role_access import AccessType
 from bisheng.database.models.user_group import UserGroupDao
+from bisheng.mcp_manage.clients.stdio import StdioClient
 from bisheng.mcp_manage.manager import ClientManager
 from bisheng.tool.domain.const import ToolPresetType
 from bisheng.tool.domain.langchain.linsight_knowledge import SearchKnowledgeBase
@@ -263,6 +265,12 @@ async def parse_mcp_schema(file_content: str) -> GptsToolsTypeRead:
             break
         if tool_type is None:
             raise ToolMcpSchemaError()
+        mcp_conf = await settings.get_mcp_conf()
+        if not mcp_conf.enable_stdio:
+            client = await ClientManager.connect_mcp_from_json(tool_type.openapi_schema)
+            if isinstance(client, StdioClient):
+                raise ToolMcpSchemaError()
+
         return tool_type
 
     @classmethod