diff --git a/cognee/infrastructure/llm/LLMAdapter.py b/cognee/infrastructure/llm/LLMAdapter.py
index 8fcad45c0..5fca9f8fb 100644
--- a/cognee/infrastructure/llm/LLMAdapter.py
+++ b/cognee/infrastructure/llm/LLMAdapter.py
@@ -21,7 +21,7 @@ class LLMAdapter:
     def acreate_structured_output(
         text_input: str, system_prompt: str, response_model: Type[BaseModel]
     ) -> BaseModel:
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.get_llm_client import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.get_llm_client import (
             get_llm_client,
         )
 
@@ -34,7 +34,7 @@ class LLMAdapter:
     def create_structured_output(
         text_input: str, system_prompt: str, response_model: Type[BaseModel]
     ) -> BaseModel:
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.get_llm_client import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.get_llm_client import (
             get_llm_client,
         )
 
@@ -45,7 +45,7 @@ class LLMAdapter:
 
     @staticmethod
     def create_transcript(input):
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.get_llm_client import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.get_llm_client import (
             get_llm_client,
         )
 
@@ -54,7 +54,7 @@ class LLMAdapter:
 
     @staticmethod
     def transcribe_image(input):
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.get_llm_client import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.get_llm_client import (
             get_llm_client,
         )
 
@@ -63,7 +63,7 @@ class LLMAdapter:
 
     @staticmethod
     def show_prompt(text_input: str, system_prompt: str) -> str:
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.get_llm_client import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.get_llm_client import (
             get_llm_client,
         )
 
@@ -88,7 +88,7 @@ class LLMAdapter:
 
             return extract_content_graph(content=content, response_model=response_model, mode=mode)
         else:
-            from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.extraction import (
+            from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.extraction import (
                 extract_content_graph,
             )
 
@@ -97,7 +97,7 @@ class LLMAdapter:
     @staticmethod
     def extract_categories(content: str, response_model: Type[BaseModel]):
         # TODO: Add BAML version of category and extraction and update function
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.extraction import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.extraction import (
             extract_categories,
         )
 
@@ -113,7 +113,7 @@ class LLMAdapter:
 
             return extract_code_summary(content=content)
         else:
-            from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.extraction import (
+            from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.extraction import (
                 extract_code_summary,
             )
 
@@ -129,7 +129,7 @@ class LLMAdapter:
 
             return extract_summary(content=content, response_model=response_model)
         else:
-            from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.extraction import (
+            from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.extraction import (
                 extract_summary,
             )
 
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_categories.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/extract_categories.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_categories.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/extract_categories.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_summary.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/extract_summary.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/extract_summary.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/extract_summary.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/knowledge_graph/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/knowledge_graph/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/knowledge_graph/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/knowledge_graph/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/knowledge_graph/extract_content_graph.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/knowledge_graph/extract_content_graph.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/knowledge_graph/extract_content_graph.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/knowledge_graph/extract_content_graph.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/texts.json b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/texts.json
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/extraction/texts.json
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/extraction/texts.json
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/anthropic/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/anthropic/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/anthropic/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py
similarity index 93%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/anthropic/adapter.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py
index e814505d6..5a3b2d5f1 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/anthropic/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/anthropic/adapter.py
@@ -3,10 +3,10 @@ from pydantic import BaseModel
 import instructor
 
 from cognee.exceptions import InvalidValueError
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
     sleep_and_retry_async,
 )
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/gemini/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/gemini/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/gemini/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py
similarity index 95%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/gemini/adapter.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py
index 4ddcefbb7..139d92980 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/gemini/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/gemini/adapter.py
@@ -6,11 +6,11 @@ from litellm import acompletion, JSONSchemaValidationError
 from cognee.shared.logging_utils import get_logger
 from cognee.modules.observability.get_observe import get_observe
 from cognee.exceptions import InvalidValueError
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
 from cognee.infrastructure.llm.LLMAdapter import LLMAdapter
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
     sleep_and_retry_async,
 )
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/generic_llm_api/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/generic_llm_api/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/generic_llm_api/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py
similarity index 96%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/generic_llm_api/adapter.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py
index 3da139709..c62ab1b2b 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/generic_llm_api/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/generic_llm_api/adapter.py
@@ -9,10 +9,10 @@ from litellm.exceptions import ContentPolicyViolationError
 from instructor.exceptions import InstructorRetryException
 
 from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
     sleep_and_retry_async,
 )
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/get_llm_client.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
similarity index 92%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/get_llm_client.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
index 7fdf67765..c52cf20ae 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/get_llm_client.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/get_llm_client.py
@@ -4,7 +4,7 @@ from enum import Enum
 
 from cognee.exceptions import InvalidValueError
 from cognee.infrastructure.llm import get_llm_config
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.ollama.adapter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.ollama.adapter import (
     OllamaAPIAdapter,
 )
 
@@ -61,7 +61,7 @@ def get_llm_client():
         if llm_config.llm_api_key is None:
             raise InvalidValueError(message="LLM API key is not set.")
 
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.openai.adapter import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.openai.adapter import (
             OpenAIAdapter,
         )
 
@@ -82,7 +82,7 @@ def get_llm_client():
         if llm_config.llm_api_key is None:
             raise InvalidValueError(message="LLM API key is not set.")
 
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.generic_llm_api import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api import (
             GenericAPIAdapter,
         )
 
@@ -95,7 +95,7 @@ def get_llm_client():
         )
 
     elif provider == LLMProvider.ANTHROPIC:
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.anthropic import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.anthropic import (
             AnthropicAdapter,
         )
 
@@ -105,7 +105,7 @@ def get_llm_client():
         if llm_config.llm_api_key is None:
             raise InvalidValueError(message="LLM API key is not set.")
 
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.generic_llm_api import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.generic_llm_api import (
             GenericAPIAdapter,
         )
 
@@ -124,7 +124,7 @@ def get_llm_client():
         if llm_config.llm_api_key is None:
             raise InvalidValueError(message="LLM API key is not set.")
 
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.gemini import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.gemini import (
             GeminiAdapter,
         )
 
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/llm_interface.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/llm_interface.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/llm_interface.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/llm_interface.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/adapter.py
similarity index 96%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/adapter.py
index 51b2e2e9f..cd7276694 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/ollama/adapter.py
@@ -4,10 +4,10 @@ from typing import Type
 from openai import OpenAI
 from pydantic import BaseModel
 
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
     sleep_and_retry_async,
 )
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/__init__.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/__init__.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/__init__.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/__init__.py
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
similarity index 98%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
index cc02fd63f..7fd0e0d95 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/openai/adapter.py
@@ -9,12 +9,12 @@ from instructor.exceptions import InstructorRetryException
 
 from cognee.exceptions import InvalidValueError
 from cognee.infrastructure.llm.LLMAdapter import LLMAdapter
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.llm_interface import (
     LLMInterface,
 )
 from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError
 from cognee.infrastructure.files.utils.open_data_file import open_data_file
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     rate_limit_async,
     rate_limit_sync,
     sleep_and_retry_async,
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/rate_limiter.py b/cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/rate_limiter.py
similarity index 100%
rename from cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/rate_limiter.py
rename to cognee/infrastructure/llm/structured_output_framework/litellm_instructor/llm/rate_limiter.py
diff --git a/cognee/infrastructure/llm/tokenizer/Gemini/adapter.py b/cognee/infrastructure/llm/tokenizer/Gemini/adapter.py
index 7db010b1d..207d6b53c 100644
--- a/cognee/infrastructure/llm/tokenizer/Gemini/adapter.py
+++ b/cognee/infrastructure/llm/tokenizer/Gemini/adapter.py
@@ -24,7 +24,7 @@ class GeminiTokenizer(TokenizerInterface):
 
         # Get LLM API key from config
         from cognee.infrastructure.databases.vector.embeddings.config import get_embedding_config
-        from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.config import (
+        from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.config import (
             get_llm_config,
         )
 
diff --git a/cognee/infrastructure/llm/utils.py b/cognee/infrastructure/llm/utils.py
index e66743762..7190db654 100644
--- a/cognee/infrastructure/llm/utils.py
+++ b/cognee/infrastructure/llm/utils.py
@@ -1,6 +1,6 @@
 import litellm
 
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.get_llm_client import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.get_llm_client import (
     get_llm_client,
 )
 from cognee.shared.logging_utils import get_logger
diff --git a/cognee/tests/unit/infrastructure/databases/test_rate_limiter.py b/cognee/tests/unit/infrastructure/databases/test_rate_limiter.py
index 2ab8e4dac..62f5b700d 100644
--- a/cognee/tests/unit/infrastructure/databases/test_rate_limiter.py
+++ b/cognee/tests/unit/infrastructure/databases/test_rate_limiter.py
@@ -2,12 +2,15 @@
 
 import pytest
 from unittest.mock import patch
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     llm_rate_limiter,
     rate_limit_async,
     rate_limit_sync,
 )
 
+LLM_RATE_LIMITER = "cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter.llm_rate_limiter"
+GET_LLM_CONFIG = "cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter.get_llm_config"
+
 
 @pytest.fixture(autouse=True)
 def reset_limiter_singleton():
@@ -18,7 +21,7 @@ def reset_limiter_singleton():
 
 def test_rate_limiter_initialization():
     """Test that the rate limiter can be initialized properly."""
-    with patch("cognee.infrastructure.llm.rate_limiter.get_llm_config") as mock_config:
+    with patch(GET_LLM_CONFIG) as mock_config:
         mock_config.return_value.llm_rate_limit_enabled = True
         mock_config.return_value.llm_rate_limit_requests = 10
         mock_config.return_value.llm_rate_limit_interval = 60  # 1 minute
@@ -32,7 +35,7 @@ def test_rate_limiter_initialization():
 
 def test_rate_limiter_disabled():
     """Test that the rate limiter is disabled by default."""
-    with patch("cognee.infrastructure.llm.rate_limiter.get_llm_config") as mock_config:
+    with patch(GET_LLM_CONFIG) as mock_config:
         mock_config.return_value.llm_rate_limit_enabled = False
 
         limiter = llm_rate_limiter()
@@ -43,7 +46,7 @@ def test_rate_limiter_disabled():
 
 def test_rate_limiter_singleton():
     """Test that the rate limiter is a singleton."""
-    with patch("cognee.infrastructure.llm.rate_limiter.get_llm_config") as mock_config:
+    with patch(GET_LLM_CONFIG) as mock_config:
         mock_config.return_value.llm_rate_limit_enabled = True
         mock_config.return_value.llm_rate_limit_requests = 5
         mock_config.return_value.llm_rate_limit_interval = 60
@@ -56,7 +59,7 @@ def test_rate_limiter_singleton():
 
 def test_sync_decorator():
     """Test the sync decorator."""
-    with patch("cognee.infrastructure.llm.rate_limiter.llm_rate_limiter") as mock_limiter_class:
+    with patch(LLM_RATE_LIMITER) as mock_limiter_class:
         mock_limiter = mock_limiter_class.return_value
         mock_limiter.wait_if_needed.return_value = 0
 
@@ -73,7 +76,7 @@ def test_sync_decorator():
 @pytest.mark.asyncio
 async def test_async_decorator():
     """Test the async decorator."""
-    with patch("cognee.infrastructure.llm.rate_limiter.llm_rate_limiter") as mock_limiter_class:
+    with patch(LLM_RATE_LIMITER) as mock_limiter_class:
         mock_limiter = mock_limiter_class.return_value
 
         # Mock an async method with a coroutine
@@ -94,7 +97,7 @@ async def test_async_decorator():
 
 def test_rate_limiting_actual():
     """Test actual rate limiting behavior with a small window."""
-    with patch("cognee.infrastructure.llm.rate_limiter.get_llm_config") as mock_config:
+    with patch(GET_LLM_CONFIG) as mock_config:
         # Configure for 3 requests per minute
         mock_config.return_value.llm_rate_limit_enabled = True
         mock_config.return_value.llm_rate_limit_requests = 3
@@ -115,7 +118,7 @@ def test_rate_limiting_actual():
 
 def test_rate_limit_60_per_minute():
     """Test rate limiting with the default 60 requests per minute limit."""
-    with patch("cognee.infrastructure.llm.rate_limiter.get_llm_config") as mock_config:
+    with patch(GET_LLM_CONFIG) as mock_config:
         # Configure for default values: 60 requests per 60 seconds
         mock_config.return_value.llm_rate_limit_enabled = True
         mock_config.return_value.llm_rate_limit_requests = 60  # 60 requests
diff --git a/cognee/tests/unit/infrastructure/test_rate_limiting_realistic.py b/cognee/tests/unit/infrastructure/test_rate_limiting_realistic.py
index c2953eb80..9ec117bcb 100644
--- a/cognee/tests/unit/infrastructure/test_rate_limiting_realistic.py
+++ b/cognee/tests/unit/infrastructure/test_rate_limiting_realistic.py
@@ -2,7 +2,7 @@ import asyncio
 import os
 from unittest.mock import patch
 from cognee.shared.logging_utils import get_logger
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     llm_rate_limiter,
 )
 from cognee.infrastructure.llm.config import (
diff --git a/cognee/tests/unit/infrastructure/test_rate_limiting_retry.py b/cognee/tests/unit/infrastructure/test_rate_limiting_retry.py
index eef14405e..b2fb23973 100644
--- a/cognee/tests/unit/infrastructure/test_rate_limiting_retry.py
+++ b/cognee/tests/unit/infrastructure/test_rate_limiting_retry.py
@@ -1,7 +1,7 @@
 import time
 import asyncio
 from cognee.shared.logging_utils import get_logger
-from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.litellm_instructor.llm.rate_limiter import (
     sleep_and_retry_sync,
     sleep_and_retry_async,
     is_rate_limit_error,