- Add 'Partner Admin' role to actual functionality in eveai_app

eveai_app/views/user_forms.py

@@ -1,9 +1,10 @@
-from flask import current_app
+from flask import current_app, session
 from flask_wtf import FlaskForm
 from wtforms import (StringField, PasswordField, BooleanField, SubmitField, EmailField, IntegerField, DateField,
                      SelectField, SelectMultipleField, FieldList, FormField, FloatField, TextAreaField)
 from wtforms.validators import DataRequired, Length, Email, NumberRange, Optional, ValidationError
 import pytz
+from flask_security import current_user
 
 from common.models.user import Role
 from common.services.user_service import UserService
@@ -24,6 +25,9 @@ class TenantForm(FlaskForm):
     timezone = SelectField('Timezone', choices=[], validators=[DataRequired()])
     # LLM fields
     llm_model = SelectField('Large Language Model', choices=[], validators=[DataRequired()])
+
+    # For Super Users only - Allow to assign the tenant to the partner
+    assign_to_partner = BooleanField('Assign to Partner', default=False)
     # Embedding variables
     submit = SubmitField('Submit')
 
@@ -40,6 +44,9 @@ class TenantForm(FlaskForm):
         self.llm_model.choices = [(model, model) for model in current_app.config['SUPPORTED_LLMS']]
         # Initialize fallback algorithms
         self.type.choices = [(t, t) for t in current_app.config['TENANT_TYPES']]
+        # Show field only for Super Users with partner in session
+        if not current_user.has_roles('Super User') or 'partner' not in session:
+            self._fields.pop('assign_to_partner', None)
 
 
 class BaseUserForm(FlaskForm):