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

test: Update embeding limiter test

Browse source
This commit is contained in:
Igor Ilic 2025-12-05 19:18:48 +01:00
parent a66b2ceeca
commit 2f572ae509
1 changed files with 0 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
cognee/tests/unit/infrastructure/test_embedding_rate_limiting_realistic.py
View file

@ -6,9 +6,6 @@ import logging
from cognee.infrastructure.llm.config import ( from cognee.infrastructure.llm.config import (
get_llm_config, get_llm_config,
) )
from cognee.infrastructure.databases.vector.embeddings.embedding_rate_limiter import (
EmbeddingRateLimiter,
)
from cognee.tests.unit.infrastructure.mock_embedding_engine import MockEmbeddingEngine from cognee.tests.unit.infrastructure.mock_embedding_engine import MockEmbeddingEngine
# Configure logging # Configure logging
@ -33,7 +30,6 @@ async def test_embedding_rate_limiting_realistic():
# Clear the config and rate limiter caches to ensure our settings are applied # Clear the config and rate limiter caches to ensure our settings are applied
get_llm_config.cache_clear() get_llm_config.cache_clear()
EmbeddingRateLimiter.reset_instance()
# Create a fresh config instance and verify settings # Create a fresh config instance and verify settings
config = get_llm_config() config = get_llm_config()
@ -170,7 +166,6 @@ async def test_with_mock_failures():
# Clear caches # Clear caches
get_llm_config.cache_clear() get_llm_config.cache_clear()
EmbeddingRateLimiter.reset_instance()
# Create a mock engine configured to fail every 3rd request # Create a mock engine configured to fail every 3rd request
engine = MockEmbeddingEngine() engine = MockEmbeddingEngine()