beaver-db 0.5.2__py3-none-any.whl → 0.5.3__py3-none-any.whl

beaver/collections.py

@@ -331,7 +331,7 @@ class CollectionWrapper:
 
 
 def rerank(
-    results: list[list[Document]],
+    *results: list[Document],
     weights: list[float] | None = None,
     k: int = 60
 ) -> list[Document]:
@@ -340,7 +340,7 @@ def rerank(
     This function is specifically designed to work with beaver.collections.Document objects.
 
     Args:
-        results (list[list[Document]]): A list of search result lists, where each
+        results (sequence of list[Document]): A sequence of search result lists, where each
             inner list contains Document objects.
         weights (list[float], optional): A list of weights corresponding to each
             result list. If None, all lists are weighted equally. Defaults to None.

{beaver_db-0.5.2.dist-info → beaver_db-0.5.3.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beaver-db
-Version: 0.5.2
+Version: 0.5.3
 Summary: Fast, embedded, and multi-modal DB based on SQLite for AI-powered applications.
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
@@ -94,14 +94,18 @@ docs.index(doc)
 
 # 1. Perform a vector search to find semantically similar documents
 query_vector = [0.7, 0.2, 0.2]
-vector_results = docs.search(vector=query_vector, top_k=1)
+vector_results = docs.search(vector=query_vector, top_k=3)
 top_doc, distance = vector_results[0]
 print(f"Vector Search Result: {top_doc.content} (distance: {distance:.2f})")
 
 # 2. Perform a full-text search to find documents with specific words
-text_results = docs.match(query="database", top_k=1)
+text_results = docs.match(query="database", top_k=3)
 top_doc, rank = text_results[0]
 print(f"Full-Text Search Result: {top_doc.content} (rank: {rank:.2f})")
+
+# 3. Combine both vector and text search for refined results
+from beaver.collections import rerank
+combined_results = rerank([d for d,_ in vector_results], [d for d,_ in text_results], weights=[2,1])
 ```
 
 ### Graph Traversal

{beaver_db-0.5.2.dist-info → beaver_db-0.5.3.dist-info}/RECORD

@@ -1,9 +1,9 @@
 beaver/__init__.py,sha256=-z5Gj6YKMOswpJOOn5Gej8z5i6k3c0Xs00DIYLA-bMI,75
-beaver/collections.py,sha256=4rdGMTD7ex4SQUH52WIHOZhOdeWe7Nqvm9TPg7flv_g,15059
+beaver/collections.py,sha256=II26aeXelbtfs0zkcHImvPDtoyDhPCCP8bsosEqoKvw,15064
 beaver/core.py,sha256=sk0Z_k7EcORe6bN8CfPukGX7eAfmCGSX_B37KpJmQJ4,7279
 beaver/lists.py,sha256=JG1JOkaYCUldADUzPJhaNi93w-k3S8mUzcCw574uht4,5915
 beaver/subscribers.py,sha256=tCty2iDbeE9IXcPicbxj2CB5gqfLufMB9-nLQwqNBUU,1944
-beaver_db-0.5.2.dist-info/METADATA,sha256=ij6szwN6Ee0MutGlFgszF2vqddR5y8OlA9O4R5Y4Nbo,5904
-beaver_db-0.5.2.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-beaver_db-0.5.2.dist-info/top_level.txt,sha256=FxA4XnX5Qm5VudEXCduFriqi4dQmDWpQ64d7g69VQKI,7
-beaver_db-0.5.2.dist-info/RECORD,,
+beaver_db-0.5.3.dist-info/METADATA,sha256=MAtHrjLqdqBKTPzHGJiX4gORwcwQl38pZXr3mSFWsk4,6105
+beaver_db-0.5.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beaver_db-0.5.3.dist-info/top_level.txt,sha256=FxA4XnX5Qm5VudEXCduFriqi4dQmDWpQ64d7g69VQKI,7
+beaver_db-0.5.3.dist-info/RECORD,,