From c8202c51a7b45995e67d307ac25e0a32f64b9b06 Mon Sep 17 00:00:00 2001 From: vasilije Date: Wed, 6 Aug 2025 19:11:39 +0200 Subject: [PATCH] format fix --- .../databases/vector/embeddings/LiteLLMEmbeddingEngine.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py index e1109cf1c..e3cdaea00 100644 --- a/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py +++ b/cognee/infrastructure/databases/vector/embeddings/LiteLLMEmbeddingEngine.py @@ -186,7 +186,9 @@ class LiteLLMEmbeddingEngine(EmbeddingEngine): tokenizer = MistralTokenizer(model=model, max_tokens=self.max_tokens) else: try: - tokenizer = HuggingFaceTokenizer(model=self.model.replace('hosted_vllm/',""), max_tokens=self.max_tokens) + tokenizer = HuggingFaceTokenizer( + model=self.model.replace("hosted_vllm/", ""), max_tokens=self.max_tokens + ) except Exception as e: logger.warning(f"Could not get tokenizer from HuggingFace due to: {e}") logger.info("Switching to TikToken default tokenizer.")