Skip to content

Commit 6b80fb2

Browse files
committed
upd vector index tests
1 parent 5a6917d commit 6b80fb2

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

tests/context/test_vector_index.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,7 @@
55
import pytest
66

77
from autointent import VectorIndex
8-
from autointent.configs import EmbedderConfig, FaissConfig, OpenSearchConfig
8+
from autointent.configs import EmbedderConfig, FaissConfig, OpenSearchConfig, get_default_embedder_config
99
from autointent.custom_types import Document
1010

1111
# Check if opensearch-py is available
@@ -56,7 +56,7 @@ class TestVectorIndex:
5656
@pytest.fixture
5757
def embedder_config(self) -> EmbedderConfig:
5858
"""Create a lightweight embedder config for testing."""
59-
return EmbedderConfig.from_search_config("sentence-transformers/all-MiniLM-L6-v2")
59+
return get_default_embedder_config(model_name="sentence-transformers/all-MiniLM-L6-v2")
6060

6161
@pytest.fixture
6262
def vector_index(self, embedder_config: EmbedderConfig, vector_config) -> VectorIndex:
@@ -242,7 +242,7 @@ def test_load_with_embedder_override(
242242
vector_index.dump(dump_path)
243243

244244
# Create override config
245-
override_config = EmbedderConfig.from_search_config("sentence-transformers/all-MiniLM-L6-v2")
245+
override_config = get_default_embedder_config(model_name="sentence-transformers/all-MiniLM-L6-v2")
246246
override_config.device = "cpu"
247247
override_config.batch_size = 1
248248

@@ -287,7 +287,7 @@ def test_abstract_config_raises_error(self):
287287
"""Test that using abstract VectorIndexConfig raises an error."""
288288
from autointent.configs import VectorIndexConfig
289289

290-
embedder_config = EmbedderConfig.from_search_config("sentence-transformers/all-MiniLM-L6-v2")
290+
embedder_config = get_default_embedder_config(model_name="sentence-transformers/all-MiniLM-L6-v2")
291291

292292
vector_index = VectorIndex(embedder_config=embedder_config, config=VectorIndexConfig())
293293
with pytest.raises(TypeError, match="Passed abstract vector index config"):

0 commit comments

Comments
 (0)