libro-sdk 0.1.0__py3-none-any.whl

libro/__init__.py

@@ -0,0 +1,3 @@
+from .client import LibroClient
+
+__all__ = ["LibroClient"]

libro/client.py

@@ -0,0 +1,77 @@
+import os
+import requests
+from typing import Dict, Any, List, Optional
+
+class LibroError(Exception):
+    """Base exception for Libro SDK errors."""
+    pass
+
+class LibroClient:
+    def __init__(self, api_key: Optional[str] = None, base_url: str = "https://api.libro.ai"):
+        self.api_key = api_key or os.environ.get("LIBRO_API_KEY")
+        if not self.api_key:
+            raise ValueError("API key is required. Pass it to the client or set LIBRO_API_KEY env var.")
+        
+        # Remove trailing slash
+        self.base_url = base_url.rstrip("/")
+        self.session = requests.Session()
+        self.session.headers.update({
+            "Authorization": f"Bearer {self.api_key}",
+            "Content-Type": "application/json"
+        })
+
+    def ingest(self, user_id: str, text: str, metadata: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Ingest a new memory for a user."""
+        payload = {"userId": user_id, "text": text}
+        if metadata is not None:
+            payload["metadata"] = metadata
+
+        res = self.session.post(f"{self.base_url}/api/v1/ingest", json=payload)
+        self._check_response(res)
+        return res.json()
+
+    def get_context(self, user_id: str, query: str) -> Dict[str, Any]:
+        """Retrieve relevant context for a user based on a query."""
+        payload = {"userId": user_id, "query": query}
+        res = self.session.post(f"{self.base_url}/api/v1/get-context", json=payload)
+        self._check_response(res)
+        return res.json()
+
+    def forget(self, user_id: str, memory_id: Optional[str] = None, query: Optional[str] = None) -> Dict[str, Any]:
+        """Delete a specific memory by ID, or memories matching a query for a user."""
+        if not memory_id and not query:
+            raise ValueError("Must provide either memory_id or query to forget.")
+        
+        payload = {"userId": user_id}
+        if memory_id:
+            payload["memoryId"] = memory_id
+        if query:
+            payload["query"] = query
+
+        res = self.session.post(f"{self.base_url}/api/v1/forget", json=payload)
+        self._check_response(res)
+        return res.json()
+
+    def update(self, user_id: str, memory_id: str, text: Optional[str] = None, metadata: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
+        """Update an existing memory's text or metadata."""
+        if not text and not metadata:
+            raise ValueError("Must provide either text or metadata to update.")
+            
+        payload = {"userId": user_id, "memoryId": memory_id}
+        if text:
+            payload["text"] = text
+        if metadata is not None:
+            payload["metadata"] = metadata
+
+        res = self.session.post(f"{self.base_url}/api/v1/update", json=payload)
+        self._check_response(res)
+        return res.json()
+
+    def _check_response(self, response: requests.Response):
+        if not response.ok:
+            try:
+                err_data = response.json()
+                msg = err_data.get("error", response.text)
+            except Exception:
+                msg = response.text
+            raise LibroError(f"API Error {response.status_code}: {msg}")

libro_sdk-0.1.0.dist-info/METADATA

@@ -0,0 +1,45 @@
+Metadata-Version: 2.4
+Name: libro-sdk
+Version: 0.1.0
+Summary: Official Python SDK for the Libro AI memory infrastructure.
+Author: Libro Team
+Requires-Python: >=3.8
+Description-Content-Type: text/markdown
+Requires-Dist: requests>=2.25.0
+Provides-Extra: dev
+Requires-Dist: pytest>=7.0; extra == "dev"
+Requires-Dist: mypy>=1.0; extra == "dev"
+
+# Libro Python SDK
+
+Official Python SDK for Libro, the open-source AI memory infrastructure.
+
+## Installation
+
+```bash
+pip install libro-sdk
+```
+
+## Quick Start
+
+```python
+from libro import LibroClient
+
+# Initialize the client
+client = LibroClient(api_key="your_api_key", base_url="http://localhost:3000")
+
+# 1. Ingest a memory
+client.ingest(
+    user_id="user_123",
+    text="User prefers Python over Java",
+    metadata={"source": "slack"}
+)
+
+# 2. Retrieve context
+context = client.get_context(
+    user_id="user_123",
+    query="What programming language do they like?"
+)
+
+print(context)
+```

libro_sdk-0.1.0.dist-info/RECORD

@@ -0,0 +1,6 @@
+libro/__init__.py,sha256=U5gE770Nq03ZujANFc4Wpmm6-XOBT_wZAghqStybE_s,59
+libro/client.py,sha256=sadhVO1Jj0cZIapB47QAGXil7VZ4mvkZQmoneK_NH4M,3160
+libro_sdk-0.1.0.dist-info/METADATA,sha256=9sjQrpB-6ckjXGacUDEiTugfQz2lnJwBONpBkEj_oP0,936
+libro_sdk-0.1.0.dist-info/WHEEL,sha256=aeYiig01lYGDzBgS8HxWXOg3uV61G9ijOsup-k9o1sk,91
+libro_sdk-0.1.0.dist-info/top_level.txt,sha256=aPNst6WSPRjN2wqWLHriT_bE2asBn6LnUNeaLh2kGMc,6
+libro_sdk-0.1.0.dist-info/RECORD,,

libro_sdk-0.1.0.dist-info/WHEEL

@@ -0,0 +1,5 @@
+Wheel-Version: 1.0
+Generator: setuptools (82.0.1)
+Root-Is-Purelib: true
+Tag: py3-none-any
+

libro_sdk-0.1.0.dist-info/top_level.txt

@@ -0,0 +1 @@
+libro