Function useForceRefreshMcpTools

apps/sim/hooks/queries/mcp.ts:180–214

()

}

export function useForceRefreshMcpTools() {
  const queryClient = useQueryClient()

  return useMutation({
    mutationFn: async (workspaceId: string) => {
      const allServers =
        queryClient.getQueryData<McpServer[]>(mcpKeys.serversList(workspaceId)) ?? []
      const servers = allServers.filter((s) => s.enabled && s.workspaceId === workspaceId)
      const results = await Promise.allSettled(
        servers.map(async (server) => {
          const tools = await fetchMcpTools(workspaceId, true, undefined, server.id)
          queryClient.setQueryData(mcpKeys.serverToolsList(workspaceId, server.id), tools)
          return tools
        })
      )
      results.forEach((result, index) => {
        if (result.status === 'rejected') {
          const failedServer = servers[index]
          if (failedServer) {
            queryClient.invalidateQueries({
              queryKey: mcpKeys.serverToolsList(workspaceId, failedServer.id),
            })
          }
        }
      })
      return results
        .filter((r): r is PromiseFulfilledResult<McpTool[]> => r.status === 'fulfilled')
        .flatMap((r) => r.value)
    },
    onSettled: (_data, _error, workspaceId) => {
      queryClient.invalidateQueries({ queryKey: mcpKeys.serversList(workspaceId) })
      queryClient.invalidateQueries({ queryKey: mcpKeys.storedToolsList(workspaceId) })
    },
  })
}

interface CreateMcpServerParams {
  workspaceId: string