fix epithermal sideeffect during retirement (#21)

* fix epthermal sideeffect during retirement

* Update tests/steps/qdrant/e2e_test.py

Author: tweigel-dev

tests/steps/qdrant/e2e_test.py

@@ -12,7 +12,7 @@
 import pytest
 
 # qdrant-Lite; See: https://qdrant.io/docs/qdrant_lite.md
-from qdrant_client import QdrantClient
+from qdrant_client import QdrantClient, models
 
 from wurzel.exceptions import StepFailed
 from wurzel.step_executor import BaseStepExecutor
@@ -95,6 +95,46 @@ def test_qdrant_collection_retirement(
         )
 
 
+def test_qdrant_get_collections_with_ephemerals(
+    input_output_folder: Tuple[Path, Path], env, dummy_collection
+):
+    input_path, output_path = input_output_folder
+    HIST_LEN = 3
+    env.set("COLLECTION_HISTORY_LEN", str(HIST_LEN))
+    env.set("COLLECTION", "tenant1-dev")
+    input_file = input_path / "qdrant_at.csv"
+    shutil.copy("./tests/data/embedded.csv", input_file)
+    client = QdrantClient(location=":memory:")
+    {
+        client.create_collection(
+            coll,
+            vectors_config=models.VectorParams(
+                size=100, distance=models.Distance.COSINE
+            ),
+        )
+        for coll in [
+            "tenant1-dev_v1",
+            "tenant1-dev_v2",
+            "tenant1-dev_v3",
+            "tenant1-dev-feature-abc_v1",
+        ]
+    }
+
+    client.close = print
+    with unittest.mock.patch("wurzel.steps.qdrant.step.QdrantClient") as mock:
+        mock.return_value = client
+        step = QdrantConnectorStep()
+        result = step._get_collection_versions()
+        assert len(result) == 3
+        assert set(result.keys()) == {1, 2, 3}
+
+        env.set("COLLECTION", "tenant1-dev-feature-abc")
+        step = QdrantConnectorStep()
+        result = step._get_collection_versions()
+        assert len(result) == 1
+        assert set(result.keys()) == {1}
+
+
 def test_qdrant_connector_csv_partially_not_same_shape(
     input_output_folder: Tuple[Path, Path],
 ):

wurzel/steps/qdrant/step.py

@@ -210,6 +210,6 @@ def _get_collection_versions(self) -> dict[int, str]:
         versioned_collections = {
             int(previous.name.split("_v")[-1]): previous.name
             for previous in previous_collections
-            if self.settings.COLLECTION in previous.name
+            if f"{self.settings.COLLECTION}_v" in previous.name
         }
         return dict(sorted(versioned_collections.items()))