Add fastembed integration

README.md

@@ -40,6 +40,7 @@ pip install qdrant-client
 - Local mode - use same API without running server
 - REST and gRPC support
 - Minimal dependencies
+- Extensive Test Coverage
 
 ## Local mode
 
@@ -66,10 +67,36 @@ Local mode is useful for development, prototyping and testing.
 - Run it in Colab or Jupyter Notebook, no extra dependencies required. See an [example](https://colab.research.google.com/drive/1Bz8RSVHwnNDaNtDwotfPj0w7AYzsdXZ-?usp=sharing)
 - When you need to scale, simply switch to server mode.
 
-### How it works?
+## Fast Embeddings + Simpler API
 
-We just implemented Qdrant API in pure Python.
-We covered it with tests extensively to be sure it works the same as the server version.
+```
+pip install fastembed qdrant-client
+```
+
+FastEmbed is a library for creating fast vector embeddings on CPU. It is based on ONNX Runtime and allows to run inference on CPU with GPU-like performance.
+
+Qdrant Client can use FastEmbed to create embeddings and upload them to Qdrant. This allows to simplify API and make it more intuitive.
+
+```python
+from qdrant_client import QdrantClient
+
+# Initialize the client
+client = QdrantClient(":memory:")  # or QdrantClient(path="path/to/db")
+
+# Prepare your documents, metadata, and IDs
+docs = ["Qdrant has Langchain integrations", "Qdrant also has Llama Index integrations"]
+metadatas = [
+    {"source": "Langchain-docs"},
+    {"source": "Linkedin-docs"},
+]
+ids = [42, 2]
+
+# Use the new add method
+client.add(collection_name="demo_collection", docs={"documents": docs, "metadatas": metadatas, "ids": ids})
+
+search_result = client.query(collection_name="demo_collection", query_texts=["This is a query document"])
+print(search_result)
+```
 
 ## Connect to Qdrant server
 

qdrant_client/qdrant_client.py

@@ -72,6 +72,18 @@ def __init__(
         path: Optional[str] = None,
         **kwargs: Any,
     ):
+        # Check if fastvector is installed
+        try:
+            from fastembed.qdrant_mixin import QdrantClientMixin
+
+            # If it is, add the mixin methods to this instance
+            for name, method in QdrantClientMixin.__dict__.items():
+                if callable(method):
+                    setattr(self, name, method.__get__(self, self.__class__))
+        except ImportError:
+            # If it's not, do nothing
+            pass
+
         self._client: QdrantBase
 
         if location == ":memory:":
@@ -115,7 +127,9 @@ def grpc_collections(self) -> grpc.CollectionsStub:
         if isinstance(self._client, QdrantRemote):
             return self._client.grpc_collections
 
-        raise NotImplementedError(f"gRPC client is not supported for {type(self._client)}")
+        raise NotImplementedError(
+            f"gRPC client is not supported for {type(self._client)}"
+        )
 
     @property
     def grpc_points(self) -> grpc.PointsStub:
@@ -127,7 +141,9 @@ def grpc_points(self) -> grpc.PointsStub:
         if isinstance(self._client, QdrantRemote):
             return self._client.grpc_points
 
-        raise NotImplementedError(f"gRPC client is not supported for {type(self._client)}")
+        raise NotImplementedError(
+            f"gRPC client is not supported for {type(self._client)}"
+        )
 
     @property
     def async_grpc_points(self) -> grpc.PointsStub:
@@ -139,7 +155,9 @@ def async_grpc_points(self) -> grpc.PointsStub:
         if isinstance(self._client, QdrantRemote):
             return self._client.async_grpc_points
 
-        raise NotImplementedError(f"gRPC client is not supported for {type(self._client)}")
+        raise NotImplementedError(
+            f"gRPC client is not supported for {type(self._client)}"
+        )
 
     @property
     def async_grpc_collections(self) -> grpc.CollectionsStub:
@@ -151,7 +169,9 @@ def async_grpc_collections(self) -> grpc.CollectionsStub:
         if isinstance(self._client, QdrantRemote):
             return self._client.async_grpc_collections
 
-        raise NotImplementedError(f"gRPC client is not supported for {type(self._client)}")
+        raise NotImplementedError(
+            f"gRPC client is not supported for {type(self._client)}"
+        )
 
     @property
     def rest(self) -> SyncApis[ApiClient]:
@@ -163,7 +183,9 @@ def rest(self) -> SyncApis[ApiClient]:
         if isinstance(self._client, QdrantRemote):
             return self._client.rest
 
-        raise NotImplementedError(f"REST client is not supported for {type(self._client)}")
+        raise NotImplementedError(
+            f"REST client is not supported for {type(self._client)}"
+        )
 
     @property
     def http(self) -> SyncApis[ApiClient]:
@@ -175,7 +197,9 @@ def http(self) -> SyncApis[ApiClient]:
         if isinstance(self._client, QdrantRemote):
             return self._client.http
 
-        raise NotImplementedError(f"REST client is not supported for {type(self._client)}")
+        raise NotImplementedError(
+            f"REST client is not supported for {type(self._client)}"
+        )
 
     def search_batch(
         self,
@@ -1173,7 +1197,9 @@ def get_collection_aliases(
         """
         assert len(kwargs) == 0, f"Unknown arguments: {list(kwargs.keys())}"
 
-        return self._client.get_collection_aliases(collection_name=collection_name, **kwargs)
+        return self._client.get_collection_aliases(
+            collection_name=collection_name, **kwargs
+        )
 
     def get_aliases(self, **kwargs: Any) -> types.CollectionsAliasesResponse:
         """Get all aliases
@@ -1195,7 +1221,9 @@ def get_collections(self, **kwargs: Any) -> types.CollectionsResponse:
 
         return self._client.get_collections(**kwargs)
 
-    def get_collection(self, collection_name: str, **kwargs: Any) -> types.CollectionInfo:
+    def get_collection(
+        self, collection_name: str, **kwargs: Any
+    ) -> types.CollectionInfo:
         """Get detailed information about specified existing collection
 
         Args:
@@ -1601,7 +1629,9 @@ def create_snapshot(
 
         return self._client.create_snapshot(collection_name=collection_name, **kwargs)
 
-    def delete_snapshot(self, collection_name: str, snapshot_name: str, **kwargs: Any) -> bool:
+    def delete_snapshot(
+        self, collection_name: str, snapshot_name: str, **kwargs: Any
+    ) -> bool:
         """Delete snapshot for a given collection.
 
         Args:
@@ -1699,3 +1729,4 @@ def get_locks(self, **kwargs: Any) -> types.LocksOption:
         assert len(kwargs) == 0, f"Unknown arguments: {list(kwargs.keys())}"
 
         return self._client.get_locks(**kwargs)
+        return self._client.get_locks(**kwargs)