diff --git a/ayushma/migrations/0042_alter_project_preset_questions.py b/ayushma/migrations/0042_alter_project_preset_questions.py new file mode 100644 index 00000000..eebe7f35 --- /dev/null +++ b/ayushma/migrations/0042_alter_project_preset_questions.py @@ -0,0 +1,20 @@ +# Generated by Django 4.2.1 on 2023-09-04 13:02 + +import django.contrib.postgres.fields +from django.db import migrations, models + + +class Migration(migrations.Migration): + dependencies = [ + ("ayushma", "0041_merge_20230904_0908"), + ] + + operations = [ + migrations.AlterField( + model_name="project", + name="preset_questions", + field=django.contrib.postgres.fields.ArrayField( + base_field=models.TextField(), blank=True, null=True, size=None + ), + ), + ] diff --git a/ayushma/models/project.py b/ayushma/models/project.py index 83e0947b..b97c29c2 100644 --- a/ayushma/models/project.py +++ b/ayushma/models/project.py @@ -16,7 +16,7 @@ class Project(BaseModel): choices=STTEngine.choices, default=STTEngine.WHISPER ) model = models.IntegerField(choices=ModelType.choices, default=ModelType.GPT_3_5) - preset_questions = ArrayField(models.TextField(), null=True, blank=True, size=100) + preset_questions = ArrayField(models.TextField(), null=True, blank=True) is_default = models.BooleanField(default=False) archived = models.BooleanField(default=False) diff --git a/ayushma/serializers/project.py b/ayushma/serializers/project.py index f6a32523..12e3d185 100644 --- a/ayushma/serializers/project.py +++ b/ayushma/serializers/project.py @@ -1,9 +1,19 @@ +from random import sample + from rest_framework import serializers from ayushma.models import Project class ProjectSerializer(serializers.ModelSerializer): + display_preset_questions = serializers.SerializerMethodField() + + def get_display_preset_questions(self, project_object): + if project_object.preset_questions and len(project_object.preset_questions) > 4: + return sample(project_object.preset_questions, 4) + else: + return project_object.preset_questions + class Meta: model = Project fields = ( @@ -15,22 +25,39 @@ class Meta: "stt_engine", "model", "is_default", + "display_preset_questions", "archived", - "preset_questions", ) - read_only_fields = ("external_id", "created_at", "modified_at") + read_only_fields = ( + "external_id", + "created_at", + "modified_at", + "display_preset_questions", + ) class ProjectUpdateSerializer(serializers.ModelSerializer): + display_preset_questions = serializers.SerializerMethodField() key_set = serializers.SerializerMethodField() + def get_display_preset_questions(self, project_object): + if project_object.preset_questions and len(project_object.preset_questions) > 4: + return sample(project_object.preset_questions, 4) + else: + return project_object.preset_questions + class Meta: model = Project - fields = ProjectSerializer.Meta.fields + ("prompt", "open_ai_key", "key_set") + fields = ProjectSerializer.Meta.fields + ( + "prompt", + "open_ai_key", + "key_set", + "preset_questions", + ) extra_kwargs = { "open_ai_key": {"write_only": True}, } - read_only_fields = ("key_set",) + read_only_fields = ("key_set", "display_preset_questions") def update(self, instance, validated_data): if validated_data.get("is_default", True):