refactor(client): rename get_by_id to get and update tests

* Update method names for consistency and clarity
* Refactor tests to use new method names
* Improve pagination and filtering examples in usage scripts

Author: atasoglu

examples/basic_usage.py

@@ -43,7 +43,7 @@ def main():
         print(f"  [{rowid}] {text[:50]}... (distance: {distance:.4f})")
 
     # Get record by ID
-    record = client.get_by_id(rowids[0])
+    record = client.get(rowids[0])
     if record:
         rowid, text, metadata, embedding = record
         print(f"\nRecord {rowid}: {text}")

examples/batch_operations.py

@@ -26,17 +26,12 @@ def main():
     rowids = client.add(texts=texts, embeddings=embeddings, metadata=metadata)
     print(f"Inserted {len(rowids)} products")
 
-    # Pagination - list first page
+    # Pagination using get_all with batch_size
     page_size = 10
-    page_1 = client.list_results(limit=page_size, offset=0, order="asc")
-    print(f"\nPage 1 ({len(page_1)} items):")
-    for rowid, text, meta, _ in page_1[:3]:
-        print(f"  [{rowid}] {text} - ${meta['price']}")
-
-    # Pagination - list second page
-    page_2 = client.list_results(limit=page_size, offset=page_size, order="asc")
-    print(f"\nPage 2 ({len(page_2)} items):")
-    for rowid, text, meta, _ in page_2[:3]:
+    print(f"\nFirst {page_size} items:")
+    for i, (rowid, text, meta, _) in enumerate(client.get_all(batch_size=page_size)):
+        if i >= 3:
+            break
         print(f"  [{rowid}] {text} - ${meta['price']}")
 
     # Batch retrieval

examples/metadata_filtering.py

@@ -2,8 +2,7 @@
 
 Demonstrates:
 - Adding records with metadata
-- Filtering by metadata
-- Filtering by text
+- Querying records with get_all
 - Updating metadata
 """
 
@@ -36,17 +35,17 @@ def main():
     rowids = client.add(texts=texts, embeddings=embeddings, metadata=metadata)
     print(f"Added {len(rowids)} articles")
 
-    # Filter by exact metadata match
-    alice_articles = client.get_by_metadata(
-        {"category": "programming", "author": "Alice", "year": 2023}
-    )
-    print(f"\nAlice's programming articles: {len(alice_articles)}")
-    for rowid, text, meta, _ in alice_articles:
-        print(f"  [{rowid}] {text} - {meta}")
+    # Query all articles and filter by author
+    print("\nAlice's articles:")
+    for rowid, text, meta, _ in client.get_all():
+        if meta.get("author") == "Alice":
+            print(f"  [{rowid}] {text} - {meta}")
 
-    # Filter by text
-    python_articles = client.get_by_text("Python for data science")
-    print(f"\nPython articles: {len(python_articles)}")
+    # Query all articles and filter by text
+    print("\nPython-related articles:")
+    for rowid, text, meta, _ in client.get_all():
+        if "Python" in text:
+            print(f"  [{rowid}] {text}")
 
     # Update metadata
     if rowids:
@@ -59,7 +58,7 @@ def main():
                 "updated": True,
             },
         )
-        updated = client.get_by_id(rowids[0])
+        updated = client.get(rowids[0])
         if updated:
             print(f"\nUpdated metadata: {updated[2]}")
 

examples/real_world_scenario.py

@@ -81,21 +81,22 @@ def main():
 
         print("Top 3 results:")
         for i, (rowid, text, distance) in enumerate(results, 1):
-            record = client.get_by_id(rowid)
+            record = client.get(rowid)
             if record:
                 _, _, meta, _ = record
                 print(f"  {i}. [{meta['category']}] {text[:60]}...")
                 print(
                     f"     Distance: {distance:.4f}, Difficulty: {meta['difficulty']}\n"
                 )
 
-        # Filter by category
-        print("All AI-related documents:")
-        ai_docs = client.get_by_metadata(
-            {"category": "ai", "language": "general", "difficulty": "intermediate"}
-        )
-        for rowid, text, meta, _ in ai_docs:
-            print(f"  • {text[:60]}...")
+        # Filter by category using get_all
+        print("All AI-related documents (intermediate):")
+        for rowid, text, meta, _ in client.get_all():
+            if (
+                meta.get("category") == "ai"
+                and meta.get("difficulty") == "intermediate"
+            ):
+                print(f"  • {text[:60]}...")
 
         # Update document
         if rowids:
@@ -114,9 +115,12 @@ def main():
         print("\nKnowledge base statistics:")
         print(f"  Total documents: {client.count()}")
 
-        # List recent documents
-        recent = client.list_results(limit=3, offset=0, order="desc")
-        print(f"  Most recent: {len(recent)} documents")
+        # List first 3 documents
+        print("  First 3 documents:")
+        for i, (rowid, text, _, _) in enumerate(client.get_all()):
+            if i >= 3:
+                break
+            print(f"    [{rowid}] {text[:50]}...")
 
 
 if __name__ == "__main__":