diff --git a/cognee/infrastructure/llm/__init__.py b/cognee/infrastructure/llm/__init__.py
index 8bc1d2082..7fb3be736 100644
--- a/cognee/infrastructure/llm/__init__.py
+++ b/cognee/infrastructure/llm/__init__.py
@@ -1 +1 @@
-from .config import llm_config
+from .config import get_llm_config