From 76143a7d48bec0d65c3e3b1edd2f103990ad5cce Mon Sep 17 00:00:00 2001
From: lxobr <122801072+lxobr@users.noreply.github.com>
Date: Mon, 1 Sep 2025 14:43:05 +0200
Subject: [PATCH] fix: update embedding exception imports

---
 .../vector/embeddings/FastembedEmbeddingEngine.py         | 2 +-
 .../databases/vector/embeddings/embedding_rate_limiter.py | 8 ++------
 2 files changed, 3 insertions(+), 7 deletions(-)

diff --git a/cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py b/cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py
index dc8443459..acb041e76 100644
--- a/cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py
+++ b/cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py
@@ -4,7 +4,7 @@ from fastembed import TextEmbedding
 import litellm
 import os
 from cognee.infrastructure.databases.vector.embeddings.EmbeddingEngine import EmbeddingEngine
-from cognee.infrastructure.databases.exceptions.EmbeddingException import EmbeddingException
+from cognee.infrastructure.databases.exceptions import EmbeddingException
 from cognee.infrastructure.llm.tokenizer.TikToken import (
     TikTokenTokenizer,
 )
diff --git a/cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py b/cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py
index 24312dab1..27688d2c9 100644
--- a/cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py
+++ b/cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py
@@ -250,9 +250,7 @@ def embedding_rate_limit_sync(func):
             logger.warning(error_msg)
 
             # Create a custom embedding rate limit exception
-            from cognee.infrastructure.databases.exceptions.EmbeddingException import (
-                EmbeddingException,
-            )
+            from cognee.infrastructure.databases.exceptions import EmbeddingException
 
             raise EmbeddingException(error_msg)
 
@@ -307,9 +305,7 @@ def embedding_rate_limit_async(func):
             logger.warning(error_msg)
 
             # Create a custom embedding rate limit exception
-            from cognee.infrastructure.databases.exceptions.EmbeddingException import (
-                EmbeddingException,
-            )
+            from cognee.infrastructure.databases.exceptions import EmbeddingException
 
             raise EmbeddingException(error_msg)