beaver-db 0.19.1__py3-none-any.whl → 0.19.2__py3-none-any.whl

beaver/__init__.py

@@ -2,4 +2,4 @@ from .core import BeaverDB
 from .types import Model
 from .collections import Document, WalkDirection
 
-__version__ = "0.19.1"
+__version__ = "0.19.2"

beaver/blobs.py

@@ -122,5 +122,16 @@ class BlobManager[M]:
             yield row["key"]
         cursor.close()
 
+    def __len__(self) -> int:
+        """Returns the number of blobs in the store."""
+        cursor = self._db.connection.cursor()
+        cursor.execute(
+            "SELECT COUNT(*) FROM beaver_blobs WHERE store_name = ?",
+            (self._name,)
+        )
+        count = cursor.fetchone()[0]
+        cursor.close()
+        return count
+
     def __repr__(self) -> str:
         return f"BlobManager(name='{self._name}')"

beaver/server.py

@@ -2,11 +2,23 @@ try:
     from typing import Any, Optional, List
     import json
     from datetime import datetime, timedelta, timezone
-    from fastapi import FastAPI, HTTPException, Body, UploadFile, File, Form, Response, WebSocket, WebSocketDisconnect
+    from fastapi import (
+        FastAPI,
+        HTTPException,
+        Body,
+        UploadFile,
+        File,
+        Form,
+        Response,
+        WebSocket,
+        WebSocketDisconnect,
+    )
     import uvicorn
     from pydantic import BaseModel, Field
 except ImportError:
-    raise ImportError("Please install server dependencies with: pip install \"beaver-db[server]\"")
+    raise ImportError(
+        'Please install server dependencies with: pip install "beaver-db[server]"'
+    )
 
 from .core import BeaverDB
 from .collections import Document, WalkDirection
@@ -14,6 +26,7 @@ from .collections import Document, WalkDirection
 
 # --- Pydantic Models for Collections ---
 
+
 class IndexRequest(BaseModel):
     id: Optional[str] = None
     embedding: Optional[List[float]] = None
@@ -21,28 +34,36 @@ class IndexRequest(BaseModel):
     fts: bool = True
     fuzzy: bool = False
 
+
 class SearchRequest(BaseModel):
     vector: List[float]
     top_k: int = 10
 
+
 class MatchRequest(BaseModel):
     query: str
     on: Optional[List[str]] = None
     top_k: int = 10
     fuzziness: int = 0
 
+
 class ConnectRequest(BaseModel):
     source_id: str
     target_id: str
     label: str
     metadata: Optional[dict] = None
 
+
 class WalkRequest(BaseModel):
     labels: List[str]
     depth: int
     direction: WalkDirection = WalkDirection.OUTGOING
 
 
+class CountResponse(BaseModel):
+    count: int
+
+
 def build(db: BeaverDB) -> FastAPI:
     """Constructs a FastAPI instance for a given BeaverDB."""
     app = FastAPI(title="BeaverDB Server")
@@ -55,7 +76,9 @@ def build(db: BeaverDB) -> FastAPI:
         d = db.dict(name)
         value = d.get(key)
         if value is None:
-            raise HTTPException(status_code=404, detail=f"Key '{key}' not found in dictionary '{name}'")
+            raise HTTPException(
+                status_code=404, detail=f"Key '{key}' not found in dictionary '{name}'"
+            )
         return value
 
     @app.put("/dicts/{name}/{key}", tags=["Dicts"])
@@ -73,8 +96,15 @@ def build(db: BeaverDB) -> FastAPI:
             del d[key]
             return {"status": "ok"}
         except KeyError:
-            raise HTTPException(status_code=404, detail=f"Key '{key}' not found in dictionary '{name}'")
+            raise HTTPException(
+                status_code=404, detail=f"Key '{key}' not found in dictionary '{name}'"
+            )
 
+    @app.get("/dicts/{name}/count", tags=["Dicts"], response_model=CountResponse)
+    def get_dict_count(name: str) -> dict:
+        """Retrieves the number of key-value pairs in the dictionary."""
+        d = db.dict(name)
+        return {"count": len(d)}
 
     # --- Lists Endpoints ---
 
@@ -91,7 +121,9 @@ def build(db: BeaverDB) -> FastAPI:
         try:
             return l[index]
         except IndexError:
-            raise HTTPException(status_code=404, detail=f"Index {index} out of bounds for list '{name}'")
+            raise HTTPException(
+                status_code=404, detail=f"Index {index} out of bounds for list '{name}'"
+            )
 
     @app.post("/lists/{name}", tags=["Lists"])
     def push_list_item(name: str, value: Any = Body(...)):
@@ -108,7 +140,9 @@ def build(db: BeaverDB) -> FastAPI:
             l[index] = value
             return {"status": "ok"}
         except IndexError:
-            raise HTTPException(status_code=404, detail=f"Index {index} out of bounds for list '{name}'")
+            raise HTTPException(
+                status_code=404, detail=f"Index {index} out of bounds for list '{name}'"
+            )
 
     @app.delete("/lists/{name}/{index}", tags=["Lists"])
     def delete_list_item(name: str, index: int):
@@ -118,7 +152,15 @@ def build(db: BeaverDB) -> FastAPI:
             del l[index]
             return {"status": "ok"}
         except IndexError:
-            raise HTTPException(status_code=404, detail=f"Index {index} out of bounds for list '{name}'")
+            raise HTTPException(
+                status_code=404, detail=f"Index {index} out of bounds for list '{name}'"
+            )
+
+    @app.get("/lists/{name}/count", tags=["Lists"], response_model=CountResponse)
+    def get_list_count(name: str) -> dict:
+        """Retrieves the number of items in the list."""
+        l = db.list(name)
+        return {"count": len(l)}
 
     # --- Queues Endpoints ---
 
@@ -149,11 +191,19 @@ def build(db: BeaverDB) -> FastAPI:
             item = q.get(block=True, timeout=timeout)
             return item
         except TimeoutError:
-            raise HTTPException(status_code=408, detail=f"Request timed out after {timeout}s waiting for an item in queue '{name}'")
+            raise HTTPException(
+                status_code=408,
+                detail=f"Request timed out after {timeout}s waiting for an item in queue '{name}'",
+            )
         except IndexError:
             # This case is less likely with block=True but good to handle
             raise HTTPException(status_code=404, detail=f"Queue '{name}' is empty")
 
+    @app.get("/queues/{name}/count", tags=["Queues"], response_model=CountResponse)
+    def get_queue_count(name: str) -> dict:
+        """RetrieVIes the number of items currently in the queue."""
+        q = db.queue(name)
+        return {"count": len(q)}
 
     # --- Blobs Endpoints ---
 
@@ -163,12 +213,20 @@ def build(db: BeaverDB) -> FastAPI:
         blobs = db.blobs(name)
         blob = blobs.get(key)
         if blob is None:
-            raise HTTPException(status_code=404, detail=f"Blob with key '{key}' not found in store '{name}'")
+            raise HTTPException(
+                status_code=404,
+                detail=f"Blob with key '{key}' not found in store '{name}'",
+            )
         # Return the raw bytes with a generic binary content type
         return Response(content=blob.data, media_type="application/octet-stream")
 
     @app.put("/blobs/{name}/{key}", tags=["Blobs"])
-    async def put_blob(name: str, key: str, data: UploadFile = File(...), metadata: Optional[str] = Form(None)):
+    async def put_blob(
+        name: str,
+        key: str,
+        data: UploadFile = File(...),
+        metadata: Optional[str] = Form(None),
+    ):
         """Stores a blob (binary file) with optional JSON metadata."""
         blobs = db.blobs(name)
         file_bytes = await data.read()
@@ -178,7 +236,9 @@ def build(db: BeaverDB) -> FastAPI:
             try:
                 meta_dict = json.loads(metadata)
             except json.JSONDecodeError:
-                raise HTTPException(status_code=400, detail="Invalid JSON format for metadata.")
+                raise HTTPException(
+                    status_code=400, detail="Invalid JSON format for metadata."
+                )
 
         blobs.put(key=key, data=file_bytes, metadata=meta_dict)
         return {"status": "ok"}
@@ -191,8 +251,16 @@ def build(db: BeaverDB) -> FastAPI:
             blobs.delete(key)
             return {"status": "ok"}
         except KeyError:
-            raise HTTPException(status_code=404, detail=f"Blob with key '{key}' not found in store '{name}'")
+            raise HTTPException(
+                status_code=404,
+                detail=f"Blob with key '{key}' not found in store '{name}'",
+            )
 
+    @app.get("/blobs/{name}/count", tags=["Blobs"], response_model=CountResponse)
+    def get_blob_count(name: str) -> dict:
+        """Retrieves the number of blobs in the store."""
+        b = db.blobs(name)
+        return {"count": len(b)}
 
     # --- Logs Endpoints ---
 
@@ -208,12 +276,25 @@ def build(db: BeaverDB) -> FastAPI:
         """Retrieves log entries within a specific time window."""
         log = db.log(name)
         # Ensure datetimes are timezone-aware (UTC) for correct comparison
-        start_utc = start.astimezone(timezone.utc) if start.tzinfo else start.replace(tzinfo=timezone.utc)
-        end_utc = end.astimezone(timezone.utc) if end.tzinfo else end.replace(tzinfo=timezone.utc)
+        start_utc = (
+            start.astimezone(timezone.utc)
+            if start.tzinfo
+            else start.replace(tzinfo=timezone.utc)
+        )
+        end_utc = (
+            end.astimezone(timezone.utc)
+            if end.tzinfo
+            else end.replace(tzinfo=timezone.utc)
+        )
         return log.range(start=start_utc, end=end_utc)
 
     @app.websocket("/logs/{name}/live", name="Logs")
-    async def live_log_feed(websocket: WebSocket, name: str, window_seconds: int = 5, period_seconds: int = 1):
+    async def live_log_feed(
+        websocket: WebSocket,
+        name: str,
+        window_seconds: int = 5,
+        period_seconds: int = 1,
+    ):
         """Streams live, aggregated log data over a WebSocket."""
         await websocket.accept()
 
@@ -222,7 +303,10 @@ def build(db: BeaverDB) -> FastAPI:
         # This simple aggregator function runs in the background and returns a
         # JSON-serializable summary of the data in the current window.
         def simple_aggregator(window):
-            return {"count": len(window), "latest_timestamp": window[-1]["timestamp"] if window else None}
+            return {
+                "count": len(window),
+                "latest_timestamp": window[-1]["timestamp"] if window else None,
+            }
 
         live_stream = async_logs.live(
             window=timedelta(seconds=window_seconds),
@@ -239,7 +323,6 @@ def build(db: BeaverDB) -> FastAPI:
             # Cleanly close the underlying iterator and its background thread.
             live_stream.close()
 
-
     # --- Channels Endpoints ---
 
     @app.post("/channels/{name}/publish", tags=["Channels"])
@@ -263,7 +346,6 @@ def build(db: BeaverDB) -> FastAPI:
         except WebSocketDisconnect:
             print(f"Client disconnected from channel '{name}' subscription.")
 
-
     # --- Collections Endpoints ---
 
     @app.get("/collections/{name}", tags=["Collections"])
@@ -282,7 +364,10 @@ def build(db: BeaverDB) -> FastAPI:
             return {"status": "ok", "id": doc.id}
         except TypeError as e:
             if "vector" in str(e):
-                raise HTTPException(status_code=501, detail="Vector indexing requires the '[vector]' extra. Install with: pip install \"beaver-db[vector]\"")
+                raise HTTPException(
+                    status_code=501,
+                    detail="Vector indexing requires the '[vector]' extra. Install with: pip install \"beaver-db[vector]\"",
+                )
             raise e
 
     @app.post("/collections/{name}/search", tags=["Collections"])
@@ -291,18 +376,29 @@ def build(db: BeaverDB) -> FastAPI:
         collection = db.collection(name)
         try:
             results = collection.search(vector=req.vector, top_k=req.top_k)
-            return [{"document": doc.to_dict(metadata_only=False), "distance": dist} for doc, dist in results]
+            return [
+                {"document": doc.to_dict(metadata_only=False), "distance": dist}
+                for doc, dist in results
+            ]
         except TypeError as e:
             if "vector" in str(e):
-                raise HTTPException(status_code=501, detail="Vector search requires the '[vector]' extra. Install with: pip install \"beaver-db[vector]\"")
+                raise HTTPException(
+                    status_code=501,
+                    detail="Vector search requires the '[vector]' extra. Install with: pip install \"beaver-db[vector]\"",
+                )
             raise e
 
     @app.post("/collections/{name}/match", tags=["Collections"])
     def match_collection(name: str, req: MatchRequest) -> List[dict]:
         """Performs a full-text or fuzzy search on the collection."""
         collection = db.collection(name)
-        results = collection.match(query=req.query, on=req.on, top_k=req.top_k, fuzziness=req.fuzziness)
-        return [{"document": doc.to_dict(metadata_only=False), "score": score} for doc, score in results]
+        results = collection.match(
+            query=req.query, on=req.on, top_k=req.top_k, fuzziness=req.fuzziness
+        )
+        return [
+            {"document": doc.to_dict(metadata_only=False), "score": score}
+            for doc, score in results
+        ]
 
     @app.post("/collections/{name}/connect", tags=["Collections"])
     def connect_documents(name: str, req: ConnectRequest):
@@ -310,11 +406,15 @@ def build(db: BeaverDB) -> FastAPI:
         collection = db.collection(name)
         source_doc = Document(id=req.source_id)
         target_doc = Document(id=req.target_id)
-        collection.connect(source=source_doc, target=target_doc, label=req.label, metadata=req.metadata)
+        collection.connect(
+            source=source_doc, target=target_doc, label=req.label, metadata=req.metadata
+        )
         return {"status": "ok"}
 
     @app.get("/collections/{name}/{doc_id}/neighbors", tags=["Collections"])
-    def get_neighbors(name: str, doc_id: str, label: Optional[str] = None) -> List[dict]:
+    def get_neighbors(
+        name: str, doc_id: str, label: Optional[str] = None
+    ) -> List[dict]:
         """Retrieves the neighboring documents for a given document."""
         collection = db.collection(name)
         doc = Document(id=doc_id)
@@ -326,9 +426,22 @@ def build(db: BeaverDB) -> FastAPI:
         """Performs a graph traversal (BFS) from a starting document."""
         collection = db.collection(name)
         source_doc = Document(id=doc_id)
-        results = collection.walk(source=source_doc, labels=req.labels, depth=req.depth, direction=req.direction)
+        results = collection.walk(
+            source=source_doc,
+            labels=req.labels,
+            depth=req.depth,
+            direction=req.direction,
+        )
         return [doc.to_dict(metadata_only=False) for doc in results]
 
+    @app.get(
+        "/collections/{name}/count", tags=["Collections"], response_model=CountResponse
+    )
+    def get_collection_count(name: str) -> dict:
+        """RetrieRetrieves the number of documents in the collection."""
+        c = db.collection(name)
+        return {"count": len(c)}
+
     return app
 
 

{beaver_db-0.19.1.dist-info → beaver_db-0.19.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beaver-db
-Version: 0.19.1
+Version: 0.19.2
 Summary: Fast, embedded, and multi-modal DB based on SQLite for AI-powered applications.
 License-File: LICENSE
 Classifier: License :: OSI Approved :: MIT License
@@ -23,7 +23,11 @@ Provides-Extra: vector
 Requires-Dist: faiss-cpu>=1.12.0; extra == 'vector'
 Description-Content-Type: text/markdown
 
-# beaver 🦫
+<div style="text-align: center;">
+  <img src="https://github.com/syalia-srl/beaver/blob/main/logo.png?raw=true" width="256px">
+</div>
+
+---
 
 <!-- Project badges -->
 ![PyPI - Version](https://img.shields.io/pypi/v/beaver-db)
@@ -32,7 +36,9 @@ Description-Content-Type: text/markdown
 ![PyPi - Downloads (Monthly)](https://img.shields.io/pypi/dm/beaver-db)
 ![Github - Commits](https://img.shields.io/github/commit-activity/m/apiad/beaver)
 
-A fast, single-file, multi-modal database for Python, built with the standard `sqlite3` library.
+> A fast, single-file, multi-modal database for Python, built with the standard `sqlite3` library.
+
+---
 
 `beaver` is the **B**ackend for **E**mbedded, **A**ll-in-one **V**ector, **E**ntity, and **R**elationship storage. It's a simple, local, and embedded database designed to manage complex, modern data types without requiring a database server, built on top of SQLite.
 

{beaver_db-0.19.1.dist-info → beaver_db-0.19.2.dist-info}/RECORD

@@ -1,5 +1,5 @@
-beaver/__init__.py,sha256=dMWanhrt58thAIt9Qj4jzxy5Dg5KzvNQenZ_cNcz_Tk,125
-beaver/blobs.py,sha256=YkIEskHD6oHRaJTF0P25HrTT8LqM-REyV_UBPVQxeqQ,4055
+beaver/__init__.py,sha256=HjTNjk3x58Pw9Iv_eAPVL088wDVEua-JEgecZ28NgMc,125
+beaver/blobs.py,sha256=U5n6NLRQGAzsePGR2SJPRXHy22K8T9cJVMIb0JGzsY0,4399
 beaver/channels.py,sha256=kIuwKMDBdDQObaKT23znsMXzfpKfE7pXSxvf-u4LlpY,9554
 beaver/cli.py,sha256=Sxm-mYU3LGd4tIqw-5LHb0ektWebjV9vn51hm-CMJD0,2232
 beaver/collections.py,sha256=UAQAuRxJRCqY5PHfxJNm3CdKqMNuyY8DOLdodvY6jpk,26107
@@ -9,11 +9,11 @@ beaver/lists.py,sha256=rfJ8uTNLkMREYc0uGx0z1VKt2m3eR9hvbdvDD58EbmQ,10140
 beaver/locks.py,sha256=GWDSRkPw2lrAQfXIRqvkc5PK9zZ2eLYWKTuzHTs9j_A,6321
 beaver/logs.py,sha256=a5xenwl5NZeegIU0dWVEs67lvaHzzw-JRAZtEzNNO3E,9529
 beaver/queues.py,sha256=Fr3oie63EtceSoiC8EOEDSLu1tDI8q2MYLXd8MEeC3g,6476
-beaver/server.py,sha256=WoNcPXU9oh6hcHtb60IbEk5DfZT5J4Fb-yubJE3YLIc,13642
+beaver/server.py,sha256=At3BoEV7JfpYjNtyHMdPUF8shj4V4D5nStXWb6Bv53A,15947
 beaver/types.py,sha256=m0ohT7A8r0Y1a7bJEx4VanLaOUWU2VYxaLHPsVPjrIw,1651
 beaver/vectors.py,sha256=EGZf1s364-rMubxkYoTcjBl72lRRxM1cUwypjsoC6ec,18499
-beaver_db-0.19.1.dist-info/METADATA,sha256=Nt_3RlU4UraPp-zJusitwvwE2yza_zUyhRvecRcEvrI,23299
-beaver_db-0.19.1.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beaver_db-0.19.1.dist-info/entry_points.txt,sha256=bd5E2s45PoBdtdR9-ToKSdLNhmHp8naV1lWP5mOzlrc,42
-beaver_db-0.19.1.dist-info/licenses/LICENSE,sha256=1xrIY5JnMk_QDQzsqmVzPIIyCgZAkWCC8kF2Ddo1UT0,1071
-beaver_db-0.19.1.dist-info/RECORD,,
+beaver_db-0.19.2.dist-info/METADATA,sha256=Iec3mTpq384nkp_R8fD2AGjXHRzarA93uxL623ZtyVE,23431
+beaver_db-0.19.2.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beaver_db-0.19.2.dist-info/entry_points.txt,sha256=bd5E2s45PoBdtdR9-ToKSdLNhmHp8naV1lWP5mOzlrc,42
+beaver_db-0.19.2.dist-info/licenses/LICENSE,sha256=1xrIY5JnMk_QDQzsqmVzPIIyCgZAkWCC8kF2Ddo1UT0,1071
+beaver_db-0.19.2.dist-info/RECORD,,