Config test

Signed-off-by: Marcel Coetzee <[email protected]>
dlt-hub · Jun 21, 2024 · 44405d7 · 44405d7
1 parent e450f47
commit 44405d7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/load/lancedb/test_config.py b/tests/load/lancedb/test_config.py
@@ -25,13 +25,14 @@ def drop_lancedb_data() -> Iterator[None]:
 
 def test_lancedb_configuration() -> None:
     os.environ["DESTINATION__LANCEDB__EMBEDDING_MODEL_PROVIDER"] = "colbert"
+    os.environ["DESTINATION__LANCEDB__EMBEDDING_MODEL"] = "text-embedding-3-small"
 
     config = resolve_configuration(
         LanceDBClientConfiguration()._bind_dataset_name(dataset_name="dataset"),
         sections=("destination", "lancedb"),
     )
     assert config.embedding_model_provider == "colbert"
-    assert config.embedding_model == "embed-english-v3.0"
+    assert config.embedding_model == "text-embedding-3-small"
     assert config.embedding_model_dimensions is None
     assert config.sentinel_table_name == "dltSentinelTable"
     assert config.id_field_name == "id__"