Spaces:
Runtime error
Runtime error
File size: 3,673 Bytes
e14ce19 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 | from fastapi import APIRouter, Request, Query
from app.core.session_manager import get_session_manager
from app.core.rag_manager import get_rag_manager
from app.core.bootstrap import chat_orchestrator
import logging
from app.api.old_routes import get_or_create_session_for_request
logger = logging.getLogger(__name__)
router = APIRouter()
session_manager = get_session_manager()
@router.get("/debug/personas")
async def debug_personas(request: Request):
try:
session_id = get_or_create_session_for_request(request)
session = session_manager.get_session(session_id)
rag_manager = get_rag_manager()
rag_stats = rag_manager.get_document_stats(session_id)
return {
"personas": {
pid: {
"name": persona.name,
"prompt": persona.system_prompt[:100] + "...",
"retrieval_keywords": chat_orchestrator._get_persona_context_keywords(pid)
} for pid, persona in chat_orchestrator.personas.items()
},
"session_info": {
"context_length": len(session.messages),
"uploaded_files": session.uploaded_files,
"rag_stats": rag_stats
}
}
except Exception as e:
logger.error(f"Error in debug endpoint: {str(e)}")
return {
"personas": {},
"session_info": {"context_length": 0},
"error": str(e)
}
@router.get("/debug/ranked-personas")
async def get_ranked_personas(request: Request, k: int = Query(3, ge=1, le=10)):
try:
session_id = get_or_create_session_for_request(request)
top_personas = await chat_orchestrator.get_top_personas(session_id=session_id, k=k)
return {
"ranked_personas": top_personas,
"available_personas": list(chat_orchestrator.personas.keys()),
"session_id": session_id
}
except Exception as e:
logger.error(f"Error in /debug/ranked-personas: {e}")
return {
"ranked_personas": [],
"error": str(e)
}
@router.get("/debug/rag-status")
async def debug_rag_status(request: Request):
try:
session_id = get_or_create_session_for_request(request)
rag_manager = get_rag_manager()
session_stats = session_manager.get_session_stats(session_id)
test_search = rag_manager.search_documents(
query="test methodology research",
session_id=session_id,
persona_context="",
n_results=3
)
return {
"rag_manager_healthy": True,
"session_id": session_id,
"session_stats": session_stats.get("rag_stats", {}),
"test_search_results": len(test_search),
"test_search_details": [
{
"relevance": chunk.get("relevance_score", 0),
"distance": chunk.get("distance", "unknown"),
"text_length": len(chunk.get("text", "")),
"filename": chunk.get("metadata", {}).get("filename", "unknown")
}
for chunk in test_search[:3]
],
"persona_keywords": {
pid: chat_orchestrator._get_persona_context_keywords(pid)
for pid in chat_orchestrator.personas.keys()
}
}
except Exception as e:
logger.error(f"Error in RAG debug: {str(e)}")
return {
"rag_manager_healthy": False,
"error": str(e),
"session_id": session_id if 'session_id' in locals() else "unknown"
}
|