fix: reset destroys app (#319)
Co-authored-by: cachho <admin@ch-webdev.com>
This commit is contained in:
@@ -37,3 +37,25 @@ class TestChromaDbHostsLoglevel(unittest.TestCase):
|
||||
app.chat("What text did I give you?")
|
||||
|
||||
self.assertEqual(mock_ec_get_llm_model_answer.call_args[1]["documents"], [knowledge])
|
||||
|
||||
def test_add_after_reset(self):
|
||||
"""
|
||||
Test if the `App` instance is correctly reconstructed after a reset.
|
||||
"""
|
||||
app = App()
|
||||
app.reset()
|
||||
|
||||
# Make sure the client is still healthy
|
||||
app.db.client.heartbeat()
|
||||
# Make sure the collection exists, and can be added to
|
||||
app.collection.add(
|
||||
embeddings=[[1.1, 2.3, 3.2], [4.5, 6.9, 4.4], [1.1, 2.3, 3.2]],
|
||||
metadatas=[
|
||||
{"chapter": "3", "verse": "16"},
|
||||
{"chapter": "3", "verse": "5"},
|
||||
{"chapter": "29", "verse": "11"},
|
||||
],
|
||||
ids=["id1", "id2", "id3"],
|
||||
)
|
||||
|
||||
app.reset()
|
||||
|
||||
@@ -245,8 +245,7 @@ class TestChromaDbCollection(unittest.TestCase):
|
||||
# Resetting the first one should reset them all.
|
||||
app1.reset()
|
||||
|
||||
# Reinstantiate them
|
||||
app1 = App(AppConfig(collection_name="one_collection", id="new_app_id_1", collect_metrics=False))
|
||||
# Reinstantiate app2-4, app1 doesn't have to be reinstantiated (PR #319)
|
||||
app2 = App(AppConfig(collection_name="one_collection", id="new_app_id_2", collect_metrics=False))
|
||||
app3 = App(AppConfig(collection_name="three_collection", id="new_app_id_3", collect_metrics=False))
|
||||
app4 = App(AppConfig(collection_name="four_collection", id="new_app_id_3", collect_metrics=False))
|
||||
|
||||
Reference in New Issue
Block a user