gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix: fixes vector index lock for vector collection creation

Browse source
This commit is contained in:
hajdul88 2025-08-01 14:37:30 +02:00
parent cd930ed646
commit f2e66bc276
1 changed files with 6 additions and 3 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
cognee/tasks/storage/index_data_points.py
View file

@ -1,9 +1,11 @@
import asyncio
from cognee.shared.logging_utils import get_logger from cognee.shared.logging_utils import get_logger
from cognee.infrastructure.databases.exceptions.EmbeddingException import EmbeddingException from cognee.infrastructure.databases.exceptions.EmbeddingException import EmbeddingException
from cognee.infrastructure.databases.vector import get_vector_engine from cognee.infrastructure.databases.vector import get_vector_engine
from cognee.infrastructure.engine import DataPoint from cognee.infrastructure.engine import DataPoint
vector_index_lock = asyncio.Lock()
logger = get_logger("index_data_points") logger = get_logger("index_data_points")
@ -22,6 +24,7 @@ async def index_data_points(data_points: list[DataPoint]):
index_name = f"{data_point_type.__name__}_{field_name}" index_name = f"{data_point_type.__name__}_{field_name}"
async with vector_index_lock:
if index_name not in created_indexes: if index_name not in created_indexes:
await vector_engine.create_vector_index(data_point_type.__name__, field_name) await vector_engine.create_vector_index(data_point_type.__name__, field_name)
created_indexes[index_name] = True created_indexes[index_name] = True