添加知识库管理

backend/ai/views/__init__.py

@@ -2,8 +2,10 @@ __all__ = [
    'AIApiKeyViewSet',
    'AIModelViewSet',
    'ToolViewSet',
+   'KnowledgeViewSet',
 ]
 
 from ai.views.ai_api_key import AIApiKeyViewSet
 from ai.views.ai_model import AIModelViewSet
-from ai.views.tool import ToolViewSet
+from ai.views.tool import ToolViewSet
+from ai.views.knowledge import KnowledgeViewSet

backend/ai/views/knowledge.py

@@ -0,0 +1,24 @@
+from ai.models import Knowledge
+from utils.serializers import CustomModelSerializer
+from utils.custom_model_viewSet import CustomModelViewSet
+
+class KnowledgeSerializer(CustomModelSerializer):
+    """
+    AI 知识库 序列化器
+    """
+    class Meta:
+        model = Knowledge
+        fields = '__all__'
+        read_only_fields = ['id', 'create_time', 'update_time']
+
+
+class KnowledgeViewSet(CustomModelViewSet):
+    """
+    AI 知识库 视图集
+    """
+    queryset = Knowledge.objects.filter(is_deleted=False).order_by('-id')
+    serializer_class = KnowledgeSerializer
+    filterset_fields = ['id', 'remark', 'creator', 'modifier', 'is_deleted', 'name', 'embedding_model', 'top_k', 'status']
+    search_fields = ['name']  # 根据实际字段调整
+    ordering_fields = ['create_time', 'id']
+    ordering = ['-create_time']