mirror of
https://github.com/hwchase17/langchain
synced 2024-11-02 09:40:22 +00:00
2f8dd1a161
Issue `langchain_community.cross_encoders` didn't have flattening namespace code in the __init__.py file. Changes: - added code to flattening namespaces (used #20050 as a template) - added ut for a change - added missed `test_imports` for `chat_loaders` and `chat_message_histories` modules
49 lines
1.5 KiB
Python
49 lines
1.5 KiB
Python
"""**Cross encoders** are wrappers around cross encoder models from different APIs and
|
|
services.
|
|
|
|
**Cross encoder models** can be LLMs or not.
|
|
|
|
**Class hierarchy:**
|
|
|
|
.. code-block::
|
|
|
|
BaseCrossEncoder --> <name>CrossEncoder # Examples: SagemakerEndpointCrossEncoder
|
|
"""
|
|
import importlib
|
|
from typing import TYPE_CHECKING, Any
|
|
|
|
if TYPE_CHECKING:
|
|
from langchain_community.cross_encoders.base import (
|
|
BaseCrossEncoder, # noqa: F401
|
|
)
|
|
from langchain_community.cross_encoders.fake import (
|
|
FakeCrossEncoder, # noqa: F401
|
|
)
|
|
from langchain_community.cross_encoders.huggingface import (
|
|
HuggingFaceCrossEncoder, # noqa: F401
|
|
)
|
|
from langchain_community.cross_encoders.sagemaker_endpoint import (
|
|
SagemakerEndpointCrossEncoder, # noqa: F401
|
|
)
|
|
|
|
__all__ = [
|
|
"BaseCrossEncoder",
|
|
"FakeCrossEncoder",
|
|
"HuggingFaceCrossEncoder",
|
|
"SagemakerEndpointCrossEncoder",
|
|
]
|
|
|
|
_module_lookup = {
|
|
"BaseCrossEncoder": "langchain_community.cross_encoders.base",
|
|
"FakeCrossEncoder": "langchain_community.cross_encoders.fake",
|
|
"HuggingFaceCrossEncoder": "langchain_community.cross_encoders.huggingface",
|
|
"SagemakerEndpointCrossEncoder": "langchain_community.cross_encoders.sagemaker_endpoint", # noqa: E501
|
|
}
|
|
|
|
|
|
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}")
|