hammad-python 0.0.19__py3-none-any.whl → 0.0.20__py3-none-any.whl

hammad/data/collections/collection.py

@@ -28,7 +28,7 @@ if TYPE_CHECKING:
         DistanceMetric,
     )
     from ..sql.types import DatabaseItemType
-    from ...genai.embedding_models.embedding_model_name import EmbeddingModelName
+    from ...genai.models.embeddings.types import EmbeddingModelName
 else:
     from .indexes.tantivy.index import TantivyCollectionIndex
     from .indexes.qdrant.index import QdrantCollectionIndex, VectorSearchResult
@@ -44,11 +44,11 @@ class Collection:
     """
     A unified collection factory that creates the appropriate collection index type
     based on the provided parameters.
-    
+
     This class acts as a factory and doesn't contain its own logic - it simply
     returns instances of TantivyCollectionIndex or QdrantCollectionIndex based on the
     vector parameter.
-    
+
     The main difference from the old approach is that now collections are 'unified'
     - there's no separate collections interface. Each collection directly uses either
     a Tantivy or Qdrant index with SQL Database as the storage backend.
@@ -83,7 +83,9 @@ class Collection:
         distance_metric: "DistanceMetric" = "dot",
         settings: Optional["QdrantCollectionIndexSettings"] = None,
         query_settings: Optional["QdrantCollectionIndexQuerySettings"] = None,
-        embedding_model: Optional["EmbeddingModelName"] = "openai/text-embedding-3-small",
+        embedding_model: Optional[
+            "EmbeddingModelName"
+        ] = "openai/text-embedding-3-small",
         embedding_dimensions: Optional[int] = None,
         embedding_api_key: Optional[str] = None,
         embedding_base_url: Optional[str] = None,
@@ -105,11 +107,20 @@ class Collection:
         # Tantivy-specific parameters
         fast: bool = True,
         # Unified settings parameters
-        settings: Optional[Union["TantivyCollectionIndexSettings", "QdrantCollectionIndexSettings"]] = None,
-        query_settings: Optional[Union["TantivyCollectionIndexQuerySettings", "QdrantCollectionIndexQuerySettings"]] = None,
+        settings: Optional[
+            Union["TantivyCollectionIndexSettings", "QdrantCollectionIndexSettings"]
+        ] = None,
+        query_settings: Optional[
+            Union[
+                "TantivyCollectionIndexQuerySettings",
+                "QdrantCollectionIndexQuerySettings",
+            ]
+        ] = None,
         # Vector/Qdrant-specific parameters
         distance_metric: "DistanceMetric" = "dot",
-        embedding_model: Optional["EmbeddingModelName"] = "openai/text-embedding-3-small",
+        embedding_model: Optional[
+            "EmbeddingModelName"
+        ] = "openai/text-embedding-3-small",
         embedding_dimensions: Optional[int] = None,
         embedding_api_key: Optional[str] = None,
         embedding_base_url: Optional[str] = None,
@@ -120,7 +131,7 @@ class Collection:
     ) -> Union["TantivyCollectionIndex", "QdrantCollectionIndex"]:
         """
         Create a collection of the specified type.
-        
+
         Args:
             name: Name of the collection
             schema: Optional schema type for validation
@@ -128,26 +139,26 @@ class Collection:
             path: File path for storage (None = in-memory)
             vector: Whether this is a vector collection (True) or text search collection (False)
             vector_size: Size of vectors (required for vector collections)
-            
+
             # Tantivy parameters (for non-vector collections):
             fast: Whether to use fast schema building & indexing
-            
+
             # Unified parameters:
             settings: Collection settings (TantivyCollectionIndexSettings or QdrantCollectionIndexSettings)
             query_settings: Query behavior settings (TantivyCollectionIndexQuerySettings or QdrantCollectionIndexQuerySettings)
-            
+
             # Qdrant parameters (for vector collections):
             distance_metric: Distance metric for similarity search
             embedding_model: The embedding model to use (e.g., 'openai/text-embedding-3-small')
             embedding_dimensions: Number of dimensions for embeddings
             embedding_api_key: API key for the embedding service
             embedding_base_url: Base URL for the embedding service
-            
+
             # Rerank parameters (for vector collections):
             rerank_model: The rerank model to use (e.g., 'cohere/rerank-english-v3.0')
             rerank_api_key: API key for the rerank service
             rerank_base_url: Base URL for the rerank service
-            
+
         Returns:
             A TantivyCollectionIndex or QdrantCollectionIndex instance
         """
@@ -181,7 +192,7 @@ class Collection:
                 settings=settings,
                 query_settings=query_settings,
             )
-        
+
 
 @overload
 def create_collection(
@@ -197,6 +208,7 @@ def create_collection(
     query_settings: Optional["TantivyCollectionIndexQuerySettings"] = None,
 ) -> "TantivyCollectionIndex": ...
 
+
 @overload
 def create_collection(
     name: str = "default",
@@ -213,6 +225,7 @@ def create_collection(
     embedding_function: Optional[Callable[[Any], List[float]]] = None,
 ) -> "QdrantCollectionIndex": ...
 
+
 def create_collection(
     name: str = "default",
     *,
@@ -224,8 +237,14 @@ def create_collection(
     # Tantivy-specific parameters
     fast: bool = True,
     # Unified settings parameters
-    settings: Optional[Union["TantivyCollectionIndexSettings", "QdrantCollectionIndexSettings"]] = None,
-    query_settings: Optional[Union["TantivyCollectionIndexQuerySettings", "QdrantCollectionIndexQuerySettings"]] = None,
+    settings: Optional[
+        Union["TantivyCollectionIndexSettings", "QdrantCollectionIndexSettings"]
+    ] = None,
+    query_settings: Optional[
+        Union[
+            "TantivyCollectionIndexQuerySettings", "QdrantCollectionIndexQuerySettings"
+        ]
+    ] = None,
     # Vector/Qdrant-specific parameters
     distance_metric: "DistanceMetric" = "dot",
     embedding_function: Optional[Callable[[Any], List[float]]] = None,
@@ -233,7 +252,7 @@ def create_collection(
     """
     Create a data collection of the specified type. Collections are a unified
     interface for creating searchable, vectorizable data stores.
-    
+
     Args:
         name: Name of the collection
         schema: Optional schema type for validation
@@ -241,21 +260,21 @@ def create_collection(
         path: File path for storage (None = in-memory)
         vector: Whether this is a vector collection (True) or text search collection (False)
         vector_size: Size of vectors (required for vector collections)
-        
+
         # Tantivy parameters (for non-vector collections):
         fast: Whether to use fast schema building & indexing
-        
+
         # Unified parameters:
         settings: Collection settings (TantivyCollectionIndexSettings or QdrantCollectionIndexSettings)
         query_settings: Query behavior settings (TantivyCollectionIndexQuerySettings or QdrantCollectionIndexQuerySettings)
-        
+
         # Qdrant parameters (for vector collections):
         distance_metric: Distance metric for similarity search
         embedding_model: The embedding model to use (e.g., 'openai/text-embedding-3-small')
         embedding_dimensions: Number of dimensions for embeddings
         embedding_api_key: API key for the embedding service
         embedding_base_url: Base URL for the embedding service
-        
+
     Returns:
         A TantivyCollectionIndex or QdrantCollectionIndex instance
     """
@@ -271,4 +290,4 @@ def create_collection(
         query_settings=query_settings,
         distance_metric=distance_metric,
         embedding_function=embedding_function,
-    )
+    )

hammad/data/collections/indexes/__init__.py

@@ -34,4 +34,4 @@ __getattr__ = create_getattr_importer(__all__)
 
 def __dir__() -> list[str]:
     """Get the attributes of the hammad.data.collections.indexes module."""
-    return list(__all__)
+    return list(__all__)

hammad/data/collections/indexes/qdrant/__init__.py

@@ -1 +1 @@
-"""hammad.lib.data.collections.indexes.qdrant"""
+"""hammad.lib.data.collections.indexes.qdrant"""