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

reformat applied

Browse source
This commit is contained in:
vasilije 2026-01-06 11:55:36 +01:00
parent 97c9cf2d8a
commit e20751688c
10 changed files with 4 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py
View file

@ -134,7 +134,9 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine):
litellm.aembedding(
model=self.model,
input=text,
api_key=self.api_key if self.api_key and self.api_key.strip() != "" else "EMPTY",
api_key=self.api_key
if self.api_key and self.api_key.strip() != ""
else "EMPTY",
api_base=self.endpoint,
api_version=self.api_version,
),

1
cognee/modules/graph/methods/delete_data_related_edges.py
View file

@ -11,4 +11,3 @@ async def delete_data_related_edges(data_id: UUID, session: AsyncSession):
edges = (await session.scalars(select(Edge).where(Edge.data_id == data_id))).all()
await session.execute(delete(Edge).where(Edge.id.in_([edge.id for edge in edges])))

1
cognee/modules/graph/methods/delete_data_related_nodes.py
View file

@ -11,4 +11,3 @@ async def delete_data_related_nodes(data_id: UUID, session: AsyncSession):
nodes = (await session.scalars(select(Node).where(Node.data_id == data_id))).all()
await session.execute(delete(Node).where(Node.id.in_([node.id for node in nodes])))

1
cognee/modules/graph/methods/delete_dataset_related_edges.py
View file

@ -11,4 +11,3 @@ async def delete_dataset_related_edges(dataset_id: UUID, session: AsyncSession):
edges = (await session.scalars(select(Edge).where(Edge.dataset_id == dataset_id))).all()
await session.execute(delete(Edge).where(Edge.id.in_([edge.id for edge in edges])))

1
cognee/modules/graph/methods/delete_dataset_related_nodes.py
View file

@ -11,4 +11,3 @@ async def delete_dataset_related_nodes(dataset_id: UUID, session: AsyncSession):
nodes = (await session.scalars(select(Node).where(Node.dataset_id == dataset_id))).all()
await session.execute(delete(Node).where(Node.id.in_([node.id for node in nodes])))

2
cognee/modules/users/authentication/get_api_auth_backend.py
View file

@ -20,7 +20,7 @@ def get_api_auth_backend():
lifetime_seconds = int(os.getenv("JWT_LIFETIME_SECONDS", "3600"))
except ValueError:
lifetime_seconds = 3600
return APIJWTStrategy(secret, lifetime_seconds=lifetime_seconds)
auth_backend = AuthenticationBackend(

1
cognee/tests/unit/modules/graph/methods/test_delete_data_related_edges.py
View file

@ -41,4 +41,3 @@ async def test_delete_data_related_edges_handles_empty_list():
session.scalars.assert_awaited_once()
session.execute.assert_awaited_once()

1
cognee/tests/unit/modules/graph/methods/test_delete_data_related_nodes.py
View file

@ -41,4 +41,3 @@ async def test_delete_data_related_nodes_handles_empty_list():
session.scalars.assert_awaited_once()
session.execute.assert_awaited_once()

1
cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_edges.py
View file

@ -41,4 +41,3 @@ async def test_delete_dataset_related_edges_handles_empty_list():
session.scalars.assert_awaited_once()
session.execute.assert_awaited_once()

1
cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_nodes.py
View file

@ -41,4 +41,3 @@ async def test_delete_dataset_related_nodes_handles_empty_list():
session.scalars.assert_awaited_once()
session.execute.assert_awaited_once()