diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py index 28fa22b4f..7a1306bf8 100644 --- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py @@ -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, ), diff --git a/cognee/modules/graph/methods/delete_data_related_edges.py b/cognee/modules/graph/methods/delete_data_related_edges.py index 374bde5f9..7e037ab7d 100644 --- a/cognee/modules/graph/methods/delete_data_related_edges.py +++ b/cognee/modules/graph/methods/delete_data_related_edges.py @@ -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]))) - diff --git a/cognee/modules/graph/methods/delete_data_related_nodes.py b/cognee/modules/graph/methods/delete_data_related_nodes.py index c6bb7c95d..adc8e8f20 100644 --- a/cognee/modules/graph/methods/delete_data_related_nodes.py +++ b/cognee/modules/graph/methods/delete_data_related_nodes.py @@ -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]))) - diff --git a/cognee/modules/graph/methods/delete_dataset_related_edges.py b/cognee/modules/graph/methods/delete_dataset_related_edges.py index 8ac0c63f1..586d58ce9 100644 --- a/cognee/modules/graph/methods/delete_dataset_related_edges.py +++ b/cognee/modules/graph/methods/delete_dataset_related_edges.py @@ -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]))) - diff --git a/cognee/modules/graph/methods/delete_dataset_related_nodes.py b/cognee/modules/graph/methods/delete_dataset_related_nodes.py index 7e251189f..403f71430 100644 --- a/cognee/modules/graph/methods/delete_dataset_related_nodes.py +++ b/cognee/modules/graph/methods/delete_dataset_related_nodes.py @@ -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]))) - diff --git a/cognee/modules/users/authentication/get_api_auth_backend.py b/cognee/modules/users/authentication/get_api_auth_backend.py index ffb591a9d..799d118f9 100644 --- a/cognee/modules/users/authentication/get_api_auth_backend.py +++ b/cognee/modules/users/authentication/get_api_auth_backend.py @@ -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( diff --git a/cognee/tests/unit/modules/graph/methods/test_delete_data_related_edges.py b/cognee/tests/unit/modules/graph/methods/test_delete_data_related_edges.py index dca5c1558..932f78975 100644 --- a/cognee/tests/unit/modules/graph/methods/test_delete_data_related_edges.py +++ b/cognee/tests/unit/modules/graph/methods/test_delete_data_related_edges.py @@ -41,4 +41,3 @@ async def test_delete_data_related_edges_handles_empty_list(): session.scalars.assert_awaited_once() session.execute.assert_awaited_once() - diff --git a/cognee/tests/unit/modules/graph/methods/test_delete_data_related_nodes.py b/cognee/tests/unit/modules/graph/methods/test_delete_data_related_nodes.py index 44ef20476..3159d5b10 100644 --- a/cognee/tests/unit/modules/graph/methods/test_delete_data_related_nodes.py +++ b/cognee/tests/unit/modules/graph/methods/test_delete_data_related_nodes.py @@ -41,4 +41,3 @@ async def test_delete_data_related_nodes_handles_empty_list(): session.scalars.assert_awaited_once() session.execute.assert_awaited_once() - diff --git a/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_edges.py b/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_edges.py index 210a98bd5..5320ad4e0 100644 --- a/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_edges.py +++ b/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_edges.py @@ -41,4 +41,3 @@ async def test_delete_dataset_related_edges_handles_empty_list(): session.scalars.assert_awaited_once() session.execute.assert_awaited_once() - diff --git a/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_nodes.py b/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_nodes.py index f52ad58ce..bc9d1d136 100644 --- a/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_nodes.py +++ b/cognee/tests/unit/modules/graph/methods/test_delete_dataset_related_nodes.py @@ -41,4 +41,3 @@ async def test_delete_dataset_related_nodes_handles_empty_list(): session.scalars.assert_awaited_once() session.execute.assert_awaited_once() -