The Afghan government has agreed on a transition plan to take over responsibility for the US-run prison at Bagram air base, where there have been allegations of human rights abuses.