diff --git a/cognee/infrastructure/llm/generic_llm_api/adapter.py b/cognee/infrastructure/llm/generic_llm_api/adapter.py
index c1cc40e38..b97d2342c 100644
--- a/cognee/infrastructure/llm/generic_llm_api/adapter.py
+++ b/cognee/infrastructure/llm/generic_llm_api/adapter.py
@@ -10,8 +10,13 @@ from litellm.exceptions import ContentPolicyViolationError
 from instructor.exceptions import InstructorRetryException
 
 from cognee.infrastructure.llm.exceptions import ContentPolicyFilterError
-from cognee.infrastructure.llm.llm_interface import LLMInterface
-from cognee.infrastructure.llm.rate_limiter import rate_limit_async, sleep_and_retry_async
+from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+    LLMInterface,
+)
+from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
+    rate_limit_async,
+    sleep_and_retry_async,
+)
 
 
 class GenericAPIAdapter(LLMInterface):
diff --git a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py
index 9bcf867e5..51b2e2e9f 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/ollama/adapter.py
@@ -4,8 +4,10 @@ from typing import Type
 from openai import OpenAI
 from pydantic import BaseModel
 
-from cognee.infrastructure.llm.llm_interface import LLMInterface
-from cognee.infrastructure.llm.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.llm_interface import (
+    LLMInterface,
+)
+from cognee.infrastructure.llm.structured_output_framework.llitellm_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/adapter.py b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py
index 066ad0157..0bf527a62 100644
--- a/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py
+++ b/cognee/infrastructure/llm/structured_output_framework/llitellm_instructor/llm/openai/adapter.py
@@ -9,10 +9,12 @@ from instructor.exceptions import InstructorRetryException
 
 from cognee.exceptions import InvalidValueError
 from cognee.infrastructure.llm.prompts import read_query_prompt
-from cognee.infrastructure.llm.llm_interface import LLMInterface
+from cognee.infrastructure.llm.structured_output_framework.llitellm_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.rate_limiter import (
+from cognee.infrastructure.llm.structured_output_framework.llitellm_instructor.llm.rate_limiter import (
     rate_limit_async,
     rate_limit_sync,
     sleep_and_retry_async,