beaver-db 0.16.8__py3-none-any.whl → 0.17.0__py3-none-any.whl

beaver/cli.py

@@ -0,0 +1,34 @@
+import typer
+import rich
+
+app = typer.Typer()
+
+
+@app.command()
+def serve(
+    database: str = typer.Option(
+        "beaver.db", "--database", "-d", help="Path to the database file."
+    ),
+    host: str = typer.Option(
+        "127.0.0.1", "--host", "-h", help="The host to bind the server to."
+    ),
+    port: int = typer.Option(
+        8000, "--port", "-p", help="The port to run the server on."
+    ),
+):
+    """
+    Starts a RESTful API server for a BeaverDB instance.
+    """
+    try:
+        from .server import serve as run_server
+    except ImportError as e:
+        rich.print(f"[red]Error: {e}[/]")
+        rich.print('[yellow]Please install the server dependencies with `pip install "beaver-db[server]"`[/]')
+        raise typer.Exit(code=1)
+
+    rich.print(f"[blue]Starting server for database at '{database}' on http://{host}:{port}[/]")
+    run_server(db_path=database, host=host, port=port)
+
+
+if __name__ == "__main__":
+    app()

beaver/server.py

@@ -0,0 +1,132 @@
+try:
+    from fastapi import FastAPI, HTTPException, Body
+    import uvicorn
+except ImportError:
+    raise ImportError(
+        "FastAPI and Uvicorn are required to serve the database. "
+        'Please install them with `pip install "beaver-db[server]"`'
+    )
+from typing import Any
+from .core import BeaverDB
+
+
+def build(db: BeaverDB) -> FastAPI:
+    """
+    Constructs a FastAPI application instance for a given BeaverDB instance.
+
+    Args:
+        db: An active BeaverDB instance.
+
+    Returns:
+        A FastAPI application with all endpoints configured.
+    """
+    app = FastAPI(
+        title="BeaverDB",
+        description="A RESTful API for a BeaverDB instance.",
+        version="0.1.0",
+    )
+
+    # --- Dicts Endpoints ---
+
+    @app.get("/dicts/{name}")
+    def get_all_dict_items(name: str) -> dict:
+        """Retrieves all key-value pairs in a dictionary."""
+        d = db.dict(name)
+        return {k: v for k, v in d.items()}
+
+    @app.get("/dicts/{name}/{key}")
+    def get_dict_item(name: str, key: str) -> Any:
+        """Retrieves the value for a specific key."""
+        d = db.dict(name)
+        try:
+            return d[key]
+        except KeyError:
+            raise HTTPException(status_code=404, detail=f"Key '{key}' not found in dictionary '{name}'")
+
+    @app.post("/dicts/{name}/{key}")
+    def set_dict_item(name: str, key: str, value: Any = Body(...)):
+        """Sets the value for a specific key."""
+        d = db.dict(name)
+        d[key] = value
+        return {"status": "ok"}
+
+    @app.delete("/dicts/{name}/{key}")
+    def delete_dict_item(name: str, key: str):
+        """Deletes a key-value pair."""
+        d = db.dict(name)
+        try:
+            del d[key]
+            return {"status": "ok"}
+        except KeyError:
+            raise HTTPException(status_code=404, detail=f"Key '{key}' not found in dictionary '{name}'")
+
+    # --- Lists Endpoints ---
+
+    @app.get("/lists/{name}")
+    def get_list(name: str) -> list:
+        """Retrieves all items in the list."""
+        l = db.list(name)
+        return l[:]
+
+    @app.get("/lists/{name}/{index}")
+    def get_list_item(name: str, index: int) -> Any:
+        """Retrieves the item at a specific index."""
+        l = db.list(name)
+        try:
+            return l[index]
+        except IndexError:
+            raise HTTPException(status_code=404, detail=f"Index {index} out of bounds for list '{name}'")
+
+    @app.post("/lists/{name}")
+    def push_list_item(name: str, value: Any = Body(...)):
+        """Adds an item to the end of the list."""
+        l = db.list(name)
+        l.push(value)
+        return {"status": "ok"}
+
+    @app.put("/lists/{name}/{index}")
+    def update_list_item(name: str, index: int, value: Any = Body(...)):
+        """Updates the item at a specific index."""
+        l = db.list(name)
+        try:
+            l[index] = value
+            return {"status": "ok"}
+        except IndexError:
+            raise HTTPException(status_code=404, detail=f"Index {index} out of bounds for list '{name}'")
+
+    @app.delete("/lists/{name}/{index}")
+    def delete_list_item(name: str, index: int):
+        """Deletes the item at a specific index."""
+        l = db.list(name)
+        try:
+            del l[index]
+            return {"status": "ok"}
+        except IndexError:
+            raise HTTPException(status_code=404, detail=f"Index {index} out of bounds for list '{name}'")
+
+    # TODO: Add endpoints for all BeaverDB modalities
+    # - Queues
+    # - Collections
+    # - Channels
+    # - Logs
+    # - Blobs
+
+    @app.get("/")
+    def read_root():
+        return {"message": "Welcome to the BeaverDB API"}
+
+    return app
+
+
+def serve(db_path: str, host: str, port: int):
+    """
+    Initializes a BeaverDB instance and runs a Uvicorn server for it.
+
+    Args:
+        db_path: The path to the SQLite database file.
+        host: The host to bind the server to.
+        port: The port to run the server on.
+    """
+    db = BeaverDB(db_path)
+    app = build(db)
+    uvicorn.run(app, host=host, port=port)

{beaver_db-0.16.8.dist-info → beaver_db-0.17.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: beaver-db
-Version: 0.16.8
+Version: 0.17.0
 Summary: Fast, embedded, and multi-modal DB based on SQLite for AI-powered applications.
 License-File: LICENSE
 Classifier: License :: OSI Approved :: MIT License
@@ -9,8 +9,17 @@ Classifier: Programming Language :: Python :: 3.13
 Classifier: Topic :: Database
 Classifier: Topic :: Software Development :: Libraries :: Application Frameworks
 Requires-Python: >=3.13
-Provides-Extra: faiss
-Requires-Dist: faiss-cpu>=1.12.0; extra == 'faiss'
+Provides-Extra: cli
+Requires-Dist: typer>=0.19.2; extra == 'cli'
+Provides-Extra: full
+Requires-Dist: faiss-cpu>=1.12.0; extra == 'full'
+Requires-Dist: fastapi[standard]>=0.118.0; extra == 'full'
+Requires-Dist: typer>=0.19.2; extra == 'full'
+Provides-Extra: server
+Requires-Dist: fastapi[standard]>=0.118.0; extra == 'server'
+Requires-Dist: typer>=0.19.2; extra == 'server'
+Provides-Extra: vector
+Requires-Dist: faiss-cpu>=1.12.0; extra == 'vector'
 Description-Content-Type: text/markdown
 
 # beaver 🦫

{beaver_db-0.16.8.dist-info → beaver_db-0.17.0.dist-info}/RECORD

@@ -1,15 +1,18 @@
 beaver/__init__.py,sha256=qyEzF1Os7w4b4Hijgz0Y0R4zTrRBrHIGT1mEkZFl2YM,101
 beaver/blobs.py,sha256=YkIEskHD6oHRaJTF0P25HrTT8LqM-REyV_UBPVQxeqQ,4055
 beaver/channels.py,sha256=kIuwKMDBdDQObaKT23znsMXzfpKfE7pXSxvf-u4LlpY,9554
+beaver/cli.py,sha256=ExphB5Tx2hiyhLdtafQmANa6GA3KE5sbbI9tHWj_yY0,948
 beaver/collections.py,sha256=Wm684pGp-E89PCq9gcbbmRC9VMtTxolRVXnrxKlw2m8,24615
 beaver/core.py,sha256=68vjuEbkJTHv4SltCLCrgs34BpLCeL602oJZ6CJ34Zo,14560
 beaver/dicts.py,sha256=Xp8lPfQt08O8zCbptQLWQLO79OxG6uAVER6ryj3SScQ,5495
 beaver/lists.py,sha256=rfJ8uTNLkMREYc0uGx0z1VKt2m3eR9hvbdvDD58EbmQ,10140
 beaver/logs.py,sha256=a5xenwl5NZeegIU0dWVEs67lvaHzzw-JRAZtEzNNO3E,9529
 beaver/queues.py,sha256=Fr3oie63EtceSoiC8EOEDSLu1tDI8q2MYLXd8MEeC3g,6476
+beaver/server.py,sha256=lmzMu51cXa1Qdezg140hmsMLCxVSq8YGX0EPQfuGidk,4043
 beaver/types.py,sha256=WZLINf7hy6zdKdAFQK0EVMSl5vnY_KnrHXNdXgAKuPg,1582
 beaver/vectors.py,sha256=qvI6RwUOGrhVH5d6PUmI3jKDaoDotMy0iy-bHyvmXks,18496
-beaver_db-0.16.8.dist-info/METADATA,sha256=SN90Fv3zk6_Xa38nkVDdWNr6mifgnfi_6dJtfEeg7PA,18240
-beaver_db-0.16.8.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
-beaver_db-0.16.8.dist-info/licenses/LICENSE,sha256=1xrIY5JnMk_QDQzsqmVzPIIyCgZAkWCC8kF2Ddo1UT0,1071
-beaver_db-0.16.8.dist-info/RECORD,,
+beaver_db-0.17.0.dist-info/METADATA,sha256=NHuwsDxsE2k8qPuBWttjBMLhfh7f9NIMIhAmFjiVTgg,18615
+beaver_db-0.17.0.dist-info/WHEEL,sha256=qtCwoSJWgHk21S1Kb4ihdzI2rlJ1ZKaIurTj_ngOhyQ,87
+beaver_db-0.17.0.dist-info/entry_points.txt,sha256=bd5E2s45PoBdtdR9-ToKSdLNhmHp8naV1lWP5mOzlrc,42
+beaver_db-0.17.0.dist-info/licenses/LICENSE,sha256=1xrIY5JnMk_QDQzsqmVzPIIyCgZAkWCC8kF2Ddo1UT0,1071
+beaver_db-0.17.0.dist-info/RECORD,,

beaver_db-0.17.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+beaver = beaver.cli:app