- cleanup of old TASKs, AGENTs and SPECIALISTs
- Add additional configuration options to agent (temperature and model choice) - Define new PROOFREADING Agents and Tasks
This commit is contained in:
@@ -2,8 +2,8 @@ from flask import session
|
||||
from flask_wtf import FlaskForm
|
||||
from wtforms import (StringField, BooleanField, SelectField, TextAreaField)
|
||||
from wtforms.fields.datetime import DateField
|
||||
from wtforms.fields.numeric import IntegerField
|
||||
from wtforms.validators import DataRequired, Length, Optional
|
||||
from wtforms.fields.numeric import IntegerField, FloatField
|
||||
from wtforms.validators import DataRequired, Length, Optional, NumberRange
|
||||
|
||||
from wtforms_sqlalchemy.fields import QuerySelectMultipleField
|
||||
|
||||
@@ -91,6 +91,16 @@ class EditEveAIAgentForm(BaseEditComponentForm):
|
||||
role = TextAreaField('Role', validators=[Optional()])
|
||||
goal = TextAreaField('Goal', validators=[Optional()])
|
||||
backstory = TextAreaField('Backstory', validators=[Optional()])
|
||||
temperature = FloatField('Temperature', validators=[Optional(), NumberRange(min=0, max=1)])
|
||||
llm_model = SelectField('LLM Model', validators=[Optional()])
|
||||
|
||||
def __init__(self, *args, **kwargs):
|
||||
super().__init__(*args, **kwargs)
|
||||
agent_config = cache_manager.agents_config_cache.get_config(self.type, self.type_version)
|
||||
if agent_config.get('allowed_models', None):
|
||||
self.llm_model.choices = agent_config.allowed_models
|
||||
else:
|
||||
self.llm_model.choices = agent_config.get('full_model_name', 'mistral.mistral-medium-latest')
|
||||
|
||||
|
||||
class EditEveAITaskForm(BaseEditComponentForm):
|
||||
|
||||
Reference in New Issue
Block a user