Added example of getting items within a certain distance with SQLAlchemy ORM

ankane · ankane · commit 8885197e0e4b · 2023-05-11T09:39:54.000-07:00
diff --git a/tests/test_sqlalchemy.py b/tests/test_sqlalchemy.py
@@ -111,6 +111,12 @@ def test_filter(self):
             items = session.query(Item).filter(Item.embedding.l2_distance([1, 1, 1]) < 1).all()
             assert [v.id for v in items] == [1]
 
+    def test_filter_orm(self):
+        create_items()
+        with Session(engine) as session:
+            items = session.scalars(select(Item).filter(Item.embedding.l2_distance([1, 1, 1]) < 1))
+            assert [v.id for v in items] == [1]
+
     def test_select(self):
         with Session(engine) as session:
             session.add(Item(embedding=[2, 3, 3]))
