I think that it could be simpler to implement the bra, operator, and ket vertex logic in form networks with two layers of partitioned graphs. The outer layer would handle the BP-like partitioned, and then each of those partitions could be partitioned into bra, operator, and ket vertices.
@JoeyT1994