mirror of
https://github.com/csunny/DB-GPT.git
synced 2025-08-13 05:55:54 +00:00
44 lines
1.1 KiB
Python
44 lines
1.1 KiB
Python
"""Module for embedding related classes and functions."""
|
|
|
|
from .embedding_factory import ( # noqa: F401
|
|
DefaultEmbeddingFactory,
|
|
EmbeddingFactory,
|
|
WrappedEmbeddingFactory,
|
|
)
|
|
from .embeddings import ( # noqa: F401
|
|
Embeddings,
|
|
HuggingFaceBgeEmbeddings,
|
|
HuggingFaceEmbeddings,
|
|
HuggingFaceInferenceAPIEmbeddings,
|
|
HuggingFaceInstructEmbeddings,
|
|
JinaEmbeddings,
|
|
OllamaEmbeddings,
|
|
OpenAPIEmbeddings,
|
|
QianFanEmbeddings,
|
|
TongYiEmbeddings,
|
|
)
|
|
from .rerank import ( # noqa: F401
|
|
CrossEncoderRerankEmbeddings,
|
|
OpenAPIRerankEmbeddings,
|
|
SiliconFlowRerankEmbeddings,
|
|
)
|
|
|
|
__ALL__ = [
|
|
"CrossEncoderRerankEmbeddings",
|
|
"DefaultEmbeddingFactory",
|
|
"EmbeddingFactory",
|
|
"Embeddings",
|
|
"HuggingFaceBgeEmbeddings",
|
|
"HuggingFaceEmbeddings",
|
|
"HuggingFaceInferenceAPIEmbeddings",
|
|
"HuggingFaceInstructEmbeddings",
|
|
"JinaEmbeddings",
|
|
"OllamaEmbeddings",
|
|
"OpenAPIEmbeddings",
|
|
"OpenAPIRerankEmbeddings",
|
|
"SiliconFlowRerankEmbeddings",
|
|
"QianFanEmbeddings",
|
|
"TongYiEmbeddings",
|
|
"WrappedEmbeddingFactory",
|
|
]
|