50 lines
1.4 KiB
Python
50 lines
1.4 KiB
Python
# Import mock data service
|
|
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__ = [
|
|
# 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",
|
|
] |