add persist

This commit is contained in:
seehi 2024-03-11 20:18:27 +08:00
parent 6a388b53f1
commit 0576ab2ed1
9 changed files with 77 additions and 21 deletions

View file

@ -150,8 +150,8 @@ class TestSimpleEngine:
def rag_key(self):
return ""
def model_dump(self):
return {}
def model_dump_json(self):
return ""
objs = [CustomTextNode(text=f"text_{i}", metadata={"obj": f"obj_{i}"}) for i in range(2)]
engine = SimpleEngine(retriever=mock_retriever, index=mocker.MagicMock())