State Based Mode: change the detection of primary_key

[dkharytonenko-blackbird]

Description

Change the detection methods for primary_key in state_based mode

Fixes # (issue)

Bug in DB updates according to primary_key detection

Type of change

• [ ] Bug fix (non-breaking change which fixes an issue)