Current version of STEP import(step model) do not offer easy way to do this. You need to find STEP_Entity by ID(this operation has bad performance). I recommend to collect to own map after import. After that you be able to get TopoDS_Shape from session by STEP_Entity.
Best regards, Commenter-1.