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

fix: update embedding exception imports

Browse source
This commit is contained in:
lxobr 2025-09-01 14:43:05 +02:00
parent 427d67babf
commit 76143a7d48
2 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/infrastructure/databases/vector/embeddings/FastembedEmbeddingEngine.py
View file

@ -4,7 +4,7 @@ from fastembed import TextEmbedding
import litellm import litellm
import os import os
from cognee.infrastructure.databases.vector.embeddings.EmbeddingEngine import EmbeddingEngine 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 ( from cognee.infrastructure.llm.tokenizer.TikToken import (
TikTokenTokenizer, TikTokenTokenizer,
) )

8
cognee/infrastructure/databases/vector/embeddings/embedding_rate_limiter.py
View file

@ -250,9 +250,7 @@ def embedding_rate_limit_sync(func):
logger.warning(error_msg) logger.warning(error_msg)
# Create a custom embedding rate limit exception # Create a custom embedding rate limit exception
from cognee.infrastructure.databases.exceptions.EmbeddingException import ( from cognee.infrastructure.databases.exceptions import EmbeddingException
EmbeddingException,
)
raise EmbeddingException(error_msg) raise EmbeddingException(error_msg)
@ -307,9 +305,7 @@ def embedding_rate_limit_async(func):
logger.warning(error_msg) logger.warning(error_msg)
# Create a custom embedding rate limit exception # Create a custom embedding rate limit exception
from cognee.infrastructure.databases.exceptions.EmbeddingException import ( from cognee.infrastructure.databases.exceptions import EmbeddingException
EmbeddingException,
)
raise EmbeddingException(error_msg) raise EmbeddingException(error_msg)