new: simplify imports (#171)

* new: simplify imports

* refactoring: update import

Author: joein

README.md

@@ -2,7 +2,7 @@
 
 FastEmbed is a lightweight, fast, Python library built for embedding generation. We [support popular text models](https://qdrant.github.io/fastembed/examples/Supported_Models/). Please [open a GitHub issue](https://github.com/qdrant/fastembed/issues/new) if you want us to add a new model.
 
-The default text embedding (`TextEmbedding`) model is Flag Embedding, the top model in the [MTEB](https://huggingface.co/spaces/mteb/leaderboard) leaderboard. It supports "query" and "passage" prefixes for the input text. Here is an example for [Retrieval Embedding Generation](https://qdrant.github.io/fastembed/examples/Retrieval_with_FastEmbed/) and how to use [FastEmbed with Qdrant](https://qdrant.github.io/fastembed/examples/Usage_With_Qdrant/).
+The default text embedding (`TextEmbedding`) model is Flag Embedding, presented in the [MTEB](https://huggingface.co/spaces/mteb/leaderboard) leaderboard. It supports "query" and "passage" prefixes for the input text. Here is an example for [Retrieval Embedding Generation](https://qdrant.github.io/fastembed/examples/Retrieval_with_FastEmbed/) and how to use [FastEmbed with Qdrant](https://qdrant.github.io/fastembed/examples/Usage_With_Qdrant/).
 
 ## 📈 Why FastEmbed?
 
@@ -23,7 +23,6 @@ pip install fastembed
 ## 📖 Quickstart
 
 ```python
-import numpy as np
 from fastembed import TextEmbedding
 from typing import List
 

fastembed/__init__.py

@@ -1,6 +1,7 @@
 import importlib.metadata
 
-from fastembed.text.text_embedding import TextEmbedding
+from fastembed.text import TextEmbedding
+from fastembed.sparse import SparseTextEmbedding, SparseEmbedding
 
 __version__ = importlib.metadata.version("fastembed")
-__all__ = ["TextEmbedding"]
+__all__ = ["TextEmbedding", "SparseTextEmbedding", "SparseEmbedding"]

fastembed/embedding.py

@@ -2,7 +2,7 @@
 
 from loguru import logger
 
-from fastembed.text.text_embedding import TextEmbedding
+from fastembed import TextEmbedding
 
 logger.warning(
     "DefaultEmbedding, FlagEmbedding, JinaEmbedding are deprecated."

fastembed/sparse/__init__.py

@@ -0,0 +1,4 @@
+from fastembed.sparse.sparse_embedding_base import SparseEmbedding
+from fastembed.sparse.sparse_text_embedding import SparseTextEmbedding
+
+__all__ = ["SparseEmbedding", "SparseTextEmbedding"]

fastembed/text/__init__.py

@@ -0,0 +1,3 @@
+from fastembed.text.text_embedding import TextEmbedding
+
+__all__ = ["TextEmbedding"]