Update default vector model logic for embeddings.batchtransform

2026-10-12 02:19:52 +00:00 · 2024-02-28 13:28:50 -05:00
parent 88430d7824
commit 9a670cd5ec
4 changed files with 21 additions and 3 deletions
@@ -685,6 +685,9 @@ class Common:
            )
            embeddings.index(data)

+            # Test transform
+            self.assertEqual(embeddings.transform("feel good story").shape, (768,))
+
            # Run search
            result = embeddings.search("feel good story", 1)[0]
            self.assertEqual(result["text"], data[4][1])