From aa4ab1ed8a58ea13b8395a248fafd86d62a1ccdc Mon Sep 17 00:00:00 2001 From: vasilije Date: Tue, 6 Jan 2026 18:05:34 +0100 Subject: [PATCH] reformat --- .../databases/vector/embeddings/LiteLLMEmbeddingEngine.py | 4 +++- cognee/modules/users/authentication/get_api_auth_backend.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) 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/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(