Ark-kun commited on
Commit
572fe9c
·
1 Parent(s): 0ba6c0b

fix: MT - Fixed the update_tenant_info_in_db function

Browse files
huggingface_overlay/start_HuggingFace_multi_tenant.py CHANGED
@@ -632,6 +632,7 @@ def update_tenant_info_in_db(request: fastapi.Request) -> TenantRow:
632
  name=oauth_user_info["preferred_username"],
633
  access_token=oauth_info["access_token"],
634
  )
 
635
  tenant_row.oauth_info = oauth_info
636
  extra_data = tenant_row.extra_data or {}
637
  extra_data["huggingface_user_info"] = huggingface_user_info
 
632
  name=oauth_user_info["preferred_username"],
633
  access_token=oauth_info["access_token"],
634
  )
635
+ session.add(tenant_row)
636
  tenant_row.oauth_info = oauth_info
637
  extra_data = tenant_row.extra_data or {}
638
  extra_data["huggingface_user_info"] = huggingface_user_info