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

uv fix for format

Browse Source
This commit is contained in:
Gaudy Blanco 2025-09-18 14:22:30 -06:00
parent efea0622c4
commit dc00cb8322
2 changed files with 12 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
tests/integration/vector_stores/test_factory.py
View File

@ -22,7 +22,9 @@ def test_create_lancedb_vector_store():
}
vector_store = VectorStoreFactory.create_vector_store(
vector_store_type=VectorStoreType.LanceDB.value,
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
vector_store_schema_config=VectorStoreSchemaConfig(
index_name="test_collection"
),
kwargs=kwargs,
)
assert isinstance(vector_store, LanceDBVectorStore)
@ -37,7 +39,9 @@ def test_create_azure_ai_search_vector_store():
}
vector_store = VectorStoreFactory.create_vector_store(
vector_store_type=VectorStoreType.AzureAISearch.value,
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
vector_store_schema_config=VectorStoreSchemaConfig(
index_name="test_collection"
),
kwargs=kwargs,
)
assert isinstance(vector_store, AzureAISearchVectorStore)
@ -52,7 +56,9 @@ def test_create_cosmosdb_vector_store():
vector_store = VectorStoreFactory.create_vector_store(
vector_store_type=VectorStoreType.CosmosDB.value,
vector_store_schema_config=VectorStoreSchemaConfig(index_name="test_collection"),
vector_store_schema_config=VectorStoreSchemaConfig(
index_name="test_collection"
),
kwargs=kwargs,
)

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

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