beaver-db 0.5.2__py3-none-any.whl → 0.6.0__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/core.py

@@ -3,6 +3,7 @@ import sqlite3
 import time
 from typing import Any
 
+from .dicts import DictWrapper
 from .lists import ListWrapper
 from .subscribers import SubWrapper
 from .collections import CollectionWrapper
@@ -30,22 +31,24 @@ class BeaverDB:
 
     def _create_all_tables(self):
         """Initializes all required tables in the database file."""
-        self._create_kv_table()
         self._create_pubsub_table()
         self._create_list_table()
         self._create_collections_table()
         self._create_fts_table()
         self._create_edges_table()
         self._create_versions_table()
+        self._create_dict_table()
 
-    def _create_kv_table(self):
-        """Creates the key-value store table."""
+    def _create_dict_table(self):
+        """Creates the namespaced dictionary table."""
         with self._conn:
             self._conn.execute(
                 """
-                CREATE TABLE IF NOT EXISTS _beaver_kv_store (
-                    key TEXT PRIMARY KEY,
-                    value TEXT NOT NULL
+                CREATE TABLE IF NOT EXISTS beaver_dicts (
+                    dict_name TEXT NOT NULL,
+                    key TEXT NOT NULL,
+                    value TEXT NOT NULL,
+                    PRIMARY KEY (dict_name, key)
                 )
             """
             )
@@ -148,39 +151,11 @@ class BeaverDB:
 
     # --- Factory and Passthrough Methods ---
 
-    def set(self, key: str, value: Any):
-        """
-        Stores a JSON-serializable value for a given key.
-        This operation is synchronous.
-        """
-        if not isinstance(key, str):
-            raise TypeError("Key must be a string.")
-
-        try:
-            json_value = json.dumps(value)
-        except TypeError as e:
-            raise TypeError("Value must be JSON-serializable.") from e
-
-        with self._conn:
-            self._conn.execute(
-                "INSERT OR REPLACE INTO _beaver_kv_store (key, value) VALUES (?, ?)",
-                (key, json_value),
-            )
-
-    def get(self, key: str) -> Any:
-        """
-        Retrieves a value for a given key.
-        This operation is synchronous.
-        """
-        if not isinstance(key, str):
-            raise TypeError("Key must be a string.")
-
-        cursor = self._conn.cursor()
-        cursor.execute("SELECT value FROM _beaver_kv_store WHERE key = ?", (key,))
-        result = cursor.fetchone()
-        cursor.close()
-
-        return json.loads(result["value"]) if result else None
+    def dict(self, name: str) -> DictWrapper:
+        """Returns a wrapper object for interacting with a named dictionary."""
+        if not isinstance(name, str) or not name:
+            raise TypeError("Dictionary name must be a non-empty string.")
+        return DictWrapper(name, self._conn)
 
     def list(self, name: str) -> ListWrapper:
         """Returns a wrapper object for interacting with a named list."""

beaver/dicts.py

@@ -0,0 +1,101 @@
+import json
+import sqlite3
+from typing import Any, Iterator, Tuple
+
+
+class DictWrapper:
+    """A wrapper providing a Pythonic interface to a dictionary in the database."""
+
+    def __init__(self, name: str, conn: sqlite3.Connection):
+        self._name = name
+        self._conn = conn
+
+    def set(self, key: str, value: Any):
+        """Sets a value for a key in the dictionary."""
+        self.__setitem__(key, value)
+
+    def __setitem__(self, key: str, value: Any):
+        """Sets a value for a key (e.g., `my_dict[key] = value`)."""
+        with self._conn:
+            self._conn.execute(
+                "INSERT OR REPLACE INTO beaver_dicts (dict_name, key, value) VALUES (?, ?, ?)",
+                (self._name, key, json.dumps(value)),
+            )
+
+    def get(self, key: str, default: Any = None) -> Any:
+        """Gets a value for a key, with a default if it doesn't exist."""
+        try:
+            return self[key]
+        except KeyError:
+            return default
+
+    def __getitem__(self, key: str) -> Any:
+        """Retrieves a value for a given key (e.g., `my_dict[key]`)."""
+        cursor = self._conn.cursor()
+        cursor.execute(
+            "SELECT value FROM beaver_dicts WHERE dict_name = ? AND key = ?",
+            (self._name, key),
+        )
+        result = cursor.fetchone()
+        cursor.close()
+        if result is None:
+            raise KeyError(f"Key '{key}' not found in dictionary '{self._name}'")
+        return json.loads(result["value"])
+
+    def __delitem__(self, key: str):
+        """Deletes a key-value pair (e.g., `del my_dict[key]`)."""
+        with self._conn:
+            cursor = self._conn.cursor()
+            cursor.execute(
+                "DELETE FROM beaver_dicts WHERE dict_name = ? AND key = ?",
+                (self._name, key),
+            )
+            if cursor.rowcount == 0:
+                raise KeyError(f"Key '{key}' not found in dictionary '{self._name}'")
+
+    def __len__(self) -> int:
+        """Returns the number of items in the dictionary."""
+        cursor = self._conn.cursor()
+        cursor.execute(
+            "SELECT COUNT(*) FROM beaver_dicts WHERE dict_name = ?", (self._name,)
+        )
+        count = cursor.fetchone()[0]
+        cursor.close()
+        return count
+
+    def __iter__(self) -> Iterator[str]:
+        """Returns an iterator over the keys of the dictionary."""
+        return self.keys()
+
+    def keys(self) -> Iterator[str]:
+        """Returns an iterator over the dictionary's keys."""
+        cursor = self._conn.cursor()
+        cursor.execute(
+            "SELECT key FROM beaver_dicts WHERE dict_name = ?", (self._name,)
+        )
+        for row in cursor:
+            yield row["key"]
+        cursor.close()
+
+    def values(self) -> Iterator[Any]:
+        """Returns an iterator over the dictionary's values."""
+        cursor = self._conn.cursor()
+        cursor.execute(
+            "SELECT value FROM beaver_dicts WHERE dict_name = ?", (self._name,)
+        )
+        for row in cursor:
+            yield json.loads(row["value"])
+        cursor.close()
+
+    def items(self) -> Iterator[Tuple[str, Any]]:
+        """Returns an iterator over the dictionary's items (key-value pairs)."""
+        cursor = self._conn.cursor()
+        cursor.execute(
+            "SELECT key, value FROM beaver_dicts WHERE dict_name = ?", (self._name,)
+        )
+        for row in cursor:
+            yield (row["key"], json.loads(row["value"]))
+        cursor.close()
+
+    def __repr__(self) -> str:
+        return f"DictWrapper(name='{self._name}')"

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beaver-db
-Version: 0.5.2
+Version: 0.6.0
 Summary: Fast, embedded, and multi-modal DB based on SQLite for AI-powered applications.
 Requires-Python: >=3.13
 Description-Content-Type: text/markdown
@@ -26,7 +26,7 @@ A fast, single-file, multi-modal database for Python, built with the standard `s
 ## Core Features
 
   - **Synchronous Pub/Sub**: A simple, thread-safe, Redis-like publish-subscribe system for real-time messaging.
-  - **Persistent Key-Value Store**: A simple `set`/`get` interface for storing any JSON-serializable object.
+  - **Namespaced Key-Value Dictionaries**: A Pythonic, dictionary-like interface for storing any JSON-serializable object within separate namespaces.
   - **Pythonic List Management**: A fluent, Redis-like interface for managing persistent, ordered lists.
   - **Efficient Vector Storage & Search**: Store vector embeddings and perform fast approximate nearest neighbor searches using an in-memory k-d tree.
   - **Full-Text Search**: Automatically index and search through document metadata using SQLite's powerful FTS5 engine.
@@ -51,17 +51,21 @@ from beaver import BeaverDB, Document
 db = BeaverDB("my_application.db")
 ```
 
-### Key-Value Store
+### Namespaced Dictionaries
 
-Use `set()` and `get()` for simple data storage. The value can be any JSON-encodable object.
+Use `db.dict()` to get a dictionary-like object for a specific namespace. The value can be any JSON-encodable object.
 
 ```python
-# Set a value
-db.set("app_config", {"theme": "dark", "user_id": 123})
+# Get a handle to the 'app_config' namespace
+config = db.dict("app_config")
+
+# Set values using standard dictionary syntax
+config["theme"] = "dark"
+config["user_id"] = 123
 
 # Get a value
-config = db.get("app_config")
-print(f"Theme: {config.get('theme')}") # Output: Theme: dark
+theme = config.get("theme")
+print(f"Theme: {theme}") # Output: Theme: dark
 ```
 
 ### List Management
@@ -94,14 +98,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.6.0.dist-info/RECORD

@@ -0,0 +1,10 @@
+beaver/__init__.py,sha256=-z5Gj6YKMOswpJOOn5Gej8z5i6k3c0Xs00DIYLA-bMI,75
+beaver/collections.py,sha256=II26aeXelbtfs0zkcHImvPDtoyDhPCCP8bsosEqoKvw,15064
+beaver/core.py,sha256=CDAnIF1InFLb_JxMiO_bWo8coXIemhB68_WBkSTTKzU,6624
+beaver/dicts.py,sha256=XgkflJegQAY0YnkxnwWw_LGAeRtGFHzGtuaLAK2GfZg,3558
+beaver/lists.py,sha256=JG1JOkaYCUldADUzPJhaNi93w-k3S8mUzcCw574uht4,5915
+beaver/subscribers.py,sha256=tCty2iDbeE9IXcPicbxj2CB5gqfLufMB9-nLQwqNBUU,1944
+beaver_db-0.6.0.dist-info/METADATA,sha256=TwpfIsOef1GXpRdCpVEaQLVWtK1Ql11M9L7QLpSkphk,6265
+beaver_db-0.6.0.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+beaver_db-0.6.0.dist-info/top_level.txt,sha256=FxA4XnX5Qm5VudEXCduFriqi4dQmDWpQ64d7g69VQKI,7
+beaver_db-0.6.0.dist-info/RECORD,,

beaver_db-0.5.2.dist-info/RECORD

@@ -1,9 +0,0 @@
-beaver/__init__.py,sha256=-z5Gj6YKMOswpJOOn5Gej8z5i6k3c0Xs00DIYLA-bMI,75
-beaver/collections.py,sha256=4rdGMTD7ex4SQUH52WIHOZhOdeWe7Nqvm9TPg7flv_g,15059
-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,,