Skip to content

fix: Resolve issue with BAML rate limit handling #1813

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
Vasilije1990 merged 9 commits into from
Dec 8, 2025
Merged

fix: Resolve issue with BAML rate limit handling #1813

Changes from 1 commit
Commits
Show all changes
9 commits
Select commit Hold shift + click to select a range
7360729
fix: Resolve issue with BAML rate limit handling
dexters1 Nov 19, 2025
a8950b2
Merge branch 'main' into baml-rate-limit-handling
dexters1 Nov 20, 2025
362aa8d
Merge branch 'main' into baml-rate-limit-handling
dexters1 Dec 1, 2025
5d0586d
Merge branch 'dev' into baml-rate-limit-handling
dexters1 Dec 5, 2025
0c97a40
feat: Add RPM control
dexters1 Dec 5, 2025
7deaa6e
feat: Add RPM limiting to Cognee
dexters1 Dec 5, 2025
a66b2ce
refactor: reduce ammount of retry attempts for baml llm calls
dexters1 Dec 5, 2025
2f572ae
test: Update embeding limiter test
dexters1 Dec 5, 2025
67a4d40
chore: regen poetry lock file
dexters1 Dec 5, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
test: Update embeding limiter test
  • Loading branch information
@dexters1
dexters1 committed Dec 5, 2025
commit 2f572ae5097f56a41c2d6a698b870c5d961d778f
5 changes: 0 additions & 5 deletions cognee/tests/unit/infrastructure/test_embedding_rate_limiting_realistic.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -6,9 +6,6 @@
from cognee.infrastructure.llm.config import (
get_llm_config,
)
from cognee.infrastructure.databases.vector.embeddings.embedding_rate_limiter import (
EmbeddingRateLimiter,
)
from cognee.tests.unit.infrastructure.mock_embedding_engine import MockEmbeddingEngine

# Configure logging
Expand All @@ -33,7 +30,6 @@ async def test_embedding_rate_limiting_realistic():

# Clear the config and rate limiter caches to ensure our settings are applied
get_llm_config.cache_clear()
EmbeddingRateLimiter.reset_instance()

# Create a fresh config instance and verify settings
config = get_llm_config()
Expand Down Expand Up @@ -170,7 +166,6 @@ async def test_with_mock_failures():

# Clear caches
get_llm_config.cache_clear()
EmbeddingRateLimiter.reset_instance()

# Create a mock engine configured to fail every 3rd request
engine = MockEmbeddingEngine()
Expand Down
Loading