hammad-python 0.0.26__py3-none-any.whl → 0.0.27__py3-none-any.whl

hammad/data/collections/collection.py

@@ -222,7 +222,16 @@ def create_collection(
     distance_metric: "DistanceMetric" = "dot",
     settings: Optional["QdrantCollectionIndexSettings"] = None,
     query_settings: Optional["QdrantCollectionIndexQuerySettings"] = None,
-    embedding_function: Optional[Callable[[Any], List[float]]] = None,
+    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,
+    # Rerank-specific parameters
+    rerank_model: Optional[str] = None,
+    rerank_api_key: Optional[str] = None,
+    rerank_base_url: Optional[str] = None,
 ) -> "QdrantCollectionIndex": ...
 
 
@@ -247,7 +256,16 @@ def create_collection(
     ] = None,
     # Vector/Qdrant-specific parameters
     distance_metric: "DistanceMetric" = "dot",
-    embedding_function: Optional[Callable[[Any], List[float]]] = None,
+    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,
+    # Rerank-specific parameters
+    rerank_model: Optional[str] = None,
+    rerank_api_key: Optional[str] = None,
+    rerank_base_url: Optional[str] = None,
 ) -> Union["TantivyCollectionIndex", "QdrantCollectionIndex"]:
     """
     Create a data collection of the specified type. Collections are a unified
@@ -275,6 +293,11 @@ def create_collection(
         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
     """
@@ -289,5 +312,11 @@ def create_collection(
         settings=settings,
         query_settings=query_settings,
         distance_metric=distance_metric,
-        embedding_function=embedding_function,
+        embedding_model=embedding_model,
+        embedding_dimensions=embedding_dimensions,
+        embedding_api_key=embedding_api_key,
+        embedding_base_url=embedding_base_url,
+        rerank_model=rerank_model,
+        rerank_api_key=rerank_api_key,
+        rerank_base_url=rerank_base_url,
     )

{hammad_python-0.0.26.dist-info → hammad_python-0.0.27.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: hammad-python
-Version: 0.0.26
+Version: 0.0.27
 Author-email: Hammad Saeed <hammadaidev@gmail.com>
 License: MIT License
         

{hammad_python-0.0.26.dist-info → hammad_python-0.0.27.dist-info}/RECORD

@@ -18,7 +18,7 @@ hammad/cli/styles/types.py,sha256=vNIeQY_23m10K8qVT7Iy-PMwosGL-La-UAZKszHJjEE,79
 hammad/cli/styles/utils.py,sha256=zzi0JdH1X7O8XWRlMVfJP2jB-OWt7zkpm_LeCHoSKVY,28287
 hammad/data/__init__.py,sha256=rga828Od--N3QeGlnOSweADD1WsRLjSTqkaZVk1VhW0,2232
 hammad/data/collections/__init__.py,sha256=xEORHnjoV75Fa6LFDMyFw90oDaJ0e9VmISLFV3mOsIQ,1110
-hammad/data/collections/collection.py,sha256=fi7jyT2GmXiGLNajjegBJMbefzggL0PIMHf-81ov7Bo,10833
+hammad/data/collections/collection.py,sha256=SvmZY0B5_VDMQGX72TxLs9BMeDs8kASmROQca_F3534,11989
 hammad/data/collections/indexes/__init__.py,sha256=RmXKWKq2sbtA1swz5vamKKWut-eKfc-Q2tUnij-E-IU,960
 hammad/data/collections/indexes/qdrant/__init__.py,sha256=KU89TIJkYmJPnVxWKHfXntkIYwhn86ejXtWG30hCyHg,49
 hammad/data/collections/indexes/qdrant/index.py,sha256=UzhW64i5Yq5AXyGZjkW4dljgQxwKUbjF6vhzendbjko,25634
@@ -129,7 +129,7 @@ hammad/web/openapi/__init__.py,sha256=JhJQ6_laBmB2djIYFc0vgGha2GsdUe4FP1LDdZCQ5J
 hammad/web/openapi/client.py,sha256=1pXz7KAO_0pN4kQZoWKWskXDYGiJ535TsPO1GGCiC0E,26816
 hammad/web/search/__init__.py,sha256=e9A6znPIiZCz-4secyHbUs0uUGf5yAqW6wGacgx961U,24
 hammad/web/search/client.py,sha256=LIx2MsHhn6cRTuq5i1mWowRTdIhPobY4GQV3S3bk9lk,36694
-hammad_python-0.0.26.dist-info/METADATA,sha256=u04KMVnt5FBw6dPi0fP19OdhLXU2UlT7CU8kaXtP-RY,6785
-hammad_python-0.0.26.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-hammad_python-0.0.26.dist-info/licenses/LICENSE,sha256=h74yFUWjbBaodcWG5wNmm30npjl8obVcxD-1nQfUp2I,1069
-hammad_python-0.0.26.dist-info/RECORD,,
+hammad_python-0.0.27.dist-info/METADATA,sha256=WfL-hAEfVrPhAwRn3cBaEdm0EfaVIv3RYfz54OkLv08,6785
+hammad_python-0.0.27.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+hammad_python-0.0.27.dist-info/licenses/LICENSE,sha256=h74yFUWjbBaodcWG5wNmm30npjl8obVcxD-1nQfUp2I,1069
+hammad_python-0.0.27.dist-info/RECORD,,