in genai-on-vertex-ai/gemini/evals_playbook/utils/evals_playbook.py [0:0]
def get_db_object(table_class):
table_name, update_keys = get_table_name_keys(table_class)
update_key_cols = [Column(key, String, primary_key=True) for key in update_keys]
return table_name, update_key_cols