diff --git a/api/server/services/MCP.js b/api/server/services/MCP.js index dbb44740a9..ccff184d4d 100644 --- a/api/server/services/MCP.js +++ b/api/server/services/MCP.js @@ -23,7 +23,7 @@ const { getFlowStateManager, getMCPManager, } = require('~/config'); -const { findToken, createToken, updateToken } = require('~/models'); +const { findToken, createToken, updateToken, deleteTokens } = require('~/models'); const { getGraphApiToken } = require('./GraphTokenService'); const { reinitMCPServer } = require('./Tools/mcp'); const { getAppConfig } = require('./Config'); @@ -644,6 +644,7 @@ function createToolInstance({ findToken, createToken, updateToken, + deleteTokens, }, oauthStart, oauthEnd, diff --git a/packages/api/src/mcp/MCPConnectionFactory.ts b/packages/api/src/mcp/MCPConnectionFactory.ts index 8fc0d87fa3..1f24fe5ec4 100644 --- a/packages/api/src/mcp/MCPConnectionFactory.ts +++ b/packages/api/src/mcp/MCPConnectionFactory.ts @@ -351,7 +351,7 @@ export class MCPConnectionFactory { config?.oauth_headers ?? {}, config?.oauth, this.allowedDomains, - this.tokenMethods?.findToken, + this.tokenMethods?.deleteTokens ? this.tokenMethods.findToken : undefined, ); if (existingFlow) {