Remove IVFIndex

langchain_postgres/indexes.py

@@ -116,33 +116,3 @@ def to_string(self) -> str:
             DeprecationWarning,
         )
         return f"ivfflat.probes = {self.probes}"
-
-
-@dataclass
-class IVFIndex(BaseIndex):
-    index_type: str = "ivf"
-    lists: int = 100
-    quantizer: str = field(
-        default="sq8", init=False
-    )  # Disable `quantizer` initialization currently only supports the value "sq8"
-
-    def index_options(self) -> str:
-        """Set index query options for vector store initialization."""
-        return f"(lists = {self.lists}, quantizer = {self.quantizer})"
-
-
-@dataclass
-class IVFQueryOptions(QueryOptions):
-    probes: int = 1
-
-    def to_parameter(self) -> list[str]:
-        """Convert index attributes to list of configurations."""
-        return [f"ivf.probes = {self.probes}"]
-
-    def to_string(self) -> str:
-        """Convert index attributes to string."""
-        warnings.warn(
-            "to_string is deprecated, use to_parameter instead.",
-            DeprecationWarning,
-        )
-        return f"ivf.probes = {self.probes}"

tests/unit_tests/test_indexes.py

@@ -7,8 +7,6 @@
     HNSWQueryOptions,
     IVFFlatIndex,
     IVFFlatQueryOptions,
-    IVFIndex,
-    IVFQueryOptions,
 )
 
 
@@ -62,21 +60,3 @@ def test_ivfflat_query_options(self) -> None:
             assert "to_string is deprecated, use to_parameter instead." in str(
                 w[-1].message
             )
-
-    def test_ivf_index(self) -> None:
-        index = IVFIndex(name="test_index", lists=200)
-        assert index.index_type == "ivf"
-        assert index.lists == 200
-        assert index.quantizer == "sq8"  # Check default value
-        assert index.index_options() == "(lists = 200, quantizer = sq8)"
-
-    def test_ivf_query_options(self) -> None:
-        options = IVFQueryOptions(probes=2)
-        assert options.to_parameter() == ["ivf.probes = 2"]
-
-        with warnings.catch_warnings(record=True) as w:
-            options.to_string()
-            assert len(w) == 1
-            assert "to_string is deprecated, use to_parameter instead." in str(
-                w[-1].message
-            )

tests/unit_tests/test_pg_vectorstore_index.py

@@ -4,7 +4,6 @@
 
 import pytest
 import pytest_asyncio
-import sqlalchemy
 from langchain_core.documents import Document
 from langchain_core.embeddings import DeterministicFakeEmbedding
 from sqlalchemy import text
@@ -15,7 +14,6 @@
     DistanceStrategy,
     HNSWIndex,
     IVFFlatIndex,
-    IVFIndex,
 )
 from tests.utils import VECTORSTORE_CONNECTION_STRING_ASYNCPG as CONNECTION_STRING
 
@@ -178,16 +176,3 @@ async def test_aapply_vector_index_ivfflat(self, vs: PGVectorStore) -> None:
     async def test_is_valid_index(self, vs: PGVectorStore) -> None:
         is_valid = await vs.ais_valid_index("invalid_index")
         assert is_valid == False
-
-    async def test_aapply_vector_index_ivf(self, vs: PGVectorStore) -> None:
-        index = IVFIndex(distance_strategy=DistanceStrategy.EUCLIDEAN)
-        await vs.aapply_vector_index(index, concurrently=True)
-        assert await vs.ais_valid_index(DEFAULT_INDEX_NAME_ASYNC)
-        index = IVFIndex(
-            name="secondindex",
-            distance_strategy=DistanceStrategy.INNER_PRODUCT,
-        )
-        await vs.aapply_vector_index(index)
-        assert await vs.ais_valid_index("secondindex")
-        await vs.adrop_vector_index("secondindex")
-        await vs.adrop_vector_index()