- Remove embedding model from Catalog. We use Mistral's embedding.

This commit is contained in:
Josako
2025-03-07 15:06:51 +01:00
parent 9e03af45e1
commit e088ef7e4e
3 changed files with 29 additions and 7 deletions

View File

@@ -32,8 +32,6 @@ class CatalogForm(FlaskForm):
type = SelectField('Catalog Type', validators=[DataRequired()])
# Selection fields for processing & creating embeddings
embedding_model = SelectField('Embedding Model', choices=[], validators=[DataRequired()])
min_chunk_size = IntegerField('Minimum Chunk Size (1500)', validators=[NumberRange(min=0), Optional()],
default=1500)
max_chunk_size = IntegerField('Maximum Chunk Size (2500)', validators=[NumberRange(min=0), Optional()],
@@ -47,7 +45,6 @@ class CatalogForm(FlaskForm):
super().__init__(*args, **kwargs)
# Dynamically populate the 'type' field using the constructor
self.type.choices = [(key, value['name']) for key, value in CATALOG_TYPES.items()]
self.embedding_model.choices = [(model, model) for model in current_app.config['SUPPORTED_EMBEDDINGS']]
class EditCatalogForm(DynamicFormBase):
@@ -58,8 +55,6 @@ class EditCatalogForm(DynamicFormBase):
type = StringField('Catalog Type', validators=[DataRequired()], render_kw={'readonly': True})
# Selection fields for processing & creating embeddings
embedding_model = StringField('Embedding Model', validators=[DataRequired()], render_kw={'readonly': True})
min_chunk_size = IntegerField('Minimum Chunk Size (2000)', validators=[NumberRange(min=0), Optional()],
default=2000)
max_chunk_size = IntegerField('Maximum Chunk Size (3000)', validators=[NumberRange(min=0), Optional()],