django-cfg 1.4.21__py3-none-any.whl → 1.4.24__py3-none-any.whl

django_cfg/apps/knowbase/views/chat_views.py

@@ -2,38 +2,38 @@
 Chat API views for RAG-powered conversations.
 """
 
+from django.utils.decorators import method_decorator
+from django_ratelimit.decorators import ratelimit
+from drf_spectacular.utils import OpenApiExample, extend_schema
 from rest_framework import status
 from rest_framework.decorators import action
 from rest_framework.response import Response
-from django_ratelimit.decorators import ratelimit
-from django.utils.decorators import method_decorator
-from drf_spectacular.utils import extend_schema, OpenApiExample
 
 from ..models import ChatSession
-from ..services import ChatService
 from ..serializers import (
-    ChatSessionCreateSerializer,
-    ChatSessionSerializer,
+    ChatHistorySerializer,
     ChatQuerySerializer,
     ChatResponseSerializer,
-    ChatHistorySerializer
+    ChatSessionCreateSerializer,
+    ChatSessionSerializer,
 )
+from ..services import ChatService
 from .base import BaseKnowledgeViewSet
 
 
 class ChatSessionViewSet(BaseKnowledgeViewSet):
     """Chat session management endpoints."""
-    
+
     queryset = ChatSession.objects.all()
     serializer_class = ChatSessionSerializer
     service_class = ChatService
-    
+
     def get_serializer_class(self):
         """Return appropriate serializer based on action."""
         if self.action == 'create':
             return ChatSessionCreateSerializer
         return ChatSessionSerializer
-    
+
     @extend_schema(
         summary="Create new chat session",
         responses={201: ChatSessionSerializer}
@@ -42,7 +42,7 @@ class ChatSessionViewSet(BaseKnowledgeViewSet):
         """Create new chat session."""
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        
+
         service = self.get_service()
         session = service.create_session(
             title=serializer.validated_data['title'],
@@ -50,10 +50,10 @@ class ChatSessionViewSet(BaseKnowledgeViewSet):
             temperature=serializer.validated_data['temperature'],
             max_context_chunks=serializer.validated_data['max_context_chunks']
         )
-        
+
         response_serializer = ChatSessionSerializer(session)
         return Response(response_serializer.data, status=status.HTTP_201_CREATED)
-    
+
     @extend_schema(
         summary="List user chat sessions",
         responses={200: ChatSessionSerializer(many=True)}
@@ -62,25 +62,25 @@ class ChatSessionViewSet(BaseKnowledgeViewSet):
         """List user chat sessions with filtering."""
         # Add filtering by is_active
         queryset = self.get_queryset()
-        
+
         # Filter by active status if requested
         is_active = request.query_params.get('is_active')
         if is_active is not None:
             is_active_bool = is_active.lower() in ('true', '1', 'yes')
             queryset = queryset.filter(is_active=is_active_bool)
-        
+
         # Order by updated_at to avoid pagination warning
         queryset = queryset.order_by('-updated_at')
-        
+
         # Apply pagination
         page = self.paginate_queryset(queryset)
         if page is not None:
             serializer = self.get_serializer(page, many=True)
             return self.get_paginated_response(serializer.data)
-        
+
         serializer = self.get_serializer(queryset, many=True)
         return Response(serializer.data)
-    
+
     @extend_schema(
         summary="Archive chat session",
         responses={200: ChatSessionSerializer}
@@ -90,10 +90,10 @@ class ChatSessionViewSet(BaseKnowledgeViewSet):
         """Archive (deactivate) chat session."""
         session = self.get_object()
         session.archive()
-        
+
         serializer = self.get_serializer(session)
         return Response(serializer.data)
-    
+
     @extend_schema(
         summary="Activate chat session",
         responses={200: ChatSessionSerializer}
@@ -103,20 +103,20 @@ class ChatSessionViewSet(BaseKnowledgeViewSet):
         """Activate chat session."""
         session = self.get_object()
         session.activate()
-        
+
         serializer = self.get_serializer(session)
         return Response(serializer.data)
 
 
 class ChatViewSet(BaseKnowledgeViewSet):
     """Chat query endpoints."""
-    
+
     # This ViewSet doesn't use standard CRUD operations
     # It only has custom actions (query, history)
     queryset = ChatSession.objects.none()  # Empty queryset since we don't use list/retrieve/etc
     service_class = ChatService
     serializer_class = ChatResponseSerializer
-    
+
     def get_serializer_class(self):
         """Return appropriate serializer based on action."""
         if self.action == 'query':
@@ -124,7 +124,7 @@ class ChatViewSet(BaseKnowledgeViewSet):
         elif self.action == 'history':
             return ChatHistorySerializer
         return ChatResponseSerializer
-    
+
     @extend_schema(
         summary="Process chat query with RAG",
         request=ChatQuerySerializer,
@@ -146,9 +146,9 @@ class ChatViewSet(BaseKnowledgeViewSet):
         """Process chat query with RAG context."""
         serializer = ChatQuerySerializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        
+
         service = self.get_service()
-        
+
         # Create session if not provided
         session_id = serializer.validated_data.get('session_id')
         if not session_id:
@@ -156,7 +156,7 @@ class ChatViewSet(BaseKnowledgeViewSet):
             session_id = str(session.id)
         else:
             session_id = str(session_id)
-        
+
         # Process query
         result = service.process_query(
             session_id=session_id,
@@ -164,9 +164,9 @@ class ChatViewSet(BaseKnowledgeViewSet):
             max_tokens=serializer.validated_data['max_tokens'],
             include_sources=serializer.validated_data['include_sources']
         )
-        
+
         return Response(result)
-    
+
     @extend_schema(
         summary="Get chat history",
         responses={200: ChatHistorySerializer}
@@ -176,12 +176,12 @@ class ChatViewSet(BaseKnowledgeViewSet):
         """Get chat session history."""
         service = self.get_service()
         messages = service.get_session_history(session_id=pk)
-        
+
         data = {
             'session_id': pk,
             'messages': messages,
             'total_messages': len(messages)
         }
-        
+
         serializer = ChatHistorySerializer(data)
-        return Response(serializer.data)
+        return Response(serializer.data)

django_cfg/apps/knowbase/views/document_views.py

@@ -2,34 +2,33 @@
 Document management API views.
 """
 
-from rest_framework import status, viewsets
+from django.utils.decorators import method_decorator
+from django_ratelimit.decorators import ratelimit
+from drf_spectacular.utils import OpenApiExample, OpenApiParameter, extend_schema
+from rest_framework import status
 from rest_framework.decorators import action
+from rest_framework.permissions import IsAdminUser, IsAuthenticated
 from rest_framework.response import Response
-from rest_framework.permissions import IsAuthenticated, IsAdminUser
-from django_ratelimit.decorators import ratelimit
-from django.utils.decorators import method_decorator
-from django.db import models
-from drf_spectacular.utils import extend_schema, OpenApiExample, OpenApiParameter
 
 from ..models import Document
-from ..services import DocumentService
 from ..serializers import (
     DocumentCreateSerializer,
+    DocumentProcessingStatusSerializer,
     DocumentSerializer,
     DocumentStatsSerializer,
-    DocumentProcessingStatusSerializer
 )
+from ..services import DocumentService
 from .base import BaseKnowledgeViewSet
 
 
 class DocumentViewSet(BaseKnowledgeViewSet):
     """Document management endpoints - Admin only."""
-    
+
     queryset = Document.objects.all()
     serializer_class = DocumentSerializer
     service_class = DocumentService
     permission_classes = [IsAuthenticated, IsAdminUser]
-    
+
     def get_serializer_class(self):
         """Return appropriate serializer based on action."""
         if self.action == 'create':
@@ -39,7 +38,7 @@ class DocumentViewSet(BaseKnowledgeViewSet):
         elif self.action == 'status':
             return DocumentProcessingStatusSerializer
         return DocumentSerializer
-    
+
     @extend_schema(
         summary="Upload new document",
         description="Upload and process a new knowledge document",
@@ -65,7 +64,7 @@ class DocumentViewSet(BaseKnowledgeViewSet):
         """Create new document with async processing."""
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        
+
         # Use service layer for business logic
         service = self.get_service()
         document = service.create_document(
@@ -74,10 +73,10 @@ class DocumentViewSet(BaseKnowledgeViewSet):
             file_type=serializer.validated_data['file_type'],
             metadata=serializer.validated_data['metadata']
         )
-        
+
         response_serializer = DocumentSerializer(document)
         return Response(response_serializer.data, status=status.HTTP_201_CREATED)
-    
+
     @extend_schema(
         summary="List user documents",
         parameters=[
@@ -93,19 +92,19 @@ class DocumentViewSet(BaseKnowledgeViewSet):
     def list(self, request, *args, **kwargs):
         """List user documents with filtering and pagination."""
         status_filter = request.query_params.get('status')
-        
+
         service = self.get_service()
         queryset = service.get_user_documents(status=status_filter)
-        
+
         # Use DRF pagination
         page = self.paginate_queryset(queryset)
         if page is not None:
             serializer = self.get_serializer(page, many=True)
             return self.get_paginated_response(serializer.data)
-        
+
         serializer = self.get_serializer(queryset, many=True)
         return Response(serializer.data)
-    
+
     @extend_schema(
         summary="Get document details",
         responses={
@@ -116,7 +115,7 @@ class DocumentViewSet(BaseKnowledgeViewSet):
     def retrieve(self, request, *args, **kwargs):
         """Get document by ID."""
         return super().retrieve(request, *args, **kwargs)
-    
+
     @extend_schema(
         summary="Delete document",
         responses={
@@ -127,7 +126,7 @@ class DocumentViewSet(BaseKnowledgeViewSet):
     def destroy(self, request, *args, **kwargs):
         """Delete document and all associated chunks."""
         return super().destroy(request, *args, **kwargs)
-    
+
     @extend_schema(
         summary="Get document processing status",
         responses={200: DocumentProcessingStatusSerializer}
@@ -136,7 +135,7 @@ class DocumentViewSet(BaseKnowledgeViewSet):
     def status(self, request, pk=None):
         """Get document processing status."""
         document = self.get_object()
-        
+
         data = {
             'id': document.id,
             'status': document.processing_status,
@@ -148,10 +147,10 @@ class DocumentViewSet(BaseKnowledgeViewSet):
             'error': document.processing_error if document.processing_error else None,
             'processing_time_seconds': document.processing_duration
         }
-        
+
         serializer = self.get_serializer(data)
         return Response(serializer.data)
-    
+
     @extend_schema(
         summary="Reprocess document",
         description="Trigger reprocessing of document chunks and embeddings"
@@ -160,15 +159,15 @@ class DocumentViewSet(BaseKnowledgeViewSet):
     def reprocess(self, request, pk=None):
         """Trigger document reprocessing."""
         document = self.get_object()
-        
+
         service = self.get_service()
         service.reprocess_document(str(document.id))
-        
+
         return Response({
             'message': 'Document reprocessing started',
             'document_id': str(document.id)
         })
-    
+
     @extend_schema(
         summary="Get processing statistics",
         responses={200: DocumentStatsSerializer}
@@ -178,6 +177,6 @@ class DocumentViewSet(BaseKnowledgeViewSet):
         """Get user's document processing statistics."""
         service = self.get_service()
         stats = service.get_processing_stats()
-        
+
         serializer = self.get_serializer(stats)
-        return Response(serializer.data)
+        return Response(serializer.data)

django_cfg/apps/knowbase/views/public_views.py

@@ -2,27 +2,27 @@
 Public API views for client access without authentication.
 """
 
+from django.db import models
+from drf_spectacular.utils import OpenApiParameter, extend_schema
 from rest_framework import viewsets
-from rest_framework.response import Response
 from rest_framework.permissions import AllowAny
-from django.db import models
-from drf_spectacular.utils import extend_schema, OpenApiParameter
+from rest_framework.response import Response
 
 from ..models import Document, DocumentCategory
 from ..serializers.public_serializers import (
-    PublicDocumentSerializer, 
+    PublicCategorySerializer,
     PublicDocumentListSerializer,
-    PublicCategorySerializer
+    PublicDocumentSerializer,
 )
 
 
 class PublicDocumentViewSet(viewsets.ReadOnlyModelViewSet):
     """Public document endpoints - read-only access for clients."""
-    
+
     serializer_class = PublicDocumentSerializer
     permission_classes = [AllowAny]
     lookup_field = 'pk'
-    
+
     def get_queryset(self):
         """Get only publicly accessible documents."""
         return Document.objects.filter(
@@ -31,13 +31,13 @@ class PublicDocumentViewSet(viewsets.ReadOnlyModelViewSet):
         ).filter(
             models.Q(categories__isnull=True) | models.Q(categories__is_public=True)
         ).prefetch_related('categories')
-    
+
     def get_serializer_class(self):
         """Return appropriate serializer based on action."""
         if self.action == 'list':
             return PublicDocumentListSerializer
         return PublicDocumentSerializer
-    
+
     @extend_schema(
         summary="List public documents",
         description="Get list of all completed and publicly accessible documents",
@@ -60,32 +60,32 @@ class PublicDocumentViewSet(viewsets.ReadOnlyModelViewSet):
     def list(self, request, *args, **kwargs):
         """List all publicly accessible documents with optional filtering."""
         queryset = self.get_queryset()
-        
+
         # Search functionality
         search = request.query_params.get('search')
         if search:
             queryset = queryset.filter(
-                models.Q(title__icontains=search) | 
+                models.Q(title__icontains=search) |
                 models.Q(content__icontains=search)
             )
-        
+
         # Category filter
         category = request.query_params.get('category')
         if category:
             queryset = queryset.filter(categories__name__iexact=category)
-        
+
         # Order by creation date (newest first)
         queryset = queryset.order_by('-created_at')
-        
+
         # Use DRF pagination
         page = self.paginate_queryset(queryset)
         if page is not None:
             serializer = self.get_serializer(page, many=True)
             return self.get_paginated_response(serializer.data)
-        
+
         serializer = self.get_serializer(queryset, many=True)
         return Response(serializer.data)
-    
+
     @extend_schema(
         summary="Get public document details",
         description="Get document details by ID (public access)",
@@ -101,12 +101,12 @@ class PublicDocumentViewSet(viewsets.ReadOnlyModelViewSet):
 
 class PublicCategoryViewSet(viewsets.ReadOnlyModelViewSet):
     """Public category endpoints - read-only access for clients."""
-    
+
     queryset = DocumentCategory.objects.filter(is_public=True)
     serializer_class = PublicCategorySerializer
     permission_classes = [AllowAny]
     lookup_field = 'pk'
-    
+
     @extend_schema(
         summary="List public categories",
         description="Get list of all public categories",
@@ -115,7 +115,7 @@ class PublicCategoryViewSet(viewsets.ReadOnlyModelViewSet):
     def list(self, request, *args, **kwargs):
         """List all public categories."""
         return super().list(request, *args, **kwargs)
-    
+
     @extend_schema(
         summary="Get public category details",
         description="Get category details by ID (public access)",