Enable model variables & start working on RAG task

This commit is contained in:
Josako
2024-05-25 20:17:02 +02:00
parent e483d6cf90
commit ce91323dc9
15 changed files with 340 additions and 46 deletions

View File

@@ -1,7 +1,7 @@
from flask import current_app
from flask_wtf import FlaskForm
from wtforms import (StringField, PasswordField, BooleanField, SubmitField, EmailField, IntegerField, DateField,
SelectField, SelectMultipleField, FieldList, FormField)
SelectField, SelectMultipleField, FieldList, FormField, FloatField)
from wtforms.validators import DataRequired, Length, Email, NumberRange, Optional
from common.models.user import Role
@@ -26,6 +26,10 @@ class TenantForm(FlaskForm):
default='p, li')
html_included_elements = StringField('HTML Included Elements', validators=[Optional()])
html_excluded_elements = StringField('HTML Excluded Elements', validators=[Optional()])
# Embedding Search variables
es_k = IntegerField('Limit for Searching Embeddings (5)', validators=[NumberRange(min=0)])
es_similarity_threshold = FloatField('Similarity Threshold for Searching Embeddings (0.5)',
validators=[NumberRange(min=0, max=1)])
submit = SubmitField('Submit')