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

django_cfg/apps/knowbase/examples/external_data_usage.py

@@ -6,28 +6,31 @@
 """
 
 from django.contrib.auth import get_user_model
-from django_cfg.apps.knowbase.utils.external_data_manager import ExternalDataManager, quick_add_model, quick_search
-from django_cfg.apps.knowbase.models.external_data import ExternalDataType
+
+from django_cfg.apps.knowbase.utils.external_data_manager import (
+    ExternalDataManager,
+    quick_search,
+)
 
 User = get_user_model()
 
 
 def example_add_django_model():
     """Пример добавления Django модели как внешнего источника данных."""
-    
+
     # Получаем пользователя
     user = User.objects.first()
     if not user:
         print("❌ No users found")
         return
-    
+
     # Создаем менеджер
     manager = ExternalDataManager(user)
-    
+
     # Добавляем модель Vehicle (если она существует)
     try:
         from apps.vehicles_data.models import Vehicle
-        
+
         external_data = manager.add_django_model(
             model_class=Vehicle,
             title="Vehicle Database",
@@ -38,14 +41,14 @@ def example_add_django_model():
             overlap_size=150,
             auto_vectorize=True
         )
-        
+
         print(f"✅ Added Vehicle model as external data: {external_data.id}")
         print(f"   Status: {external_data.status}")
         print(f"   Total chunks: {external_data.total_chunks}")
-        
+
     except ImportError:
         print("⚠️ Vehicle model not found, using example data instead")
-        
+
         # Добавляем произвольные данные
         external_data = manager.add_custom_data(
             title="Sample Car Data",
@@ -59,27 +62,27 @@ def example_add_django_model():
             description="Sample car data for testing",
             tags=['cars', 'vehicles', 'sample']
         )
-        
+
         print(f"✅ Added sample car data: {external_data.id}")
 
 
 def example_search_external_data():
     """Пример поиска по внешним данным."""
-    
+
     user = User.objects.first()
     if not user:
         print("❌ No users found")
         return
-    
+
     manager = ExternalDataManager(user)
-    
+
     # Поиск по запросу
     results = manager.search(
         query="reliable car with good fuel economy",
         limit=3,
         threshold=0.6
     )
-    
+
     print(f"🔍 Search results ({len(results)} found):")
     for i, result in enumerate(results, 1):
         print(f"  {i}. {result['source_title']} (similarity: {result['similarity']:.3f})")
@@ -89,15 +92,15 @@ def example_search_external_data():
 
 def example_get_statistics():
     """Пример получения статистики."""
-    
+
     user = User.objects.first()
     if not user:
         print("❌ No users found")
         return
-    
+
     manager = ExternalDataManager(user)
     stats = manager.get_statistics()
-    
+
     print("📊 External Data Statistics:")
     print(f"   Total sources: {stats.total_sources}")
     print(f"   Active sources: {stats.active_sources}")
@@ -111,15 +114,15 @@ def example_get_statistics():
 
 def example_health_check():
     """Пример проверки здоровья системы."""
-    
+
     user = User.objects.first()
     if not user:
         print("❌ No users found")
         return
-    
+
     manager = ExternalDataManager(user)
     health = manager.health_check()
-    
+
     print("🏥 System Health Check:")
     print(f"   Status: {health.status}")
     print(f"   Healthy: {'✅' if health.healthy else '❌'}")
@@ -130,7 +133,7 @@ def example_health_check():
     print(f"   Active sources: {health.active_sources}")
     print(f"   Pending processing: {health.pending_processing}")
     print(f"   Failed processing: {health.failed_processing}")
-    
+
     if health.issues:
         print(f"   Issues: {health.issues}")
     if health.warnings:
@@ -139,19 +142,19 @@ def example_health_check():
 
 def example_quick_functions():
     """Пример использования быстрых функций."""
-    
+
     user = User.objects.first()
     if not user:
         print("❌ No users found")
         return
-    
+
     # Быстрый поиск
     results = quick_search(
         user=user,
         query="electric vehicle",
         limit=2
     )
-    
+
     print(f"⚡ Quick search results: {len(results)} found")
     for result in results:
         print(f"   - {result['source_title']}: {result['similarity']:.3f}")
@@ -159,28 +162,28 @@ def example_quick_functions():
 
 def run_all_examples():
     """Запустить все примеры."""
-    
+
     print("🚀 Running External Data Manager Examples")
     print("=" * 50)
-    
+
     try:
         print("\n1. Adding Django Model:")
         example_add_django_model()
-        
+
         print("\n2. Searching External Data:")
         example_search_external_data()
-        
+
         print("\n3. Getting Statistics:")
         example_get_statistics()
-        
+
         print("\n4. Health Check:")
         example_health_check()
-        
+
         print("\n5. Quick Functions:")
         example_quick_functions()
-        
+
         print("\n✅ All examples completed successfully!")
-        
+
     except Exception as e:
         print(f"\n❌ Error running examples: {e}")
         import traceback

django_cfg/apps/knowbase/management/commands/knowbase_stats.py

@@ -2,20 +2,20 @@
 Knowledge Base statistics command.
 """
 
-from django.core.management.base import BaseCommand
-from django.db.models import Count, Sum, Avg, Q, F, ExpressionWrapper
-from django.db.models.functions import Extract
 from django.contrib.auth import get_user_model
+from django.core.management.base import BaseCommand
 from django.db import models
+from django.db.models import Avg, Count, ExpressionWrapper, F, Q, Sum
+from django.db.models.functions import Extract
 
 User = get_user_model()
 
 
 class Command(BaseCommand):
     """Display Knowledge Base statistics."""
-    
+
     help = 'Display Knowledge Base usage statistics'
-    
+
     def add_arguments(self, parser):
         parser.add_argument(
             '--user',
@@ -27,17 +27,17 @@ class Command(BaseCommand):
             action='store_true',
             help='Show detailed statistics',
         )
-    
+
     def handle(self, *args, **options):
         """Display statistics."""
-        
-        from ...models import Document, DocumentChunk, ChatSession, ChatMessage
-        
+
+        from ...models import ChatSession, Document
+
         self.stdout.write(
             self.style.SUCCESS('📊 Knowledge Base Statistics')
         )
         self.stdout.write('=' * 50)
-        
+
         # Filter by user if specified
         user_filter = {}
         if options['user']:
@@ -51,14 +51,14 @@ class Command(BaseCommand):
                     self.style.ERROR(f"User '{options['user']}' not found")
                 )
                 return
-        
+
         # Document statistics
         # Calculate processing duration in database
         processing_time_expr = ExpressionWrapper(
             Extract(F('processing_completed_at') - F('processing_started_at'), 'epoch'),
             output_field=models.FloatField()
         )
-        
+
         doc_stats = Document.objects.filter(**user_filter).aggregate(
             total_docs=Count('id'),
             completed_docs=Count('id', filter=Q(processing_status='completed')),
@@ -70,19 +70,19 @@ class Command(BaseCommand):
                 processing_completed_at__isnull=False
             ))
         )
-        
+
         self.stdout.write("📄 Documents:")
         self.stdout.write(f"  Total: {doc_stats['total_docs'] or 0}")
         self.stdout.write(f"  Completed: {doc_stats['completed_docs'] or 0}")
         self.stdout.write(f"  Success Rate: {((doc_stats['completed_docs'] or 0) / (doc_stats['total_docs'] or 1) * 100):.1f}%")
-        
+
         self.stdout.write("\n📝 Content:")
         self.stdout.write(f"  Total Chunks: {doc_stats['total_chunks'] or 0}")
         self.stdout.write(f"  Total Tokens: {doc_stats['total_tokens'] or 0}")
-        
+
         self.stdout.write("\n💰 Costs:")
         self.stdout.write(f"  Total Cost: ${(doc_stats['total_cost'] or 0):.6f}")
-        
+
         # Chat statistics
         chat_stats = ChatSession.objects.filter(**user_filter).aggregate(
             total_sessions=Count('id'),
@@ -91,68 +91,69 @@ class Command(BaseCommand):
             total_chat_tokens=Sum('total_tokens_used'),
             total_chat_cost=Sum('total_cost_usd')
         )
-        
+
         self.stdout.write("\n💬 Chat:")
         self.stdout.write(f"  Total Sessions: {chat_stats['total_sessions'] or 0}")
         self.stdout.write(f"  Active Sessions: {chat_stats['active_sessions'] or 0}")
         self.stdout.write(f"  Total Messages: {chat_stats['total_messages'] or 0}")
         self.stdout.write(f"  Chat Tokens: {chat_stats['total_chat_tokens'] or 0}")
         self.stdout.write(f"  Chat Cost: ${(chat_stats['total_chat_cost'] or 0):.6f}")
-        
+
         # Detailed statistics
         if options['detailed']:
             self.show_detailed_stats(user_filter)
-    
+
     def show_detailed_stats(self, user_filter):
         """Show detailed statistics."""
-        from ...models import Document, ChatSession
-        
+        from ...models import ChatSession, Document
+
         self.stdout.write("\n" + "=" * 50)
         self.stdout.write("📈 Detailed Statistics")
         self.stdout.write("=" * 50)
-        
+
         # Processing status breakdown
         status_counts = Document.objects.filter(**user_filter).values(
             'processing_status'
         ).annotate(count=Count('id'))
-        
+
         self.stdout.write("\n📊 Document Status Breakdown:")
         for status in status_counts:
             self.stdout.write(f"  {status['processing_status']}: {status['count']}")
-        
+
         # Model usage breakdown
         model_counts = ChatSession.objects.filter(**user_filter).values(
             'model_name'
         ).annotate(count=Count('id'))
-        
+
         self.stdout.write("\n🤖 Model Usage:")
         for model in model_counts:
             self.stdout.write(f"  {model['model_name']}: {model['count']} sessions")
-        
+
         # Top documents by cost
         top_docs = Document.objects.filter(
             **user_filter
         ).order_by('-total_cost_usd')[:5]
-        
+
         self.stdout.write("\n💸 Most Expensive Documents:")
         for doc in top_docs:
             self.stdout.write(f"  {doc.title[:40]}...: ${doc.total_cost_usd:.6f}")
-        
+
         # Recent activity
-        from django.utils import timezone
         from datetime import timedelta
-        
+
+        from django.utils import timezone
+
         week_ago = timezone.now() - timedelta(days=7)
         recent_docs = Document.objects.filter(
             **user_filter,
             created_at__gte=week_ago
         ).count()
-        
+
         recent_sessions = ChatSession.objects.filter(
             **user_filter,
             created_at__gte=week_ago
         ).count()
-        
+
         self.stdout.write("\n📅 Recent Activity (Last 7 Days):")
         self.stdout.write(f"  New Documents: {recent_docs}")
         self.stdout.write(f"  New Chat Sessions: {recent_sessions}")

django_cfg/apps/knowbase/management/commands/setup_knowbase.py

@@ -4,51 +4,50 @@ Setup command for Knowledge Base application.
 
 from django.core.management.base import BaseCommand
 from django.db import connection
-from django.core.management import call_command
 
 
 class Command(BaseCommand):
     """Setup Knowledge Base with pgvector extension and initial data."""
-    
+
     help = 'Setup Knowledge Base with pgvector extension and run migrations'
-    
+
     def add_arguments(self, parser):
         parser.add_argument(
             '--skip-extensions',
             action='store_true',
             help='Skip creating PostgreSQL extensions',
         )
-    
+
     def handle(self, *args, **options):
         """Execute the setup process."""
-        
+
         self.stdout.write(
             self.style.SUCCESS('🚀 Setting up Knowledge Base...')
         )
-        
+
         # Step 1: Create PostgreSQL extensions
         if not options['skip_extensions']:
             self.create_extensions()
-        
-        
+
+
         self.stdout.write(
             self.style.SUCCESS('✅ Knowledge Base setup completed!')
         )
-    
+
     def create_extensions(self):
         """Create required PostgreSQL extensions."""
         self.stdout.write('📦 Creating PostgreSQL extensions...')
-        
+
         try:
             with connection.cursor() as cursor:
                 # Create pgvector extension
                 cursor.execute("CREATE EXTENSION IF NOT EXISTS vector;")
                 self.stdout.write('  ✓ pgvector extension created')
-                
+
                 # Create pg_trgm for text search
                 cursor.execute("CREATE EXTENSION IF NOT EXISTS pg_trgm;")
                 self.stdout.write('  ✓ pg_trgm extension created')
-                
+
         except Exception as e:
             self.stdout.write(
                 self.style.ERROR(f'  ❌ Failed to create extensions: {e}')
@@ -56,4 +55,3 @@ class Command(BaseCommand):
             self.stdout.write(
                 self.style.WARNING('  ⚠️  You may need to create extensions manually as superuser')
             )
-    

django_cfg/apps/knowbase/managers/__init__.py

@@ -2,10 +2,10 @@
 Custom managers for knowledge base models.
 """
 
+from .archive import *
 from .base import *
-from .document import *
 from .chat import *
-from .archive import *
+from .document import *
 from .external_data import *
 
 __all__ = [