I would naively think that putting the bigger part of the battery in the trailer is a good idea. Now someone can tell me why it isn't.
Trailers come in many form factors: container, flat bet, end-dump. Heck, sometimes you just want to tow another tractor. Those different form factors have different constraints in terms of battery size and shape.
You'd still need big batteries in some tractor units so that they could drop off trailers. All tractor units would need small batteries so they could go short distances without a trailer.