mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-07-30 07:11:56 +00:00
167d972093
Co-authored-by: hzh97 <2976151305@qq.com> Co-authored-by: Fangyin Cheng <staneyffer@gmail.com> Co-authored-by: licunxing <864255598@qq.com>
61 lines
1.4 KiB
Python
61 lines
1.4 KiB
Python
"""Module Of Knowledge."""
|
|
|
|
from typing import Any, Dict
|
|
|
|
from dbgpt.rag.knowledge.factory import KnowledgeFactory
|
|
|
|
_MODULE_CACHE: Dict[str, Any] = {}
|
|
|
|
|
|
def __getattr__(name: str):
|
|
# Lazy load
|
|
import importlib
|
|
|
|
if name in _MODULE_CACHE:
|
|
return _MODULE_CACHE[name]
|
|
|
|
_LIBS = {
|
|
"KnowledgeFactory": "factory",
|
|
"Knowledge": "base",
|
|
"KnowledgeType": "base",
|
|
"ChunkStrategy": "base",
|
|
"CSVKnowledge": "csv",
|
|
"DatasourceKnowledge": "datasource",
|
|
"DocxKnowledge": "docx",
|
|
"HTMLKnowledge": "html",
|
|
"MarkdownKnowledge": "markdown",
|
|
"PDFKnowledge": "pdf",
|
|
"PPTXKnowledge": "pptx",
|
|
"StringKnowledge": "string",
|
|
"TXTKnowledge": "txt",
|
|
"URLKnowledge": "url",
|
|
"ExcelKnowledge": "xlsx",
|
|
}
|
|
|
|
if name in _LIBS:
|
|
module_path = "." + _LIBS[name]
|
|
module = importlib.import_module(module_path, __name__)
|
|
attr = getattr(module, name)
|
|
_MODULE_CACHE[name] = attr
|
|
return attr
|
|
raise AttributeError(f"module {__name__!r} has no attribute {name!r}")
|
|
|
|
|
|
__all__ = [
|
|
"KnowledgeFactory",
|
|
"Knowledge",
|
|
"KnowledgeType",
|
|
"ChunkStrategy",
|
|
"CSVKnowledge",
|
|
"DatasourceKnowledge",
|
|
"DocxKnowledge",
|
|
"HTMLKnowledge",
|
|
"MarkdownKnowledge",
|
|
"PDFKnowledge",
|
|
"PPTXKnowledge",
|
|
"StringKnowledge",
|
|
"TXTKnowledge",
|
|
"URLKnowledge",
|
|
"ExcelKnowledge",
|
|
]
|