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

Merge branch 'feature/cog-3502-tool-logging-with-redis' of github.com:topoteretes/cognee into feature/cog-3502-tool-logging-with-redis

Browse source
This commit is contained in:
hajdul88 2026-01-19 10:22:15 +01:00
commit 489cf4b79a
4 changed files with 60 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

30
cognee/infrastructure/databases/graph/get_graph_engine.py
View file

@ -24,7 +24,6 @@ async def get_graph_engine() -> GraphDBInterface:
return graph_client return graph_client
@lru_cache
def create_graph_engine( def create_graph_engine(
graph_database_provider, graph_database_provider,
graph_file_path, graph_file_path,
@ -35,6 +34,35 @@ def create_graph_engine(
graph_database_port="", graph_database_port="",
graph_database_key="", graph_database_key="",
graph_dataset_database_handler="", graph_dataset_database_handler="",
):
"""
Wrapper function to call create graph engine with caching.
For a detailed description, see _create_graph_engine.
"""
return _create_graph_engine(
graph_database_provider,
graph_file_path,
graph_database_url,
graph_database_name,
graph_database_username,
graph_database_password,
graph_database_port,
graph_database_key,
graph_dataset_database_handler,
)
@lru_cache
def _create_graph_engine(
graph_database_provider,
graph_file_path,
graph_database_url="",
graph_database_name="",
graph_database_username="",
graph_database_password="",
graph_database_port="",
graph_database_key="",
graph_dataset_database_handler="",
): ):
""" """
Create a graph engine based on the specified provider type. Create a graph engine based on the specified provider type.

24
cognee/infrastructure/databases/vector/create_vector_engine.py
View file

@ -7,7 +7,6 @@ from cognee.infrastructure.databases.graph.config import get_graph_context_confi
from functools import lru_cache from functools import lru_cache
@lru_cache
def create_vector_engine( def create_vector_engine(
vector_db_provider: str, vector_db_provider: str,
vector_db_url: str, vector_db_url: str,
@ -15,6 +14,29 @@ def create_vector_engine(
vector_db_port: str = "", vector_db_port: str = "",
vector_db_key: str = "", vector_db_key: str = "",
vector_dataset_database_handler: str = "", vector_dataset_database_handler: str = "",
):
"""
Wrapper function to call create vector engine with caching.
For a detailed description, see _create_vector_engine.
"""
return _create_vector_engine(
vector_db_provider,
vector_db_url,
vector_db_name,
vector_db_port,
vector_db_key,
vector_dataset_database_handler,
)
@lru_cache
def _create_vector_engine(
vector_db_provider: str,
vector_db_url: str,
vector_db_name: str,
vector_db_port: str = "",
vector_db_key: str = "",
vector_dataset_database_handler: str = "",
): ):
""" """
Create a vector database engine based on the specified provider. Create a vector database engine based on the specified provider.

8
cognee/tests/test_permissions.py
View file

@ -41,14 +41,14 @@ async def _reset_engines_and_prune() -> None:
except Exception: except Exception:
pass pass
from cognee.infrastructure.databases.graph.get_graph_engine import create_graph_engine
from cognee.infrastructure.databases.relational.create_relational_engine import ( from cognee.infrastructure.databases.relational.create_relational_engine import (
create_relational_engine, create_relational_engine,
) )
from cognee.infrastructure.databases.vector.create_vector_engine import create_vector_engine from cognee.infrastructure.databases.vector.create_vector_engine import _create_vector_engine
from cognee.infrastructure.databases.graph.get_graph_engine import _create_graph_engine
create_graph_engine.cache_clear() _create_graph_engine.cache_clear()
create_vector_engine.cache_clear() _create_vector_engine.cache_clear()
create_relational_engine.cache_clear() create_relational_engine.cache_clear()
await cognee.prune.prune_data() await cognee.prune.prune_data()

8
cognee/tests/test_search_db.py
View file

@ -48,14 +48,14 @@ async def _reset_engines_and_prune() -> None:
# Engine might not exist yet # Engine might not exist yet
pass pass
from cognee.infrastructure.databases.graph.get_graph_engine import create_graph_engine from cognee.infrastructure.databases.graph.get_graph_engine import _create_graph_engine
from cognee.infrastructure.databases.vector.create_vector_engine import create_vector_engine from cognee.infrastructure.databases.vector.create_vector_engine import _create_vector_engine
from cognee.infrastructure.databases.relational.create_relational_engine import ( from cognee.infrastructure.databases.relational.create_relational_engine import (
create_relational_engine, create_relational_engine,
) )
create_graph_engine.cache_clear() _create_graph_engine.cache_clear()
create_vector_engine.cache_clear() _create_vector_engine.cache_clear()
create_relational_engine.cache_clear() create_relational_engine.cache_clear()
await cognee.prune.prune_data() await cognee.prune.prune_data()