Skip to content

Commit dfbf5aa

Browse files
committed
chore: remove TODO comments and update mcp_servers_config key to use app name
1 parent e728647 commit dfbf5aa

File tree

2 files changed

+1
-3
lines changed

2 files changed

+1
-3
lines changed

apps/application/chat_pipeline/step/chat_step/impl/base_chat_step.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -270,7 +270,6 @@ def _handle_mcp_request(self, mcp_enable, tool_enable, mcp_source, mcp_servers,
270270
for application_id in application_ids:
271271
app = QuerySet(Application).filter(id=application_id).first()
272272
app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first()
273-
# TODO 处理api
274273
if app_key is not None:
275274
api_key = app_key.secret_key
276275
else:

apps/application/flow/step_node/ai_chat_step_node/impl/base_chat_node.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -257,14 +257,13 @@ def _handle_mcp_request(self, mcp_enable, tool_enable, mcp_source, mcp_servers,
257257
for application_id in application_ids:
258258
app = QuerySet(Application).filter(id=application_id).first()
259259
app_key = QuerySet(ApplicationApiKey).filter(application_id=application_id, is_active=True).first()
260-
# TODO 处理api
261260
if app_key is not None:
262261
api_key = app_key.secret_key
263262
else:
264263
continue
265264
executor = ToolExecutor()
266265
app_config = executor.get_app_mcp_config(api_key)
267-
mcp_servers_config[str(app.id)] = app_config
266+
mcp_servers_config[app.name] = app_config
268267

269268
if len(mcp_servers_config) > 0:
270269
r = mcp_response_generator(chat_model, message_list, json.dumps(mcp_servers_config), mcp_output_enable)

0 commit comments

Comments
 (0)