diff --git a/cognee/infrastructure/llm/__init__.py b/cognee/infrastructure/llm/__init__.py
index 8e0cce1e6..09c049ddb 100644
--- a/cognee/infrastructure/llm/__init__.py
+++ b/cognee/infrastructure/llm/__init__.py
@@ -10,3 +10,4 @@ from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.l
 from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.utils import (
     test_embedding_connection,
 )
+from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm import rate_limiter