Spaces:
Running
Running
| from django.db import models | |
| from django.urls import reverse | |
| from django.forms import ModelForm | |
| class Query(models.Model): | |
| query = models.TextField( | |
| max_length=1000, help_text="Unesite upit") | |
| reply = models.TextField( | |
| max_length=1000, help_text="Unesite odgovor") | |
| jezik = models.CharField( | |
| max_length=16, help_text="Odaberite jezik!") | |
| def get_absolute_url(self): | |
| return reverse('query', args=[str(self.id)]) | |
| def __str__(self): | |
| return self.text | |
| #class QueryForm(ModelForm): | |
| # class Meta: | |
| # model = Query | |
| # fields = ['query', 'reply', 'jezik'] | |
| class Language(models.Model): | |
| name = models.CharField(max_length=16, | |
| unique=True, | |
| default="srpski", | |
| help_text="Unesite jezik") | |
| lid = models.CharField(max_length=2, unique=False, default="sr", help_text="Unesite kratak naziv jezika") | |
| def get_absolute_url(self): | |
| """Returns the url to access a particular language instance.""" | |
| return reverse('bot-detail', args=[str(self.id)]) | |
| def __str__(self): | |
| """String for representing the Model object (in Admin site etc.)""" | |
| return self.name | |
| class LanguageForm(ModelForm): | |
| class Meta: | |
| model = Language | |
| fields = ['name', 'lid'] | |
| class Param(models.Model): | |
| system = models.TextField(max_length=2048, unique=False, blank=False, default="System prpmpt", help_text="Enter parameter value") | |
| ctxpre = models.CharField(max_length=80, unique=False, blank=True, help_text="Enter parameter value") | |
| msg_content = models.CharField(unique=False, blank=True, max_length=80, help_text="Enter parameter value") | |
| min_len = models.IntegerField(unique=False, blank=False, default=40, help_text="Enter parameter value") | |
| CHUNK_SIZE = models.IntegerField(unique=False, blank=False, default=800, help_text="Enter parameter value") | |
| CHUNK_OVERLAP = models.IntegerField(unique=False, blank=False, default=50, help_text="Enter parameter value") | |
| max_results = models.IntegerField(unique=False, blank=False, default=3, help_text="Enter parameter value") | |
| EMBED_MODEL = models.CharField(max_length=256, default="embeddings model", help_text="Enter parameter value") | |
| model_id = models.CharField(max_length=256, unique=False, default="LLM model", blank=False, help_text="Enter parameter value") | |
| max_conv = models.IntegerField(unique=False, blank=False, default=3, help_text="Enter parameter value") | |
| min_distance = models.FloatField(unique=False, blank=False, default=0.35, help_text="Enter parameter value") | |
| max_distance = models.FloatField(unique=False, blank=False, default=0.6, help_text="Enter parameter value") | |
| temperature = models.FloatField(unique=False, blank=False, default=0.55, help_text="Enter parameter value") | |
| max_tokens = models.IntegerField(unique=False, blank=False, default=3072, help_text="Enter parameter value") | |
| top_p = models.FloatField(unique=False, blank=False, default=0.8, help_text="Enter parameter value") | |
| frequency_penalty = models.FloatField(unique=False, default=0.0, blank=False, help_text="Enter parameter value") | |
| presence_penalty = models.FloatField(unique=False, default=0.0, blank=False, help_text="Enter parameter value") | |
| DEBUG = models.BooleanField(unique=False, blank=False, default=True, help_text="Enter parameter value") | |
| jezik = models.CharField(max_length=16, unique=False, blank=False, default="srpski", help_text="Uneti jezik") | |
| class ParamModelForm(ModelForm): | |
| class Meta: | |
| model = Param | |
| fields = [ 'system', 'ctxpre', 'msg_content', 'min_len', 'CHUNK_SIZE', 'CHUNK_OVERLAP', 'max_results', 'EMBED_MODEL', 'model_id', 'max_conv', 'min_distance', 'max_distance', 'temperature', 'max_tokens', 'top_p', 'frequency_penalty', 'presence_penalty', 'DEBUG', 'jezik'] | |