mirror of
https://github.com/hwchase17/langchain
synced 2024-11-11 19:11:02 +00:00
4cb5f4c353
This PR should make it easier for linters to do type checking and for IDEs to jump to definition of code. See #20050 as a template for this PR. - As a byproduct: Added 3 missed `test_imports`. - Added missed `SolarChat` in to __init___.py Added it into test_import ut. - Added `# type: ignore` to fix linting. It is not clear, why linting errors appear after ^ changes. --------- Co-authored-by: Eugene Yurtsev <eyurtsev@gmail.com>
49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
"""**Docstores** are classes to store and load Documents.
|
|
|
|
The **Docstore** is a simplified version of the Document Loader.
|
|
|
|
**Class hierarchy:**
|
|
|
|
.. code-block::
|
|
|
|
Docstore --> <name> # Examples: InMemoryDocstore, Wikipedia
|
|
|
|
**Main helpers:**
|
|
|
|
.. code-block::
|
|
|
|
Document, AddableMixin
|
|
"""
|
|
|
|
import importlib
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
if TYPE_CHECKING:
|
|
from langchain_community.docstore.arbitrary_fn import (
|
|
DocstoreFn, # noqa: F401
|
|
)
|
|
from langchain_community.docstore.in_memory import (
|
|
InMemoryDocstore, # noqa: F401
|
|
)
|
|
from langchain_community.docstore.wikipedia import (
|
|
Wikipedia, # noqa: F401
|
|
)
|
|
|
|
__all__ = ["DocstoreFn", "InMemoryDocstore", "Wikipedia"]
|
|
|
|
_module_lookup = {
|
|
"DocstoreFn": "langchain_community.docstore.arbitrary_fn",
|
|
"InMemoryDocstore": "langchain_community.docstore.in_memory",
|
|
"Wikipedia": "langchain_community.docstore.wikipedia",
|
|
}
|
|
|
|
|
|
def __getattr__(name: str) -> Any:
|
|
if name in _module_lookup:
|
|
module = importlib.import_module(_module_lookup[name])
|
|
return getattr(module, name)
|
|
raise AttributeError(f"module {__name__} has no attribute {name}")
|
|
|
|
|
|
__all__ = list(_module_lookup.keys())
|