Correcting the retrieval of relevant documents

eveai_app/views/user_forms.py

@@ -3,6 +3,8 @@ from flask_wtf import FlaskForm
 from wtforms import (StringField, PasswordField, BooleanField, SubmitField, EmailField, IntegerField, DateField,
                      SelectField, SelectMultipleField, FieldList, FormField, FloatField)
 from wtforms.validators import DataRequired, Length, Email, NumberRange, Optional
+import pytz
+
 from common.models.user import Role
 
 
@@ -12,6 +14,8 @@ class TenantForm(FlaskForm):
     # language fields
     default_language = SelectField('Default Language', choices=[], validators=[DataRequired()])
     allowed_languages = SelectMultipleField('Allowed Languages', choices=[], validators=[DataRequired()])
+    # Timezone
+    timezone = SelectField('Timezone', choices=[], validators=[DataRequired()])
     # LLM fields
     embedding_model = SelectField('Embedding Model', choices=[], validators=[DataRequired()])
     llm_model = SelectField('Large Language Model', choices=[], validators=[DataRequired()])
@@ -44,6 +48,8 @@ class TenantForm(FlaskForm):
         # initialise language fields
         self.default_language.choices = [(lang, lang.lower()) for lang in current_app.config['SUPPORTED_LANGUAGES']]
         self.allowed_languages.choices = [(lang, lang.lower()) for lang in current_app.config['SUPPORTED_LANGUAGES']]
+        # initialise timezone
+        self.timezone.choices = [(tz, tz) for tz in pytz.all_timezones]
         # initialise LLM fields
         self.embedding_model.choices = [(model, model) for model in current_app.config['SUPPORTED_EMBEDDINGS']]
         self.llm_model.choices = [(model, model) for model in current_app.config['SUPPORTED_LLMS']]

eveai_app/views/user_views.py

@@ -49,6 +49,7 @@ def tenant():
                             website=form.website.data,
                             default_language=form.default_language.data,
                             allowed_languages=form.allowed_languages.data,
+                            timezone=form.timezone.data,
                             embedding_model=form.embedding_model.data,
                             llm_model=form.llm_model.data,
                             license_start_date=form.license_start_date.data,