khoj 1.33.3.dev32__py3-none-any.whl

khoj/database/migrations/0049_texttoimagemodelconfig_api_key_and_more.py

@@ -0,0 +1,58 @@
+# Generated by Django 4.2.11 on 2024-06-26 03:27
+
+import django.db.models.deletion
+from django.conf import settings
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0048_voicemodeloption_uservoicemodelconfig"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="texttoimagemodelconfig",
+            name="api_key",
+            field=models.CharField(blank=True, default=None, max_length=200, null=True),
+        ),
+        migrations.AddField(
+            model_name="texttoimagemodelconfig",
+            name="openai_config",
+            field=models.ForeignKey(
+                blank=True,
+                default=None,
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                to="database.openaiprocessorconversationconfig",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="texttoimagemodelconfig",
+            name="model_type",
+            field=models.CharField(
+                choices=[("openai", "Openai"), ("stability-ai", "Stabilityai")], default="openai", max_length=200
+            ),
+        ),
+        migrations.CreateModel(
+            name="UserTextToImageModelConfig",
+            fields=[
+                ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                ("created_at", models.DateTimeField(auto_now_add=True)),
+                ("updated_at", models.DateTimeField(auto_now=True)),
+                (
+                    "setting",
+                    models.ForeignKey(
+                        on_delete=django.db.models.deletion.CASCADE, to="database.texttoimagemodelconfig"
+                    ),
+                ),
+                (
+                    "user",
+                    models.OneToOneField(on_delete=django.db.models.deletion.CASCADE, to=settings.AUTH_USER_MODEL),
+                ),
+            ],
+            options={
+                "abstract": False,
+            },
+        ),
+    ]

khoj/database/migrations/0050_alter_processlock_name.py

@@ -0,0 +1,25 @@
+# Generated by Django 4.2.11 on 2024-06-28 06:11
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0049_datastore"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="processlock",
+            name="name",
+            field=models.CharField(
+                choices=[
+                    ("index_content", "Index Content"),
+                    ("scheduled_job", "Scheduled Job"),
+                    ("schedule_leader", "Schedule Leader"),
+                ],
+                max_length=200,
+                unique=True,
+            ),
+        ),
+    ]

khoj/database/migrations/0051_merge_20240702_1220.py

@@ -0,0 +1,14 @@
+# Generated by Django 4.2.11 on 2024-07-02 12:20
+
+from typing import List
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0049_texttoimagemodelconfig_api_key_and_more"),
+        ("database", "0050_alter_processlock_name"),
+    ]
+
+    operations: List[str] = []

khoj/database/migrations/0052_alter_searchmodelconfig_bi_encoder_docs_encode_config_and_more.py

@@ -0,0 +1,27 @@
+# Generated by Django 4.2.11 on 2024-07-08 09:27
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0051_merge_20240702_1220"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="searchmodelconfig",
+            name="bi_encoder_docs_encode_config",
+            field=models.JSONField(blank=True, default=dict),
+        ),
+        migrations.AlterField(
+            model_name="searchmodelconfig",
+            name="bi_encoder_model_config",
+            field=models.JSONField(blank=True, default=dict),
+        ),
+        migrations.AlterField(
+            model_name="searchmodelconfig",
+            name="bi_encoder_query_encode_config",
+            field=models.JSONField(blank=True, default=dict),
+        ),
+    ]

khoj/database/migrations/0053_agent_style_color_agent_style_icon.py

@@ -0,0 +1,61 @@
+# Generated by Django 5.0.6 on 2024-07-13 16:26
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0052_alter_searchmodelconfig_bi_encoder_docs_encode_config_and_more"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            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"),
+                    ("fuschia", "Fuschia"),
+                    ("rose", "Rose"),
+                    ("sky", "Sky"),
+                    ("amber", "Amber"),
+                    ("emerald", "Emerald"),
+                ],
+                default="blue",
+                max_length=200,
+            ),
+        ),
+        migrations.AddField(
+            model_name="agent",
+            name="style_icon",
+            field=models.CharField(
+                choices=[
+                    ("Lightbulb", "Lighbulb"),
+                    ("Health", "Health"),
+                    ("Robot", "Robot"),
+                    ("Aperture", "Aperture"),
+                    ("GraduationCap", "Graduation Cap"),
+                    ("Jeep", "Jeep"),
+                    ("Island", "Island"),
+                    ("MathOperations", "Math Operations"),
+                    ("Asclepius", "Asclepius"),
+                    ("Couch", "Couch"),
+                    ("Code", "Code"),
+                    ("Atom", "Atom"),
+                    ("ClockCounterClockwise", "Clock Counter Clockwise"),
+                ],
+                default="Lightbulb",
+                max_length=200,
+            ),
+        ),
+    ]

khoj/database/migrations/0054_alter_agent_style_color.py

@@ -0,0 +1,38 @@
+# Generated by Django 5.0.7 on 2024-08-05 06:19
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0053_agent_style_color_agent_style_icon"),
+    ]
+
+    operations = [
+        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="blue",
+                max_length=200,
+            ),
+        ),
+    ]

khoj/database/migrations/0055_alter_agent_style_icon.py

@@ -0,0 +1,37 @@
+# Generated by Django 5.0.7 on 2024-08-05 11:51
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0054_alter_agent_style_color"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="agent",
+            name="style_icon",
+            field=models.CharField(
+                choices=[
+                    ("Lightbulb", "Lighbulb"),
+                    ("Health", "Health"),
+                    ("Robot", "Robot"),
+                    ("Aperture", "Aperture"),
+                    ("GraduationCap", "Graduation Cap"),
+                    ("Jeep", "Jeep"),
+                    ("Island", "Island"),
+                    ("MathOperations", "Math Operations"),
+                    ("Asclepius", "Asclepius"),
+                    ("Couch", "Couch"),
+                    ("Code", "Code"),
+                    ("Atom", "Atom"),
+                    ("ClockCounterClockwise", "Clock Counter Clockwise"),
+                    ("PencilLine", "Pencil Line"),
+                    ("Chalkboard", "Chalkboard"),
+                ],
+                default="Lightbulb",
+                max_length=200,
+            ),
+        ),
+    ]

khoj/database/migrations/0056_chatmodeloptions_vision_enabled.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.7 on 2024-08-14 19:43
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0055_alter_agent_style_icon"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="chatmodeloptions",
+            name="vision_enabled",
+            field=models.BooleanField(default=False),
+        ),
+    ]

khoj/database/migrations/0056_searchmodelconfig_cross_encoder_model_config.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.7 on 2024-08-07 09:12
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0055_alter_agent_style_icon"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="searchmodelconfig",
+            name="cross_encoder_model_config",
+            field=models.JSONField(blank=True, default=dict),
+        ),
+    ]

khoj/database/migrations/0057_merge_20240816_1409.py

@@ -0,0 +1,13 @@
+# Generated by Django 5.0.7 on 2024-08-16 18:09
+
+from typing import List
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0056_chatmodeloptions_vision_enabled"),
+        ("database", "0056_searchmodelconfig_cross_encoder_model_config"),
+    ]
+    operations: List[migrations.operations.base.Operation] = []

khoj/database/migrations/0057_remove_serverchatsettings_default_model_and_more.py

@@ -0,0 +1,51 @@
+# Generated by Django 5.0.7 on 2024-08-16 18:18
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0056_searchmodelconfig_cross_encoder_model_config"),
+    ]
+
+    operations = [
+        migrations.RenameField(
+            model_name="serverchatsettings",
+            old_name="default_model",
+            new_name="chat_default",
+        ),
+        migrations.RemoveField(
+            model_name="serverchatsettings",
+            name="summarizer_model",
+        ),
+        migrations.AddField(
+            model_name="chatmodeloptions",
+            name="subscribed_max_prompt_size",
+            field=models.IntegerField(blank=True, default=None, null=True),
+        ),
+        migrations.AddField(
+            model_name="serverchatsettings",
+            name="chat_advanced",
+            field=models.ForeignKey(
+                blank=True,
+                default=None,
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="chat_advanced",
+                to="database.chatmodeloptions",
+            ),
+        ),
+        migrations.AlterField(
+            model_name="serverchatsettings",
+            name="chat_default",
+            field=models.ForeignKey(
+                blank=True,
+                default=None,
+                null=True,
+                on_delete=django.db.models.deletion.CASCADE,
+                related_name="chat_default",
+                to="database.chatmodeloptions",
+            ),
+        ),
+    ]

khoj/database/migrations/0058_alter_chatmodeloptions_chat_model.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.7 on 2024-08-19 12:37
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0057_remove_serverchatsettings_default_model_and_more"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="chatmodeloptions",
+            name="chat_model",
+            field=models.CharField(default="bartowski/Meta-Llama-3.1-8B-Instruct-GGUF", max_length=200),
+        ),
+    ]

khoj/database/migrations/0059_searchmodelconfig_bi_encoder_confidence_threshold.py

@@ -0,0 +1,17 @@
+# Generated by Django 5.0.7 on 2024-08-24 18:19
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0058_alter_chatmodeloptions_chat_model"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="searchmodelconfig",
+            name="bi_encoder_confidence_threshold",
+            field=models.FloatField(default=0.18),
+        ),
+    ]

khoj/database/migrations/0060_merge_20240905_1828.py

@@ -0,0 +1,14 @@
+# Generated by Django 5.0.7 on 2024-09-05 18:28
+
+from typing import List
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0057_merge_20240816_1409"),
+        ("database", "0059_searchmodelconfig_bi_encoder_confidence_threshold"),
+    ]
+
+    operations: List[str] = []

khoj/database/migrations/0061_alter_chatmodeloptions_model_type.py

@@ -0,0 +1,26 @@
+# Generated by Django 5.0.8 on 2024-09-12 20:06
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0060_merge_20240905_1828"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="chatmodeloptions",
+            name="model_type",
+            field=models.CharField(
+                choices=[
+                    ("openai", "Openai"),
+                    ("offline", "Offline"),
+                    ("anthropic", "Anthropic"),
+                    ("google", "Google"),
+                ],
+                default="offline",
+                max_length=200,
+            ),
+        ),
+    ]

khoj/database/migrations/0061_alter_texttoimagemodelconfig_model_type.py

@@ -0,0 +1,21 @@
+# Generated by Django 5.0.7 on 2024-09-12 05:43
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0060_merge_20240905_1828"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="texttoimagemodelconfig",
+            name="model_type",
+            field=models.CharField(
+                choices=[("openai", "Openai"), ("stability-ai", "Stabilityai"), ("replicate", "Replicate")],
+                default="openai",
+                max_length=200,
+            ),
+        ),
+    ]

khoj/database/migrations/0062_merge_20240913_0222.py

@@ -0,0 +1,14 @@
+# Generated by Django 5.0.8 on 2024-09-13 02:22
+
+from typing import List
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0061_alter_chatmodeloptions_model_type"),
+        ("database", "0061_alter_texttoimagemodelconfig_model_type"),
+    ]
+
+    operations: List[str] = []

khoj/database/migrations/0063_conversation_temp_id.py

@@ -0,0 +1,36 @@
+# Generated by Django 5.0.8 on 2024-09-19 15:53
+
+import uuid
+
+from django.db import migrations, models
+
+
+def create_uuid(apps, schema_editor):
+    Conversation = apps.get_model("database", "Conversation")
+    for conversation in Conversation.objects.all():
+        conversation.temp_id = uuid.uuid4()
+        conversation.save()
+
+
+def remove_uuid(apps, schema_editor):
+    pass
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0062_merge_20240913_0222"),
+    ]
+
+    operations = [
+        migrations.AddField(
+            model_name="conversation",
+            name="temp_id",
+            field=models.UUIDField(default=uuid.uuid4, editable=False),
+        ),
+        migrations.RunPython(create_uuid, reverse_code=remove_uuid),
+        migrations.AlterField(
+            model_name="conversation",
+            name="temp_id",
+            field=models.UUIDField(default=uuid.uuid4, editable=False, unique=True),
+        ),
+    ]

khoj/database/migrations/0064_remove_conversation_temp_id_alter_conversation_id.py

@@ -0,0 +1,86 @@
+# Generated by Django 5.0.8 on 2024-09-19 15:59
+
+import json
+import pickle
+import uuid
+
+from django.db import migrations, models
+
+
+def reverse_remove_bigint_id(apps, schema_editor):
+    Conversation = apps.get_model("database", "Conversation")
+    index = 1
+    for conversation in Conversation.objects.all():
+        conversation.id = index
+        conversation.save()
+        index += 1
+
+
+def update_conversation_id_in_job_state(apps, schema_editor):
+    try:
+        DjangoJob = apps.get_model("django_apscheduler", "DjangoJob")
+        Conversation = apps.get_model("database", "Conversation")
+
+        for job in DjangoJob.objects.all():
+            job_state = pickle.loads(job.job_state)
+            kwargs = job_state.get("kwargs")
+            conversation_id = kwargs.get("conversation_id") if kwargs else None
+            automation_metadata = json.loads(job_state.get("name", "{}"))
+
+            if not conversation_id:
+                job.delete()
+
+            if conversation_id:
+                try:
+                    conversation = Conversation.objects.get(id=conversation_id)
+                    automation_metadata["conversation_id"] = str(conversation.temp_id)
+                    name = json.dumps(automation_metadata)
+                    job_state["name"] = name
+                    job_state["kwargs"]["conversation_id"] = str(conversation.temp_id)
+                    job.job_state = pickle.dumps(job_state)
+                    job.save()
+                except Conversation.DoesNotExist:
+                    pass
+    except LookupError as e:
+        pass
+
+
+def no_op(apps, schema_editor):
+    pass
+
+
+def disable_triggers(apps, schema_editor):
+    schema_editor.execute('ALTER TABLE "database_conversation" DISABLE TRIGGER ALL;')
+
+
+def enable_triggers(apps, schema_editor):
+    schema_editor.execute('ALTER TABLE "database_conversation" ENABLE TRIGGER ALL;')
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0063_conversation_temp_id"),
+    ]
+
+    operations = [
+        migrations.RunPython(no_op, reverse_code=enable_triggers),
+        migrations.RunPython(update_conversation_id_in_job_state, reverse_code=no_op),
+        migrations.RemoveField(
+            model_name="conversation",
+            name="id",
+        ),
+        migrations.RenameField(
+            model_name="conversation",
+            old_name="temp_id",
+            new_name="id",
+        ),
+        migrations.AlterField(
+            model_name="conversation",
+            name="id",
+            field=models.UUIDField(
+                db_index=True, default=uuid.uuid4, editable=False, primary_key=True, serialize=False, unique=True
+            ),
+        ),
+        migrations.RunPython(no_op, reverse_code=reverse_remove_bigint_id),
+        migrations.RunPython(no_op, reverse_code=disable_triggers),
+    ]

khoj/database/migrations/0065_remove_agent_avatar_remove_agent_public_and_more.py

@@ -0,0 +1,49 @@
+# Generated by Django 5.0.8 on 2024-09-18 02:54
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+    dependencies = [
+        ("database", "0064_remove_conversation_temp_id_alter_conversation_id"),
+    ]
+
+    operations = [
+        migrations.RemoveField(
+            model_name="agent",
+            name="avatar",
+        ),
+        migrations.RemoveField(
+            model_name="agent",
+            name="public",
+        ),
+        migrations.AddField(
+            model_name="agent",
+            name="privacy_level",
+            field=models.CharField(
+                choices=[("public", "Public"), ("private", "Private"), ("protected", "Protected")],
+                default="private",
+                max_length=30,
+            ),
+        ),
+        migrations.AddField(
+            model_name="entry",
+            name="agent",
+            field=models.ForeignKey(
+                blank=True, default=None, null=True, on_delete=django.db.models.deletion.CASCADE, to="database.agent"
+            ),
+        ),
+        migrations.AddField(
+            model_name="fileobject",
+            name="agent",
+            field=models.ForeignKey(
+                blank=True, default=None, null=True, on_delete=django.db.models.deletion.CASCADE, to="database.agent"
+            ),
+        ),
+        migrations.AlterField(
+            model_name="agent",
+            name="slug",
+            field=models.CharField(max_length=200, unique=True),
+        ),
+    ]