From 183d6c74d587184002db18241ce31ad3002c1f28 Mon Sep 17 00:00:00 2001
From: wxg0103 <727495428@qq.com>
Date: Thu, 26 Jun 2025 11:40:57 +0800
Subject: [PATCH] refactor: knowledge model

---
 .../0002_alter_document_status_and_more.py    | 37 +++++++++++++++++++
 apps/knowledge/models/knowledge.py            |  2 +-
 ..._model_unique_together_alter_model_user.py | 24 ++++++++++++
 3 files changed, 62 insertions(+), 1 deletion(-)
 create mode 100644 apps/knowledge/migrations/0002_alter_document_status_and_more.py
 create mode 100644 apps/models_provider/migrations/0002_alter_model_unique_together_alter_model_user.py

diff --git a/apps/knowledge/migrations/0002_alter_document_status_and_more.py b/apps/knowledge/migrations/0002_alter_document_status_and_more.py
new file mode 100644
index 000000000..f679351b1
--- /dev/null
+++ b/apps/knowledge/migrations/0002_alter_document_status_and_more.py
@@ -0,0 +1,37 @@
+# Generated by Django 5.2.3 on 2025-06-26 03:39
+
+import django.db.models.deletion
+import knowledge.models.knowledge
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('knowledge', '0001_initial'),
+        ('models_provider', '0001_initial'),
+        ('users', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name='document',
+            name='status',
+            field=models.CharField(default=knowledge.models.knowledge.Status.__str__, max_length=20, verbose_name='状态'),
+        ),
+        migrations.AlterField(
+            model_name='knowledge',
+            name='embedding_model',
+            field=models.ForeignKey(blank=True, db_constraint=False, null=True, on_delete=django.db.models.deletion.SET_NULL, to='models_provider.model'),
+        ),
+        migrations.AlterField(
+            model_name='knowledge',
+            name='user',
+            field=models.ForeignKey(blank=True, db_constraint=False, null=True, on_delete=django.db.models.deletion.SET_NULL, to='users.user'),
+        ),
+        migrations.AlterField(
+            model_name='paragraph',
+            name='status',
+            field=models.CharField(default=knowledge.models.knowledge.Status.__str__, max_length=20, verbose_name='状态'),
+        ),
+    ]
diff --git a/apps/knowledge/models/knowledge.py b/apps/knowledge/models/knowledge.py
index 76d703d54..2099ce628 100644
--- a/apps/knowledge/models/knowledge.py
+++ b/apps/knowledge/models/knowledge.py
@@ -125,7 +125,7 @@ class Knowledge(AppModelMixin):
     scope = models.CharField(max_length=20, verbose_name='可用范围', choices=KnowledgeScope.choices,
                              default=KnowledgeScope.WORKSPACE)
     folder = models.ForeignKey(KnowledgeFolder, on_delete=models.DO_NOTHING, verbose_name="文件夹id", default='default')
-    embedding_model = models.ForeignKey(Model, on_delete=models.DO_NOTHING, verbose_name="向量模型")
+    embedding_model = models.ForeignKey(Model, on_delete=models.SET_NULL, db_constraint=False, blank=True, null=True)
     file_size_limit = models.IntegerField(verbose_name="文件大小限制", default=100)
     file_count_limit = models.IntegerField(verbose_name="文件数量限制", default=50)
     meta = models.JSONField(verbose_name="元数据", default=dict)
diff --git a/apps/models_provider/migrations/0002_alter_model_unique_together_alter_model_user.py b/apps/models_provider/migrations/0002_alter_model_unique_together_alter_model_user.py
new file mode 100644
index 000000000..7eeb4e73c
--- /dev/null
+++ b/apps/models_provider/migrations/0002_alter_model_unique_together_alter_model_user.py
@@ -0,0 +1,24 @@
+# Generated by Django 5.2.3 on 2025-06-26 03:39
+
+import django.db.models.deletion
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ('models_provider', '0001_initial'),
+        ('users', '0001_initial'),
+    ]
+
+    operations = [
+        migrations.AlterUniqueTogether(
+            name='model',
+            unique_together={('name', 'workspace_id')},
+        ),
+        migrations.AlterField(
+            model_name='model',
+            name='user',
+            field=models.ForeignKey(blank=True, db_constraint=False, null=True, on_delete=django.db.models.deletion.SET_NULL, to='users.user'),
+        ),
+    ]