添加聊天对话列表
This commit is contained in:
@@ -3,9 +3,11 @@ __all__ = [
|
||||
'AIModelViewSet',
|
||||
'ToolViewSet',
|
||||
'KnowledgeViewSet',
|
||||
'ChatConversationViewSet',
|
||||
]
|
||||
|
||||
from ai.views.ai_api_key import AIApiKeyViewSet
|
||||
from ai.views.ai_model import AIModelViewSet
|
||||
from ai.views.tool import ToolViewSet
|
||||
from ai.views.knowledge import KnowledgeViewSet
|
||||
from ai.views.knowledge import KnowledgeViewSet
|
||||
from ai.views.chat_conversation import ChatConversationViewSet
|
||||
39
backend/ai/views/chat_conversation.py
Normal file
39
backend/ai/views/chat_conversation.py
Normal file
@@ -0,0 +1,39 @@
|
||||
from rest_framework import serializers
|
||||
|
||||
from ai.models import ChatConversation
|
||||
from utils.serializers import CustomModelSerializer
|
||||
from utils.custom_model_viewSet import CustomModelViewSet
|
||||
from django_filters import rest_framework as filters
|
||||
|
||||
|
||||
class ChatConversationSerializer(CustomModelSerializer):
|
||||
username = serializers.CharField(source='user.username', read_only=True)
|
||||
|
||||
"""
|
||||
AI 聊天对话 序列化器
|
||||
"""
|
||||
class Meta:
|
||||
model = ChatConversation
|
||||
fields = '__all__'
|
||||
read_only_fields = ['id', 'create_time', 'update_time']
|
||||
|
||||
class ChatConversationFilter(filters.FilterSet):
|
||||
|
||||
class Meta:
|
||||
model = ChatConversation
|
||||
fields = ['id', 'remark', 'creator', 'modifier', 'is_deleted', 'title', 'pinned', 'model',
|
||||
'system_message', 'max_tokens', 'max_contexts']
|
||||
|
||||
|
||||
class ChatConversationViewSet(CustomModelViewSet):
|
||||
"""
|
||||
AI 聊天对话 视图集
|
||||
"""
|
||||
queryset = ChatConversation.objects.filter(is_deleted=False).order_by('-id')
|
||||
serializer_class = ChatConversationSerializer
|
||||
filterset_class = ChatConversationFilter
|
||||
search_fields = ['name'] # 根据实际字段调整
|
||||
ordering_fields = ['create_time', 'id']
|
||||
ordering = ['-create_time']
|
||||
|
||||
# 移入urls中
|
||||
Reference in New Issue
Block a user