feat: Add embedding manager (#570)

2023-09-11 23:43:53 -07:00
parent ba208f5b48
commit 2bd6881361
16 changed files with 311 additions and 73 deletions
--- a/tests/chunkers/test_text.py
+++ b/tests/chunkers/test_text.py
@@ -69,9 +69,12 @@ class MockLoader:
        Mock loader that returns a list of data dictionaries.
        Adjust this method to return different data for testing.
        """
-        return [
-            {
-                "content": src,
-                "meta_data": {"url": "none"},
-            }
-        ]
+        return {
+            "doc_id": "123",
+            "data": [
+                {
+                    "content": src,
+                    "meta_data": {"url": "none"},
+                }
+            ]
+        }