django-cfg 1.4.20__py3-none-any.whl → 1.4.23__py3-none-any.whl

django_cfg/apps/knowbase/serializers/archive_serializers.py

@@ -2,15 +2,23 @@
 Archive serializers for DRF API.
 """
 
+from typing import Any, Dict, List
+
 from rest_framework import serializers
-from typing import Dict, Any, List
-from ..models.archive import DocumentArchive, ArchiveItem, ArchiveItemChunk, ArchiveType, ContentType, ChunkType
+
+from ..models.archive import (
+    ArchiveItem,
+    ArchiveItemChunk,
+    ChunkType,
+    ContentType,
+    DocumentArchive,
+)
 from ..models.document import DocumentCategory
 
 
 class DocumentArchiveCreateSerializer(serializers.Serializer):
     """Document archive creation request serializer."""
-    
+
     archive_file = serializers.FileField(
         help_text="Archive file to upload"
     )
@@ -39,19 +47,19 @@ class DocumentArchiveCreateSerializer(serializers.Serializer):
         default=True,
         help_text="Process archive synchronously"
     )
-    
+
     def validate_archive_file(self, value):
         """Validate uploaded archive file."""
         if not value:
             raise serializers.ValidationError('Archive file is required')
-        
+
         # Check file size (max 100MB)
         max_size = 100 * 1024 * 1024  # 100MB
         if value.size > max_size:
             raise serializers.ValidationError(
                 f'File too large. Maximum size is {max_size // (1024*1024)}MB'
             )
-        
+
         # Check file extension
         allowed_extensions = [
             '.zip', '.jar', '.war', '.ear',  # ZIP formats
@@ -64,15 +72,15 @@ class DocumentArchiveCreateSerializer(serializers.Serializer):
             raise serializers.ValidationError(
                 f'Unsupported file format. Allowed: {", ".join(allowed_extensions)}'
             )
-        
+
         return value
-    
+
     def validate_title(self, value: str) -> str:
         """Validate title format."""
         if value and not value.strip():
             raise serializers.ValidationError('Title cannot be empty')
         return value.strip() if value else value
-    
+
     def validate_category_ids(self, value: List[str]) -> List[str]:
         """Validate category IDs exist."""
         if value:
@@ -81,19 +89,19 @@ class DocumentArchiveCreateSerializer(serializers.Serializer):
                     id__in=value
                 ).values_list('id', flat=True)
             )
-            
+
             invalid_ids = set(str(id) for id in value) - set(str(id) for id in existing_ids)
             if invalid_ids:
                 raise serializers.ValidationError(
                     f'Invalid category IDs: {", ".join(invalid_ids)}'
                 )
-        
+
         return value
 
 
 class DocumentCategorySerializer(serializers.ModelSerializer):
     """Document category serializer."""
-    
+
     class Meta:
         model = DocumentCategory
         fields = ['id', 'name', 'description', 'is_public', 'created_at']
@@ -102,7 +110,7 @@ class DocumentCategorySerializer(serializers.ModelSerializer):
 
 class ArchiveItemSerializer(serializers.ModelSerializer):
     """Archive item serializer."""
-    
+
     class Meta:
         model = ArchiveItem
         fields = [
@@ -120,19 +128,19 @@ class ArchiveItemSerializer(serializers.ModelSerializer):
 
 class ArchiveItemDetailSerializer(ArchiveItemSerializer):
     """Detailed archive item serializer with content."""
-    
+
     raw_content = serializers.CharField(read_only=True)
     metadata = serializers.JSONField(read_only=True)
-    
+
     class Meta(ArchiveItemSerializer.Meta):
         fields = ArchiveItemSerializer.Meta.fields + ['raw_content', 'metadata']
 
 
 class ArchiveItemChunkSerializer(serializers.ModelSerializer):
     """Archive item chunk serializer."""
-    
+
     context_summary = serializers.SerializerMethodField()
-    
+
     class Meta:
         model = ArchiveItemChunk
         fields = [
@@ -144,7 +152,7 @@ class ArchiveItemChunkSerializer(serializers.ModelSerializer):
             'id', 'token_count', 'character_count', 'embedding_model',
             'embedding_cost', 'context_summary', 'created_at'
         ]
-    
+
     def get_context_summary(self, obj: ArchiveItemChunk) -> Dict[str, Any]:
         """Get context summary for display."""
         return obj.get_context_summary()
@@ -152,21 +160,21 @@ class ArchiveItemChunkSerializer(serializers.ModelSerializer):
 
 class ArchiveItemChunkDetailSerializer(ArchiveItemChunkSerializer):
     """Detailed chunk serializer with full context."""
-    
+
     context_metadata = serializers.JSONField(read_only=True)
-    
+
     class Meta(ArchiveItemChunkSerializer.Meta):
         fields = ArchiveItemChunkSerializer.Meta.fields + ['context_metadata']
 
 
 class DocumentArchiveSerializer(serializers.ModelSerializer):
     """Document archive serializer."""
-    
+
     categories = DocumentCategorySerializer(many=True, read_only=True)
     processing_progress = serializers.ReadOnlyField()
     vectorization_progress = serializers.ReadOnlyField()
     is_processed = serializers.ReadOnlyField()
-    
+
     class Meta:
         model = DocumentArchive
         fields = [
@@ -190,13 +198,13 @@ class DocumentArchiveSerializer(serializers.ModelSerializer):
 
 class DocumentArchiveDetailSerializer(DocumentArchiveSerializer):
     """Detailed archive serializer with items."""
-    
+
     items = ArchiveItemSerializer(many=True, read_only=True)
     file_tree = serializers.SerializerMethodField()
-    
+
     class Meta(DocumentArchiveSerializer.Meta):
         fields = DocumentArchiveSerializer.Meta.fields + ['items', 'file_tree', 'metadata']
-    
+
     def get_file_tree(self, obj: DocumentArchive) -> Dict[str, Any]:
         """Get hierarchical file tree."""
         return obj.get_file_tree()
@@ -204,10 +212,10 @@ class DocumentArchiveDetailSerializer(DocumentArchiveSerializer):
 
 class DocumentArchiveListSerializer(serializers.ModelSerializer):
     """Simplified archive serializer for list views."""
-    
+
     categories = DocumentCategorySerializer(many=True, read_only=True)
     processing_progress = serializers.ReadOnlyField()
-    
+
     class Meta:
         model = DocumentArchive
         fields = [
@@ -222,7 +230,7 @@ class DocumentArchiveListSerializer(serializers.ModelSerializer):
 
 class ArchiveProcessingResultSerializer(serializers.Serializer):
     """Archive processing result serializer."""
-    
+
     archive_id = serializers.UUIDField(read_only=True)
     status = serializers.CharField(read_only=True)
     processing_time_ms = serializers.IntegerField(read_only=True)
@@ -235,7 +243,7 @@ class ArchiveProcessingResultSerializer(serializers.Serializer):
 
 class ArchiveSearchRequestSerializer(serializers.Serializer):
     """Archive search request serializer."""
-    
+
     query = serializers.CharField(
         min_length=1,
         max_length=500,
@@ -277,7 +285,7 @@ class ArchiveSearchRequestSerializer(serializers.Serializer):
 
 class ArchiveSearchResultSerializer(serializers.Serializer):
     """Archive search result serializer."""
-    
+
     chunk = ArchiveItemChunkSerializer(read_only=True)
     similarity_score = serializers.FloatField(read_only=True)
     context_summary = serializers.DictField(read_only=True)
@@ -287,7 +295,7 @@ class ArchiveSearchResultSerializer(serializers.Serializer):
 
 class ArchiveStatisticsSerializer(serializers.Serializer):
     """Archive statistics serializer."""
-    
+
     total_archives = serializers.IntegerField(read_only=True)
     processed_archives = serializers.IntegerField(read_only=True)
     failed_archives = serializers.IntegerField(read_only=True)
@@ -302,7 +310,7 @@ class ArchiveStatisticsSerializer(serializers.Serializer):
 
 class VectorizationStatisticsSerializer(serializers.Serializer):
     """Vectorization statistics serializer."""
-    
+
     total_chunks = serializers.IntegerField(read_only=True)
     vectorized_chunks = serializers.IntegerField(read_only=True)
     pending_chunks = serializers.IntegerField(read_only=True)
@@ -315,7 +323,7 @@ class VectorizationStatisticsSerializer(serializers.Serializer):
 
 class ContentTypeDistributionSerializer(serializers.Serializer):
     """Content type distribution serializer."""
-    
+
     content_type = serializers.CharField(read_only=True)
     count = serializers.IntegerField(read_only=True)
     percentage = serializers.FloatField(read_only=True)
@@ -323,7 +331,7 @@ class ContentTypeDistributionSerializer(serializers.Serializer):
 
 class LanguageDistributionSerializer(serializers.Serializer):
     """Language distribution serializer."""
-    
+
     language = serializers.CharField(read_only=True)
     count = serializers.IntegerField(read_only=True)
     percentage = serializers.FloatField(read_only=True)
@@ -331,36 +339,36 @@ class LanguageDistributionSerializer(serializers.Serializer):
 
 class ArchiveUploadSerializer(serializers.Serializer):
     """Archive file upload serializer."""
-    
+
     file = serializers.FileField(
         help_text="Archive file to upload"
     )
-    
+
     def validate_file(self, value):
         """Validate uploaded file."""
-        
+
         # Check file size (200MB limit)
         max_size = 200 * 1024 * 1024
         if value.size > max_size:
             raise serializers.ValidationError(
                 f'File too large. Maximum size is {max_size // (1024*1024)}MB'
             )
-        
+
         # Check file extension
         allowed_extensions = ['.zip', '.tar', '.tar.gz', '.tgz', '.tar.bz2', '.tbz2']
         filename = value.name.lower()
-        
+
         if not any(filename.endswith(ext) for ext in allowed_extensions):
             raise serializers.ValidationError(
                 f'Unsupported file type. Allowed: {", ".join(allowed_extensions)}'
             )
-        
+
         return value
 
 
 class ChunkRevectorizationRequestSerializer(serializers.Serializer):
     """Chunk re-vectorization request serializer."""
-    
+
     chunk_ids = serializers.ListField(
         child=serializers.UUIDField(),
         min_length=1,
@@ -374,7 +382,7 @@ class ChunkRevectorizationRequestSerializer(serializers.Serializer):
 
 class VectorizationResultSerializer(serializers.Serializer):
     """Vectorization result serializer."""
-    
+
     vectorized_count = serializers.IntegerField(read_only=True)
     failed_count = serializers.IntegerField(read_only=True)
     total_tokens = serializers.IntegerField(read_only=True)

django_cfg/apps/knowbase/serializers/chat_serializers.py

@@ -3,13 +3,14 @@ Chat serializers for DRF API.
 """
 
 from rest_framework import serializers
-from ..models import ChatSession, ChatMessage
+
+from ..models import ChatMessage, ChatSession
 from ..utils.validation import is_valid_float, safe_float
 
 
 class ChatSessionCreateSerializer(serializers.Serializer):
     """Chat session creation request serializer."""
-    
+
     title = serializers.CharField(
         max_length=255,
         default="",
@@ -37,7 +38,7 @@ class ChatSessionCreateSerializer(serializers.Serializer):
 
 class ChatQuerySerializer(serializers.Serializer):
     """Chat query request serializer."""
-    
+
     session_id = serializers.UUIDField(
         required=False,
         allow_null=True,
@@ -58,7 +59,7 @@ class ChatQuerySerializer(serializers.Serializer):
         default=True,
         help_text="Include source documents in response"
     )
-    
+
     def validate_query(self, value):
         """Validate query content."""
         if not value.strip():
@@ -68,11 +69,11 @@ class ChatQuerySerializer(serializers.Serializer):
 
 class ChatSourceSerializer(serializers.Serializer):
     """Chat source document information serializer."""
-    
+
     document_title = serializers.CharField()
     chunk_content = serializers.CharField()
     similarity = serializers.FloatField()
-    
+
     def validate_similarity(self, value):
         """Validate similarity value to prevent NaN in JSON."""
         if not is_valid_float(value):
@@ -82,7 +83,7 @@ class ChatSourceSerializer(serializers.Serializer):
 
 class ChatResponseSerializer(serializers.Serializer):
     """Chat response serializer."""
-    
+
     message_id = serializers.UUIDField()
     content = serializers.CharField()
     tokens_used = serializers.IntegerField()
@@ -94,14 +95,14 @@ class ChatResponseSerializer(serializers.Serializer):
 
 class ChatSessionSerializer(serializers.ModelSerializer):
     """Chat session response serializer."""
-    
+
     id = serializers.UUIDField(read_only=True)
     total_cost_usd = serializers.FloatField(read_only=True)
-    
+
     def validate_total_cost_usd(self, value):
         """Validate cost value to prevent NaN in JSON."""
         return safe_float(value, 0.0)
-    
+
     class Meta:
         model = ChatSession
         fields = [
@@ -113,14 +114,14 @@ class ChatSessionSerializer(serializers.ModelSerializer):
 
 class ChatMessageSerializer(serializers.ModelSerializer):
     """Chat message response serializer."""
-    
+
     id = serializers.UUIDField(read_only=True)
     cost_usd = serializers.FloatField(read_only=True)
-    
+
     def validate_cost_usd(self, value):
         """Validate cost value to prevent NaN in JSON."""
         return safe_float(value, 0.0)
-    
+
     class Meta:
         model = ChatMessage
         fields = [
@@ -131,7 +132,7 @@ class ChatMessageSerializer(serializers.ModelSerializer):
 
 class ChatHistorySerializer(serializers.Serializer):
     """Chat history response serializer."""
-    
+
     session_id = serializers.UUIDField()
     messages = ChatMessageSerializer(many=True)
-    total_messages = serializers.IntegerField()
+    total_messages = serializers.IntegerField()

django_cfg/apps/knowbase/serializers/document_serializers.py

@@ -2,14 +2,15 @@
 Document serializers for DRF API.
 """
 
+
 from rest_framework import serializers
-from typing import Dict, Any
-from ..models import Document, ProcessingStatus
+
+from ..models import Document
 
 
 class DocumentCreateSerializer(serializers.Serializer):
     """Document creation request serializer."""
-    
+
     title = serializers.CharField(
         max_length=512,
         help_text="Document title"
@@ -28,19 +29,19 @@ class DocumentCreateSerializer(serializers.Serializer):
         default=dict,
         help_text="Additional metadata"
     )
-    
+
     def validate_content(self, value):
         """Validate content for security."""
         dangerous_patterns = [
             '<script', 'javascript:', 'data:',
             'vbscript:', 'onload=', 'onerror='
         ]
-        
+
         if any(pattern in value.lower() for pattern in dangerous_patterns):
             raise serializers.ValidationError('Content contains potentially unsafe elements')
-        
+
         return value
-    
+
     def validate_title(self, value):
         """Validate title format."""
         if not value.strip():
@@ -50,7 +51,7 @@ class DocumentCreateSerializer(serializers.Serializer):
 
 class DocumentSerializer(serializers.ModelSerializer):
     """Document response serializer."""
-    
+
     id = serializers.UUIDField(read_only=True)
     processing_status = serializers.CharField(read_only=True)
     chunks_count = serializers.IntegerField(read_only=True)
@@ -61,7 +62,7 @@ class DocumentSerializer(serializers.ModelSerializer):
     processing_started_at = serializers.DateTimeField(read_only=True)
     processing_completed_at = serializers.DateTimeField(read_only=True)
     processing_error = serializers.CharField(read_only=True)
-    
+
     class Meta:
         model = Document
         fields = [
@@ -74,7 +75,7 @@ class DocumentSerializer(serializers.ModelSerializer):
 
 class DocumentStatsSerializer(serializers.Serializer):
     """Document processing statistics serializer."""
-    
+
     total_documents = serializers.IntegerField()
     completed_documents = serializers.IntegerField()
     processing_success_rate = serializers.FloatField()
@@ -86,9 +87,9 @@ class DocumentStatsSerializer(serializers.Serializer):
 
 class DocumentProcessingStatusSerializer(serializers.Serializer):
     """Document processing status serializer."""
-    
+
     id = serializers.UUIDField()
     status = serializers.CharField()
     progress = serializers.JSONField()
     error = serializers.CharField(allow_null=True, required=False)
-    processing_time_seconds = serializers.FloatField(allow_null=True, required=False)
+    processing_time_seconds = serializers.FloatField(allow_null=True, required=False)