Hi, you can do what you're describing.

By default, if you use Firefox tab groups, then each tab group will be stored separately, and can be restored separately.

If you want to store a whole window full of tab groups, and have the entire window restore at once with all of the tab groups inside, then you can do this by going to OneTab Options and choosing the "Send the current window to OneTab, including all tabs and tab groups" option.