diff --git a/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py b/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py index 21afade51..8511d3ce5 100644 --- a/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py +++ b/apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py @@ -270,7 +270,6 @@ class BaseChatStep(IChatStep): for application_id in application_ids: app = QuerySet(Application).filter(id=application_id).first() app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first() - # TODO 处理api if app_key is not None: api_key = app_key.secret_key else: diff --git a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py index f71192b6c..cc646c642 100644 --- a/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py +++ b/apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py @@ -257,14 +257,13 @@ class BaseChatNode(IChatNode): for application_id in application_ids: app = QuerySet(Application).filter(id=application_id).first() app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first() - # TODO 处理api if app_key is not None: api_key = app_key.secret_key else: continue executor = ToolExecutor() app_config = executor.get_app_mcp_config(api_key) - mcp_servers_config[str(app.id)] = app_config + mcp_servers_config[app.name] = app_config if len(mcp_servers_config) > 0: r = mcp_response_generator(chat_model, message_list, json.dumps(mcp_servers_config), mcp_output_enable)