kanshan/graphrag
1
0
Fork 0
mirror of https://github.com/microsoft/graphrag.git synced 2026-01-14 09:07:20 +08:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

integration test fix for lancedb

Browse Source
This commit is contained in:
Gaudy Blanco 2025-09-18 14:16:56 -06:00
parent 2f247da3ee
commit efea0622c4
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/integration/vector_stores/test_lancedb.py
View File

@ -20,7 +20,8 @@ def test_vector_store_operations():
try:
vector_store = LanceDBVectorStore(
vector_store_schema_config=VectorStoreSchemaConfig(
index_name="test_collection"
index_name="test_collection",
vector_size=5
)
)
vector_store.connect(db_uri=temp_dir)
@ -98,7 +99,8 @@ def test_empty_collection():
try:
vector_store = LanceDBVectorStore(
vector_store_schema_config=VectorStoreSchemaConfig(
index_name="empty_collection"
index_name="empty_collection",
vector_size=5
)
)
vector_store.connect(db_uri=temp_dir)
@ -142,7 +144,8 @@ def test_filter_search():
try:
vector_store = LanceDBVectorStore(
vector_store_schema_config=VectorStoreSchemaConfig(
index_name="filter_collection"
index_name="filter_collection",
vector_size=5
)
)