Allow agent personality to be nullable, in which case the default prompt will be used.

This commit is contained in:
sabaimran
2025-01-24 08:24:53 -08:00
parent 2b9a61c987
commit 43c9ec260d
2 changed files with 72 additions and 1 deletions

View File

@@ -0,0 +1,71 @@
# Generated by Django 5.0.10 on 2025-01-24 16:23
import django.contrib.postgres.fields
from django.db import migrations, models
class Migration(migrations.Migration):
dependencies = [
("database", "0082_merge_20250121_1842"),
]
operations = [
migrations.AlterField(
model_name="agent",
name="output_modes",
field=django.contrib.postgres.fields.ArrayField(
base_field=models.CharField(
choices=[("image", "Image"), ("automation", "Automation"), ("diagram", "Diagram")], max_length=200
),
blank=True,
default=list,
null=True,
size=None,
),
),
migrations.AlterField(
model_name="agent",
name="personality",
field=models.TextField(blank=True, default=None, null=True),
),
migrations.AlterField(
model_name="agent",
name="style_color",
field=models.CharField(
choices=[
("blue", "Blue"),
("green", "Green"),
("red", "Red"),
("yellow", "Yellow"),
("orange", "Orange"),
("purple", "Purple"),
("pink", "Pink"),
("teal", "Teal"),
("cyan", "Cyan"),
("lime", "Lime"),
("indigo", "Indigo"),
("fuchsia", "Fuchsia"),
("rose", "Rose"),
("sky", "Sky"),
("amber", "Amber"),
("emerald", "Emerald"),
],
default="orange",
max_length=200,
),
),
migrations.AlterField(
model_name="processlock",
name="name",
field=models.CharField(
choices=[
("index_content", "Index Content"),
("scheduled_job", "Scheduled Job"),
("schedule_leader", "Schedule Leader"),
("apply_migrations", "Apply Migrations"),
],
max_length=200,
unique=True,
),
),
]

View File

@@ -293,7 +293,7 @@ class Agent(DbBaseModel):
KhojUser, on_delete=models.CASCADE, default=None, null=True, blank=True
) # Creator will only be null when the agents are managed by admin
name = models.CharField(max_length=200)
personality = models.TextField()
personality = models.TextField(default=None, null=True, blank=True)
input_tools = ArrayField(
models.CharField(max_length=200, choices=InputToolOptions.choices), default=list, null=True, blank=True
)