AIRegulation-Demo-Test-Backend/app/services/__init__.py

# Import services
from .minio import minio_service, MinioService
from .database import db_service, DatabaseService, init_db, Document, ParseTask
from .tasks import task_manager, get_task_status, set_task_status, generate_task_id

# Import mock data service (for development)
from .mock_data import (
    get_mock_documents,
    get_mock_quick_questions,
    get_mock_retrieval,
    get_mock_rag_answer,
    get_mock_compliance_result,
    get_mock_compliance_chat_response,
    MOCK_SYSTEM_STATS,
    MOCK_SYSTEM_CONFIG,
)

# Try importing real services (may fail if dependencies not installed)
try:
    from .llm import llm_service, LLMService
    from .embedding import embedding_service, EmbeddingService
    from .milvus import milvus_service, MilvusService
    from .document import DocumentService, get_document_service
    _real_services_available = True
except ImportError:
    _real_services_available = False
    llm_service = None
    LLMService = None
    embedding_service = None
    EmbeddingService = None
    milvus_service = None
    MilvusService = None
    DocumentService = None
    get_document_service = None

__all__ = [
    # Core services
    "minio_service",
    "MinioService",
    "db_service",
    "DatabaseService",
    "init_db",
    "Document",
    "ParseTask",
    "task_manager",
    "get_task_status",
    "set_task_status",
    "generate_task_id",
    # Mock data services
    "get_mock_documents",
    "get_mock_quick_questions",
    "get_mock_retrieval",
    "get_mock_rag_answer",
    "get_mock_compliance_result",
    "get_mock_compliance_chat_response",
    "MOCK_SYSTEM_STATS",
    "MOCK_SYSTEM_CONFIG",
    # Real services (may be None if not available)
    "llm_service",
    "LLMService",
    "embedding_service",
    "EmbeddingService",
    "milvus_service",
    "MilvusService",
    "DocumentService",
    "get_document_service",
]