use correct invitation id

Josh-XT · Jan 25, 2025 · dc203cb · dc203cb
1 parent e9e79bb
commit dc203cb
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/agixt/MagicalAuth.py b/agixt/MagicalAuth.py
@@ -1740,7 +1740,9 @@ def create_invitation(self, invitation: InvitationCreate) -> InvitationResponse:
                         email=invitation.email, otp=pyotp.TOTP(user.mfa_token).now()
                     )
                     if company_id is not None:
-                        default_agent["settings"]["company_id"] = str(company_id)
+                        default_agent["settings"]["company_id"] = str(
+                            invitation.company_id
+                        )
                     agixt.add_agent(
                         agent_name=company.agent_name,
                         settings=default_agent["settings"],