save relationship of objects

2023-09-13 17:55:27 +08:00
parent 8ac6065332
commit 6e5beb4cdf
7 changed files with 131 additions and 11 deletions
@@ -14,13 +14,13 @@ import asyncio
 import unittest


-async def test_embedding_email():
+async def test_vector():
    storage = ChromaVectorStore("test")
    await storage.insert([1, 2, 3], "test")
    ids = await storage.query([1, 2, 3], 10)
    print(ids)

-class TestVectorSTorage(unittest.TestCase):
+class TestVectorStorage(unittest.TestCase):
    def test_run(self):
        asyncio.run(test_vector())