@@ -51,7 +51,7 @@ class FileEmbeddingStoreTest {
5151 )
5252
5353 store.add(embeddings)
54- val loaded = store.getAll ()
54+ val loaded = store.get ()
5555
5656 Assertions .assertEquals(2 , loaded.size)
5757 Assertions .assertEquals(embeddings[0 ].id, loaded[0 ].id)
@@ -72,7 +72,7 @@ class FileEmbeddingStoreTest {
7272 store.add(first)
7373 store.add(second)
7474
75- val all = store.getAll ()
75+ val all = store.get ()
7676 Assertions .assertEquals(2 , all.size)
7777 Assertions .assertEquals(1 , all[0 ].id)
7878 Assertions .assertEquals(2 , all[1 ].id)
@@ -90,7 +90,7 @@ class FileEmbeddingStoreTest {
9090 store.add(embeddings)
9191 store.remove(listOf (2L ))
9292
93- val remaining = store.getAll ()
93+ val remaining = store.get ()
9494 Assertions .assertEquals(2 , remaining.size)
9595 Assertions .assertFalse(remaining.any { it.id == 2L })
9696 }
@@ -101,7 +101,7 @@ class FileEmbeddingStoreTest {
101101 val embeddings = listOf (embedding(1 , 100 , FloatArray (embeddingLength) { 0.1f }))
102102 store.add(embeddings)
103103
104- val firstLoad = store.getAll ()
104+ val firstLoad = store.get ()
105105 if (store.useCache){
106106 Assertions .assertTrue(store.isCached)
107107 }else {
@@ -111,7 +111,7 @@ class FileEmbeddingStoreTest {
111111 store.clear()
112112 Assertions .assertFalse(store.isCached)
113113
114- val secondLoad = store.getAll ()
114+ val secondLoad = store.get ()
115115 assertTrue(firstLoad.zip(secondLoad).all { (a, b) ->
116116 a.id == b.id && a.date == b.date && a.embeddings.contentEquals(b.embeddings)
117117 })
@@ -123,13 +123,13 @@ class FileEmbeddingStoreTest {
123123 val embeddings = listOf (embedding(1 , 100 , FloatArray (embeddingLength) { 0.1f }))
124124 store.add(embeddings)
125125
126- val firstLoad = store.getAll ()
126+ val firstLoad = store.get ()
127127 Assertions .assertFalse(store.isCached)
128128
129129 store.clear()
130130 Assertions .assertFalse(store.isCached)
131131
132- val secondLoad = store.getAll ()
132+ val secondLoad = store.get ()
133133 assertTrue(firstLoad.zip(secondLoad).all { (a, b) ->
134134 a.id == b.id && a.date == b.date && a.embeddings.contentEquals(b.embeddings)
135135 })
0 commit comments