MOD assign context company_id in partner creation

energy_communities/models/res_partner.py

@@ -9,6 +9,15 @@ class ResPartner(models.Model):
 
     gender = fields.Selection(selection_add=[("not_binary", "Not binary"),
                                              ("not_share", "I prefer to not share it")])
+    @api.model
+    def create(self, vals):
+        current_company = self.env.company
+        if self.env.user not in (self.env.ref("base.user_root"), self.env.ref("base.user_admin")):
+            if vals.get('company_ids', False):
+                vals['company_ids'][0][-1].append(current_company.id)
+
+        new_partner = super(ResPartner, self).create(vals)
+        return new_partner
 
     def cron_update_company_ids_from_user(self):
         partner_with_users = self.search([('user_ids', '!=', False), ('user_ids.id', '!=', SUPERUSER_ID)])