15308555518/ MaxKB
1
0
Fork 0
mirror of https://github.com/1Panel-dev/MaxKB.git synced 2025-12-26 01:33:05 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

feat: add MCP and tool configuration options in application settings

This commit is contained in:
CaptainB 2025-09-08 11:24:27 +08:00 committed by 刘瑞斌
parent 76766ebc23
commit 4a8cd95119
10 changed files with 493 additions and 86 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
apps/application/chat_pipeline/step/chat_step/i_chat_step.py
View File

@ -82,6 +82,12 @@ class IChatStep(IBaseChatPipelineStep):
model_params_setting = serializers.DictField(required=False, allow_null=True,
label=_("Model parameter settings"))
mcp_enable = serializers.BooleanField(label="MCP否启用", required=False, default=False)
mcp_tool_ids = serializers.JSONField(label="MCP工具ID列表", required=False, default=list)
mcp_servers = serializers.JSONField(label="MCP服务列表", required=False, default=dict)
mcp_source = serializers.CharField(label="MCP Source", required=False, default="referencing")
tool_enable = serializers.BooleanField(label="工具是否启用", required=False, default=False)
tool_ids = serializers.JSONField(label="工具ID列表", required=False, default=list)
def is_valid(self, *, raise_exception=False):
super().is_valid(raise_exception=True)
@ -106,5 +112,8 @@ class IChatStep(IBaseChatPipelineStep):
paragraph_list=None,
manage: PipelineManage = None,
padding_problem_text: str = None, stream: bool = True, chat_user_id=None, chat_user_type=None,
no_references_setting=None, model_params_setting=None, model_setting=None, **kwargs):
no_references_setting=None, model_params_setting=None, model_setting=None,
mcp_enable=False, mcp_tool_ids=None, mcp_servers='', mcp_source="referencing",
tool_enable=False, tool_ids=None,
**kwargs):
pass

130
apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py
View File

@ -6,7 +6,8 @@
@date：2024/1/9 18:25
@desc: 对话step Base实现
"""
import logging
import json
import os
import time
import traceback
import uuid_utils.compat as uuid
@ -24,10 +25,14 @@ from rest_framework import status
from application.chat_pipeline.I_base_chat_pipeline import ParagraphPipelineModel
from application.chat_pipeline.pipeline_manage import PipelineManage
from application.chat_pipeline.step.chat_step.i_chat_step import IChatStep, PostResponseHandler
from application.flow.tools import Reasoning
from application.flow.tools import Reasoning, mcp_response_generator
from application.models import ApplicationChatUserStats, ChatUserType
from common.utils.logger import maxkb_logger
from common.utils.rsa_util import rsa_long_decrypt
from common.utils.tool_code import ToolExecutor
from maxkb.const import CONFIG
from models_provider.tools import get_model_instance_by_model_workspace_id
from tools.models import Tool
def add_access_num(chat_user_id=None, chat_user_type=None, application_id=None):
@ -54,6 +59,7 @@ def write_context(step, manage, request_token, response_token, all_text):
manage.context['answer_tokens'] = manage.context['answer_tokens'] + response_token
def event_content(response,
chat_id,
chat_record_id,
@ -169,6 +175,12 @@ class BaseChatStep(IChatStep):
no_references_setting=None,
model_params_setting=None,
model_setting=None,
mcp_enable=False,
mcp_tool_ids=None,
mcp_servers='',
mcp_source="referencing",
tool_enable=False,
tool_ids=None,
**kwargs):
chat_model = get_model_instance_by_model_workspace_id(model_id, workspace_id,
**model_params_setting) if model_id is not None else None
@ -177,14 +189,24 @@ class BaseChatStep(IChatStep):
paragraph_list,
manage, padding_problem_text, chat_user_id, chat_user_type,
no_references_setting,
model_setting)
model_setting,
mcp_enable, mcp_tool_ids, mcp_servers, mcp_source, tool_enable, tool_ids)
else:
return self.execute_block(message_list, chat_id, problem_text, post_response_handler, chat_model,
paragraph_list,
manage, padding_problem_text, chat_user_id, chat_user_type, no_references_setting,
model_setting)
model_setting,
mcp_enable, mcp_tool_ids, mcp_servers, mcp_source, tool_enable, tool_ids)
def get_details(self, manage, **kwargs):
# 删除临时生成的MCP代码文件
if self.context.get('execute_ids'):
executor = ToolExecutor(CONFIG.get('SANDBOX'))
# 清理工具代码文件，延时删除，避免文件被占用
for tool_id in self.context.get('execute_ids'):
code_path = f'{executor.sandbox_path}/execute/{tool_id}.py'
if os.path.exists(code_path):
os.remove(code_path)
return {
'step_type': 'chat_step',
'run_time': self.context['run_time'],
@ -206,12 +228,63 @@ class BaseChatStep(IChatStep):
result.append({'role': 'ai', 'content': answer_text})
return result
@staticmethod
def get_stream_result(message_list: List[BaseMessage],
def _handle_mcp_request(self, mcp_enable, tool_enable, mcp_source, mcp_servers, mcp_tool_ids, tool_ids,
chat_model, message_list):
if not mcp_enable and not tool_enable:
return None
mcp_servers_config = {}
# 迁移过来mcp_source是None
if mcp_source is None:
mcp_source = 'custom'
if mcp_enable:
# 兼容老数据
if not mcp_tool_ids:
mcp_tool_ids = []
if mcp_source == 'custom' and mcp_servers is not None and '"stdio"' not in mcp_servers:
mcp_servers_config = json.loads(mcp_servers)
elif mcp_tool_ids:
mcp_tools = QuerySet(Tool).filter(id__in=mcp_tool_ids).values()
for mcp_tool in mcp_tools:
if mcp_tool and mcp_tool['is_active']:
mcp_servers_config = {**mcp_servers_config, **json.loads(mcp_tool['code'])}
if tool_enable:
if tool_ids and len(tool_ids) > 0: # 如果有工具ID，则将其转换为MCP
self.context['tool_ids'] = tool_ids
self.context['execute_ids'] = []
for tool_id in tool_ids:
tool = QuerySet(Tool).filter(id=tool_id).first()
if not tool.is_active:
continue
executor = ToolExecutor(CONFIG.get('SANDBOX'))
if tool.init_params is not None:
params = json.loads(rsa_long_decrypt(tool.init_params))
else:
params = {}
_id, tool_config = executor.get_tool_mcp_config(tool.code, params)
self.context['execute_ids'].append(_id)
mcp_servers_config[str(tool.id)] = tool_config
if len(mcp_servers_config) > 0:
return mcp_response_generator(chat_model, message_list, json.dumps(mcp_servers_config))
return None
def get_stream_result(self, message_list: List[BaseMessage],
chat_model: BaseChatModel = None,
paragraph_list=None,
no_references_setting=None,
problem_text=None):
problem_text=None,
mcp_enable=False,
mcp_tool_ids=None,
mcp_servers='',
mcp_source="referencing",
tool_enable=False,
tool_ids=None):
if paragraph_list is None:
paragraph_list = []
directly_return_chunk_list = [AIMessageChunk(content=paragraph.content)
@ -227,6 +300,12 @@ class BaseChatStep(IChatStep):
return iter([AIMessageChunk(
_('Sorry, the AI model is not configured. Please go to the application to set up the AI model first.'))]), False
else:
# 处理 MCP 请求
mcp_result = self._handle_mcp_request(
mcp_enable, tool_enable, mcp_source, mcp_servers, mcp_tool_ids, tool_ids, chat_model, message_list,
)
if mcp_result:
return mcp_result, True
return chat_model.stream(message_list), True
def execute_stream(self, message_list: List[BaseMessage],
@ -239,9 +318,15 @@ class BaseChatStep(IChatStep):
padding_problem_text: str = None,
chat_user_id=None, chat_user_type=None,
no_references_setting=None,
model_setting=None):
model_setting=None,
mcp_enable=False,
mcp_tool_ids=None,
mcp_servers='',
mcp_source="referencing",
tool_enable=False,
tool_ids=None):
chat_result, is_ai_chat = self.get_stream_result(message_list, chat_model, paragraph_list,
no_references_setting, problem_text)
no_references_setting, problem_text, mcp_enable, mcp_tool_ids, mcp_servers, mcp_source, tool_enable, tool_ids)
chat_record_id = uuid.uuid7()
r = StreamingHttpResponse(
streaming_content=event_content(chat_result, chat_id, chat_record_id, paragraph_list,
@ -253,12 +338,17 @@ class BaseChatStep(IChatStep):
r['Cache-Control'] = 'no-cache'
return r
@staticmethod
def get_block_result(message_list: List[BaseMessage],
def get_block_result(self, message_list: List[BaseMessage],
chat_model: BaseChatModel = None,
paragraph_list=None,
no_references_setting=None,
problem_text=None):
problem_text=None,
mcp_enable=False,
mcp_tool_ids=None,
mcp_servers='',
mcp_source="referencing",
tool_enable=False,
tool_ids=None):
if paragraph_list is None:
paragraph_list = []
directly_return_chunk_list = [AIMessageChunk(content=paragraph.content)
@ -273,6 +363,12 @@ class BaseChatStep(IChatStep):
return AIMessage(
_('Sorry, the AI model is not configured. Please go to the application to set up the AI model first.')), False
else:
# 处理 MCP 请求
mcp_result = self._handle_mcp_request(
mcp_enable, tool_enable, mcp_source, mcp_servers, mcp_tool_ids, tool_ids, chat_model, message_list,
)
if mcp_result:
return mcp_result, True
return chat_model.invoke(message_list), True
def execute_block(self, message_list: List[BaseMessage],
@ -284,7 +380,13 @@ class BaseChatStep(IChatStep):
manage: PipelineManage = None,
padding_problem_text: str = None,
chat_user_id=None, chat_user_type=None, no_references_setting=None,
model_setting=None):
model_setting=None,
mcp_enable=False,
mcp_tool_ids=None,
mcp_servers='',
mcp_source="referencing",
tool_enable=False,
tool_ids=None):
reasoning_content_enable = model_setting.get('reasoning_content_enable', False)
reasoning_content_start = model_setting.get('reasoning_content_start', '<think>')
reasoning_content_end = model_setting.get('reasoning_content_end', '</think>')
@ -294,7 +396,7 @@ class BaseChatStep(IChatStep):
# 调用模型
try:
chat_result, is_ai_chat = self.get_block_result(message_list, chat_model, paragraph_list,
no_references_setting, problem_text)
no_references_setting, problem_text, mcp_enable, mcp_tool_ids, mcp_servers, mcp_source, tool_enable, tool_ids)
if is_ai_chat:
request_token = chat_model.get_num_tokens_from_messages(message_list)
response_token = chat_model.get_num_tokens(chat_result.content)

70
apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py
View File

@ -6,27 +6,21 @@
@date：2024/6/4 14:30
@desc:
"""
import asyncio
import json
import os
import re
import sys
import time
import traceback
from functools import reduce
from typing import List, Dict
import uuid_utils.compat as uuid
from django.db.models import QuerySet
from langchain.schema import HumanMessage, SystemMessage
from langchain_core.messages import BaseMessage, AIMessage, AIMessageChunk, ToolMessage
from langchain_mcp_adapters.client import MultiServerMCPClient
from langgraph.prebuilt import create_react_agent
from langchain_core.messages import BaseMessage, AIMessage
from application.flow.i_step_node import NodeResult, INode
from application.flow.step_node.ai_chat_step_node.i_chat_node import IChatNode
from application.flow.tools import Reasoning
from common.utils.logger import maxkb_logger
from application.flow.tools import Reasoning, mcp_response_generator
from common.utils.rsa_util import rsa_long_decrypt
from common.utils.tool_code import ToolExecutor
from maxkb.const import CONFIG
@ -34,31 +28,6 @@ from models_provider.models import Model
from models_provider.tools import get_model_credential, get_model_instance_by_model_workspace_id
from tools.models import Tool
tool_message_template = """
<details>
<summary>
<strong>Called MCP Tool: <em>%s</em></strong>
</summary>
%s
</details>
"""
tool_message_json_template = """
```json
%s
```
"""
def generate_tool_message_template(name, context):
if '```' in context:
return tool_message_template % (name, context)
else:
return tool_message_template % (name, tool_message_json_template % (context))
def _write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow, answer: str,
reasoning_content: str):
@ -122,39 +91,6 @@ def write_context_stream(node_variable: Dict, workflow_variable: Dict, node: INo
_write_context(node_variable, workflow_variable, node, workflow, answer, reasoning_content)
async def _yield_mcp_response(chat_model, message_list, mcp_servers):
client = MultiServerMCPClient(json.loads(mcp_servers))
tools = await client.get_tools()
agent = create_react_agent(chat_model, tools)
response = agent.astream({"messages": message_list}, stream_mode='messages')
async for chunk in response:
if isinstance(chunk[0], ToolMessage):
content = generate_tool_message_template(chunk[0].name, chunk[0].content)
chunk[0].content = content
yield chunk[0]
if isinstance(chunk[0], AIMessageChunk):
yield chunk[0]
def mcp_response_generator(chat_model, message_list, mcp_servers):
loop = asyncio.new_event_loop()
try:
async_gen = _yield_mcp_response(chat_model, message_list, mcp_servers)
while True:
try:
chunk = loop.run_until_complete(anext_async(async_gen))
yield chunk
except StopAsyncIteration:
break
except Exception as e:
maxkb_logger.error(f'Exception: {e}', traceback.format_exc())
finally:
loop.close()
async def anext_async(agen):
return await agen.__anext__()
def write_context(node_variable: Dict, workflow_variable: Dict, node: INode, workflow):
"""

67
apps/application/flow/tools.py
View File

@ -6,14 +6,18 @@
@date：2024/6/6 15:15
@desc:
"""
import asyncio
import json
import traceback
from typing import Iterator
from django.http import StreamingHttpResponse
from langchain_core.messages import BaseMessageChunk, BaseMessage
from langchain_core.messages import BaseMessageChunk, BaseMessage, ToolMessage, AIMessageChunk
from langchain_mcp_adapters.client import MultiServerMCPClient
from langgraph.prebuilt import create_react_agent
from application.flow.i_step_node import WorkFlowPostHandler
from common.result import result
from common.utils.logger import maxkb_logger
class Reasoning:
@ -196,3 +200,62 @@ def to_stream_response_simple(stream_event):
r['Cache-Control'] = 'no-cache'
return r
tool_message_template = """
<details>
<summary>
<strong>Called MCP Tool: <em>%s</em></strong>
</summary>
%s
</details>
"""
tool_message_json_template = """
```json
%s
```
"""
def generate_tool_message_template(name, context):
if '```' in context:
return tool_message_template % (name, context)
else:
return tool_message_template % (name, tool_message_json_template % (context))
async def _yield_mcp_response(chat_model, message_list, mcp_servers):
client = MultiServerMCPClient(json.loads(mcp_servers))
tools = await client.get_tools()
agent = create_react_agent(chat_model, tools)
response = agent.astream({"messages": message_list}, stream_mode='messages')
async for chunk in response:
if isinstance(chunk[0], ToolMessage):
content = generate_tool_message_template(chunk[0].name, chunk[0].content)
chunk[0].content = content
yield chunk[0]
if isinstance(chunk[0], AIMessageChunk):
yield chunk[0]
def mcp_response_generator(chat_model, message_list, mcp_servers):
loop = asyncio.new_event_loop()
try:
async_gen = _yield_mcp_response(chat_model, message_list, mcp_servers)
while True:
try:
chunk = loop.run_until_complete(anext_async(async_gen))
yield chunk
except StopAsyncIteration:
break
except Exception as e:
maxkb_logger.error(f'Exception: {e}', traceback.format_exc())
finally:
loop.close()
async def anext_async(agen):
return await agen.__anext__()

73
apps/application/migrations/0002_application_mcp_enable_application_mcp_servers_and_more.py Normal file
View File

@ -0,0 +1,73 @@
# Generated by Django 5.2.4 on 2025-09-08 03:16
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
('application', '0001_initial'),
]
operations = [
migrations.AddField(
model_name='application',
name='mcp_enable',
field=models.BooleanField(default=False, verbose_name='MCP否启用'),
),
migrations.AddField(
model_name='application',
name='mcp_servers',
field=models.JSONField(default=dict, verbose_name='MCP服务列表'),
),
migrations.AddField(
model_name='application',
name='mcp_source',
field=models.CharField(default='referencing', max_length=20, verbose_name='MCP Source'),
),
migrations.AddField(
model_name='application',
name='mcp_tool_ids',
field=models.JSONField(default=list, verbose_name='MCP工具ID列表'),
),
migrations.AddField(
model_name='application',
name='tool_enable',
field=models.BooleanField(default=False, verbose_name='工具是否启用'),
),
migrations.AddField(
model_name='application',
name='tool_ids',
field=models.JSONField(default=list, verbose_name='工具ID列表'),
),
migrations.AddField(
model_name='applicationversion',
name='mcp_enable',
field=models.BooleanField(default=False, verbose_name='MCP否启用'),
),
migrations.AddField(
model_name='applicationversion',
name='mcp_servers',
field=models.JSONField(default=dict, verbose_name='MCP服务列表'),
),
migrations.AddField(
model_name='applicationversion',
name='mcp_source',
field=models.CharField(default='referencing', max_length=20, verbose_name='MCP Source'),
),
migrations.AddField(
model_name='applicationversion',
name='mcp_tool_ids',
field=models.JSONField(default=list, verbose_name='MCP工具ID列表'),
),
migrations.AddField(
model_name='applicationversion',
name='tool_enable',
field=models.BooleanField(default=False, verbose_name='工具是否启用'),
),
migrations.AddField(
model_name='applicationversion',
name='tool_ids',
field=models.JSONField(default=list, verbose_name='工具ID列表'),
),
]

12
apps/application/models/application.py
View File

@ -93,6 +93,12 @@ class Application(AppModelMixin):
publish_time = models.DateTimeField(verbose_name="发布时间", default=None, null=True, blank=True)
file_upload_enable = models.BooleanField(verbose_name="文件上传是否启用", default=False)
file_upload_setting = models.JSONField(verbose_name="文件上传相关设置", default=dict)
mcp_enable = models.BooleanField(verbose_name="MCP否启用", default=False)
mcp_tool_ids = models.JSONField(verbose_name="MCP工具ID列表", default=list)
mcp_servers = models.JSONField(verbose_name="MCP服务列表", default=dict)
mcp_source = models.CharField(verbose_name="MCP Source", max_length=20, default="referencing")
tool_enable = models.BooleanField(verbose_name="工具是否启用", default=False)
tool_ids = models.JSONField(verbose_name="工具ID列表", default=list)
@staticmethod
def get_default_model_prompt():
@ -158,6 +164,12 @@ class ApplicationVersion(AppModelMixin):
clean_time = models.IntegerField(verbose_name="清理时间", default=180)
file_upload_enable = models.BooleanField(verbose_name="文件上传是否启用", default=False)
file_upload_setting = models.JSONField(verbose_name="文件上传相关设置", default=dict)
mcp_enable = models.BooleanField(verbose_name="MCP否启用", default=False)
mcp_tool_ids = models.JSONField(verbose_name="MCP工具ID列表", default=list)
mcp_servers = models.JSONField(verbose_name="MCP服务列表", default=dict)
mcp_source = models.CharField(verbose_name="MCP Source", max_length=20, default="referencing")
tool_enable = models.BooleanField(verbose_name="工具是否启用", default=False)
tool_ids = models.JSONField(verbose_name="工具ID列表", default=list)
class Meta:
db_table = "application_version"

3
apps/application/serializers/application.py
View File

@ -706,6 +706,8 @@ class ApplicationOperateSerializer(serializers.Serializer):
'stt_model_enable': 'stt_model_enable', 'tts_type': 'tts_type',
'tts_autoplay': 'tts_autoplay', 'stt_autosend': 'stt_autosend', 'file_upload_enable': 'file_upload_enable',
'file_upload_setting': 'file_upload_setting',
'mcp_enable': 'mcp_enable', 'mcp_tool_ids': 'mcp_tool_ids', 'mcp_servers': 'mcp_servers',
'mcp_source': 'mcp_source', 'tool_enable': 'tool_enable', 'tool_ids': 'tool_ids',
'type': 'type'
}
@ -829,6 +831,7 @@ class ApplicationOperateSerializer(serializers.Serializer):
'stt_model_id', 'tts_model_id', 'tts_model_enable', 'stt_model_enable', 'tts_type',
'tts_autoplay', 'stt_autosend', 'file_upload_enable', 'file_upload_setting',
'api_key_is_active', 'icon', 'work_flow', 'model_params_setting', 'tts_model_params_setting',
'mcp_enable', 'mcp_tool_ids', 'mcp_servers', 'mcp_source', 'tool_enable', 'tool_ids',
'problem_optimization_prompt', 'clean_time', 'folder_id']
for update_key in update_keys:
if update_key in instance and instance.get(update_key) is not None:

8
apps/application/serializers/common.py
View File

@ -154,7 +154,13 @@ class ChatInfo:
'search_mode': self.application.knowledge_setting.get('search_mode') or 'embedding',
'no_references_setting': self.get_no_references_setting(self.application.knowledge_setting, model_setting),
'workspace_id': self.application.workspace_id,
'application_id': self.application.id
'application_id': self.application.id,
'mcp_enable': self.application.mcp_enable,
'mcp_tool_ids': self.application.mcp_tool_ids,
'mcp_servers': self.application.mcp_servers,
'mcp_source': self.application.mcp_source,
'tool_enable': self.application.tool_enable,
'tool_ids': self.application.tool_ids,
}
def to_pipeline_manage_params(self, problem_text: str, post_response_handler: PostResponseHandler,

6
ui/src/api/type/application.ts
View File

@ -26,6 +26,12 @@ interface ApplicationFormType {
stt_autosend?: boolean
folder_id?: string
workspace_id?: string
mcp_enable?: boolean
mcp_servers?: string
mcp_tool_ids?: string[]
mcp_source?: string
tool_enable?: boolean
tool_ids?: string[]
}
interface Chunk {
real_node_id: string

199
ui/src/views/application/ApplicationSetting.vue
View File

@ -273,6 +273,111 @@
@submitDialog="submitPrologueDialog"
/>
</el-form-item>
<!-- MCP-->
<el-form-item @click.prevent>
<template #label>
<div class="flex-between">
<span>MCP</span>
<div class="flex">
<el-button
type="primary"
link
@click="openMcpServersDialog"
@refreshForm="refreshParam"
v-if="applicationForm.mcp_enable"
>
<AppIcon iconName="app-setting"></AppIcon>
</el-button>
<el-switch class="ml-8" size="small" v-model="applicationForm.mcp_enable" />
</div>
</div>
</template>
</el-form-item>
<div
class="w-full mb-16"
v-if="applicationForm.mcp_tool_ids &&
applicationForm.mcp_tool_ids.length > 0 || (applicationForm.mcp_servers && applicationForm.mcp_servers.length > 0)
"
>
<template v-for="(item, index) in applicationForm.mcp_tool_ids" :key="index">
<div class="flex-between border border-r-6 white-bg mb-4" style="padding: 5px 8px"
v-if="relatedObject(mcpToolSelectOptions, item, 'id')"
>
<div class="flex align-center" style="line-height: 20px">
<el-avatar
v-if="relatedObject(mcpToolSelectOptions, item, 'id')?.icon"
shape="square"
:size="20"
style="background: none"
class="mr-8"
>
<img :src="resetUrl(relatedObject(mcpToolSelectOptions, item, 'id')?.icon)" alt="" />
</el-avatar>
<ToolIcon v-else type="MCP" class="mr-8" :size="20" />
<div
class="ellipsis"
:title="relatedObject(mcpToolSelectOptions, item, 'id')?.name"
>
{{
relatedObject(mcpToolSelectOptions, item, 'id')?.name ||
$t('common.custom') + ' MCP'
}}
</div>
</div>
<el-button text @click="removeMcpTool(item)">
<el-icon><Close /></el-icon>
</el-button>
</div>
</template>
</div>
<!-- 工具 -->
<el-form-item @click.prevent>
<template #label>
<div class="flex-between">
<span class="mr-4">
{{ $t('views.tool.title') }}
</span>
<div class="flex">
<el-button
type="primary"
link
@click="openToolDialog"
@refreshForm="refreshParam"
v-if="applicationForm.tool_enable"
>
<AppIcon iconName="app-setting"></AppIcon>
</el-button>
<el-switch class="ml-8" size="small" v-model="applicationForm.tool_enable" />
</div>
</div>
</template>
</el-form-item>
<div class="w-full mb-16" v-if="applicationForm.tool_ids && applicationForm.tool_ids.length > 0">
<template v-for="(item, index) in applicationForm.tool_ids" :key="index">
<div class="flex-between border border-r-6 white-bg mb-4" style="padding: 5px 8px">
<div class="flex align-center" style="line-height: 20px">
<el-avatar
v-if="relatedObject(toolSelectOptions, item, 'id')?.icon"
shape="square"
:size="20"
style="background: none"
class="mr-8"
>
<img :src="resetUrl(relatedObject(toolSelectOptions, item, 'id')?.icon)" alt="" />
</el-avatar>
<ToolIcon v-else class="mr-8" :size="20" />
<div class="ellipsis" :title="relatedObject(toolSelectOptions, item, 'id')?.name">
{{ relatedObject(toolSelectOptions, item, 'id')?.name }}
</div>
</div>
<el-button text @click="removeTool(item)">
<el-icon><Close /></el-icon>
</el-button>
</div>
</template>
</div>
<el-form-item @click.prevent>
<template #label>
<div class="flex-between">
@ -436,12 +541,14 @@
ref="ReasoningParamSettingDialogRef"
@refresh="submitReasoningDialog"
/>
<McpServersDialog ref="mcpServersDialogRef" @refresh="submitMcpServersDialog" />
<ToolDialog ref="toolDialogRef" @refresh="submitToolDialog" />
</div>
</template>
<script setup lang="ts">
import { reactive, ref, onMounted, computed, onBeforeMount } from 'vue'
import { useRoute, useRouter } from 'vue-router'
import { groupBy } from 'lodash'
import { groupBy, set } from 'lodash'
import AIModeParamSettingDialog from './component/AIModeParamSettingDialog.vue'
import ParamSettingDialog from './component/ParamSettingDialog.vue'
import AddKnowledgeDialog from './component/AddKnowledgeDialog.vue'
@ -456,6 +563,9 @@ import permissionMap from '@/permission'
import { EditionConst } from '@/utils/permission/data'
import { hasPermission } from '@/utils/permission/index'
import { loadSharedApi } from '@/utils/dynamics-api/shared-api'
import { resetUrl } from "@/utils/common.ts";
import McpServersDialog from "@/views/application/component/McpServersDialog.vue";
import ToolDialog from "@/views/application/component/ToolDialog.vue";
const route = useRoute()
const router = useRouter()
const {
@ -528,6 +638,12 @@ const applicationForm = ref<ApplicationFormType>({
tts_model_enable: false,
tts_type: 'BROWSER',
type: 'SIMPLE',
mcp_enable: false,
mcp_tool_ids: [],
mcp_servers: '',
mcp_source: 'referencing',
tool_enable: false,
tool_ids: [],
})
const themeDetail = ref({})
@ -628,6 +744,85 @@ const openParamSettingDialog = () => {
ParamSettingDialogRef.value?.open(applicationForm.value)
}
function removeTool(id: any) {
applicationForm.value.tool_ids = applicationForm.value.tool_ids.filter((v: any) => v !== id)
}
function removeMcpTool(id: any) {
applicationForm.value.mcp_tool_ids = applicationForm.value.mcp_tool_ids.filter((v: any) => v !== id)
}
const mcpServersDialogRef = ref()
function openMcpServersDialog() {
const config = {
mcp_servers: applicationForm.value.mcp_servers,
mcp_tool_ids: applicationForm.value.mcp_tool_ids,
mcp_source: applicationForm.value.mcp_source,
}
mcpServersDialogRef.value.open(config, mcpToolSelectOptions.value)
}
function submitMcpServersDialog(config: any) {
applicationForm.value.mcp_servers = config.mcp_servers
applicationForm.value.mcp_tool_ids = config.mcp_tool_ids
applicationForm.value.mcp_source = config.mcp_source
}
const toolDialogRef = ref()
function openToolDialog() {
toolDialogRef.value.open(applicationForm.value.tool_ids)
}
function submitToolDialog(config: any) {
applicationForm.value.tool_ids = config.tool_ids
}
const toolSelectOptions = ref<any[]>([])
function getToolSelectOptions() {
const obj =
apiType.value === 'systemManage'
? {
scope: 'WORKSPACE',
tool_type: 'CUSTOM',
workspace_id: application.value?.workspace_id,
}
: {
scope: 'WORKSPACE',
tool_type: 'CUSTOM',
}
loadSharedApi({ type: 'tool', systemType: apiType.value })
.getAllToolList(obj)
.then((res: any) => {
toolSelectOptions.value = [...res.data.shared_tools, ...res.data.tools].filter(
(item: any) => item.is_active,
)
})
}
const mcpToolSelectOptions = ref<any[]>([])
function getMcpToolSelectOptions() {
const obj =
apiType.value === 'systemManage'
? {
scope: 'WORKSPACE',
tool_type: 'MCP',
workspace_id: application.value?.workspace_id,
}
: {
scope: 'WORKSPACE',
tool_type: 'MCP',
}
loadSharedApi({ type: 'tool', systemType: apiType.value })
.getAllToolList(obj)
.then((res: any) => {
mcpToolSelectOptions.value = [...res.data.shared_tools, ...res.data.tools].filter(
(item: any) => item.is_active,
)
})
}
function refreshParam(data: any) {
applicationForm.value = { ...applicationForm.value, ...data }
}
@ -783,6 +978,8 @@ onMounted(() => {
getDetail()
getSTTModel()
getTTSModel()
getToolSelectOptions()
getMcpToolSelectOptions()
})
</script>
<style lang="scss" scoped>