33 lines
1,010 B
Python
33 lines
1,010 B
Python
from cognee.infrastructure.loaders.external import PyPdfLoader
|
|
from cognee.infrastructure.loaders.core import TextLoader, AudioLoader, ImageLoader, CsvLoader
|
|
|
|
# Registry for loader implementations
|
|
supported_loaders = {
|
|
PyPdfLoader.loader_name: PyPdfLoader,
|
|
TextLoader.loader_name: TextLoader,
|
|
ImageLoader.loader_name: ImageLoader,
|
|
AudioLoader.loader_name: AudioLoader,
|
|
CsvLoader.loader_name: CsvLoader,
|
|
}
|
|
|
|
# Try adding optional loaders
|
|
try:
|
|
from cognee.infrastructure.loaders.external import UnstructuredLoader
|
|
|
|
supported_loaders[UnstructuredLoader.loader_name] = UnstructuredLoader
|
|
except ImportError:
|
|
pass
|
|
|
|
try:
|
|
from cognee.infrastructure.loaders.external import AdvancedPdfLoader
|
|
|
|
supported_loaders[AdvancedPdfLoader.loader_name] = AdvancedPdfLoader
|
|
except ImportError:
|
|
pass
|
|
|
|
try:
|
|
from cognee.infrastructure.loaders.external import BeautifulSoupLoader
|
|
|
|
supported_loaders[BeautifulSoupLoader.loader_name] = BeautifulSoupLoader
|
|
except ImportError:
|
|
pass
|