From 6a513f25ee49b3a960084954389e315892214aa0 Mon Sep 17 00:00:00 2001
From: Vasilije <8619304+Vasilije1990@users.noreply.github.com>
Date: Sat, 25 May 2024 14:37:42 +0200
Subject: [PATCH] try few things

---
 cognee/infrastructure/llm/get_llm_client.py | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/cognee/infrastructure/llm/get_llm_client.py b/cognee/infrastructure/llm/get_llm_client.py
index 3842084bd..1aadd12bb 100644
--- a/cognee/infrastructure/llm/get_llm_client.py
+++ b/cognee/infrastructure/llm/get_llm_client.py
@@ -2,8 +2,9 @@
 from enum import Enum
 import json
 import logging
-from cognee.infrastructure.llm import llm_config
+# from cognee.infrastructure.llm import llm_config
 
+from cognee.config import Config
 # Define an Enum for LLM Providers
 class LLMProvider(Enum):
     OPENAI = "openai"
@@ -13,8 +14,8 @@ class LLMProvider(Enum):
 
 def get_llm_client():
     """Get the LLM client based on the configuration using Enums."""
-    logging.error(json.dumps(llm_config.to_dict()))
-    provider = LLMProvider(llm_config.llm_provider)
+    # logging.error(json.dumps(llm_config.to_dict()))
+    provider = LLMProvider(Config.llm_provider)
 
     if provider == LLMProvider.OPENAI:
         from .openai.adapter import OpenAIAdapter