Changes Documents - llm and languagefields on tenant, processing on documents
first version of Adding Documents (excl. embeddings)
This commit is contained in:
30
eveai_app/views/basic_forms.py
Normal file
30
eveai_app/views/basic_forms.py
Normal file
@@ -0,0 +1,30 @@
|
||||
from flask import session
|
||||
from flask_security import current_user
|
||||
from flask_wtf import FlaskForm
|
||||
from wtforms import StringField, SelectField
|
||||
from wtforms.validators import DataRequired
|
||||
|
||||
|
||||
class SessionDefaultsForm(FlaskForm):
|
||||
user_name = StringField('User Name', validators=[DataRequired()])
|
||||
user_email = StringField('User', validators=[DataRequired()])
|
||||
|
||||
# Tenant Defaults
|
||||
tenant_name = StringField('Tenant Name', validators=[DataRequired()])
|
||||
default_language = SelectField(u'Default Language', choices=[], validators=[DataRequired()])
|
||||
default_embedding_model = SelectField(u'Default Embedding Model', choices=[], validators=[DataRequired()])
|
||||
default_llm_model = SelectField(u'Default LLM Model', choices=[], validators=[DataRequired()])
|
||||
|
||||
def __init__(self):
|
||||
super(SessionDefaultsForm, self).__init__()
|
||||
self.user_name.data = current_user.user_name
|
||||
self.user_email.data = current_user.email
|
||||
self.tenant_name.data = session.get('tenant').get('name')
|
||||
self.default_language.choices = [(lang, lang.lower()) for lang in
|
||||
session.get('tenant').get('allowed_languages')]
|
||||
self.default_language.data = session.get('default_language')
|
||||
self.default_embedding_model.choices = [(model, model) for model in
|
||||
session.get('tenant').get('allowed_embedding_models')]
|
||||
self.default_embedding_model.data = session.get('default_embedding_model')
|
||||
self.default_llm_model.choices = [(model, model) for model in session.get('tenant').get('allowed_llm_models')]
|
||||
self.default_llm_model.data = session.get('default_llm_model')
|
||||
Reference in New Issue
Block a user