gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

try few things

Browse source
This commit is contained in:
Vasilije 2024-05-25 14:38:16 +02:00
parent 6a513f25ee
commit fa5028a072
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/infrastructure/llm/get_llm_client.py
View file

@ -12,10 +12,12 @@ class LLMProvider(Enum):
ANTHROPIC = "anthropic" ANTHROPIC = "anthropic"
CUSTOM = "custom" CUSTOM = "custom"
config = Config()
config.load()
def get_llm_client(): def get_llm_client():
"""Get the LLM client based on the configuration using Enums.""" """Get the LLM client based on the configuration using Enums."""
# logging.error(json.dumps(llm_config.to_dict())) # logging.error(json.dumps(llm_config.to_dict()))
provider = LLMProvider(Config.llm_provider) provider = LLMProvider(config.llm_provider)
if provider == LLMProvider.OPENAI: if provider == LLMProvider.OPENAI:
from .openai.adapter import OpenAIAdapter from .openai.adapter import OpenAIAdapter