|
5 | 5 | import pytest |
6 | 6 |
|
7 | 7 | from autointent import VectorIndex |
8 | | -from autointent.configs import EmbedderConfig, FaissConfig, OpenSearchConfig |
| 8 | +from autointent.configs import EmbedderConfig, FaissConfig, OpenSearchConfig, get_default_embedder_config |
9 | 9 | from autointent.custom_types import Document |
10 | 10 |
|
11 | 11 | # Check if opensearch-py is available |
@@ -56,7 +56,7 @@ class TestVectorIndex: |
56 | 56 | @pytest.fixture |
57 | 57 | def embedder_config(self) -> EmbedderConfig: |
58 | 58 | """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") |
60 | 60 |
|
61 | 61 | @pytest.fixture |
62 | 62 | def vector_index(self, embedder_config: EmbedderConfig, vector_config) -> VectorIndex: |
@@ -242,7 +242,7 @@ def test_load_with_embedder_override( |
242 | 242 | vector_index.dump(dump_path) |
243 | 243 |
|
244 | 244 | # 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") |
246 | 246 | override_config.device = "cpu" |
247 | 247 | override_config.batch_size = 1 |
248 | 248 |
|
@@ -287,7 +287,7 @@ def test_abstract_config_raises_error(self): |
287 | 287 | """Test that using abstract VectorIndexConfig raises an error.""" |
288 | 288 | from autointent.configs import VectorIndexConfig |
289 | 289 |
|
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") |
291 | 291 |
|
292 | 292 | vector_index = VectorIndex(embedder_config=embedder_config, config=VectorIndexConfig()) |
293 | 293 | with pytest.raises(TypeError, match="Passed abstract vector index config"): |
|
0 commit comments