mega changes

backend/app/api/internal_v1/debugging.py

@@ -20,23 +20,28 @@ router = APIRouter()
 async def get_chatbot_config_debug(
     chatbot_id: str,
     db: Session = Depends(get_db),
-    current_user: User = Depends(get_current_user)
+    current_user: User = Depends(get_current_user),
 ):
     """Get detailed configuration for debugging a specific chatbot"""
 
     # Get chatbot instance
-    chatbot = db.query(ChatbotInstance).filter(
-        ChatbotInstance.id == chatbot_id,
-        ChatbotInstance.user_id == current_user.id
-    ).first()
+    chatbot = (
+        db.query(ChatbotInstance)
+        .filter(
+            ChatbotInstance.id == chatbot_id, ChatbotInstance.user_id == current_user.id
+        )
+        .first()
+    )
 
     if not chatbot:
         raise HTTPException(status_code=404, detail="Chatbot not found")
 
     # Get prompt template
-    prompt_template = db.query(PromptTemplate).filter(
-        PromptTemplate.type == chatbot.chatbot_type
-    ).first()
+    prompt_template = (
+        db.query(PromptTemplate)
+        .filter(PromptTemplate.type == chatbot.chatbot_type)
+        .first()
+    )
 
     # Get RAG collections if configured
     rag_collections = []
@@ -44,31 +49,37 @@ async def get_chatbot_config_debug(
         collection_ids = chatbot.rag_collection_ids
         if isinstance(collection_ids, str):
             import json
+
             try:
                 collection_ids = json.loads(collection_ids)
             except:
                 collection_ids = []
 
         if collection_ids:
-            collections = db.query(RagCollection).filter(
-                RagCollection.id.in_(collection_ids)
-            ).all()
+            collections = (
+                db.query(RagCollection)
+                .filter(RagCollection.id.in_(collection_ids))
+                .all()
+            )
             rag_collections = [
                 {
                     "id": col.id,
                     "name": col.name,
                     "document_count": col.document_count,
                     "qdrant_collection_name": col.qdrant_collection_name,
-                    "is_active": col.is_active
+                    "is_active": col.is_active,
                 }
                 for col in collections
             ]
 
     # Get recent conversations count
     from app.models.chatbot import ChatbotConversation
-    conversation_count = db.query(ChatbotConversation).filter(
-        ChatbotConversation.chatbot_instance_id == chatbot_id
-    ).count()
+
+    conversation_count = (
+        db.query(ChatbotConversation)
+        .filter(ChatbotConversation.chatbot_instance_id == chatbot_id)
+        .count()
+    )
 
     return {
         "chatbot": {
@@ -78,20 +89,20 @@ async def get_chatbot_config_debug(
             "description": chatbot.description,
             "created_at": chatbot.created_at,
             "is_active": chatbot.is_active,
-            "conversation_count": conversation_count
+            "conversation_count": conversation_count,
         },
         "prompt_template": {
             "type": prompt_template.type if prompt_template else None,
             "system_prompt": prompt_template.system_prompt if prompt_template else None,
-            "variables": prompt_template.variables if prompt_template else []
+            "variables": prompt_template.variables if prompt_template else [],
         },
         "rag_collections": rag_collections,
         "configuration": {
             "max_tokens": chatbot.max_tokens,
             "temperature": chatbot.temperature,
             "streaming": chatbot.streaming,
-            "memory_config": chatbot.memory_config
-        }
+            "memory_config": chatbot.memory_config,
+        },
     }
 
 
@@ -101,15 +112,18 @@ async def test_rag_search(
     query: str = "test query",
     top_k: int = 5,
     db: Session = Depends(get_db),
-    current_user: User = Depends(get_current_user)
+    current_user: User = Depends(get_current_user),
 ):
     """Test RAG search for a specific chatbot"""
 
     # Get chatbot instance
-    chatbot = db.query(ChatbotInstance).filter(
-        ChatbotInstance.id == chatbot_id,
-        ChatbotInstance.user_id == current_user.id
-    ).first()
+    chatbot = (
+        db.query(ChatbotInstance)
+        .filter(
+            ChatbotInstance.id == chatbot_id, ChatbotInstance.user_id == current_user.id
+        )
+        .first()
+    )
 
     if not chatbot:
         raise HTTPException(status_code=404, detail="Chatbot not found")
@@ -123,6 +137,7 @@ async def test_rag_search(
         if chatbot.rag_collection_ids:
             if isinstance(chatbot.rag_collection_ids, str):
                 import json
+
                 try:
                     collection_ids = json.loads(chatbot.rag_collection_ids)
                 except:
@@ -134,22 +149,19 @@ async def test_rag_search(
             return {
                 "query": query,
                 "results": [],
-                "message": "No RAG collections configured for this chatbot"
+                "message": "No RAG collections configured for this chatbot",
             }
 
         # Perform search
         search_results = await rag_module.search(
-            query=query,
-            collection_ids=collection_ids,
-            top_k=top_k,
-            score_threshold=0.5
+            query=query, collection_ids=collection_ids, top_k=top_k, score_threshold=0.5
         )
 
         return {
             "query": query,
             "results": search_results,
             "collections_searched": collection_ids,
-            "result_count": len(search_results)
+            "result_count": len(search_results),
         }
 
     except Exception as e:
@@ -157,14 +169,13 @@ async def test_rag_search(
             "query": query,
             "results": [],
             "error": str(e),
-            "message": "RAG search failed"
+            "message": "RAG search failed",
         }
 
 
 @router.get("/system/status")
 async def get_system_status(
-    db: Session = Depends(get_db),
-    current_user: User = Depends(get_current_user)
+    db: Session = Depends(get_db), current_user: User = Depends(get_current_user)
 ):
     """Get system status for debugging"""
 
@@ -179,11 +190,12 @@ async def get_system_status(
     module_status = {}
     try:
         from app.services.module_manager import module_manager
+
         modules = module_manager.list_modules()
         for module_name, module_info in modules.items():
             module_status[module_name] = {
                 "status": module_info.get("status", "unknown"),
-                "enabled": module_info.get("enabled", False)
+                "enabled": module_info.get("enabled", False),
             }
     except Exception as e:
         module_status = {"error": str(e)}
@@ -192,6 +204,7 @@ async def get_system_status(
     redis_status = "not configured"
     try:
         from app.core.cache import core_cache
+
         await core_cache.ping()
         redis_status = "healthy"
     except Exception as e:
@@ -201,6 +214,7 @@ async def get_system_status(
     qdrant_status = "not configured"
     try:
         from app.services.qdrant_service import qdrant_service
+
         collections = await qdrant_service.list_collections()
         qdrant_status = f"healthy ({len(collections)} collections)"
     except Exception as e:
@@ -211,5 +225,5 @@ async def get_system_status(
         "modules": module_status,
         "redis": redis_status,
         "qdrant": qdrant_status,
-        "timestamp": "UTC"
-    }
+        "timestamp": "UTC",
+    }